texgen (WIP)

1 files changed, 29 insertions, 9 deletions

diff --git a/web.py b/web.py
index 7aadac4..007e93d 100644
--- a/web.py
+++ b/web.py
@@ -2,26 +2,46 @@ import re
 
 import aiohttp
 from fastapi import FastAPI, Response
+from fastapi.responses import FileResponse
 
 import main
+import texgen
 
 app = FastAPI()
 
 utaten_pattern = re.compile(r'[a-z0-9]+')
+tex_generator = texgen.TexGenerator('pdf_cache', 'temp', 20)
 
 
[email protected]("/utaten/{item_id}/pdf")
-async def get_utaten_lyric_pdf(item_id: str):
-    raise NotImplementedError
+class TexSourceGenerationError(Exception):
+    pass
 
 
[email protected]("/utaten/{item_id}.tex")
-async def get_utaten_lyric_pdf(item_id: str, resp: Response):
+async def _get_utaten_tex_source(item_id: str) -> str:
     async with aiohttp.ClientSession() as ses:
         async with ses.get(f'https://utaten.com/lyric/{item_id}/') as r:
             if not r.ok:
-                resp.status_code = 503
-                return
+                raise TexSourceGenerationError('HTTP request failed when reading page source')
             html = await r.text()
-            tex = main.html_to_tex(html)
-            return Response(content=tex, media_type='application/x-tex')
+            return main.html_to_tex(html)
+
+
[email protected]("/utaten/{item_id}.pdf")
+async def get_utaten_lyric_pdf(item_id: str):
+    try:
+        print('_get_utaten_tex_source')
+        tex = await _get_utaten_tex_source(item_id)
+        print('xelatex')
+        pdf_path = await tex_generator.xelatex(tex)
+        return FileResponse(pdf_path, media_type='application/pdf')
+    except texgen.TexGenerationError as e:
+        return Response(content=f'Failed to generate tex file: {e}', status_code=502)
+
+
[email protected]("/utaten/{item_id}.tex")
+async def get_utaten_lyric_tex(item_id: str):
+    try:
+        tex = await _get_utaten_tex_source(item_id)
+        return Response(content=tex, media_type='application/x-tex')
+    except TexSourceGenerationError as e:
+        return Response(content=str(e), status_code=503)