mirror of
https://github.com/revanced/revanced-releases-api.git
synced 2025-04-30 06:24:27 +02:00
101 lines
2.9 KiB
Python
Executable File
101 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import toml
|
|
import uvicorn
|
|
from fastapi import FastAPI, Request, Response
|
|
from modules.Releases import Releases
|
|
from fastapi.responses import RedirectResponse
|
|
import modules.ResponseModels as ResponseModels
|
|
from slowapi import Limiter, _rate_limit_exceeded_handler
|
|
from slowapi.util import get_remote_address
|
|
from slowapi.errors import RateLimitExceeded
|
|
from fastapi_cache import FastAPICache
|
|
from fastapi_cache.backends.redis import RedisBackend
|
|
from fastapi_cache.decorator import cache
|
|
import aioredis
|
|
|
|
"""Get latest ReVanced releases from GitHub API."""
|
|
|
|
# Load config
|
|
|
|
config = toml.load("config.toml")
|
|
|
|
# Create releases instance
|
|
|
|
releases = Releases()
|
|
|
|
# Create FastAPI instance
|
|
|
|
app = FastAPI(title=config['docs']['title'],
|
|
description=config['docs']['description'],
|
|
version=config['docs']['version'],
|
|
license_info={"name": config['license']['name'],
|
|
"url": config['license']['url']
|
|
})
|
|
|
|
# Slowapi limiter
|
|
|
|
limiter = Limiter(key_func=get_remote_address, headers_enabled=True)
|
|
app.state.limiter = limiter
|
|
app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
|
|
|
|
# Setup cache
|
|
|
|
@cache()
|
|
async def get_cache():
|
|
return 1
|
|
|
|
# Routes
|
|
|
|
@app.get("/", response_class=RedirectResponse, status_code=301)
|
|
@limiter.limit(config['slowapi']['limit'])
|
|
async def root(request: Request, response: Response) -> RedirectResponse:
|
|
"""Brings up API documentation
|
|
|
|
Returns:
|
|
None: Redirects to /docs
|
|
"""
|
|
return RedirectResponse(url="/docs")
|
|
|
|
@app.get('/tools', response_model=ResponseModels.ToolsResponseModel)
|
|
@limiter.limit(config['slowapi']['limit'])
|
|
@cache(expire=60)
|
|
async def tools(request: Request, response: Response) -> dict:
|
|
"""Get patching tools' latest version.
|
|
|
|
Returns:
|
|
json: information about the patching tools' latest version
|
|
"""
|
|
return await releases.get_latest_releases(config['app']['repositories'])
|
|
|
|
@app.get('/apps', response_model=ResponseModels.AppsResponseModel)
|
|
@limiter.limit(config['slowapi']['limit'])
|
|
@cache(expire=60)
|
|
async def apps(request: Request, response: Response) -> dict:
|
|
"""Get patchable apps.
|
|
|
|
Returns:
|
|
json: list of supported apps
|
|
"""
|
|
return await releases.get_patchable_apps()
|
|
|
|
@app.get('/patches', response_model=ResponseModels.PatchesResponseModel)
|
|
@limiter.limit(config['slowapi']['limit'])
|
|
@cache(expire=60)
|
|
async def patches(request: Request, response: Response) -> dict:
|
|
"""Get latest patches.
|
|
|
|
Returns:
|
|
json: list of latest patches
|
|
"""
|
|
|
|
return await releases.get_patches_json()
|
|
|
|
@app.on_event("startup")
|
|
async def startup():
|
|
redis = aioredis.from_url("redis://localhost", encoding="utf8", decode_responses=True)
|
|
FastAPICache.init(RedisBackend(redis), prefix="fastapi-cache")
|
|
|
|
# Run app
|
|
if __name__ == '__main__':
|
|
uvicorn.run(app, host=config['uvicorn']['host'], port=config['uvicorn']['port']) |