package controllers
import (
"Xcx_New/enums"
"Xcx_New/models"
"Xcx_New/service"
"bytes"
"crypto/md5"
"encoding/hex"
"encoding/json"
"encoding/xml"
"errors"
"fmt"
"github.com/astaxie/beego"
"io/ioutil"
"math/rand"
"net/http"
"regexp"
"sort"
"strconv"
"strings"
"time"
)
const (
pay_url = "https://api.mch.weixin.qq.com/pay/unifiedorder"
)
type PayApiController struct {
BaseServeAPIController
parameters map[string]string
resultParameters map[string]string
payUrl string
}
type PayApiController2 struct {
beego.Controller
parameters map[string]string
resultParameters map[string]string
payUrl string
}
func PayApiRegistRouters() {
beego.Router("/api/my/service", &PayApiController{}, "Get:GetMyService")
beego.Router("/api/product", &PayApiController{}, "Get:GetProduct")
beego.Router("/api/pay", &PayApiController{}, "Get:GetPayUrl")
beego.Router("/api/pay/transfer", &PayApiController{}, "Post:PostTransferStatus")
beego.Router("/api/order", &PayApiController{}, "Post:PostOrderInfo")
beego.Router("/api/order/get", &PayApiController{}, "Get:GetOrderInfo")
beego.Router("/api/pay/notify", &PayApiController2{}, "Post:WxPaySuccessNotify")
beego.Router("/api/order/cancel", &PayApiController{}, "Post:CancelOrder")
beego.Router("/api/order/list", &PayApiController{}, "Get:GetOrderList")
beego.Router("/api/order/hetong", &PayApiController{}, "Get:GetHeTong")
beego.Router("/api/order/hetong", &PayApiController{}, "Post:CreateHeTong")
}
func (c *PayApiController) GetOrderList() {
adminUserInfo := c.GetAdminUserInfo()
serviceOrderList, err := service.GetOrderList(adminUserInfo.CurrentOrgId)
if err == nil {
c.ServeSuccessJSON(map[string]interface{}{
"list": serviceOrderList,
})
} else {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
}
}
func (c *PayApiController) CancelOrder() {
adminUserInfo := c.GetAdminUserInfo()
orderId, _ := c.GetInt64("id", 0)
if orderId == 0 {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeParamWrong)
return
}
order, err := service.FindServeOrderByID(adminUserInfo.CurrentOrgId, orderId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if order == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
return
}
errs := service.UpdateOrderStatus(adminUserInfo.CurrentOrgId, orderId)
if errs == nil {
c.ServeSuccessJSON(map[string]interface{}{
"msg": "取消成功",
})
} else {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
}
}
func (c *PayApiController) PostTransferStatus() {
adminUserInfo := c.GetAdminUserInfo()
orderId, _ := c.GetInt64("id", 0)
if orderId == 0 {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeParamWrong)
return
}
order, err := service.FindServeOrderByID(adminUserInfo.CurrentOrgId, orderId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if order == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
return
}
errs := service.UpdateOrderPayType(adminUserInfo.CurrentOrgId, orderId)
if errs == nil {
c.ServeSuccessJSON(map[string]interface{}{
"msg": "确认成功",
})
} else {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
}
}
func (c *PayApiController) GetOrderInfo() {
adminUserInfo := c.GetAdminUserInfo()
orderId, _ := c.GetInt64("id", 0)
order, err := service.FindServeOrderByID(adminUserInfo.CurrentOrgId, orderId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if order == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
return
}
orderInfo, err := service.FindOrderInfomationByID(order.OrderNumber, adminUserInfo.CurrentOrgId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if orderInfo == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
return
}
c.ServeSuccessJSON(map[string]interface{}{
"total": order.PaymentAmount,
"orderNumber": order.OrderNumber,
"OrgName": adminUserInfo.Orgs[adminUserInfo.CurrentOrgId].OrgName,
"order": order,
"orderInfo": orderInfo,
})
}
func (c *PayApiController) PostOrderInfo() {
adminUserInfo := c.GetAdminUserInfo()
amount, _ := c.GetInt64("amount", 0)
productId, _ := c.GetInt64("id", 0)
product, err := service.FindProductByID(productId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if product == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeProductError)
return
}
timestamp := c.GetTimestamp()
//自定义订单号
bill_no := "S" + strconv.FormatInt(timestamp, 10) + strconv.FormatInt(adminUserInfo.CurrentOrgId, 10)
order := models.ServeOrder{
OrgId: adminUserInfo.CurrentOrgId,
Period: amount * 12,
Status: 1,
CreatedTime: time.Now().Unix(),
UpdatedTime: time.Now().Unix(),
OrderNumber: bill_no,
OrderStatus: 1,
PayableAmount: float64(amount) * product.Price,
PaymentAmount: float64(amount) * product.Price,
ServeName: product.ServeName,
ServeDesc: product.ServeDesc,
OrderExpireTime: time.Now().Unix() + (7 * 24 * 3600),
Quantity: amount,
Price: product.Price,
ServeId: productId,
}
infomation := models.ServeOrderInfomation{
OrgId: adminUserInfo.CurrentOrgId,
OrderNumber: bill_no,
Status: 1,
ProductId: product.ID,
ProductName: product.ServeName,
ProductDesc: product.ServeDesc,
Price: product.Price,
Quantity: amount,
MarketPrice: product.OriginalPrice,
}
service.CreateOrderRecord(&order)
service.CreateOrderInfomation(&infomation)
c.ServeSuccessJSON(map[string]interface{}{
"order": order,
"msg": "提交订单成功",
})
}
func (c *PayApiController) GetProduct() {
adminUserInfo := c.GetAdminUserInfo()
products, err := service.FindAllProduct()
subscibe, _ := service.FindServiceSubscibeByOrgId(adminUserInfo.CurrentOrgId)
if err == nil {
c.ServeSuccessJSON(map[string]interface{}{
"products": products,
"subscibe": subscibe,
"serviceTime": time.Now().Unix(),
//"OrgName": adminUserInfo.Orgs[adminUserInfo.CurrentOrgId].OrgName,
})
} else {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeSystemError)
}
}
func (c *PayApiController) GetMyService() {
adminUserInfo := c.GetAdminUserInfo()
subscibe, err := service.FindServiceSubscibeByOrgId(adminUserInfo.CurrentOrgId)
if err == nil {
c.ServeSuccessJSON(map[string]interface{}{
"subscibe": subscibe,
"serviceTime": time.Now().Unix(),
})
} else {
}
}
func (c *PayApiController) GetPayUrl() {
adminUserInfo := c.GetAdminUserInfo()
orderId, _ := c.GetInt64("id", 0)
if orderId == 0 {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeParamWrong)
return
}
order, err := service.FindServeOrderByID(adminUserInfo.CurrentOrgId, orderId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if order == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
return
}
orderInfo, err := service.FindOrderInfomationByID(order.OrderNumber, adminUserInfo.CurrentOrgId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if orderInfo == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
return
}
totalPrice := orderInfo.Price * float64(orderInfo.Quantity)
totalFee := fmt.Sprintf("%.0f", totalPrice*100)
c.SetParameter("out_trade_no", orderInfo.OrderNumber)
c.SetParameter("total_fee", totalFee)
c.SetParameter("trade_type", "NATIVE")
c.SetParameter("body", order.ServeName)
notify_url := beego.AppConfig.String("httpdomain") + "/api/pay/notify"
c.SetParameter("product_id", orderInfo.OrderNumber)
c.SetParameter("notify_url", notify_url)
c.SetParameter("spbill_create_ip", c.GetClientIp())
fmt.Println(c.GetClientIp())
url, err := c.GetPayCodeUrl()
fmt.Println(err)
if err == nil {
c.SetPayUrl(url)
c.ServeSuccessJSON(map[string]interface{}{
"payUrl": url,
"price": order.PayableAmount,
//": adminUserInfo.Orgs[adminUserInfo.CurrentOrgId].OrgName,
})
} else {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
// utils.LogError(err)
}
}
func (this *PayApiController2) WxPaySuccessNotify() {
type PaySuccessXmlResp struct {
ReturnCode string `xml:"return_code"`
Appid string `xml:"appid"`
BankType string `xml:"bank_type"`
CashFee int64 `xml:"cash_fee"`
FeeType string `xml:"fee_type"`
IsSubscribe string `xml:"is_subscribe"`
MchId string `xml:"mch_id"`
NonceStr string `xml:"nonce_str"`
OpenId string `xml:"openid"`
OutTradeNo string `xml:"out_trade_no"`
ResultCode string `xml:"result_code"`
Sign string `xml:"sign"`
TimeEnd string `xml:"time_end"`
TotalFee string `xml:"total_fee"`
TradeType string `xml:"trade_type"`
TransactionId string `xml:"transaction_id"`
}
defer this.Ctx.Request.Body.Close()
result, _ := ioutil.ReadAll(this.Ctx.Request.Body)
res := new(PaySuccessXmlResp)
xmlErr := xml.Unmarshal(result, &res)
if xmlErr != nil {
}
order, _ := service.FindServeOrderByOrderNumber(res.OutTradeNo)
service.UpdateOrder(order.OrderNumber, order.Quantity, order.OrgId, res.TransactionId, order.PeriodEnd, order.PeriodStart)
this.SetResultParameter("return_code", res.ReturnCode)
results := this.ParamsToXml2(this.resultParameters)
this.Ctx.WriteString(results)
this.ServeXML()
}
func (this *PayApiController) GetTimestamp() int64 {
return time.Now().UnixNano() / 1000000 //毫秒
}
func (this *PayApiController) GetClientIp() string {
ip := this.Ctx.Request.Header.Get("Remote_addr")
if ip == "" {
ip = this.Ctx.Request.RemoteAddr
}
if strings.Contains(ip, ":") {
ip = this.Substr(ip, 0, strings.Index(ip, ":"))
}
return ip
}
//截取字符串 start 起点下标 end 终点下标(不包括)
func (this *PayApiController) Substr(str string, start int, end int) string {
rs := []rune(str)
length := len(rs)
if start < 0 || start > length {
return ""
}
if end < 0 || end > length {
return ""
}
return string(rs[start:end])
}
//设置请求参数
func (this *PayApiController) SetParameter(key string, value string) {
if this.parameters == nil {
this.parameters = make(map[string]string)
}
this.parameters[key] = value
}
//设置请求参数
func (this *PayApiController2) SetResultParameter(key string, value string) {
if this.resultParameters == nil {
this.resultParameters = make(map[string]string)
}
this.resultParameters[key] = value
}
//设置prepay_id
func (this *PayApiController) SetPayUrl(payUrl string) {
this.payUrl = payUrl
}
type XmlResp struct {
Return_code string `xml:"return_code"`
Return_msg string `xml:"return_msg"`
Result_code string `xml:"result_code"`
Err_code string `xml:"err_code"`
Err_code_des string `xml:"err_code_des"`
Prepay_id string `xml:"prepay_id"`
Code_url string `xml:"code_url"`
}
func (this *PayApiController) GetPayCodeUrl() (string, error) {
strXml, err := this.CreateXml()
if err != nil {
return "", this.Error("get pay_code_url error", err)
}
result, err := this.http_post(pay_url, strXml)
if err != nil {
return "", this.Error("get pay_code_url error", err)
}
res := new(XmlResp)
xmlErr := xml.Unmarshal(result, &res)
if xmlErr != nil {
return "", this.Error("get pay_code_url xml error", xmlErr)
}
if res.Return_code != "SUCCESS" {
return "", this.Error("get pay_code_url result error: "+res.Return_msg, nil)
}
if res.Result_code != "SUCCESS" {
return "", this.Error("get pay_code_url result error: "+res.Err_code+"-"+res.Err_code_des, nil)
}
if res.Code_url == "" {
return "", this.Error("get pay_code_url result error: not get pay_code_url", nil)
}
return res.Code_url, nil
}
func (this *PayApiController) http_post(url string, xml string) ([]byte, error) {
bc := &http.Client{
Timeout: 30 * time.Second, //设置超时时间30s
}
res, err := bc.Post(url, "text/xml:charset=UTF-8", strings.NewReader(xml))
if err != nil {
return nil, this.Error("post", err)
}
result, err := ioutil.ReadAll(res.Body)
res.Body.Close()
if err != nil {
return nil, this.Error("post result err", err)
}
return result, nil
}
func (this *PayApiController) CreateXml() (string, error) {
//检测必填参数
if this.parameters["out_trade_no"] == "" {
return "", this.Error("缺少统一支付接口必填参数out_trade_no(商户订单号)!", nil)
} else if this.parameters["body"] == "" {
return "", this.Error("缺少统一支付接口必填参数body(商品描述)!", nil)
} else if this.parameters["total_fee"] == "" {
return "", this.Error("缺少统一支付接口必填参数total_fee(交易金额)!", nil)
} else if this.parameters["notify_url"] == "" {
return "", this.Error("缺少统一支付接口必填参数notify_url(异步接收微信支付结果通知的回调地址)!", nil)
} else if this.parameters["trade_type"] == "" {
return "", this.Error("缺少统一支付接口必填参数trade_type(交易类型)!", nil)
} else if this.parameters["spbill_create_ip"] == "" {
return "", this.Error("缺少统一支付接口必填参数spbill_create_ip(终端ip)", nil)
} else if this.parameters["trade_type"] == "NATIVE" && this.parameters["product_id"] == "" {
return "", this.Error("统一支付接口中,缺少必填参数product_id!trade_type为NATIVE时,product_id为必填参数!", nil)
}
this.parameters["appid"] = beego.AppConfig.String("appid") //公众账号ID
this.parameters["mch_id"] = beego.AppConfig.String("mchid") //商户号
this.parameters["nonce_str"] = this.CreateNoncestr(32) //随机字符串
this.parameters["sign"] = this.GetSign(this.parameters, beego.AppConfig.String("key")) //签名
return this.ParamsToXml(this.parameters), nil
}
//产生随机字符串,不长于32位
func (this *PayApiController) CreateNoncestr(length int) (nonceStr string) {
chars := "abcdefghijklmnopqrstuvwxyz0123456789"
for i := 0; i < length; i++ {
idx := rand.Intn(len(chars) - 1)
nonceStr += chars[idx : idx+1]
}
return
}
func (this *PayApiController) Error(strMsg string, err error) error {
if err == nil {
return errors.New(strMsg)
} else {
return errors.New(strMsg + ": " + err.Error())
}
}
//格式化参数,签名过程需要使用
func (this *PayApiController) FormatParams(paramsMap map[string]string) string {
//STEP 1, 对key进行升序排序.
var sorted_keys []string
for k, _ := range paramsMap {
sorted_keys = append(sorted_keys, k)
}
sort.Strings(sorted_keys)
//STEP2, 对key=value的键值对用&连接起来,略过空值
var paramsStr []string
for _, k := range sorted_keys {
v := fmt.Sprintf("%v", strings.TrimSpace(paramsMap[k]))
if v != "" {
paramsStr = append(paramsStr, fmt.Sprintf("%s=%s", k, v))
}
}
return strings.Join(paramsStr, "&")
}
//生成签名
func (this *PayApiController) GetSign(paramsMap map[string]string, wxKey string) string {
//STEP 1:按字典序排序参数
paramsStr := this.FormatParams(paramsMap)
//STEP 2:在string后加入KEY
signStr := paramsStr + "&key=" + wxKey
//STEP 3:MD5加密
sign := md5.New()
sign.Write([]byte(signStr))
//STEP 3:所有字符转为大写
return strings.ToUpper(hex.EncodeToString(sign.Sum(nil)))
}
//xml结构
func (this *PayApiController) ParamsToXml(data map[string]string) string {
fmt.Println(data)
buf := bytes.NewBufferString("")
for k, v := range data {
str := ""
flag, _ := regexp.MatchString("^\\d+\\.?\\d*$", v)
if flag {
str = "%s"
}
buf.WriteString(fmt.Sprintf("<%s>"+str+"%s>", k, v, k))
}
buf.WriteString("")
return buf.String()
}
//xml结构
func (this *PayApiController2) ParamsToXml2(data map[string]string) string {
fmt.Println(data)
buf := bytes.NewBufferString("")
for k, v := range data {
str := ""
flag, _ := regexp.MatchString("^\\d+\\.?\\d*$", v)
if flag {
str = "%s"
}
buf.WriteString(fmt.Sprintf("<%s>"+str+"%s>", k, v, k))
}
buf.WriteString("")
return buf.String()
}
func (this *PayApiController) SetUrl(payUrl string) {
this.payUrl = payUrl
}
func (c *PayApiController) GetHeTong() {
order_id, _ := c.GetInt64("order_id", 0)
if order_id <= 0 {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeParamWrong)
return
}
admin := c.GetAdminUserInfo()
ht, err := service.GetHetong(admin.CurrentOrgId, order_id)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
c.ServeSuccessJSON(map[string]interface{}{
"ht": ht,
})
}
func (c *PayApiController) CreateHeTong() {
orderId, _ := c.GetInt64("order_id", 0)
if orderId <= 0 {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeParamWrong)
return
}
admin := c.GetAdminUserInfo()
order, err := service.FindServeOrderByID(admin.CurrentOrgId, orderId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if order == nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeServeNotExist)
return
}
ht, err := service.GetHetong(admin.CurrentOrgId, orderId)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeDataException)
return
}
if ht != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeHetongHad)
return
}
var hetong models.ServeOrderContract
err = json.Unmarshal(c.Ctx.Input.RequestBody, &hetong)
if err != nil {
fmt.Println(err)
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeParamWrong)
return
}
hetong.OrderId = orderId
hetong.OrderNumber = order.OrderNumber
hetong.CreatedTime = time.Now().Unix()
hetong.UpdatedTime = time.Now().Unix()
hetong.Status = 1
hetong.OrgId = admin.CurrentOrgId
err = service.CreateHetong(&hetong)
if err != nil {
c.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeCreateHetongFail)
return
}
c.ServeSuccessJSON(map[string]interface{}{
"ht": hetong,
})
}