package chat

import (
	"context"
	"time"

	"github.com/openimsdk/tools/db/pagination"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

// 敏感词相关结构体定义
type SensitiveWord struct {
	ID         string    `bson:"_id" json:"id"`                  // 主键ID
	Word       string    `bson:"word" json:"word"`               // 敏感词内容
	Level      int32     `bson:"level" json:"level"`             // 敏感词级别 1:低 2:中 3:高
	Type       int32     `bson:"type" json:"type"`               // 敏感词类型 1:政治 2:色情 3:暴力 4:广告 5:其他
	Action     int32     `bson:"action" json:"action"`           // 处理动作 1:替换为*** 2:拦截不发
	Status     int32     `bson:"status" json:"status"`           // 状态 1:启用 0:禁用
	Creator    string    `bson:"creator" json:"creator"`         // 创建者
	Updater    string    `bson:"updater" json:"updater"`         // 更新者
	CreateTime time.Time `bson:"create_time" json:"create_time"` // 创建时间
	UpdateTime time.Time `bson:"update_time" json:"update_time"` // 更新时间
	Remark     string    `bson:"remark" json:"remark"`           // 备注
}

func (SensitiveWord) TableName() string {
	return "sensitive_words"
}

type SensitiveWordLog struct {
	ID            primitive.ObjectID `bson:"_id,omitempty"`
	UserID        string             `bson:"user_id"`        // 触发用户ID
	GroupID       string             `bson:"group_id"`       // 触发群组ID (如果是在群聊中)
	Content       string             `bson:"content"`        // 原始消息内容
	MatchedWords  []string           `bson:"matched_words"`  // 匹配到的敏感词
	Action        int32              `bson:"action"`         // 采取的动作
	ProcessedText string             `bson:"processed_text"` // 处理后的文本 (如果被替换)
	CreateTime    time.Time          `bson:"create_time"`
}

func (SensitiveWordLog) TableName() string {
	return "sensitive_word_logs"
}

type SensitiveWordGroup struct {
	ID         primitive.ObjectID `bson:"_id,omitempty"`
	Name       string             `bson:"name"`   // 分组名称
	Remark     string             `bson:"remark"` // 备注
	CreateTime time.Time          `bson:"create_time"`
	UpdateTime time.Time          `bson:"update_time"`
}

func (SensitiveWordGroup) TableName() string {
	return "sensitive_word_groups"
}

type SensitiveWordConfig struct {
	ID              string    `bson:"_id" json:"id"`                            // 主键ID
	EnableFilter    bool      `bson:"enable_filter" json:"enable_filter"`       // 是否启用过滤
	FilterMode      int32     `bson:"filter_mode" json:"filter_mode"`           // 过滤模式
	ReplaceChar     string    `bson:"replace_char" json:"replace_char"`         // 替换字符，默认***
	WhitelistUsers  []string  `bson:"whitelist_users" json:"whitelist_users"`   // 白名单用户
	WhitelistGroups []string  `bson:"whitelist_groups" json:"whitelist_groups"` // 白名单群组
	LogEnabled      bool      `bson:"log_enabled" json:"log_enabled"`           // 是否记录日志
	AutoApprove     bool      `bson:"auto_approve" json:"auto_approve"`         // 是否自动审核
	UpdateTime      time.Time `bson:"update_time" json:"update_time"`           // 更新时间
}

func (SensitiveWordConfig) TableName() string {
	return "sensitive_word_configs"
}

// 敏感词级别常量
const (
	SensitiveLevelLow    = 1 // 低级别
	SensitiveLevelMedium = 2 // 中级别
	SensitiveLevelHigh   = 3 // 高级别
)

// 敏感词类型常量
const (
	SensitiveTypePolitical = 1 // 政治
	SensitiveTypePorn      = 2 // 色情
	SensitiveTypeViolence  = 3 // 暴力
	SensitiveTypeAd        = 4 // 广告
	SensitiveTypeOther     = 5 // 其他
)

// 处理动作常量
const (
	SensitiveActionReplace = 1 // 替换为***
	SensitiveActionBlock   = 2 // 拦截不发
)

// 状态常量
const (
	SensitiveStatusDisabled = 0 // 禁用
	SensitiveStatusEnabled  = 1 // 启用
)

type SensitiveWordInterface interface {
	// 敏感词管理
	CreateSensitiveWord(ctx context.Context, word *SensitiveWord) error
	UpdateSensitiveWord(ctx context.Context, id string, data map[string]any) error
	DeleteSensitiveWord(ctx context.Context, ids []string) error
	GetSensitiveWord(ctx context.Context, id string) (*SensitiveWord, error)
	SearchSensitiveWords(ctx context.Context, keyword string, action int32, status int32, pagination pagination.Pagination) (int64, []*SensitiveWord, error)
	GetAllSensitiveWords(ctx context.Context) ([]*SensitiveWord, error)
	GetEnabledSensitiveWords(ctx context.Context) ([]*SensitiveWord, error)

	// 敏感词检测
	CheckSensitiveWords(ctx context.Context, content string) ([]*SensitiveWord, error)
	FilterContent(ctx context.Context, content string) (string, []*SensitiveWord, error)

	// 敏感词日志
	CreateSensitiveWordLog(ctx context.Context, log *SensitiveWordLog) error
	GetSensitiveWordLogs(ctx context.Context, userID string, groupID string, pagination pagination.Pagination) (int64, []*SensitiveWordLog, error)
	DeleteSensitiveWordLogs(ctx context.Context, ids []string) error

	// 敏感词分组管理
	CreateSensitiveWordGroup(ctx context.Context, group *SensitiveWordGroup) error
	UpdateSensitiveWordGroup(ctx context.Context, id string, data map[string]any) error
	DeleteSensitiveWordGroup(ctx context.Context, ids []string) error
	GetSensitiveWordGroup(ctx context.Context, id string) (*SensitiveWordGroup, error)
	GetAllSensitiveWordGroups(ctx context.Context) ([]*SensitiveWordGroup, error)

	// 敏感词配置管理
	GetSensitiveWordConfig(ctx context.Context) (*SensitiveWordConfig, error)
	UpdateSensitiveWordConfig(ctx context.Context, config *SensitiveWordConfig) error
	IsFilterEnabled(ctx context.Context) (bool, error)
	GetFilterMode(ctx context.Context) (int32, error)
	GetReplaceChar(ctx context.Context) (string, error)
	IsUserInWhitelist(ctx context.Context, userID string) (bool, error)
	IsGroupInWhitelist(ctx context.Context, groupID string) (bool, error)

	// 批量操作
	BatchCreateSensitiveWords(ctx context.Context, words []*SensitiveWord) error
	BatchUpdateSensitiveWords(ctx context.Context, updates map[string]map[string]any) error
	BatchDeleteSensitiveWords(ctx context.Context, ids []string) error

	// 统计信息
	GetSensitiveWordStats(ctx context.Context) (map[string]int64, error)
	GetSensitiveWordLogStats(ctx context.Context, startTime, endTime time.Time) (map[string]int64, error)
}