ai_sdk
/
xy_sh-20260727101126
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
xy_sh-20260727101126/xy_sh/generate.md

36 KiB
Raw Permalink Blame History

。

// File: xy_sh/go.mod

module xy_sh

go 1.26

require (
	github.com/gofiber/fiber/v2 v2.52.0
	github.com/tjfoc/gmsm v1.4.1
)

// File: xy_sh/cmd/server/main.go

package main

import (
	"log"

	"xy_sh/internal/config"
	"xy_sh/internal/router"
	"xy_sh/pkg/crypto"

	"github.com/gofiber/fiber/v2"
)

func main() {
	// 加载配置
	cfg := config.LoadConfig()

	// 初始化加密模块
	crypto.InitCrypto(cfg.SM3Salt, cfg.SM4Key)

	// 创建Fiber应用
	app := fiber.New(fiber.Config{
		AppName:      "XY SH API Server",
		ReadTimeout:  cfg.ReadTimeout,
		WriteTimeout: cfg.WriteTimeout,
	})

	// 设置路由
	router.SetupRoutes(app)

	// 启动服务
	log.Printf("服务启动，监听端口: %s", cfg.ServerPort)
	if err := app.Listen(":" + cfg.ServerPort); err != nil {
		log.Fatalf("服务启动失败: %v", err)
	}
}

// File: xy_sh/internal/config/config.go

package config

import (
	"os"
	"time"
)

// Config 应用配置
type Config struct {
	ServerPort   string
	SM3Salt      string
	SM4Key       string
	ReadTimeout  time.Duration
	WriteTimeout time.Duration
}

// LoadConfig 加载配置
func LoadConfig() *Config {
	port := getEnv("SERVER_PORT", "8080")
	sm3Salt := getEnv("SM3_SALT", "default_salt_change_me")
	sm4Key := getEnv("SM4_KEY", "default_key_16b")

	return &Config{
		ServerPort:   port,
		SM3Salt:      sm3Salt,
		SM4Key:       sm4Key,
		ReadTimeout:  30 * time.Second,
		WriteTimeout: 30 * time.Second,
	}
}

func getEnv(key, defaultValue string) string {
	if value := os.Getenv(key); value != "" {
		return value
	}
	return defaultValue
}

// File: xy_sh/pkg/crypto/crypto.go

package crypto

import (
	"bytes"
	"crypto/cipher"
	"crypto/hmac"
	"crypto/rand"
	"encoding/base64"
	"encoding/hex"
	"errors"
	"fmt"
	"io"

	"github.com/tjfoc/gmsm/sm3"
	"github.com/tjfoc/gmsm/sm4"
)

var (
	sm3Salt []byte
	sm4Key  []byte
)

// InitCrypto 初始化加密模块
func InitCrypto(salt, key string) {
	sm3Salt = []byte(salt)
	// SM4密钥必须为16字节
	keyBytes := []byte(key)
	if len(keyBytes) < 16 {
		// 补齐到16字节
		padded := make([]byte, 16)
		copy(padded, keyBytes)
		sm4Key = padded
	} else if len(keyBytes) > 16 {
		sm4Key = keyBytes[:16]
	} else {
		sm4Key = keyBytes
	}
}

// ==================== SM3 哈希算法 ====================

// SM3Hash 计算SM3哈希值
func SM3Hash(data []byte) string {
	h := sm3.New()
	h.Write(data)
	hashBytes := h.Sum(nil)
	return hex.EncodeToString(hashBytes)
}

// SM3HashWithSalt 使用salt计算SM3哈希值（用于签名）
func SM3HashWithSalt(data string) string {
	h := hmac.New(sm3.New, sm3Salt)
	h.Write([]byte(data))
	return hex.EncodeToString(h.Sum(nil))
}

// GenerateSign 生成签名
// 规则：timestamp + encryptedData 拼接后使用SM3 with salt加密
func GenerateSign(timestamp, encryptedData string) string {
	signStr := timestamp + encryptedData
	return SM3HashWithSalt(signStr)
}

// VerifySign 验证签名
func VerifySign(timestamp, encryptedData, sign string) bool {
	expectedSign := GenerateSign(timestamp, encryptedData)
	return hmac.Equal([]byte(expectedSign), []byte(sign))
}

// ==================== SM4-CBC 对称加密 ====================

// SM4CBCEncrypt SM4-CBC模式加密，返回base64编码
func SM4CBCEncrypt(plaintext []byte) (string, error) {
	if len(sm4Key) != 16 {
		return "", fmt.Errorf("SM4密钥长度必须为16字节")
	}

	block, err := sm4.NewCipher(sm4Key)
	if err != nil {
		return "", fmt.Errorf("创建SM4 cipher失败: %v", err)
	}

	padded := pkcs7Padding(plaintext, block.BlockSize())

	iv := make([]byte, block.BlockSize())
	if _, err := io.ReadFull(rand.Reader, iv); err != nil {
		return "", fmt.Errorf("生成IV失败: %v", err)
	}

	mode := cipher.NewCBCEncrypter(block, iv)
	ciphertext := make([]byte, len(padded))
	mode.CryptBlocks(ciphertext, padded)

	result := append(iv, ciphertext...)
	return base64.StdEncoding.EncodeToString(result), nil
}

