import argparse
import os.path
import sys
from pathlib import Path
from typing import List, Union, Optional

from jinja2 import Environment, PackageLoader, select_autoescape

from ee.kicad import read_schematics
from ee.project import Project


def ninja_path_filter(s: Union[Path, str, List[str]]) -> str:
    if isinstance(s, Path):
        s = str(s)

    if isinstance(s, str):
        return s. \
            replace("$", "$$"). \
            replace(" ", "$ ")
    elif isinstance(s, list):

        if len(s) == 1:
            return ninja_path_filter(s[0])

        return "$\n" + " $\n".join(["  " + ninja_path_filter(path) for path in s])
    else:
        raise Exception("Unsupported argument type: {}".format(type(s)))


def parent_dir_filter(s: Union[str, Path]) -> Path:
    return Path(s).parent


def basename_filter(s: Union[str, Path]) -> str:
    return os.path.basename(str(s))


def noext_filter(s: Union[str, Path]) -> str:
    return os.path.splitext(os.path.basename(str(s)))[0]


def generate(project: Project, sch_path: Path, kicad_bom_strategy: Optional[str]):
    def _create_env():
        e = Environment(
            loader=PackageLoader(__name__, "templates"),
            autoescape=select_autoescape(["html", "xml"]),
            keep_trailing_newline=True,
        )
        e.filters["ninja_path"] = ninja_path_filter
        e.filters["parent_dir"] = parent_dir_filter
        e.filters["basename"] = basename_filter
        e.filters["noext"] = noext_filter
        return e

    gerber_zip = "prod/gerber.zip"

    sch = read_schematics(str(sch_path))

    sch_files = sorted([s.path for s in sch.schematics])

    part_dbs = []
    params = {
        "ee": "{} -m ee".format(os.path.relpath(sys.executable, Path("."))),
        "project": project,
        "sch": sch_path,
        "sch_files": sch_files, "kicad_bom_strategy": kicad_bom_strategy,
        "pcb": str(sch_path).replace(".sch", ".kicad_pcb"),
        "part_dbs": part_dbs,
    }

    # TODO: read from config
    distributors = ["digikey"]
    params["distributors"] = distributors

    if gerber_zip is not None:
        params["gerber_zip"] = gerber_zip

    # ee_dir = sch_path.parent / "ee"
    ee_dir = Path(".")
    build_ninja = ee_dir / "build.ninja"

    parts_yaml_files = [path for path in ee_dir.iterdir() if str(path).endswith("-parts.yaml")]
    params["parts_yaml_files"] = parts_yaml_files

    # Local part databases first
    part_dbs.extend([parent_dir_filter(p) / noext_filter(p) for p in parts_yaml_files])
    part_dbs.extend([Path("ee") / d / "normalized" for d in distributors])

    with build_ninja.open("w") as f:
        env = _create_env()
        template = env.get_template("build.ninja.j2")
        f.write(template.render(**params))


parser = argparse.ArgumentParser()

parser.add_argument("--sch",
                    required=True,
                    metavar="FILE")

parser.add_argument("--kicad-bom-strategy",
                    required=False,
                    metavar="PY CALLABLE")

args = parser.parse_args()

project = Project.load()
generate(project, Path(args.sch), args.kicad_bom_strategy)