import os
import importlib
import inspect
import subprocess
import time
from typing import TypedDict

def get_defined_functions_and_classes(module):
    functions = []
    classes = []

    for name, obj in inspect.getmembers(module):
        if inspect.isfunction(obj) and obj.__module__ == module.__name__:
            functions.append(name)
        elif inspect.isclass(obj) and obj.__module__ == module.__name__:
            classes.append(name)

    return functions, classes


class IGenerateSphinx(TypedDict):
    module_path: str
    module_name: str
    functions: list[str]
    classes: list[str]

def generate_sphinx_output(data: list[IGenerateSphinx]):
    output = []
    output.append("API Reference\n===============\n")
    tmp = {}
    for i in hierarchy.values():
        tmp[i] = []
        #tmp[i].append(f"{i}\n{'-'*len(i)}")
    
    tmp["__OTHER"] = []

    def add_data(_data: IGenerateSphinx, add_to: list):
        if _data['functions']:
            for function in _data['functions']:
                add_to.append(f"\n{function}\n" + "~" * len(function))
                add_to.append(f".. autofunction:: {_data['module_name']}.{function}")
        
        if _data['classes']:
            for class_name in _data['classes']:
                add_to.append(f"\n{class_name}\n" + "~" * len(class_name))
                full_class_name = f"{_data['module_name']}.{class_name}"
                add_to.append(f".. attributetable:: {full_class_name}")
                add_to.append(f".. autoclass:: {full_class_name}\n    :members:")

    for i in data:
        add_to = "__OTHER"
        for category in hierarchy.keys():
            if category in  i['module_path']:
                add_to = hierarchy[category]
        add_data(i, tmp[add_to])

    for k,v in tmp.items():
        output.append(f"{k}\n{'-'*len(k)}")
        output.append("\n\n".join(v))
    return '\n'.join(output)


def generate_documentation(base_path, output_file, hierarchy, exclude_files=[]):
    documentation = []
    items: list[IGenerateSphinx] = []

    for file in [{'filename': file, 'root': root} for root, dirs, files in os.walk(base_path) if root != "__pycache__" for file in files]:
        if file['filename'].endswith('.py') and file['filename'] not in exclude_files:

            module_path = os.path.join(file['root'], file['filename'])
            module_import_path = module_path.replace(os.path.sep, '.').replace('.py', '')
            
            module = importlib.import_module(module_import_path)
            functions, classes = get_defined_functions_and_classes(module)
            if functions or classes:
                items.append({"module_path":module_path, "module_name": module_import_path,"functions": functions,"classes": classes})
    doc_content = generate_sphinx_output(items)
    documentation.append(doc_content)
    with open(output_file, 'w', encoding='utf-8') as doc_file:
        doc_file.write('\n\n'.join(documentation))


if __name__ == "__main__":
    hierarchy = {
        "mipac/models": "Misskey Models",
        "mipac/manager": "Managers",
        "mipac/actions": "Actions",
        "mipac/types": "Type class",
        "mipac/errors": "Errors"

    }
    source_path = "mipac"
    output_path = "docs/index.rst"
    excluded_files = ["__init__.py", "_version.py"]  # Add files to exclude here
    generate_documentation(source_path, output_path, hierarchy, exclude_files=excluded_files)
    os.chdir("./docs")
    subprocess.run("make gettext && sphinx-intl update -p _build/gettext && ./doc_builder.sh", shell=True)