131 lines
3.5 KiB
Go
131 lines
3.5 KiB
Go
package feishu
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"io"
|
||
"log"
|
||
"net/http"
|
||
"strings"
|
||
"time"
|
||
|
||
"xiaji-go/internal/channel"
|
||
"xiaji-go/internal/service"
|
||
"xiaji-go/models"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// DefaultUserID 统一用户ID，使所有平台共享同一份账本
|
||
const DefaultUserID int64 = 1
|
||
|
||
type Bot struct {
|
||
db *gorm.DB
|
||
finance *service.FinanceService
|
||
appID string
|
||
appSecret string
|
||
verificationToken string
|
||
encryptKey string
|
||
}
|
||
|
||
func NewBot(db *gorm.DB, finance *service.FinanceService, appID, appSecret, verificationToken, encryptKey string) *Bot {
|
||
return &Bot{
|
||
db: db,
|
||
finance: finance,
|
||
appID: appID,
|
||
appSecret: appSecret,
|
||
verificationToken: verificationToken,
|
||
encryptKey: encryptKey,
|
||
}
|
||
}
|
||
|
||
func (b *Bot) Start(ctx context.Context) {
|
||
log.Printf("🚀 Feishu Bot 已启用 app_id=%s", maskID(b.appID))
|
||
<-ctx.Done()
|
||
log.Printf("⏳ Feishu Bot 已停止")
|
||
}
|
||
|
||
func (b *Bot) RegisterRoutes(r *gin.Engine) {
|
||
r.POST("/webhook/feishu", b.handleWebhook)
|
||
}
|
||
|
||
func (b *Bot) handleWebhook(c *gin.Context) {
|
||
body, err := io.ReadAll(io.LimitReader(c.Request.Body, 1<<20))
|
||
if err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": "bad request"})
|
||
return
|
||
}
|
||
|
||
// 统一走 channel 包解析，便于后续扩展验签/解密
|
||
msg, verifyChallenge, err := channel.ParseFeishuInbound(body, b.verificationToken)
|
||
if err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
if verifyChallenge != "" {
|
||
c.JSON(http.StatusOK, gin.H{"challenge": verifyChallenge})
|
||
return
|
||
}
|
||
if msg == nil {
|
||
c.JSON(http.StatusOK, gin.H{"code": 0})
|
||
return
|
||
}
|
||
|
||
// 幂等去重
|
||
var existed models.MessageDedup
|
||
if err := b.db.Where("platform = ? AND event_id = ?", "feishu", msg.EventID).First(&existed).Error; err == nil {
|
||
c.JSON(http.StatusOK, gin.H{"code": 0})
|
||
return
|
||
}
|
||
_ = b.db.Create(&models.MessageDedup{Platform: "feishu", EventID: msg.EventID, ProcessedAt: time.Now()}).Error
|
||
|
||
reply := b.handleText(msg.Text)
|
||
if reply != "" && msg.UserID != "" {
|
||
tk, err := channel.GetFeishuTenantToken(c.Request.Context(), b.appID, b.appSecret)
|
||
if err == nil {
|
||
_ = channel.SendFeishuText(c.Request.Context(), tk, msg.UserID, reply)
|
||
}
|
||
}
|
||
|
||
c.JSON(http.StatusOK, gin.H{"code": 0})
|
||
}
|
||
|
||
func (b *Bot) handleText(text string) string {
|
||
trim := strings.TrimSpace(text)
|
||
switch trim {
|
||
case "帮助", "help", "/help", "菜单", "功能", "/start":
|
||
return "🦞 虾记记账\n\n直接发送消费描述即可记账：\n• 午饭 25元\n• 打车 ¥30\n\n📋 命令：记录/查看、今日/今天、统计"
|
||
case "查看", "记录", "列表", "最近":
|
||
items, err := b.finance.GetTransactions(DefaultUserID, 10)
|
||
if err != nil {
|
||
return "❌ 查询失败"
|
||
}
|
||
if len(items) == 0 {
|
||
return "📭 暂无记录"
|
||
}
|
||
var sb strings.Builder
|
||
sb.WriteString("📋 最近记录：\n\n")
|
||
for _, item := range items {
|
||
sb.WriteString(fmt.Sprintf("%s %s %.2f元\n", item.Date, item.Category, item.AmountYuan()))
|
||
}
|
||
return sb.String()
|
||
}
|
||
|
||
amount, category, err := b.finance.AddTransaction(DefaultUserID, trim)
|
||
if err != nil {
|
||
return "❌ 记账失败，请稍后重试"
|
||
}
|
||
if amount == 0 {
|
||
return "📍 没看到金额，这笔花了多少钱？"
|
||
}
|
||
return fmt.Sprintf("✅ 已记入【%s】：%.2f元\n📝 备注：%s", category, float64(amount)/100.0, trim)
|
||
}
|
||
|
||
func maskID(s string) string {
|
||
if len(s) <= 6 {
|
||
return "***"
|
||
}
|
||
return s[:3] + "***" + s[len(s)-3:]
|
||
}
|