117 lines
3.8 KiB
Python
117 lines
3.8 KiB
Python
"""
|
|
Art-DAG L2 Server Application Factory.
|
|
|
|
Creates and configures the FastAPI application with all routers and middleware.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
from contextlib import asynccontextmanager
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.responses import JSONResponse, HTMLResponse
|
|
|
|
from artdag_common import create_jinja_env
|
|
from artdag_common.middleware.auth import get_user_from_cookie
|
|
|
|
from .config import settings
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""Manage database connection pool lifecycle."""
|
|
import db
|
|
await db.init_pool()
|
|
yield
|
|
await db.close_pool()
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
"""
|
|
Create and configure the L2 FastAPI application.
|
|
|
|
Returns:
|
|
Configured FastAPI instance
|
|
"""
|
|
app = FastAPI(
|
|
title="Art-DAG L2 Server",
|
|
description="ActivityPub server for Art-DAG ownership and federation",
|
|
version="1.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
# Coop fragment pre-fetch — inject nav-tree, auth-menu, cart-mini
|
|
_FRAG_SKIP = ("/auth/", "/.well-known/", "/health",
|
|
"/internal/", "/static/", "/inbox")
|
|
|
|
@app.middleware("http")
|
|
async def coop_fragments_middleware(request: Request, call_next):
|
|
path = request.url.path
|
|
if (
|
|
request.method != "GET"
|
|
or any(path.startswith(p) for p in _FRAG_SKIP)
|
|
or request.headers.get("hx-request")
|
|
):
|
|
request.state.nav_tree_html = ""
|
|
request.state.auth_menu_html = ""
|
|
request.state.cart_mini_html = ""
|
|
return await call_next(request)
|
|
|
|
from artdag_common.fragments import fetch_fragments as _fetch_frags
|
|
|
|
user = get_user_from_cookie(request)
|
|
auth_params = {"email": user.email} if user and user.email else {}
|
|
nav_params = {"app_name": "artdag", "path": path}
|
|
|
|
try:
|
|
nav_tree_html, auth_menu_html, cart_mini_html = await _fetch_frags([
|
|
("blog", "nav-tree", nav_params),
|
|
("account", "auth-menu", auth_params or None),
|
|
("cart", "cart-mini", None),
|
|
])
|
|
except Exception:
|
|
nav_tree_html = auth_menu_html = cart_mini_html = ""
|
|
|
|
request.state.nav_tree_html = nav_tree_html
|
|
request.state.auth_menu_html = auth_menu_html
|
|
request.state.cart_mini_html = cart_mini_html
|
|
|
|
return await call_next(request)
|
|
|
|
# Initialize Jinja2 templates
|
|
template_dir = Path(__file__).parent / "templates"
|
|
app.state.templates = create_jinja_env(template_dir)
|
|
|
|
# Custom 404 handler
|
|
@app.exception_handler(404)
|
|
async def not_found_handler(request: Request, exc):
|
|
from artdag_common.middleware import wants_html
|
|
if wants_html(request):
|
|
from artdag_common import render
|
|
return render(app.state.templates, "404.html", request,
|
|
user=None,
|
|
)
|
|
return JSONResponse({"detail": "Not found"}, status_code=404)
|
|
|
|
# Include routers
|
|
from .routers import auth, assets, activities, anchors, storage, users, renderers
|
|
|
|
# Root routes
|
|
app.include_router(auth.router, prefix="/auth", tags=["auth"])
|
|
app.include_router(users.router, tags=["users"])
|
|
|
|
# Feature routers
|
|
app.include_router(assets.router, prefix="/assets", tags=["assets"])
|
|
app.include_router(activities.router, prefix="/activities", tags=["activities"])
|
|
app.include_router(anchors.router, prefix="/anchors", tags=["anchors"])
|
|
app.include_router(storage.router, prefix="/storage", tags=["storage"])
|
|
app.include_router(renderers.router, prefix="/renderers", tags=["renderers"])
|
|
|
|
# WebFinger and ActivityPub discovery
|
|
from .routers import federation
|
|
app.include_router(federation.router, tags=["federation"])
|
|
|
|
return app
|
|
|
|
|
|
# Create the default app instance
|
|
app = create_app()
|