From 23a00d2c0616dad4654e44ccfab791703f2a0523 Mon Sep 17 00:00:00 2001
From: OleSTEEP <krinkinoleg@gmail.com>
Date: Sun, 5 Oct 2025 20:18:05 +0300
Subject: [PATCH] backend: complete file uploading

---
 backend/api.py     | 48 ++++++++++++++++++++++++++++++++++------------
 backend/utils.py   |  8 ++++++++
 backend/vntypes.py | 10 +++++++---
 3 files changed, 51 insertions(+), 15 deletions(-)

diff --git a/backend/api.py b/backend/api.py
index affd950..7770657 100644
--- a/backend/api.py
+++ b/backend/api.py
@@ -1,9 +1,8 @@
 from typing import Annotated
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, File, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from pathlib import Path
 
-from vntypes import Novel, Mark
+from vntypes import *
 from utils import *
 from db import VNDB
 
@@ -23,24 +22,49 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+@app.get("/api/queue")
+def get_post_queue() -> list[FullNovel]:
+    raise HTTPException(status_code=501, detail="Method is not implemented yet!")
+
+
 @app.get("/api/posts/{post_id}")
 def get_post(post_id: int):
-    return "NOT IMPLEMENTED!"
+    raise HTTPException(status_code=501, detail="Method is not implemented yet!")
 
-@app.post("/api/new_post")
-def new_post(novel: Novel):
+
+@app.post("/api/posts/{post_id}")
+def new_post(novel: FullNovel):
     print(novel)
     return "yay!"
 
-@app.post("/api/new_mark")
+
+@app.patch("/api/posts/{post_id}")
+def edit_post(novel: FullNovel):
+    print(novel)
+    return "yay!"
+
+
+@app.post("/api/mark")
 def new_mark(mark: Mark):
     database.insert_mark(mark.type, mark.value)
     return "yay!"
 
-@app.post("/api/novel_thumb")
-async def novel_thumb(thumb: Annotated[bytes, File()], filename: str):
-    return {"file_size": len(thumb)}
+
+@app.post("/api/marks")
+def search_marks(part: str):
+    raise HTTPException(status_code=501, detail="Method is not implemented yet!")
+
+
+@app.post("/api/thumb")
+async def upload_thumb(thumb: Annotated[bytes, File()], filename: str):
+    return {"file_size": save_image(thumb, "thumbs", filename)}
+
 
 @app.post("/api/screenshot")
-async def screenshot(scrshot: Annotated[bytes, File()], filename: str):
-    return {"file_size": save_image(scrshot, "screens", filename)}
\ No newline at end of file
+async def upload_screenshot(scrshot: Annotated[bytes, File()], filename: str):
+    return {"file_size": save_image(scrshot, "screens", filename)}
+
+
+@app.post("/api/file")
+async def upload_file(file: Annotated[bytes, File()], filename: str):
+    return {"file_size": save_file(file, "files", filename)}
\ No newline at end of file
diff --git a/backend/utils.py b/backend/utils.py
index 7f3bdfd..7b1405b 100644
--- a/backend/utils.py
+++ b/backend/utils.py
@@ -24,3 +24,11 @@ def save_image(img: bytes, dir: str, name: str) -> int:
     with open(Path(path, name+'.jpg'), "wb") as file:
         file.write(img)
     return len(img)
+
+
+def save_file(file_b: bytes, dir: str, name: str) -> int:
+    path = asset(dir)
+    path.mkdir(exist_ok=True)
+    with open(Path(path, name+'.jpg'), "wb") as file:
+        file.write(file_b)
+    return len(file_b)
\ No newline at end of file
diff --git a/backend/vntypes.py b/backend/vntypes.py
index 3b843e4..eeb8a78 100644
--- a/backend/vntypes.py
+++ b/backend/vntypes.py
@@ -7,6 +7,10 @@ class Mark(BaseModel):
     type: Literal["tag", "badge", "genre"]
     value: str
 
+class NovelFile(BaseModel):
+    filename: str
+    platform: Literal["android", "ios", "windows", "linux", "macos"]
+
 class Novel(BaseModel):
     title: str
     description: str
@@ -21,9 +25,9 @@ class Novel(BaseModel):
     tg_post: str | None = None #url::Url
     post_at: datetime | None = None
 
-class FullNovel:
+class FullNovel(BaseModel):
     data: Novel
 
-    upload_queue: list[str]
-    files: list[str]
+    #upload_queue: list[str]
+    files: list[NovelFile]
     screenshots: list[str]