add profile picture support and rearrange things to be slightly more sane

cef_3M/__init__.py

@@ -1,21 +1,13 @@
 from fastapi import FastAPI, UploadFile, Request, Depends
 from fastapi.middleware.cors import CORSMiddleware
-from minio import Minio
 import mimetypes
-import re
 
 from .auth import JWTBearer
 from .sql import SessionMaker, Uploads
 from . import util
 import config
 from datetime import datetime, timedelta
-
-
-minioClient = Minio(
-    config.MINIO_ADDR,
-    access_key=config.MINIO_ACCESS_KEY,
-    secret_key=config.MINIO_SECRET_KEY,
-)
+from . import endpoints
 
 app = FastAPI()
 app.add_middleware(
@@ -26,6 +18,8 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+app.include_router(endpoints.router)
+
 @app.post("/upload", dependencies=[Depends(JWTBearer())])
 async def upload(file: UploadFile, request: Request):
     if file.size > config.MAX_FILE_SIZE:
@@ -40,11 +34,12 @@ async def upload(file: UploadFile, request: Request):
         existing.expiry = datetime.now() + timedelta(days=7)
     else:
         mime = mimetypes.guess_type(safeFilename)
-        minioClient.put_object("uploads", sha, file.file, file.size, content_type=mime[0])
+        util.minioClient.put_object("uploads", sha, file.file, file.size, content_type=mime[0])
         up = Uploads(hash=sha)
         session.add(up)
     session.commit()
     return {"url": f"https://{config.MINIO_ADDR}/uploads/{sha}/{safeFilename}"}
 
 
+
 __all__ = ["sql", "auth", "util"]

cef_3M/endpoints/__init__.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter
+import os
+import importlib
+router = APIRouter()
+
+for module in os.listdir(os.path.dirname(__file__)):
+    if module == '__init__.py' or module[-3:] != '.py':
+        continue
+    importlib.import_module("."+module[:-3], package="cef_3M.endpoints")

cef_3M/endpoints/pfp.py

@@ -0,0 +1,66 @@
+import mimetypes
+import time
+
+from . import router
+from fastapi import UploadFile, Request, Depends
+
+from ..util import minioClient
+from ..auth import JWTBearer
+import config
+
+from pywuffs import ImageDecoderType
+from pywuffs.aux import (
+    ImageDecoder,
+    ImageDecoderConfig,
+)
+pfpConfig = ImageDecoderConfig()
+pfpConfig.max_incl_dimension = 400
+pfpConfig.enabled_decoders = [
+    ImageDecoderType.GIF,
+    ImageDecoderType.PNG,
+    ImageDecoderType.JPEG,
+]
+
+iconConfig = ImageDecoderConfig()
+iconConfig.max_incl_dimension = 24
+iconConfig.enabled_decoders = [
+    ImageDecoderType.PNG,
+]
+
+@router.post("/pfp/upload", dependencies=[Depends(JWTBearer())])
+async def pfpUpload(file: UploadFile, request: Request):
+    if file.size > config.MAX_PFP_SIZE:
+        return {"error": "file too big"}
+    whoami = request.state.jwt
+    username = whoami["account"].lower()
+
+    # It's not the path I exactly wanted, but this will have to do - WUFFS ensures that the file is valid then we just save it to the server
+    # I hope there's no issue in doing that...
+    decoder = ImageDecoder(pfpConfig)
+    data = await file.read()
+    decoded = decoder.decode(data)
+    if decoded.error_message:
+        return {"error": "invalid file"}
+    file.file.seek(0)
+
+    mime = mimetypes.guess_type(file.filename)
+    minioClient.put_object("pfp", username, file.file, file.size, content_type=mime[0])
+    return {"url": f"https://{config.MINIO_ADDR}/pfp/{username}?{time.time():.0f}"}
+
+@router.post("/pfp/uploadIcon", dependencies=[Depends(JWTBearer())])
+async def pfpUpload(file: UploadFile, request: Request):
+    if file.size > config.MAX_PFP_SIZE:
+        return {"error": "file too big"}
+    whoami = request.state.jwt
+    username = whoami["account"].lower()
+
+    decoder = ImageDecoder(iconConfig)
+    data = await file.read()
+    decoded = decoder.decode(data)
+    if decoded.error_message:
+        return {"error": "invalid file"}
+    file.file.seek(0)
+
+    mime = mimetypes.guess_type(file.filename)
+    minioClient.put_object("pfp", username+"/icon", file.file, file.size, content_type=mime[0])
+    return {"url": f"https://{config.MINIO_ADDR}/pfp/{username}/icon?{time.time():.0f}"}

cef_3M/util.py

@@ -1,5 +1,8 @@
 import hashlib
 import re
+import config
+from minio import Minio
+
 
 from fastapi import UploadFile
 
@@ -13,3 +16,9 @@ async def SHA256(f: UploadFile) -> str:
         sha.update(data)
     await f.seek(0)
     return sha.hexdigest()
+
+minioClient = Minio(
+    config.MINIO_ADDR,
+    access_key=config.MINIO_ACCESS_KEY,
+    secret_key=config.MINIO_SECRET_KEY,
+)