gogobee/internal/plugin/llm_passive.go

package plugin

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"log/slog"
	"math/rand"
	"net/http"
	"os"
	"regexp"
	"strconv"
	"strings"
	"sync"
	"time"
	"unicode"

	"gogobee/internal/db"

	"maunium.net/go/mautrix"
	"maunium.net/go/mautrix/id"
)

// profanityKeywords is a basic list used for pre-filtering.
var profanityKeywords = []string{
	"fuck", "shit", "damn", "hell", "ass", "bitch", "crap", "dick",
	"bastard", "piss", "cock", "cunt", "douche", "dumbass", "idiot",
	"moron", "stupid", "stfu", "wtf", "lmao", "lmfao",
}

var thanksPatternRe = regexp.MustCompile(`(?i)\b(thanks|thank\s+you|thankyou|thx|ty|tysm|tyvm|appreciate)\b`)
var mentionRe = regexp.MustCompile(`@[a-zA-Z0-9._=-]+:[a-zA-Z0-9.-]+`)

// classificationResult holds the parsed LLM JSON response.
type classificationResult struct {
	Sentiment         string   `json:"sentiment"`
	SentimentScore    float64  `json:"sentiment_score"`
	Topics            []string `json:"topics"`
	Profanity         bool     `json:"profanity"`
	ProfanitySeverity int      `json:"profanity_severity"`
	InsultTarget      string   `json:"insult_target"`
	WOTDUsed          bool     `json:"wotd_used"`
	GratitudeTarget   string   `json:"gratitude_target"`
}

// queueItem holds a message pending classification.
type queueItem struct {
	UserID  id.UserID
	RoomID  id.RoomID
	EventID id.EventID
	Body    string
}

// LLMPassivePlugin classifies messages using Ollama and reacts accordingly.
type LLMPassivePlugin struct {
	Base
	xp         *XPPlugin
	ollamaHost  string
	ollamaModel string
	sampleRate  float64
	enabled     bool

	mu      sync.Mutex
	queue   []queueItem
	backoff time.Duration

	httpClient *http.Client
	stopCh     chan struct{}
}

// NewLLMPassivePlugin creates a new LLM passive classification plugin.
func NewLLMPassivePlugin(client *mautrix.Client, xp *XPPlugin) *LLMPassivePlugin {
	host := os.Getenv("OLLAMA_HOST")
	model := os.Getenv("OLLAMA_MODEL")
	enabled := host != "" && model != ""

	sampleRate := 0.15
	if v := os.Getenv("LLM_SAMPLE_RATE"); v != "" {
		if parsed, err := strconv.ParseFloat(v, 64); err == nil && parsed >= 0 && parsed <= 1 {
			sampleRate = parsed
		}
	}

	p := &LLMPassivePlugin{
		Base:        NewBase(client),
		xp:          xp,
		ollamaHost:  host,
		ollamaModel: model,
		sampleRate:  sampleRate,
		enabled:     enabled,
		backoff:     5 * time.Second,
		httpClient:  &http.Client{Timeout: 30 * time.Second},
		stopCh:      make(chan struct{}),
	}

	return p
}

func (p *LLMPassivePlugin) Name() string { return "llm_passive" }

func (p *LLMPassivePlugin) Commands() []CommandDef {
	return []CommandDef{
		{Name: "potty", Description: "Show profanity count for a user", Usage: "!potty [@user]", Category: "LLM & Sentiment"},
		{Name: "pottyboard", Description: "Top 10 most profane users", Usage: "!pottyboard", Category: "LLM & Sentiment"},
		{Name: "insults", Description: "Show insult stats for a user", Usage: "!insults [@user]", Category: "LLM & Sentiment"},
		{Name: "insultboard", Description: "Top 10 most insulted users", Usage: "!insultboard", Category: "LLM & Sentiment"},
		{Name: "sentiment", Description: "Show sentiment stats for a user", Usage: "!sentiment [@user]", Category: "LLM & Sentiment"},
	}
}

