eafonyang
293 lines
7.7 KiB
Go
293 lines
7.7 KiB
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
|
|
maxPerMac int
|
|
}
|
|
|
|
func NewShareCodeHandler(shareCache *cache.ShareCodeCache, log *zap.Logger, maxPerMac int) *ShareCodeHandler {
|
|
return &ShareCodeHandler{cache: shareCache, log: log, maxPerMac: maxPerMac}
|
|
}
|
|
|
|
// ExportShareCode 导出分享码
|
|
//
|
|
// @Summary 创建 EQ 分享码
|
|
// @Description 将用户的 EQ 数据生成一个 5 位分享码，有效期由 SHARE_CODE_TTL_MIN 配置决定（支持 30、30m、24h、30d 等格式）
|
|
// @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()
|
|
|
|
// 限制同一 MAC 在有效期内的分享码数量
|
|
existing, err := h.cache.ListByMac(ctx, mac)
|
|
if err != nil {
|
|
h.log.Error("check existing share codes by mac failed", zap.String("mac", mac), zap.Error(err))
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 500,
|
|
"msg": "系统错误",
|
|
})
|
|
return
|
|
}
|
|
if len(existing) >= h.maxPerMac {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"msg": "已有未过期的分享码，请稍后再试",
|
|
})
|
|
return
|
|
}
|
|
|
|
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,
|
|
})
|
|
}
|
|
|
|
// ListShareCodesByMac 查询某 MAC 尚未过期的分享码
|
|
//
|
|
// @Summary 查询未过期分享码
|
|
// @Description 根据设备 MAC 地址查询该设备尚未过期的所有分享码（仅查询 Redis，依赖 TTL 自动过期）
|
|
// @Tags ShareCode
|
|
// @Produce json
|
|
// @Param mac query string true "设备 MAC 地址"
|
|
// @Success 200 {object} map[string]any "成功返回 share_codes 列表"
|
|
// @Failure 400 {object} map[string]any "参数校验失败"
|
|
// @Failure 500 {object} map[string]any "系统错误"
|
|
// @Router /audio/shareList [get]
|
|
func (h *ShareCodeHandler) ListShareCodesByMac(c *gin.Context) {
|
|
mac := strings.TrimSpace(c.Query("mac"))
|
|
if mac == "" {
|
|
c.JSON(http.StatusOK, gin.H{"code": 400, "msg": "参数校验失败"})
|
|
return
|
|
}
|
|
|
|
ctx := c.Request.Context()
|
|
|
|
items, err := h.cache.ListByMac(ctx, mac)
|
|
if err != nil {
|
|
h.log.Error("list share codes by mac failed", zap.String("mac", mac), zap.Error(err))
|
|
c.JSON(http.StatusOK, gin.H{"code": 500, "msg": "系统错误"})
|
|
return
|
|
}
|
|
|
|
type shareItem struct {
|
|
ShareCode string `json:"share_code"`
|
|
ExpireAt string `json:"expire_at"`
|
|
EqData any `json:"eq_data"`
|
|
}
|
|
list := make([]shareItem, 0, len(items))
|
|
for _, item := range items {
|
|
var eqData any
|
|
if err := json.Unmarshal([]byte(item.EqData), &eqData); err != nil {
|
|
eqData = item.EqData
|
|
}
|
|
list = append(list, shareItem{
|
|
ShareCode: item.ShareCode,
|
|
ExpireAt: item.ExpireAt.Format("2006-01-02 15:04:05"),
|
|
EqData: eqData,
|
|
})
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 200,
|
|
"msg": "操作成功",
|
|
"share_codes": list,
|
|
})
|
|
}
|
|
|
|
// QueryShareCode 查询分享码（预览，不记录导入）
|
|
//
|
|
// @Summary 查询 EQ 分享码
|
|
// @Description 根据分享码预览 EQ 数据，不写入导入流水
|
|
// @Tags ShareCode
|
|
// @Produce json
|
|
// @Param shareCode query string true "5 位分享码"
|
|
// @Success 200 {object} map[string]any "成功返回 eq_data、expire_at"
|
|
// @Failure 400 {object} map[string]any "参数校验失败"
|
|
// @Failure 500 {object} map[string]any "系统错误"
|
|
// @Router /audio/shareQuery [get]
|
|
//
|
|
// GET /audio/shareQuery?shareCode=ABC12
|
|
func (h *ShareCodeHandler) QueryShareCode(c *gin.Context) {
|
|
shareCode := strings.TrimSpace(c.Query("shareCode"))
|
|
data, ok := h.lookupShareCode(c, shareCode)
|
|
if !ok {
|
|
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,
|
|
"expire_at": data.ExpireAt.Format("2006-01-02 15:04:05"),
|
|
})
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
data, ok := h.lookupShareCode(c, shareCode)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
ctx := c.Request.Context()
|
|
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,
|
|
})
|
|
}
|
|
|
|
func (h *ShareCodeHandler) lookupShareCode(c *gin.Context, shareCode string) (*cache.ShareCodeData, bool) {
|
|
if shareCode == "" {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 400,
|
|
"msg": "参数校验失败",
|
|
})
|
|
return nil, false
|
|
}
|
|
if len(shareCode) != 5 {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"msg": "分享码无效或已过期",
|
|
})
|
|
return nil, false
|
|
}
|
|
|
|
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 nil, false
|
|
}
|
|
if data == nil {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"msg": "分享码无效或已过期",
|
|
})
|
|
return nil, false
|
|
}
|
|
return data, true
|
|
}
|