| import PTT |
| import RTN |
|
|
| from fastapi import APIRouter, Request |
| from fastapi.responses import RedirectResponse |
| from fastapi.templating import Jinja2Templates |
|
|
| from comet.utils.models import settings |
| from comet.utils.general import config_check, get_debrid_extension |
|
|
| templates = Jinja2Templates("comet/templates") |
| main = APIRouter() |
|
|
|
|
| @main.get("/", status_code=200) |
| async def root(): |
| return RedirectResponse("/configure") |
|
|
|
|
| @main.get("/health", status_code=200) |
| async def health(): |
| return {"status": "ok"} |
|
|
|
|
| indexers = settings.INDEXER_MANAGER_INDEXERS |
| languages = [language for language in PTT.parse.LANGUAGES_TRANSLATION_TABLE.values()] |
| languages.insert(0, "Multi") |
| web_config = { |
| "indexers": [indexer.replace(" ", "_").lower() for indexer in indexers], |
| "languages": languages, |
| "resolutions": [resolution.value for resolution in RTN.models.Resolution], |
| "resultFormat": ["Title", "Metadata", "Size", "Tracker", "Languages"], |
| } |
|
|
|
|
| @main.get("/configure") |
| @main.get("/{b64config}/configure") |
| async def configure(request: Request): |
| return templates.TemplateResponse( |
| "index.html", |
| { |
| "request": request, |
| "CUSTOM_HEADER_HTML": settings.CUSTOM_HEADER_HTML |
| if settings.CUSTOM_HEADER_HTML |
| else "", |
| "webConfig": web_config, |
| "indexerManager": settings.INDEXER_MANAGER_TYPE, |
| "proxyDebridStream": settings.PROXY_DEBRID_STREAM, |
| }, |
| ) |
|
|
|
|
| @main.get("/manifest.json") |
| @main.get("/{b64config}/manifest.json") |
| async def manifest(b64config: str = None): |
| config = config_check(b64config) |
| if not config: |
| config = {"debridService": None} |
|
|
| debrid_extension = get_debrid_extension(config["debridService"]) |
|
|
| return { |
| "id": settings.ADDON_ID, |
| "name": f"{settings.ADDON_NAME}{(' | ' + debrid_extension) if debrid_extension is not None else ''}", |
| "description": "Stremio's fastest torrent/debrid search add-on.", |
| "version": "1.0.0", |
| "catalogs": [], |
| "resources": [ |
| { |
| "name": "stream", |
| "types": ["movie", "series"], |
| "idPrefixes": ["tt", "kitsu"], |
| } |
| ], |
| "types": ["movie", "series", "anime", "other"], |
| "logo": "https://i.imgur.com/jmVoVMu.jpeg", |
| "background": "https://i.imgur.com/WwnXB3k.jpeg", |
| "behaviorHints": {"configurable": True, "configurationRequired": False}, |
| } |
|
|