func (p *LLMPassivePlugin) Init() error {
	if p.enabled {
		slog.Info("llm_passive: enabled", "host", p.ollamaHost, "model", p.ollamaModel, "sample_rate", p.sampleRate)
		go p.processQueue()
	} else {
		slog.Warn("llm_passive: disabled (OLLAMA_HOST or OLLAMA_MODEL not set)",
			"host", p.ollamaHost, "model", p.ollamaModel)
	}
	return nil
}

func (p *LLMPassivePlugin) OnReaction(_ ReactionContext) error { return nil }

func (p *LLMPassivePlugin) OnMessage(ctx MessageContext) error {
	// Handle commands
	if p.IsCommand(ctx.Body, "potty") {
		return p.handlePotty(ctx)
	}
	if p.IsCommand(ctx.Body, "pottyboard") {
		return p.handlePottyboard(ctx)
	}
	if p.IsCommand(ctx.Body, "insults") {
		return p.handleInsults(ctx)
	}
	if p.IsCommand(ctx.Body, "insultboard") {
		return p.handleInsultboard(ctx)
	}
	if p.IsCommand(ctx.Body, "sentiment") {
		return p.handleSentiment(ctx)
	}

	if !p.enabled {
		return nil
	}

	// Skip command messages
	if ctx.IsCommand {
		return nil
	}

	// Pre-filter: only classify messages that match certain criteria
	if !p.shouldClassify(ctx.Body) {
		slog.Debug("llm_passive: message did not pass pre-filter", "body_len", len(ctx.Body))
		return nil
	}

	// Enqueue for async classification
	slog.Debug("llm_passive: enqueuing message for classification", "user", ctx.Sender, "body_len", len(ctx.Body))
	p.mu.Lock()
	p.queue = append(p.queue, queueItem{
		UserID:  ctx.Sender,
		RoomID:  ctx.RoomID,
		EventID: ctx.EventID,
		Body:    ctx.Body,
	})
	p.mu.Unlock()

	return nil
}

// shouldClassify applies pre-filtering heuristics.
func (p *LLMPassivePlugin) shouldClassify(body string) bool {
	// Skip single-character messages (trivia answers, etc.)
	if len(strings.TrimSpace(body)) <= 1 {
		return false
	}

	lower := strings.ToLower(body)

	// Check profanity keywords
	for _, kw := range profanityKeywords {
		if strings.Contains(lower, kw) {
			return true
		}
	}

	// Check mentions
	if mentionRe.MatchString(body) {
		return true
	}

	// Check WOTD pattern (simple heuristic)
	if strings.Contains(lower, "wotd") {
		return true
	}

	// Check non-ASCII
	for _, r := range body {
		if r > unicode.MaxASCII {
			return true
		}
	}

	// Check thanks patterns
	if thanksPatternRe.MatchString(body) {
		return true
	}

	// Random sample of remaining messages (controlled by LLM_SAMPLE_RATE)
	if p.sampleRate >= 1.0 || rand.Float64() < p.sampleRate {
		return true
	}

	return false
}

// processQueue processes the classification queue with backoff.
func (p *LLMPassivePlugin) processQueue() {
	for {
		select {
		case <-p.stopCh:
			return
		default:
		}

		p.mu.Lock()
		if len(p.queue) == 0 {
			p.mu.Unlock()
			time.Sleep(1 * time.Second)
			continue
		}
		item := p.queue[0]
		p.queue = p.queue[1:]
		p.mu.Unlock()

		slog.Debug("llm_passive: processing queued item", "user", item.UserID, "body_preview", truncate(item.Body, 50))
		err := p.classifyAndProcess(item)
		if err != nil {
			slog.Error("llm: classification failed", "err", err, "user", item.UserID)
			// Backoff on error
			p.mu.Lock()
			p.backoff *= 2
			if p.backoff > 5*time.Minute {
				p.backoff = 5 * time.Minute
			}
			p.mu.Unlock()
			time.Sleep(p.backoff)
		} else {
			// Reset backoff on success
			p.mu.Lock()
			p.backoff = 5 * time.Second
			p.mu.Unlock()
		}
	}
}

