mirror of
https://github.com/UrloMythus/UnHided.git
synced 2026-04-09 02:40:47 +00:00
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
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()
|