Add files via upload

mediaflow_proxy/main.py

@@ -0,0 +1,99 @@
+import logging
+from importlib import resources
+
+from fastapi import FastAPI, Depends, Security, HTTPException
+from fastapi.security import APIKeyQuery, APIKeyHeader
+from starlette.middleware.cors import CORSMiddleware
+from starlette.responses import RedirectResponse
+from starlette.staticfiles import StaticFiles
+
+from mediaflow_proxy.configs import settings
+from mediaflow_proxy.routes import proxy_router, extractor_router, speedtest_router
+from mediaflow_proxy.schemas import GenerateUrlRequest
+from mediaflow_proxy.utils.crypto_utils import EncryptionHandler, EncryptionMiddleware
+from mediaflow_proxy.utils.http_utils import encode_mediaflow_proxy_url
+
+logging.basicConfig(level=settings.log_level, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+app = FastAPI()
+api_password_query = APIKeyQuery(name="api_password", auto_error=False)
+api_password_header = APIKeyHeader(name="api_password", auto_error=False)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+app.add_middleware(EncryptionMiddleware)
+
+
+async def verify_api_key(api_key: str = Security(api_password_query), api_key_alt: str = Security(api_password_header)):
+    """
+    Verifies the API key for the request.
+
+    Args:
+        api_key (str): The API key to validate.
+        api_key_alt (str): The alternative API key to validate.
+
+    Raises:
+        HTTPException: If the API key is invalid.
+    """
+    if not settings.api_password:
+        return
+
+    if api_key == settings.api_password or api_key_alt == settings.api_password:
+        return
+
+    raise HTTPException(status_code=403, detail="Could not validate credentials")
+
+
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy"}
+
+
+@app.get("/favicon.ico")
+async def get_favicon():
+    return RedirectResponse(url="/logo.png")
+
+
+@app.get("/speedtest")
+async def show_speedtest_page():
+    return RedirectResponse(url="/speedtest.html")
+
+
+@app.post("/generate_encrypted_or_encoded_url")
+async def generate_encrypted_or_encoded_url(request: GenerateUrlRequest):
+    if "api_password" not in request.query_params:
+        request.query_params["api_password"] = request.api_password
+
+    encoded_url = encode_mediaflow_proxy_url(
+        request.mediaflow_proxy_url,
+        request.endpoint,
+        request.destination_url,
+        request.query_params,
+        request.request_headers,
+        request.response_headers,
+        EncryptionHandler(request.api_password) if request.api_password else None,
+        request.expiration,
+        str(request.ip) if request.ip else None,
+    )
+    return {"encoded_url": encoded_url}
+
+
+app.include_router(proxy_router, prefix="/proxy", tags=["proxy"], dependencies=[Depends(verify_api_key)])
+app.include_router(extractor_router, prefix="/extractor", tags=["extractors"], dependencies=[Depends(verify_api_key)])
+app.include_router(speedtest_router, prefix="/speedtest", tags=["speedtest"], dependencies=[Depends(verify_api_key)])
+
+static_path = resources.files("mediaflow_proxy").joinpath("static")
+app.mount("/", StaticFiles(directory=str(static_path), html=True), name="static")
+
+
+def run():
+    import uvicorn
+
+    uvicorn.run(app, host="0.0.0.0", port=8888, log_level="info", workers=3)
+
+
+if __name__ == "__main__":
+    run()