Genarrative/server-rs/crates/api-server/src/modules/profile.rs

use axum::{
    Router, middleware,
    routing::{delete, get, patch, post},
};

use crate::{
    auth::require_bearer_auth,
    external_api_keys::{
        create_external_api_key, ensure_llm_router_api_key, list_external_api_keys,
        revoke_external_api_key,
    },
    payment::{
        create_payment_app, get_public_payment_checkout, list_payment_apps, revoke_payment_api_key,
        update_payment_app,
    },
    profile_identity::update_profile_identity,
    runtime_profile::{
        claim_profile_task_reward, confirm_wechat_profile_recharge_order,
        create_profile_recharge_order, get_profile_analytics_metric, get_profile_dashboard,
        get_profile_membership_upgrade_quote, get_profile_recharge_center,
        get_profile_referral_invite_center, get_profile_task_center, get_profile_wallet_ledger,
        redeem_profile_referral_invite_code, redeem_profile_reward_code,
        stream_wechat_profile_recharge_order_events, submit_profile_feedback,
    },
    state::AppState,
};

pub fn router(state: AppState) -> Router<AppState> {
    Router::new()
        .route(
            "/api/profile/me",
            patch(update_profile_identity).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/dashboard",
            get(get_profile_dashboard).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/api-keys/llm-router",
            post(ensure_llm_router_api_key).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/api-keys",
            get(list_external_api_keys)
                .post(create_external_api_key)
                .route_layer(middleware::from_fn_with_state(
                    state.clone(),
                    require_bearer_auth,
                )),
        )
        .route(
            "/api/profile/api-keys/{key_id}",
            delete(revoke_external_api_key).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/payment/apps",
            get(list_payment_apps).post(create_payment_app).route_layer(
                middleware::from_fn_with_state(state.clone(), require_bearer_auth),
            ),
        )
        .route(
            "/api/profile/payment/apps/{app_id}",
            axum::routing::patch(update_payment_app).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/payment/keys/{key_id}",
            axum::routing::delete(revoke_payment_api_key).route_layer(
                middleware::from_fn_with_state(state.clone(), require_bearer_auth),
            ),
        )
        .route(
            "/api/payment/checkout/{checkout_token}",
            get(get_public_payment_checkout),
        )
        .route(
            "/api/profile/wallet-ledger",
            get(get_profile_wallet_ledger).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/recharge-center",
            get(get_profile_recharge_center).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/membership/upgrade-quote",
            post(get_profile_membership_upgrade_quote).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/recharge/orders",
            post(create_profile_recharge_order).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/recharge/orders/{order_id}/wechat/confirm",
            post(confirm_wechat_profile_recharge_order).route_layer(
                middleware::from_fn_with_state(state.clone(), require_bearer_auth),
            ),
        )
        .route(
            "/api/profile/recharge/orders/{order_id}/wechat/events",
            get(stream_wechat_profile_recharge_order_events).route_layer(
                middleware::from_fn_with_state(state.clone(), require_bearer_auth),
            ),
        )
        .route(
            "/api/profile/feedback",
            post(submit_profile_feedback)
                .layer(axum::extract::DefaultBodyLimit::max(6 * 1024 * 1024))
                .route_layer(middleware::from_fn_with_state(
                    state.clone(),
                    require_bearer_auth,
                )),
        )
        .route(
            "/api/profile/referrals/invite-center",
            get(get_profile_referral_invite_center).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/referrals/redeem-code",
            post(redeem_profile_referral_invite_code).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/redeem-codes/redeem",
            post(redeem_profile_reward_code).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/analytics/metric",
            get(get_profile_analytics_metric).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/tasks",
            get(get_profile_task_center).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
        .route(
            "/api/profile/tasks/{task_id}/claim",
            post(claim_profile_task_reward).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
}