rename build.py and enable CORS

jakobkogler · jakobkogler · commit 4f0576646a77 · 2021-10-10T20:40:25.000+02:00
diff --git a/build/Dockerfile b/build/Dockerfile
@@ -5,10 +5,10 @@ WORKDIR /app/
 COPY build/requirements.txt .
 RUN pip install -r requirements.txt
 
-COPY build/build.py build.py
+COPY build/main.py main.py
 
 ENV BASEURL="https://cp-algorithms.com"
 VOLUME /input
 VOLUME /output
 
-CMD python build.py -i /input/src -o /output --template-dir /input/src/_templates --static-dir /input/static --baseurl $BASEURL
+CMD python main.py -i /input/src -o /output --template-dir /input/src/_templates --static-dir /input/static --baseurl $BASEURL
diff --git a/build/main.py b/build/main.py
@@ -12,6 +12,7 @@
 from urllib.parse import urljoin
 import shutil
 from tempfile import TemporaryDirectory
+from typing import Dict, Tuple
 
 import markdown  # type: ignore
 from tqdm import tqdm  # type: ignore
@@ -137,7 +138,17 @@ def convert(self, md_content: str, relative_url: Path) -> str:
         return content
 
 
-def convert(request) -> str:
+def convert(request) -> Tuple[str, int, Dict]:
+    # Set CORS headers for the preflight request
+    if request.method == 'OPTIONS':
+        headers = {
+            'Access-Control-Allow-Origin': '*',
+            'Access-Control-Allow-Methods': 'POST',
+            'Access-Control-Allow-Headers': 'Content-Type',
+            'Access-Control-Max-Age': '3600'
+        }
+        return ('', 204, headers)
+
     request_json = request.get_json()
 
     # prepare markdown converter
@@ -160,7 +171,10 @@ def convert(request) -> str:
         md_content = request_json['markdown']
         html_content = md.convert(md_content, Path("."))
 
-    return html_content
+    headers = {
+        'Access-Control-Allow-Origin': '*'
+    }
+    return (html_content, 200, headers)
 
 
 def main():
