initial commit — dropLake

backend/main.py

@@ -0,0 +1,70 @@
+import logging
+import os
+from contextlib import asynccontextmanager
+from pathlib import Path
+
+from fastapi import FastAPI, Request
+from fastapi.responses import FileResponse
+from fastapi.staticfiles import StaticFiles
+from slowapi import Limiter, _rate_limit_exceeded_handler
+from slowapi.errors import RateLimitExceeded
+from slowapi.util import get_remote_address
+
+from config import settings
+from database import Base, engine
+from routes.download import router as download_router
+from routes.upload import router as upload_router
+from utils.cleanup import start_scheduler
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s %(levelname)s %(name)s — %(message)s",
+)
+
+FRONTEND = Path(__file__).parent.parent / "frontend"
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    os.makedirs(settings.storage_path, exist_ok=True)
+    Base.metadata.create_all(bind=engine)
+    scheduler = start_scheduler()
+    yield
+    scheduler.shutdown(wait=False)
+
+
+limiter = Limiter(key_func=get_remote_address)
+
+app = FastAPI(
+    title="dropLake",
+    lifespan=lifespan,
+    docs_url=None,    # no Swagger UI exposed
+    redoc_url=None,
+    openapi_url=None,
+)
+
+app.state.limiter = limiter
+app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
+
+# API routes
+app.include_router(upload_router, prefix="/api")
+app.include_router(download_router, prefix="/api")
+
+# Static assets (CSS, JS)
+app.mount("/static", StaticFiles(directory=str(FRONTEND / "static")), name="static")
+
+
+@app.get("/")
+async def serve_index():
+    return FileResponse(str(FRONTEND / "index.html"))
+
+
+@app.get("/s/{share_id}")
+async def serve_share_page(share_id: str):
+    # The JS on this page reads share_id from the URL path and the key from the fragment
+    return FileResponse(str(FRONTEND / "s.html"))
+
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}