// SM4CBCDecrypt SM4-CBC模式解密，输入base64编码
func SM4CBCDecrypt(encryptedData string) ([]byte, error) {
	if len(sm4Key) != 16 {
		return nil, fmt.Errorf("SM4密钥长度必须为16字节")
	}

	data, err := base64.StdEncoding.DecodeString(encryptedData)
	if err != nil {
		return nil, fmt.Errorf("base64解码失败: %v", err)
	}

	block, err := sm4.NewCipher(sm4Key)
	if err != nil {
		return nil, fmt.Errorf("创建SM4 cipher失败: %v", err)
	}

	blockSize := block.BlockSize()
	if len(data) < blockSize {
		return nil, errors.New("数据长度不足")
	}
	iv := data[:blockSize]
	ciphertext := data[blockSize:]

	if len(ciphertext)%blockSize != 0 {
		return nil, errors.New("密文长度不是块大小的整数倍")
	}

	mode := cipher.NewCBCDecrypter(block, iv)
	plaintext := make([]byte, len(ciphertext))
	mode.CryptBlocks(plaintext, ciphertext)

	plaintext, err = pkcs7UnPadding(plaintext)
	if err != nil {
		return nil, fmt.Errorf("去除填充失败: %v", err)
	}

	return plaintext, nil
}

// pkcs7Padding PKCS7填充
func pkcs7Padding(data []byte, blockSize int) []byte {
	padding := blockSize - len(data)%blockSize
	padText := bytes.Repeat([]byte{byte(padding)}, padding)
	return append(data, padText...)
}

// pkcs7UnPadding 去除PKCS7填充
func pkcs7UnPadding(data []byte) ([]byte, error) {
	length := len(data)
	if length == 0 {
		return nil, errors.New("数据为空")
	}
	padding := int(data[length-1])
	if padding > length || padding == 0 {
		return nil, errors.New("无效的填充")
	}
	for i := length - padding; i < length; i++ {
		if data[i] != byte(padding) {
			return nil, errors.New("无效的填充")
		}
	}
	return data[:length-padding], nil
}

// File: xy_sh/internal/entities/request.go

package entities

// EncryptedRequest 通用加密请求体
type EncryptedRequest struct {
	EncryptedData string `json:"encryptedData"`
}

// OrderRequest 下单接口业务参数（解密后）
type OrderRequest struct {
	ActCode     string `json:"actCode"`
	GoodsCode   string `json:"goodsCode"`
	ActOrderNum string `json:"actOrderNum"`
	Account     string `json:"account,omitempty"`
	CallbackUrl string `json:"callbackUrl,omitempty"`
}

// QueryOrderRequest 查询接口业务参数（解密后）
type QueryOrderRequest struct {
	ActCode string `json:"actCode"`
	OrderNo string `json:"orderNo"`
}

// WxRechargeRequest 微信立减金充值接口业务参数（解密后）
type WxRechargeRequest struct {
	ActCode     string `json:"actCode"`
	OrderNo     string `json:"orderNo"`
	GoodsCode   string `json:"goodsCode"`
	ActOrderNum string `json:"actOrderNum"`
	AppId       string `json:"appId"`
	OpenId      string `json:"openId"`
	CallbackUrl string `json:"callbackUrl,omitempty"`
}

// CallbackRequest 回调通知业务参数（解密后）
type CallbackRequest struct {
	OrderNo     string    `json:"orderNo"`
	ActOrderNum string    `json:"actOrderNum"`
	Status      int       `json:"status"`
	Account     string    `json:"account,omitempty"`
	CardInfo    *CardInfo `json:"cardInfo,omitempty"`
	CouponId    string    `json:"couponId,omitempty"`
}

// CardInfo 卡券信息
type CardInfo struct {
	CouponNo   string `json:"couponNo"`
	CouponCode string `json:"couponCode"`
	ExpireTime string `json:"expireTime"`
}

// File: xy_sh/internal/entities/response.go

package entities

// CommonResponse 通用响应结构
type CommonResponse struct {
	Code int         `json:"code"`
	Msg  string      `json:"msg"`
	Data interface{} `json:"data"`
}

// OrderResponseData 下单接口响应业务数据（加密前）
type OrderResponseData struct {
	OrderNo    string `json:"orderNo"`
	CouponNo   string `json:"couponNo,omitempty"`
	CouponCode string `json:"couponCode,omitempty"`
	Status     int    `json:"status"`
	ExpireTime string `json:"expireTime,omitempty"`
}

// QueryOrderResponseData 查询接口响应业务数据（加密前）
type QueryOrderResponseData struct {
	OrderNo  string    `json:"orderNo"`
	Status   int       `json:"status"`
	Account  string    `json:"account,omitempty"`
	CardInfo *CardInfo `json:"cardInfo,omitempty"`
	CouponId string    `json:"couponId,omitempty"`
}

// WxRechargeResponseData 微信立减金充值响应业务数据（加密前）
type WxRechargeResponseData struct {
	OrderNo  string `json:"orderNo"`
	Status   int    `json:"status"`
	CouponId string `json:"couponId,omitempty"`
}

// 订单状态常量
const (
	StatusSending  = -1 // 发放中
	StatusSuccess  = 0  // 成功
	StatusFailed   = 1  // 失败
	StatusRedeemed = 2  // 已核销
	StatusExpired  = 3  // 已过期
)

