import os
import re
import shutil
import yaml

SOURCE_DIR = "docs"
ENGLISH_DIR = os.path.join(SOURCE_DIR, "en")
MKDOCS_YML = "mkdocs.yml"

# Kopiera alla filer från docs/ till docs/en/
if os.path.exists(ENGLISH_DIR):
    shutil.rmtree(ENGLISH_DIR)
shutil.copytree(SOURCE_DIR, ENGLISH_DIR)

def clean_file(path, keep_lang):
    with open(path, "r", encoding="utf-8") as f:
        content = f.read()

    # Ta bort block som inte ska vara kvar
    if keep_lang == "sv":
        content = re.sub(r":en\n.*?\n:::", "", content, flags=re.DOTALL)
    elif keep_lang == "en":
        content = re.sub(r":sv\n.*?\n:::", "", content, flags=re.DOTALL)

    # Ta bort block-taggar för behållna block
    content = re.sub(r":(sv|en)\n", "", content)
    content = re.sub(r"\n:::", "", content)

    # Hantera inline-taggar
    def inline_replacer(match):
        lang, text = match.groups()
        return text if lang == keep_lang else ""

    content = re.sub(r"\[:(sv|en)](.*?)(?=(\[:|$))", inline_replacer, content)

    with open(path, "w", encoding="utf-8") as f:
        f.write(content.strip() + "\n")

# Rensa svenska versioner
for filename in os.listdir(SOURCE_DIR):
    full_path = os.path.join(SOURCE_DIR, filename)
    if filename.endswith(".md") and os.path.isfile(full_path):
        clean_file(full_path, keep_lang="sv")

# Rensa engelska versioner
for filename in os.listdir(ENGLISH_DIR):
    full_path = os.path.join(ENGLISH_DIR, filename)
    if filename.endswith(".md") and os.path.isfile(full_path):
        clean_file(full_path, keep_lang="en")

# Skapa .pages för engelska versionen utifrån mkdocs.yml
def extract_nav_from_mkdocs_yml(path):
    with open(path, "r", encoding="utf-8") as f:
        data = yaml.safe_load(f)
    return {
        "title": data.get("site_name", "SA6ANW"),
        "nav": data.get("nav", [])
    }

english_pages = extract_nav_from_mkdocs_yml(MKDOCS_YML)

with open(os.path.join(ENGLISH_DIR, ".pages"), "w", encoding="utf-8") as f:
    yaml.dump(english_pages, f, allow_unicode=True)