Genarrative/server-rs/crates/api-server/src/frontend_runtime_config.rs

use axum::{
    extract::{Extension, State},
    http::{HeaderMap, StatusCode, header},
    response::{IntoResponse, Response},
};
use serde::Serialize;
use serde_json::json;

use crate::{
    api_response::json_success_body, auth::optional_access_token_from_headers,
    http_error::AppError, request_context::RequestContext, state::AppState,
};

#[derive(Debug, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct FrontendRuntimeConfigResponse {
    pub image_editor_agent_sidebar_enabled: bool,
    pub agc_template_library_enabled: bool,
}

pub async fn get_frontend_runtime_config(
    State(state): State<AppState>,
    Extension(request_context): Extension<RequestContext>,
    headers: HeaderMap,
) -> Result<Response, AppError> {
    let authenticated = optional_access_token_from_headers(
        &state,
        "/api/runtime/frontend-config".to_string(),
        headers,
        request_context.request_id().to_string(),
    )
    .await?;
    let user_id = authenticated
        .as_ref()
        .map(|authenticated| authenticated.claims().user_id());
    let image_editor_agent_sidebar_enabled = state
        .is_image_editor_agent_sidebar_enabled_for_user(user_id)
        .await
        .map_err(|error| {
            AppError::from_status(StatusCode::BAD_GATEWAY)
                .with_message("读取前端运行时配置失败")
                .with_details(json!({
                    "provider": "spacetimedb",
                    "message": error.to_string(),
                }))
        })?;

    let agc_template_library_enabled = state
        .is_agc_template_library_enabled_for_user(user_id)
        .await
        .map_err(|error| {
            AppError::from_status(StatusCode::BAD_GATEWAY)
                .with_message("读取前端运行时配置失败")
                .with_details(json!({
                    "provider": "spacetimedb",
                    "message": error.to_string(),
                }))
        })?;

    Ok((
        [
            (header::CACHE_CONTROL, "no-store"),
            (header::VARY, "Authorization"),
        ],
        json_success_body(
            Some(&request_context),
            FrontendRuntimeConfigResponse {
                image_editor_agent_sidebar_enabled,
                agc_template_library_enabled,
            },
        ),
    )
        .into_response())
}