import sys, platform
if sys.platform != 'win32':
    print("This script is for Windows only! Use configure.sh instead.")
    exit(1)
if sys.version_info.major != 3 or sys.version_info.minor != 8:
    print("You neeed to use Python 3.8.x (due to the requirements.txt). You are using %s instead." % platform.python_version())
    exit(1)

import argparse
import subprocess
import os
import stat
import urllib.request
import shutil 
import zipfile

from venv import EnvBuilder

parser = argparse.ArgumentParser(description='ODM Windows Configure Script')
parser.add_argument('action',
                type=str,
                choices=["build", "clean", "dist", "vcpkg_export"],
                help='Action: %(choices)s')
parser.add_argument('--build-vcpkg',
                    type=bool,
                    help='Build VCPKG environment from scratch instead of downloading prebuilt one.')
parser.add_argument('--vcpkg-archive-url',
                    type=str,
                    default='https://github.com/OpenDroneMap/windows-deps/releases/download/2.5.0/vcpkg-export-250.zip',
                    required=False,
                    help='Path to VCPKG export archive')
args = parser.parse_args()

def run(cmd, cwd=os.getcwd()):
    env = os.environ.copy()
    print(cmd)
    p = subprocess.Popen(cmd, shell=True, env=env, cwd=cwd)
    retcode = p.wait()
    if retcode != 0:
        raise Exception("Command returned %s" % retcode)

# https://izziswift.com/shutil-rmtree-fails-on-windows-with-access-is-denied/
def rmtree(top):
    for root, dirs, files in os.walk(top, topdown=False):
        for name in files:
            filename = os.path.join(root, name)
            os.chmod(filename, stat.S_IWUSR)
            os.remove(filename)
        for name in dirs:
            os.rmdir(os.path.join(root, name))
    os.rmdir(top)      

def vcpkg_requirements():
    with open("vcpkg-requirements.txt") as f:
        pckgs = list(filter(lambda l: len(l) > 0, map(str.strip, f.read().split("\n"))))
    return pckgs
        

def build():
    # Create python virtual env
    if not os.path.isdir("venv"):
        print("Creating virtual env --> venv/")
        ebuilder = EnvBuilder(with_pip=True)
        ebuilder.create("venv")

    run("venv\\Scripts\\pip install --ignore-installed -r requirements.txt")
    
    # Download / build VCPKG environment
    if not os.path.isdir("vcpkg"):
        if args.build_vcpkg:
            print("TODO")
            # git clone vcpkg repo
            # bootstrap
            # install requirements

        else:
            if not os.path.exists("vcpkg-env.zip"):
                print("Downloading %s" % args.vcpkg_archive_url)
                with urllib.request.urlopen(args.vcpkg_archive_url) as response, open( "vcpkg-env.zip", 'wb') as out_file:
                    shutil.copyfileobj(response, out_file)
            if not os.path.exists("vcpkg"):
                print("Extracting vcpkg-env.zip --> vcpkg/")
                with zipfile.ZipFile("vcpkg-env.zip") as z:
                    top_dir = z.namelist()[0]
                    z.extractall(".")

                    if os.path.exists(top_dir):
                        os.rename(top_dir, "vcpkg")
                    else:
                        print("Warning! Something looks wrong in the VCPKG archive... check the vcpkg/ directory.")

    if not os.path.exists(os.path.join("SuperBuild", "build")) or not os.path.exists(os.path.join("SuperBuild", "install")):
        print("Compiling SuperBuild")
        
        build_dir = os.path.join("SuperBuild", "build")
        if not os.path.isdir(build_dir):
            os.mkdir(build_dir)

        toolchain_file = os.path.join(os.getcwd(), "vcpkg", "scripts", "buildsystems", "vcpkg.cmake")
        run("cmake .. -DCMAKE_TOOLCHAIN_FILE=\"%s\"" % toolchain_file,  cwd=build_dir)
        run("cmake --build . --config Release", cwd=build_dir)

def vcpkg_export():
    if not os.path.exists("vcpkg"):
        print("vcpkg directory does not exist. Did you build the environment?")
        exit(1)

    pkgs = vcpkg_requirements()
    out = "vcpkg-export-%s" % odm_version().replace(".", "")
    run("vcpkg\\vcpkg export %s --output=%s --zip" % (" ".join(pkgs), out))

def odm_version():
    with open("VERSION") as f:
        return f.read().split("\n")[0].strip()

def safe_remove(path):
    if os.path.isdir(path):
        rmtree(path)
    elif os.path.isfile(path):
        os.remove(path)

def clean():
    safe_remove("vcpkg-download.zip")
    safe_remove("vcpkg")
    safe_remove("venv")
    safe_remove(os.path.join("SuperBuild", "build"))
    safe_remove(os.path.join("SuperBuild", "download"))
    safe_remove(os.path.join("SuperBuild", "src"))
    safe_remove(os.path.join("SuperBuild", "install"))

def dist():
    # D:\Program Files (x86)\Inno Setup 6
    run("compil32 /cc \"innosetup.iss\")

if args.action == 'build':
    build()
elif args.action == 'vcpkg_export':
    vcpkg_export()
elif args.action == 'clean':
    clean()
else:
    args.print_help()
    exit(1)