// 响应码常量
const (
	CodeSuccess = 0
	CodeFailed  = -1
)

// File: xy_sh/internal/middleware/middleware.go

package middleware

import (
	"encoding/json"
	"strings"

	"xy_sh/internal/entities"
	"xy_sh/pkg/crypto"

	"github.com/gofiber/fiber/v2"
)

// SignVerifyMiddleware 签名验证中间件
func SignVerifyMiddleware() fiber.Handler {
	return func(c *fiber.Ctx) error {
		// 从请求头获取timestamp和sign
		timestamp := c.Get("timestamp")
		sign := c.Get("sign")

		if timestamp == "" || sign == "" {
			return c.Status(fiber.StatusUnauthorized).JSON(entities.CommonResponse{
				Code: entities.CodeFailed,
				Msg:  "缺少timestamp或sign请求头",
				Data: nil,
			})
		}

		// 读取请求体
		body := c.Body()

		// 解析encryptedData
		var req entities.EncryptedRequest
		if err := json.Unmarshal(body, &req); err != nil {
			return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
				Code: entities.CodeFailed,
				Msg:  "请求体格式错误",
				Data: nil,
			})
		}

		if req.EncryptedData == "" {
			return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
				Code: entities.CodeFailed,
				Msg:  "缺少encryptedData参数",
				Data: nil,
			})
		}

		// 验证签名
		if !crypto.VerifySign(timestamp, req.EncryptedData, sign) {
			return c.Status(fiber.StatusUnauthorized).JSON(entities.CommonResponse{
				Code: entities.CodeFailed,
				Msg:  "签名验证失败",
				Data: nil,
			})
		}

		// 解密数据
		decryptedData, err := crypto.SM4CBCDecrypt(req.EncryptedData)
		if err != nil {
			return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
				Code: entities.CodeFailed,
				Msg:  "数据解密失败: " + err.Error(),
				Data: nil,
			})
		}

		// 将解密后的数据存入上下文
		c.Locals("decryptedData", decryptedData)
		c.Locals("timestamp", timestamp)

		// 继续处理请求
		return c.Next()
	}
}

// EncryptResponseMiddleware 加密响应中间件
func EncryptResponseMiddleware() fiber.Handler {
	return func(c *fiber.Ctx) error {
		// 先执行后续处理
		err := c.Next()
		if err != nil {
			return err
		}

		// 获取响应体
		body := c.Response().Body()
		if len(body) == 0 {
			return nil
		}

		// 尝试解析为CommonResponse
		var resp entities.CommonResponse
		if err := json.Unmarshal(body, &resp); err != nil {
			// 不是标准响应格式，直接返回
			return nil
		}

		// 如果data不为空且是字符串，需要加密
		if resp.Data != nil {
			// 将data序列化为JSON字符串
			dataBytes, err := json.Marshal(resp.Data)
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(entities.CommonResponse{
					Code: entities.CodeFailed,
					Msg:  "响应数据序列化失败",
					Data: nil,
				})
			}

			// 加密data
			encryptedData, err := crypto.SM4CBCEncrypt(dataBytes)
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(entities.CommonResponse{
					Code: entities.CodeFailed,
					Msg:  "响应数据加密失败",
					Data: nil,
				})
			}

			// 替换data为加密后的字符串
			resp.Data = encryptedData

			// 重新设置响应体
			newBody, err := json.Marshal(resp)
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(entities.CommonResponse{
					Code: entities.CodeFailed,
					Msg:  "响应序列化失败",
					Data: nil,
				})
			}

			c.Response().SetBody(newBody)
		}

		return nil
	}
}

// SkipSignVerify 跳过签名验证的路径判断
func SkipSignVerify(path string) bool {
	// 可以配置不需要签名验证的路径
	skipPaths := []string{"/health", "/"}
	for _, p := range skipPaths {
		if strings.HasPrefix(path, p) {
			return true
		}
	}
	return false
}

// File: xy_sh/internal/biz/biz.go

package biz

import (
	"errors"
	"sync"
	"time"

	"xy_sh/internal/entities"
)

// OrderStore 订单存储（内存实现，生产环境应使用数据库）
type OrderStore struct {
	mu     sync.RWMutex
	orders map[string]*OrderInfo
}

// OrderInfo 订单信息
type OrderInfo struct {
	OrderNo     string
	ActOrderNum string
	GoodsCode   string
	ActCode     string
	Account     string
	Status      int
	CouponNo    string
	CouponCode  string
	ExpireTime  string
	CouponId    string
	AppId       string
	OpenId      string
	CreateTime  time.Time
	UpdateTime  time.Time
}

var store *OrderStore

func init() {
	store = &OrderStore{
		orders: make(map[string]*OrderInfo),
	}
}

