feat: add uwu top yappers

Author: polarhive

ash.go

@@ -1,11 +1,18 @@
-// Package main provides unit tests for the ash Matrix bot functionality.
 package main
 
 import (
+	"context"
+	"database/sql"
+	"fmt"
+	"strings"
 	"testing"
+	"time"
+
+	_ "github.com/mattn/go-sqlite3"
+	"maunium.net/go/mautrix/event"
+	"maunium.net/go/mautrix/id"
 )
 
-// TestExtractLinks tests the URL extraction from text messages.
 func TestExtractLinks(t *testing.T) {
 	tests := []struct {
 		name  string
@@ -37,7 +44,6 @@ func TestExtractLinks(t *testing.T) {
 	}
 }
 
-// TestIsBlacklisted tests URL blacklisting functionality.
 func TestIsBlacklisted(t *testing.T) {
 	blacklist, err := LoadBlacklist("blacklist.json")
 	if err != nil {
@@ -47,7 +53,6 @@ func TestIsBlacklisted(t *testing.T) {
 	_ = IsBlacklisted("https://example.com", blacklist)
 }
 
-// TestExtractJSONPath tests JSON path extraction for bot commands.
 func TestExtractJSONPath(t *testing.T) {
 	root := map[string]interface{}{
 		"a": map[string]interface{}{
@@ -84,7 +89,6 @@ func TestExtractJSONPath(t *testing.T) {
 	}
 }
 
-// TestFormatPosts tests formatting of post arrays for bot responses.
 func TestFormatPosts(t *testing.T) {
 	posts := []interface{}{
 		map[string]interface{}{"title": "Post 1", "url": "https://a.com"},
@@ -102,7 +106,6 @@ func TestFormatPosts(t *testing.T) {
 	}
 }
 
-// TestFormatPostsLimit tests that formatPosts limits output to 5 posts.
 func TestFormatPostsLimit(t *testing.T) {
 	// More than 5 posts should be capped
 	posts := make([]interface{}, 10)
@@ -125,7 +128,6 @@ func TestFormatPostsLimit(t *testing.T) {
 	}
 }
 
-// TestTruncateText tests text truncation for AI prompts.
 func TestTruncateText(t *testing.T) {
 	tests := []struct {
 		name       string
@@ -147,7 +149,6 @@ func TestTruncateText(t *testing.T) {
 	}
 }
 
-// TestStripCommandPrefix tests removal of bot command prefixes.
 func TestStripCommandPrefix(t *testing.T) {
 	tests := []struct {
 		input string
@@ -170,7 +171,6 @@ func TestStripCommandPrefix(t *testing.T) {
 	}
 }
 
-// TestResolveReplyLabel tests reply label resolution logic.
 func TestResolveReplyLabel(t *testing.T) {
 	tests := []struct {
 		name   string
@@ -194,7 +194,6 @@ func TestResolveReplyLabel(t *testing.T) {
 	}
 }
 
-// TestInSlice tests string slice membership checking.
 func TestInSlice(t *testing.T) {
 	slice := []string{"a", "b", "c"}
 	if !inSlice(slice, "b") {
@@ -208,7 +207,6 @@ func TestInSlice(t *testing.T) {
 	}
 }
 
-// TestTruncate tests string truncation utility.
 func TestTruncate(t *testing.T) {
 	if truncate("hello", 10) != "hello" {
 		t.Error("truncate should not truncate short string")
@@ -219,7 +217,6 @@ func TestTruncate(t *testing.T) {
 	}
 }
 
-// TestGenerateHelpMessage tests help message generation for bot commands.
 func TestGenerateHelpMessage(t *testing.T) {
 	botCfg := &BotConfig{
 		Commands: map[string]BotCommand{
@@ -245,7 +242,6 @@ func TestGenerateHelpMessage(t *testing.T) {
 	}
 }
 
-// TestLoadBotConfig tests loading of bot configuration from file.
 func TestLoadBotConfig(t *testing.T) {
 	cfg, err := LoadBotConfig("bot.json")
 	if err != nil {
@@ -275,12 +271,10 @@ func TestLoadBotConfig(t *testing.T) {
 
 // helpers
 
-// contains checks if a substring is present in a string.
 func contains(s, substr string) bool {
 	return len(s) >= len(substr) && searchString(s, substr)
 }
 
-// searchString performs a simple substring search.
 func searchString(s, substr string) bool {
 	for i := 0; i <= len(s)-len(substr); i++ {
 		if s[i:i+len(substr)] == substr {
@@ -290,7 +284,6 @@ func searchString(s, substr string) bool {
 	return false
 }
 
-// splitLines splits a string into lines by newline characters.
 func splitLines(s string) []string {
 	var lines []string
 	start := 0
@@ -305,3 +298,150 @@ func splitLines(s string) []string {
 	}
 	return lines
 }
+
+func TestUwuify(t *testing.T) {
+	tests := []struct {
+		name  string
+		input string
+		check func(string) bool
+		desc  string
+	}{
+		{
+			"replaces r/l with w",
+			"really cool",
+			func(s string) bool { return strings.Contains(s, "w") },
+			"should replace r and l with w",
+		},
+		{
+			"replaces th with d",
+			"the weather",
+			func(s string) bool { return strings.Contains(s, "da") && strings.Contains(s, "wead") },
+			"should replace 'the ' with 'da ' and 'th' with 'd'",
+		},
+		{
+			"replaces love with wuv",
+			"I love you",
+			func(s string) bool { return strings.Contains(s, "wuv") },
+			"should replace love with wuv",
+		},
+		{
+			"appends kaomoji",
+			"hello world",
+			func(s string) bool {
+				faces := []string{"uwu", "owo", ">w<", "^w^", "◕ᴗ◕✿", "✧w✧", "~nyaa"}
+				for _, f := range faces {
+					if strings.HasSuffix(s, f) {
+						return true
+					}
+				}
+				return false
+			},
+			"should end with a kaomoji",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := uwuify(tt.input)
+			if !tt.check(got) {
+				t.Errorf("uwuify(%q) = %q: %s", tt.input, got, tt.desc)
+			}
+		})
+	}
+}
+
+func TestQueryTopYappers(t *testing.T) {
+	db, err := sql.Open("sqlite3", ":memory:")
+	if err != nil {
+		t.Fatalf("open db: %v", err)
+	}
+	defer db.Close()
+
+	_, err = db.Exec(`CREATE TABLE IF NOT EXISTS messages (
+		id TEXT PRIMARY KEY,
+		room_id TEXT,
+		sender TEXT,
+		ts_ms INTEGER,
+		body TEXT,
+		msgtype TEXT,
+		raw_json TEXT
+	)`)
+	if err != nil {
+		t.Fatalf("create table: %v", err)
+	}
+
+	now := time.Now().UnixMilli()
+	room := "!testroom:example.com"
+
+	// Insert test messages: alice=5, bob=3, carol=1, plus some bot messages that should be excluded.
+	for i := 0; i < 5; i++ {
+		_, _ = db.Exec(`INSERT INTO messages(id, room_id, sender, ts_ms, body, msgtype) VALUES (?, ?, ?, ?, ?, ?)`,
+			fmt.Sprintf("alice-%d", i), room, "@alice:example.com", now-int64(i*1000), fmt.Sprintf("hello %d", i), "m.text")
+	}
+	for i := 0; i < 3; i++ {
+		_, _ = db.Exec(`INSERT INTO messages(id, room_id, sender, ts_ms, body, msgtype) VALUES (?, ?, ?, ?, ?, ?)`,
+			fmt.Sprintf("bob-%d", i), room, "@bob:example.com", now-int64(i*1000), fmt.Sprintf("hey %d", i), "m.text")
+	}
+	_, _ = db.Exec(`INSERT INTO messages(id, room_id, sender, ts_ms, body, msgtype) VALUES (?, ?, ?, ?, ?, ?)`,
+		"carol-0", room, "@carol:example.com", now, "sup", "m.text")
+
+	// Bot messages — should be excluded.
+	_, _ = db.Exec(`INSERT INTO messages(id, room_id, sender, ts_ms, body, msgtype) VALUES (?, ?, ?, ?, ?, ?)`,
+		"bot-1", room, "@bot:example.com", now, "[BOT] hello", "m.text")
+	_, _ = db.Exec(`INSERT INTO messages(id, room_id, sender, ts_ms, body, msgtype) VALUES (?, ?, ?, ?, ?, ?)`,
+		"bot-2", room, "@bot:example.com", now, "/bot help", "m.text")
+
+	// Old message — should be excluded (>24h ago).
+	_, _ = db.Exec(`INSERT INTO messages(id, room_id, sender, ts_ms, body, msgtype) VALUES (?, ?, ?, ?, ?, ?)`,
+		"old-1", room, "@old:example.com", now-100000000, "ancient msg", "m.text")
+
+	// Different room — should be excluded.
+	_, _ = db.Exec(`INSERT INTO messages(id, room_id, sender, ts_ms, body, msgtype) VALUES (?, ?, ?, ?, ?, ?)`,
+		"other-1", "!otherroom:example.com", "@other:example.com", now, "wrong room", "m.text")
+
+	ev := &event.Event{
+		RoomID: id.RoomID(room),
+	}
+
+	ctx := context.Background()
+
+	// Test default (top 5).
+	result, err := queryTopYappers(ctx, db, nil, ev, "")
+	if err != nil {
+		t.Fatalf("queryTopYappers: %v", err)
+	}
+	if !strings.Contains(result, "alice") {
+		t.Errorf("expected alice in result, got: %s", result)
+	}
+	if !strings.Contains(result, "5 msgs") {
+		t.Errorf("expected '5 msgs' for alice, got: %s", result)
+	}
+	if !strings.Contains(result, "bob") {
+		t.Errorf("expected bob in result, got: %s", result)
+	}
+	// alice should be ranked #1.
+	if !strings.Contains(result, "1. alice") {
+		t.Errorf("expected alice at rank 1, got: %s", result)
+	}
+
+	// Test with limit.
+	result2, err := queryTopYappers(ctx, db, nil, ev, "2")
+	if err != nil {
+		t.Fatalf("queryTopYappers with limit: %v", err)
+	}
+	lines := strings.Split(strings.TrimSpace(result2), "\n")
+	// Header line + 2 results.
+	if len(lines) != 3 {
+		t.Errorf("expected 3 lines (header + 2 results), got %d: %s", len(lines), result2)
+	}
+
+	// Bot and old messages should not appear.
+	if strings.Contains(result, "bot") {
+		t.Errorf("bot messages should be excluded, got: %s", result)
+	}
+	if strings.Contains(result, "old") {
+		t.Errorf("old messages should be excluded, got: %s", result)
+	}
+	if strings.Contains(result, "other") {
+		t.Errorf("messages from other rooms should be excluded, got: %s", result)
+	}
+}