#!/usr/bin/env python3

import argparse, sys, shutil, os

sys.path.append(os.path.dirname(__file__) + '/lib')

import tamarin, system, rkt

def configure_args_parser():

    profile_names = tamarin.get_available_profile_names()

    parser = argparse.ArgumentParser(description="Generate packages for various GNU/Linux distribution")

    # Define available/required arguments and flags
    parser.add_argument("project_path", help="The path to your project to package")
    parser.add_argument("-p", "--profile", help="The profile to use to package this project (default: debian)", choices=profile_names, default='debian')

    return parser

if __name__ == "__main__":

    parser = configure_args_parser()
    args = parser.parse_args()

    profile = tamarin.load_profile(args.profile)

    workspace = tamarin.get_workspace_dir()
    workspace_tmp = tamarin.get_workspace_subdir('tmp')

    local_rkt_dir = tamarin.get_workspace_subdir('rkt')
    if not system.which('rkt', local_rkt_dir):
        # Download and extract rkt
        rkt_archive_path = tamarin.download_rkt()
        system.extract_tar(rkt_archive_path, workspace_tmp)
        rkt_archive_dir = tamarin.get_rkt_achive_dest_dir()
        shutil.rmtree(local_rkt_dir, ignore_errors=True)
        os.rename(rkt_archive_dir, local_rkt_dir)

    local_acbuild_dir = tamarin.get_workspace_subdir('acbuild')
    if not system.which('acbuild', local_acbuild_dir):
        # Download and extract acbuild
        acbuild_archive_path = tamarin.download_acbuild()
        system.extract_tar(acbuild_archive_path, workspace_tmp)
        acbuild_archive_dir = tamarin.get_acbuild_achive_dest_dir()
        shutil.rmtree(local_acbuild_dir, ignore_errors=True)
        os.rename(acbuild_archive_dir, local_acbuild_dir)

    acbuild_workspace = tamarin.get_acbuild_workspace_dir()
    pid = os.getpid()
    image_name = "image_{:d}.aci".format(pid)
    image_path = os.path.join(os.sep, acbuild_workspace, image_name)
    acbuild_flags = ["--work-path", acbuild_workspace]

    rkt_flags = ["--dir={:s}".format(workspace_tmp)]

    base_image = profile['profile']['default_image']
    # If the base image is Docker-based, preload it and get its name from the store
    if base_image.startswith('docker://'):
        rkt.run([
            "fetch",
            "--insecure-options=image",
            base_image
        ] + rkt_flags)

    name_pattern = base_image.split('/')[-1] + '$'
    image = rkt.find_image_by_name(name_pattern, rkt_flags=rkt_flags)


    base_aci_name = image['id'][:21]+'.aci'
    base_aci_file = workspace_tmp+'/'+base_aci_name

    if not os.path.isfile(base_aci_file):
        rkt.export_image(image['id'], workspace_tmp+'/'+base_aci_name, rkt_flags=rkt_flags);

    # Start building image
    tamarin.run_acbuild(acbuild_flags+["begin", base_aci_file])
    tamarin.run_acbuild(acbuild_flags+["set-name", "image_{:d}".format(pid)])
    tamarin.run_acbuild(acbuild_flags+["set-exec", "--", "/bin/sh", "-c", "uname -a"])
    tamarin.run_acbuild(acbuild_flags+["write", image_path])
    tamarin.run_acbuild(acbuild_flags+["end"])

    # rkt run --insecure-options=image ./nginx.aci --volume html,kind=host,source=/path/to/test --net=host
    rkt.run([
        "run",
        "--insecure-options=image",
        image_path, "--net=host"
    ] + rkt_flags, as_root=True)

    # Cleanup

    rkt.run([
        "gc",
        "--grace-period=0"
    ] + rkt_flags, as_root=True)

    rkt.run([
        "image",
        "gc"
    ] + rkt_flags, as_root=True)

    shutil.rmtree(acbuild_workspace, ignore_errors=True)