GenarrativeAI/Genarrative
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

接入资产生成发布叙世币消耗
Some checks failed
CI / verify (push) Has been cancelled
Details

Browse Source
This commit is contained in:
kdletters
2026-04-27 23:32:34 +08:00
parent 1348b2e940
commit abe44948ee
11 changed files with 548 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files

121
server-rs/crates/api-server/src/big_fish.rs
View File

@@ -46,6 +46,7 @@ use crate::{
AiGenerationDraftContext, AiGenerationDraftSink, AiGenerationDraftWriter,
},
api_response::json_success_body,
asset_billing::{consume_asset_operation_points, refund_asset_operation_points},
auth::AuthenticatedAccessToken,
http_error::AppError,
request_context::RequestContext,
@@ -471,9 +472,30 @@ pub async fn execute_big_fish_action(
let owner_user_id = authenticated.claims().user_id().to_string();
let now = current_utc_micros();
let session = match payload.action.trim() {
let action = payload.action.trim().to_string();
let billed_asset_kind = match action.as_str() {
"big_fish_generate_level_main_image" => Some("big_fish_level_main_image"),
"big_fish_generate_level_motion" => Some("big_fish_level_motion"),
"big_fish_generate_stage_background" => Some("big_fish_stage_background"),
"big_fish_publish_game" => Some("big_fish_publish_game"),
_ => None,
};
let billing_asset_id = format!("{session_id}:{now}");
if let Some(asset_kind) = billed_asset_kind {
consume_asset_operation_points(&state, &owner_user_id, asset_kind, &billing_asset_id)
.await
.map_err(|error| big_fish_error_response(&request_context, error))?;
}
let session_result = match action.as_str() {
"big_fish_compile_draft" => {
compile_big_fish_draft_with_all_assets(&state, session_id, owner_user_id, now).await
compile_big_fish_draft_with_all_assets(
&state,
session_id.clone(),
owner_user_id.clone(),
now,
)
.await
}
"big_fish_generate_level_main_image" => {
let asset_url = generate_big_fish_formal_asset(
@@ -486,12 +508,30 @@ pub async fn execute_big_fish_action(
now,
)
.await
.map_err(|error| big_fish_error_response(&request_context, error))?;
.map_err(|error| {
if let Some(asset_kind) = billed_asset_kind {
tokio::spawn({
let state = state.clone();
let owner_user_id = owner_user_id.clone();
let billing_asset_id = billing_asset_id.clone();
async move {
refund_asset_operation_points(
&state,
&owner_user_id,
asset_kind,
&billing_asset_id,
)
.await;
}
});
}
big_fish_error_response(&request_context, error)
})?;
state
.spacetime_client()
.generate_big_fish_asset(BigFishAssetGenerateRecordInput {
session_id,
owner_user_id,
owner_user_id: owner_user_id.clone(),
session_id: session_id.clone(),
asset_kind: "level_main_image".to_string(),
level: payload.level,
motion_key: None,
@@ -511,12 +551,30 @@ pub async fn execute_big_fish_action(
now,
)
.await
.map_err(|error| big_fish_error_response(&request_context, error))?;
.map_err(|error| {
if let Some(asset_kind) = billed_asset_kind {
tokio::spawn({
let state = state.clone();
let owner_user_id = owner_user_id.clone();
let billing_asset_id = billing_asset_id.clone();
async move {
refund_asset_operation_points(
&state,
&owner_user_id,
asset_kind,
&billing_asset_id,
)
.await;
}
});
}
big_fish_error_response(&request_context, error)
})?;
state
.spacetime_client()
.generate_big_fish_asset(BigFishAssetGenerateRecordInput {
session_id,
owner_user_id,
owner_user_id: owner_user_id.clone(),
session_id: session_id.clone(),
asset_kind: "level_motion".to_string(),
level: payload.level,
motion_key: payload.motion_key,
@@ -536,12 +594,30 @@ pub async fn execute_big_fish_action(
now,
)
.await
.map_err(|error| big_fish_error_response(&request_context, error))?;
.map_err(|error| {
if let Some(asset_kind) = billed_asset_kind {
tokio::spawn({
let state = state.clone();
let owner_user_id = owner_user_id.clone();
let billing_asset_id = billing_asset_id.clone();
async move {
refund_asset_operation_points(
&state,
&owner_user_id,
asset_kind,
&billing_asset_id,
)
.await;
}
});
}
big_fish_error_response(&request_context, error)
})?;
state
.spacetime_client()
.generate_big_fish_asset(BigFishAssetGenerateRecordInput {
session_id,
owner_user_id,
owner_user_id: owner_user_id.clone(),
session_id: session_id.clone(),
asset_kind: "stage_background".to_string(),
level: None,
motion_key: None,
@@ -553,7 +629,7 @@ pub async fn execute_big_fish_action(
"big_fish_publish_game" => {
state
.spacetime_client()
.publish_big_fish_game(session_id, owner_user_id, now)
.publish_big_fish_game(session_id, owner_user_id.clone(), now)
.await
}
other => {
@@ -562,8 +638,25 @@ pub async fn execute_big_fish_action(
format!("action `{other}` is not supported").as_str(),
));
}
}
.map_err(|error| big_fish_error_response(&request_context, map_big_fish_client_error(error)))?;
};
let session = match session_result {
Ok(session) => session,
Err(error) => {
if let Some(asset_kind) = billed_asset_kind {
refund_asset_operation_points(
&state,
&owner_user_id,
asset_kind,
&billing_asset_id,
)
.await;
}
return Err(big_fish_error_response(
&request_context,
map_big_fish_client_error(error),
));
}
};
Ok(json_success_body(
Some(&request_context),