71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
import mimetypes
|
|
import time
|
|
|
|
from . import router
|
|
from fastapi import UploadFile, Request, Depends
|
|
|
|
from ..util import minioClient, ergo
|
|
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])
|
|
await ergo.broadcastAs(username, "CACHEBUST")
|
|
return {"url": f"https://{config.MINIO_EXTERNAL_ADDR}/pfp/{username}?{time.time():.0f}"}
|
|
|
|
|
|
@router.post("/pfp/uploadIcon", dependencies=[Depends(JWTBearer())])
|
|
async def IconUpload(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])
|
|
await ergo.broadcastAs(username, "CACHEBUST")
|
|
return {"url": f"https://{config.MINIO_EXTERNAL_ADDR}/pfp/{username}.icon?{time.time():.0f}"}
|