#!/usr/bin/env python3

import argparse
import subprocess
import os

def parse_args():
    parser = argparse.ArgumentParser(description='frappe_docker common CI elements', add_help=True)

    parser.add_argument('service', action='store', type=str, help='Name of the service to build: "erpnext" or "frappe"')
    parser.add_argument('-o', '--tag-only', required=False, action='store_true', dest='tag_only', help='Only tag an image and push it.')

    image_type = parser.add_mutually_exclusive_group(required=True)
    image_type.add_argument('-a', '--nginx', action='store_const', dest='image_type', const='nginx', help='Build the nginx + static assets image')
    image_type.add_argument('-s', '--socketio', action='store_const', dest='image_type', const='socketio', help='Build the frappe-socketio image')
    image_type.add_argument('-w', '--worker', action='store_const', dest='image_type', const='worker', help='Build the python environment image')

    tag_type = parser.add_mutually_exclusive_group(required=True)
    tag_type.add_argument('-g', '--git-version', action='store', type=str, dest='version', help='The version number of service (i.e. "11", "12", etc.)')
    tag_type.add_argument('-t', '--tag', action='store', type=str, dest='tag', help='The image tag (i.e. erpnext-worker:$TAG )')

    args = parser.parse_args()
    return args


def git_version(service, version, branch):
    print(f'Pulling {service} v{version}')
    subprocess.run(f'git clone https://github.com/frappe/{service} --branch {branch}', shell=True)
    cd = os.getcwd()
    os.chdir(os.getcwd() + f'/{service}')
    subprocess.run('git fetch --tags', shell=True)

    # XX-beta becomes XX for tags search
    version = version.split('-')[0]

    version_tag = subprocess.check_output(f'git tag --list --sort=-version:refname "v{version}*" | sed -n 1p | sed -e \'s#.*@\(\)#\\1#\'', shell=True).strip().decode('ascii')
    os.chdir(cd)
    return version_tag


def build(service, tag, image, branch):
    build_args = f'--build-arg GIT_BRANCH={branch}'
    if service == 'nginx' and branch == 'version-11':
        build_args += f' --build-arg NODE_IMAGE_TAG=10-prod'

    print(f'Building {service} {image} image')
    subprocess.run(f'docker build {build_args} -t {service}-{image} -f build/{service}-{image}/Dockerfile .', shell=True)
    tag_and_push(f'{service}-{image}', tag)


def tag_and_push(image_name, tag):
    print(f'Tagging {image_name} as "{tag}" and pushing')
    subprocess.run(f'docker tag {image_name} frappe/{image_name}:{tag}', shell=True)
    subprocess.run(f'docker push frappe/{image_name}:{tag}', shell=True)


def main():
    args = parse_args()
    tag = args.tag
    branch = 'develop'

    if args.version:
        branch = 'version-' + args.version
        tag = git_version(args.service, args.version, branch)

    if args.tag_only:
        tag_and_push(f'{args.service}-{args.image_type}', tag)
    else:
        build(args.service, tag, args.image_type, branch)

if __name__ == "__main__":
    main()