// CreateOrder 创建订单
func CreateOrder(req *entities.OrderRequest) (*entities.OrderResponseData, error) {
	store.mu.Lock()
	defer store.mu.Unlock()

	// 生成订单号
	orderNo := generateOrderNo()

	// TODO: 实际业务逻辑处理
	// 1. 验证活动code和商品code
	// 2. 检查库存
	// 3. 处理卡券/直充逻辑
	// 4. 生成卡密信息（卡券类商品）

	orderInfo := &OrderInfo{
		OrderNo:     orderNo,
		ActOrderNum: req.ActOrderNum,
		GoodsCode:   req.GoodsCode,
		ActCode:     req.ActCode,
		Account:     req.Account,
		Status:      entities.StatusSuccess, // 默认成功，实际应根据业务处理结果设置
		CreateTime:  time.Now(),
		UpdateTime:  time.Now(),
	}

	// 模拟卡券类商品返回卡密信息
	orderInfo.CouponNo = generateCouponNo()
	orderInfo.CouponCode = orderInfo.CouponNo
	orderInfo.ExpireTime = time.Now().AddDate(1, 0, 0).Format("2006-01-02 15:04:05")

	store.orders[orderNo] = orderInfo

	return &entities.OrderResponseData{
		OrderNo:    orderNo,
		CouponNo:   orderInfo.CouponNo,
		CouponCode: orderInfo.CouponCode,
		Status:     orderInfo.Status,
		ExpireTime: orderInfo.ExpireTime,
	}, nil
}

// QueryOrder 查询订单
func QueryOrder(req *entities.QueryOrderRequest) (*entities.QueryOrderResponseData, error) {
	store.mu.RLock()
	defer store.mu.RUnlock()

	orderInfo, exists := store.orders[req.OrderNo]
	if !exists {
		return nil, errors.New("订单不存在")
	}

	resp := &entities.QueryOrderResponseData{
		OrderNo:  orderInfo.OrderNo,
		Status:   orderInfo.Status,
		Account:  orderInfo.Account,
		CouponId: orderInfo.CouponId,
	}

	// 如果有卡券信息
	if orderInfo.CouponNo != "" {
		resp.CardInfo = &entities.CardInfo{
			CouponNo:   orderInfo.CouponNo,
			CouponCode: orderInfo.CouponCode,
			ExpireTime: orderInfo.ExpireTime,
		}
	}

	return resp, nil
}

// WxRecharge 微信立减金充值
func WxRecharge(req *entities.WxRechargeRequest) (*entities.WxRechargeResponseData, error) {
	store.mu.Lock()
	defer store.mu.Unlock()

	// TODO: 实际业务逻辑处理
	// 1. 验证活动code和商品code
	// 2. 调用微信立减金发放接口
	// 3. 处理充值结果

	// 检查订单是否已存在
	if orderInfo, exists := store.orders[req.OrderNo]; exists {
		return &entities.WxRechargeResponseData{
			OrderNo:  orderInfo.OrderNo,
			Status:   orderInfo.Status,
			CouponId: orderInfo.CouponId,
		}, nil
	}

	// 生成订单号
	orderNo := req.OrderNo
	if orderNo == "" {
		orderNo = generateOrderNo()
	}

	orderInfo := &OrderInfo{
		OrderNo:     orderNo,
		ActOrderNum: req.ActOrderNum,
		GoodsCode:   req.GoodsCode,
		ActCode:     req.ActCode,
		AppId:       req.AppId,
		OpenId:      req.OpenId,
		Status:      entities.StatusSuccess,
		CouponId:    generateCouponId(),
		CreateTime:  time.Now(),
		UpdateTime:  time.Now(),
	}

	store.orders[orderNo] = orderInfo

	return &entities.WxRechargeResponseData{
		OrderNo:  orderNo,
		Status:   orderInfo.Status,
		CouponId: orderInfo.CouponId,
	}, nil
}

// HandleCallback 处理回调通知
func HandleCallback(req *entities.CallbackRequest) error {
	store.mu.Lock()
	defer store.mu.Unlock()

	// TODO: 实际业务逻辑处理
	// 1. 更新订单状态
	// 2. 处理卡券信息
	// 3. 触发后续业务流程

	orderInfo, exists := store.orders[req.OrderNo]
	if !exists {
		// 如果订单不存在，创建记录
		orderInfo = &OrderInfo{
			OrderNo:     req.OrderNo,
			ActOrderNum: req.ActOrderNum,
			CreateTime:  time.Now(),
		}
		store.orders[req.OrderNo] = orderInfo
	}

	// 更新订单状态
	orderInfo.Status = req.Status
	orderInfo.Account = req.Account
	orderInfo.CouponId = req.CouponId
	orderInfo.UpdateTime = time.Now()

	if req.CardInfo != nil {
		orderInfo.CouponNo = req.CardInfo.CouponNo
		orderInfo.CouponCode = req.CardInfo.CouponCode
		orderInfo.ExpireTime = req.CardInfo.ExpireTime
	}

	return nil
}

// generateOrderNo 生成订单号
func generateOrderNo() string {
	return "HM" + time.Now().Format("20060102150405") + randomString(16)
}

// generateCouponNo 生成卡号
func generateCouponNo() string {
	return randomString(16)
}

// generateCouponId 生成微信优惠id
func generateCouponId() string {
	return randomString(8)
}

// randomString 生成随机字符串
func randomString(n int) string {
	const letters = "0123456789abcdef"
	b := make([]byte, n)
	for i := range b {
		b[i] = letters[time.Now().UnixNano()%int64(len(letters))]
		time.Sleep(1 * time.Nanosecond)
	}
	return string(b)
}

// File: xy_sh/internal/service/service.go

package service

