import os
import markdown
from docx import Document
from ebooklib import epub
from core import utils


def create_readme(folder, bp):
    meta = bp['book_metadata']
    ls = bp['length_settings']
    content = f"""# {meta['title']}\n**Generated by BookApp**\n\n## Stats Used\n- **Type:** {ls.get('label', 'Custom')}\n- **Planned Chapters:** {ls['chapters']}\n- **Logic Depth:** {ls['depth']}\n- **Target Words:** {ls.get('words', 'Unknown')}"""
    with open(os.path.join(folder, "README.md"), "w") as f: f.write(content)


def compile_files(bp, ms, folder):
    utils.log("SYSTEM", "Compiling EPUB and DOCX...")
    meta = bp.get('book_metadata', {})
    title = meta.get('title', 'Untitled')

    if meta.get('filename'):
        safe = meta['filename']
    else:
        safe = utils.sanitize_filename(title)

    doc = Document(); doc.add_heading(title, 0)
    book = epub.EpubBook(); book.set_title(title); spine = ['nav']

    cover_path = os.path.join(folder, "cover.png")
    if os.path.exists(cover_path):
        with open(cover_path, 'rb') as f:
            book.set_cover("cover.png", f.read())

    ms.sort(key=utils.chapter_sort_key)

    for c in ms:
        num_str = str(c['num']).lower()
        if num_str == '0' or 'prologue' in num_str:
            filename = "prologue.xhtml"
            default_header = f"Prologue: {c['title']}"
        elif 'epilogue' in num_str:
            filename = "epilogue.xhtml"
            default_header = f"Epilogue: {c['title']}"
        else:
            filename = f"ch_{c['num']}.xhtml"
            default_header = f"Ch {c['num']}: {c['title']}"

        content = c['content'].strip()
        clean_content = content.replace("```markdown", "").replace("```", "").strip()
        lines = clean_content.split('\n')

        ai_header = None
        body_content = clean_content

        if lines and lines[0].strip().startswith('# '):
            ai_header = lines[0].strip().replace('#', '').strip()
            header = ai_header
            body_content = "\n".join(lines[1:]).strip()
        else:
            header = default_header

        doc.add_heading(header, 1)
        doc.add_paragraph(body_content)

        ch = epub.EpubHtml(title=header, file_name=filename)

        clean_content = clean_content.replace(f"{folder}\\", "").replace(f"{folder}/", "")
        html_content = markdown.markdown(clean_content)
        ch.content = html_content if ai_header else f"<h1>{header}</h1>{html_content}"

        book.add_item(ch); spine.append(ch)

    doc.save(os.path.join(folder, f"{safe}.docx"))
    book.spine = spine; book.add_item(epub.EpubNcx()); book.add_item(epub.EpubNav())
    epub.write_epub(os.path.join(folder, f"{safe}.epub"), book, {})
    create_readme(folder, bp)