mirror of
https://github.com/prosolis/gogobee.git
synced 2026-07-16 00:52:40 +00:00
Rewrite from TypeScript to Go
Complete rewrite of the Freebee Matrix bot as GogoBee using mautrix-go. - E2EE with goolm (pure Go, no CGo/libolm) and cross-signing bootstrap - 35+ plugins with dependency injection and ordered registration - SQLite storage via modernc.org/sqlite (no CGo) - Scheduler via robfig/cron for WOTD, holidays, birthdays, releases, etc. - Optional LLM integration (Ollama) for sentiment, profanity, roasts, vibes - Threaded trivia, three-tier profanity tracking, WOTD usage verification - Multi-country holiday support with deduplication - Quadratic XP curve, configurable bot display name, encrypted DMs Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
prosolis
678
internal/plugin/llm_passive.go
Normal file
678
internal/plugin/llm_passive.go
Normal file
@@ -0,0 +1,678 @@
|
||||
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)
|
||||
}
|
||||
Reference in New Issue
Block a user