import glob
import os
import pathlib
import re
import shutil
from SCons.Script import *


_BUILT_STAMPFILE = '.spp_built'

def _git_cook(env: Environment, repo: dict) -> dict:
    if os.name == 'nt':
        def run_cmd(args):
            if env.Execute(' '.join([str(s) for s in args])):
                Exit(1)
        install_dir = os.path.join(repo['checkout_root'], 'install')
        stamp_file = pathlib.Path(install_dir, _BUILT_STAMPFILE)

        include_path = os.path.join(install_dir, 'include')
        bin_path = os.path.join(install_dir, 'bin')
        lib_path = os.path.join(install_dir, 'lib')
        if not stamp_file.exists():
            os.makedirs(include_path, exist_ok=True)
            os.makedirs(bin_path, exist_ok=True)
            os.makedirs(lib_path, exist_ok=True)

            old_cwd = os.getcwd()
            os.chdir(repo['checkout_root'])
            run_cmd(['nmake', '/f', 'Makefile.msc', 'sqlite3.dll'])
            os.chdir(old_cwd)

            include_files = ['sqlite3.h']
            bin_files = ['sqlite3.dll']
            lib_files = ['sqlite3.def', 'sqlite3.lib']
            for file in include_files:
                shutil.copy(os.path.join(repo['checkout_root'], file), include_path)
            for file in bin_files:
                shutil.copy(os.path.join(repo['checkout_root'], file), bin_path)
            for file in lib_files:
                shutil.copy(os.path.join(repo['checkout_root'], file), lib_path)
            stamp_file.touch()
        return {
            'CPPPATH': [include_path],
            'LIBPATH': [lib_path],
            'LIBS': ['sqlite3']
        }
    else:
        checkout_root = repo['checkout_root']
        build_result = env.AutotoolsProject(checkout_root, skip_steps=('install',))

        # the install script seems borked, do the install manually
        include_dir = os.path.join(build_result['install_dir'], 'include')
        lib_dir = os.path.join(build_result['install_dir'], 'lib')
        os.makedirs(include_dir, exist_ok=True)
        os.makedirs(lib_dir, exist_ok=True)

        for ext in ('*.a', '*.so'):
            lib_files = glob.glob(os.path.join(build_result['build_dir'], ext))
            for lib_file in lib_files:
                out_path = os.path.join(lib_dir, os.path.basename(lib_file))
                if not os.path.exists(out_path):
                    shutil.copy(lib_file, out_path)
        for h_file in ('sqlite3.h',):
            out_path = os.path.join(include_dir, h_file)
            if not os.path.exists(out_path):
                in_path = os.path.join(build_result['build_dir'], h_file)
                shutil.copy(in_path, out_path)

        return {
            'CPPPATH': build_result['CPPPATH'],
            'LIBS': [env.FindLib('sqlite3', paths=build_result['LIBPATH'])]
        }

env.GitRecipe(
    globals = globals(),
    repo_name = 'sqlite',
    repo_url = 'https://github.com/sqlite/sqlite.git',
    tag_pattern = re.compile(r'^version-([0-9]+)\.([0-9]+)\.([0-9]+)$'),
    tag_fn = lambda version: f'version-{version[0]}.{version[1]}.{version[2]}',
    cook_fn = _git_cook
)