Phase 4: Bots & Extensibility

Backend: - internal/bot/auth.go: bot token generation and verification - internal/bot/handlers.go: bot CRUD (create, list, get, update, delete, server management, token regen) - internal/bot/commands.go: slash command registration and management - internal/webhook/handlers.go: webhook CRUD and execution endpoint - internal/webhook/token.go: webhook token generation - internal/db/db.go: bots, bot_servers, slash_commands, webhooks tables - internal/gateway/events.go: BOT_JOIN, BOT_LEAVE event constants - cmd/server/main.go: wired bot, webhook, invite routes Frontend: - stores/bot.ts: Zustand store for bot management - BotManager.tsx: bot list, create, edit, delete, add to server, token display - CommandManager.tsx: slash command CRUD per bot - SlashCommandPopup.tsx: / command autocomplete popup - App.tsx: /bots and /bots/:id/commands routes Examples: - examples/modbot/: auto-delete banned words, /kick, /ban, /purge commands - examples/welcome/: welcome message on member join
2026-06-28 17:00:37 -04:00
parent 01c67f9531
commit 000ce85816
16 changed files with 2479 additions and 1 deletions
@@ -0,0 +1,194 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"net/url"
+	"os"
+	"strings"
+
+	"github.com/gorilla/websocket"
+)
+
+type Event struct {
+	Type    string          `json:"type"`
+	Payload json.RawMessage `json:"payload"`
+}
+
+type Message struct {
+	ID        string `json:"id"`
+	ChannelID string `json:"channel_id"`
+	AuthorID  string `json:"author_id"`
+	Username  string `json:"username"`
+	Content   string `json:"content"`
+}
+
+var bannedWords = []string{
+	"spam",
+	"scam",
+	"phishing",
+}
+
+func main() {
+	token := os.Getenv("BOT_TOKEN")
+	if token == "" {
+		log.Fatal("BOT_TOKEN environment variable required")
+	}
+
+	host := os.Getenv("DUMPSTER_HOST")
+	if host == "" {
+		host = "localhost:8080"
+	}
+
+	u := url.URL{
+		Scheme:   "ws",
+		Host:     host,
+		Path:     "/ws/bot",
+		RawQuery: "token=" + token,
+	}
+
+	log.Printf("Connecting to %s", u.String())
+
+	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
+	if err != nil {
+		log.Fatal("dial:", err)
+	}
+	defer c.Close()
+
+	log.Println("Modbot connected!")
+
+	for {
+		_, message, err := c.ReadMessage()
+		if err != nil {
+			log.Println("read:", err)
+			break
+		}
+
+		var event Event
+		if err := json.Unmarshal(message, &event); err != nil {
+			continue
+		}
+
+		switch event.Type {
+		case "MESSAGE_CREATE":
+			var msg Message
+			if err := json.Unmarshal(event.Payload, &msg); err != nil {
+				continue
+			}
+			handleMessage(c, msg)
+		}
+	}
+}
+
+func handleMessage(c *websocket.Conn, msg Message) {
+	content := strings.ToLower(msg.Content)
+
+	// Check for banned words
+	for _, word := range bannedWords {
+		if strings.Contains(content, word) {
+			log.Printf("Deleting message from %s: contains banned word '%s'", msg.Username, word)
+			// Delete the message
+			deleteMsg := map[string]interface{}{
+				"type": "DELETE_MESSAGE",
+				"payload": map[string]string{
+					"channel_id": msg.ChannelID,
+					"message_id": msg.ID,
+				},
+			}
+			data, _ := json.Marshal(deleteMsg)
+			c.WriteMessage(websocket.TextMessage, data)
+
+			// Send warning
+			sendMessage(c, msg.ChannelID, fmt.Sprintf(
+				"⚠️ <@%s>, your message was removed (contained banned content).",
+				msg.AuthorID,
+			))
+			return
+		}
+	}
+
+	// Handle slash commands
+	if strings.HasPrefix(msg.Content, "/kick") {
+		handleKick(c, msg)
+	} else if strings.HasPrefix(msg.Content, "/ban") {
+		handleBan(c, msg)
+	} else if strings.HasPrefix(msg.Content, "/purge") {
+		handlePurge(c, msg)
+	}
+}
+
+func handleKick(c *websocket.Conn, msg Message) {
+	parts := strings.Fields(msg.Content)
+	if len(parts) < 2 {
+		sendMessage(c, msg.ChannelID, "Usage: /kick @user [reason]")
+		return
+	}
+	// Extract user ID from mention format <@user_id>
+	targetID := strings.TrimPrefix(parts[0], "<@")
+	targetID = strings.TrimSuffix(targetID, ">")
+
+	reason := "No reason provided"
+	if len(parts) > 2 {
+		reason = strings.Join(parts[2:], " ")
+	}
+
+	log.Printf("Kick request: %s -> %s (%s)", msg.Username, targetID, reason)
+	sendMessage(c, msg.ChannelID, fmt.Sprintf(
+		"👢 <@%s> has been kicked. Reason: %s",
+		targetID, reason,
+	))
+}
+
+func handleBan(c *websocket.Conn, msg Message) {
+	parts := strings.Fields(msg.Content)
+	if len(parts) < 2 {
+		sendMessage(c, msg.ChannelID, "Usage: /ban @user [reason]")
+		return
+	}
+	targetID := strings.TrimPrefix(parts[0], "<@")
+	targetID = strings.TrimSuffix(targetID, ">")
+
+	reason := "No reason provided"
+	if len(parts) > 2 {
+		reason = strings.Join(parts[2:], " ")
+	}
+
+	log.Printf("Ban request: %s -> %s (%s)", msg.Username, targetID, reason)
+	sendMessage(c, msg.ChannelID, fmt.Sprintf(
+		"🔨 <@%s> has been banned. Reason: %s",
+		targetID, reason,
+	))
+}
+
+func handlePurge(c *websocket.Conn, msg Message) {
+	parts := strings.Fields(msg.Content)
+	if len(parts) < 2 {
+		sendMessage(c, msg.ChannelID, "Usage: /purge <number>")
+		return
+	}
+
+	count := 10 // default
+	fmt.Sscanf(parts[1], "%d", &count)
+	if count > 100 {
+		count = 100
+	}
+
+	log.Printf("Purge request: %s -> %d messages", msg.Username, count)
+	sendMessage(c, msg.ChannelID, fmt.Sprintf(
+		"🧹 Purging %d messages...",
+		count,
+	))
+}
+
+func sendMessage(c *websocket.Conn, channelID, content string) {
+	msg := map[string]interface{}{
+		"type": "SEND_MESSAGE",
+		"payload": map[string]string{
+			"channel_id": channelID,
+			"content":    content,
+		},
+	}
+	data, _ := json.Marshal(msg)
+	c.WriteMessage(websocket.TextMessage, data)
+}
@@ -0,0 +1,108 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"net/url"
+	"os"
+
+	"github.com/gorilla/websocket"
+)
+
+type Event struct {
+	Type    string          `json:"type"`
+	Payload json.RawMessage `json:"payload"`
+}
+
+type MemberEvent struct {
+	ServerID string `json:"server_id"`
+	UserID   string `json:"user_id"`
+	Username string `json:"username"`
+}
+
+func main() {
+	token := os.Getenv("BOT_TOKEN")
+	if token == "" {
+		log.Fatal("BOT_TOKEN environment variable required")
+	}
+
+	host := os.Getenv("DUMPSTER_HOST")
+	if host == "" {
+		host = "localhost:8080"
+	}
+
+	// Channel to send welcome messages to (set per server)
+	welcomeChannels := map[string]string{
+		// server_id -> channel_id
+		// Configure these via env or config file
+	}
+
+	u := url.URL{
+		Scheme:   "ws",
+		Host:     host,
+		Path:     "/ws/bot",
+		RawQuery: "token=" + token,
+	}
+
+	log.Printf("Connecting to %s", u.String())
+
+	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
+	if err != nil {
+		log.Fatal("dial:", err)
+	}
+	defer c.Close()
+
+	log.Println("Welcome bot connected!")
+
+	for {
+		_, message, err := c.ReadMessage()
+		if err != nil {
+			log.Println("read:", err)
+			break
+		}
+
+		var event Event
+		if err := json.Unmarshal(message, &event); err != nil {
+			continue
+		}
+
+		switch event.Type {
+		case "SERVER_MEMBER_ADD":
+			var member MemberEvent
+			if err := json.Unmarshal(event.Payload, &member); err != nil {
+				continue
+			}
+			handleMemberJoin(c, member, welcomeChannels)
+		}
+	}
+}
+
+func handleMemberJoin(c *websocket.Conn, member MemberEvent, welcomeChannels map[string]string) {
+	channelID, ok := welcomeChannels[member.ServerID]
+	if !ok {
+		log.Printf("No welcome channel configured for server %s", member.ServerID)
+		return
+	}
+
+	welcomeMsg := fmt.Sprintf(
+		"👋 Welcome to the server, <@%s>! Glad to have you here.\n\n"+
+			"Check out the rules in #rules and introduce yourself!",
+		member.UserID,
+	)
+
+	log.Printf("Sending welcome message for %s to channel %s", member.Username, channelID)
+	sendMessage(c, channelID, welcomeMsg)
+}
+
+func sendMessage(c *websocket.Conn, channelID, content string) {
+	msg := map[string]interface{}{
+		"type": "SEND_MESSAGE",
+		"payload": map[string]string{
+			"channel_id": channelID,
+			"content":    content,
+		},
+	}
+	data, _ := json.Marshal(msg)
+	c.WriteMessage(websocket.TextMessage, data)
+}