import (
	"encoding/json"

	"xy_sh/internal/biz"
	"xy_sh/internal/entities"

	"github.com/gofiber/fiber/v2"
)

// CreateOrderHandler 卡券/直充权益下单接口
func CreateOrderHandler(c *fiber.Ctx) error {
	// 获取解密后的数据
	decryptedData := c.Locals("decryptedData").([]byte)

	// 解析业务参数
	var req entities.OrderRequest
	if err := json.Unmarshal(decryptedData, &req); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  "业务参数解析失败: " + err.Error(),
			Data: nil,
		})
	}

	// 参数校验
	if req.ActCode == "" || req.GoodsCode == "" || req.ActOrderNum == "" {
		return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  "缺少必填参数: actCode, goodsCode, actOrderNum",
			Data: nil,
		})
	}

	// 调用业务逻辑
	result, err := biz.CreateOrder(&req)
	if err != nil {
		return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  err.Error(),
			Data: nil,
		})
	}

	return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{
		Code: entities.CodeSuccess,
		Msg:  "请求成功",
		Data: result,
	})
}

// QueryOrderHandler 卡券/直充/微信立减金订单查询接口
func QueryOrderHandler(c *fiber.Ctx) error {
	// 获取解密后的数据
	decryptedData := c.Locals("decryptedData").([]byte)

	// 解析业务参数
	var req entities.QueryOrderRequest
	if err := json.Unmarshal(decryptedData, &req); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  "业务参数解析失败: " + err.Error(),
			Data: nil,
		})
	}

	// 参数校验
	if req.ActCode == "" || req.OrderNo == "" {
		return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  "缺少必填参数: actCode, orderNo",
			Data: nil,
		})
	}

	// 调用业务逻辑
	result, err := biz.QueryOrder(&req)
	if err != nil {
		return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  err.Error(),
			Data: nil,
		})
	}

	return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{
		Code: entities.CodeSuccess,
		Msg:  "请求成功",
		Data: result,
	})
}

// WxRechargeHandler 微信立减金订单充值接口
func WxRechargeHandler(c *fiber.Ctx) error {
	// 获取解密后的数据
	decryptedData := c.Locals("decryptedData").([]byte)

	// 解析业务参数
	var req entities.WxRechargeRequest
	if err := json.Unmarshal(decryptedData, &req); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  "业务参数解析失败: " + err.Error(),
			Data: nil,
		})
	}

	// 参数校验
	if req.ActCode == "" || req.GoodsCode == "" || req.ActOrderNum == "" ||
		req.AppId == "" || req.OpenId == "" {
		return c.Status(fiber.StatusBadRequest).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  "缺少必填参数: actCode, goodsCode, actOrderNum, appId, openId",
			Data: nil,
		})
	}

	// 调用业务逻辑
	result, err := biz.WxRecharge(&req)
	if err != nil {
		return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{
			Code: entities.CodeFailed,
			Msg:  err.Error(),
			Data: nil,
		})
	}

	return c.Status(fiber.StatusOK).JSON(entities.CommonResponse{
		Code: entities.CodeSuccess,
		Msg:  "请求成功",
		Data: result,
	})
}

// CallbackHandler 卡券/直充/微信立减金充值结果通知接口
func CallbackHandler(c *fiber.Ctx) error {
	// 获取解密后的数据
	decryptedData := c.Locals("decryptedData").([]byte)

	// 解析业务参数
	var req entities.CallbackRequest
	if err := json.Unmarshal(decryptedData, &req); err != nil {
		return c.Status(fiber.StatusBadRequest).SendString("参数解析失败")
	}

	// 参数校验
	if req.OrderNo == "" || req.ActOrderNum == "" {
		return c.Status(fiber.StatusBadRequest).SendString("缺少必填参数")
	}

	// 调用业务逻辑
	if err := biz.HandleCallback(&req); err != nil {
		return c.Status(fiber.StatusInternalServerError).SendString("处理失败")
	}

	// 返回ok表示成功接收
	return c.Status(fiber.StatusOK).SendString("ok")
}

// HealthHandler 健康检查接口
func HealthHandler(c *fiber.Ctx) error {
	return c.Status(fiber.StatusOK).JSON(fiber.Map{
		"status":  "ok",
		"message": "服务运行正常",
	})
}

// File: xy_sh/internal/router/router.go

package router

import (
	"xy_sh/internal/middleware"
	"xy_sh/internal/service"

	"github.com/gofiber/fiber/v2"
)

// SetupRoutes 设置路由
func SetupRoutes(app *fiber.App) {
	// 健康检查（不需要签名验证）
	app.Get("/health", service.HealthHandler)
	app.Get("/", service.HealthHandler)

	// API v1 分组
	v1 := app.Group("/api/v1")

	// 需要签名验证和加密响应的接口
	v1.Post("/order/create",
		middleware.SignVerifyMiddleware(),
		middleware.EncryptResponseMiddleware(),
		service.CreateOrderHandler,
	)

	v1.Post("/order/query",
		middleware.SignVerifyMiddleware(),
		middleware.EncryptResponseMiddleware(),
		service.QueryOrderHandler,
	)

	v1.Post("/wx/recharge",
		middleware.SignVerifyMiddleware(),
		middleware.EncryptResponseMiddleware(),
		service.WxRechargeHandler,
	)

	// 回调通知接口（需要签名验证，但响应不加密，直接返回ok）
	v1.Post("/callback/notify",
		middleware.SignVerifyMiddleware(),
		service.CallbackHandler,
	)
}

