import os
import shutil
import json

import mistune
import minify_html

markdown_renderer = mistune.create_markdown(
    plugins=[
        "math",
        "strikethrough",
        "footnotes",
        "table",
        "url",
        "task_lists",
        "abbr",
        "mark",
        "subscript",
        "spoiler",
    ]
)

shutil.rmtree("build", ignore_errors=True)
os.makedirs("build")

with open("templates/general.html", "r") as template_file:
    template_text = template_file.read()

with open("templates/blog.html", "r") as blog_template_file:
    blog_template_text = blog_template_file.read()

with open("templates/gallery.html", "r") as gallery_template_file:
    gallery_template_text = gallery_template_file.read()

with open("templates/meta.json", "r") as meta_file:
    base_meta = json.load(meta_file)


def parse_meta(meta_data):
    """Generate meta tags for the HTML head section."""
    meta_tags = []

    for key, value in meta_data.get("tags", {}).items():
        meta_tags.append(f'<meta name="{key}" content="{value}">')

    for key, value in meta_data.get("og", {}).items():
        meta_tags.append(f'<meta property="og:{key}" content="{value}">')

    return "".join(meta_tags)


def generate_html_content(directory, filename, status=0):
    """Generate HTML content from a markdown file and metadata."""
    file_path = os.path.join(directory, filename)

    with open(file_path, "r") as file:
        md_text = file.read()

    meta_path = os.path.join(directory, "meta.json")
    meta = base_meta
    if os.path.exists(meta_path):
        with open(meta_path, "r") as meta_file:
            meta = json.load(meta_file)

    meta_data = parse_meta(meta)

    if status == 1:
        return (
            blog_template_text.replace("{{%CONTENT%}}", markdown_renderer(md_text))
            .replace("{{%TITLE%}}", meta.get("title", ""))
            .replace("{{%META%}}", meta_data)
        )
    elif status == 2:
        gallery_path = os.path.join(directory, "gallery.data")
        gallery = "[]"
        if os.path.exists(gallery_path):
            with open(gallery_path, "r") as gallery_file:
                gallery = gallery_file.read()

        return (
            gallery_template_text.replace("{{%CONTENT%}}", markdown_renderer(md_text))
            .replace("{{%GALLERY%}}", gallery)
            .replace("{{%TITLE%}}", meta.get("title", ""))
            .replace("{{%META%}}", meta_data)
        )
    else:
        return minify_html.minify(
            template_text.replace("{{%CONTENT%}}", markdown_renderer(md_text))
            .replace("{{%TITLE%}}", meta.get("title", ""))
            .replace("{{%META%}}", meta_data),
            minify_js=True,
            remove_processing_instructions=True,
        )


def copy_static_files(static_dir, build_dir="build"):
    """Copy static files and directories to the build folder."""
    for element in os.listdir(static_dir):
        src = os.path.join(static_dir, element)
        dst = os.path.join(build_dir, element)

        if os.path.isdir(src):
            shutil.copytree(src, dst)
        else:
            shutil.copy(src, dst)


def process_content_directory(source_dir, target_dir):
    """Recursively process the content directory and generate HTML files."""
    for filename in os.listdir(source_dir):
        source_path = os.path.join(source_dir, filename)
        target_path = os.path.join(target_dir, filename)

        if os.path.isdir(source_path):
            os.makedirs(target_path, exist_ok=True)
            process_content_directory(source_path, target_path)
        else:
            if filename.endswith(".md"):
                par = os.path.relpath(source_path, start="content").split(os.sep)
                status = 0

                if "blog" in par:
                    status = 1
                if "gallery" in par and par[-2] != "gallery":
                    status = 2

                html_filename = os.path.splitext(filename)[0] + ".html"
                output_file = os.path.join(target_dir, html_filename)
                content = generate_html_content(source_dir, filename, status=status)

                with open(output_file, "w") as file:
                    file.write(content)


copy_static_files("static")
process_content_directory("content", "build")