rust_pay_wf 0.5.0

Rust 微信、支付宝、银联支付SDK
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

use crate::config::AlipayConfig;
use crate::errors::PayError;
use crate::utils::rsa_verify_sha256_pem;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::sync::Arc;

#[derive(Debug, Serialize, Deserialize)]
pub struct AlipayNotifyData {
    pub app_id: String,
    pub out_trade_no: String,
    pub trade_no: String,
    pub trade_status: String,
    pub total_amount: String,
    pub seller_id: Option<String>,
    pub sub_merchant_id: Option<String>,
    // keep other parameters if needed
    #[serde(flatten)]
    pub others: HashMap<String, String>,
}

pub struct AlipayNotify {
    cfg: Arc<AlipayConfig>,
    mode: crate::config::Mode,
}

impl AlipayNotify {
    pub fn new(cfg: Arc<AlipayConfig>, mode: crate::config::Mode) -> Self {
        Self { cfg, mode }
    }

    /// Verify Alipay notify parameters (form-encoded -> HashMap)
    /// Steps:
    /// 1. Extract sign and sign_type
    /// 2. Build pre-sign string by sorting params excluding sign & sign_type, joining k=v with &
    /// 3. Verify using RSA2 with alipay public key (sandbox/public as mode requires)
    /// 4. Validate trade_status and sub_merchant_id if service mode configured
    pub fn verify_notify(
        &self,
        params: &HashMap<String, String>,
    ) -> Result<AlipayNotifyData, PayError> {
        let sign = params
            .get("sign")
            .ok_or(PayError::Other("missing sign".to_string()))?;
        // sign_type may be optional but typically present
        // Build pre-sign string
        let mut kv: Vec<(&String, &String)> = params
            .iter()
            .filter(|&(k, _)| k != "sign" && k != "sign_type")
            .collect();
        kv.sort_by(|a, b| a.0.cmp(b.0));
        let content = kv
            .iter()
            .map(|(k, v)| format!("{}={}", k, v))
            .collect::<Vec<String>>()
            .join("&");

        // Choose public key: sandbox or normal config (here we assume cfg.alipay_public_key_pem works for both if set by user)
        let pubkey = &self.cfg.alipay_public_key_pem;
        let verified = rsa_verify_sha256_pem(pubkey, &content, sign)
            .map_err(|e| PayError::Crypto(format!("rsa verify error: {}", e)))?;
        if !verified {
            return Err(PayError::Other(
                "alipay notify signature invalid".to_string(),
            ));
        }

        // Extract fields and build data
        let app_id = params.get("app_id").cloned().unwrap_or_default();
        let out_trade_no = params.get("out_trade_no").cloned().unwrap_or_default();
        let trade_no = params.get("trade_no").cloned().unwrap_or_default();
        let trade_status = params.get("trade_status").cloned().unwrap_or_default();
        let total_amount = params.get("total_amount").cloned().unwrap_or_default();
        let seller_id = params.get("seller_id").cloned();
        let sub_merchant_id = params.get("sub_merchant_id").cloned();

        // business status check per Alipay doc: treat TRADE_SUCCESS or TRADE_FINISHED as success
        if trade_status != "TRADE_SUCCESS" && trade_status != "TRADE_FINISHED" {
            return Err(PayError::Other(format!(
                "trade_status not success: {}",
                trade_status
            )));
        }

        // service mode extra check: if configured, ensure notify contains expected sub_merchant_id (if exists)
        if let crate::config::Mode::Service = self.mode {
            if let Some(cfg_sub) = &self.cfg.sub_merchant_id {
                if let Some(notify_sub) = &sub_merchant_id {
                    if notify_sub != cfg_sub {
                        return Err(PayError::Other(format!(
                            "sub_merchant_id mismatch notify={} cfg={}",
                            notify_sub, cfg_sub
                        )));
                    }
                }
            }
        }

        // collect other params
        let mut others = HashMap::new();
        for (k, v) in params.iter() {
            if k != "sign" && k != "sign_type" {
                others.insert(k.clone(), v.clone());
            }
        }

        Ok(AlipayNotifyData {
            app_id,
            out_trade_no,
            trade_no,
            trade_status,
            total_amount,
            seller_id,
            sub_merchant_id,
            others,
        })
    }

    /// success response body to reply to Alipay
    pub fn success_response(&self) -> &'static str {
        "success"
    }
}