yyfbaloot/servers/payment/kaya/withdraw.go

package kaya

import (
	"encoding/base64"
	"encoding/json"
	"fmt"
	"net/http"
	"net/url"
	"strconv"
	"strings"
	"time"

	"bet24.com/log"
	"bet24.com/public"
	notification "bet24.com/servers/micros/notification/proto"
	"bet24.com/servers/payment/config"
	"bet24.com/servers/payment/db"
	"github.com/gin-gonic/gin"
)

// 提现请求
func WithdrawOrder(c *gin.Context) {
	obj := db.NewKayaWithdrawReq()
	if err := c.ShouldBind(&obj.In); err != nil {
		log.Debug("%s query params err %v", "kaya.WithdrawOrder", err)
		c.String(http.StatusOK, "fail")
		return
	}

	obj.In.IPAddress = strings.Split(c.Request.RemoteAddr, ":")[0]
	obj.DoAction()
	if obj.Out.OrderID == "" {
		log.Debug("%s GenOrder fail obj.In=%+v obj.Out=%+v", "kaya.WithdrawOrder", obj.In, obj.Out)
		c.String(http.StatusOK, "fail")
		return
	}

	// 请求时，没有传手机号
	if obj.In.Mobile == "" {
		obj.In.Mobile = obj.Out.Tel
	}

	// 请求withdrawOrder的代码
	req := withdraw_req{
		MerchantCode: config.Server.KayaPay.MerchantCode,
		OrderNum:     obj.Out.OrderID,
		Money:        obj.In.Amount,
		Description:  "withdraw",
		Name:         obj.In.RealName,
		BankCode:     obj.In.BankName,
		Number:       obj.In.BankCard,
		NotifyUrl:    config.Server.KayaPay.Url_withdraw_notify,
		FeeType:      0,
		DateTime:     TIME_FORMAT,
	}

	params := url.Values{}
	params.Set("merchantCode", req.MerchantCode)
	params.Set("orderNum", req.OrderNum)
	params.Set("money", strconv.Itoa(req.Money))
	params.Set("description", req.Description)
	params.Set("name", req.Name)
	params.Set("bankCode", req.BankCode)
	params.Set("number", req.Number)
	params.Set("notifyUrl", req.NotifyUrl)
	params.Set("feeType", strconv.Itoa(req.FeeType))
	params.Set("dateTime", req.DateTime)

	// 生成签名
	checkContent := createEncryptStr(params)
	log.Debug("kaya.WithdrawOrder.checkContent ==> %s", checkContent)

	// 商户请求我们接口时使用商户私钥对请求参数 进行加密
	mchPriKey := fmt.Sprintf(`-----BEGIN PRIVATE KEY-----
%s
-----END PRIVATE KEY-----
`, config.Server.KayaPay.MCH_PRIVATE_KEY)

	if err := public.RSA.SetPrivateKey(mchPriKey); err != nil {
		log.Error("kaya.WithdrawOrder set private key :%v ==> %+v", err, params)
		return
	}

	// 私钥加密
	prienctypt, err := public.RSA.PriKeyENCTYPT([]byte(checkContent))
	if err != nil {
		log.Error("kaya.WithdrawOrder RSA.PriKeyENCTYPT err %v", err)
		return
	}

	req.Sign = base64.StdEncoding.EncodeToString(prienctypt)
	// log.Debug("kaya.payOrder req ==> %+v ", params)

	buf, _ := json.Marshal(req)
	respBody := public.HttpPostByJson(config.Server.KayaPay.Url_withdraw_order, string(buf))
	log.Debug("kaya.WithdrawOrder req ==> %+v  resp ==> %+v", params, respBody)

	var resp withdraw_resp
	if err := json.Unmarshal([]byte(respBody), &resp); err != nil {
		log.Error("kaya.WithdrawOrder json unmarshal req ==> %+v resp ==> %+v fail %v", params, respBody, err)
		c.String(http.StatusOK, "fail")
		return
	}

	log.Debug("kaya.WithdrawOrder resp ==> %+v", resp)

	// 请求响应码，00000表示成功，其他失败
	if resp.PlatRespCode != RESP_CODE_SUCCESS {
		log.Error("kaya.WithdrawOrder post return resp fail ==> %+v", resp)
		c.String(http.StatusOK, "fail")
		return
	}

	go notification.AddNotification(obj.In.UserID, notification.Notification_Gold, "")

	// 更新银行信息
	objBank := db.NewBankInfoUp()
	objBank.In.UserID = obj.In.UserID
	objBank.In.RealName = req.Name
	objBank.In.BankCard = req.Number
	objBank.In.BankName = req.BankCode
	objBank.In.Mobile = obj.In.Mobile
	objBank.In.EMail = obj.In.Email
	objBank.In.Address = obj.In.Address
	objBank.DoAction()

	log.Debug("kaya.WithdrawNotify need save bankInfo")

	c.String(http.StatusOK, "success")
	return
}