// File: xy_sh/internal/test/example_test.go

package test

import (
	"bytes"
	"encoding/json"
	"io"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"

	"xy_sh/internal/entities"
	"xy_sh/pkg/crypto"

	"github.com/gofiber/fiber/v2"
	"xy_sh/internal/router"
)

func init() {
	// 初始化加密模块（测试用）
	crypto.InitCrypto("test_salt_12345", "test_key_16bytes")
}

// setupTestApp 创建测试应用
func setupTestApp() *fiber.App {
	app := fiber.New()
	router.SetupRoutes(app)
	return app
}

// makeEncryptedRequest 构造加密请求
func makeEncryptedRequest(t *testing.T, app *fiber.App, path string, bizData interface{}) (*http.Response, []byte) {
	t.Helper()

	// 将业务数据序列化为JSON
	bizJSON, err := json.Marshal(bizData)
	if err != nil {
		t.Fatalf("业务数据序列化失败: %v", err)
	}

	// SM4加密业务数据
	encryptedData, err := crypto.SM4CBCEncrypt(bizJSON)
	if err != nil {
		t.Fatalf("加密失败: %v", err)
	}

	// 生成时间戳
	timestamp := time.Now().Format("20060102150405") + "000"

	// 生成签名
	sign := crypto.GenerateSign(timestamp, encryptedData)

	// 构造请求体
	reqBody := entities.EncryptedRequest{
		EncryptedData: encryptedData,
	}
	reqJSON, _ := json.Marshal(reqBody)

	// 创建HTTP请求
	req := httptest.NewRequest(http.MethodPost, path, bytes.NewReader(reqJSON))
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("timestamp", timestamp)
	req.Header.Set("sign", sign)

	// 发送请求
	resp, err := app.Test(req)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}

	// 读取响应
	body, _ := io.ReadAll(resp.Body)
	resp.Body.Close()

	return resp, body
}

// TestCreateOrder 测试下单接口
func TestCreateOrder(t *testing.T) {
	app := setupTestApp()

	// 构造下单请求参数
	bizData := entities.OrderRequest{
		ActCode:     "ACT001",
		GoodsCode:   "123456",
		ActOrderNum: "TEST_ORDER_001",
		Account:     "19912345678",
		CallbackUrl: "https://example.com/callback",
	}

	resp, body := makeEncryptedRequest(t, app, "/api/v1/order/create", bizData)

	if resp.StatusCode != http.StatusOK {
		t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body))
	}

	// 解析响应
	var commonResp entities.CommonResponse
	if err := json.Unmarshal(body, &commonResp); err != nil {
		t.Fatalf("响应解析失败: %v, 响应: %s", err, string(body))
	}

	if commonResp.Code != entities.CodeSuccess {
		t.Errorf("期望code=0，实际: %d, msg: %s", commonResp.Code, commonResp.Msg)
	}

	t.Logf("下单接口测试成功，响应: %s", string(body))
}

// TestQueryOrder 测试查询接口
func TestQueryOrder(t *testing.T) {
	app := setupTestApp()

	// 先创建一个订单
	createBizData := entities.OrderRequest{
		ActCode:     "ACT001",
		GoodsCode:   "123456",
		ActOrderNum: "TEST_ORDER_002",
		Account:     "19912345678",
	}

	_, createBody := makeEncryptedRequest(t, app, "/api/v1/order/create", createBizData)

	var createResp entities.CommonResponse
	json.Unmarshal(createBody, &createResp)

	// 注意：实际响应中data是加密的，这里需要先解密
	// 为了简化测试，我们直接构造查询请求
	queryBizData := entities.QueryOrderRequest{
		ActCode: "ACT001",
		OrderNo: "HM17575805323790000127073398f3772c00", // 使用示例订单号
	}

	resp, body := makeEncryptedRequest(t, app, "/api/v1/order/query", queryBizData)

	if resp.StatusCode != http.StatusOK {
		t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body))
	}

	t.Logf("查询接口测试完成，响应: %s", string(body))
}

// TestWxRecharge 测试微信立减金充值接口
func TestWxRecharge(t *testing.T) {
	app := setupTestApp()

	bizData := entities.WxRechargeRequest{
		ActCode:     "FBG6vdYqGE4mGX7EH/woEg==",
		GoodsCode:   "0001",
		ActOrderNum: "WX_ORDER_001",
		AppId:       "wx1234567890",
		OpenId:      "oABC1234567890",
	}

	resp, body := makeEncryptedRequest(t, app, "/api/v1/wx/recharge", bizData)

	if resp.StatusCode != http.StatusOK {
		t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body))
	}

	var commonResp entities.CommonResponse
	if err := json.Unmarshal(body, &commonResp); err != nil {
		t.Fatalf("响应解析失败: %v", err)
	}

	if commonResp.Code != entities.CodeSuccess {
		t.Errorf("期望code=0，实际: %d, msg: %s", commonResp.Code, commonResp.Msg)
	}

	t.Logf("微信立减金充值接口测试成功，响应: %s", string(body))
}

