package payment

import (
	"encoding/json"
	"fmt"
	"time"
)

// OrderQuery 查询订单支付状态
// 对齐 YouChuKoffee 的 OrderQuery 流程：
// 1. 构建完整 {head, body} 结构
// 2. 整个 {head, body} 一起加密（EncryptRequest）
// 3. head 同时作为 HTTP Header 明文传输
func (c *Client) OrderQuery(orderNo string) (*OrderQueryResponse, error) {
	now := time.Now().Format("20060102150405")
	BusiMainId := now + RandomNumber(10)

	// 构建完整 request（head+body 一起加密，对齐 YouChuKoffee）
	requestBody := map[string]interface{}{
		"head": map[string]string{
			"partnerTxSriNo": BusiMainId,
			"reqTime":        now,
			"method":         "b2c.gatewaypay.orderQuery",
			"version":        "1",
			"merchantId":     c.cfg.MerchantId,
			"appID":          c.cfg.AppID,
			"accessType":     "API",
			"reserve":        "",
		},
		"body": map[string]interface{}{
			"busiMainId":   BusiMainId,
			"reqTransTime": now,
			"data": map[string]string{
				"txnCode":    "1004",
				"sourceId":   "16",
				"reqTraceId": now + RandomNumber(10),
				"reqDate":    now, // 14 位 YYYYMMDDHHmmss
				"txnDt":      now,
				"mchtNo":     c.cfg.MchtNo,
				"oldSeqNo":   orderNo,
			},
		},
	}

	// 整个 {head, body} 一起加密
	encryptedReq, err := EncryptRequest(requestBody, c.cfg)
	if err != nil {
		return nil, fmt.Errorf("加密请求失败: %v", err)
	}

	encryptedBytes, _ := json.Marshal(encryptedReq)
	url := c.cfg.OrderHost + c.cfg.MerchantId + ".htm?partnerTxSriNo=" + BusiMainId

	// head 同时作为 HTTP Header 明文传输
	head := requestBody["head"].(map[string]string)
	body, err := c.doPost(url, head, encryptedBytes)
	if err != nil {
		return nil, err
	}

	responseData, err := c.DecryptResponse(string(body), false)
	if err != nil {
		return nil, fmt.Errorf("解密响应失败: %v，原始响应: %s", err, string(body))
	}

	var result OrderQueryResponse
	if err := json.Unmarshal([]byte(responseData), &result); err != nil {
		return nil, fmt.Errorf("解析响应失败: %v，响应内容: %s", err, responseData)
	}

	if result.RespCode != "" && result.RespCode != "0000" && result.RespCode != "00" {
		return nil, fmt.Errorf("订单查询失败，错误码: %s，错误信息: %s", result.RespCode, result.RespMsg)
	}

	return &result, nil
}