package pays_with_go

import (
	"encoding/xml"
	"errors"
	"fmt"
	"gitee.com/shirdonl/pays_with_go/client"
	"gitee.com/shirdonl/pays_with_go/common"
	"gitee.com/shirdonl/pays_with_go/util"
	"io/ioutil"
	"net/http"
	"sort"
	"strings"
)

// WeChatCallback 微信app支付
func WeChatAppCallback(w http.ResponseWriter, r *http.Request) (*common.WeChatPayResult, error) {
	var returnCode = "FAIL"
	var returnMsg = ""
	defer func() {
		formatStr := `<xml><return_code><![CDATA[%s]]></return_code>
                  <return_msg>![CDATA[%s]]</return_msg></xml>`
		returnBody := fmt.Sprintf(formatStr, returnCode, returnMsg)
		w.Write([]byte(returnBody))
	}()
	var reXML common.WeChatPayResult
	//body := cb.Ctx.Input.RequestBody
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		//log.Error(string(body))
		returnCode = "FAIL"
		returnMsg = "Bodyerror"
		return nil, err
	}
	err = xml.Unmarshal(body, &reXML)
	if err != nil {
		//log.Error(err, string(body))
		returnMsg = "参数错误"
		returnCode = "FAIL"
		return nil, err
	}

	if reXML.ReturnCode != "SUCCESS" {
		//log.Error(reXML)
		returnCode = "FAIL"
		return &reXML, errors.New(reXML.ReturnCode)
	}
	m, err := util.XmlToMap(body)
	if err != nil {
		//log.Error(err, body)
		returnMsg = "参数错误"
		returnCode = "FAIL"
		return nil, err
	}
	//log.Info(m)
	var signData []string
	for k, v := range m {
		if k == "sign" {
			continue
		}
		signData = append(signData, fmt.Sprintf("%v=%v", k, v))
	}
	sort.Strings(signData)
	signData2 := strings.Join(signData, "&")
	err = client.DefaultWechatAppClient().CheckSign(signData2, m["sign"])
	if err != nil {
		returnCode = "FAIL"
		return nil, err
	}

	returnCode = "SUCCESS"
	return &reXML, nil
}

//支付宝web端回调
func AliWebCallback(w http.ResponseWriter, r *http.Request) (*common.AliWebPayResult, error) {
	var m = make(map[string]string)
	var signSlice []string
	r.ParseForm()
	for k, v := range r.Form {
		// k不会有多个值的情况
		m[k] = v[0]
		if k == "sign" || k == "sign_type" {
			continue
		}
		signSlice = append(signSlice, fmt.Sprintf("%s=%s", k, v[0]))
	}

	sort.Strings(signSlice)
	signData := strings.Join(signSlice, "&")
	if m["sign_type"] != "RSA" {
		//错误日志
		return nil, errors.New("签名类型未知")
	}

	err := client.DefaultAliWebClient().CheckSign(signData, m["sign"])
	if err != nil {
		//log.Error("签名验证失败：", err, signData, m)
		return nil, err
	}

	var aliPay common.AliWebPayResult
	err = util.MapStringToStruct(m, &aliPay)
	if err != nil {
		//log.Error(err)
		w.Write([]byte("error"))
		return nil, err
	}

	// err = biz.AliWebCallBack(&aliPay)
	// if err != nil {
	// 	//log.Error(err)
	// 	w.Write([]byte("error"))
	// 	return nil, err
	// }
	w.Write([]byte("success"))
	return &aliPay, nil
}

// 支付宝app支付回调
func AliAppCallback(w http.ResponseWriter, r *http.Request) (*common.AliWebPayResult, error) {
	var m = make(map[string]string)
	var signSlice []string
	r.ParseForm()
	for k, v := range r.Form {
		m[k] = v[0]
		if k == "sign" || k == "sign_type" {
			continue
		}
		signSlice = append(signSlice, fmt.Sprintf("%s=%s", k, v[0]))
	}
	sort.Strings(signSlice)
	signData := strings.Join(signSlice, "&")
	if m["sign_type"] != "RSA" {
		//log.Error(m)
		return nil, errors.New("签名类型未知")
	}

	err := client.DefaultAliAppClient().CheckSign(signData, m["sign"])
	if err != nil {
		//log.Error(err, m, signData)
		w.Write([]byte("error"))
		return nil, err
	}

	var aliPay common.AliWebPayResult
	err = util.MapStringToStruct(m, &aliPay)
	if err != nil {
		//log.Error(err)
		w.Write([]byte("error"))
		return nil, err
	}

	// err = biz.AliAppCallBack(&aliPay)
	// if err != nil {
	// 	//log.Error(err)
	// 	w.Write([]byte("error"))
	// }

	w.Write([]byte("success"))
	return &aliPay, nil
}