import bs4
import sys
import os
from glob import glob
from .formatters import formatters


def get_required_libraries(soup: bs4.BeautifulSoup) -> list[str]:
    if html := soup.find("html"):
        if isinstance(html, bs4.Tag):
            if html.has_attr("dj-libs"):
                dj_libs = html["dj-libs"]
                if isinstance(dj_libs, str):
                    return dj_libs.split(";")
    return []


def convert_file(path: str, overwrite: bool = False) -> None:
    with open(path, "r") as file:
        soup = bs4.BeautifulSoup(file, "html.parser")

    required_libraries = get_required_libraries(soup)

    for formatter in formatters:
        formatter.format(soup)

    if overwrite:
        out_path = path
    else:
        out_path = path.replace(".html", ".n.html")
    with open(out_path, "w") as file:
        for lib in required_libraries:
            file.write(f"{{% load {lib} %}}\n")
        file.write(soup
                   .prettify(formatter=bs4.formatter.HTMLFormatter(indent=4))
                   .replace(""", '"'))  # Could this cause any issues? (maybe in attributes)


def convert_dir(path: str, overwrite: bool = False) -> None:
    for filepath in glob(f"{path}/**/*.html", recursive=True):
        convert_file(filepath, overwrite)


def command_entry() -> None:
    path = os.path.abspath(sys.argv[1])
    if os.path.isdir(path):
        convert_dir(path)
    elif os.path.isfile(path):
        convert_file(path)