// classifyAndProcess sends a message to Ollama and processes the result.
func (p *LLMPassivePlugin) classifyAndProcess(item queueItem) error {
	result, err := p.callOllama(item.Body)
	if err != nil {
		return fmt.Errorf("ollama call: %w", err)
	}

	d := db.Get()

	// Store classification
	topicsJSON, _ := json.Marshal(result.Topics)
	profanityInt := 0
	if result.Profanity {
		profanityInt = 1
	}
	wotdInt := 0
	if result.WOTDUsed {
		wotdInt = 1
	}

	_, err = d.Exec(
		`INSERT INTO llm_classifications (user_id, room_id, message_text, sentiment, sentiment_score, topics, profanity, profanity_severity, insult_target, wotd_used, gratitude_target)
		 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
		string(item.UserID), string(item.RoomID), item.Body,
		result.Sentiment, result.SentimentScore, string(topicsJSON),
		profanityInt, result.ProfanitySeverity, result.InsultTarget, wotdInt, result.GratitudeTarget,
	)
	if err != nil {
		slog.Error("llm: store classification", "err", err)
	}

	// Aggregate sentiment stats
	switch result.Sentiment {
	case "positive":
		_, _ = d.Exec(
			`INSERT INTO sentiment_stats (user_id, positive, total_score) VALUES (?, 1, ?)
			 ON CONFLICT(user_id) DO UPDATE SET positive = positive + 1, total_score = total_score + ?`,
			string(item.UserID), result.SentimentScore, result.SentimentScore,
		)
	case "negative":
		_, _ = d.Exec(
			`INSERT INTO sentiment_stats (user_id, negative, total_score) VALUES (?, 1, ?)
			 ON CONFLICT(user_id) DO UPDATE SET negative = negative + 1, total_score = total_score + ?`,
			string(item.UserID), result.SentimentScore, result.SentimentScore,
		)
	default:
		_, _ = d.Exec(
			`INSERT INTO sentiment_stats (user_id, neutral, total_score) VALUES (?, 1, ?)
			 ON CONFLICT(user_id) DO UPDATE SET neutral = neutral + 1, total_score = total_score + ?`,
			string(item.UserID), result.SentimentScore, result.SentimentScore,
		)
	}

	// Track profanity with severity
	if result.Profanity {
		severity := result.ProfanitySeverity
		if severity < 1 {
			severity = 1
		}
		if severity > 3 {
			severity = 3
		}
		mildInc, modInc, scorchInc := 0, 0, 0
		switch severity {
		case 1:
			mildInc = 1
		case 2:
			modInc = 1
		case 3:
			scorchInc = 1
		}
		_, _ = d.Exec(
			`INSERT INTO potty_mouth (user_id, count, mild, moderate, scorching) VALUES (?, 1, ?, ?, ?)
			 ON CONFLICT(user_id) DO UPDATE SET count = count + 1, mild = mild + ?, moderate = moderate + ?, scorching = scorching + ?`,
			string(item.UserID), mildInc, modInc, scorchInc, mildInc, modInc, scorchInc,
		)
	}

	// React if someone insults the bot
	if result.InsultTarget == string(p.Client.UserID) {
		botReactions := []string{
			"\U0001f595", // 🖕
			"\U0001f52a", // 🔪
			"\U0001f5e1", // 🗡️
			"\U0001fa78", // 🩸
			"\U0001f480", // 💀
		}
		emoji := botReactions[rand.Intn(len(botReactions))]
		_ = p.SendReact(item.RoomID, item.EventID, emoji)
	}

	// Track insults
	if result.InsultTarget != "" {
		_, _ = d.Exec(
			`INSERT INTO insult_log (user_id, times_insulting) VALUES (?, 1)
			 ON CONFLICT(user_id) DO UPDATE SET times_insulting = times_insulting + 1`,
			string(item.UserID),
		)
		_, _ = d.Exec(
			`INSERT INTO insult_log (user_id, times_insulted) VALUES (?, 1)
			 ON CONFLICT(user_id) DO UPDATE SET times_insulted = times_insulted + 1`,
			result.InsultTarget,
		)
	}

	// React with emojis based on classification
	if result.Sentiment == "positive" && result.SentimentScore > 0.7 {
		_ = p.SendReact(item.RoomID, item.EventID, "\U0001f44d") // thumbsup
	}
	if result.Sentiment == "negative" && result.SentimentScore < -0.7 {
		_ = p.SendReact(item.RoomID, item.EventID, "\U0001f44e") // thumbsdown
	}
	if result.Profanity {
		switch result.ProfanitySeverity {
		case 3:
			_ = p.SendReact(item.RoomID, item.EventID, "\U0001f92c") // 🤬
		case 2:
			_ = p.SendReact(item.RoomID, item.EventID, "\U0001f632") // 😲
		default:
			_ = p.SendReact(item.RoomID, item.EventID, "\U0001fae3") // 🫣
		}
	}
	if result.WOTDUsed {
		_ = p.SendReact(item.RoomID, item.EventID, "\U0001f4d6") // open book
	}
	if result.GratitudeTarget != "" {
		_ = p.SendReact(item.RoomID, item.EventID, "\U0001f49c") // purple heart
	}

	// Grant XP for gratitude
	if result.GratitudeTarget != "" && p.xp != nil {
		targetID := id.UserID(result.GratitudeTarget)
		if targetID != item.UserID {
			p.xp.GrantXP(targetID, 5, "gratitude")
		}
	}

	return nil
}

// ollamaRequest is the request body for the Ollama API.
type ollamaRequest struct {
	Model  string `json:"model"`
	Prompt string `json:"prompt"`
	Stream bool   `json:"stream"`
}

// ollamaResponse is the response from the Ollama API.
type ollamaResponse struct {
	Response string `json:"response"`
}

// callOllama sends a classification prompt to Ollama and parses the JSON result.
func (p *LLMPassivePlugin) callOllama(messageText string) (*classificationResult, error) {
	prompt := fmt.Sprintf(`Classify the following chat message. Respond ONLY with valid JSON (no markdown, no explanation).

JSON schema:
{
  "sentiment": "positive" | "negative" | "neutral",
  "sentiment_score": number between -1.0 and 1.0,
  "topics": ["topic1", "topic2"],
  "profanity": true | false,
  "profanity_severity": 0 | 1 | 2 | 3 (0=none, 1=mild e.g. damn/hell/crap, 2=moderate e.g. shit/ass/bitch, 3=scorching e.g. fuck/cunt and slurs),
  "insult_target": "" or "@user:server" if someone is being insulted,
  "wotd_used": true | false (if the message uses an unusual/sophisticated word),
  "gratitude_target": "" or "@user:server" if thanking someone
}

Message: %s`, messageText)

	reqBody := ollamaRequest{
		Model:  p.ollamaModel,
		Prompt: prompt,
		Stream: false,
	}

	body, err := json.Marshal(reqBody)
	if err != nil {
		return nil, fmt.Errorf("marshal request: %w", err)
	}

	url := strings.TrimRight(p.ollamaHost, "/") + "/api/generate"
	slog.Debug("llm_passive: calling ollama", "url", url, "model", p.ollamaModel)
	resp, err := p.httpClient.Post(url, "application/json", bytes.NewReader(body))
	if err != nil {
		return nil, fmt.Errorf("ollama request: %w", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		respBody, _ := io.ReadAll(resp.Body)
		return nil, fmt.Errorf("ollama status %d: %s", resp.StatusCode, string(respBody))
	}

	var ollamaResp ollamaResponse
	if err := json.NewDecoder(resp.Body).Decode(&ollamaResp); err != nil {
		return nil, fmt.Errorf("decode ollama response: %w", err)
	}

	result, err := parseClassification(ollamaResp.Response)
	if err != nil {
		return nil, fmt.Errorf("parse classification: %w", err)
	}

	return result, nil
}

// parseClassification parses a JSON classification response with repair logic.
func parseClassification(raw string) (*classificationResult, error) {
	cleaned := raw

	// Remove markdown code fences
	cleaned = strings.TrimSpace(cleaned)
	if strings.HasPrefix(cleaned, "```json") {
		cleaned = strings.TrimPrefix(cleaned, "```json")
	} else if strings.HasPrefix(cleaned, "```") {
		cleaned = strings.TrimPrefix(cleaned, "```")
	}
	if strings.HasSuffix(cleaned, "```") {
		cleaned = strings.TrimSuffix(cleaned, "```")
	}
	cleaned = strings.TrimSpace(cleaned)

	// Try to parse directly first
	var result classificationResult
	if err := json.Unmarshal([]byte(cleaned), &result); err == nil {
		return &result, nil
	}

	// Repair: replace single quotes with double quotes
	repaired := strings.ReplaceAll(cleaned, "'", "\"")

	// Repair: remove trailing commas before } or ]
	trailingCommaRe := regexp.MustCompile(`,\s*([}\]])`)
	repaired = trailingCommaRe.ReplaceAllString(repaired, "$1")

	if err := json.Unmarshal([]byte(repaired), &result); err != nil {
		return nil, fmt.Errorf("JSON parse failed after repair: %w (raw: %s)", err, raw)
	}

	return &result, nil
}

