from pathlib import Path
from dataclasses import dataclass, field
from codegen.constants import WAPP_SRC_ROOT
from codegen.utils import load_func_body_from_file
from codegen.datatypes import (
    CDataType,
    CStruct,
    CFunc,
    CHeader,
    CSource,
    CArg,
    CType,
    CPointer,
    CPointerType,
    CQualifier,
    CInclude,
    get_datatype_string,
)


@dataclass
class DblListData:
    out_dir: Path
    hdr_includes: list[CInclude] = field(default_factory=list)
    src_includes: list[CInclude] = field(default_factory=list)
    hdr_decl_types: list[CStruct] = field(default_factory=list)
    src_decl_types: list[CStruct] = field(default_factory=list)


def make_dbl_list(user_datatypes: dict[CDataType, DblListData] = {}):
    def __format_func_body(filename: Path, type_string: str):
        return load_func_body_from_file(filename).format(
            T=type_string,
            Ttitle=type_string.title(),
            Tupper=type_string.upper(),
            Tlower=type_string.lower(),
        )

    common_local_include_files = [
        (WAPP_SRC_ROOT / "common" / "aliases" / "aliases.h")
    ]
    common_includes: list[CInclude] = []

    common_decl_types: list[CStruct] = []

    datatypes: dict[CDataType, DblListData] = {
        "Str8": DblListData(
            hdr_decl_types=[
                CStruct(name="str8", cargs=[], typedef_name="Str8"),
            ],
            out_dir=WAPP_SRC_ROOT / "core" / "strings" / "str8",
        ),
    }
    datatypes.update(user_datatypes)

    snippets_dir = Path(__file__).parent / "snippets"

    for _type, dbl_list_data in datatypes.items():
        type_string = get_datatype_string(_type)

        node = CStruct(
            name=f"{type_string.title()}Node",
            cargs=[
                CArg(name="item", _type=type_string, pointer=CPointer(_type=CPointerType.SINGLE)),
            ],
        )
        node.cargs.extend([
            CArg(name="prev", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
            CArg(name="next", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
        ])
        
        dl_list = CStruct(
            name=f"{type_string.title()}List",
            cargs=[
                CArg(name="first", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
                CArg(name="last", _type=node, pointer=CPointer(_type=CPointerType.SINGLE)),
                CArg(name="node_count", _type=CType.U64),
            ],
        )

        node_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_node",
            ret_type=node,
            args=[
                CArg(name="item", _type=type_string, pointer=CPointer(CPointerType.SINGLE)),
            ],
            body=__format_func_body(snippets_dir / "list_node", type_string),
        )

        get_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_get",
            ret_type=node,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE), qualifier=CQualifier.CONST),
                CArg(name="index", _type=CType.U64),
            ],
            body=__format_func_body(snippets_dir / "list_get", type_string),
            pointer=CPointer(CPointerType.SINGLE),
        )

        push_front_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_push_front",
            ret_type=CType.VOID,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
            ],
            body=__format_func_body(snippets_dir / "list_push_front", type_string),
        )

        push_back_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_push_back",
            ret_type=CType.VOID,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
            ],
            body=__format_func_body(snippets_dir / "list_push_back", type_string),
        )

        insert_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_insert",
            ret_type=CType.VOID,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
                CArg(name="index", _type=CType.U64),
            ],
            body=__format_func_body(snippets_dir / "list_insert", type_string),
        )

        pop_front_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_pop_front",
            ret_type=node,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
            ],
            body=__format_func_body(snippets_dir / "list_pop_front", type_string),
            pointer=CPointer(CPointerType.SINGLE),
        )

        pop_back_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_pop_back",
            ret_type=node,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
            ],
            body=__format_func_body(snippets_dir / "list_pop_back", type_string),
            pointer=CPointer(CPointerType.SINGLE),
        )

        remove_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_remove",
            ret_type=node,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
                CArg(name="index", _type=CType.U64),
            ],
            body=__format_func_body(snippets_dir / "list_remove", type_string),
            pointer=CPointer(CPointerType.SINGLE),
        )

        empty_func = CFunc(
            name=f"wapp_{type_string.lower()}_list_empty",
            ret_type=CType.VOID,
            args=[
                CArg(name="list", _type=dl_list, pointer=CPointer(CPointerType.SINGLE)),
            ],
            body=__format_func_body(snippets_dir / "list_empty", type_string),
        )

        node_to_list_func = CFunc(
            name=f"{type_string.lower()}_node_to_list",
            ret_type=dl_list,
            args=[
                CArg(name="node", _type=node, pointer=CPointer(CPointerType.SINGLE)),
            ],
            body=__format_func_body(snippets_dir / "node_to_list", type_string),
            qualifiers=[CQualifier.INTERNAL],
        )

        header = CHeader(
            name=f"{type_string.lower()}_list",
            decl_types=common_decl_types + dbl_list_data.hdr_decl_types,
            includes=[],
            types=[node, dl_list],
            funcs=[
                node_func,
                get_func,
                push_front_func,
                push_back_func,
                insert_func,
                pop_front_func,
                pop_back_func,
                remove_func,
                empty_func,
           ]
        )

        source = CSource(
            name=header.name,
            decl_types=common_decl_types + dbl_list_data.src_decl_types,
            includes=[CInclude(header, local=True, same_dir=True), CInclude(header="stddef.h")],
            internal_funcs=[node_to_list_func],
            funcs=header.funcs
        )

        if len(common_includes) > 0:
            header.includes.extend(common_includes)
            source.includes.extend(common_includes)

        for include_file in common_local_include_files:
            include = CInclude(header=str(include_file.relative_to(dbl_list_data.out_dir, walk_up=True)), local=True)
            header.includes.append(include)
            source.includes.append(include)

        if len(dbl_list_data.hdr_includes) > 0:
            header.includes.extend(dbl_list_data.hdr_includes)

        if len(dbl_list_data.src_includes) > 0:
            source.includes.extend(dbl_list_data.src_includes)

        header.save(dbl_list_data.out_dir)
        source.save(dbl_list_data.out_dir)