pycdn/server.py

from typing import Union, Annotated
import uvicorn
from routes.user_routes import user_route
from bearer.routes.auth import auth
from db.DB import DBException
import mimetypes
from mime.mime import mime_content_type
from fastapi.middleware.cors import CORSMiddleware

from fastapi import FastAPI, File, UploadFile, Form, Response, status, HTTPException

import os
import magic

app = FastAPI()
app.include_router(user_route)
app.include_router(auth)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

@app.get("/")
def read_root():
    return {"Hello": "World"}


@app.get("/items/{name}/{version}/{file}")
def read_item(version: str, name: str, file: str = 'index.js'):
    file_path = f"packages/{name}/{version}/{file}"
    file_mime = mime_content_type(file_path)
    print(file_mime)
    if os.path.exists(file_path):
        with open(file_path, 'rb') as f:
            return Response(content=f.read(), media_type=file_mime)

    return status.HTTP_404_NOT_FOUND


@app.post("/files/")
async def create_file(file: Annotated[bytes | None, File()] = None):
    if not file:
        return {"message": "No file sent"}
    else:
        return {"file_size": len(file)}


@app.post("/uploadfile/")
async def create_upload_file(file: UploadFile | None, package: Annotated[str, Form()], version: Annotated[str, Form()], filename: Annotated[str, Form()]):
    try:
        contents = file.file.read()
        dir_file = "packages/{package}/{version}".format(package=package, version=version)
        dir_last = "packages/{package}/latest".format(package=package)
        dir_stable = "packages/{package}/stable".format(package=package)
        if not os.path.exists(dir_file):
            os.makedirs(dir_file)
        if not os.path.exists(dir_last):
            os.makedirs(dir_last)
        if not os.path.exists(dir_stable):
            os.makedirs(dir_stable)
        with open(f"{dir_file}/{filename}", 'wb') as f:
            f.write(contents)
        with open(f"{dir_last}/{filename}", 'wb') as f:
            f.write(contents)
    except Exception as err:
        return {"message": "There was an error uploading the file, error {err}".format(err=err)}
    finally:
        file.file.close()

    return {"message": f"Successfully uploaded {file.filename}"}


if __name__ == "__main__":
    uvicorn.run("server:app", port=5044, log_level="info")