func truncate(s string, n int) string {
	if len(s) <= n {
		return s
	}
	return s[:n] + "..."
}

// Command handlers

func (p *LLMPassivePlugin) handlePotty(ctx MessageContext) error {
	target := ctx.Sender
	args := p.GetArgs(ctx.Body, "potty")
	if args != "" {
		cleaned := strings.TrimSpace(strings.TrimPrefix(args, "@"))
		if cleaned != "" {
			target = id.UserID(cleaned)
		}
	}

	d := db.Get()
	var count, mild, moderate, scorching int
	err := d.QueryRow(
		`SELECT COALESCE(count, 0), COALESCE(mild, 0), COALESCE(moderate, 0), COALESCE(scorching, 0)
		 FROM potty_mouth WHERE user_id = ?`,
		string(target),
	).Scan(&count, &mild, &moderate, &scorching)
	if err != nil {
		count = 0
	}

	msg := fmt.Sprintf("Profanity for %s: %s total (🫣 mild: %s, 😲 moderate: %s, 🤬 scorching: %s)",
		string(target), formatNumber(count), formatNumber(mild), formatNumber(moderate), formatNumber(scorching))
	return p.SendReply(ctx.RoomID, ctx.EventID, msg)
}

func (p *LLMPassivePlugin) handlePottyboard(ctx MessageContext) error {
	d := db.Get()
	rows, err := d.Query(
		`SELECT user_id, count FROM potty_mouth ORDER BY count DESC LIMIT 10`,
	)
	if err != nil {
		slog.Error("llm: pottyboard query", "err", err)
		return p.SendReply(ctx.RoomID, ctx.EventID, "Failed to load potty board.")
	}
	defer rows.Close()

	var sb strings.Builder
	sb.WriteString("Potty Mouth Board — Top 10\n\n")

	medals := []string{"\U0001f947", "\U0001f948", "\U0001f949"}
	i := 0
	for rows.Next() {
		var userID string
		var count int
		if err := rows.Scan(&userID, &count); err != nil {
			continue
		}
		prefix := fmt.Sprintf("#%d", i+1)
		if i < len(medals) {
			prefix = medals[i]
		}
		sb.WriteString(fmt.Sprintf("%s %s — %s incidents\n", prefix, userID, formatNumber(count)))
		i++
	}

	if i == 0 {
		return p.SendReply(ctx.RoomID, ctx.EventID, "No profanity data yet.")
	}

	return p.SendReply(ctx.RoomID, ctx.EventID, sb.String())
}

