package feishu

import (
	"context"
	"io"
	"log"
	"strings"
	"time"

	"ops-assistant/internal/channel"
	"ops-assistant/internal/core/ops"
	"ops-assistant/models"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

// DefaultUserID 统一用户ID
const DefaultUserID int64 = 1

type Bot struct {
	db                *gorm.DB
	opsSvc            *ops.Service
	appID             string
	appSecret         string
	verificationToken string
	encryptKey        string
}

func NewBot(db *gorm.DB, opsSvc *ops.Service, appID, appSecret, verificationToken, encryptKey string) *Bot {
	return &Bot{
		db:                db,
		opsSvc:            opsSvc,
		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(400, gin.H{"error": "bad request"})
		return
	}

	// 统一走 channel 包解析，便于后续扩展验签/解密
	msg, verifyChallenge, err := channel.ParseFeishuInbound(body, b.verificationToken)
	if err != nil {
		c.JSON(400, gin.H{"error": err.Error()})
		return
	}
	if verifyChallenge != "" {
		c.JSON(200, gin.H{"challenge": verifyChallenge})
		return
	}
	if msg == nil {
		c.JSON(200, 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(200, 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(200, gin.H{"code": 0})
}

func (b *Bot) handleText(text string) string {
	trim := strings.TrimSpace(text)
	if b.opsSvc != nil {
		if handled, out := b.opsSvc.Handle(DefaultUserID, trim); handled {
			return out
		}
	}
	switch trim {
	case "帮助", "help", "/help", "菜单", "功能", "/start":
		return "🛠️ Ops-Assistant\n\n模块化运维命令助手，支持：\n• /cpa - CPA 管理\n• /cf - Cloudflare 管理\n• /cfdav - cfdav 管理\n• /sub - 订阅管理\n• /mail - 邮箱管理"
	default:
		return "❓ 无法识别，请使用 OPS 命令（发送「帮助」查看说明）。"
	}
}

func maskID(s string) string {
	if len(s) <= 6 {
		return "***"
	}
	return s[:3] + "***" + s[len(s)-3:]
}