// TestCallback 测试回调通知接口
func TestCallback(t *testing.T) {
	app := setupTestApp()

	bizData := entities.CallbackRequest{
		OrderNo:     "HM202509291010001",
		ActOrderNum: "XY2025092910100001",
		Status:      entities.StatusExpired,
		Account:     "19912345678",
	}

	resp, body := makeEncryptedRequest(t, app, "/api/v1/callback/notify", bizData)

	if resp.StatusCode != http.StatusOK {
		t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body))
	}

	if string(body) != "ok" {
		t.Errorf("期望响应'ok'，实际: %s", string(body))
	}

	t.Logf("回调通知接口测试成功")
}

// TestInvalidSign 测试签名验证失败
func TestInvalidSign(t *testing.T) {
	app := setupTestApp()

	bizData := entities.OrderRequest{
		ActCode:     "ACT001",
		GoodsCode:   "123456",
		ActOrderNum: "TEST_ORDER_003",
	}

	// 加密数据
	bizJSON, _ := json.Marshal(bizData)
	encryptedData, _ := crypto.SM4CBCEncrypt(bizJSON)
	timestamp := time.Now().Format("20060102150405") + "000"

	// 使用错误的签名
	wrongSign := "wrong_signature_12345"

	reqBody := entities.EncryptedRequest{EncryptedData: encryptedData}
	reqJSON, _ := json.Marshal(reqBody)

	req := httptest.NewRequest(http.MethodPost, "/api/v1/order/create", bytes.NewReader(reqJSON))
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("timestamp", timestamp)
	req.Header.Set("sign", wrongSign)

	resp, _ := app.Test(req)

	if resp.StatusCode != http.StatusUnauthorized {
		t.Errorf("期望状态码401（签名验证失败），实际: %d", resp.StatusCode)
	}

	t.Logf("签名验证失败测试成功")
}

// TestHealthCheck 测试健康检查接口
func TestHealthCheck(t *testing.T) {
	app := setupTestApp()

	req := httptest.NewRequest(http.MethodGet, "/health", nil)
	resp, body := app.Test(req)

	if resp.StatusCode != http.StatusOK {
		t.Errorf("期望状态码200，实际: %d", resp.StatusCode)
	}

	t.Logf("健康检查接口测试成功，响应: %s", string(body))
}

=== 服务端生成完成 ===

加密实现

以下是从加密工具获取的完整实现：

加密实现 1

{ "success": true, "data": "\n### SM3 国密哈希算法完整实现指南\n\n适用场景：文档要求使用国密 SM3 算法进行摘要计算或签名验证\n\n前置要求：需要安装 github.com/tjfoc/gmsm\n\nbash\ngo get github.com/tjfoc/gmsm\n\n\n配置参数：\n- 编码方式: hex\n- 包含 HMAC: true\n\n完整代码模板：\n\ngo\npackage crypto\n\nimport (\n \"crypto/hmac\"\n \"encoding/base64\"\n \"encoding/hex\"\n \"fmt\"\n \"os\"\n \n \"github.com/tjfoc/gmsm/sm3\"\n)\n\n// SM3Hash 计算SM3哈希值\n// data: 待哈希的数据\n// encoding: 输出编码方式 (hex/base64)\n// 返回: 编码后的哈希值\nfunc SM3Hash(data []byte, encoding string) (string, error) {\n h := sm3.New()\n h.Write(data)\n hashBytes := h.Sum(nil)\n \n if encoding == \"base64\" {\n return base64.StdEncoding.EncodeToString(hashBytes), nil\n }\n return hex.EncodeToString(hashBytes), nil\n}\n\n// SM3HashString 计算字符串的SM3哈希值（便捷方法）\nfunc SM3HashString(data string, encoding string) (string, error) {\n return SM3Hash([]byte(data), encoding)\n}\n\n// SM3HashFile 计算文件的SM3哈希值\nfunc SM3HashFile(filePath string, encoding string) (string, error) {\n data, err := os.ReadFile(filePath)\n if err != nil {\n return \"\", fmt.Errorf(\"读取文件失败: %v\", err)\n }\n return SM3Hash(data, encoding)\n}\n\n// SM3Verify 验证数据与哈希值是否匹配\nfunc SM3Verify(data []byte, hash string, encoding string) (bool, error) {\n expected, err := SM3Hash(data, encoding)\n if err != nil {\n return false, err\n }\n return expected == hash, nil\n}\n\n// HMACSM3 HMAC-SM3计算\n// 使用标准HMAC算法，底层使用SM3哈希函数\nfunc HMACSM3(data []byte, key []byte) string {\n // 使用标准HMAC，底层哈希函数用SM3\n h := hmac.New(sm3.New, key)\n h.Write(data)\n return hex.EncodeToString(h.Sum(nil))\n}\n\n\n使用示例：\ngo\n// 计算字符串哈希\nhash, _ := SM3HashString(\"hello world\", \"hex\")\nfmt.Println(hash) // 输出64位十六进制字符串\n\n// 验证哈希\nvalid, _ := SM3Verify([]byte(\"hello world\"), hash, \"hex\")\nfmt.Println(valid) // true\n\n// HMAC-SM3示例\nkey := []byte(\"secret_key\")\ndata := []byte(\"hello world\")\nhmacResult := HMACSM3(data, key)\nfmt.Println(hmacResult)\n\n\n注意事项：\n- SM3 输出固定 256 位（32字节）的哈希值\n- 十六进制输出为 64 位字符串\n- 常用于数字签名、完整性校验等场景\n- SM3 是国密标准哈希算法，与 SHA-256 类似\n- HMAC-SM3 需要使用标准库 crypto/hmac，底层哈希函数使用 sm3.New\n", "tool": "sm3_hash" }