func (p *LLMPassivePlugin) handleInsults(ctx MessageContext) error {
	target := ctx.Sender
	args := p.GetArgs(ctx.Body, "insults")
	if args != "" {
		cleaned := strings.TrimSpace(strings.TrimPrefix(args, "@"))
		if cleaned != "" {
			target = id.UserID(cleaned)
		}
	}

	d := db.Get()
	var insulted, insulting int
	err := d.QueryRow(
		`SELECT COALESCE(times_insulted, 0), COALESCE(times_insulting, 0) FROM insult_log WHERE user_id = ?`,
		string(target),
	).Scan(&insulted, &insulting)
	if err != nil {
		insulted = 0
		insulting = 0
	}

	msg := fmt.Sprintf("Insult stats for %s:\nTimes insulted: %s\nTimes insulting others: %s",
		string(target), formatNumber(insulted), formatNumber(insulting))
	return p.SendReply(ctx.RoomID, ctx.EventID, msg)
}

func (p *LLMPassivePlugin) handleInsultboard(ctx MessageContext) error {
	d := db.Get()
	rows, err := d.Query(
		`SELECT user_id, times_insulted FROM insult_log ORDER BY times_insulted DESC LIMIT 10`,
	)
	if err != nil {
		slog.Error("llm: insultboard query", "err", err)
		return p.SendReply(ctx.RoomID, ctx.EventID, "Failed to load insult board.")
	}
	defer rows.Close()

	var sb strings.Builder
	sb.WriteString("Most Insulted — Top 10\n\n")

	medals := []string{"\U0001f947", "\U0001f948", "\U0001f949"}
	i := 0
	for rows.Next() {
		var userID string
		var count int
		if err := rows.Scan(&userID, &count); err != nil {
			continue
		}
		prefix := fmt.Sprintf("#%d", i+1)
		if i < len(medals) {
			prefix = medals[i]
		}
		sb.WriteString(fmt.Sprintf("%s %s — %s times\n", prefix, userID, formatNumber(count)))
		i++
	}

	if i == 0 {
		return p.SendReply(ctx.RoomID, ctx.EventID, "No insult data yet.")
	}

	return p.SendReply(ctx.RoomID, ctx.EventID, sb.String())
}

