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 starlette.middleware import Middleware
from starlette.middleware.cors import CORSMiddleware

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

import os
import magic

middleware = [
    Middleware(
        CORSMiddleware,
        allow_origins=['*'],
        allow_credentials=True,
        allow_methods=['*'],
        allow_headers=['*']
    )
]

app = FastAPI(middleware=middleware)

# app.add_middleware(
#     CORSMiddleware,
#     allow_origins=['*'],
#     allow_credentials=True,
#     allow_methods=['*'],
#     allow_headers=['*'],
#     max_age=3600
# )

app.include_router(user_route)
app.include_router(auth)


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


@app.get("/items/{name}/{version}/{file}")
async 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, headers={"Access-Control-Allow-Origin": "*"})

    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()], stable: Annotated[bool, 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)
        if stable:
            with open(f"{dir_stable}/{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")