api/base.py

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)

origins = ["https://codey.lol", "https://api.codey.lol"]

app.add_middleware(
    CORSMiddleware,  # type: ignore
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["POST", "GET", "HEAD"],
    allow_headers=["*"],
)  # type: ignore

"""
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):
    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
"""

"""
Actionable Routes
"""

routes: dict = {
    "randmsg": importlib.import_module("endpoints.rand_msg").RandMsg(
        app, util, constants
    ),
    "transcriptions": importlib.import_module(
        "endpoints.transcriptions"
    ).Transcriptions(app, util, constants),
    "lyrics": importlib.import_module("endpoints.lyric_search").LyricSearch(
        app, util, constants
    ),
    "lastfm": importlib.import_module("endpoints.lastfm").LastFM(app, util, constants),
    "yt": importlib.import_module("endpoints.yt").YT(app, util, constants),
    "karma": importlib.import_module("endpoints.karma").Karma(app, util, constants),
    "radio": importlib.import_module("endpoints.radio").Radio(app, util, constants),
    "mgr": importlib.import_module("endpoints.mgr.mgr_test").Mgr(app, util, constants),
}

# Misc endpoint depends on radio endpoint instance
radio_endpoint = routes.get("radio")
if radio_endpoint:
    routes["misc"] = importlib.import_module("endpoints.misc").Misc(
        app, util, constants, radio_endpoint
    )

"""
End Actionable Routes
"""


"""
Startup
"""

redis = redis_cache.RedisCache()
loop.create_task(redis.create_index())