from fastapi import APIRouter, Request
from fastapi.responses import HTMLResponse, FileResponse
from fastapi.templating import Jinja2Templates



import pdfkit
from PyPDF2 import PdfMerger

from pathlib import Path

from app.entity.models import Entity
from app.template.models import ProvisionTemplate
from ..schemas import ContractDraft


async def build_model(model):
    parties = []
    for p in model.parties:
        party = {
            "entity": await Entity.get(p.entity_id),
            "part": p.part
        }
        if p.representative_id:
            party['representative'] = await Entity.get(p.representative_id)

        parties.append(party)


    model.parties = parties

    provisions = []
    for p in model.provisions:
        if p.provision.type == "template":
            provisions.append(await ProvisionTemplate.get(p.provision.provision_template_id))
        else:
            provisions.append(p.provision)
    model.provisions = provisions

    model.location = "Toulouse"
    model.date = "01/01/1970"
    return model


BASE_PATH = Path(__file__).resolve().parent

print_router = APIRouter()


templates = Jinja2Templates(directory=str(BASE_PATH / "templates"))

options = {
    'encoding': 'UTF-8',
    'margin-left': '20mm',
    'margin-right': '20mm',
    'margin-bottom': '20mm',
    'margin-top': '20mm'
}
options_content = options.copy()
options_content['footer-html'] = 'footer.html'
options_content['margin-bottom'] = '60mm'



async def render_template(host):
    draft = await ContractDraft.get("63e92534aafed8b509f229c4")
    lawyer = {
        "firstname": "Nathaniel",
        "lastname": "Toshi",
    }

    template = templates.get_template("print.html")
    return template.render({
        "draft": await build_model(draft),
        "lawyer": lawyer,
        "static_host": host
    })


async def render_frontpage(host, draft, lawyer, ):
    template = templates.get_template("frontpage.html")
    return template.render({
        "draft": draft,
        "lawyer": lawyer,
        "static_host": host
    })


async def render_content(host, draft, ):
    template = templates.get_template("content.html")
    return template.render({
        "draft": draft,
        "static_host": host
    })


async def render_footer(host, draft, ):
    template = templates.get_template("footer.html")
    return template.render({
        "draft": draft,
        "static_host": host
    })


@print_router.get("/frontpage", response_class=HTMLResponse)
async def frontpage() -> str:
    draft = await build_model(await ContractDraft.get("63e92534aafed8b509f229c4"))
    lawyer = {
        "firstname": "Nathaniel",
        "lastname": "Toshi",
    }
    return await render_frontpage('localhost', draft, lawyer)


@print_router.get("/content", response_class=HTMLResponse)
async def content() -> str:
    draft = await build_model(await ContractDraft.get("63e92534aafed8b509f229c4"))
    return await render_content('localhost', draft)


@print_router.get("/footer", response_class=HTMLResponse)
async def footer() -> str:
    draft = await build_model(await ContractDraft.get("63e92534aafed8b509f229c4"))
    return await render_footer('localhost', draft)


@print_router.get("/", response_class=HTMLResponse)
async def create() -> str:
    return await render_template('localhost')


@print_router.get("/pdf", response_class=FileResponse)
async def create_pdf() -> str:
    draft = await build_model(await ContractDraft.get("63e92534aafed8b509f229c4"))
    lawyer = {
        "firstname": "Nathaniel",
        "lastname": "Toshi",
    }

    merger = PdfMerger()

    pdfkit.from_string(await render_frontpage('nginx', draft, lawyer), 'front.pdf', options=options)
    merger.append('front.pdf')
    with open('footer.html', 'w') as f:
        f.write(await render_footer('nginx', draft))
    pdfkit.from_string(await render_content('nginx', draft), 'content.pdf', options=options_content)
    merger.append('content.pdf')

    merger.write("out.pdf")
    merger.close()
    print(options)
    return FileResponse(
        "out.pdf",
        media_type="application/pdf",
        filename=draft.name)