feat: sync AI compose and ChatBot features

internal/domain/ai.go

@@ -0,0 +1,149 @@
+package domain
+
+import "errors"
+
+const (
+	// AICompose 最大文本长度沿用普通消息正文限制，避免 provider 请求被客户端巨大值打爆。
+	MaxAIComposeTextLength = MaxMessageTextLength
+	// MaxAIComposeEntityCount 限制 TextWithEntities 的实体数量。
+	MaxAIComposeEntityCount = MaxMessageEntityCount
+	// AICompose tone 的 appConfig 默认值与 TDesktop/DrKLO 消费点一致。
+	MaxAIComposeToneTitleLength    = 12
+	MaxAIComposeTonePromptLength   = 1024
+	AIComposeToneSavedLimitDefault = 5
+	AIComposeToneSavedLimitPremium = 20
+	AIComposeToneExamplesNum       = 3
+)
+
+var (
+	ErrAIComposeDisabled            = errors.New("ai compose disabled")
+	ErrAIComposeInvalid             = errors.New("ai compose invalid")
+	ErrAIComposeToneInvalid         = errors.New("ai compose tone invalid")
+	ErrAIComposeToneNotFound        = errors.New("ai compose tone not found")
+	ErrAIComposeToneLimitExceeded   = errors.New("ai compose tone limit exceeded")
+	ErrAIComposeRateLimited         = errors.New("ai compose rate limited")
+	ErrAIComposeProviderUnavailable = errors.New("ai compose provider unavailable")
+	ErrAIComposeProviderTimeout     = errors.New("ai compose provider timeout")
+)
+
+type AIComposeText struct {
+	Text     string
+	Entities []MessageEntity
+}
+
+func (t AIComposeText) Clone() AIComposeText {
+	out := t
+	out.Entities = append([]MessageEntity(nil), t.Entities...)
+	return out
+}
+
+type AIComposeToneRefKind string
+
+const (
+	AIComposeToneRefDefault AIComposeToneRefKind = "default"
+	AIComposeToneRefID      AIComposeToneRefKind = "id"
+	AIComposeToneRefSlug    AIComposeToneRefKind = "slug"
+)
+
+type AIComposeToneRef struct {
+	Kind        AIComposeToneRefKind
+	DefaultTone string
+	ID          int64
+	AccessHash  int64
+	Slug        string
+}
+
+func (r AIComposeToneRef) Empty() bool {
+	return r.Kind == "" && r.DefaultTone == "" && r.ID == 0 && r.Slug == ""
+}
+
+type AIComposeRequest struct {
+	UserID          int64
+	Text            AIComposeText
+	Proofread       bool
+	Emojify         bool
+	TranslateToLang string
+	Tone            AIComposeToneRef
+}
+
+type AIComposeResult struct {
+	ResultText AIComposeText
+	DiffText   *AIComposeText
+}
+
+type AITextGenerationRequest struct {
+	UserID      int64
+	Text        AIComposeText
+	Instruction string
+}
+
+type AIComposeTone struct {
+	Default        bool
+	Creator        bool
+	ID             int64
+	AccessHash     int64
+	OwnerUserID    int64
+	Slug           string
+	Title          string
+	EmojiID        int64
+	Prompt         string
+	InstallsCount  int
+	AuthorID       int64
+	DisplayAuthor  bool
+	CreatedAt      int64
+	UpdatedAt      int64
+	Saved          bool
+	ExampleEnglish *AIComposeToneExample
+}
+
+func (t AIComposeTone) Clone() AIComposeTone {
+	out := t
+	if t.ExampleEnglish != nil {
+		ex := t.ExampleEnglish.Clone()
+		out.ExampleEnglish = &ex
+	}
+	return out
+}
+
+type AIComposeToneExample struct {
+	From AIComposeText
+	To   AIComposeText
+}
+
+func (e AIComposeToneExample) Clone() AIComposeToneExample {
+	return AIComposeToneExample{
+		From: e.From.Clone(),
+		To:   e.To.Clone(),
+	}
+}
+
+type AIComposeToneInput struct {
+	UserID        int64
+	DisplayAuthor bool
+	EmojiID       int64
+	Title         string
+	Prompt        string
+}
+
+type AIComposeToneUpdate struct {
+	Ref           AIComposeToneRef
+	UserID        int64
+	DisplayAuthor *bool
+	EmojiID       *int64
+	Title         *string
+	Prompt        *string
+}
+
+type AIComposeTones struct {
+	Hash  int64
+	Tones []AIComposeTone
+}
+
+func (t AIComposeTones) Clone() AIComposeTones {
+	out := t
+	out.Tones = make([]AIComposeTone, 0, len(t.Tones))
+	for _, tone := range t.Tones {
+		out.Tones = append(out.Tones, tone.Clone())
+	}
+	return out
+}