import os
from tableofcontents_generator import generate_table_of_contents
from contents_generator import generate_contents
from mainheader_generator import generate_mainheader

# List of target platforms
platforms = ["all", "windows", "macos", "linux", "selfhost"]

# Map platform to the header file to use
header_files = {
    "all": "source/components/header.md",
    "windows": "source/components/windowsheader.md",
    "macos": "source/components/macosheader.md",
    "linux": "source/components/linuxheader.md",
    "selfhost": "source/components/selfhostheader.md"
}

def generate_readme_for_platform(platform):
    content = ""
    header_file = header_files.get(platform, "source/components/header.md")

    # Inject mainheader with dynamic project count
    if platform == "all":
        content += generate_mainheader()
    
    # Inject header (for 'all', use header.md; for others, use the platform-specific header)
    with open(header_file, "r", encoding="utf-8") as f:
        content += f.read() + "\n"
    
    # Inject tags.md
    with open("source/components/tags.md", "r", encoding="utf-8") as f:
        content += f.read() + "\n"
    
    # Generate Table of Contents using the imported function
    toc_md = generate_table_of_contents()
    content += toc_md + "\n"
    
    # Generate the actual markdown list of contents for the given platform
    contents_md = generate_contents(platform)
    content += contents_md + "\n"
    
    # Inject footer.md
    with open("source/components/footer.md", "r", encoding="utf-8") as f:
        content += f.read() + "\n"
    
    # Write output file
    output_filename = "README.md" if platform == "all" else f"readmes/{platform}.md"
    with open(output_filename, "w", encoding="utf-8") as f:
        f.write(content)
    print(f"Generated {output_filename}")

if __name__ == "__main__":
    for platform in platforms:
        generate_readme_for_platform(platform)