Aether/apps/aether-gateway/src/handlers/admin/billing/wallets/mutations/adjust.rs

use super::super::shared::{
    admin_wallet_id_from_suffix_path, admin_wallet_operator_id,
    build_admin_wallet_not_found_response, build_admin_wallet_summary_payload_with_package,
    build_admin_wallet_transaction_payload, build_admin_wallets_bad_request_response,
    build_admin_wallets_data_unavailable_response, normalize_admin_wallet_balance_type,
    normalize_admin_wallet_description, normalize_admin_wallet_non_zero_amount,
    resolve_admin_wallet_owner_summary, AdminWalletAdjustRequest,
    ADMIN_WALLETS_API_KEY_GIFT_ADJUST_DETAIL,
};
use crate::handlers::admin::request::{AdminAppState, AdminRequestContext};
use crate::handlers::admin::shared::{attach_admin_audit_response, unix_secs_to_rfc3339};
use crate::GatewayError;
use aether_data::repository::wallet::stored_timestamp_unix_secs;
use axum::{
    body::Body,
    response::{IntoResponse, Response},
    Json,
};
use serde_json::json;

pub(in super::super) async fn build_admin_wallet_adjust_response(
    state: &AdminAppState<'_>,
    request_context: &AdminRequestContext<'_>,
    request_body: Option<&axum::body::Bytes>,
) -> Result<Response<Body>, GatewayError> {
    let Some(wallet_id) = admin_wallet_id_from_suffix_path(request_context.path(), "/adjust")
    else {
        return Ok(build_admin_wallets_bad_request_response("wallet_id 无效"));
    };
    let Some(request_body) = request_body else {
        return Ok(build_admin_wallets_bad_request_response("请求体不能为空"));
    };
    let payload = match serde_json::from_slice::<AdminWalletAdjustRequest>(request_body) {
        Ok(value) => value,
        Err(_) => return Ok(build_admin_wallets_bad_request_response("请求体格式无效")),
    };
    let amount_usd = match normalize_admin_wallet_non_zero_amount(payload.amount_usd, "amount_usd")
    {
        Ok(value) => value,
        Err(detail) => return Ok(build_admin_wallets_bad_request_response(detail)),
    };
    let balance_type = match normalize_admin_wallet_balance_type(payload.balance_type) {
        Ok(value) => value,
        Err(detail) => return Ok(build_admin_wallets_bad_request_response(detail)),
    };
    let description = match normalize_admin_wallet_description(payload.description) {
        Ok(value) => value,
        Err(detail) => return Ok(build_admin_wallets_bad_request_response(detail)),
    };

    let Some(existing_wallet) = state
        .find_wallet(aether_data::repository::wallet::WalletLookupKey::WalletId(
            &wallet_id,
        ))
        .await?
    else {
        return Ok(build_admin_wallet_not_found_response());
    };
    if existing_wallet.api_key_id.is_some() && balance_type == "gift" {
        return Ok(build_admin_wallets_bad_request_response(
            ADMIN_WALLETS_API_KEY_GIFT_ADJUST_DETAIL,
        ));
    }
    let operator_id = admin_wallet_operator_id(request_context);
    let has_wallet_writer = state.has_wallet_data_writer();
    let Some((wallet, Some(transaction))) = state
        .admin_adjust_wallet_balance(
            &wallet_id,
            amount_usd,
            &balance_type,
            operator_id.as_deref(),
            description.as_deref(),
            false,
        )
        .await?
    else {
        return if has_wallet_writer {
            Ok(build_admin_wallet_not_found_response())
        } else {
            Ok(build_admin_wallets_data_unavailable_response())
        };
    };
    let owner = resolve_admin_wallet_owner_summary(state, &wallet).await?;
    let wallet_payload =
        build_admin_wallet_summary_payload_with_package(state, &wallet, &owner).await?;
    let transaction_payload = build_admin_wallet_transaction_payload(
        &wallet,
        &owner,
        transaction.id,
        &transaction.category,
        &transaction.reason_code,
        transaction.amount,
        transaction.balance_before,
        transaction.balance_after,
        transaction.recharge_balance_before,
        transaction.recharge_balance_after,
        transaction.gift_balance_before,
        transaction.gift_balance_after,
        transaction.link_type.as_deref(),
        transaction.link_id.as_deref(),
        transaction.operator_id.as_deref(),
        transaction.description.as_deref(),
        unix_secs_to_rfc3339(stored_timestamp_unix_secs(transaction.created_at_unix_ms)),
    );
    let response = Json(json!({
        "wallet": wallet_payload,
        "transaction": transaction_payload,
    }))
    .into_response();
    Ok(attach_admin_audit_response(
        response,
        "admin_wallet_balance_adjusted",
        "adjust_wallet_balance",
        "wallet",
        &wallet_id,
    ))
}