package payment

// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
// * Use of this source code is governed by a Apache-2.0 license
// * that can be found in the LICENSE file.
// * @Author yangjian102621@163.com
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

import (
	"crypto/md5"
	"encoding/hex"
	"errors"
	"fmt"
	"geekai/core/types"
	"geekai/utils"
	"io"
	"net/http"
	"net/url"
	"sort"
	"strconv"
	"strings"
	"time"
)

type HuPiPayService struct {
	appId     string
	appSecret string
	apiURL    string
}

func NewHuPiPay(config *types.AppConfig) *HuPiPayService {
	return &HuPiPayService{
		appId:     config.HuPiPayConfig.AppId,
		appSecret: config.HuPiPayConfig.AppSecret,
		apiURL:    config.HuPiPayConfig.ApiURL,
	}
}

type HuPiPayParams struct {
	AppId        string `json:"appid"`
	Version      string `json:"version"`
	TradeOrderId string `json:"trade_order_id"`
	TotalFee     string `json:"total_fee"`
	Title        string `json:"title"`
	NotifyURL    string `json:"notify_url"`
	ReturnURL    string `json:"return_url"`
	WapName      string `json:"wap_name"`
	CallbackURL  string `json:"callback_url"`
	Time         string `json:"time"`
	NonceStr     string `json:"nonce_str"`
	Type         string `json:"type"`
	WapUrl       string `json:"wap_url"`
}

type HuPiPayResp struct {
	Openid    interface{} `json:"openid"`
	UrlQrcode string      `json:"url_qrcode"`
	URL       string      `json:"url"`
	ErrCode   int         `json:"errcode"`
	ErrMsg    string      `json:"errmsg,omitempty"`
}

// Pay 执行支付请求操作
func (s *HuPiPayService) Pay(params HuPiPayParams) (HuPiPayResp, error) {
	data := url.Values{}
	simple := strconv.FormatInt(time.Now().Unix(), 10)
	params.AppId = s.appId
	params.Time = simple
	params.NonceStr = simple
	encode := utils.JsonEncode(params)
	m := make(map[string]string)
	_ = utils.JsonDecode(encode, &m)
	for k, v := range m {
		data.Add(k, fmt.Sprintf("%v", v))
	}
	// 生成签名
	data.Add("hash", s.Sign(data))
	// 发送支付请求
	apiURL := fmt.Sprintf("%s/payment/do.html", s.apiURL)
	resp, err := http.PostForm(apiURL, data)
	if err != nil {
		return HuPiPayResp{}, fmt.Errorf("error with requst api: %v", err)
	}
	defer resp.Body.Close()
	all, err := io.ReadAll(resp.Body)
	if err != nil {
		return HuPiPayResp{}, fmt.Errorf("error with reading response: %v", err)
	}

	var res HuPiPayResp
	err = utils.JsonDecode(string(all), &res)
	if err != nil {
		return HuPiPayResp{}, fmt.Errorf("error with decode payment result: %v", err)
	}

	if res.ErrCode != 0 {
		return HuPiPayResp{}, fmt.Errorf("error with generate pay url: %s", res.ErrMsg)
	}

	return res, nil
}

// Sign 签名方法
func (s *HuPiPayService) Sign(params url.Values) string {
	params.Del(`Sign`)
	var keys = make([]string, 0, 0)
	for key := range params {
		if params.Get(key) != `` {
			keys = append(keys, key)
		}
	}
	sort.Strings(keys)

	var pList = make([]string, 0, 0)
	for _, key := range keys {
		var value = strings.TrimSpace(params.Get(key))
		if len(value) > 0 {
			pList = append(pList, key+"="+value)
		}
	}
	var src = strings.Join(pList, "&")
	src += s.appSecret

	md5bs := md5.Sum([]byte(src))
	return hex.EncodeToString(md5bs[:])
}

// Check 校验订单状态
func (s *HuPiPayService) Check(outTradeNo string) error {
	data := url.Values{}
	data.Add("appid", s.appId)
	data.Add("out_trade_order", outTradeNo)
	stamp := strconv.FormatInt(time.Now().Unix(), 10)
	data.Add("time", stamp)
	data.Add("nonce_str", stamp)
	data.Add("hash", s.Sign(data))

	apiURL := fmt.Sprintf("%s/payment/query.html", s.apiURL)
	resp, err := http.PostForm(apiURL, data)
	if err != nil {
		return fmt.Errorf("error with http reqeust: %v", err)
	}

	defer resp.Body.Close()
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return fmt.Errorf("error with reading response: %v", err)
	}

	var r struct {
		ErrCode int `json:"errcode"`
		Data    struct {
			Status      string `json:"status"`
			OpenOrderId string `json:"open_order_id"`
		} `json:"data,omitempty"`
		ErrMsg string `json:"errmsg"`
		Hash   string `json:"hash"`
	}
	err = utils.JsonDecode(string(body), &r)
	if err != nil {
		return fmt.Errorf("error with decode response: %v", err)
	}

	if r.ErrCode == 0 && r.Data.Status == "OD" {
		return nil
	} else {
		logger.Debugf("%+v", r)
		return errors.New("order not paid：" + r.ErrMsg)
	}
}