114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
#!/usr/bin/env python3.12
|
|
|
|
import importlib
|
|
import sys
|
|
sys.path.insert(0, ".")
|
|
import logging
|
|
import asyncio
|
|
from typing import Any
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from lyric_search.sources import redis_cache
|
|
|
|
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.INFO)
|
|
|
|
loop = asyncio.get_event_loop()
|
|
app = FastAPI(title="codey.lol API",
|
|
version="1.0",
|
|
contact={
|
|
'name': 'codey'
|
|
},
|
|
redirect_slashes=False,
|
|
loop=loop)
|
|
|
|
constants = importlib.import_module("constants").Constants()
|
|
util = importlib.import_module("util").Utilities(app, constants)
|
|
glob_state = importlib.import_module("state").State(app, util, constants)
|
|
|
|
|
|
origins = [
|
|
"https://codey.lol",
|
|
"https://api.codey.lol"
|
|
]
|
|
|
|
app.add_middleware(CORSMiddleware,
|
|
allow_origins=origins,
|
|
allow_credentials=True,
|
|
allow_methods=["POST", "GET", "HEAD"],
|
|
allow_headers=["*"])
|
|
|
|
|
|
|
|
# pylint: disable=missing-function-docstring
|
|
|
|
"""
|
|
Blacklisted routes
|
|
"""
|
|
|
|
@app.get("/", include_in_schema=False)
|
|
def disallow_get():
|
|
return util.get_blocked_response()
|
|
|
|
@app.head("/", include_in_schema=False)
|
|
def base_head():
|
|
return
|
|
|
|
@app.get("/{path}", include_in_schema=False)
|
|
def disallow_get_any(request: Request, var: Any = None): # pylint: disable=unused-argument
|
|
path = request.path_params['path']
|
|
if not (
|
|
isinstance(path, str)
|
|
and
|
|
path.split("/", maxsplit=1) == "widget"
|
|
):
|
|
return util.get_blocked_response()
|
|
else:
|
|
logging.info("OK, %s",
|
|
path)
|
|
|
|
@app.post("/", include_in_schema=False)
|
|
def disallow_base_post():
|
|
return util.get_blocked_response()
|
|
|
|
"""
|
|
End Blacklisted Routes
|
|
"""
|
|
|
|
# pylint: enable=missing-function-docstring
|
|
|
|
|
|
"""
|
|
Actionable Routes
|
|
"""
|
|
|
|
randmsg_endpoint = importlib.import_module("endpoints.rand_msg").RandMsg(app, util, constants, glob_state)
|
|
transcription_endpoints = importlib.import_module("endpoints.transcriptions").Transcriptions(app, util, constants, glob_state)
|
|
ai_endpoints = importlib.import_module("endpoints.ai").AI(app, util, constants, glob_state)
|
|
# Below also provides: /lyric_cache_list/ (in addition to /lyric_search/)
|
|
lyric_search_endpoint = importlib.import_module("endpoints.lyric_search").LyricSearch(app, util, constants, glob_state)
|
|
# Below provides numerous LastFM-fed endpoints
|
|
lastfm_endpoints = importlib.import_module("endpoints.lastfm").LastFM(app, util, constants, glob_state)
|
|
# Below: YT endpoint(s)
|
|
yt_endpoints = importlib.import_module("endpoints.yt").YT(app, util, constants, glob_state)
|
|
# Below: XC endpoint(s)
|
|
xc_endpoints = importlib.import_module("endpoints.xc").XC(app, util, constants, glob_state)
|
|
# Below: Karma endpoint(s)
|
|
karma_endpoints = importlib.import_module("endpoints.karma").Karma(app, util, constants, glob_state)
|
|
# Below: Radio endpoint(s) - in development, sporadically loaded as needed
|
|
radio_endpoints = importlib.import_module("endpoints.radio").Radio(app, util, constants, glob_state)
|
|
# Below: Misc endpoints
|
|
misc_endpoints = importlib.import_module("endpoints.misc").Misc(app, util, constants, glob_state, radio_endpoints)
|
|
|
|
"""
|
|
End Actionable Routes
|
|
"""
|
|
|
|
|
|
"""
|
|
Startup
|
|
"""
|
|
redis_cache = redis_cache.RedisCache()
|
|
asyncio.get_event_loop().create_task(
|
|
redis_cache.create_index()) |