from fastapi.responses import FileResponse, RedirectResponse
from fastapi.staticfiles import StaticFiles
from fastapi import FastAPI
import os
import pathlib


def add_static_file(app: FastAPI, fname: str, folder: str):
    print(f"Adding route for /{fname}")
    p = os.path.join(folder, fname)
    app.get(f"/{fname}", response_class=FileResponse, include_in_schema=False)(
        lambda: FileResponse(p)
    )


def add_static_files(app: FastAPI):
    # with importlib.resources.as_file(openflexure_microscope_server) as p:
    #   static_path = p.join("/static/")
    # TODO: don't hard code this!
    search_paths = [
        "/var/openflexure/application/openflexure-microscope-server/src/openflexure_microscope_server/static",
        pathlib.Path().absolute()
        / "application/openflexure-microscope-server/src/openflexure_microscope_server/static",
    ]
    if __file__:
        search_paths.append(pathlib.Path(__file__).parent.parent / "static")

    for static_path in search_paths:
        if os.path.isdir(static_path):
            break  # stop once one of the paths exists
    else:
        # If we get to the else: block, no pat was found.
        raise RuntimeError("Can't find static files :(")

    @app.get("/", response_class=RedirectResponse)
    async def redirect_fastapi():
        return "/index.html"

    for fname in os.listdir(static_path):
        fpath = os.path.join(static_path, fname)
        if os.path.isfile(fpath):
            add_static_file(app, fname, static_path)
        elif os.path.isdir(fpath):
            app.mount(
                f"/{fname}/",
                StaticFiles(directory=fpath),
                name=f"static_{fname}",
            )