func (p *LLMPassivePlugin) handleSentiment(ctx MessageContext) error {
	target := ctx.Sender
	args := p.GetArgs(ctx.Body, "sentiment")
	if args != "" {
		cleaned := strings.TrimSpace(strings.TrimPrefix(args, "@"))
		if cleaned != "" {
			target = id.UserID(cleaned)
		}
	}

	d := db.Get()
	var positive, negative, neutral int
	var totalScore float64
	err := d.QueryRow(
		`SELECT COALESCE(positive, 0), COALESCE(negative, 0), COALESCE(neutral, 0), COALESCE(total_score, 0)
		 FROM sentiment_stats WHERE user_id = ?`,
		string(target),
	).Scan(&positive, &negative, &neutral, &totalScore)
	if err != nil {
		return p.SendReply(ctx.RoomID, ctx.EventID, fmt.Sprintf("No sentiment data for %s yet.", string(target)))
	}

	total := positive + negative + neutral
	avgScore := 0.0
	if total > 0 {
		avgScore = totalScore / float64(total)
	}

	mood := "neutral"
	if avgScore > 0.3 {
		mood = "mostly positive"
	} else if avgScore > 0.1 {
		mood = "leaning positive"
	} else if avgScore < -0.3 {
		mood = "mostly negative"
	} else if avgScore < -0.1 {
		mood = "leaning negative"
	}

	msg := fmt.Sprintf("Sentiment for %s:\n😊 Positive: %s  😐 Neutral: %s  😠 Negative: %s\nAverage mood: %.2f (%s)",
		string(target), formatNumber(positive), formatNumber(neutral), formatNumber(negative), avgScore, mood)
	return p.SendReply(ctx.RoomID, ctx.EventID, msg)
}