// 提现通知
func WithdrawNotify(c *gin.Context) {
	var resp withdrawNotify
	if err := c.ShouldBind(&resp); err != nil {
		log.Debug("%s query params err %v", "kaya.WithdrawNotify", err)
		c.String(http.StatusOK, "")
		return
	}

	log.Debug("kaya.WithdrawNotify resp ==> %+v", resp)

	params := url.Values{}
	params.Set("bankCode", resp.BankCode)
	params.Set("description", resp.Description)
	params.Set("fee", strconv.Itoa(resp.Fee))
	params.Set("feeType", strconv.Itoa(resp.FeeType))
	params.Set("money", strconv.Itoa(resp.Money))
	params.Set("name", resp.Name)
	params.Set("number", resp.Number)
	params.Set("orderNum", resp.OrderNum)
	params.Set("platOrderNum", resp.PlatOrderNum)
	params.Set("status", resp.Status)
	params.Set("statusMsg", resp.StatusMsg)

	// 生成签名
	checkContent := createEncryptStr(params)
	log.Debug("kaya.WithdrawNotify checkContent ==> %s", checkContent)

	// 商户使用 商户后台显示的平台公钥 进行解密
	platPubKey := fmt.Sprintf(`-----BEGIN PUBLIC KEY-----
%s
-----END PUBLIC KEY-----
`, config.Server.KayaPay.PLAT_PUBLIC_KEY)

	if err := public.RSA.SetPublicKey(platPubKey); err != nil {
		log.Error("kaya.WithdrawNotify set public key :%v ==> %+v", err, params)
		return
	}

	data, err := base64.StdEncoding.DecodeString(resp.PlatSign)
	if err != nil {
		log.Error("kaya.WithdrawNotify base64.StdEncoding.DecodeString err %v", err)
		return
	}

	// 公钥解密
	pubdecrypt, err := public.RSA.PubKeyDECRYPT(data)
	if err != nil {
		log.Error("kaya.WithdrawNotify RSA.PubKeyDECRYPT err %v", err)
		return
	}

	// 校验信息
	if checkContent != string(pubdecrypt) {
		log.Error("kaya.payOrder 签名失败 ==> %+v", resp)
		return
	}

	log.Debug("kaya.PayNotify 签名成功")

	// 状态:2-成功 4-失败
	if resp.Status != WITHDRAW_SUCCESS && resp.Status != WITHDRAW_FAIL {
		log.Error("kaya.WithdrawNotify req ==> %+v", resp)
		c.String(http.StatusOK, "")
		return
	}

	// 查询账户余额
	balance := queryAccount()

	// 状态(0=Failure 1=Success  2=Pending(Success))
	status := 0
	switch resp.Status {
	case "2": // 2=成功
		status = 1
	case "4": // 4=失败
		status = 0
	default:
		status = 2
	}

	obj := db.NewKayaWithdrawNotify()
	obj.In.OrderID = resp.OrderNum
	obj.In.DfTransactionId = resp.PlatOrderNum
	obj.In.Status = status
	obj.In.DfDesc = fmt.Sprintf("%s，手续费: %d", resp.StatusMsg, resp.Fee)
	obj.In.Balance = balance
	obj.DoAction()

	log.Debug("kaya.WithdrawNotify obj.In=%+v obj.Out=%+v", obj.In, obj.Out)

	// 提现失败,退款
	if resp.Status == WITHDRAW_FAIL {
		go notification.AddNotification(obj.Out.UserID, notification.Notification_Gold, "")
	}

	c.String(http.StatusOK, "success")
	return
}

// 账户余额
func queryAccount() int {
	req := queryAccount_req{
		MerchantCode: config.Server.KayaPay.MerchantCode,
		DateTime:     time.Now().Format(TIME_FORMAT),
	}

	params := url.Values{}
	params.Set("merchantCode", req.MerchantCode)
	params.Set("dateTime", req.DateTime)

	// 生成签名
	checkContent := createEncryptStr(params)
	log.Debug("kaya.queryAccount.checkContent ==> %s", checkContent)

	// 商户请求我们接口时使用商户私钥对请求参数 进行加密
	mchPriKey := fmt.Sprintf(`-----BEGIN PRIVATE KEY-----
%s
-----END PRIVATE KEY-----
`, config.Server.KayaPay.MCH_PRIVATE_KEY)

	if err := public.RSA.SetPrivateKey(mchPriKey); err != nil {
		log.Error("kaya.queryAccount set private key :%v ==> %+v", err, params)
		return -1
	}

	// 私钥加密
	prienctypt, err := public.RSA.PriKeyENCTYPT([]byte(checkContent))
	if err != nil {
		log.Error("kaya.queryAccount RSA.PriKeyENCTYPT err %v", err)
		return -1
	}

	req.Sign = base64.StdEncoding.EncodeToString(prienctypt)
	// log.Debug("kaya.queryAccount req ==> %+v ", params)

	buf, _ := json.Marshal(req)
	respBody := public.HttpPostByJson(config.Server.KayaPay.Url_queryAccount, string(buf))
	log.Debug("kaya.queryAccount req ==> %+v  resp ==> %+v", params, respBody)

	var resp queryAccount_resp
	if err := json.Unmarshal([]byte(respBody), &resp); err != nil {
		log.Error("kaya.queryAccount json unmarshal req ==> %+v resp ==> %+v fail %v", params, respBody, err)
		return -1
	}

	log.Debug("kaya.queryAccount resp ==> %+v", resp)

	// 请求响应码
	if resp.PlatRespCode != RESP_CODE_SUCCESS {
		log.Error("kaya.queryAccount post return resp fail ==> %+v", resp)
		return -1
	}

	balance, err := strconv.Atoi(resp.Balance)
	if err != nil {
		log.Error("kaya.queryAccount get balance %v", err)
		return -1
	}

	return balance
}