feat: 接入微信小程序支付
This commit is contained in:
2
server-rs/Cargo.lock
generated
2
server-rs/Cargo.lock
generated
@@ -81,6 +81,7 @@ dependencies = [
|
||||
"async-stream",
|
||||
"axum",
|
||||
"base64 0.22.1",
|
||||
"bytes",
|
||||
"dotenvy",
|
||||
"futures-util",
|
||||
"hmac",
|
||||
@@ -109,6 +110,7 @@ dependencies = [
|
||||
"platform-oss",
|
||||
"platform-speech",
|
||||
"reqwest 0.12.28",
|
||||
"ring",
|
||||
"serde",
|
||||
"serde_json",
|
||||
"sha2",
|
||||
|
||||
@@ -93,6 +93,7 @@ langchainrust = "0.2.18"
|
||||
log = "0.4"
|
||||
rand_core = "0.6"
|
||||
reqwest = { version = "0.12", default-features = false }
|
||||
ring = "0.17"
|
||||
serde = { version = "1", features = ["derive"] }
|
||||
serde_json = "1"
|
||||
serde_urlencoded = "0.7"
|
||||
|
||||
@@ -8,6 +8,7 @@ license.workspace = true
|
||||
async-stream = { workspace = true }
|
||||
axum = { workspace = true, features = ["ws"] }
|
||||
base64 = { workspace = true }
|
||||
bytes = { workspace = true }
|
||||
dotenvy = { workspace = true }
|
||||
image = { workspace = true, features = ["jpeg", "png", "webp"] }
|
||||
reqwest = { workspace = true, features = ["json", "multipart", "rustls-tls"] }
|
||||
@@ -34,6 +35,7 @@ platform-auth = { workspace = true }
|
||||
platform-llm = { workspace = true }
|
||||
platform-oss = { workspace = true }
|
||||
platform-speech = { workspace = true }
|
||||
ring = { workspace = true }
|
||||
serde = { workspace = true }
|
||||
serde_json = { workspace = true }
|
||||
shared-contracts = { workspace = true, features = ["oss-contracts"] }
|
||||
|
||||
@@ -179,6 +179,7 @@ use crate::{
|
||||
wechat_auth::{
|
||||
bind_wechat_phone, handle_wechat_callback, login_wechat_mini_program, start_wechat_login,
|
||||
},
|
||||
wechat_pay::handle_wechat_pay_notify,
|
||||
};
|
||||
|
||||
const PUZZLE_REFERENCE_IMAGE_BODY_LIMIT_BYTES: usize = 12 * 1024 * 1024;
|
||||
@@ -1410,6 +1411,10 @@ pub fn build_router(state: AppState) -> Router {
|
||||
require_bearer_auth,
|
||||
)),
|
||||
)
|
||||
.route(
|
||||
"/api/profile/recharge/wechat/notify",
|
||||
post(handle_wechat_pay_notify),
|
||||
)
|
||||
.route(
|
||||
"/api/profile/feedback",
|
||||
post(submit_profile_feedback)
|
||||
|
||||
@@ -71,6 +71,18 @@ pub struct AppConfig {
|
||||
pub wechat_mock_union_id: Option<String>,
|
||||
pub wechat_mock_display_name: String,
|
||||
pub wechat_mock_avatar_url: Option<String>,
|
||||
pub wechat_pay_enabled: bool,
|
||||
pub wechat_pay_provider: String,
|
||||
pub wechat_pay_mch_id: Option<String>,
|
||||
pub wechat_pay_merchant_serial_no: Option<String>,
|
||||
pub wechat_pay_private_key_pem: Option<String>,
|
||||
pub wechat_pay_private_key_path: Option<PathBuf>,
|
||||
pub wechat_pay_platform_public_key_pem: Option<String>,
|
||||
pub wechat_pay_platform_public_key_path: Option<PathBuf>,
|
||||
pub wechat_pay_platform_serial_no: Option<String>,
|
||||
pub wechat_pay_api_v3_key: Option<String>,
|
||||
pub wechat_pay_notify_url: Option<String>,
|
||||
pub wechat_pay_jsapi_endpoint: String,
|
||||
pub oss_bucket: Option<String>,
|
||||
pub oss_endpoint: Option<String>,
|
||||
pub oss_access_key_id: Option<String>,
|
||||
@@ -189,6 +201,19 @@ impl Default for AppConfig {
|
||||
wechat_mock_union_id: Some("wx-mock-union".to_string()),
|
||||
wechat_mock_display_name: "微信旅人".to_string(),
|
||||
wechat_mock_avatar_url: None,
|
||||
wechat_pay_enabled: false,
|
||||
wechat_pay_provider: "mock".to_string(),
|
||||
wechat_pay_mch_id: None,
|
||||
wechat_pay_merchant_serial_no: None,
|
||||
wechat_pay_private_key_pem: None,
|
||||
wechat_pay_private_key_path: None,
|
||||
wechat_pay_platform_public_key_pem: None,
|
||||
wechat_pay_platform_public_key_path: None,
|
||||
wechat_pay_platform_serial_no: None,
|
||||
wechat_pay_api_v3_key: None,
|
||||
wechat_pay_notify_url: None,
|
||||
wechat_pay_jsapi_endpoint: "https://api.mch.weixin.qq.com/v3/pay/transactions/jsapi"
|
||||
.to_string(),
|
||||
oss_bucket: None,
|
||||
oss_endpoint: None,
|
||||
oss_access_key_id: None,
|
||||
@@ -458,6 +483,33 @@ impl AppConfig {
|
||||
}
|
||||
config.wechat_mock_avatar_url = read_first_non_empty_env(&["WECHAT_MOCK_AVATAR_URL"]);
|
||||
|
||||
if let Some(wechat_pay_enabled) = read_first_bool_env(&["WECHAT_PAY_ENABLED"]) {
|
||||
config.wechat_pay_enabled = wechat_pay_enabled;
|
||||
}
|
||||
if let Some(wechat_pay_provider) = read_first_non_empty_env(&["WECHAT_PAY_PROVIDER"]) {
|
||||
config.wechat_pay_provider = wechat_pay_provider;
|
||||
}
|
||||
config.wechat_pay_mch_id = read_first_non_empty_env(&["WECHAT_PAY_MCH_ID"]);
|
||||
config.wechat_pay_merchant_serial_no =
|
||||
read_first_non_empty_env(&["WECHAT_PAY_MERCHANT_SERIAL_NO"]);
|
||||
config.wechat_pay_private_key_pem =
|
||||
read_first_non_empty_env(&["WECHAT_PAY_PRIVATE_KEY_PEM"]);
|
||||
config.wechat_pay_private_key_path =
|
||||
read_first_non_empty_env(&["WECHAT_PAY_PRIVATE_KEY_PATH"]).map(PathBuf::from);
|
||||
config.wechat_pay_platform_public_key_pem =
|
||||
read_first_non_empty_env(&["WECHAT_PAY_PLATFORM_PUBLIC_KEY_PEM"]);
|
||||
config.wechat_pay_platform_public_key_path =
|
||||
read_first_non_empty_env(&["WECHAT_PAY_PLATFORM_PUBLIC_KEY_PATH"]).map(PathBuf::from);
|
||||
config.wechat_pay_platform_serial_no =
|
||||
read_first_non_empty_env(&["WECHAT_PAY_PLATFORM_SERIAL_NO"]);
|
||||
config.wechat_pay_api_v3_key = read_first_non_empty_env(&["WECHAT_PAY_API_V3_KEY"]);
|
||||
config.wechat_pay_notify_url = read_first_non_empty_env(&["WECHAT_PAY_NOTIFY_URL"]);
|
||||
if let Some(wechat_pay_jsapi_endpoint) =
|
||||
read_first_non_empty_env(&["WECHAT_PAY_JSAPI_ENDPOINT"])
|
||||
{
|
||||
config.wechat_pay_jsapi_endpoint = wechat_pay_jsapi_endpoint;
|
||||
}
|
||||
|
||||
config.oss_bucket = read_first_non_empty_env(&["ALIYUN_OSS_BUCKET"]);
|
||||
config.oss_endpoint = read_first_non_empty_env(&["ALIYUN_OSS_ENDPOINT"]);
|
||||
config.oss_access_key_id = read_first_non_empty_env(&["ALIYUN_OSS_ACCESS_KEY_ID"]);
|
||||
@@ -1081,6 +1133,74 @@ mod tests {
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_env_reads_wechat_pay_settings() {
|
||||
let _guard = ENV_LOCK
|
||||
.get_or_init(|| Mutex::new(()))
|
||||
.lock()
|
||||
.expect("env lock should not poison");
|
||||
|
||||
unsafe {
|
||||
std::env::remove_var("WECHAT_PAY_ENABLED");
|
||||
std::env::remove_var("WECHAT_PAY_PROVIDER");
|
||||
std::env::remove_var("WECHAT_PAY_MCH_ID");
|
||||
std::env::remove_var("WECHAT_PAY_MERCHANT_SERIAL_NO");
|
||||
std::env::remove_var("WECHAT_PAY_PRIVATE_KEY_PATH");
|
||||
std::env::remove_var("WECHAT_PAY_PLATFORM_PUBLIC_KEY_PATH");
|
||||
std::env::remove_var("WECHAT_PAY_PLATFORM_SERIAL_NO");
|
||||
std::env::remove_var("WECHAT_PAY_API_V3_KEY");
|
||||
std::env::remove_var("WECHAT_PAY_NOTIFY_URL");
|
||||
std::env::set_var("WECHAT_PAY_ENABLED", "true");
|
||||
std::env::set_var("WECHAT_PAY_PROVIDER", "real");
|
||||
std::env::set_var("WECHAT_PAY_MCH_ID", "1900000109");
|
||||
std::env::set_var("WECHAT_PAY_MERCHANT_SERIAL_NO", "serial-001");
|
||||
std::env::set_var("WECHAT_PAY_PRIVATE_KEY_PATH", "certs/apiclient_key.pem");
|
||||
std::env::set_var(
|
||||
"WECHAT_PAY_PLATFORM_PUBLIC_KEY_PATH",
|
||||
"certs/wechatpay_platform.pem",
|
||||
);
|
||||
std::env::set_var("WECHAT_PAY_PLATFORM_SERIAL_NO", "platform-serial-001");
|
||||
std::env::set_var("WECHAT_PAY_API_V3_KEY", "12345678901234567890123456789012");
|
||||
std::env::set_var(
|
||||
"WECHAT_PAY_NOTIFY_URL",
|
||||
"https://api.example.com/api/profile/recharge/wechat/notify",
|
||||
);
|
||||
}
|
||||
|
||||
let config = AppConfig::from_env();
|
||||
assert!(config.wechat_pay_enabled);
|
||||
assert_eq!(config.wechat_pay_provider, "real");
|
||||
assert_eq!(config.wechat_pay_mch_id.as_deref(), Some("1900000109"));
|
||||
assert_eq!(
|
||||
config.wechat_pay_private_key_path.as_deref(),
|
||||
Some(std::path::Path::new("certs/apiclient_key.pem"))
|
||||
);
|
||||
assert_eq!(
|
||||
config.wechat_pay_notify_url.as_deref(),
|
||||
Some("https://api.example.com/api/profile/recharge/wechat/notify")
|
||||
);
|
||||
assert_eq!(
|
||||
config.wechat_pay_platform_public_key_path.as_deref(),
|
||||
Some(std::path::Path::new("certs/wechatpay_platform.pem"))
|
||||
);
|
||||
assert_eq!(
|
||||
config.wechat_pay_platform_serial_no.as_deref(),
|
||||
Some("platform-serial-001")
|
||||
);
|
||||
|
||||
unsafe {
|
||||
std::env::remove_var("WECHAT_PAY_ENABLED");
|
||||
std::env::remove_var("WECHAT_PAY_PROVIDER");
|
||||
std::env::remove_var("WECHAT_PAY_MCH_ID");
|
||||
std::env::remove_var("WECHAT_PAY_MERCHANT_SERIAL_NO");
|
||||
std::env::remove_var("WECHAT_PAY_PRIVATE_KEY_PATH");
|
||||
std::env::remove_var("WECHAT_PAY_PLATFORM_PUBLIC_KEY_PATH");
|
||||
std::env::remove_var("WECHAT_PAY_PLATFORM_SERIAL_NO");
|
||||
std::env::remove_var("WECHAT_PAY_API_V3_KEY");
|
||||
std::env::remove_var("WECHAT_PAY_NOTIFY_URL");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_env_ignores_zero_spacetime_pool_size() {
|
||||
let _guard = ENV_LOCK
|
||||
|
||||
@@ -75,6 +75,7 @@ mod vector_engine_audio_generation;
|
||||
mod visual_novel;
|
||||
mod volcengine_speech;
|
||||
mod wechat_auth;
|
||||
mod wechat_pay;
|
||||
mod wechat_provider;
|
||||
mod work_author;
|
||||
mod work_play_tracking;
|
||||
|
||||
@@ -6,15 +6,16 @@ use axum::{
|
||||
};
|
||||
use module_runtime::{
|
||||
AnalyticsGranularity, PROFILE_RECHARGE_PAYMENT_CHANNEL_MOCK,
|
||||
RuntimeProfileFeedbackEvidenceRecord, RuntimeProfileFeedbackEvidenceSnapshot,
|
||||
RuntimeProfileFeedbackSubmissionRecord, RuntimeProfileInviteCodeRecord,
|
||||
RuntimeProfileMembershipBenefitRecord, RuntimeProfileRechargeCenterRecord,
|
||||
RuntimeProfileRechargeOrderRecord, RuntimeProfileRechargeProductRecord,
|
||||
RuntimeProfileRedeemCodeMode, RuntimeProfileRedeemCodeRecord,
|
||||
RuntimeProfileRewardCodeRedeemRecord, RuntimeProfileTaskCenterRecord,
|
||||
RuntimeProfileTaskClaimRecord, RuntimeProfileTaskConfigRecord, RuntimeProfileTaskCycle,
|
||||
RuntimeProfileTaskItemRecord, RuntimeProfileTaskStatus, RuntimeProfileWalletLedgerSourceType,
|
||||
RuntimeReferralInviteCenterRecord, RuntimeTrackingScopeKind,
|
||||
PROFILE_RECHARGE_PAYMENT_CHANNEL_WECHAT_MINI_PROGRAM, RuntimeProfileFeedbackEvidenceRecord,
|
||||
RuntimeProfileFeedbackEvidenceSnapshot, RuntimeProfileFeedbackSubmissionRecord,
|
||||
RuntimeProfileInviteCodeRecord, RuntimeProfileMembershipBenefitRecord,
|
||||
RuntimeProfileRechargeCenterRecord, RuntimeProfileRechargeOrderRecord,
|
||||
RuntimeProfileRechargeProductRecord, RuntimeProfileRedeemCodeMode,
|
||||
RuntimeProfileRedeemCodeRecord, RuntimeProfileRewardCodeRedeemRecord,
|
||||
RuntimeProfileTaskCenterRecord, RuntimeProfileTaskClaimRecord, RuntimeProfileTaskConfigRecord,
|
||||
RuntimeProfileTaskCycle, RuntimeProfileTaskItemRecord, RuntimeProfileTaskStatus,
|
||||
RuntimeProfileWalletLedgerSourceType, RuntimeReferralInviteCenterRecord,
|
||||
RuntimeTrackingScopeKind,
|
||||
};
|
||||
use serde::Deserialize;
|
||||
use serde_json::{Value, json};
|
||||
@@ -56,8 +57,13 @@ use spacetime_client::SpacetimeClientError;
|
||||
use time::OffsetDateTime;
|
||||
|
||||
use crate::{
|
||||
admin::AuthenticatedAdmin, api_response::json_success_body, auth::AuthenticatedAccessToken,
|
||||
http_error::AppError, request_context::RequestContext, state::AppState,
|
||||
admin::AuthenticatedAdmin,
|
||||
api_response::json_success_body,
|
||||
auth::AuthenticatedAccessToken,
|
||||
http_error::AppError,
|
||||
request_context::RequestContext,
|
||||
state::AppState,
|
||||
wechat_pay::{build_wechat_payment_request, current_unix_micros, map_wechat_pay_error},
|
||||
};
|
||||
|
||||
pub async fn get_profile_dashboard(
|
||||
@@ -186,14 +192,15 @@ pub async fn create_profile_recharge_order(
|
||||
let payment_channel = payload
|
||||
.payment_channel
|
||||
.unwrap_or_else(|| PROFILE_RECHARGE_PAYMENT_CHANNEL_MOCK.to_string());
|
||||
let created_at_micros = OffsetDateTime::now_utc().unix_timestamp_nanos() / 1_000;
|
||||
let payment_channel = payment_channel.trim().to_string();
|
||||
let created_at_micros = current_unix_micros();
|
||||
let (center, order) = state
|
||||
.spacetime_client()
|
||||
.create_profile_recharge_order(
|
||||
user_id,
|
||||
payload.product_id,
|
||||
payment_channel,
|
||||
created_at_micros as i64,
|
||||
payment_channel.clone(),
|
||||
created_at_micros,
|
||||
)
|
||||
.await
|
||||
.map_err(|error| {
|
||||
@@ -203,11 +210,36 @@ pub async fn create_profile_recharge_order(
|
||||
)
|
||||
})?;
|
||||
|
||||
let wechat_mini_program_pay_params = if payment_channel
|
||||
== PROFILE_RECHARGE_PAYMENT_CHANNEL_WECHAT_MINI_PROGRAM
|
||||
{
|
||||
let identity = resolve_wechat_identity_for_payment(&state, &order.user_id)
|
||||
.await
|
||||
.map_err(|error| runtime_profile_error_response(&request_context, error))?;
|
||||
Some(
|
||||
state
|
||||
.wechat_pay_client()
|
||||
.create_mini_program_order(build_wechat_payment_request(
|
||||
order.order_id.clone(),
|
||||
order.product_title.clone(),
|
||||
order.amount_cents,
|
||||
identity,
|
||||
))
|
||||
.await
|
||||
.map_err(|error| {
|
||||
runtime_profile_error_response(&request_context, map_wechat_pay_error(error))
|
||||
})?,
|
||||
)
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
Ok(json_success_body(
|
||||
Some(&request_context),
|
||||
CreateProfileRechargeOrderResponse {
|
||||
order: build_profile_recharge_order_response(order),
|
||||
center: build_profile_recharge_center_response(center),
|
||||
wechat_mini_program_pay_params,
|
||||
},
|
||||
))
|
||||
}
|
||||
@@ -750,6 +782,25 @@ fn runtime_profile_error_response(request_context: &RequestContext, error: AppEr
|
||||
error.into_response_with_context(Some(request_context))
|
||||
}
|
||||
|
||||
async fn resolve_wechat_identity_for_payment(
|
||||
state: &AppState,
|
||||
user_id: &str,
|
||||
) -> Result<String, AppError> {
|
||||
if let Some(identity) = state
|
||||
.wechat_auth_service()
|
||||
.get_identity_by_user_id(user_id)
|
||||
.map_err(|error| {
|
||||
AppError::from_status(StatusCode::INTERNAL_SERVER_ERROR)
|
||||
.with_message(format!("读取微信身份失败:{error}"))
|
||||
})?
|
||||
{
|
||||
return Ok(identity.provider_uid);
|
||||
}
|
||||
|
||||
Err(AppError::from_status(StatusCode::BAD_REQUEST)
|
||||
.with_message("当前账号缺少微信小程序身份,请在小程序内重新登录后再支付"))
|
||||
}
|
||||
|
||||
fn build_profile_recharge_center_response(
|
||||
record: RuntimeProfileRechargeCenterRecord,
|
||||
) -> ProfileRechargeCenterResponse {
|
||||
@@ -825,6 +876,7 @@ fn build_profile_recharge_order_response(
|
||||
status: record.status.as_str().to_string(),
|
||||
payment_channel: record.payment_channel,
|
||||
paid_at: record.paid_at,
|
||||
provider_transaction_id: record.provider_transaction_id,
|
||||
created_at: record.created_at,
|
||||
points_delta: record.points_delta,
|
||||
membership_expires_at: record.membership_expires_at,
|
||||
|
||||
@@ -30,6 +30,7 @@ use time::OffsetDateTime;
|
||||
use tracing::{info, warn};
|
||||
|
||||
use crate::config::AppConfig;
|
||||
use crate::wechat_pay::{WechatPayClient, map_wechat_pay_init_error};
|
||||
use crate::wechat_provider::build_wechat_provider;
|
||||
|
||||
const ADMIN_ROLE: &str = "admin";
|
||||
@@ -55,6 +56,7 @@ pub struct AppState {
|
||||
wechat_auth_state_service: WechatAuthStateService,
|
||||
wechat_auth_service: WechatAuthService,
|
||||
wechat_provider: WechatProvider,
|
||||
wechat_pay_client: WechatPayClient,
|
||||
#[cfg_attr(not(test), allow(dead_code))]
|
||||
ai_task_service: AiTaskService,
|
||||
spacetime_client: SpacetimeClient,
|
||||
@@ -110,6 +112,7 @@ pub enum AppStateInitError {
|
||||
RefreshCookie(RefreshCookieError),
|
||||
AuthStore(String),
|
||||
SmsProvider(SmsProviderError),
|
||||
WechatPay(String),
|
||||
Oss(OssError),
|
||||
Llm(LlmError),
|
||||
}
|
||||
@@ -174,6 +177,8 @@ impl AppState {
|
||||
WechatAuthStateService::new(auth_store.clone(), config.wechat_state_ttl_minutes);
|
||||
let wechat_auth_service = WechatAuthService::new(auth_store.clone());
|
||||
let wechat_provider = build_wechat_provider(&config);
|
||||
let wechat_pay_client =
|
||||
WechatPayClient::from_config(&config).map_err(map_wechat_pay_init_error)?;
|
||||
let refresh_session_service =
|
||||
RefreshSessionService::new(auth_store.clone(), config.refresh_session_ttl_days);
|
||||
// AI 编排服务当前先挂接内存态 store,后续再按 task table / procedure 接到 SpacetimeDB 真相源。
|
||||
@@ -206,6 +211,7 @@ impl AppState {
|
||||
wechat_auth_state_service,
|
||||
wechat_auth_service,
|
||||
wechat_provider,
|
||||
wechat_pay_client,
|
||||
ai_task_service,
|
||||
spacetime_client,
|
||||
llm_client,
|
||||
@@ -454,6 +460,10 @@ impl AppState {
|
||||
&self.wechat_provider
|
||||
}
|
||||
|
||||
pub fn wechat_pay_client(&self) -> &WechatPayClient {
|
||||
&self.wechat_pay_client
|
||||
}
|
||||
|
||||
#[cfg_attr(not(test), allow(dead_code))]
|
||||
pub fn ai_task_service(&self) -> &AiTaskService {
|
||||
&self.ai_task_service
|
||||
@@ -860,7 +870,7 @@ impl fmt::Display for AppStateInitError {
|
||||
match self {
|
||||
Self::Jwt(error) => write!(f, "{error}"),
|
||||
Self::RefreshCookie(error) => write!(f, "{error}"),
|
||||
Self::AuthStore(error) => write!(f, "{error}"),
|
||||
Self::AuthStore(error) | Self::WechatPay(error) => write!(f, "{error}"),
|
||||
Self::SmsProvider(error) => write!(f, "{error}"),
|
||||
Self::Oss(error) => write!(f, "{error}"),
|
||||
Self::Llm(error) => write!(f, "{error}"),
|
||||
|
||||
780
server-rs/crates/api-server/src/wechat_pay.rs
Normal file
780
server-rs/crates/api-server/src/wechat_pay.rs
Normal file
@@ -0,0 +1,780 @@
|
||||
use std::{fs, path::Path, sync::Arc};
|
||||
|
||||
use axum::{
|
||||
extract::State,
|
||||
http::{HeaderMap, StatusCode},
|
||||
};
|
||||
use base64::{Engine as _, engine::general_purpose::STANDARD as BASE64_STANDARD};
|
||||
use bytes::Bytes;
|
||||
use ring::{
|
||||
aead,
|
||||
rand::{SecureRandom, SystemRandom},
|
||||
signature,
|
||||
};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use serde_json::{Value, json};
|
||||
use sha2::{Digest, Sha256};
|
||||
use shared_contracts::runtime::WechatMiniProgramPayParamsResponse;
|
||||
use shared_kernel::offset_datetime_to_unix_micros;
|
||||
use time::OffsetDateTime;
|
||||
use tracing::{info, warn};
|
||||
|
||||
use crate::{http_error::AppError, state::AppState};
|
||||
|
||||
const WECHAT_PAY_PROVIDER_MOCK: &str = "mock";
|
||||
const WECHAT_PAY_PROVIDER_REAL: &str = "real";
|
||||
const WECHAT_PAY_BODY_SIGNATURE_METHOD: &str = "WECHATPAY2-SHA256-RSA2048";
|
||||
const WECHAT_PAY_PAY_SIGN_TYPE: &str = "RSA";
|
||||
const WECHAT_PAY_NOTIFY_SUCCESS: &str = "<xml><return_code><![CDATA[SUCCESS]]></return_code></xml>";
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub enum WechatPayClient {
|
||||
Disabled,
|
||||
Mock,
|
||||
Real(Arc<RealWechatPayClient>),
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct RealWechatPayClient {
|
||||
client: reqwest::Client,
|
||||
app_id: String,
|
||||
mch_id: String,
|
||||
merchant_serial_no: String,
|
||||
private_key: Arc<signature::RsaKeyPair>,
|
||||
platform_public_key_der: Vec<u8>,
|
||||
platform_serial_no: String,
|
||||
api_v3_key: String,
|
||||
notify_url: String,
|
||||
jsapi_endpoint: String,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct WechatMiniProgramOrderRequest {
|
||||
pub order_id: String,
|
||||
pub description: String,
|
||||
pub amount_cents: u64,
|
||||
pub payer_openid: String,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct WechatPayNotifyOrder {
|
||||
pub out_trade_no: String,
|
||||
pub transaction_id: Option<String>,
|
||||
pub trade_state: String,
|
||||
pub success_time: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum WechatPayError {
|
||||
Disabled,
|
||||
InvalidConfig(String),
|
||||
InvalidRequest(String),
|
||||
RequestFailed(String),
|
||||
Upstream(String),
|
||||
Deserialize(String),
|
||||
Crypto(String),
|
||||
InvalidSignature,
|
||||
}
|
||||
|
||||
#[derive(Serialize)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
struct WechatJsapiOrderRequest<'a> {
|
||||
appid: &'a str,
|
||||
mchid: &'a str,
|
||||
description: &'a str,
|
||||
out_trade_no: &'a str,
|
||||
notify_url: &'a str,
|
||||
amount: WechatJsapiAmount,
|
||||
payer: WechatJsapiPayer<'a>,
|
||||
}
|
||||
|
||||
#[derive(Serialize)]
|
||||
struct WechatJsapiAmount {
|
||||
total: i64,
|
||||
currency: &'static str,
|
||||
}
|
||||
|
||||
#[derive(Serialize)]
|
||||
struct WechatJsapiPayer<'a> {
|
||||
openid: &'a str,
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct WechatJsapiOrderResponse {
|
||||
prepay_id: Option<String>,
|
||||
code: Option<String>,
|
||||
message: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct WechatPayNotifyBody {
|
||||
#[serde(default)]
|
||||
resource: Option<WechatPayNotifyResource>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct WechatPayNotifyResource {
|
||||
ciphertext: String,
|
||||
nonce: String,
|
||||
#[serde(default)]
|
||||
associated_data: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct WechatPayTransactionResource {
|
||||
out_trade_no: String,
|
||||
#[serde(default)]
|
||||
transaction_id: Option<String>,
|
||||
trade_state: String,
|
||||
#[serde(default)]
|
||||
success_time: Option<String>,
|
||||
}
|
||||
|
||||
impl WechatPayClient {
|
||||
pub fn from_config(config: &crate::config::AppConfig) -> Result<Self, WechatPayError> {
|
||||
if !config.wechat_pay_enabled {
|
||||
return Ok(Self::Disabled);
|
||||
}
|
||||
|
||||
if config
|
||||
.wechat_pay_provider
|
||||
.trim()
|
||||
.eq_ignore_ascii_case(WECHAT_PAY_PROVIDER_MOCK)
|
||||
{
|
||||
return Ok(Self::Mock);
|
||||
}
|
||||
|
||||
if !config
|
||||
.wechat_pay_provider
|
||||
.trim()
|
||||
.eq_ignore_ascii_case(WECHAT_PAY_PROVIDER_REAL)
|
||||
{
|
||||
return Err(WechatPayError::InvalidConfig(
|
||||
"WECHAT_PAY_PROVIDER 仅支持 mock 或 real".to_string(),
|
||||
));
|
||||
}
|
||||
|
||||
let app_id = config
|
||||
.wechat_mini_program_app_id
|
||||
.as_ref()
|
||||
.or(config.wechat_app_id.as_ref())
|
||||
.map(|value| value.trim())
|
||||
.filter(|value| !value.is_empty())
|
||||
.ok_or_else(|| WechatPayError::InvalidConfig("微信支付缺少小程序 AppID".to_string()))?
|
||||
.to_string();
|
||||
let mch_id = required_config(config.wechat_pay_mch_id.as_deref(), "WECHAT_PAY_MCH_ID")?;
|
||||
let merchant_serial_no = required_config(
|
||||
config.wechat_pay_merchant_serial_no.as_deref(),
|
||||
"WECHAT_PAY_MERCHANT_SERIAL_NO",
|
||||
)?;
|
||||
let private_key_pem = read_private_key_pem(
|
||||
config.wechat_pay_private_key_pem.as_deref(),
|
||||
config.wechat_pay_private_key_path.as_deref(),
|
||||
)?;
|
||||
let private_key = Arc::new(parse_rsa_private_key(&private_key_pem)?);
|
||||
let platform_public_key_pem = read_pem(
|
||||
config.wechat_pay_platform_public_key_pem.as_deref(),
|
||||
config.wechat_pay_platform_public_key_path.as_deref(),
|
||||
"WECHAT_PAY_PLATFORM_PUBLIC_KEY_PEM 或 WECHAT_PAY_PLATFORM_PUBLIC_KEY_PATH 未配置",
|
||||
"读取微信支付平台公钥失败",
|
||||
)?;
|
||||
let platform_public_key_der = parse_public_key_pem(&platform_public_key_pem)?;
|
||||
let platform_serial_no = required_config(
|
||||
config.wechat_pay_platform_serial_no.as_deref(),
|
||||
"WECHAT_PAY_PLATFORM_SERIAL_NO",
|
||||
)?;
|
||||
let api_v3_key = required_config(
|
||||
config.wechat_pay_api_v3_key.as_deref(),
|
||||
"WECHAT_PAY_API_V3_KEY",
|
||||
)?;
|
||||
if api_v3_key.as_bytes().len() != 32 {
|
||||
return Err(WechatPayError::InvalidConfig(
|
||||
"WECHAT_PAY_API_V3_KEY 必须是 32 字节字符串".to_string(),
|
||||
));
|
||||
}
|
||||
let notify_url = required_config(
|
||||
config.wechat_pay_notify_url.as_deref(),
|
||||
"WECHAT_PAY_NOTIFY_URL",
|
||||
)?;
|
||||
let jsapi_endpoint = normalize_required_url(
|
||||
&config.wechat_pay_jsapi_endpoint,
|
||||
"WECHAT_PAY_JSAPI_ENDPOINT",
|
||||
)?;
|
||||
|
||||
Ok(Self::Real(Arc::new(RealWechatPayClient {
|
||||
client: reqwest::Client::new(),
|
||||
app_id,
|
||||
mch_id,
|
||||
merchant_serial_no,
|
||||
private_key,
|
||||
platform_public_key_der,
|
||||
platform_serial_no,
|
||||
api_v3_key,
|
||||
notify_url,
|
||||
jsapi_endpoint,
|
||||
})))
|
||||
}
|
||||
|
||||
pub async fn create_mini_program_order(
|
||||
&self,
|
||||
request: WechatMiniProgramOrderRequest,
|
||||
) -> Result<WechatMiniProgramPayParamsResponse, WechatPayError> {
|
||||
match self {
|
||||
Self::Disabled => Err(WechatPayError::Disabled),
|
||||
Self::Mock => Ok(build_mock_pay_params(&request.order_id)),
|
||||
Self::Real(client) => client.create_mini_program_order(request).await,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn parse_notify(
|
||||
&self,
|
||||
headers: &HeaderMap,
|
||||
body: &[u8],
|
||||
) -> Result<WechatPayNotifyOrder, WechatPayError> {
|
||||
match self {
|
||||
Self::Disabled => Err(WechatPayError::Disabled),
|
||||
Self::Mock => parse_mock_notify(body),
|
||||
Self::Real(client) => client.parse_notify(headers, body),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl RealWechatPayClient {
|
||||
async fn create_mini_program_order(
|
||||
&self,
|
||||
request: WechatMiniProgramOrderRequest,
|
||||
) -> Result<WechatMiniProgramPayParamsResponse, WechatPayError> {
|
||||
let amount_total = i64::try_from(request.amount_cents)
|
||||
.map_err(|_| WechatPayError::InvalidRequest("微信支付金额超出 i64 范围".to_string()))?;
|
||||
let body = serde_json::to_string(&WechatJsapiOrderRequest {
|
||||
appid: &self.app_id,
|
||||
mchid: &self.mch_id,
|
||||
description: &request.description,
|
||||
out_trade_no: &request.order_id,
|
||||
notify_url: &self.notify_url,
|
||||
amount: WechatJsapiAmount {
|
||||
total: amount_total,
|
||||
currency: "CNY",
|
||||
},
|
||||
payer: WechatJsapiPayer {
|
||||
openid: &request.payer_openid,
|
||||
},
|
||||
})
|
||||
.map_err(|error| WechatPayError::Deserialize(format!("微信支付请求序列化失败:{error}")))?;
|
||||
let timestamp = OffsetDateTime::now_utc().unix_timestamp().to_string();
|
||||
let nonce = create_nonce()?;
|
||||
let authorization = self.build_authorization(
|
||||
"POST",
|
||||
"/v3/pay/transactions/jsapi",
|
||||
×tamp,
|
||||
&nonce,
|
||||
&body,
|
||||
)?;
|
||||
let response = self
|
||||
.client
|
||||
.post(&self.jsapi_endpoint)
|
||||
.header("Authorization", authorization)
|
||||
.header("Accept", "application/json")
|
||||
.header("Content-Type", "application/json")
|
||||
.body(body)
|
||||
.send()
|
||||
.await
|
||||
.map_err(|error| {
|
||||
WechatPayError::RequestFailed(format!("微信支付 JSAPI 下单请求失败:{error}"))
|
||||
})?;
|
||||
let status = response.status();
|
||||
let response_text = response.text().await.map_err(|error| {
|
||||
WechatPayError::Deserialize(format!("微信支付 JSAPI 下单响应读取失败:{error}"))
|
||||
})?;
|
||||
let payload =
|
||||
serde_json::from_str::<WechatJsapiOrderResponse>(&response_text).map_err(|error| {
|
||||
WechatPayError::Deserialize(format!("微信支付 JSAPI 下单响应解析失败:{error}"))
|
||||
})?;
|
||||
|
||||
if !status.is_success() {
|
||||
return Err(WechatPayError::Upstream(format!(
|
||||
"微信支付 JSAPI 下单失败:{}",
|
||||
payload
|
||||
.message
|
||||
.or(payload.code)
|
||||
.unwrap_or_else(|| format!("HTTP {status}"))
|
||||
)));
|
||||
}
|
||||
|
||||
let prepay_id = payload
|
||||
.prepay_id
|
||||
.map(|value| value.trim().to_string())
|
||||
.filter(|value| !value.is_empty())
|
||||
.ok_or_else(|| WechatPayError::Upstream("微信支付未返回 prepay_id".to_string()))?;
|
||||
self.build_pay_params(&prepay_id)
|
||||
}
|
||||
|
||||
fn build_authorization(
|
||||
&self,
|
||||
method: &str,
|
||||
canonical_url: &str,
|
||||
timestamp: &str,
|
||||
nonce: &str,
|
||||
body: &str,
|
||||
) -> Result<String, WechatPayError> {
|
||||
let message = format!("{method}\n{canonical_url}\n{timestamp}\n{nonce}\n{body}\n");
|
||||
let signature = self.sign_message(&message)?;
|
||||
Ok(format!(
|
||||
"{WECHAT_PAY_BODY_SIGNATURE_METHOD} mchid=\"{}\",nonce_str=\"{}\",timestamp=\"{}\",serial_no=\"{}\",signature=\"{}\"",
|
||||
self.mch_id, nonce, timestamp, self.merchant_serial_no, signature
|
||||
))
|
||||
}
|
||||
|
||||
fn build_pay_params(
|
||||
&self,
|
||||
prepay_id: &str,
|
||||
) -> Result<WechatMiniProgramPayParamsResponse, WechatPayError> {
|
||||
let time_stamp = OffsetDateTime::now_utc().unix_timestamp().to_string();
|
||||
let nonce_str = create_nonce()?;
|
||||
let package = format!("prepay_id={prepay_id}");
|
||||
let message = format!(
|
||||
"{}\n{}\n{}\n{}\n",
|
||||
self.app_id, time_stamp, nonce_str, package
|
||||
);
|
||||
let pay_sign = self.sign_message(&message)?;
|
||||
|
||||
Ok(WechatMiniProgramPayParamsResponse {
|
||||
time_stamp,
|
||||
nonce_str,
|
||||
package,
|
||||
sign_type: WECHAT_PAY_PAY_SIGN_TYPE.to_string(),
|
||||
pay_sign,
|
||||
})
|
||||
}
|
||||
|
||||
fn parse_notify(
|
||||
&self,
|
||||
headers: &HeaderMap,
|
||||
body: &[u8],
|
||||
) -> Result<WechatPayNotifyOrder, WechatPayError> {
|
||||
self.verify_notify_signature(headers, body)?;
|
||||
let notify = serde_json::from_slice::<WechatPayNotifyBody>(body).map_err(|error| {
|
||||
WechatPayError::Deserialize(format!("微信支付通知解析失败:{error}"))
|
||||
})?;
|
||||
let resource = notify.resource.ok_or_else(|| {
|
||||
WechatPayError::InvalidRequest("微信支付通知缺少 resource".to_string())
|
||||
})?;
|
||||
let plain_text = decrypt_aes_256_gcm(
|
||||
self.api_v3_key.as_bytes(),
|
||||
resource.nonce.as_bytes(),
|
||||
resource.associated_data.as_deref().unwrap_or("").as_bytes(),
|
||||
resource.ciphertext.as_str(),
|
||||
)?;
|
||||
let transaction = serde_json::from_slice::<WechatPayTransactionResource>(&plain_text)
|
||||
.map_err(|error| {
|
||||
WechatPayError::Deserialize(format!("微信支付通知资源解析失败:{error}"))
|
||||
})?;
|
||||
|
||||
Ok(WechatPayNotifyOrder {
|
||||
out_trade_no: transaction.out_trade_no,
|
||||
transaction_id: transaction
|
||||
.transaction_id
|
||||
.map(|value| value.trim().to_string())
|
||||
.filter(|value| !value.is_empty()),
|
||||
trade_state: transaction.trade_state,
|
||||
success_time: transaction.success_time,
|
||||
})
|
||||
}
|
||||
|
||||
fn verify_notify_signature(
|
||||
&self,
|
||||
headers: &HeaderMap,
|
||||
body: &[u8],
|
||||
) -> Result<(), WechatPayError> {
|
||||
let timestamp = read_required_header(headers, "Wechatpay-Timestamp")?;
|
||||
let nonce = read_required_header(headers, "Wechatpay-Nonce")?;
|
||||
let signature = read_required_header(headers, "Wechatpay-Signature")?;
|
||||
let serial = read_required_header(headers, "Wechatpay-Serial")?;
|
||||
if serial != self.platform_serial_no {
|
||||
return Err(WechatPayError::InvalidSignature);
|
||||
}
|
||||
|
||||
let message = format!(
|
||||
"{}\n{}\n{}\n",
|
||||
timestamp,
|
||||
nonce,
|
||||
String::from_utf8_lossy(body)
|
||||
);
|
||||
let signature_bytes = BASE64_STANDARD
|
||||
.decode(signature)
|
||||
.map_err(|_| WechatPayError::InvalidSignature)?;
|
||||
let public_key = signature::UnparsedPublicKey::new(
|
||||
&signature::RSA_PKCS1_2048_8192_SHA256,
|
||||
&self.platform_public_key_der,
|
||||
);
|
||||
public_key
|
||||
.verify(message.as_bytes(), &signature_bytes)
|
||||
.map_err(|_| WechatPayError::InvalidSignature)
|
||||
}
|
||||
|
||||
fn sign_message(&self, message: &str) -> Result<String, WechatPayError> {
|
||||
let rng = SystemRandom::new();
|
||||
let mut signature = vec![0_u8; self.private_key.public().modulus_len()];
|
||||
self.private_key
|
||||
.sign(
|
||||
&signature::RSA_PKCS1_SHA256,
|
||||
&rng,
|
||||
message.as_bytes(),
|
||||
&mut signature,
|
||||
)
|
||||
.map_err(|_| WechatPayError::Crypto("微信支付签名失败".to_string()))?;
|
||||
Ok(BASE64_STANDARD.encode(signature))
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn handle_wechat_pay_notify(
|
||||
State(state): State<AppState>,
|
||||
headers: HeaderMap,
|
||||
body: Bytes,
|
||||
) -> Result<&'static str, AppError> {
|
||||
let notify = state
|
||||
.wechat_pay_client()
|
||||
.parse_notify(&headers, &body)
|
||||
.map_err(map_wechat_pay_notify_error)?;
|
||||
if notify.trade_state != "SUCCESS" {
|
||||
info!(
|
||||
order_id = notify.out_trade_no.as_str(),
|
||||
trade_state = notify.trade_state.as_str(),
|
||||
"收到非成功微信支付通知"
|
||||
);
|
||||
return Ok(WECHAT_PAY_NOTIFY_SUCCESS);
|
||||
}
|
||||
|
||||
let paid_at_micros = notify
|
||||
.success_time
|
||||
.as_deref()
|
||||
.and_then(|value| shared_kernel::parse_rfc3339(value).ok())
|
||||
.map(offset_datetime_to_unix_micros)
|
||||
.unwrap_or_else(current_unix_micros);
|
||||
|
||||
state
|
||||
.spacetime_client()
|
||||
.mark_profile_recharge_order_paid(
|
||||
notify.out_trade_no.clone(),
|
||||
paid_at_micros,
|
||||
notify.transaction_id.clone(),
|
||||
)
|
||||
.await
|
||||
.map_err(|error| {
|
||||
AppError::from_status(StatusCode::BAD_GATEWAY)
|
||||
.with_message(format!("确认微信支付订单失败:{error}"))
|
||||
})?;
|
||||
info!(
|
||||
order_id = notify.out_trade_no.as_str(),
|
||||
"微信支付通知已确认订单入账"
|
||||
);
|
||||
|
||||
Ok(WECHAT_PAY_NOTIFY_SUCCESS)
|
||||
}
|
||||
|
||||
pub fn map_wechat_pay_error(error: WechatPayError) -> AppError {
|
||||
match error {
|
||||
WechatPayError::Disabled => AppError::from_status(StatusCode::BAD_REQUEST)
|
||||
.with_message("微信支付暂未启用")
|
||||
.with_details(json!({ "provider": "wechat_pay" })),
|
||||
WechatPayError::InvalidConfig(message) => {
|
||||
AppError::from_status(StatusCode::SERVICE_UNAVAILABLE)
|
||||
.with_message(message)
|
||||
.with_details(json!({ "provider": "wechat_pay" }))
|
||||
}
|
||||
WechatPayError::InvalidRequest(message) => AppError::from_status(StatusCode::BAD_REQUEST)
|
||||
.with_message(message)
|
||||
.with_details(json!({ "provider": "wechat_pay" })),
|
||||
WechatPayError::RequestFailed(message)
|
||||
| WechatPayError::Upstream(message)
|
||||
| WechatPayError::Deserialize(message)
|
||||
| WechatPayError::Crypto(message) => AppError::from_status(StatusCode::BAD_GATEWAY)
|
||||
.with_message(message)
|
||||
.with_details(json!({ "provider": "wechat_pay" })),
|
||||
WechatPayError::InvalidSignature => AppError::from_status(StatusCode::UNAUTHORIZED)
|
||||
.with_message("微信支付通知签名无效")
|
||||
.with_details(json!({ "provider": "wechat_pay" })),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn map_wechat_pay_init_error(error: WechatPayError) -> crate::state::AppStateInitError {
|
||||
crate::state::AppStateInitError::WechatPay(error.to_string())
|
||||
}
|
||||
|
||||
pub fn build_wechat_payment_request(
|
||||
order_id: String,
|
||||
product_title: String,
|
||||
amount_cents: u64,
|
||||
payer_openid: String,
|
||||
) -> WechatMiniProgramOrderRequest {
|
||||
WechatMiniProgramOrderRequest {
|
||||
order_id,
|
||||
description: format!("百梦 - {product_title}"),
|
||||
amount_cents,
|
||||
payer_openid,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn current_unix_micros() -> i64 {
|
||||
let value = OffsetDateTime::now_utc().unix_timestamp_nanos() / 1_000;
|
||||
i64::try_from(value).unwrap_or(i64::MAX)
|
||||
}
|
||||
|
||||
fn map_wechat_pay_notify_error(error: WechatPayError) -> AppError {
|
||||
warn!(error = %error, "微信支付通知处理失败");
|
||||
map_wechat_pay_error(error)
|
||||
}
|
||||
|
||||
fn build_mock_pay_params(order_id: &str) -> WechatMiniProgramPayParamsResponse {
|
||||
let time_stamp = OffsetDateTime::now_utc().unix_timestamp().to_string();
|
||||
let nonce_str = "mock-nonce".to_string();
|
||||
let package = format!("prepay_id=mock-{order_id}");
|
||||
let pay_sign = hex_sha256(format!("{time_stamp}\n{nonce_str}\n{package}\n").as_bytes());
|
||||
|
||||
WechatMiniProgramPayParamsResponse {
|
||||
time_stamp,
|
||||
nonce_str,
|
||||
package,
|
||||
sign_type: WECHAT_PAY_PAY_SIGN_TYPE.to_string(),
|
||||
pay_sign,
|
||||
}
|
||||
}
|
||||
|
||||
fn parse_mock_notify(body: &[u8]) -> Result<WechatPayNotifyOrder, WechatPayError> {
|
||||
let value = serde_json::from_slice::<Value>(body).map_err(|error| {
|
||||
WechatPayError::Deserialize(format!("mock 微信支付通知解析失败:{error}"))
|
||||
})?;
|
||||
Ok(WechatPayNotifyOrder {
|
||||
out_trade_no: value
|
||||
.get("outTradeNo")
|
||||
.or_else(|| value.get("out_trade_no"))
|
||||
.and_then(Value::as_str)
|
||||
.map(str::trim)
|
||||
.filter(|value| !value.is_empty())
|
||||
.ok_or_else(|| {
|
||||
WechatPayError::InvalidRequest("mock 微信支付通知缺少 outTradeNo".to_string())
|
||||
})?
|
||||
.to_string(),
|
||||
transaction_id: value
|
||||
.get("transactionId")
|
||||
.or_else(|| value.get("transaction_id"))
|
||||
.and_then(Value::as_str)
|
||||
.map(str::trim)
|
||||
.filter(|value| !value.is_empty())
|
||||
.map(ToOwned::to_owned),
|
||||
trade_state: value
|
||||
.get("tradeState")
|
||||
.or_else(|| value.get("trade_state"))
|
||||
.and_then(Value::as_str)
|
||||
.unwrap_or("SUCCESS")
|
||||
.to_string(),
|
||||
success_time: value
|
||||
.get("successTime")
|
||||
.or_else(|| value.get("success_time"))
|
||||
.and_then(Value::as_str)
|
||||
.map(ToOwned::to_owned),
|
||||
})
|
||||
}
|
||||
|
||||
fn required_config(value: Option<&str>, key: &str) -> Result<String, WechatPayError> {
|
||||
value
|
||||
.map(str::trim)
|
||||
.filter(|value| !value.is_empty())
|
||||
.map(ToOwned::to_owned)
|
||||
.ok_or_else(|| WechatPayError::InvalidConfig(format!("{key} 未配置")))
|
||||
}
|
||||
|
||||
fn normalize_required_url(value: &str, key: &str) -> Result<String, WechatPayError> {
|
||||
let value = value.trim();
|
||||
if value.starts_with("https://") {
|
||||
return Ok(value.to_string());
|
||||
}
|
||||
|
||||
Err(WechatPayError::InvalidConfig(format!(
|
||||
"{key} 必须是 https 地址"
|
||||
)))
|
||||
}
|
||||
|
||||
fn read_private_key_pem(
|
||||
inline_pem: Option<&str>,
|
||||
path: Option<&Path>,
|
||||
) -> Result<String, WechatPayError> {
|
||||
read_pem(
|
||||
inline_pem,
|
||||
path,
|
||||
"WECHAT_PAY_PRIVATE_KEY_PEM 或 WECHAT_PAY_PRIVATE_KEY_PATH 未配置",
|
||||
"读取微信支付私钥失败",
|
||||
)
|
||||
}
|
||||
|
||||
fn read_pem(
|
||||
inline_pem: Option<&str>,
|
||||
path: Option<&Path>,
|
||||
missing_message: &str,
|
||||
read_error_prefix: &str,
|
||||
) -> Result<String, WechatPayError> {
|
||||
if let Some(value) = inline_pem.map(str::trim).filter(|value| !value.is_empty()) {
|
||||
return Ok(value.replace("\\n", "\n"));
|
||||
}
|
||||
let Some(path) = path else {
|
||||
return Err(WechatPayError::InvalidConfig(missing_message.to_string()));
|
||||
};
|
||||
fs::read_to_string(path).map_err(|error| {
|
||||
WechatPayError::InvalidConfig(format!("{read_error_prefix}:{}:{error}", path.display()))
|
||||
})
|
||||
}
|
||||
|
||||
fn parse_rsa_private_key(pem: &str) -> Result<signature::RsaKeyPair, WechatPayError> {
|
||||
let (label, der) = parse_single_pem_block(pem)?;
|
||||
match label.as_str() {
|
||||
"PRIVATE KEY" => signature::RsaKeyPair::from_pkcs8(&der),
|
||||
"RSA PRIVATE KEY" => signature::RsaKeyPair::from_der(&der),
|
||||
_ => {
|
||||
return Err(WechatPayError::InvalidConfig(
|
||||
"微信支付私钥必须是 PRIVATE KEY 或 RSA PRIVATE KEY PEM".to_string(),
|
||||
));
|
||||
}
|
||||
}
|
||||
.map_err(|error| WechatPayError::InvalidConfig(format!("微信支付私钥解析失败:{error}")))
|
||||
}
|
||||
|
||||
fn parse_public_key_pem(pem: &str) -> Result<Vec<u8>, WechatPayError> {
|
||||
let (label, der) = parse_single_pem_block(pem)?;
|
||||
if label != "PUBLIC KEY" {
|
||||
return Err(WechatPayError::InvalidConfig(
|
||||
"微信支付平台公钥必须是 PUBLIC KEY PEM".to_string(),
|
||||
));
|
||||
}
|
||||
Ok(der)
|
||||
}
|
||||
|
||||
fn parse_single_pem_block(pem: &str) -> Result<(String, Vec<u8>), WechatPayError> {
|
||||
let mut label: Option<String> = None;
|
||||
let mut content = String::new();
|
||||
for line in pem.lines().map(str::trim).filter(|line| !line.is_empty()) {
|
||||
if let Some(raw_label) = line
|
||||
.strip_prefix("-----BEGIN ")
|
||||
.and_then(|value| value.strip_suffix("-----"))
|
||||
{
|
||||
label = Some(raw_label.trim().to_string());
|
||||
continue;
|
||||
}
|
||||
if line.starts_with("-----END ") {
|
||||
break;
|
||||
}
|
||||
if label.is_some() {
|
||||
content.push_str(line);
|
||||
}
|
||||
}
|
||||
let label = label
|
||||
.ok_or_else(|| WechatPayError::InvalidConfig("微信支付 PEM 缺少 BEGIN 标记".to_string()))?;
|
||||
let der = BASE64_STANDARD
|
||||
.decode(content)
|
||||
.map_err(|_| WechatPayError::InvalidConfig("微信支付 PEM base64 无效".to_string()))?;
|
||||
if der.is_empty() {
|
||||
return Err(WechatPayError::InvalidConfig(
|
||||
"微信支付 PEM 内容为空".to_string(),
|
||||
));
|
||||
}
|
||||
Ok((label, der))
|
||||
}
|
||||
|
||||
fn create_nonce() -> Result<String, WechatPayError> {
|
||||
let mut bytes = [0_u8; 16];
|
||||
SystemRandom::new()
|
||||
.fill(&mut bytes)
|
||||
.map_err(|_| WechatPayError::Crypto("生成微信支付 nonce 失败".to_string()))?;
|
||||
Ok(hex_encode(&bytes))
|
||||
}
|
||||
|
||||
fn decrypt_aes_256_gcm(
|
||||
key: &[u8],
|
||||
nonce: &[u8],
|
||||
associated_data: &[u8],
|
||||
ciphertext_base64: &str,
|
||||
) -> Result<Vec<u8>, WechatPayError> {
|
||||
let mut ciphertext = BASE64_STANDARD
|
||||
.decode(ciphertext_base64)
|
||||
.map_err(|_| WechatPayError::Crypto("微信支付通知密文 base64 无效".to_string()))?;
|
||||
if ciphertext.len() < aead::AES_256_GCM.tag_len() {
|
||||
return Err(WechatPayError::Crypto(
|
||||
"微信支付通知密文长度无效".to_string(),
|
||||
));
|
||||
}
|
||||
let nonce = aead::Nonce::try_assume_unique_for_key(nonce)
|
||||
.map_err(|_| WechatPayError::Crypto("微信支付通知 nonce 长度无效".to_string()))?;
|
||||
let key = aead::UnboundKey::new(&aead::AES_256_GCM, key)
|
||||
.map_err(|_| WechatPayError::Crypto("微信支付通知解密 key 无效".to_string()))?;
|
||||
let plain_text = aead::LessSafeKey::new(key)
|
||||
.open_in_place(
|
||||
nonce,
|
||||
aead::Aad::from(associated_data),
|
||||
ciphertext.as_mut_slice(),
|
||||
)
|
||||
.map_err(|_| WechatPayError::Crypto("微信支付通知认证或解密失败".to_string()))?;
|
||||
Ok(plain_text.to_vec())
|
||||
}
|
||||
|
||||
fn read_required_header<'a>(
|
||||
headers: &'a HeaderMap,
|
||||
name: &'static str,
|
||||
) -> Result<&'a str, WechatPayError> {
|
||||
headers
|
||||
.get(name)
|
||||
.and_then(|value| value.to_str().ok())
|
||||
.map(str::trim)
|
||||
.filter(|value| !value.is_empty())
|
||||
.ok_or(WechatPayError::InvalidSignature)
|
||||
}
|
||||
|
||||
fn hex_sha256(content: &[u8]) -> String {
|
||||
let mut hasher = Sha256::new();
|
||||
hasher.update(content);
|
||||
hex_encode(&hasher.finalize())
|
||||
}
|
||||
|
||||
fn hex_encode(bytes: &[u8]) -> String {
|
||||
bytes.iter().map(|byte| format!("{byte:02x}")).collect()
|
||||
}
|
||||
|
||||
impl std::fmt::Display for WechatPayError {
|
||||
fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
Self::Disabled => formatter.write_str("微信支付暂未启用"),
|
||||
Self::InvalidConfig(message)
|
||||
| Self::InvalidRequest(message)
|
||||
| Self::RequestFailed(message)
|
||||
| Self::Upstream(message)
|
||||
| Self::Deserialize(message)
|
||||
| Self::Crypto(message) => formatter.write_str(message),
|
||||
Self::InvalidSignature => formatter.write_str("微信支付通知签名无效"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl std::error::Error for WechatPayError {}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn mock_pay_params_use_request_payment_shape() {
|
||||
let params = build_mock_pay_params("recharge:user:1:points_60");
|
||||
|
||||
assert!(!params.time_stamp.is_empty());
|
||||
assert_eq!(params.sign_type, "RSA");
|
||||
assert!(params.package.starts_with("prepay_id=mock-"));
|
||||
assert!(!params.pay_sign.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn parse_mock_notify_defaults_success_state() {
|
||||
let notify =
|
||||
parse_mock_notify(br#"{"outTradeNo":"order-1"}"#).expect("mock notify should parse");
|
||||
|
||||
assert_eq!(notify.out_trade_no, "order-1");
|
||||
assert_eq!(notify.transaction_id, None);
|
||||
assert_eq!(notify.trade_state, "SUCCESS");
|
||||
}
|
||||
}
|
||||
@@ -118,6 +118,14 @@ pub struct WechatIdentityProfile {
|
||||
pub avatar_url: Option<String>,
|
||||
}
|
||||
|
||||
/// 已绑定微信身份快照。
|
||||
#[derive(Clone, Debug, PartialEq, Eq)]
|
||||
pub struct WechatIdentityRecord {
|
||||
pub user_id: String,
|
||||
pub provider_uid: String,
|
||||
pub provider_union_id: Option<String>,
|
||||
}
|
||||
|
||||
/// 微信授权 state 快照。
|
||||
#[derive(Clone, Debug, PartialEq, Eq)]
|
||||
pub struct WechatAuthStateRecord {
|
||||
|
||||
@@ -797,6 +797,13 @@ impl WechatAuthService {
|
||||
created: true,
|
||||
})
|
||||
}
|
||||
|
||||
pub fn get_identity_by_user_id(
|
||||
&self,
|
||||
user_id: &str,
|
||||
) -> Result<Option<WechatIdentityRecord>, WechatAuthError> {
|
||||
self.store.get_wechat_identity_by_user_id(user_id)
|
||||
}
|
||||
}
|
||||
|
||||
impl AuthUserService {
|
||||
@@ -1342,6 +1349,29 @@ impl InMemoryAuthStore {
|
||||
.map(|stored| stored.user.clone()))
|
||||
}
|
||||
|
||||
fn get_wechat_identity_by_user_id(
|
||||
&self,
|
||||
user_id: &str,
|
||||
) -> Result<Option<WechatIdentityRecord>, WechatAuthError> {
|
||||
let state = self
|
||||
.inner
|
||||
.lock()
|
||||
.map_err(|_| WechatAuthError::Store("用户仓储锁已中毒".to_string()))?;
|
||||
let Some(identity) = state
|
||||
.wechat_identity_by_provider_uid
|
||||
.values()
|
||||
.find(|identity| identity.user_id == user_id.trim())
|
||||
else {
|
||||
return Ok(None);
|
||||
};
|
||||
|
||||
Ok(Some(WechatIdentityRecord {
|
||||
user_id: identity.user_id.clone(),
|
||||
provider_uid: identity.provider_uid.clone(),
|
||||
provider_union_id: identity.provider_union_id.clone(),
|
||||
}))
|
||||
}
|
||||
|
||||
fn refresh_wechat_identity_profile(
|
||||
&self,
|
||||
user_id: &str,
|
||||
|
||||
@@ -190,8 +190,9 @@ pub fn build_runtime_profile_recharge_order_record(
|
||||
amount_cents: snapshot.amount_cents,
|
||||
status: snapshot.status,
|
||||
payment_channel: snapshot.payment_channel,
|
||||
paid_at: format_utc_micros(snapshot.paid_at_micros),
|
||||
paid_at: snapshot.paid_at_micros.map(format_utc_micros),
|
||||
paid_at_micros: snapshot.paid_at_micros,
|
||||
provider_transaction_id: snapshot.provider_transaction_id,
|
||||
created_at: format_utc_micros(snapshot.created_at_micros),
|
||||
created_at_micros: snapshot.created_at_micros,
|
||||
points_delta: snapshot.points_delta,
|
||||
|
||||
@@ -265,6 +265,20 @@ pub fn build_runtime_profile_recharge_order_create_input(
|
||||
})
|
||||
}
|
||||
|
||||
pub fn build_runtime_profile_recharge_order_paid_input(
|
||||
order_id: String,
|
||||
paid_at_micros: i64,
|
||||
provider_transaction_id: Option<String>,
|
||||
) -> Result<RuntimeProfileRechargeOrderPaidInput, RuntimeProfileFieldError> {
|
||||
let order_id =
|
||||
normalize_required_string(order_id).ok_or(RuntimeProfileFieldError::MissingOrderId)?;
|
||||
Ok(RuntimeProfileRechargeOrderPaidInput {
|
||||
order_id,
|
||||
paid_at_micros,
|
||||
provider_transaction_id: provider_transaction_id.and_then(normalize_required_string),
|
||||
})
|
||||
}
|
||||
|
||||
pub fn build_runtime_profile_feedback_submission_input(
|
||||
user_id: String,
|
||||
description: String,
|
||||
|
||||
@@ -33,6 +33,7 @@ pub const PROFILE_TASK_DEFAULT_THRESHOLD: u32 = 1;
|
||||
pub const SAVE_SNAPSHOT_VERSION: u32 = 2;
|
||||
pub const DEFAULT_SAVE_ARCHIVE_SUMMARY_TEXT: &str = "继续推进上一次保存的故事。";
|
||||
pub const PROFILE_RECHARGE_PAYMENT_CHANNEL_MOCK: &str = "mock";
|
||||
pub const PROFILE_RECHARGE_PAYMENT_CHANNEL_WECHAT_MINI_PROGRAM: &str = "wechat_mp";
|
||||
pub const PROFILE_FEEDBACK_DESCRIPTION_MIN_CHARS: usize = 10;
|
||||
pub const PROFILE_FEEDBACK_DESCRIPTION_MAX_CHARS: usize = 200;
|
||||
pub const PROFILE_FEEDBACK_CONTACT_PHONE_MAX_CHARS: usize = 40;
|
||||
@@ -951,13 +952,21 @@ impl RuntimeProfileMembershipTier {
|
||||
#[cfg_attr(feature = "spacetime-types", derive(SpacetimeType))]
|
||||
#[derive(Clone, Copy, Debug, PartialEq, Eq, Serialize, Deserialize)]
|
||||
pub enum RuntimeProfileRechargeOrderStatus {
|
||||
Pending,
|
||||
Paid,
|
||||
Failed,
|
||||
Closed,
|
||||
Refunded,
|
||||
}
|
||||
|
||||
impl RuntimeProfileRechargeOrderStatus {
|
||||
pub fn as_str(&self) -> &'static str {
|
||||
match self {
|
||||
Self::Pending => "pending",
|
||||
Self::Paid => "paid",
|
||||
Self::Failed => "failed",
|
||||
Self::Closed => "closed",
|
||||
Self::Refunded => "refunded",
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1009,7 +1018,8 @@ pub struct RuntimeProfileRechargeOrderSnapshot {
|
||||
pub amount_cents: u64,
|
||||
pub status: RuntimeProfileRechargeOrderStatus,
|
||||
pub payment_channel: String,
|
||||
pub paid_at_micros: i64,
|
||||
pub paid_at_micros: Option<i64>,
|
||||
pub provider_transaction_id: Option<String>,
|
||||
pub created_at_micros: i64,
|
||||
pub points_delta: i64,
|
||||
pub membership_expires_at_micros: Option<i64>,
|
||||
@@ -1059,6 +1069,14 @@ pub struct RuntimeProfileRechargeOrderCreateInput {
|
||||
pub created_at_micros: i64,
|
||||
}
|
||||
|
||||
#[cfg_attr(feature = "spacetime-types", derive(SpacetimeType))]
|
||||
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
|
||||
pub struct RuntimeProfileRechargeOrderPaidInput {
|
||||
pub order_id: String,
|
||||
pub paid_at_micros: i64,
|
||||
pub provider_transaction_id: Option<String>,
|
||||
}
|
||||
|
||||
#[cfg_attr(feature = "spacetime-types", derive(SpacetimeType))]
|
||||
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
|
||||
pub struct RuntimeProfileWalletLedgerEntrySnapshot {
|
||||
@@ -1471,8 +1489,9 @@ pub struct RuntimeProfileRechargeOrderRecord {
|
||||
pub amount_cents: u64,
|
||||
pub status: RuntimeProfileRechargeOrderStatus,
|
||||
pub payment_channel: String,
|
||||
pub paid_at: String,
|
||||
pub paid_at_micros: i64,
|
||||
pub paid_at: Option<String>,
|
||||
pub paid_at_micros: Option<i64>,
|
||||
pub provider_transaction_id: Option<String>,
|
||||
pub created_at: String,
|
||||
pub created_at_micros: i64,
|
||||
pub points_delta: i64,
|
||||
|
||||
@@ -72,6 +72,7 @@ pub enum RuntimeProfileFieldError {
|
||||
TaskDisabled,
|
||||
TaskNotClaimable,
|
||||
TaskAlreadyClaimed,
|
||||
MissingOrderId,
|
||||
MissingProductId,
|
||||
MissingWorldKey,
|
||||
MissingBottomTab,
|
||||
@@ -133,6 +134,7 @@ impl std::fmt::Display for RuntimeProfileFieldError {
|
||||
Self::TaskDisabled => f.write_str("任务已停用"),
|
||||
Self::TaskNotClaimable => f.write_str("任务尚未达成"),
|
||||
Self::TaskAlreadyClaimed => f.write_str("任务奖励已领取"),
|
||||
Self::MissingOrderId => f.write_str("recharge.order_id 不能为空"),
|
||||
Self::MissingProductId => f.write_str("recharge.product_id 不能为空"),
|
||||
Self::MissingWorldKey => f.write_str("profile.world_key 不能为空"),
|
||||
Self::MissingBottomTab => f.write_str("runtime_snapshot.bottom_tab 不能为空"),
|
||||
|
||||
@@ -222,12 +222,23 @@ pub struct ProfileRechargeOrderResponse {
|
||||
pub amount_cents: u64,
|
||||
pub status: String,
|
||||
pub payment_channel: String,
|
||||
pub paid_at: String,
|
||||
pub paid_at: Option<String>,
|
||||
pub provider_transaction_id: Option<String>,
|
||||
pub created_at: String,
|
||||
pub points_delta: i64,
|
||||
pub membership_expires_at: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
pub struct WechatMiniProgramPayParamsResponse {
|
||||
pub time_stamp: String,
|
||||
pub nonce_str: String,
|
||||
pub package: String,
|
||||
pub sign_type: String,
|
||||
pub pay_sign: String,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
pub struct ProfileRechargeCenterResponse {
|
||||
@@ -253,6 +264,8 @@ pub struct CreateProfileRechargeOrderRequest {
|
||||
pub struct CreateProfileRechargeOrderResponse {
|
||||
pub order: ProfileRechargeOrderResponse,
|
||||
pub center: ProfileRechargeCenterResponse,
|
||||
#[serde(default)]
|
||||
pub wechat_mini_program_pay_params: Option<WechatMiniProgramPayParamsResponse>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
|
||||
|
||||
@@ -176,6 +176,18 @@ impl From<module_runtime::RuntimeProfileRechargeOrderCreateInput>
|
||||
}
|
||||
}
|
||||
|
||||
impl From<module_runtime::RuntimeProfileRechargeOrderPaidInput>
|
||||
for RuntimeProfileRechargeOrderPaidInput
|
||||
{
|
||||
fn from(input: module_runtime::RuntimeProfileRechargeOrderPaidInput) -> Self {
|
||||
Self {
|
||||
order_id: input.order_id,
|
||||
paid_at_micros: input.paid_at_micros,
|
||||
provider_transaction_id: input.provider_transaction_id,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<module_runtime::RuntimeProfileFeedbackSubmissionInput>
|
||||
for RuntimeProfileFeedbackSubmissionInput
|
||||
{
|
||||
@@ -2217,6 +2229,7 @@ pub(crate) fn map_runtime_profile_recharge_order_snapshot(
|
||||
status: map_runtime_profile_recharge_order_status_back(snapshot.status),
|
||||
payment_channel: snapshot.payment_channel,
|
||||
paid_at_micros: snapshot.paid_at_micros,
|
||||
provider_transaction_id: snapshot.provider_transaction_id,
|
||||
created_at_micros: snapshot.created_at_micros,
|
||||
points_delta: snapshot.points_delta,
|
||||
membership_expires_at_micros: snapshot.membership_expires_at_micros,
|
||||
@@ -5026,9 +5039,21 @@ pub(crate) fn map_runtime_profile_recharge_order_status_back(
|
||||
value: crate::module_bindings::RuntimeProfileRechargeOrderStatus,
|
||||
) -> module_runtime::RuntimeProfileRechargeOrderStatus {
|
||||
match value {
|
||||
crate::module_bindings::RuntimeProfileRechargeOrderStatus::Pending => {
|
||||
module_runtime::RuntimeProfileRechargeOrderStatus::Pending
|
||||
}
|
||||
crate::module_bindings::RuntimeProfileRechargeOrderStatus::Paid => {
|
||||
module_runtime::RuntimeProfileRechargeOrderStatus::Paid
|
||||
}
|
||||
crate::module_bindings::RuntimeProfileRechargeOrderStatus::Failed => {
|
||||
module_runtime::RuntimeProfileRechargeOrderStatus::Failed
|
||||
}
|
||||
crate::module_bindings::RuntimeProfileRechargeOrderStatus::Closed => {
|
||||
module_runtime::RuntimeProfileRechargeOrderStatus::Closed
|
||||
}
|
||||
crate::module_bindings::RuntimeProfileRechargeOrderStatus::Refunded => {
|
||||
module_runtime::RuntimeProfileRechargeOrderStatus::Refunded
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,62 @@
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED BY SPACETIMEDB. EDITS TO THIS FILE
|
||||
// WILL NOT BE SAVED. MODIFY TABLES IN YOUR MODULE SOURCE CODE INSTEAD.
|
||||
|
||||
#![allow(unused, clippy::all)]
|
||||
use spacetimedb_sdk::__codegen::{self as __sdk, __lib, __sats, __ws};
|
||||
|
||||
use super::runtime_profile_recharge_center_procedure_result_type::RuntimeProfileRechargeCenterProcedureResult;
|
||||
use super::runtime_profile_recharge_order_paid_input_type::RuntimeProfileRechargeOrderPaidInput;
|
||||
|
||||
#[derive(__lib::ser::Serialize, __lib::de::Deserialize, Clone, PartialEq, Debug)]
|
||||
#[sats(crate = __lib)]
|
||||
struct MarkProfileRechargeOrderPaidAndReturnArgs {
|
||||
pub input: RuntimeProfileRechargeOrderPaidInput,
|
||||
}
|
||||
|
||||
impl __sdk::InModule for MarkProfileRechargeOrderPaidAndReturnArgs {
|
||||
type Module = super::RemoteModule;
|
||||
}
|
||||
|
||||
#[allow(non_camel_case_types)]
|
||||
/// Extension trait for access to the procedure `mark_profile_recharge_order_paid_and_return`.
|
||||
///
|
||||
/// Implemented for [`super::RemoteProcedures`].
|
||||
pub trait mark_profile_recharge_order_paid_and_return {
|
||||
fn mark_profile_recharge_order_paid_and_return(
|
||||
&self,
|
||||
input: RuntimeProfileRechargeOrderPaidInput,
|
||||
) {
|
||||
self.mark_profile_recharge_order_paid_and_return_then(input, |_, _| {});
|
||||
}
|
||||
|
||||
fn mark_profile_recharge_order_paid_and_return_then(
|
||||
&self,
|
||||
input: RuntimeProfileRechargeOrderPaidInput,
|
||||
|
||||
__callback: impl FnOnce(
|
||||
&super::ProcedureEventContext,
|
||||
Result<RuntimeProfileRechargeCenterProcedureResult, __sdk::InternalError>,
|
||||
) + Send
|
||||
+ 'static,
|
||||
);
|
||||
}
|
||||
|
||||
impl mark_profile_recharge_order_paid_and_return for super::RemoteProcedures {
|
||||
fn mark_profile_recharge_order_paid_and_return_then(
|
||||
&self,
|
||||
input: RuntimeProfileRechargeOrderPaidInput,
|
||||
|
||||
__callback: impl FnOnce(
|
||||
&super::ProcedureEventContext,
|
||||
Result<RuntimeProfileRechargeCenterProcedureResult, __sdk::InternalError>,
|
||||
) + Send
|
||||
+ 'static,
|
||||
) {
|
||||
self.imp
|
||||
.invoke_procedure_with_callback::<_, RuntimeProfileRechargeCenterProcedureResult>(
|
||||
"mark_profile_recharge_order_paid_and_return",
|
||||
MarkProfileRechargeOrderPaidAndReturnArgs { input },
|
||||
__callback,
|
||||
);
|
||||
}
|
||||
}
|
||||
@@ -367,6 +367,7 @@ pub mod list_puzzle_works_procedure;
|
||||
pub mod list_square_hole_works_procedure;
|
||||
pub mod list_visual_novel_runtime_history_procedure;
|
||||
pub mod list_visual_novel_works_procedure;
|
||||
pub mod mark_profile_recharge_order_paid_and_return_procedure;
|
||||
pub mod match_3_d_agent_message_finalize_input_type;
|
||||
pub mod match_3_d_agent_message_row_type;
|
||||
pub mod match_3_d_agent_message_submit_input_type;
|
||||
@@ -616,6 +617,7 @@ pub mod runtime_profile_recharge_center_get_input_type;
|
||||
pub mod runtime_profile_recharge_center_procedure_result_type;
|
||||
pub mod runtime_profile_recharge_center_snapshot_type;
|
||||
pub mod runtime_profile_recharge_order_create_input_type;
|
||||
pub mod runtime_profile_recharge_order_paid_input_type;
|
||||
pub mod runtime_profile_recharge_order_snapshot_type;
|
||||
pub mod runtime_profile_recharge_order_status_type;
|
||||
pub mod runtime_profile_recharge_product_kind_type;
|
||||
@@ -1177,6 +1179,7 @@ pub use list_puzzle_works_procedure::list_puzzle_works;
|
||||
pub use list_square_hole_works_procedure::list_square_hole_works;
|
||||
pub use list_visual_novel_runtime_history_procedure::list_visual_novel_runtime_history;
|
||||
pub use list_visual_novel_works_procedure::list_visual_novel_works;
|
||||
pub use mark_profile_recharge_order_paid_and_return_procedure::mark_profile_recharge_order_paid_and_return;
|
||||
pub use match_3_d_agent_message_finalize_input_type::Match3DAgentMessageFinalizeInput;
|
||||
pub use match_3_d_agent_message_row_type::Match3DAgentMessageRow;
|
||||
pub use match_3_d_agent_message_submit_input_type::Match3DAgentMessageSubmitInput;
|
||||
@@ -1426,6 +1429,7 @@ pub use runtime_profile_recharge_center_get_input_type::RuntimeProfileRechargeCe
|
||||
pub use runtime_profile_recharge_center_procedure_result_type::RuntimeProfileRechargeCenterProcedureResult;
|
||||
pub use runtime_profile_recharge_center_snapshot_type::RuntimeProfileRechargeCenterSnapshot;
|
||||
pub use runtime_profile_recharge_order_create_input_type::RuntimeProfileRechargeOrderCreateInput;
|
||||
pub use runtime_profile_recharge_order_paid_input_type::RuntimeProfileRechargeOrderPaidInput;
|
||||
pub use runtime_profile_recharge_order_snapshot_type::RuntimeProfileRechargeOrderSnapshot;
|
||||
pub use runtime_profile_recharge_order_status_type::RuntimeProfileRechargeOrderStatus;
|
||||
pub use runtime_profile_recharge_product_kind_type::RuntimeProfileRechargeProductKind;
|
||||
|
||||
@@ -18,7 +18,8 @@ pub struct ProfileRechargeOrder {
|
||||
pub amount_cents: u64,
|
||||
pub status: RuntimeProfileRechargeOrderStatus,
|
||||
pub payment_channel: String,
|
||||
pub paid_at: __sdk::Timestamp,
|
||||
pub paid_at: Option<__sdk::Timestamp>,
|
||||
pub provider_transaction_id: Option<String>,
|
||||
pub created_at: __sdk::Timestamp,
|
||||
pub points_delta: i64,
|
||||
pub membership_expires_at: Option<__sdk::Timestamp>,
|
||||
@@ -41,7 +42,8 @@ pub struct ProfileRechargeOrderCols {
|
||||
pub status:
|
||||
__sdk::__query_builder::Col<ProfileRechargeOrder, RuntimeProfileRechargeOrderStatus>,
|
||||
pub payment_channel: __sdk::__query_builder::Col<ProfileRechargeOrder, String>,
|
||||
pub paid_at: __sdk::__query_builder::Col<ProfileRechargeOrder, __sdk::Timestamp>,
|
||||
pub paid_at: __sdk::__query_builder::Col<ProfileRechargeOrder, Option<__sdk::Timestamp>>,
|
||||
pub provider_transaction_id: __sdk::__query_builder::Col<ProfileRechargeOrder, Option<String>>,
|
||||
pub created_at: __sdk::__query_builder::Col<ProfileRechargeOrder, __sdk::Timestamp>,
|
||||
pub points_delta: __sdk::__query_builder::Col<ProfileRechargeOrder, i64>,
|
||||
pub membership_expires_at:
|
||||
@@ -61,6 +63,10 @@ impl __sdk::__query_builder::HasCols for ProfileRechargeOrder {
|
||||
status: __sdk::__query_builder::Col::new(table_name, "status"),
|
||||
payment_channel: __sdk::__query_builder::Col::new(table_name, "payment_channel"),
|
||||
paid_at: __sdk::__query_builder::Col::new(table_name, "paid_at"),
|
||||
provider_transaction_id: __sdk::__query_builder::Col::new(
|
||||
table_name,
|
||||
"provider_transaction_id",
|
||||
),
|
||||
created_at: __sdk::__query_builder::Col::new(table_name, "created_at"),
|
||||
points_delta: __sdk::__query_builder::Col::new(table_name, "points_delta"),
|
||||
membership_expires_at: __sdk::__query_builder::Col::new(
|
||||
|
||||
@@ -0,0 +1,17 @@
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED BY SPACETIMEDB. EDITS TO THIS FILE
|
||||
// WILL NOT BE SAVED. MODIFY TABLES IN YOUR MODULE SOURCE CODE INSTEAD.
|
||||
|
||||
#![allow(unused, clippy::all)]
|
||||
use spacetimedb_sdk::__codegen::{self as __sdk, __lib, __sats, __ws};
|
||||
|
||||
#[derive(__lib::ser::Serialize, __lib::de::Deserialize, Clone, PartialEq, Debug)]
|
||||
#[sats(crate = __lib)]
|
||||
pub struct RuntimeProfileRechargeOrderPaidInput {
|
||||
pub order_id: String,
|
||||
pub paid_at_micros: i64,
|
||||
pub provider_transaction_id: Option<String>,
|
||||
}
|
||||
|
||||
impl __sdk::InModule for RuntimeProfileRechargeOrderPaidInput {
|
||||
type Module = super::RemoteModule;
|
||||
}
|
||||
@@ -18,7 +18,8 @@ pub struct RuntimeProfileRechargeOrderSnapshot {
|
||||
pub amount_cents: u64,
|
||||
pub status: RuntimeProfileRechargeOrderStatus,
|
||||
pub payment_channel: String,
|
||||
pub paid_at_micros: i64,
|
||||
pub paid_at_micros: Option<i64>,
|
||||
pub provider_transaction_id: Option<String>,
|
||||
pub created_at_micros: i64,
|
||||
pub points_delta: i64,
|
||||
pub membership_expires_at_micros: Option<i64>,
|
||||
|
||||
@@ -8,7 +8,15 @@ use spacetimedb_sdk::__codegen::{self as __sdk, __lib, __sats, __ws};
|
||||
#[sats(crate = __lib)]
|
||||
#[derive(Copy, Eq, Hash)]
|
||||
pub enum RuntimeProfileRechargeOrderStatus {
|
||||
Pending,
|
||||
|
||||
Paid,
|
||||
|
||||
Failed,
|
||||
|
||||
Closed,
|
||||
|
||||
Refunded,
|
||||
}
|
||||
|
||||
impl __sdk::InModule for RuntimeProfileRechargeOrderStatus {
|
||||
|
||||
@@ -268,6 +268,42 @@ impl SpacetimeClient {
|
||||
.await
|
||||
}
|
||||
|
||||
pub async fn mark_profile_recharge_order_paid(
|
||||
&self,
|
||||
order_id: String,
|
||||
paid_at_micros: i64,
|
||||
provider_transaction_id: Option<String>,
|
||||
) -> Result<
|
||||
(
|
||||
RuntimeProfileRechargeCenterRecord,
|
||||
RuntimeProfileRechargeOrderRecord,
|
||||
),
|
||||
SpacetimeClientError,
|
||||
> {
|
||||
let procedure_input = module_runtime::build_runtime_profile_recharge_order_paid_input(
|
||||
order_id,
|
||||
paid_at_micros,
|
||||
provider_transaction_id,
|
||||
)
|
||||
.map_err(SpacetimeClientError::validation_failed)?
|
||||
.into();
|
||||
|
||||
self.call_after_connect(move |connection, sender| {
|
||||
connection
|
||||
.procedures()
|
||||
.mark_profile_recharge_order_paid_and_return_then(
|
||||
procedure_input,
|
||||
move |_, result| {
|
||||
let mapped = result
|
||||
.map_err(SpacetimeClientError::from_sdk_error)
|
||||
.and_then(map_runtime_profile_recharge_order_procedure_result);
|
||||
send_once(&sender, mapped);
|
||||
},
|
||||
);
|
||||
})
|
||||
.await
|
||||
}
|
||||
|
||||
pub async fn submit_profile_feedback(
|
||||
&self,
|
||||
user_id: String,
|
||||
|
||||
@@ -1151,6 +1151,14 @@ fn normalize_migration_row(table_name: &str, value: &serde_json::Value) -> serde
|
||||
.or_insert_with(|| serde_json::Value::String("{}".to_string()));
|
||||
}
|
||||
}
|
||||
if table_name == "profile_recharge_order" {
|
||||
if let Some(object) = next_value.as_object_mut() {
|
||||
// 中文注释:真实微信支付接入后才有平台交易号,旧迁移包按未回填处理。
|
||||
object
|
||||
.entry("provider_transaction_id".to_string())
|
||||
.or_insert(serde_json::Value::Null);
|
||||
}
|
||||
}
|
||||
if table_name == "big_fish_creation_session" {
|
||||
if let Some(object) = next_value.as_object_mut() {
|
||||
// 中文注释:旧迁移包没有公开游玩次数字段,导入时按新建作品默认 0 兼容。
|
||||
|
||||
@@ -336,6 +336,7 @@ pub struct ProfileMembership {
|
||||
btree(columns = [user_id, created_at])
|
||||
)
|
||||
)]
|
||||
#[derive(Clone)]
|
||||
pub struct ProfileRechargeOrder {
|
||||
#[primary_key]
|
||||
pub(crate) order_id: String,
|
||||
@@ -346,7 +347,10 @@ pub struct ProfileRechargeOrder {
|
||||
pub(crate) amount_cents: u64,
|
||||
pub(crate) status: RuntimeProfileRechargeOrderStatus,
|
||||
pub(crate) payment_channel: String,
|
||||
pub(crate) paid_at: Timestamp,
|
||||
#[default(None::<Timestamp>)]
|
||||
pub(crate) paid_at: Option<Timestamp>,
|
||||
#[default(None::<String>)]
|
||||
pub(crate) provider_transaction_id: Option<String>,
|
||||
pub(crate) created_at: Timestamp,
|
||||
pub(crate) points_delta: i64,
|
||||
pub(crate) membership_expires_at: Option<Timestamp>,
|
||||
@@ -767,7 +771,6 @@ pub fn get_profile_recharge_center(
|
||||
}
|
||||
}
|
||||
|
||||
// 当前阶段没有真实支付网关,下单后在服务端模拟支付成功并立即写入权益。
|
||||
#[spacetimedb::procedure]
|
||||
pub fn create_profile_recharge_order_and_return(
|
||||
ctx: &mut ProcedureContext,
|
||||
@@ -789,6 +792,27 @@ pub fn create_profile_recharge_order_and_return(
|
||||
}
|
||||
}
|
||||
|
||||
#[spacetimedb::procedure]
|
||||
pub fn mark_profile_recharge_order_paid_and_return(
|
||||
ctx: &mut ProcedureContext,
|
||||
input: RuntimeProfileRechargeOrderPaidInput,
|
||||
) -> RuntimeProfileRechargeCenterProcedureResult {
|
||||
match ctx.try_with_tx(|tx| mark_profile_recharge_order_paid_record(tx, input.clone())) {
|
||||
Ok((record, order)) => RuntimeProfileRechargeCenterProcedureResult {
|
||||
ok: true,
|
||||
record: Some(record),
|
||||
order: Some(order),
|
||||
error_message: None,
|
||||
},
|
||||
Err(message) => RuntimeProfileRechargeCenterProcedureResult {
|
||||
ok: false,
|
||||
record: None,
|
||||
order: None,
|
||||
error_message: Some(message),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
#[spacetimedb::procedure]
|
||||
pub fn submit_profile_feedback_and_return(
|
||||
ctx: &mut ProcedureContext,
|
||||
@@ -2049,36 +2073,24 @@ fn create_profile_recharge_order_record(
|
||||
let product = runtime_profile_recharge_product_by_id(&validated_input.product_id)
|
||||
.ok_or_else(|| "recharge.product_id 不存在".to_string())?;
|
||||
let created_at = Timestamp::from_micros_since_unix_epoch(validated_input.created_at_micros);
|
||||
|
||||
let (points_delta, membership_expires_at) = match product.kind {
|
||||
RuntimeProfileRechargeProductKind::Points => {
|
||||
let has_recharged = has_profile_points_recharged(ctx, &validated_input.user_id);
|
||||
let points_delta =
|
||||
resolve_runtime_profile_points_recharge_delta(&product, has_recharged);
|
||||
apply_profile_wallet_delta(
|
||||
ctx,
|
||||
&validated_input.user_id,
|
||||
points_delta,
|
||||
RuntimeProfileWalletLedgerSourceType::PointsRecharge,
|
||||
&build_runtime_profile_recharge_wallet_ledger_id(
|
||||
&validated_input.user_id,
|
||||
validated_input.created_at_micros,
|
||||
&product.product_id,
|
||||
),
|
||||
created_at,
|
||||
)?;
|
||||
(points_delta as i64, None)
|
||||
}
|
||||
RuntimeProfileRechargeProductKind::Membership => {
|
||||
let expires_at = apply_profile_membership_purchase(
|
||||
ctx,
|
||||
&validated_input.user_id,
|
||||
product.tier,
|
||||
product.duration_days,
|
||||
created_at,
|
||||
);
|
||||
(0, Some(expires_at))
|
||||
}
|
||||
let should_settle_immediately =
|
||||
validated_input.payment_channel == PROFILE_RECHARGE_PAYMENT_CHANNEL_MOCK;
|
||||
let (status, paid_at, points_delta, membership_expires_at) = if should_settle_immediately {
|
||||
let (points_delta, membership_expires_at) = apply_profile_recharge_purchase(
|
||||
ctx,
|
||||
&validated_input.user_id,
|
||||
&product,
|
||||
validated_input.created_at_micros,
|
||||
created_at,
|
||||
)?;
|
||||
(
|
||||
RuntimeProfileRechargeOrderStatus::Paid,
|
||||
Some(created_at),
|
||||
points_delta,
|
||||
membership_expires_at,
|
||||
)
|
||||
} else {
|
||||
(RuntimeProfileRechargeOrderStatus::Pending, None, 0, None)
|
||||
};
|
||||
|
||||
let order = ProfileRechargeOrder {
|
||||
@@ -2092,9 +2104,10 @@ fn create_profile_recharge_order_record(
|
||||
product_title: product.title.clone(),
|
||||
kind: product.kind,
|
||||
amount_cents: product.price_cents,
|
||||
status: RuntimeProfileRechargeOrderStatus::Paid,
|
||||
status,
|
||||
payment_channel: validated_input.payment_channel,
|
||||
paid_at: created_at,
|
||||
paid_at,
|
||||
provider_transaction_id: None,
|
||||
created_at,
|
||||
points_delta,
|
||||
membership_expires_at,
|
||||
@@ -2109,6 +2122,106 @@ fn create_profile_recharge_order_record(
|
||||
))
|
||||
}
|
||||
|
||||
fn mark_profile_recharge_order_paid_record(
|
||||
ctx: &ReducerContext,
|
||||
input: RuntimeProfileRechargeOrderPaidInput,
|
||||
) -> Result<
|
||||
(
|
||||
RuntimeProfileRechargeCenterSnapshot,
|
||||
RuntimeProfileRechargeOrderSnapshot,
|
||||
),
|
||||
String,
|
||||
> {
|
||||
let validated_input = build_runtime_profile_recharge_order_paid_input(
|
||||
input.order_id,
|
||||
input.paid_at_micros,
|
||||
input.provider_transaction_id,
|
||||
)
|
||||
.map_err(|error| error.to_string())?;
|
||||
let mut order = ctx
|
||||
.db
|
||||
.profile_recharge_order()
|
||||
.order_id()
|
||||
.find(&validated_input.order_id)
|
||||
.ok_or_else(|| "profile_recharge_order 不存在".to_string())?;
|
||||
|
||||
if order.status == RuntimeProfileRechargeOrderStatus::Paid {
|
||||
return Ok((
|
||||
build_profile_recharge_center_snapshot(ctx, &order.user_id),
|
||||
build_profile_recharge_order_snapshot_from_row(&order),
|
||||
));
|
||||
}
|
||||
if order.status != RuntimeProfileRechargeOrderStatus::Pending {
|
||||
return Err("profile_recharge_order 当前状态不能确认支付".to_string());
|
||||
}
|
||||
|
||||
let product = runtime_profile_recharge_product_by_id(&order.product_id)
|
||||
.ok_or_else(|| "recharge.product_id 不存在".to_string())?;
|
||||
let paid_at = Timestamp::from_micros_since_unix_epoch(validated_input.paid_at_micros);
|
||||
let (points_delta, membership_expires_at) = apply_profile_recharge_purchase(
|
||||
ctx,
|
||||
&order.user_id,
|
||||
&product,
|
||||
order.created_at.to_micros_since_unix_epoch(),
|
||||
paid_at,
|
||||
)?;
|
||||
|
||||
ctx.db
|
||||
.profile_recharge_order()
|
||||
.order_id()
|
||||
.delete(&order.order_id);
|
||||
order.status = RuntimeProfileRechargeOrderStatus::Paid;
|
||||
order.paid_at = Some(paid_at);
|
||||
order.provider_transaction_id = validated_input.provider_transaction_id;
|
||||
order.points_delta = points_delta;
|
||||
order.membership_expires_at = membership_expires_at;
|
||||
ctx.db.profile_recharge_order().insert(order.clone());
|
||||
|
||||
Ok((
|
||||
build_profile_recharge_center_snapshot(ctx, &order.user_id),
|
||||
build_profile_recharge_order_snapshot_from_row(&order),
|
||||
))
|
||||
}
|
||||
|
||||
fn apply_profile_recharge_purchase(
|
||||
ctx: &ReducerContext,
|
||||
user_id: &str,
|
||||
product: &RuntimeProfileRechargeProductSnapshot,
|
||||
order_created_at_micros: i64,
|
||||
paid_at: Timestamp,
|
||||
) -> Result<(i64, Option<Timestamp>), String> {
|
||||
match product.kind {
|
||||
RuntimeProfileRechargeProductKind::Points => {
|
||||
let has_recharged = has_profile_points_recharged(ctx, user_id);
|
||||
let points_delta =
|
||||
resolve_runtime_profile_points_recharge_delta(product, has_recharged);
|
||||
apply_profile_wallet_delta(
|
||||
ctx,
|
||||
user_id,
|
||||
points_delta,
|
||||
RuntimeProfileWalletLedgerSourceType::PointsRecharge,
|
||||
&build_runtime_profile_recharge_wallet_ledger_id(
|
||||
user_id,
|
||||
order_created_at_micros,
|
||||
&product.product_id,
|
||||
),
|
||||
paid_at,
|
||||
)?;
|
||||
Ok((points_delta as i64, None))
|
||||
}
|
||||
RuntimeProfileRechargeProductKind::Membership => {
|
||||
let expires_at = apply_profile_membership_purchase(
|
||||
ctx,
|
||||
user_id,
|
||||
product.tier,
|
||||
product.duration_days,
|
||||
paid_at,
|
||||
);
|
||||
Ok((0, Some(expires_at)))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn submit_profile_feedback_record(
|
||||
ctx: &ReducerContext,
|
||||
input: RuntimeProfileFeedbackSubmissionInput,
|
||||
@@ -3745,7 +3858,8 @@ fn build_profile_recharge_order_snapshot_from_row(
|
||||
amount_cents: row.amount_cents,
|
||||
status: row.status,
|
||||
payment_channel: row.payment_channel.clone(),
|
||||
paid_at_micros: row.paid_at.to_micros_since_unix_epoch(),
|
||||
paid_at_micros: row.paid_at.map(|value| value.to_micros_since_unix_epoch()),
|
||||
provider_transaction_id: row.provider_transaction_id.clone(),
|
||||
created_at_micros: row.created_at.to_micros_since_unix_epoch(),
|
||||
points_delta: row.points_delta,
|
||||
membership_expires_at_micros: row
|
||||
|
||||
Reference in New Issue
Block a user