app-api/internal/handler/share_code.go

package handler

import (
	"encoding/json"
	"net/http"
	"strings"

	"github.com/gin-gonic/gin"
	"github.com/luxsin/app-api/internal/cache"
	"github.com/luxsin/app-api/pkg/encode"
	"go.uber.org/zap"
)

type ShareCodeHandler struct {
	cache *cache.ShareCodeCache
	log   *zap.Logger
}

func NewShareCodeHandler(shareCache *cache.ShareCodeCache, log *zap.Logger) *ShareCodeHandler {
	return &ShareCodeHandler{cache: shareCache, log: log}
}

// ExportShareCode 导出分享码
//
//	@Summary	创建 EQ 分享码
//	@Description	将用户的 EQ 数据生成一个 5 位分享码，有效期 30 分钟
//	@Tags		ShareCode
//	@Accept		json
//	@Produce	json
//	@Param		body	body		object{mac=string,eq_data=object}	true	"分享请求"
//	@Success	200		{object}	map[string]any	"成功返回 share_code、expire_at、eq_data"
//	@Failure	400		{object}	map[string]any	"参数校验失败"
//	@Failure	500		{object}	map[string]any	"系统错误"
//	@Router		/audio/shareCreate [post]
//
// POST /audio/shareCreate
// Body: {"mac": "xx", "eq_data": {...}}
func (h *ShareCodeHandler) ExportShareCode(c *gin.Context) {
	var req struct {
		Mac    string          `json:"mac"`
		EqData json.RawMessage `json:"eq_data"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusOK, gin.H{
			"code": 400,
			"msg":  "参数校验失败",
		})
		return
	}

	mac := strings.TrimSpace(req.Mac)
	eqDataRaw := strings.TrimSpace(string(req.EqData))
	clientIP := encode.ClientPublicIP(c)

	if mac == "" || eqDataRaw == "" {
		c.JSON(http.StatusOK, gin.H{
			"code": 400,
			"msg":  "参数校验失败",
		})
		return
	}

	if !json.Valid([]byte(eqDataRaw)) {
		c.JSON(http.StatusOK, gin.H{
			"code": 400,
			"msg":  "eq_data 格式错误",
		})
		return
	}

	ctx := c.Request.Context()
	data, err := h.cache.Create(ctx, mac, clientIP, []byte(eqDataRaw))
	if err != nil {
		h.log.Error("create share code failed", zap.Error(err))
		c.JSON(http.StatusOK, gin.H{
			"code": 500,
			"msg":  "系统错误",
		})
		return
	}

	var eqData any
	if err := json.Unmarshal([]byte(data.EqData), &eqData); err != nil {
		eqData = data.EqData
	}

	c.JSON(http.StatusOK, gin.H{
		"code":       200,
		"msg":        "操作成功",
		"share_code": data.ShareCode,
		"expire_at":  data.ExpireAt.Format("2006-01-02 15:04:05"),
		"eq_data":    eqData,
	})
}

// ImportShareCode 导入分享码
//
//	@Summary	导入 EQ 分享码
//	@Description	根据分享码获取他人分享的 EQ 数据
//	@Tags		ShareCode
//	@Produce	json
//	@Param		mac			query		string	true	"设备 MAC 地址"
//	@Param		shareCode	query		string	true	"5 位分享码"
//	@Success	200			{object}	map[string]any	"成功返回 eq_data"
//	@Failure	400			{object}	map[string]any	"参数校验失败"
//	@Failure	500			{object}	map[string]any	"系统错误"
//	@Router		/audio/shareAccept [get]
//
// GET /audio/shareAccept?mac=xx&shareCode=ABC12
func (h *ShareCodeHandler) ImportShareCode(c *gin.Context) {
	mac := strings.TrimSpace(c.Query("mac"))
	shareCode := strings.TrimSpace(c.Query("shareCode"))
	clientIP := encode.ClientPublicIP(c)

	if mac == "" || shareCode == "" {
		c.JSON(http.StatusOK, gin.H{
			"code": 400,
			"msg":  "参数校验失败",
		})
		return
	}
	if len(shareCode) != 5 {
		c.JSON(http.StatusOK, gin.H{
			"code": 0,
			"msg":  "分享码无效或已过期",
		})
		return
	}

	ctx := c.Request.Context()
	data, err := h.cache.Get(ctx, shareCode)
	if err != nil {
		h.log.Error("get share code failed", zap.String("share_code", shareCode), zap.Error(err))
		c.JSON(http.StatusOK, gin.H{
			"code": 500,
			"msg":  "系统错误",
		})
		return
	}
	if data == nil {
		c.JSON(http.StatusOK, gin.H{
			"code": 0,
			"msg":  "分享码无效或已过期",
		})
		return
	}

	if err := h.cache.EnqueueImportLog(ctx, mac, shareCode, clientIP, data.EqData, data.ExpireAt); err != nil {
		h.log.Error("enqueue share import log failed",
			zap.String("share_code", shareCode),
			zap.Error(err),
		)
		c.JSON(http.StatusOK, gin.H{
			"code": 500,
			"msg":  "系统错误",
		})
		return
	}

	var eqData any
	if err := json.Unmarshal([]byte(data.EqData), &eqData); err != nil {
		eqData = data.EqData
	}

	c.JSON(http.StatusOK, gin.H{
		"code":    200,
		"msg":     "操作成功",
		"eq_data": eqData,
	})
}