#!/usr/bin/env python3

import sys
import os
import glob
from os import path

# base dir of sqlx workspace
dir_workspace = path.dirname(path.dirname(path.realpath(__file__)))

# dir of tests
dir_tests = path.join(dir_workspace, "tests")

# extend import path to tests/
sys.path.append(dir_tests)

import subprocess
import time
import argparse
import runpy
from docker import start_database

parser = argparse.ArgumentParser()
parser.add_argument("-p", "--project")
parser.add_argument("-l", "--list-projects", action="store_true")

argv, unknown = parser.parse_known_args()


def run(command, env=None, cwd=None, display=None):
    if display:
        print(f"\x1b[93m $ {display}\x1b[0m")

    else:
        print(f"\x1b[93m $ {command}\x1b[0m")

    res = subprocess.run(
        command.split(" "),
        env=os.environ | env,
        cwd=cwd,
    )

    if res.returncode != 0:
        sys.exit(res.returncode)


def sqlx(command, url, cwd=None):
    run(f"cargo --quiet run -p sqlx-cli --bin sqlx -- {command} --database-url {url}", cwd=cwd, env={},
        display=f"sqlx {command}")


def project(name, database=None, driver=None):
    if argv.list_projects:
        print(f"{name}")
        return

    if argv.project and name != argv.project:
        return

    print(f"\x1b[2m # {name}\x1b[0m")

    env = {}

    cwd = path.join(dir_workspace, "examples", name)

    if database is not None:
        database_url = start_database(driver, database, cwd=cwd)
        env["DATABASE_URL"] = database_url

        # show the database url
        print(f"\x1b[94m @ {database_url}\x1b[0m")

        # database drop (if exists)
        sqlx("db drop -y", database_url, cwd=cwd)

        # database create
        sqlx("db create", database_url, cwd=cwd)

        if path.exists(path.join(name, "setup.py")):
            setup = runpy.run_path(path.join(name, "setup.py"))
            res = setup["setup"](database_url=database_url,cwd=cwd,sqlx=sqlx)

            if type(res) is dict:
                env |= res

        # migrate
        if path.exists(path.join(name, "migrations")) or path.exists(path.join(name, "src/migrations")):
            sqlx("migrate run", database_url, cwd=cwd)

    # check
    run("cargo check", cwd=cwd, env=env)


# MySQL
for example_path in sorted(glob.iglob("mysql/*")):
    if not path.isdir(example_path):
        continue

    example = path.relpath(example_path, start="mysql")
    project(example_path, driver="mysql_8", database=f"example-{example}")

# Postgres
for example_path in sorted(glob.iglob("postgres/*")):
    if not path.isdir(example_path):
        continue

    example = path.relpath(example_path, start="postgres")
    project(example_path, driver="postgres_17", database=f"example-{example}")

# SQLite
for example_path in sorted(glob.iglob("sqlite/*")):
    if not path.isdir(example_path):
        continue

    example = path.relpath(example_path, start="sqlite")
    project(example_path, driver="sqlite", database=f"example-{example}.db")