from pathlib import Path

import argparse

import fontmake.instantiator
import fontTools.designspaceLib
import ufo2ft
import ufoLib2
import vttLib

INPUT_DIR = Path("sources")
OUTPUT_DIR = Path("build")
VTT_DATA_FILE = INPUT_DIR / "vtt_data" / "CascadiaCode.ttx"


def step_set_font_name(n):
    def _set(instance):
        instance.info.familyName = n
        # We have to change the style map family name because that's what
        # Windows uses to map Bold/Regular/Medium/etc. fonts
        instance.info.styleMapFamilyName = n

    return _set


def step_merge_glyphs_from_ufo(path):
    def _merge(instance):
        ufo = ufoLib2.Font.open(path)
        print(f"[{instance.info.familyName}] Merging {path}")
        for glyph in ufo.glyphOrder:
            if glyph not in instance.glyphOrder:
                instance.addGlyph(ufo[glyph])

    return _merge


def step_set_feature_file(n):
    fea = n.read_text()

    def _set(instance):
        instance.features.text = fea

    return _set


def build_font_instance(generator, instance_descriptor, *steps):
    instance = generator.generate_instance(instance_descriptor)

    for step in steps:
        step(instance)

    setattr(instance.info, "openTypeOS2Panose", [2, 11, 6, 9, 2, 0, 0, 2, 0, 4])

    instance.info.openTypeGaspRangeRecords =[
        {
            "rangeMaxPPEM" : 9,
            "rangeGaspBehavior" : [1,3]
        },
        {
            "rangeMaxPPEM" : 50,
            "rangeGaspBehavior" : [0,1,2,3]
        },
        {
            "rangeMaxPPEM" : 65535,
            "rangeGaspBehavior" : [1,3]
        },
    ]

    familyName = instance.info.familyName
    file_name = f"{familyName}.ttf".replace(" ", "")
    file_path = OUTPUT_DIR / file_name

    print(f"[{familyName}] Compiling")
    instance_font = ufo2ft.compileTTF(instance, removeOverlaps=True, inplace=True)

    print(f"[{familyName}] Merging VTT")
    vttLib.transfer.merge_from_file(instance_font, VTT_DATA_FILE)

    print(f"[{familyName}] Saving")
    instance_font.save(file_path)

    print(f"[{familyName}] Done: {file_path}")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="build some fonts")
    parser.add_argument("-N", "--no-nerdfonts", default=False, action="store_true")
    parser.add_argument("-P", "--no-powerline", default=False, action="store_true")
    parser.add_argument("-M", "--no-mono", default=False, action="store_true")
    args = parser.parse_args()

    # 1. Load Designspace and filter out instances that are marked as non-exportable.
    designspace = fontTools.designspaceLib.DesignSpaceDocument.fromfile(
        INPUT_DIR / "CascadiaCode-Regular.designspace"
    )
    designspace.instances = [
        s
        for s in designspace.instances
        if s.lib.get("com.schriftgestaltung.export", True)
    ]

    # 2. Prepare masters.
    generator = fontmake.instantiator.Instantiator.from_designspace(designspace)

    OUTPUT_DIR.mkdir(exist_ok=True)

    step_remove_ligatures = step_set_feature_file(
        INPUT_DIR / "features" / "features.fea"
    )
    step_merge_pl = step_merge_glyphs_from_ufo(
        INPUT_DIR / "nerdfonts" / "NerdfontsPL.ufo"
    )

    nf_path = INPUT_DIR / "nerdfonts" / "NerdfontsNF.ufo"
    if not nf_path.exists():
        args.no_nerdfonts = True  # No NF = don't try to build those fonts.

    step_merge_nf = None
    if not args.no_nerdfonts:
        step_merge_nf = step_merge_glyphs_from_ufo(
            INPUT_DIR / "nerdfonts" / "NerdfontsNF.ufo"
        )

    for instance_descriptor in designspace.instances:

        build_font_instance(generator, instance_descriptor)

        if not args.no_mono:
            build_font_instance(
                generator,
                instance_descriptor,
                step_set_font_name("Cascadia Mono"),
                step_remove_ligatures,
            )

        if not args.no_powerline:
            build_font_instance(
                generator,
                instance_descriptor,
                step_set_font_name("Cascadia Code PL"),
                step_merge_pl,
            )

            if not args.no_mono:
                build_font_instance(
                    generator,
                    instance_descriptor,
                    step_set_font_name("Cascadia Mono PL"),
                    step_remove_ligatures,
                    step_merge_pl,
                )

        if not args.no_nerdfonts:
            build_font_instance(
                generator,
                instance_descriptor,
                step_set_font_name("Cascadia Code NF"),
                step_merge_nf,
            )

            if not args.no_mono:
                build_font_instance(
                    generator,
                    instance_descriptor,
                    step_set_font_name("Cascadia Mono NF"),
                    step_remove_ligatures,
                    step_merge_nf,
                )

        print("All done")
        print("*** *** *** *** *** *** *** *** *** *** *** *** *** *** *** *** *** ***")