1use br_reqwest::Client;
2use std::collections::{HashMap};
3use base64::{Engine};
4use base64::engine::general_purpose::STANDARD;
5use chrono::{Local};
6use crate::{PayMode, PayNotify, RefundNotify, RefundStatus, TradeState, TradeType, Types};
7use json::{array, object, JsonValue};
8use openssl::hash::MessageDigest;
9use openssl::pkey::{PKey};
10use openssl::rsa::Rsa;
11use openssl::sign::{Signer, Verifier};
12use urlencoding::{decode, encode};
13use log::error;
14
15#[derive(Clone)]
16pub struct AliPay {
17 pub appid: String,
19 pub sp_mchid: String,
21 pub app_auth_token: String,
23 pub app_private: String,
25 pub content_encryp: String,
27 pub alipay_public_key: String,
29 pub notify_url: String,
30}
31impl AliPay {
32 pub fn sign(&mut self, txt: &str) -> Result<JsonValue, String> {
33 let t = self.app_private.as_bytes().chunks(64).map(|chunk| std::str::from_utf8(chunk).unwrap_or("")).collect::<Vec<&str>>().join("\n");
34 let cart = format!("-----BEGIN PRIVATE KEY-----\n{t}\n-----END PRIVATE KEY-----\n");
35
36 let rsa = match Rsa::private_key_from_pem(cart.as_bytes()) {
38 Ok(e) => e,
39 Err(e) => return Err(e.to_string())
40 };
41
42 let pkey = match PKey::from_rsa(rsa) {
43 Ok(e) => e,
44 Err(e) => return Err(e.to_string())
45 };
46
47 let mut signer = match Signer::new(MessageDigest::sha256(), &pkey) {
49 Ok(e) => e,
50 Err(e) => return Err(e.to_string())
51 };
52 match signer.update(txt.as_bytes()) {
53 Ok(()) => {}
54 Err(e) => return Err(e.to_string())
55 };
56 let signature = match signer.sign_to_vec() {
58 Ok(e) => e,
59 Err(e) => return Err(e.to_string())
60 };
61 Ok(STANDARD.encode(signature).into())
63 }
64 pub fn http(&mut self, method: &str, biz_content: JsonValue) -> Result<JsonValue, String> {
65 let mut http = Client::new();
66 let sign = "";
68
69 let now = Local::now();
70 let timestamp = now.format("%Y-%m-%d %H:%M:%S").to_string();
71
72 let mut data = object! {
73 "charset":"UTF-8",
74 "method":method,
75 "app_id":self.appid.clone(),
76 "app_private_key":self.app_private.clone(),
77 "version":"1.0",
78 "sign_type":"RSA2",
79 "timestamp":timestamp,
80 "alipay_public_key":self.alipay_public_key.clone(),
81 "sign":sign
82 };
83 if !self.app_auth_token.is_empty() {
84 data["app_auth_token"] = self.app_auth_token.clone().into();
85 }
86 if method.contains("alipay.trade.") {
87 data["notify_url"] = self.notify_url.clone().into();
88 }
89 for (key, value) in biz_content.entries() {
90 data[key] = value.clone()
91 }
92 let mut map = HashMap::new();
93 for (key, value) in data.entries() {
94 if key == "sign" {
95 continue;
96 }
97 if value.is_empty() {
98 continue;
99 }
100 map.insert(key, value);
101 }
102
103 let mut keys: Vec<_> = map.keys().cloned().collect();
104 keys.sort();
105 let mut txt = vec![];
106 for key in keys {
107 txt.push(format!("{}={}", key, map.get(&key).unwrap()));
108 }
109 let txt = txt.join("&");
110 data["sign"] = self.sign(&txt)?;
111
112 let mut new_data = object! {};
113 for (key, value) in data.entries() {
114 let t = encode(value.to_string().as_str()).to_string();
115 new_data[key] = t.into();
116 }
117 data = new_data;
118 let url = "https://openapi.alipay.com/gateway.do".to_string();
119 let res = match method {
120 "alipay.trade.wap.pay" => {
121 let tt = http.get(url.as_str()).query(data);
122 return Ok(tt.url.clone().into());
123 }
124 _ => {
125 match http.get(&url).query(data).form_data(biz_content).send() {
126 Ok(e) => e,
127 Err(e) => return Err(e.to_string())
128 }
129 }
130 };
131
132 let res = res.json()?;
133 if res.has_key("error_response") {
134 return Err(res["error_response"]["sub_msg"].to_string());
135 }
136 let key = method.replace(".", "_");
137 let key = format!("{key}_response");
138 let data = res[key].clone();
139 if data.has_key("code") {
140 if data["code"] != "10000" {
141 Err(data["sub_msg"].to_string())
142 } else {
143 Ok(data)
144 }
145 } else {
146 Err(data.to_string())
147 }
148 }
149 pub fn https(&mut self, method: &str, biz_content: JsonValue) -> Result<JsonValue, String> {
150 let mut http = Client::new();
151 let sign = "";
153
154 let now = Local::now();
155 let timestamp = now.format("%Y-%m-%d %H:%M:%S").to_string();
156
157 let mut data = object! {
158 "charset":"UTF-8",
159 "method":method,
160 "app_id":self.appid.clone(),
161 "app_private_key":self.app_private.clone(),
162 "version":"1.0",
163 "sign_type":"RSA2",
164 "timestamp":timestamp,
165 "alipay_public_key":self.alipay_public_key.clone(),
166 "sign":sign
167 };
168 if !self.app_auth_token.is_empty() {
169 data["app_auth_token"] = self.app_auth_token.clone().into();
170 }
171 if method.contains("alipay.trade.") {
172 data["notify_url"] = self.notify_url.clone().into();
173 }
174 for (key, value) in biz_content.entries() {
175 data[key] = value.clone()
176 }
177 let mut map = HashMap::new();
178 for (key, value) in data.entries() {
179 if key == "sign" {
180 continue;
181 }
182 if value.is_empty() {
183 continue;
184 }
185 map.insert(key, value);
186 }
187
188 let mut keys: Vec<_> = map.keys().cloned().collect();
189 keys.sort();
190 let mut txt = vec![];
191 for key in keys {
192 txt.push(format!("{}={}", key, map.get(&key).unwrap()));
193 }
194 let txt = txt.join("&");
195 data["sign"] = self.sign(&txt)?;
196
197 let mut new_data = object! {};
198 for (key, value) in data.entries() {
199 let t = encode(value.to_string().as_str()).to_string();
200 new_data[key] = t.into();
201 }
202 data = new_data;
203 let url = "https://openapi.alipay.com/gateway.do".to_string();
204 let res = match method {
205 "alipay.trade.wap.pay" => {
206 let tt = http.get(url.as_str()).query(data);
207 return Ok(tt.url.clone().into());
208 }
209 _ => {
210 match http.get(&url).query(data).form_data(biz_content).send() {
211 Ok(e) => e,
212 Err(e) => return Err(e.to_string())
213 }
214 }
215 };
216
217 let res = res.json()?;
218 if res.has_key("error_response") {
219 return Err(res["error_response"]["sub_msg"].to_string());
220 }
221 let key = method.replace(".", "_");
222 let key = format!("{key}_response");
223 let data = res[key].clone();
224 Ok(data)
225 }
226}
227impl PayMode for AliPay {
228
229
230 fn get_sub_mchid(&mut self, sub_mchid: &str) -> Result<JsonValue, String> {
231 let res = self.https("alipay.open.agent.signstatus.query", object! {
232 "biz_content":{
233 "pid":sub_mchid,
234 "product_codes":array!["QUICK_WAP_WAY"]
235 }
236 })?;
237 if !res["code"].eq("10000") {
238 return Err(res["msg"].to_string());
239 }
240 for item in res["sign_status_list"].members() {
241 if item["status"].eq("none") {
242 return Err(format!("{} 未开通", res["product_name"]));
243 }
244 }
245 Ok(true.into())
246 }
247
248 fn notify(&mut self, data: JsonValue) -> Result<JsonValue, String> {
249 let sign = match STANDARD.decode(data["sign"].to_string()) {
250 Ok(e) => e,
251 Err(e) => return Err(format!("decode sign: {e}"))
252 };
253 let mut map = HashMap::new();
254 for (key, value) in data.entries() {
255 if key == "sign" {
256 continue;
257 }
258 if value.is_empty() {
259 continue;
260 }
261 map.insert(key, value);
262 }
263
264 let mut keys: Vec<_> = map.keys().cloned().collect();
265 keys.sort();
266
267 let mut txt = vec![];
268 for key in keys {
269 let value = decode(map.get(&key).unwrap().to_string().as_str()).unwrap().to_string();
270 txt.push(format!("{key}={value}"));
271 }
272 let txt = txt.join("&");
273
274 let public_key_pem = self.alipay_public_key.clone();
275 let public_key_pem = format!("-----BEGIN PUBLIC KEY-----\n{public_key_pem}\n-----END PUBLIC KEY-----\n");
276 let public_key = match PKey::public_key_from_pem(public_key_pem.as_bytes()) {
277 Ok(e) => e,
278 Err(e) => return Err(format!("Invalid public key: {e}"))
279 };
280
281 let mut verifier = match Verifier::new(MessageDigest::sha256(), &public_key) {
283 Ok(e) => e,
284 Err(e) => return Err(format!("Invalid verifier: {e}"))
285 };
286 match verifier.update(txt.as_bytes()) {
287 Ok(_) => {}
288 Err(_) => return Err("Invalid transaction signature".to_string())
289 };
290
291 let result = match verifier.verify(&sign) {
292 Ok(e) => e,
293 Err(_) => return Err("Invalid transaction signature".to_string())
294 };
295 if !result {
296 return Err("sign error".to_string());
297 }
298 if data.has_key("service") {
299 let service = data["service"].to_string();
300 if service.as_str() == "alipay.service.check" {
301 let sign_txt = "<success>true</success>";
302 let sign = self.sign(sign_txt)?;
303 let text = format!(r#"<?xml version="1.0" encoding="GBK"?><alipay><response><success>true</success></response><sign>{sign}</sign><sign_type>RSA2</sign_type></alipay>"#);
304 return Ok(JsonValue::String(text));
305 }
306 }
307 Ok(result.into())
308 }
309
310 fn config(&mut self) -> JsonValue {
311 todo!()
312 }
313
314 fn login(&mut self, code: &str) -> Result<JsonValue, String> {
315 let res = self.https("alipay.system.oauth.token", object! {
316 "grant_type":"authorization_code",
317 "code":code
318 })?;
319 Ok(res)
320 }
321
322 fn auth(&mut self, code: &str) -> Result<JsonValue, String> {
323 let biz_content = object! {
324 "biz_content":{
325 "grant_type":"authorization_code",
326 "code":code
327 }
328 };
329 let res = match self.http("alipay.open.auth.token.app", biz_content) {
330 Ok(e) => e,
331 Err(e) => {
332 error!("Err: {e:#}");
333 return Err(e);
334 }
335 };
336
337 let data = object! {
338 data:res.clone(),
339 user_id : res["user_id"].clone(),
340 auth_app_id : res["auth_app_id"].clone(),
341 re_expires_in :res["re_expires_in"].clone(),
342 app_auth_token:res["app_auth_token"].clone(),
343 app_refresh_token:res["app_refresh_token"].clone()
344 };
345 Ok(data)
346 }
347
348 fn pay(&mut self, types: Types, sub_mchid: &str, out_trade_no: &str, description: &str, total_fee: f64, sp_openid: &str) -> Result<JsonValue, String> {
349 let mut api = "";
350 let mut order = object! {
351 out_trade_no:out_trade_no,
352 total_amount:total_fee,
353 subject:description,
354 product_code:"JSAPI_PAY",
355 op_app_id:sub_mchid,
356 buyer_open_id:sp_openid
357 };
358
359 match types {
360 Types::MiniJsapi => {
361 api = "alipay.trade.create";
362 order["product_code"] = "JSAPI_PAY".into();
363 order["op_app_id"] = self.appid.clone().into();
364 order["buyer_open_id"] = sp_openid.into();
365 self.app_auth_token = "".to_string();
366 }
367 Types::Jsapi => {
368 api = "alipay.trade.create";
369 order["product_code"] = "JSAPI_PAY".into();
370 }
371 Types::H5 => {
372 api = "alipay.trade.wap.pay";
373 order["product_code"] = "QUICK_WAP_WAY".into();
374 }
375 Types::Native => {
376 api = "alipay.trade.wap.pay";
377 order["product_code"] = "QUICK_WAP_WAY".into();
378 }
379 _ => {
380 order["product_code"] = "JSAPI_PAY".into();
381 }
382 };
383 match self.http(api, object! {"biz_content":order}) {
384 Ok(e) => {
385 match types {
386 Types::Jsapi => {}
387 Types::Native => {}
388 Types::H5 => {
389 return Ok(object! {url:e});
390 }
391 Types::MiniJsapi => {
392 println!("alipay.trade.wap.pay:{e:#}");
393 return Ok(e);
394 }
395 Types::App => {}
396 Types::Micropay => {}
397 }
398 Ok(e)
399 }
400 Err(e) => {
401 println!("Err: {e:#}");
402 Err(e)
403 }
404 }
405 }
406
407 fn micropay(&mut self, auth_code: &str, sub_mchid: &str, out_trade_no: &str, description: &str, total_fee: f64, org_openid: &str, _ip: &str) -> Result<JsonValue, String> {
408 let order = object! {
409 out_trade_no:out_trade_no,
410 total_amount:total_fee,
411 subject:description,
412 seller_id:sub_mchid,
413 auth_code:auth_code,
414 scene:"bar_code",
415 operator_id:org_openid,
416 };
417
418 match self.http("alipay.trade.create", object! {"biz_content":order}) {
419 Ok(e) => {
420 Ok(e)
421 }
422 Err(e) => {
423 println!("Err: {e:#}");
424 Err(e)
425 }
426 }
427 }
428
429
430 fn close(&mut self, out_trade_no: &str, sub_mchid: &str) -> Result<JsonValue, String> {
431 let order = object! {
432 "biz_content"=> object! {
433 out_trade_no:out_trade_no,
434 operator_id:sub_mchid
435 }
436 };
437 match self.http("alipay.trade.close", order) {
438 Ok(_) => {
439 Ok(true.into())
440 }
441 Err(e) => {
442 if e.contains("交易不存在") {
443 return Ok(true.into());
444 }
445 Err(e)
446 }
447 }
448 }
449
450 fn pay_query(&mut self, out_trade_no: &str, sub_mchid: &str) -> Result<JsonValue, String> {
451 let order = object! {
452 "biz_content"=> object! {
453 out_trade_no:out_trade_no
454 }
455 };
456 match self.http("alipay.trade.query", order) {
457 Ok(e) => {
458 if e.has_key("code") && e["code"].as_str().unwrap() != "10000" {
459 return Err(e["msg"].to_string());
460 }
461 let buyer_open_id = if e.has_key("buyer_open_id") {
462 e["buyer_open_id"].to_string()
463 } else {
464 e["buyer_user_id"].to_string()
465 };
466 let res = PayNotify {
467 trade_type: TradeType::None,
468 out_trade_no: e["out_trade_no"].to_string(),
469 sp_mchid: "".to_string(),
470 sub_mchid: sub_mchid.to_string(),
471 sp_appid: "".to_string(),
472 transaction_id: e["trade_no"].to_string(),
473 success_time: PayNotify::alipay_time(e["send_pay_date"].as_str().unwrap()),
474 sp_openid: buyer_open_id.clone(),
475 sub_openid: buyer_open_id.clone(),
476 total: (e["total_amount"].to_string().parse::<f64>().unwrap_or(0.0)),
477 payer_total: (e["total_amount"].to_string().parse::<f64>().unwrap_or(0.0)),
478 currency: "CNY".to_string(),
479 payer_currency: "CNY".to_string(),
480 trade_state: TradeState::from(e["trade_status"].as_str().unwrap()),
481 };
482 Ok(res.json())
483 }
484 Err(e) => Err(e)
485 }
486 }
487
488 fn pay_micropay_query(&mut self, _out_trade_no: &str, _sub_mchid: &str) -> Result<JsonValue, String> {
489 todo!()
490 }
491
492 fn pay_notify(&mut self, _nonce: &str, _ciphertext: &str, _associated_data: &str) -> Result<JsonValue, String> {
493 todo!()
494 }
495
496 fn refund(&mut self, sub_mchid: &str, out_trade_no: &str, transaction_id: &str, out_refund_no: &str, amount: f64, total: f64, _currency: &str) -> Result<JsonValue, String> {
497 let body = object! {
498 "biz_content"=> object! {
499 "trade_no"=>transaction_id,
500 "out_trade_no"=>out_trade_no,
501 "out_request_no"=>out_refund_no,
502 "refund_amount"=>format!("{:.2}",amount),
503 }
504 };
505 match self.http("alipay.trade.refund", body.clone()) {
506 Ok(e) => {
507 if e.has_key("code") && e["code"].as_str().unwrap() != "10000" {
508 return Err(e["msg"].to_string());
509 }
510 let res = RefundNotify {
511 out_trade_no: e["out_trade_no"].to_string(),
512 refund_no: out_refund_no.to_string(),
513 sp_mchid: "".to_string(),
514 sub_mchid: sub_mchid.to_string(),
515 transaction_id: e["trade_no"].to_string(),
516 refund_id: out_refund_no.to_string(),
517 success_time: PayNotify::alipay_time(e["gmt_refund_pay"].as_str().unwrap()),
518 total,
519 refund: e["refund_fee"].to_string().parse::<f64>().unwrap(),
520 payer_total: e["refund_fee"].to_string().parse::<f64>().unwrap(),
521 payer_refund: e["send_back_fee"].to_string().parse::<f64>().unwrap(),
522 status: RefundStatus::from(e["fund_change"].as_str().unwrap()),
523 };
524 Ok(res.json())
525 }
526 Err(e) => Err(e)
527 }
528 }
529
530 fn micropay_refund(&mut self, _sub_mchid: &str, _out_trade_no: &str, _transaction_id: &str, _out_refund_no: &str, _amount: f64, _total: f64, _currency: &str, _refund_text: &str) -> Result<JsonValue, String> {
531 todo!()
532 }
533
534 fn refund_notify(&mut self, _nonce: &str, _ciphertext: &str, _associated_data: &str) -> Result<JsonValue, String> {
535 todo!()
536 }
537
538 fn refund_query(&mut self, trade_no: &str, out_refund_no: &str, sub_mchid: &str) -> Result<JsonValue, String> {
539 let body = object! {
540 "biz_content"=> object! {
541 "out_request_no"=>out_refund_no,
542 "trade_no"=>trade_no,
543 }
544 };
545 match self.http("alipay.trade.fastpay.refund.query", body.clone()) {
546 Ok(e) => {
547 if e.has_key("code") && e["code"].as_str().unwrap() != "10000" {
548 return Err(e["msg"].to_string());
549 }
550 let res = RefundNotify {
551 out_trade_no: e["out_trade_no"].to_string(),
552 refund_no: e["out_request_no"].to_string(),
553 sp_mchid: "".to_string(),
554 sub_mchid: sub_mchid.to_string(),
555 transaction_id: e["trade_no"].to_string(),
556 refund_id: e["out_request_no"].to_string(),
557 success_time: Local::now().timestamp(),
558 total: e["total_amount"].to_string().parse::<f64>().unwrap(),
559 payer_total: e["total_amount"].to_string().parse::<f64>().unwrap(),
560 refund: e["refund_amount"].to_string().parse::<f64>().unwrap(),
561 payer_refund: e["refund_amount"].to_string().parse::<f64>().unwrap(),
562 status: RefundStatus::from(e["refund_status"].as_str().unwrap()),
563 };
564 Ok(res.json())
565 }
566 Err(e) => Err(e)
567 }
568 }
569
570 fn incoming(&mut self, _business_code: &str, _contact_info: JsonValue, _subject_info: JsonValue, _business_info: JsonValue, _settlement_info: JsonValue, _bank_account_info: JsonValue) -> Result<JsonValue, String> {
571 todo!()
572 }
573}