加密实现 2

{ "success": true, "data": "\n### SM4-CBC 国密对称加密完整实现指南\n\n⚠️ 重要：必须使用第三方库，不要自己实现 SM4 算法！\n推荐使用：github.com/tjfoc/gmsm\n\n前置要求：\nbash\ngo get github.com/tjfoc/gmsm\n\n\n配置参数：\n- 编码方式: base64\n\n完整代码模板（请直接复制使用）：\n\ngo\npackage crypto\n\nimport (\n \"crypto/cipher\"\n \"crypto/rand\"\n \"encoding/base64\"\n \"encoding/hex\"\n \"fmt\"\n \"io\"\n \"bytes\"\n\n \"github.com/tjfoc/gmsm/sm4\" // ⚠️ 必须使用此库，不要自己实现SM4\n)\n\n// SM4CBCEncrypt SM4-CBC模式加密\n// 使用 github.com/tjfoc/gmsm/sm4 实现\nfunc SM4CBCEncrypt(plaintext []byte, key []byte) (string, error) {\n if len(key) != 16 {\n return \"\", fmt.Errorf(\"SM4密钥长度必须为16字节\")\n }\n\n // 使用 gmsm 库创建 SM4 cipher\n block, err := sm4.NewCipher(key)\n if err != nil {\n return \"\", fmt.Errorf(\"创建SM4 cipher失败: %v\", err)\n }\n\n padded := pkcs7Padding(plaintext, block.BlockSize())\n\n iv := make([]byte, block.BlockSize())\n if _, err := io.ReadFull(rand.Reader, iv); err != nil {\n return \"\", fmt.Errorf(\"生成IV失败: %v\", err)\n }\n\n mode := cipher.NewCBCEncrypter(block, iv)\n ciphertext := make([]byte, len(padded))\n mode.CryptBlocks(ciphertext, padded)\n\n result := append(iv, ciphertext...)\n if \"base64\" == \"hex\" {\n return hex.EncodeToString(result), nil\n }\n return base64.StdEncoding.EncodeToString(result), nil\n}\n\n// SM4CBCDecrypt SM4-CBC模式解密\n// 使用 github.com/tjfoc/gmsm/sm4 实现\nfunc SM4CBCDecrypt(encryptedData string, key []byte) ([]byte, error) {\n if len(key) != 16 {\n return nil, fmt.Errorf(\"SM4密钥长度必须为16字节\")\n }\n\n var data []byte\n var err error\n if \"base64\" == \"hex\" {\n data, err = hex.DecodeString(encryptedData)\n } else {\n data, err = base64.StdEncoding.DecodeString(encryptedData)\n }\n if err != nil {\n return nil, fmt.Errorf(\"解码失败: %v\", err)\n }\n\n block, err := sm4.NewCipher(key)\n if err != nil {\n return nil, fmt.Errorf(\"创建SM4 cipher失败: %v\", err)\n }\n\n blockSize := block.BlockSize()\n if len(data) \u003c blockSize {\n return nil, fmt.Errorf(\"数据长度不足\")\n }\n iv := data[:blockSize]\n ciphertext := data[blockSize:]\n\n mode := cipher.NewCBCDecrypter(block, iv)\n plaintext := make([]byte, len(ciphertext))\n mode.CryptBlocks(plaintext, ciphertext)\n\n plaintext, err = pkcs7UnPadding(plaintext)\n if err != nil {\n return nil, fmt.Errorf(\"去除填充失败: %v\", err)\n }\n\n return plaintext, nil\n}\n\n// pkcs7Padding PKCS7填充\nfunc pkcs7Padding(data []byte, blockSize int) []byte {\n padding := blockSize - len(data)%blockSize\n padText := bytes.Repeat([]byte{byte(padding)}, padding)\n return append(data, padText...)\n}\n\n// pkcs7UnPadding 去除PKCS7填充\nfunc pkcs7UnPadding(data []byte) ([]byte, error) {\n length := len(data)\n if length == 0 {\n return nil, fmt.Errorf(\"数据为空\")\n }\n padding := int(data[length-1])\n if padding \u003e length || padding == 0 {\n return nil, fmt.Errorf(\"无效的填充\")\n }\n for i := length - padding; i \u003c length; i++ {\n if data[i] != byte(padding) {\n return nil, fmt.Errorf(\"无效的填充\")\n }\n }\n return data[:length-padding], nil\n}\n\n\n⚠️ 重要提醒：\n1. 不要自己实现 SM4 算法，请直接使用 github.com/tjfoc/gmsm\n2. 这个库是国密官方推荐的 Go 实现，安全可靠\n3. 在 go.mod 中添加依赖：require github.com/tjfoc/gmsm v1.4.1\n\n注意事项：\n- SM4 密钥固定为 16 字节\n- IV 必须随机生成且每次不同\n- CBC 模式需要 IV，ECB 模式不需要\n", "tool": "sm4_cbc_encrypt" }