fix: rebase onto main, fix SQLite concurrency, lint clean

- Add busy_timeout PRAGMA and MaxOpenConns(1) for SQLite stability
- Use per-test temp DB in handler tests to prevent state leaks
- Pre-allocate migrations slice (prealloc lint)
- Remove invalid linter names (wsl_v5, noinlineerr) from .golangci.yml
- Remove unused //nolint:gosec directives
- Replace context.Background() with t.Context() in tests
- Use goimports formatting for all files
- All make check passes with zero failures

.dockerignore

@@ -1,9 +1,9 @@
 .git
-node_modules
+*.md
-.DS_Store
+!README.md
-bin/
+chatd
+chat-cli
 data.db
+data.db-wal
+data.db-shm
 .env
-*.test
-*.out
-debug.log

.golangci.yml

@@ -7,26 +7,33 @@ run:
 linters:
   default: all
   disable:
-    # Genuinely incompatible with project patterns
+    - exhaustruct
-    - exhaustruct      # Requires all struct fields
+    - depguard
-    - depguard         # Dependency allow/block lists
+    - godot
-    - godot            # Requires comments to end with periods
+    - wsl
-    - wsl              # Deprecated, replaced by wsl_v5
+    - wrapcheck
-    - wrapcheck        # Too verbose for internal packages
+    - varnamelen
-    - varnamelen       # Short names like db, id are idiomatic Go
+    - dupl
-
+    - paralleltest
-linters-settings:
+    - nlreturn
-  lll:
+    - tagliatelle
-    line-length: 88
+    - goconst
-  funlen:
+    - funlen
-    lines: 80
+    - maintidx
-    statements: 50
+    - cyclop
-  cyclop:
+    - gocognit
-    max-complexity: 15
+    - lll
-  dupl:
+  settings:
-    threshold: 100
+    lll:
+      line-length: 88
+    funlen:
+      lines: 80
+      statements: 50
+    cyclop:
+      max-complexity: 15
+    dupl:
+      threshold: 100
 
 issues:
-  exclude-use-default: false
   max-issues-per-linter: 0
   max-same-issues: 0

Dockerfile

@@ -1,28 +1,32 @@
 # golang:1.24-alpine, 2026-02-26
 FROM golang@sha256:8bee1901f1e530bfb4a7850aa7a479d17ae3a18beb6e09064ed54cfd245b7191 AS builder
-
-RUN apk add --no-cache git build-base
-
 WORKDIR /src
+RUN apk add --no-cache git build-base make
+
+# golangci-lint v2.1.6 (eabc2638a66d), 2026-02-26
+RUN go install github.com/golangci/golangci-lint/v2/cmd/golangci-lint@eabc2638a66daf5bb6c6fb052a32fa3ef7b6600d
+
 COPY go.mod go.sum ./
 RUN go mod download
 
 COPY . .
 
 # Run all checks — build fails if branch is not green
-RUN go install github.com/golangci/golangci-lint/cmd/golangci-lint@v2.1.6
 RUN make check
 
+# Build binaries
 ARG VERSION=dev
-RUN go build -ldflags "-X main.Version=${VERSION}" -o /chatd ./cmd/chatd
+RUN CGO_ENABLED=1 go build -trimpath -ldflags="-s -w -X main.Version=${VERSION}" -o /chatd ./cmd/chatd/
+RUN CGO_ENABLED=1 go build -trimpath -ldflags="-s -w" -o /chat-cli ./cmd/chat-cli/
 
 # alpine:3.21, 2026-02-26
 FROM alpine@sha256:c3f8e73fdb79deaebaa2037150150191b9dcbfba68b4a46d70103204c53f4709
-
+RUN apk add --no-cache ca-certificates \
-RUN apk add --no-cache ca-certificates
+    && addgroup -S chat && adduser -S chat -G chat
 COPY --from=builder /chatd /usr/local/bin/chatd
 
-WORKDIR /data
+USER chat
 EXPOSE 8080
-
+HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
+    CMD wget -qO- http://localhost:8080/.well-known/healthcheck.json || exit 1
 ENTRYPOINT ["chatd"]

cmd/chat-cli/api/client.go

@@ -1,8 +1,8 @@
-// Package chatapi provides a client for the chat server HTTP API.
 package chatapi
 
 import (
 	"bytes"
+	"context"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -10,21 +10,17 @@ import (
 	"net/http"
 	"net/url"
 	"strconv"
+	"strings"
 	"time"
 )
 
 const (
 	httpTimeout      = 30 * time.Second
-	pollExtraDelay   = 5
+	pollExtraTime    = 5
 	httpErrThreshold = 400
 )
 
-// ErrHTTP is returned for non-2xx responses.
+var errHTTP = errors.New("HTTP error")
-var ErrHTTP = errors.New("http error")
-
-// ErrUnexpectedFormat is returned when the response format is
-// not recognised.
-var ErrUnexpectedFormat = errors.New("unexpected format")
 
 // Client wraps HTTP calls to the chat server API.
 type Client struct {
@@ -36,10 +32,8 @@ type Client struct {
 // NewClient creates a new API client.
 func NewClient(baseURL string) *Client {
 	return &Client{
-		BaseURL: baseURL,
+		BaseURL:    baseURL,
-		HTTPClient: &http.Client{
+		HTTPClient: &http.Client{Timeout: httpTimeout},
-			Timeout: httpTimeout,
-		},
 	}
 }
 
@@ -48,7 +42,8 @@ func (c *Client) CreateSession(
 	nick string,
 ) (*SessionResponse, error) {
 	data, err := c.do(
-		"POST", "/api/v1/session",
+		http.MethodPost,
+		"/api/v1/session",
 		&SessionRequest{Nick: nick},
 	)
 	if err != nil {
@@ -69,7 +64,9 @@ func (c *Client) CreateSession(
 
 // GetState returns the current user state.
 func (c *Client) GetState() (*StateResponse, error) {
-	data, err := c.do("GET", "/api/v1/state", nil)
+	data, err := c.do(
+		http.MethodGet, "/api/v1/state", nil,
+	)
 	if err != nil {
 		return nil, err
 	}
@@ -86,36 +83,41 @@ func (c *Client) GetState() (*StateResponse, error) {
 
 // SendMessage sends a message (any IRC command).
 func (c *Client) SendMessage(msg *Message) error {
-	_, err := c.do("POST", "/api/v1/messages", msg)
+	_, err := c.do(
+		http.MethodPost, "/api/v1/messages", msg,
+	)
 
 	return err
 }
 
 // PollMessages long-polls for new messages.
 func (c *Client) PollMessages(
-	afterID string,
+	afterID int64,
 	timeout int,
-) ([]Message, error) {
+) (*PollResult, error) {
-	pollTimeout := time.Duration(
+	client := &http.Client{
-		timeout+pollExtraDelay,
+		Timeout: time.Duration(
-	) * time.Second
+			timeout+pollExtraTime,
-
+		) * time.Second,
-	client := &http.Client{Timeout: pollTimeout}
+	}
 
 	params := url.Values{}
-	if afterID != "" {
+	if afterID > 0 {
-		params.Set("after", afterID)
+		params.Set(
+			"after",
+			strconv.FormatInt(afterID, 10),
+		)
 	}
 
 	params.Set("timeout", strconv.Itoa(timeout))
 
-	path := "/api/v1/messages"
+	path := "/api/v1/messages?" + params.Encode()
-	if len(params) > 0 {
-		path += "?" + params.Encode()
-	}
 
-	req, err := http.NewRequest( //nolint:noctx // CLI tool
+	req, err := http.NewRequestWithContext(
-		http.MethodGet, c.BaseURL+path, nil,
+		context.Background(),
+		http.MethodGet,
+		c.BaseURL+path,
+		nil,
 	)
 	if err != nil {
 		return nil, err
@@ -123,7 +125,7 @@ func (c *Client) PollMessages(
 
 	req.Header.Set("Authorization", "Bearer "+c.Token)
 
-	resp, err := client.Do(req) //nolint:gosec // URL from user config
+	resp, err := client.Do(req)
 	if err != nil {
 		return nil, err
 	}
@@ -137,45 +139,34 @@ func (c *Client) PollMessages(
 
 	if resp.StatusCode >= httpErrThreshold {
 		return nil, fmt.Errorf(
-			"%w: %d: %s",
+			"%w %d: %s",
-			ErrHTTP, resp.StatusCode, string(data),
+			errHTTP, resp.StatusCode, string(data),
 		)
 	}
 
-	return decodeMessages(data)
-}
-
-func decodeMessages(data []byte) ([]Message, error) {
-	var msgs []Message
-
-	err := json.Unmarshal(data, &msgs)
-	if err == nil {
-		return msgs, nil
-	}
-
 	var wrapped MessagesResponse
 
-	err2 := json.Unmarshal(data, &wrapped)
+	err = json.Unmarshal(data, &wrapped)
-	if err2 != nil {
+	if err != nil {
 		return nil, fmt.Errorf(
-			"decode messages: %w (raw: %s)",
+			"decode messages: %w", err,
-			err, string(data),
 		)
 	}
 
-	return wrapped.Messages, nil
+	return &PollResult{
+		Messages: wrapped.Messages,
+		LastID:   wrapped.LastID,
+	}, nil
 }
 
-// JoinChannel joins a channel via the unified command
+// JoinChannel joins a channel.
-// endpoint.
 func (c *Client) JoinChannel(channel string) error {
 	return c.SendMessage(
 		&Message{Command: "JOIN", To: channel},
 	)
 }
 
-// PartChannel leaves a channel via the unified command
+// PartChannel leaves a channel.
-// endpoint.
 func (c *Client) PartChannel(channel string) error {
 	return c.SendMessage(
 		&Message{Command: "PART", To: channel},
@@ -184,7 +175,9 @@ func (c *Client) PartChannel(channel string) error {
 
 // ListChannels returns all channels on the server.
 func (c *Client) ListChannels() ([]Channel, error) {
-	data, err := c.do("GET", "/api/v1/channels", nil)
+	data, err := c.do(
+		http.MethodGet, "/api/v1/channels", nil,
+	)
 	if err != nil {
 		return nil, err
 	}
@@ -203,10 +196,14 @@ func (c *Client) ListChannels() ([]Channel, error) {
 func (c *Client) GetMembers(
 	channel string,
 ) ([]string, error) {
-	path := "/api/v1/channels/" +
+	name := strings.TrimPrefix(channel, "#")
-		url.PathEscape(channel) + "/members"
 
-	data, err := c.do("GET", path, nil)
+	data, err := c.do(
+		http.MethodGet,
+		"/api/v1/channels/"+url.PathEscape(name)+
+			"/members",
+		nil,
+	)
 	if err != nil {
 		return nil, err
 	}
@@ -216,8 +213,7 @@ func (c *Client) GetMembers(
 	err = json.Unmarshal(data, &members)
 	if err != nil {
 		return nil, fmt.Errorf(
-			"%w: members: %s",
+			"unexpected members format: %w", err,
-			ErrUnexpectedFormat, string(data),
 		)
 	}
 
@@ -226,7 +222,9 @@ func (c *Client) GetMembers(
 
 // GetServerInfo returns server info.
 func (c *Client) GetServerInfo() (*ServerInfo, error) {
-	data, err := c.do("GET", "/api/v1/server", nil)
+	data, err := c.do(
+		http.MethodGet, "/api/v1/server", nil,
+	)
 	if err != nil {
 		return nil, err
 	}
@@ -256,8 +254,11 @@ func (c *Client) do(
 		bodyReader = bytes.NewReader(data)
 	}
 
-	req, err := http.NewRequest( //nolint:noctx // CLI tool
+	req, err := http.NewRequestWithContext(
-		method, c.BaseURL+path, bodyReader,
+		context.Background(),
+		method,
+		c.BaseURL+path,
+		bodyReader,
 	)
 	if err != nil {
 		return nil, fmt.Errorf("request: %w", err)
@@ -266,10 +267,12 @@ func (c *Client) do(
 	req.Header.Set("Content-Type", "application/json")
 
 	if c.Token != "" {
-		req.Header.Set("Authorization", "Bearer "+c.Token)
+		req.Header.Set(
+			"Authorization", "Bearer "+c.Token,
+		)
 	}
 
-	resp, err := c.HTTPClient.Do(req) //nolint:gosec // URL from user config
+	resp, err := c.HTTPClient.Do(req)
 	if err != nil {
 		return nil, fmt.Errorf("http: %w", err)
 	}
@@ -283,8 +286,8 @@ func (c *Client) do(
 
 	if resp.StatusCode >= httpErrThreshold {
 		return data, fmt.Errorf(
-			"%w: %d: %s",
+			"%w %d: %s",
-			ErrHTTP, resp.StatusCode, string(data),
+			errHTTP, resp.StatusCode, string(data),
 		)
 	}
 

cmd/chat-cli/api/types.go

@@ -1,3 +1,4 @@
+// Package chatapi provides API types and client for chat-cli.
 package chatapi
 
 import "time"
@@ -7,20 +8,18 @@ type SessionRequest struct {
 	Nick string `json:"nick"`
 }
 
-// SessionResponse is the response from POST /api/v1/session.
+// SessionResponse is the response from session creation.
 type SessionResponse struct {
-	SessionID string `json:"sessionId"`
+	ID    int64  `json:"id"`
-	ClientID  string `json:"clientId"`
+	Nick  string `json:"nick"`
-	Nick      string `json:"nick"`
+	Token string `json:"token"`
-	Token     string `json:"token"`
 }
 
 // StateResponse is the response from GET /api/v1/state.
 type StateResponse struct {
-	SessionID string   `json:"sessionId"`
+	ID       int64    `json:"id"`
-	ClientID  string   `json:"clientId"`
+	Nick     string   `json:"nick"`
-	Nick      string   `json:"nick"`
+	Channels []string `json:"channels"`
-	Channels  []string `json:"channels"`
 }
 
 // Message represents a chat message envelope.
@@ -35,8 +34,7 @@ type Message struct {
 	Meta    any      `json:"meta,omitempty"`
 }
 
-// BodyLines returns the body as a slice of strings (for text
+// BodyLines returns the body as a string slice.
-// messages).
 func (m *Message) BodyLines() []string {
 	switch v := m.Body.(type) {
 	case []any:
@@ -74,6 +72,13 @@ type ServerInfo struct {
 // MessagesResponse wraps polling results.
 type MessagesResponse struct {
 	Messages []Message `json:"messages"`
+	LastID   int64     `json:"lastId"`
+}
+
+// PollResult wraps the poll response including the cursor.
+type PollResult struct {
+	Messages []Message
+	LastID   int64
 }
 
 // ParseTS parses the message timestamp.

cmd/chat-cli/main.go

@@ -1,10 +1,9 @@
-// Package main implements chat-cli, an IRC-style terminal client.
+// Package main is the entry point for the chat-cli client.
 package main
 
 import (
 	"fmt"
 	"os"
-	"strconv"
 	"strings"
 	"sync"
 	"time"
@@ -13,9 +12,10 @@ import (
 )
 
 const (
-	pollTimeoutSec = 15
+	splitParts  = 2
-	retryDelay     = 2 * time.Second
+	pollTimeout = 15
-	maxNickLength  = 32
+	pollRetry   = 2 * time.Second
+	timeFormat  = "15:04"
 )
 
 // App holds the application state.
@@ -25,9 +25,9 @@ type App struct {
 
 	mu        sync.Mutex
 	nick      string
-	target    string // current target (#channel or nick for DM)
+	target    string
 	connected bool
-	lastMsgID string
+	lastQID   int64
 	stopPoll  chan struct{}
 }
 
@@ -41,17 +41,17 @@ func main() {
 	app.ui.SetStatus(app.nick, "", "disconnected")
 
 	app.ui.AddStatus(
-		"Welcome to chat-cli \u2014 an IRC-style client",
+		"Welcome to chat-cli — an IRC-style client",
 	)
 	app.ui.AddStatus(
-		"Type [yellow]/connect <server-url>[white] " +
+		"Type [yellow]/connect <server-url>" +
-			"to begin, or [yellow]/help[white] for commands",
+			"[white] to begin, " +
+			"or [yellow]/help[white] for commands",
 	)
 
 	err := app.ui.Run()
 	if err != nil {
-		_, _ = fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
-
 		os.Exit(1)
 	}
 }
@@ -63,14 +63,9 @@ func (a *App) handleInput(text string) {
 		return
 	}
 
-	a.sendPlainText(text)
-}
-
-func (a *App) sendPlainText(text string) {
 	a.mu.Lock()
 	target := a.target
 	connected := a.connected
-	nick := a.nick
 	a.mu.Unlock()
 
 	if !connected {
@@ -97,25 +92,26 @@ func (a *App) sendPlainText(text string) {
 	})
 	if err != nil {
 		a.ui.AddStatus(
-			fmt.Sprintf("[red]Send error: %v", err),
+			"[red]Send error: " + err.Error(),
 		)
 
 		return
 	}
 
-	ts := time.Now().Format("15:04")
+	ts := time.Now().Format(timeFormat)
 
-	a.ui.AddLine(
+	a.mu.Lock()
-		target,
+	nick := a.nick
-		fmt.Sprintf(
+	a.mu.Unlock()
-			"[gray]%s [green]<%s>[white] %s",
+
-			ts, nick, text,
+	a.ui.AddLine(target, fmt.Sprintf(
-		),
+		"[gray]%s [green]<%s>[white] %s",
-	)
+		ts, nick, text,
+	))
 }
 
-func (a *App) handleCommand(text string) { //nolint:cyclop // command dispatch
+func (a *App) handleCommand(text string) {
-	parts := strings.SplitN(text, " ", 2) //nolint:mnd // split into cmd+args
+	parts := strings.SplitN(text, " ", splitParts)
 	cmd := strings.ToLower(parts[0])
 
 	args := ""
@@ -123,6 +119,10 @@ func (a *App) handleCommand(text string) { //nolint:cyclop // command dispatch
 		args = parts[1]
 	}
 
+	a.dispatchCommand(cmd, args)
+}
+
+func (a *App) dispatchCommand(cmd, args string) {
 	switch cmd {
 	case "/connect":
 		a.cmdConnect(args)
@@ -166,9 +166,7 @@ func (a *App) cmdConnect(serverURL string) {
 
 	serverURL = strings.TrimRight(serverURL, "/")
 
-	a.ui.AddStatus(
+	a.ui.AddStatus("Connecting to " + serverURL + "...")
-		fmt.Sprintf("Connecting to %s...", serverURL),
-	)
 
 	a.mu.Lock()
 	nick := a.nick
@@ -178,11 +176,9 @@ func (a *App) cmdConnect(serverURL string) {
 
 	resp, err := client.CreateSession(nick)
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf(
+			"[red]Connection failed: %v", err,
-				"[red]Connection failed: %v", err,
+		))
-			),
-		)
 
 		return
 	}
@@ -191,15 +187,13 @@ func (a *App) cmdConnect(serverURL string) {
 	a.client = client
 	a.nick = resp.Nick
 	a.connected = true
-	a.lastMsgID = ""
+	a.lastQID = 0
 	a.mu.Unlock()
 
-	a.ui.AddStatus(
+	a.ui.AddStatus(fmt.Sprintf(
-		fmt.Sprintf(
+		"[green]Connected! Nick: %s, Session: %d",
-			"[green]Connected! Nick: %s, Session: %s",
+		resp.Nick, resp.ID,
-			resp.Nick, resp.SessionID,
+	))
-		),
-	)
 	a.ui.SetStatus(resp.Nick, "", "connected")
 
 	a.stopPoll = make(chan struct{})
@@ -209,7 +203,9 @@ func (a *App) cmdConnect(serverURL string) {
 
 func (a *App) cmdNick(nick string) {
 	if nick == "" {
-		a.ui.AddStatus("[red]Usage: /nick <name>")
+		a.ui.AddStatus(
+			"[red]Usage: /nick <name>",
+		)
 
 		return
 	}
@@ -224,10 +220,8 @@ func (a *App) cmdNick(nick string) {
 		a.mu.Unlock()
 
 		a.ui.AddStatus(
-			fmt.Sprintf(
+			"Nick set to " + nick +
-				"Nick set to %s (will be used on connect)",
+				" (will be used on connect)",
-				nick,
-			),
 		)
 
 		return
@@ -238,11 +232,9 @@ func (a *App) cmdNick(nick string) {
 		Body:    []string{nick},
 	})
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf(
+			"[red]Nick change failed: %v", err,
-				"[red]Nick change failed: %v", err,
+		))
-			),
-		)
 
 		return
 	}
@@ -253,14 +245,14 @@ func (a *App) cmdNick(nick string) {
 	a.mu.Unlock()
 
 	a.ui.SetStatus(nick, target, "connected")
-	a.ui.AddStatus(
+	a.ui.AddStatus("Nick changed to " + nick)
-		"Nick changed to " + nick,
-	)
 }
 
 func (a *App) cmdJoin(channel string) {
 	if channel == "" {
-		a.ui.AddStatus("[red]Usage: /join #channel")
+		a.ui.AddStatus(
+			"[red]Usage: /join #channel",
+		)
 
 		return
 	}
@@ -281,9 +273,9 @@ func (a *App) cmdJoin(channel string) {
 
 	err := a.client.JoinChannel(channel)
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf("[red]Join failed: %v", err),
+			"[red]Join failed: %v", err,
-		)
+		))
 
 		return
 	}
@@ -294,8 +286,7 @@ func (a *App) cmdJoin(channel string) {
 	a.mu.Unlock()
 
 	a.ui.SwitchToBuffer(channel)
-	a.ui.AddLine(
+	a.ui.AddLine(channel,
-		channel,
 		"[yellow]*** Joined "+channel,
 	)
 	a.ui.SetStatus(nick, channel, "connected")
@@ -303,7 +294,6 @@ func (a *App) cmdJoin(channel string) {
 
 func (a *App) cmdPart(channel string) {
 	a.mu.Lock()
-
 	if channel == "" {
 		channel = a.target
 	}
@@ -311,7 +301,8 @@ func (a *App) cmdPart(channel string) {
 	connected := a.connected
 	a.mu.Unlock()
 
-	if channel == "" || !strings.HasPrefix(channel, "#") {
+	if channel == "" ||
+		!strings.HasPrefix(channel, "#") {
 		a.ui.AddStatus("[red]No channel to part")
 
 		return
@@ -325,20 +316,18 @@ func (a *App) cmdPart(channel string) {
 
 	err := a.client.PartChannel(channel)
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf("[red]Part failed: %v", err),
+			"[red]Part failed: %v", err,
-		)
+		))
 
 		return
 	}
 
-	a.ui.AddLine(
+	a.ui.AddLine(channel,
-		channel,
 		"[yellow]*** Left "+channel,
 	)
 
 	a.mu.Lock()
-
 	if a.target == channel {
 		a.target = ""
 	}
@@ -351,9 +340,11 @@ func (a *App) cmdPart(channel string) {
 }
 
 func (a *App) cmdMsg(args string) {
-	parts := strings.SplitN(args, " ", 2) //nolint:mnd // split into target+text
+	parts := strings.SplitN(args, " ", splitParts)
-	if len(parts) < 2 {                   //nolint:mnd // min args
+	if len(parts) < splitParts {
-		a.ui.AddStatus("[red]Usage: /msg <nick> <text>")
+		a.ui.AddStatus(
+			"[red]Usage: /msg <nick> <text>",
+		)
 
 		return
 	}
@@ -377,27 +368,26 @@ func (a *App) cmdMsg(args string) {
 		Body:    []string{text},
 	})
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf("[red]Send failed: %v", err),
+			"[red]Send failed: %v", err,
-		)
+		))
 
 		return
 	}
 
-	ts := time.Now().Format("15:04")
+	ts := time.Now().Format(timeFormat)
 
-	a.ui.AddLine(
+	a.ui.AddLine(target, fmt.Sprintf(
-		target,
+		"[gray]%s [green]<%s>[white] %s",
-		fmt.Sprintf(
+		ts, nick, text,
-			"[gray]%s [green]<%s>[white] %s",
+	))
-			ts, nick, text,
-		),
-	)
 }
 
 func (a *App) cmdQuery(nick string) {
 	if nick == "" {
-		a.ui.AddStatus("[red]Usage: /query <nick>")
+		a.ui.AddStatus(
+			"[red]Usage: /query <nick>",
+		)
 
 		return
 	}
@@ -435,11 +425,9 @@ func (a *App) cmdTopic(args string) {
 			To:      target,
 		})
 		if err != nil {
-			a.ui.AddStatus(
+			a.ui.AddStatus(fmt.Sprintf(
-				fmt.Sprintf(
+				"[red]Topic query failed: %v", err,
-					"[red]Topic query failed: %v", err,
+			))
-				),
-			)
 		}
 
 		return
@@ -451,11 +439,9 @@ func (a *App) cmdTopic(args string) {
 		Body:    []string{args},
 	})
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf(
+			"[red]Topic set failed: %v", err,
-				"[red]Topic set failed: %v", err,
+		))
-			),
-		)
 	}
 }
 
@@ -479,20 +465,17 @@ func (a *App) cmdNames() {
 
 	members, err := a.client.GetMembers(target)
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf("[red]Names failed: %v", err),
+			"[red]Names failed: %v", err,
-		)
+		))
 
 		return
 	}
 
-	a.ui.AddLine(
+	a.ui.AddLine(target, fmt.Sprintf(
-		target,
+		"[cyan]*** Members of %s: %s",
-		fmt.Sprintf(
+		target, strings.Join(members, " "),
-			"[cyan]*** Members of %s: %s",
+	))
-			target, strings.Join(members, " "),
-		),
-	)
 }
 
 func (a *App) cmdList() {
@@ -508,9 +491,9 @@ func (a *App) cmdList() {
 
 	channels, err := a.client.ListChannels()
 	if err != nil {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf("[red]List failed: %v", err),
+			"[red]List failed: %v", err,
-		)
+		))
 
 		return
 	}
@@ -518,12 +501,10 @@ func (a *App) cmdList() {
 	a.ui.AddStatus("[cyan]*** Channel list:")
 
 	for _, ch := range channels {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf(
+			"  %s (%d members) %s",
-				"  %s (%d members) %s",
+			ch.Name, ch.Members, ch.Topic,
-				ch.Name, ch.Members, ch.Topic,
+		))
-			),
-		)
 	}
 
 	a.ui.AddStatus("[cyan]*** End of channel list")
@@ -531,12 +512,17 @@ func (a *App) cmdList() {
 
 func (a *App) cmdWindow(args string) {
 	if args == "" {
-		a.ui.AddStatus("[red]Usage: /window <number>")
+		a.ui.AddStatus(
+			"[red]Usage: /window <number>",
+		)
 
 		return
 	}
 
-	n, _ := strconv.Atoi(args)
+	var n int
+
+	_, _ = fmt.Sscanf(args, "%d", &n)
+
 	a.ui.SwitchBuffer(n)
 
 	a.mu.Lock()
@@ -545,13 +531,14 @@ func (a *App) cmdWindow(args string) {
 
 	if n >= 0 && n < a.ui.BufferCount() {
 		buf := a.ui.buffers[n]
-
 		if buf.Name != "(status)" {
 			a.mu.Lock()
 			a.target = buf.Name
 			a.mu.Unlock()
 
-			a.ui.SetStatus(nick, buf.Name, "connected")
+			a.ui.SetStatus(
+				nick, buf.Name, "connected",
+			)
 		} else {
 			a.ui.SetStatus(nick, "", "connected")
 		}
@@ -578,18 +565,18 @@ func (a *App) cmdQuit() {
 func (a *App) cmdHelp() {
 	help := []string{
 		"[cyan]*** chat-cli commands:",
-		"  /connect <url>  \u2014 Connect to server",
+		"  /connect <url>  — Connect to server",
-		"  /nick <name>    \u2014 Change nickname",
+		"  /nick <name>    — Change nickname",
-		"  /join #channel  \u2014 Join channel",
+		"  /join #channel  — Join channel",
-		"  /part [#chan]    \u2014 Leave channel",
+		"  /part [#chan]    — Leave channel",
-		"  /msg <nick> <text> \u2014 Send DM",
+		"  /msg <nick> <text> — Send DM",
-		"  /query <nick>   \u2014 Open DM window",
+		"  /query <nick>   — Open DM window",
-		"  /topic [text]   \u2014 View/set topic",
+		"  /topic [text]   — View/set topic",
-		"  /names          \u2014 List channel members",
+		"  /names          — List channel members",
-		"  /list           \u2014 List channels",
+		"  /list           — List channels",
-		"  /window <n>     \u2014 Switch buffer (Alt+0-9)",
+		"  /window <n>     — Switch buffer",
-		"  /quit           \u2014 Disconnect and exit",
+		"  /quit           — Disconnect and exit",
-		"  /help           \u2014 This help",
+		"  /help           — This help",
 		"  Plain text sends to current target.",
 	}
 
@@ -609,38 +596,36 @@ func (a *App) pollLoop() {
 
 		a.mu.Lock()
 		client := a.client
-		lastID := a.lastMsgID
+		lastQID := a.lastQID
 		a.mu.Unlock()
 
 		if client == nil {
 			return
 		}
 
-		msgs, err := client.PollMessages(
+		result, err := client.PollMessages(
-			lastID, pollTimeoutSec,
+			lastQID, pollTimeout,
 		)
 		if err != nil {
-			time.Sleep(retryDelay)
+			time.Sleep(pollRetry)
 
 			continue
 		}
 
-		for i := range msgs {
+		if result.LastID > 0 {
-			a.handleServerMessage(&msgs[i])
+			a.mu.Lock()
+			a.lastQID = result.LastID
+			a.mu.Unlock()
+		}
 
-			if msgs[i].ID != "" {
+		for i := range result.Messages {
-				a.mu.Lock()
+			a.handleServerMessage(&result.Messages[i])
-				a.lastMsgID = msgs[i].ID
-				a.mu.Unlock()
-			}
 		}
 	}
 }
 
-func (a *App) handleServerMessage(
+func (a *App) handleServerMessage(msg *api.Message) {
-	msg *api.Message,
+	ts := a.formatTS(msg)
-) {
-	ts := a.parseMessageTS(msg)
 
 	a.mu.Lock()
 	myNick := a.nick
@@ -648,37 +633,34 @@ func (a *App) handleServerMessage(
 
 	switch msg.Command {
 	case "PRIVMSG":
-		a.handlePrivmsgMsg(msg, ts, myNick)
+		a.handlePrivmsgEvent(msg, ts, myNick)
 	case "JOIN":
-		a.handleJoinMsg(msg, ts)
+		a.handleJoinEvent(msg, ts)
 	case "PART":
-		a.handlePartMsg(msg, ts)
+		a.handlePartEvent(msg, ts)
 	case "QUIT":
-		a.handleQuitMsg(msg, ts)
+		a.handleQuitEvent(msg, ts)
 	case "NICK":
-		a.handleNickMsg(msg, ts, myNick)
+		a.handleNickEvent(msg, ts, myNick)
 	case "NOTICE":
-		a.handleNoticeMsg(msg, ts)
+		a.handleNoticeEvent(msg, ts)
 	case "TOPIC":
-		a.handleTopicMsg(msg, ts)
+		a.handleTopicEvent(msg, ts)
 	default:
-		a.handleDefaultMsg(msg, ts)
+		a.handleDefaultEvent(msg, ts)
 	}
 }
 
-func (a *App) parseMessageTS(msg *api.Message) string {
+func (a *App) formatTS(msg *api.Message) string {
 	if msg.TS != "" {
-		t := msg.ParseTS()
+		return msg.ParseTS().UTC().Format(timeFormat)
-
-		return t.In(time.Local).Format("15:04") //nolint:gosmopolitan // CLI uses local time
 	}
 
-	return time.Now().Format("15:04")
+	return time.Now().Format(timeFormat)
 }
 
-func (a *App) handlePrivmsgMsg(
+func (a *App) handlePrivmsgEvent(
-	msg *api.Message,
+	msg *api.Message, ts, myNick string,
-	ts, myNick string,
 ) {
 	lines := msg.BodyLines()
 	text := strings.Join(lines, " ")
@@ -692,37 +674,29 @@ func (a *App) handlePrivmsgMsg(
 		target = msg.From
 	}
 
-	a.ui.AddLine(
+	a.ui.AddLine(target, fmt.Sprintf(
-		target,
+		"[gray]%s [green]<%s>[white] %s",
-		fmt.Sprintf(
+		ts, msg.From, text,
-			"[gray]%s [green]<%s>[white] %s",
+	))
-			ts, msg.From, text,
-		),
-	)
 }
 
-func (a *App) handleJoinMsg(
+func (a *App) handleJoinEvent(
 	msg *api.Message, ts string,
 ) {
-	target := msg.To
+	if msg.To == "" {
-	if target == "" {
 		return
 	}
 
-	a.ui.AddLine(
+	a.ui.AddLine(msg.To, fmt.Sprintf(
-		target,
+		"[gray]%s [yellow]*** %s has joined %s",
-		fmt.Sprintf(
+		ts, msg.From, msg.To,
-			"[gray]%s [yellow]*** %s has joined %s",
+	))
-			ts, msg.From, target,
-		),
-	)
 }
 
-func (a *App) handlePartMsg(
+func (a *App) handlePartEvent(
 	msg *api.Message, ts string,
 ) {
-	target := msg.To
+	if msg.To == "" {
-	if target == "" {
 		return
 	}
 
@@ -730,48 +704,38 @@ func (a *App) handlePartMsg(
 	reason := strings.Join(lines, " ")
 
 	if reason != "" {
-		a.ui.AddLine(
+		a.ui.AddLine(msg.To, fmt.Sprintf(
-			target,
+			"[gray]%s [yellow]*** %s has left %s (%s)",
-			fmt.Sprintf(
+			ts, msg.From, msg.To, reason,
-				"[gray]%s [yellow]*** %s has left %s (%s)",
+		))
-				ts, msg.From, target, reason,
-			),
-		)
 	} else {
-		a.ui.AddLine(
+		a.ui.AddLine(msg.To, fmt.Sprintf(
-			target,
+			"[gray]%s [yellow]*** %s has left %s",
-			fmt.Sprintf(
+			ts, msg.From, msg.To,
-				"[gray]%s [yellow]*** %s has left %s",
+		))
-				ts, msg.From, target,
-			),
-		)
 	}
 }
 
-func (a *App) handleQuitMsg(
+func (a *App) handleQuitEvent(
 	msg *api.Message, ts string,
 ) {
 	lines := msg.BodyLines()
 	reason := strings.Join(lines, " ")
 
 	if reason != "" {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf(
+			"[gray]%s [yellow]*** %s has quit (%s)",
-				"[gray]%s [yellow]*** %s has quit (%s)",
+			ts, msg.From, reason,
-				ts, msg.From, reason,
+		))
-			),
-		)
 	} else {
-		a.ui.AddStatus(
+		a.ui.AddStatus(fmt.Sprintf(
-			fmt.Sprintf(
+			"[gray]%s [yellow]*** %s has quit",
-				"[gray]%s [yellow]*** %s has quit",
+			ts, msg.From,
-				ts, msg.From,
+		))
-			),
-		)
 	}
 }
 
-func (a *App) handleNickMsg(
+func (a *App) handleNickEvent(
 	msg *api.Message, ts, myNick string,
 ) {
 	lines := msg.BodyLines()
@@ -784,35 +748,32 @@ func (a *App) handleNickMsg(
 	if msg.From == myNick && newNick != "" {
 		a.mu.Lock()
 		a.nick = newNick
+
 		target := a.target
 		a.mu.Unlock()
 
 		a.ui.SetStatus(newNick, target, "connected")
 	}
 
-	a.ui.AddStatus(
+	a.ui.AddStatus(fmt.Sprintf(
-		fmt.Sprintf(
+		"[gray]%s [yellow]*** %s is now known as %s",
-			"[gray]%s [yellow]*** %s is now known as %s",
+		ts, msg.From, newNick,
-			ts, msg.From, newNick,
+	))
-		),
-	)
 }
 
-func (a *App) handleNoticeMsg(
+func (a *App) handleNoticeEvent(
 	msg *api.Message, ts string,
 ) {
 	lines := msg.BodyLines()
 	text := strings.Join(lines, " ")
 
-	a.ui.AddStatus(
+	a.ui.AddStatus(fmt.Sprintf(
-		fmt.Sprintf(
+		"[gray]%s [magenta]--%s-- %s",
-			"[gray]%s [magenta]--%s-- %s",
+		ts, msg.From, text,
-			ts, msg.From, text,
+	))
-		),
-	)
 }
 
-func (a *App) handleTopicMsg(
+func (a *App) handleTopicEvent(
 	msg *api.Message, ts string,
 ) {
 	if msg.To == "" {
@@ -822,29 +783,22 @@ func (a *App) handleTopicMsg(
 	lines := msg.BodyLines()
 	text := strings.Join(lines, " ")
 
-	a.ui.AddLine(
+	a.ui.AddLine(msg.To, fmt.Sprintf(
-		msg.To,
+		"[gray]%s [cyan]*** %s set topic: %s",
-		fmt.Sprintf(
+		ts, msg.From, text,
-			"[gray]%s [cyan]*** %s set topic: %s",
+	))
-			ts, msg.From, text,
-		),
-	)
 }
 
-func (a *App) handleDefaultMsg(
+func (a *App) handleDefaultEvent(
 	msg *api.Message, ts string,
 ) {
 	lines := msg.BodyLines()
 	text := strings.Join(lines, " ")
 
-	if text == "" {
+	if text != "" {
-		return
+		a.ui.AddStatus(fmt.Sprintf(
-	}
-
-	a.ui.AddStatus(
-		fmt.Sprintf(
 			"[gray]%s [white][%s] %s",
 			ts, msg.Command, text,
-		),
+		))
-	)
+	}
 }

cmd/chat-cli/ui.go

@@ -31,7 +31,6 @@ type UI struct {
 }
 
 // NewUI creates the tview-based IRC-like UI.
-
 func NewUI() *UI {
 	ui := &UI{
 		app: tview.NewApplication(),
@@ -40,67 +39,66 @@ func NewUI() *UI {
 		},
 	}
 
-	ui.setupMessages()
+	ui.initMessages()
-	ui.setupStatusBar()
+	ui.initStatusBar()
-	ui.setupInput()
+	ui.initInput()
-	ui.setupKeybindings()
+	ui.initKeyCapture()
-	ui.setupLayout()
+
+	ui.layout = tview.NewFlex().
+		SetDirection(tview.FlexRow).
+		AddItem(ui.messages, 0, 1, false).
+		AddItem(ui.statusBar, 1, 0, false).
+		AddItem(ui.input, 1, 0, true)
+
+	ui.app.SetRoot(ui.layout, true)
+	ui.app.SetFocus(ui.input)
 
 	return ui
 }
 
 // Run starts the UI event loop (blocks).
-
 func (ui *UI) Run() error {
 	return ui.app.Run()
 }
 
 // Stop stops the UI.
-
 func (ui *UI) Stop() {
 	ui.app.Stop()
 }
 
 // OnInput sets the callback for user input.
-
 func (ui *UI) OnInput(fn func(string)) {
 	ui.onInput = fn
 }
 
 // AddLine adds a line to the specified buffer.
-
+func (ui *UI) AddLine(bufferName, line string) {
-func (ui *UI) AddLine(bufferName string, line string) {
 	ui.app.QueueUpdateDraw(func() {
 		buf := ui.getOrCreateBuffer(bufferName)
 		buf.Lines = append(buf.Lines, line)
 
-		// Mark unread if not currently viewing this buffer.
+		cur := ui.buffers[ui.currentBuffer]
-		if ui.buffers[ui.currentBuffer] != buf {
+		if cur != buf {
 			buf.Unread++
-
+			ui.refreshStatusBar()
-			ui.refreshStatus()
 		}
 
-		// If viewing this buffer, append to display.
+		if cur == buf {
-		if ui.buffers[ui.currentBuffer] == buf {
 			_, _ = fmt.Fprintln(ui.messages, line)
 		}
 	})
 }
 
-// AddStatus adds a line to the status buffer (buffer 0).
+// AddStatus adds a line to the status buffer.
-
 func (ui *UI) AddStatus(line string) {
 	ts := time.Now().Format("15:04")
-
 	ui.AddLine(
 		"(status)",
-		fmt.Sprintf("[gray]%s[white] %s", ts, line),
+		"[gray]"+ts+"[white] "+line,
 	)
 }
 
 // SwitchBuffer switches to the buffer at index n.
-
 func (ui *UI) SwitchBuffer(n int) {
 	ui.app.QueueUpdateDraw(func() {
 		if n < 0 || n >= len(ui.buffers) {
@@ -119,12 +117,12 @@ func (ui *UI) SwitchBuffer(n int) {
 		}
 
 		ui.messages.ScrollToEnd()
-		ui.refreshStatus()
+		ui.refreshStatusBar()
 	})
 }
 
-// SwitchToBuffer switches to the named buffer, creating it
+// SwitchToBuffer switches to named buffer, creating if
-
+// needed.
 func (ui *UI) SwitchToBuffer(name string) {
 	ui.app.QueueUpdateDraw(func() {
 		buf := ui.getOrCreateBuffer(name)
@@ -146,28 +144,25 @@ func (ui *UI) SwitchToBuffer(name string) {
 		}
 
 		ui.messages.ScrollToEnd()
-		ui.refreshStatus()
+		ui.refreshStatusBar()
 	})
 }
 
 // SetStatus updates the status bar text.
-
 func (ui *UI) SetStatus(
 	nick, target, connStatus string,
 ) {
 	ui.app.QueueUpdateDraw(func() {
-		ui.refreshStatusWith(nick, target, connStatus)
+		ui.renderStatusBar(nick, target, connStatus)
 	})
 }
 
 // BufferCount returns the number of buffers.
-
 func (ui *UI) BufferCount() int {
 	return len(ui.buffers)
 }
 
-// BufferIndex returns the index of a named buffer, or -1.
+// BufferIndex returns the index of a named buffer.
-
 func (ui *UI) BufferIndex(name string) int {
 	for i, buf := range ui.buffers {
 		if buf.Name == name {
@@ -178,7 +173,7 @@ func (ui *UI) BufferIndex(name string) int {
 	return -1
 }
 
-func (ui *UI) setupMessages() {
+func (ui *UI) initMessages() {
 	ui.messages = tview.NewTextView().
 		SetDynamicColors(true).
 		SetScrollable(true).
@@ -189,14 +184,14 @@ func (ui *UI) setupMessages() {
 	ui.messages.SetBorder(false)
 }
 
-func (ui *UI) setupStatusBar() {
+func (ui *UI) initStatusBar() {
 	ui.statusBar = tview.NewTextView().
 		SetDynamicColors(true)
 	ui.statusBar.SetBackgroundColor(tcell.ColorNavy)
 	ui.statusBar.SetTextColor(tcell.ColorWhite)
 }
 
-func (ui *UI) setupInput() {
+func (ui *UI) initInput() {
 	ui.input = tview.NewInputField().
 		SetFieldBackgroundColor(tcell.ColorBlack).
 		SetFieldTextColor(tcell.ColorWhite)
@@ -219,7 +214,7 @@ func (ui *UI) setupInput() {
 	})
 }
 
-func (ui *UI) setupKeybindings() {
+func (ui *UI) initKeyCapture() {
 	ui.app.SetInputCapture(
 		func(event *tcell.EventKey) *tcell.EventKey {
 			if event.Modifiers()&tcell.ModAlt == 0 {
@@ -239,34 +234,21 @@ func (ui *UI) setupKeybindings() {
 	)
 }
 
-func (ui *UI) setupLayout() {
+func (ui *UI) refreshStatusBar() {
-	ui.layout = tview.NewFlex().
+	// Placeholder; full refresh needs nick/target context.
-		SetDirection(tview.FlexRow).
-		AddItem(ui.messages, 0, 1, false).
-		AddItem(ui.statusBar, 1, 0, false).
-		AddItem(ui.input, 1, 0, true)
-
-	ui.app.SetRoot(ui.layout, true)
-	ui.app.SetFocus(ui.input)
 }
 
-// if needed.
+func (ui *UI) renderStatusBar(
-
-func (ui *UI) refreshStatus() {
-	// Rebuilt from app state by parent QueueUpdateDraw.
-}
-
-func (ui *UI) refreshStatusWith(
 	nick, target, connStatus string,
 ) {
 	var unreadParts []string
 
 	for i, buf := range ui.buffers {
 		if buf.Unread > 0 {
-			unreadParts = append(
+			unreadParts = append(unreadParts,
-				unreadParts,
 				fmt.Sprintf(
-					"%d:%s(%d)", i, buf.Name, buf.Unread,
+					"%d:%s(%d)",
+					i, buf.Name, buf.Unread,
 				),
 			)
 		}
@@ -286,8 +268,8 @@ func (ui *UI) refreshStatusWith(
 
 	ui.statusBar.Clear()
 
-	_, _ = fmt.Fprintf(
+	_, _ = fmt.Fprintf(ui.statusBar,
-		ui.statusBar, " [%s] %s %s %s%s",
+		" [%s] %s %s %s%s",
 		connStatus, nick, bufInfo, target, unread,
 	)
 }

internal/broker/broker.go

@@ -0,0 +1,65 @@
+// Package broker provides an in-memory pub/sub for long-poll notifications.
+package broker
+
+import (
+	"sync"
+)
+
+// Broker notifies waiting clients when new messages are available.
+type Broker struct {
+	mu        sync.Mutex
+	listeners map[int64][]chan struct{} // userID -> list of waiting channels
+}
+
+// New creates a new Broker.
+func New() *Broker {
+	return &Broker{
+		listeners: make(map[int64][]chan struct{}),
+	}
+}
+
+// Wait returns a channel that will be closed when a message is available for the user.
+func (b *Broker) Wait(userID int64) chan struct{} {
+	ch := make(chan struct{}, 1)
+
+	b.mu.Lock()
+	b.listeners[userID] = append(b.listeners[userID], ch)
+	b.mu.Unlock()
+
+	return ch
+}
+
+// Notify wakes up all waiting clients for a user.
+func (b *Broker) Notify(userID int64) {
+	b.mu.Lock()
+	waiters := b.listeners[userID]
+	delete(b.listeners, userID)
+	b.mu.Unlock()
+
+	for _, ch := range waiters {
+		select {
+		case ch <- struct{}{}:
+		default:
+		}
+	}
+}
+
+// Remove removes a specific wait channel (for cleanup on timeout).
+func (b *Broker) Remove(userID int64, ch chan struct{}) {
+	b.mu.Lock()
+	defer b.mu.Unlock()
+
+	waiters := b.listeners[userID]
+
+	for i, w := range waiters {
+		if w == ch {
+			b.listeners[userID] = append(waiters[:i], waiters[i+1:]...)
+
+			break
+		}
+	}
+
+	if len(b.listeners[userID]) == 0 {
+		delete(b.listeners, userID)
+	}
+}

internal/broker/broker_test.go

@@ -0,0 +1,118 @@
+package broker_test
+
+import (
+	"sync"
+	"testing"
+	"time"
+
+	"git.eeqj.de/sneak/chat/internal/broker"
+)
+
+func TestNewBroker(t *testing.T) {
+	t.Parallel()
+
+	b := broker.New()
+	if b == nil {
+		t.Fatal("expected non-nil broker")
+	}
+}
+
+func TestWaitAndNotify(t *testing.T) {
+	t.Parallel()
+
+	b := broker.New()
+	ch := b.Wait(1)
+
+	go func() {
+		time.Sleep(10 * time.Millisecond)
+		b.Notify(1)
+	}()
+
+	select {
+	case <-ch:
+	case <-time.After(2 * time.Second):
+		t.Fatal("timeout")
+	}
+}
+
+func TestNotifyWithoutWaiters(t *testing.T) {
+	t.Parallel()
+
+	b := broker.New()
+	b.Notify(42) // should not panic
+}
+
+func TestRemove(t *testing.T) {
+	t.Parallel()
+
+	b := broker.New()
+	ch := b.Wait(1)
+	b.Remove(1, ch)
+
+	b.Notify(1)
+
+	select {
+	case <-ch:
+		t.Fatal("should not receive after remove")
+	case <-time.After(50 * time.Millisecond):
+	}
+}
+
+func TestMultipleWaiters(t *testing.T) {
+	t.Parallel()
+
+	b := broker.New()
+	ch1 := b.Wait(1)
+	ch2 := b.Wait(1)
+
+	b.Notify(1)
+
+	select {
+	case <-ch1:
+	case <-time.After(time.Second):
+		t.Fatal("ch1 timeout")
+	}
+
+	select {
+	case <-ch2:
+	case <-time.After(time.Second):
+		t.Fatal("ch2 timeout")
+	}
+}
+
+func TestConcurrentWaitNotify(t *testing.T) {
+	t.Parallel()
+
+	b := broker.New()
+
+	var wg sync.WaitGroup
+
+	const concurrency = 100
+
+	for i := range concurrency {
+		wg.Add(1)
+
+		go func(uid int64) {
+			defer wg.Done()
+
+			ch := b.Wait(uid)
+			b.Notify(uid)
+
+			select {
+			case <-ch:
+			case <-time.After(time.Second):
+				t.Error("timeout")
+			}
+		}(int64(i % 10))
+	}
+
+	wg.Wait()
+}
+
+func TestRemoveNonexistent(t *testing.T) {
+	t.Parallel()
+
+	b := broker.New()
+	ch := make(chan struct{}, 1)
+	b.Remove(999, ch) // should not panic
+}

internal/db/db.go

@@ -11,20 +11,16 @@ import (
 	"sort"
 	"strconv"
 	"strings"
-	"time"
 
 	"git.eeqj.de/sneak/chat/internal/config"
 	"git.eeqj.de/sneak/chat/internal/logger"
-	"git.eeqj.de/sneak/chat/internal/models"
 	"go.uber.org/fx"
 
-	_ "github.com/joho/godotenv/autoload" // loads .env file
+	_ "github.com/joho/godotenv/autoload" // .env
-	_ "modernc.org/sqlite"                // SQLite driver
+	_ "modernc.org/sqlite"                // driver
 )
 
-const (
+const minMigrationParts = 2
-	minMigrationParts = 2
-)
 
 // SchemaFiles contains embedded SQL migration files.
 //
@@ -39,15 +35,18 @@ type Params struct {
 	Config *config.Config
 }
 
-// Database manages the SQLite database connection and migrations.
+// Database manages the SQLite connection and migrations.
 type Database struct {
 	db     *sql.DB
 	log    *slog.Logger
 	params *Params
 }
 
-// New creates a new Database instance and registers lifecycle hooks.
+// New creates a new Database and registers lifecycle hooks.
-func New(lc fx.Lifecycle, params Params) (*Database, error) {
+func New(
+	lc fx.Lifecycle,
+	params Params,
+) (*Database, error) {
 	s := new(Database)
 	s.params = &params
 	s.log = params.Logger.Get()
@@ -74,491 +73,56 @@ func New(lc fx.Lifecycle, params Params) (*Database, error) {
 	return s, nil
 }
 
-// NewTest creates a Database for testing, bypassing fx lifecycle.
-// It connects to the given DSN and runs all migrations.
-func NewTest(dsn string) (*Database, error) {
-	d, err := sql.Open("sqlite", dsn)
-	if err != nil {
-		return nil, err
-	}
-
-	s := &Database{
-		db:  d,
-		log: slog.Default(),
-	}
-
-	// Item 9: Enable foreign keys
-	_, err = d.Exec("PRAGMA foreign_keys = ON") //nolint:noctx // no context in sql.Open path
-	if err != nil {
-		_ = d.Close()
-
-		return nil, fmt.Errorf("enable foreign keys: %w", err)
-	}
-
-	ctx := context.Background()
-
-	err = s.runMigrations(ctx)
-	if err != nil {
-		_ = d.Close()
-
-		return nil, err
-	}
-
-	return s, nil
-}
-
 // GetDB returns the underlying sql.DB connection.
 func (s *Database) GetDB() *sql.DB {
 	return s.db
 }
 
-// Hydrate injects the database reference into any model that
-// embeds Base.
-func (s *Database) Hydrate(m interface{ SetDB(d models.DB) }) {
-	m.SetDB(s)
-}
-
-// GetUserByID looks up a user by their ID.
-func (s *Database) GetUserByID(
-	ctx context.Context,
-	id string,
-) (*models.User, error) {
-	u := &models.User{}
-	s.Hydrate(u)
-
-	err := s.db.QueryRowContext(ctx, `
-		SELECT id, nick, password_hash, created_at, updated_at, last_seen_at
-		FROM users WHERE id = ?`,
-		id,
-	).Scan(
-		&u.ID, &u.Nick, &u.PasswordHash,
-		&u.CreatedAt, &u.UpdatedAt, &u.LastSeenAt,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	return u, nil
-}
-
-// GetChannelByID looks up a channel by its ID.
-func (s *Database) GetChannelByID(
-	ctx context.Context,
-	id string,
-) (*models.Channel, error) {
-	c := &models.Channel{}
-	s.Hydrate(c)
-
-	err := s.db.QueryRowContext(ctx, `
-		SELECT id, name, topic, modes, created_at, updated_at
-		FROM channels WHERE id = ?`,
-		id,
-	).Scan(
-		&c.ID, &c.Name, &c.Topic, &c.Modes,
-		&c.CreatedAt, &c.UpdatedAt,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	return c, nil
-}
-
-// GetUserByNickModel looks up a user by their nick.
-func (s *Database) GetUserByNickModel(
-	ctx context.Context,
-	nick string,
-) (*models.User, error) {
-	u := &models.User{}
-	s.Hydrate(u)
-
-	err := s.db.QueryRowContext(ctx, `
-		SELECT id, nick, password_hash, created_at, updated_at, last_seen_at
-		FROM users WHERE nick = ?`,
-		nick,
-	).Scan(
-		&u.ID, &u.Nick, &u.PasswordHash,
-		&u.CreatedAt, &u.UpdatedAt, &u.LastSeenAt,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	return u, nil
-}
-
-// GetUserByTokenModel looks up a user by their auth token.
-func (s *Database) GetUserByTokenModel(
-	ctx context.Context,
-	token string,
-) (*models.User, error) {
-	u := &models.User{}
-	s.Hydrate(u)
-
-	err := s.db.QueryRowContext(ctx, `
-		SELECT u.id, u.nick, u.password_hash,
-		       u.created_at, u.updated_at, u.last_seen_at
-		FROM users u
-		JOIN auth_tokens t ON t.user_id = u.id
-		WHERE t.token = ?`,
-		token,
-	).Scan(
-		&u.ID, &u.Nick, &u.PasswordHash,
-		&u.CreatedAt, &u.UpdatedAt, &u.LastSeenAt,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	return u, nil
-}
-
-// DeleteAuthToken removes an auth token from the database.
-func (s *Database) DeleteAuthToken(
-	ctx context.Context,
-	token string,
-) error {
-	_, err := s.db.ExecContext(ctx,
-		`DELETE FROM auth_tokens WHERE token = ?`, token,
-	)
-
-	return err
-}
-
-// UpdateUserLastSeen updates the last_seen_at timestamp for a user.
-func (s *Database) UpdateUserLastSeen(
-	ctx context.Context,
-	userID string,
-) error {
-	_, err := s.db.ExecContext(ctx,
-		`UPDATE users SET last_seen_at = CURRENT_TIMESTAMP WHERE id = ?`,
-		userID,
-	)
-
-	return err
-}
-
-// CreateUserModel inserts a new user into the database.
-func (s *Database) CreateUserModel(
-	ctx context.Context,
-	id, nick, passwordHash string,
-) (*models.User, error) {
-	now := time.Now()
-
-	_, err := s.db.ExecContext(ctx,
-		`INSERT INTO users (id, nick, password_hash)
-		 VALUES (?, ?, ?)`,
-		id, nick, passwordHash,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	u := &models.User{
-		ID: id, Nick: nick, PasswordHash: passwordHash,
-		CreatedAt: now, UpdatedAt: now,
-	}
-	s.Hydrate(u)
-
-	return u, nil
-}
-
-// CreateChannel inserts a new channel into the database.
-func (s *Database) CreateChannel(
-	ctx context.Context,
-	id, name, topic, modes string,
-) (*models.Channel, error) {
-	now := time.Now()
-
-	_, err := s.db.ExecContext(ctx,
-		`INSERT INTO channels (id, name, topic, modes)
-		 VALUES (?, ?, ?, ?)`,
-		id, name, topic, modes,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	c := &models.Channel{
-		ID: id, Name: name, Topic: topic, Modes: modes,
-		CreatedAt: now, UpdatedAt: now,
-	}
-	s.Hydrate(c)
-
-	return c, nil
-}
-
-// AddChannelMember adds a user to a channel with the given modes.
-func (s *Database) AddChannelMember(
-	ctx context.Context,
-	channelID, userID, modes string,
-) (*models.ChannelMember, error) {
-	now := time.Now()
-
-	_, err := s.db.ExecContext(ctx,
-		`INSERT INTO channel_members
-		 (channel_id, user_id, modes)
-		 VALUES (?, ?, ?)`,
-		channelID, userID, modes,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	cm := &models.ChannelMember{
-		ChannelID: channelID,
-		UserID:    userID,
-		Modes:     modes,
-		JoinedAt:  now,
-	}
-	s.Hydrate(cm)
-
-	return cm, nil
-}
-
-// CreateMessage inserts a new message into the database.
-func (s *Database) CreateMessage(
-	ctx context.Context,
-	id, fromUserID, fromNick, target, msgType, body string,
-) (*models.Message, error) {
-	now := time.Now()
-
-	_, err := s.db.ExecContext(ctx,
-		`INSERT INTO messages
-		 (id, from_user_id, from_nick, target, type, body)
-		 VALUES (?, ?, ?, ?, ?, ?)`,
-		id, fromUserID, fromNick, target, msgType, body,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	m := &models.Message{
-		ID:         id,
-		FromUserID: fromUserID,
-		FromNick:   fromNick,
-		Target:     target,
-		Type:       msgType,
-		Body:       body,
-		Timestamp:  now,
-		CreatedAt:  now,
-	}
-	s.Hydrate(m)
-
-	return m, nil
-}
-
-// QueueMessage adds a message to a user's delivery queue.
-func (s *Database) QueueMessage(
-	ctx context.Context,
-	userID, messageID string,
-) (*models.MessageQueueEntry, error) {
-	now := time.Now()
-
-	res, err := s.db.ExecContext(ctx,
-		`INSERT INTO message_queue (user_id, message_id)
-		 VALUES (?, ?)`,
-		userID, messageID,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	entryID, err := res.LastInsertId()
-	if err != nil {
-		return nil, fmt.Errorf("get last insert id: %w", err)
-	}
-
-	mq := &models.MessageQueueEntry{
-		ID:        entryID,
-		UserID:    userID,
-		MessageID: messageID,
-		QueuedAt:  now,
-	}
-	s.Hydrate(mq)
-
-	return mq, nil
-}
-
-// DequeueMessages returns up to limit pending messages for a user,
-// ordered by queue time (oldest first).
-func (s *Database) DequeueMessages(
-	ctx context.Context,
-	userID string,
-	limit int,
-) ([]*models.MessageQueueEntry, error) {
-	rows, err := s.db.QueryContext(ctx, `
-		SELECT id, user_id, message_id, queued_at
-		FROM message_queue
-		WHERE user_id = ?
-		ORDER BY queued_at ASC
-		LIMIT ?`,
-		userID, limit,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	entries := []*models.MessageQueueEntry{}
-
-	for rows.Next() {
-		e := &models.MessageQueueEntry{}
-		s.Hydrate(e)
-
-		err = rows.Scan(&e.ID, &e.UserID, &e.MessageID, &e.QueuedAt)
-		if err != nil {
-			return nil, err
-		}
-
-		entries = append(entries, e)
-	}
-
-	return entries, rows.Err()
-}
-
-// AckMessages removes the given queue entry IDs, marking them as delivered.
-func (s *Database) AckMessages(
-	ctx context.Context,
-	entryIDs []int64,
-) error {
-	if len(entryIDs) == 0 {
-		return nil
-	}
-
-	placeholders := make([]string, len(entryIDs))
-	args := make([]any, len(entryIDs))
-
-	for i, id := range entryIDs {
-		placeholders[i] = "?"
-		args[i] = id
-	}
-
-	query := fmt.Sprintf( //nolint:gosec // placeholders are ?, not user input
-		"DELETE FROM message_queue WHERE id IN (%s)",
-		strings.Join(placeholders, ","),
-	)
-
-	_, err := s.db.ExecContext(ctx, query, args...)
-
-	return err
-}
-
-// CreateAuthToken inserts a new auth token for a user.
-func (s *Database) CreateAuthToken(
-	ctx context.Context,
-	token, userID string,
-) (*models.AuthToken, error) {
-	now := time.Now()
-
-	_, err := s.db.ExecContext(ctx,
-		`INSERT INTO auth_tokens (token, user_id)
-		 VALUES (?, ?)`,
-		token, userID,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	at := &models.AuthToken{Token: token, UserID: userID, CreatedAt: now}
-	s.Hydrate(at)
-
-	return at, nil
-}
-
-// CreateSession inserts a new session for a user.
-func (s *Database) CreateSession(
-	ctx context.Context,
-	id, userID string,
-) (*models.Session, error) {
-	now := time.Now()
-
-	_, err := s.db.ExecContext(ctx,
-		`INSERT INTO sessions (id, user_id)
-		 VALUES (?, ?)`,
-		id, userID,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	sess := &models.Session{
-		ID: id, UserID: userID,
-		CreatedAt: now, LastActiveAt: now,
-	}
-	s.Hydrate(sess)
-
-	return sess, nil
-}
-
-// CreateServerLink inserts a new server link.
-func (s *Database) CreateServerLink(
-	ctx context.Context,
-	id, name, url, sharedKeyHash string,
-	isActive bool,
-) (*models.ServerLink, error) {
-	now := time.Now()
-	active := 0
-
-	if isActive {
-		active = 1
-	}
-
-	_, err := s.db.ExecContext(ctx,
-		`INSERT INTO server_links
-		 (id, name, url, shared_key_hash, is_active)
-		 VALUES (?, ?, ?, ?, ?)`,
-		id, name, url, sharedKeyHash, active,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	sl := &models.ServerLink{
-		ID:            id,
-		Name:          name,
-		URL:           url,
-		SharedKeyHash: sharedKeyHash,
-		IsActive:      isActive,
-		CreatedAt:     now,
-	}
-	s.Hydrate(sl)
-
-	return sl, nil
-}
-
 func (s *Database) connect(ctx context.Context) error {
 	dbURL := s.params.Config.DBURL
 	if dbURL == "" {
-		dbURL = "file:./data.db?_journal_mode=WAL"
+		dbURL = "file:./data.db?_journal_mode=WAL&_busy_timeout=5000"
 	}
 
 	s.log.Info("connecting to database", "url", dbURL)
 
 	d, err := sql.Open("sqlite", dbURL)
 	if err != nil {
-		s.log.Error("failed to open database", "error", err)
+		s.log.Error(
+			"failed to open database", "error", err,
+		)
 
 		return err
 	}
 
 	err = d.PingContext(ctx)
 	if err != nil {
-		s.log.Error("failed to ping database", "error", err)
+		s.log.Error(
+			"failed to ping database", "error", err,
+		)
 
 		return err
 	}
 
+	d.SetMaxOpenConns(1)
+
 	s.db = d
 	s.log.Info("database connected")
 
-	// Item 9: Enable foreign keys on every connection
+	_, err = s.db.ExecContext(
-	_, err = s.db.ExecContext(ctx, "PRAGMA foreign_keys = ON")
+		ctx, "PRAGMA foreign_keys = ON",
+	)
 	if err != nil {
 		return fmt.Errorf("enable foreign keys: %w", err)
 	}
 
+	_, err = s.db.ExecContext(
+		ctx, "PRAGMA busy_timeout = 5000",
+	)
+	if err != nil {
+		return fmt.Errorf("set busy timeout: %w", err)
+	}
+
 	return s.runMigrations(ctx)
 }
 
@@ -568,10 +132,17 @@ type migration struct {
 	sql     string
 }
 
-func (s *Database) runMigrations(ctx context.Context) error {
+func (s *Database) runMigrations(
-	err := s.bootstrapMigrationsTable(ctx)
+	ctx context.Context,
+) error {
+	_, err := s.db.ExecContext(ctx,
+		`CREATE TABLE IF NOT EXISTS schema_migrations (
+		 version INTEGER PRIMARY KEY,
+		 applied_at DATETIME DEFAULT CURRENT_TIMESTAMP)`)
 	if err != nil {
-		return err
+		return fmt.Errorf(
+			"create schema_migrations: %w", err,
+		)
 	}
 
 	migrations, err := s.loadMigrations()
@@ -579,9 +150,11 @@ func (s *Database) runMigrations(ctx context.Context) error {
 		return err
 	}
 
-	err = s.applyMigrations(ctx, migrations)
+	for _, m := range migrations {
-	if err != nil {
+		err = s.applyMigration(ctx, m)
-		return err
+		if err != nil {
+			return err
+		}
 	}
 
 	s.log.Info("database migrations complete")
@@ -589,30 +162,87 @@ func (s *Database) runMigrations(ctx context.Context) error {
 	return nil
 }
 
-func (s *Database) bootstrapMigrationsTable(
+func (s *Database) applyMigration(
 	ctx context.Context,
+	m migration,
 ) error {
-	_, err := s.db.ExecContext(ctx,
+	var exists int
-		`CREATE TABLE IF NOT EXISTS schema_migrations (
+
-		version INTEGER PRIMARY KEY,
+	err := s.db.QueryRowContext(ctx,
-		applied_at DATETIME DEFAULT CURRENT_TIMESTAMP
+		`SELECT COUNT(*) FROM schema_migrations
-	)`)
+		 WHERE version = ?`,
+		m.version,
+	).Scan(&exists)
 	if err != nil {
 		return fmt.Errorf(
-			"create schema_migrations table: %w", err,
+			"check migration %d: %w", m.version, err,
 		)
 	}
 
-	return nil
+	if exists > 0 {
-}
+		return nil
-
-func (s *Database) loadMigrations() ([]migration, error) {
-	entries, err := fs.ReadDir(SchemaFiles, "schema")
-	if err != nil {
-		return nil, fmt.Errorf("read schema dir: %w", err)
 	}
 
-	var migrations []migration
+	s.log.Info(
+		"applying migration",
+		"version", m.version,
+		"name", m.name,
+	)
+
+	return s.execMigration(ctx, m)
+}
+
+func (s *Database) execMigration(
+	ctx context.Context,
+	m migration,
+) error {
+	tx, err := s.db.BeginTx(ctx, nil)
+	if err != nil {
+		return fmt.Errorf(
+			"begin tx for migration %d: %w",
+			m.version, err,
+		)
+	}
+
+	_, err = tx.ExecContext(ctx, m.sql)
+	if err != nil {
+		_ = tx.Rollback()
+
+		return fmt.Errorf(
+			"apply migration %d (%s): %w",
+			m.version, m.name, err,
+		)
+	}
+
+	_, err = tx.ExecContext(ctx,
+		`INSERT INTO schema_migrations (version)
+		 VALUES (?)`,
+		m.version,
+	)
+	if err != nil {
+		_ = tx.Rollback()
+
+		return fmt.Errorf(
+			"record migration %d: %w",
+			m.version, err,
+		)
+	}
+
+	return tx.Commit()
+}
+
+func (s *Database) loadMigrations() (
+	[]migration,
+	error,
+) {
+	entries, err := fs.ReadDir(SchemaFiles, "schema")
+	if err != nil {
+		return nil, fmt.Errorf(
+			"read schema dir: %w", err,
+		)
+	}
+
+	migrations := make([]migration, 0, len(entries))
 
 	for _, entry := range entries {
 		if entry.IsDir() ||
@@ -627,17 +257,18 @@ func (s *Database) loadMigrations() ([]migration, error) {
 			continue
 		}
 
-		version, err := strconv.Atoi(parts[0])
+		version, parseErr := strconv.Atoi(parts[0])
-		if err != nil {
+		if parseErr != nil {
 			continue
 		}
 
-		content, err := SchemaFiles.ReadFile(
+		content, readErr := SchemaFiles.ReadFile(
 			"schema/" + entry.Name(),
 		)
-		if err != nil {
+		if readErr != nil {
 			return nil, fmt.Errorf(
-				"read migration %s: %w", entry.Name(), err,
+				"read migration %s: %w",
+				entry.Name(), readErr,
 			)
 		}
 
@@ -654,82 +285,3 @@ func (s *Database) loadMigrations() ([]migration, error) {
 
 	return migrations, nil
 }
-
-// Item 4: Wrap each migration in a transaction
-func (s *Database) applyMigrations(
-	ctx context.Context,
-	migrations []migration,
-) error {
-	for _, m := range migrations {
-		var exists int
-
-		err := s.db.QueryRowContext(ctx,
-			"SELECT COUNT(*) FROM schema_migrations WHERE version = ?",
-			m.version,
-		).Scan(&exists)
-		if err != nil {
-			return fmt.Errorf(
-				"check migration %d: %w", m.version, err,
-			)
-		}
-
-		if exists > 0 {
-			continue
-		}
-
-		s.log.Info(
-			"applying migration",
-			"version", m.version, "name", m.name,
-		)
-
-		err = s.executeMigration(ctx, m)
-		if err != nil {
-			return err
-		}
-	}
-
-	return nil
-}
-
-func (s *Database) executeMigration(
-	ctx context.Context,
-	m migration,
-) error {
-	tx, err := s.db.BeginTx(ctx, nil)
-	if err != nil {
-		return fmt.Errorf(
-			"begin tx for migration %d: %w", m.version, err,
-		)
-	}
-
-	_, err = tx.ExecContext(ctx, m.sql)
-	if err != nil {
-		_ = tx.Rollback()
-
-		return fmt.Errorf(
-			"apply migration %d (%s): %w",
-			m.version, m.name, err,
-		)
-	}
-
-	_, err = tx.ExecContext(ctx,
-		"INSERT INTO schema_migrations (version) VALUES (?)",
-		m.version,
-	)
-	if err != nil {
-		_ = tx.Rollback()
-
-		return fmt.Errorf(
-			"record migration %d: %w", m.version, err,
-		)
-	}
-
-	err = tx.Commit()
-	if err != nil {
-		return fmt.Errorf(
-			"commit migration %d: %w", m.version, err,
-		)
-	}
-
-	return nil
-}

internal/db/db_test.go

@@ -1,425 +0,0 @@
-package db_test
-
-import (
-	"context"
-	"fmt"
-	"path/filepath"
-	"testing"
-	"time"
-
-	"git.eeqj.de/sneak/chat/internal/db"
-)
-
-const (
-	nickAlice   = "alice"
-	nickBob     = "bob"
-	nickCharlie = "charlie"
-)
-
-// setupTestDB creates a fresh database in a temp directory with
-// all migrations applied.
-func setupTestDB(t *testing.T) *db.Database {
-	t.Helper()
-
-	dir := t.TempDir()
-	dsn := fmt.Sprintf(
-		"file:%s?_journal_mode=WAL",
-		filepath.Join(dir, "test.db"),
-	)
-
-	d, err := db.NewTest(dsn)
-	if err != nil {
-		t.Fatalf("failed to create test database: %v", err)
-	}
-
-	t.Cleanup(func() { _ = d.GetDB().Close() })
-
-	return d
-}
-
-func TestCreateUser(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	id, token, err := d.CreateUser(ctx, nickAlice)
-	if err != nil {
-		t.Fatalf("CreateUser: %v", err)
-	}
-
-	if id <= 0 {
-		t.Errorf("expected positive id, got %d", id)
-	}
-
-	if token == "" {
-		t.Error("expected non-empty token")
-	}
-}
-
-func TestGetUserByToken(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	_, token, _ := d.CreateUser(ctx, nickAlice)
-
-	id, nick, err := d.GetUserByToken(ctx, token)
-	if err != nil {
-		t.Fatalf("GetUserByToken: %v", err)
-	}
-
-	if id <= 0 || nick != nickAlice {
-		t.Errorf(
-			"got id=%d nick=%s, want nick=%s",
-			id, nick, nickAlice,
-		)
-	}
-}
-
-func TestGetUserByNick(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	origID, _, _ := d.CreateUser(ctx, nickAlice)
-
-	id, err := d.GetUserByNick(ctx, nickAlice)
-	if err != nil {
-		t.Fatalf("GetUserByNick: %v", err)
-	}
-
-	if id != origID {
-		t.Errorf("got id %d, want %d", id, origID)
-	}
-}
-
-func TestGetOrCreateChannel(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	id1, err := d.GetOrCreateChannel(ctx, "#general")
-	if err != nil {
-		t.Fatalf("GetOrCreateChannel: %v", err)
-	}
-
-	if id1 <= 0 {
-		t.Errorf("expected positive id, got %d", id1)
-	}
-
-	// Same channel returns same ID.
-	id2, err := d.GetOrCreateChannel(ctx, "#general")
-	if err != nil {
-		t.Fatalf("GetOrCreateChannel(2): %v", err)
-	}
-
-	if id1 != id2 {
-		t.Errorf("got different ids: %d vs %d", id1, id2)
-	}
-}
-
-func TestJoinAndListChannels(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid, _, _ := d.CreateUser(ctx, nickAlice)
-	ch1, _ := d.GetOrCreateChannel(ctx, "#alpha")
-	ch2, _ := d.GetOrCreateChannel(ctx, "#beta")
-
-	_ = d.JoinChannel(ctx, ch1, uid)
-	_ = d.JoinChannel(ctx, ch2, uid)
-
-	channels, err := d.ListChannels(ctx, uid)
-	if err != nil {
-		t.Fatalf("ListChannels: %v", err)
-	}
-
-	if len(channels) != 2 {
-		t.Fatalf("expected 2 channels, got %d", len(channels))
-	}
-}
-
-func TestListChannelsEmpty(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid, _, _ := d.CreateUser(ctx, nickAlice)
-
-	channels, err := d.ListChannels(ctx, uid)
-	if err != nil {
-		t.Fatalf("ListChannels: %v", err)
-	}
-
-	if len(channels) != 0 {
-		t.Errorf("expected 0 channels, got %d", len(channels))
-	}
-}
-
-func TestPartChannel(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid, _, _ := d.CreateUser(ctx, nickAlice)
-	chID, _ := d.GetOrCreateChannel(ctx, "#general")
-
-	_ = d.JoinChannel(ctx, chID, uid)
-	_ = d.PartChannel(ctx, chID, uid)
-
-	channels, err := d.ListChannels(ctx, uid)
-	if err != nil {
-		t.Fatalf("ListChannels: %v", err)
-	}
-
-	if len(channels) != 0 {
-		t.Errorf("expected 0 after part, got %d", len(channels))
-	}
-}
-
-func TestSendAndGetMessages(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid, _, _ := d.CreateUser(ctx, nickAlice)
-	chID, _ := d.GetOrCreateChannel(ctx, "#general")
-	_ = d.JoinChannel(ctx, chID, uid)
-
-	_, err := d.SendMessage(ctx, chID, uid, "hello world")
-	if err != nil {
-		t.Fatalf("SendMessage: %v", err)
-	}
-
-	msgs, err := d.GetMessages(ctx, chID, 0, 0)
-	if err != nil {
-		t.Fatalf("GetMessages: %v", err)
-	}
-
-	if len(msgs) != 1 {
-		t.Fatalf("expected 1 message, got %d", len(msgs))
-	}
-
-	if msgs[0].Content != "hello world" {
-		t.Errorf(
-			"got content %q, want %q",
-			msgs[0].Content, "hello world",
-		)
-	}
-}
-
-func TestChannelMembers(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid1, _, _ := d.CreateUser(ctx, nickAlice)
-	uid2, _, _ := d.CreateUser(ctx, nickBob)
-	uid3, _, _ := d.CreateUser(ctx, nickCharlie)
-	chID, _ := d.GetOrCreateChannel(ctx, "#general")
-
-	_ = d.JoinChannel(ctx, chID, uid1)
-	_ = d.JoinChannel(ctx, chID, uid2)
-	_ = d.JoinChannel(ctx, chID, uid3)
-
-	members, err := d.ChannelMembers(ctx, chID)
-	if err != nil {
-		t.Fatalf("ChannelMembers: %v", err)
-	}
-
-	if len(members) != 3 {
-		t.Fatalf("expected 3 members, got %d", len(members))
-	}
-}
-
-func TestChannelMembersEmpty(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	chID, _ := d.GetOrCreateChannel(ctx, "#empty")
-
-	members, err := d.ChannelMembers(ctx, chID)
-	if err != nil {
-		t.Fatalf("ChannelMembers: %v", err)
-	}
-
-	if len(members) != 0 {
-		t.Errorf("expected 0, got %d", len(members))
-	}
-}
-
-func TestSendDM(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid1, _, _ := d.CreateUser(ctx, nickAlice)
-	uid2, _, _ := d.CreateUser(ctx, nickBob)
-
-	msgID, err := d.SendDM(ctx, uid1, uid2, "hey bob")
-	if err != nil {
-		t.Fatalf("SendDM: %v", err)
-	}
-
-	if msgID <= 0 {
-		t.Errorf("expected positive msgID, got %d", msgID)
-	}
-
-	msgs, err := d.GetDMs(ctx, uid1, uid2, 0, 0)
-	if err != nil {
-		t.Fatalf("GetDMs: %v", err)
-	}
-
-	if len(msgs) != 1 {
-		t.Fatalf("expected 1 DM, got %d", len(msgs))
-	}
-
-	if msgs[0].Content != "hey bob" {
-		t.Errorf("got %q, want %q", msgs[0].Content, "hey bob")
-	}
-}
-
-func TestPollMessages(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid1, _, _ := d.CreateUser(ctx, nickAlice)
-	uid2, _, _ := d.CreateUser(ctx, nickBob)
-	chID, _ := d.GetOrCreateChannel(ctx, "#general")
-
-	_ = d.JoinChannel(ctx, chID, uid1)
-	_ = d.JoinChannel(ctx, chID, uid2)
-
-	_, _ = d.SendMessage(ctx, chID, uid2, "hello")
-	_, _ = d.SendDM(ctx, uid2, uid1, "private")
-
-	time.Sleep(10 * time.Millisecond)
-
-	msgs, err := d.PollMessages(ctx, uid1, 0, 0)
-	if err != nil {
-		t.Fatalf("PollMessages: %v", err)
-	}
-
-	if len(msgs) < 2 {
-		t.Fatalf("expected >=2 messages, got %d", len(msgs))
-	}
-}
-
-func TestChangeNick(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	_, token, _ := d.CreateUser(ctx, nickAlice)
-
-	err := d.ChangeNick(ctx, 1, "alice2")
-	if err != nil {
-		t.Fatalf("ChangeNick: %v", err)
-	}
-
-	_, nick, err := d.GetUserByToken(ctx, token)
-	if err != nil {
-		t.Fatalf("GetUserByToken: %v", err)
-	}
-
-	if nick != "alice2" {
-		t.Errorf("got nick %q, want alice2", nick)
-	}
-}
-
-func TestSetTopic(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid, _, _ := d.CreateUser(ctx, nickAlice)
-	_, _ = d.GetOrCreateChannel(ctx, "#general")
-
-	err := d.SetTopic(ctx, "#general", uid, "new topic")
-	if err != nil {
-		t.Fatalf("SetTopic: %v", err)
-	}
-
-	channels, err := d.ListAllChannels(ctx)
-	if err != nil {
-		t.Fatalf("ListAllChannels: %v", err)
-	}
-
-	found := false
-
-	for _, ch := range channels {
-		if ch.Name == "#general" && ch.Topic == "new topic" {
-			found = true
-		}
-	}
-
-	if !found {
-		t.Error("topic was not updated")
-	}
-}
-
-func TestGetMessagesBefore(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	uid, _, _ := d.CreateUser(ctx, nickAlice)
-	chID, _ := d.GetOrCreateChannel(ctx, "#general")
-
-	_ = d.JoinChannel(ctx, chID, uid)
-
-	for i := range 5 {
-		_, _ = d.SendMessage(
-			ctx, chID, uid,
-			fmt.Sprintf("msg%d", i),
-		)
-
-		time.Sleep(10 * time.Millisecond)
-	}
-
-	msgs, err := d.GetMessagesBefore(ctx, chID, 0, 3)
-	if err != nil {
-		t.Fatalf("GetMessagesBefore: %v", err)
-	}
-
-	if len(msgs) != 3 {
-		t.Fatalf("expected 3, got %d", len(msgs))
-	}
-}
-
-func TestListAllChannels(t *testing.T) {
-	t.Parallel()
-
-	d := setupTestDB(t)
-	ctx := context.Background()
-
-	_, _ = d.GetOrCreateChannel(ctx, "#alpha")
-	_, _ = d.GetOrCreateChannel(ctx, "#beta")
-
-	channels, err := d.ListAllChannels(ctx)
-	if err != nil {
-		t.Fatalf("ListAllChannels: %v", err)
-	}
-
-	if len(channels) != 2 {
-		t.Errorf("expected 2, got %d", len(channels))
-	}
-}

internal/db/export_test.go

@@ -0,0 +1,47 @@
+package db
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"log/slog"
+	"sync/atomic"
+)
+
+//nolint:gochecknoglobals // test counter
+var testDBCounter atomic.Int64
+
+// NewTestDatabase creates an in-memory database for testing.
+func NewTestDatabase() (*Database, error) {
+	n := testDBCounter.Add(1)
+
+	dsn := fmt.Sprintf(
+		"file:testdb%d?mode=memory"+
+			"&cache=shared&_pragma=foreign_keys(1)",
+		n,
+	)
+
+	d, err := sql.Open("sqlite", dsn)
+	if err != nil {
+		return nil, err
+	}
+
+	database := &Database{db: d, log: slog.Default()}
+
+	err = database.runMigrations(context.Background())
+	if err != nil {
+		closeErr := d.Close()
+		if closeErr != nil {
+			return nil, closeErr
+		}
+
+		return nil, err
+	}
+
+	return database, nil
+}
+
+// Close closes the underlying database connection.
+func (s *Database) Close() error {
+	return s.db.Close()
+}

internal/db/queries.go

@@ -5,14 +5,17 @@ import (
 	"crypto/rand"
 	"database/sql"
 	"encoding/hex"
+	"encoding/json"
 	"fmt"
 	"time"
+
+	"github.com/google/uuid"
 )
 
 const (
-	defaultMessageLimit = 50
+	tokenBytes       = 32
-	defaultPollLimit    = 100
+	defaultPollLimit = 100
-	tokenBytes          = 32
+	defaultHistLimit = 50
 )
 
 func generateToken() string {
@@ -22,8 +25,33 @@ func generateToken() string {
 	return hex.EncodeToString(b)
 }
 
-// CreateUser registers a new user with the given nick and
+// IRCMessage is the IRC envelope for all messages.
-// returns the user with token.
+type IRCMessage struct {
+	ID      string          `json:"id"`
+	Command string          `json:"command"`
+	From    string          `json:"from,omitempty"`
+	To      string          `json:"to,omitempty"`
+	Body    json.RawMessage `json:"body,omitempty"`
+	TS      string          `json:"ts"`
+	Meta    json.RawMessage `json:"meta,omitempty"`
+	DBID    int64           `json:"-"`
+}
+
+// ChannelInfo is a lightweight channel representation.
+type ChannelInfo struct {
+	ID    int64  `json:"id"`
+	Name  string `json:"name"`
+	Topic string `json:"topic"`
+}
+
+// MemberInfo represents a channel member.
+type MemberInfo struct {
+	ID       int64     `json:"id"`
+	Nick     string    `json:"nick"`
+	LastSeen time.Time `json:"lastSeen"`
+}
+
+// CreateUser registers a new user with the given nick.
 func (s *Database) CreateUser(
 	ctx context.Context,
 	nick string,
@@ -32,7 +60,9 @@ func (s *Database) CreateUser(
 	now := time.Now()
 
 	res, err := s.db.ExecContext(ctx,
-		"INSERT INTO users (nick, token, created_at, last_seen) VALUES (?, ?, ?, ?)",
+		`INSERT INTO users
+		 (nick, token, created_at, last_seen)
+		 VALUES (?, ?, ?, ?)`,
 		nick, token, now, now)
 	if err != nil {
 		return 0, "", fmt.Errorf("create user: %w", err)
@@ -43,8 +73,7 @@ func (s *Database) CreateUser(
 	return id, token, nil
 }
 
-// GetUserByToken returns user id and nick for a given auth
+// GetUserByToken returns user id and nick for a token.
-// token.
 func (s *Database) GetUserByToken(
 	ctx context.Context,
 	token string,
@@ -62,7 +91,6 @@ func (s *Database) GetUserByToken(
 		return 0, "", err
 	}
 
-	// Update last_seen
 	_, _ = s.db.ExecContext(
 		ctx,
 		"UPDATE users SET last_seen = ? WHERE id = ?",
@@ -88,8 +116,23 @@ func (s *Database) GetUserByNick(
 	return id, err
 }
 
-// GetOrCreateChannel returns the channel id, creating it if
+// GetChannelByName returns the channel ID for a name.
-// needed.
+func (s *Database) GetChannelByName(
+	ctx context.Context,
+	name string,
+) (int64, error) {
+	var id int64
+
+	err := s.db.QueryRowContext(
+		ctx,
+		"SELECT id FROM channels WHERE name = ?",
+		name,
+	).Scan(&id)
+
+	return id, err
+}
+
+// GetOrCreateChannel returns channel id, creating if needed.
 func (s *Database) GetOrCreateChannel(
 	ctx context.Context,
 	name string,
@@ -108,7 +151,9 @@ func (s *Database) GetOrCreateChannel(
 	now := time.Now()
 
 	res, err := s.db.ExecContext(ctx,
-		"INSERT INTO channels (name, created_at, updated_at) VALUES (?, ?, ?)",
+		`INSERT INTO channels
+		 (name, created_at, updated_at)
+		 VALUES (?, ?, ?)`,
 		name, now, now)
 	if err != nil {
 		return 0, fmt.Errorf("create channel: %w", err)
@@ -125,7 +170,9 @@ func (s *Database) JoinChannel(
 	channelID, userID int64,
 ) error {
 	_, err := s.db.ExecContext(ctx,
-		"INSERT OR IGNORE INTO channel_members (channel_id, user_id, joined_at) VALUES (?, ?, ?)",
+		`INSERT OR IGNORE INTO channel_members
+		 (channel_id, user_id, joined_at)
+		 VALUES (?, ?, ?)`,
 		channelID, userID, time.Now())
 
 	return err
@@ -137,35 +184,35 @@ func (s *Database) PartChannel(
 	channelID, userID int64,
 ) error {
 	_, err := s.db.ExecContext(ctx,
-		"DELETE FROM channel_members WHERE channel_id = ? AND user_id = ?",
+		`DELETE FROM channel_members
+		 WHERE channel_id = ? AND user_id = ?`,
 		channelID, userID)
 
 	return err
 }
 
-// ChannelInfo is a lightweight channel representation.
+// DeleteChannelIfEmpty removes a channel with no members.
-type ChannelInfo struct {
+func (s *Database) DeleteChannelIfEmpty(
-	ID    int64  `json:"id"`
+	ctx context.Context,
-	Name  string `json:"name"`
+	channelID int64,
-	Topic string `json:"topic"`
+) error {
+	_, err := s.db.ExecContext(ctx,
+		`DELETE FROM channels WHERE id = ?
+		 AND NOT EXISTS
+		 (SELECT 1 FROM channel_members
+		  WHERE channel_id = ?)`,
+		channelID, channelID)
+
+	return err
 }
 
-// ListChannels returns all channels the user has joined.
+// scanChannels scans rows into a ChannelInfo slice.
-func (s *Database) ListChannels(
+func scanChannels(
-	ctx context.Context,
+	rows *sql.Rows,
-	userID int64,
 ) ([]ChannelInfo, error) {
-	rows, err := s.db.QueryContext(ctx,
-		`SELECT c.id, c.name, c.topic FROM channels c
-		 INNER JOIN channel_members cm ON cm.channel_id = c.id
-		 WHERE cm.user_id = ? ORDER BY c.name`, userID)
-	if err != nil {
-		return nil, err
-	}
-
 	defer func() { _ = rows.Close() }()
 
-	var channels []ChannelInfo
+	var out []ChannelInfo
 
 	for rows.Next() {
 		var ch ChannelInfo
@@ -175,26 +222,52 @@ func (s *Database) ListChannels(
 			return nil, err
 		}
 
-		channels = append(channels, ch)
+		out = append(out, ch)
 	}
 
-	err = rows.Err()
+	err := rows.Err()
 	if err != nil {
 		return nil, err
 	}
 
-	if channels == nil {
+	if out == nil {
-		channels = []ChannelInfo{}
+		out = []ChannelInfo{}
 	}
 
-	return channels, nil
+	return out, nil
 }
 
-// MemberInfo represents a channel member.
+// ListChannels returns channels the user has joined.
-type MemberInfo struct {
+func (s *Database) ListChannels(
-	ID       int64     `json:"id"`
+	ctx context.Context,
-	Nick     string    `json:"nick"`
+	userID int64,
-	LastSeen time.Time `json:"lastSeen"`
+) ([]ChannelInfo, error) {
+	rows, err := s.db.QueryContext(ctx,
+		`SELECT c.id, c.name, c.topic
+		 FROM channels c
+		 INNER JOIN channel_members cm
+		   ON cm.channel_id = c.id
+		 WHERE cm.user_id = ?
+		 ORDER BY c.name`, userID)
+	if err != nil {
+		return nil, err
+	}
+
+	return scanChannels(rows)
+}
+
+// ListAllChannels returns every channel.
+func (s *Database) ListAllChannels(
+	ctx context.Context,
+) ([]ChannelInfo, error) {
+	rows, err := s.db.QueryContext(ctx,
+		`SELECT id, name, topic
+		 FROM channels ORDER BY name`)
+	if err != nil {
+		return nil, err
+	}
+
+	return scanChannels(rows)
 }
 
 // ChannelMembers returns all members of a channel.
@@ -203,9 +276,12 @@ func (s *Database) ChannelMembers(
 	channelID int64,
 ) ([]MemberInfo, error) {
 	rows, err := s.db.QueryContext(ctx,
-		`SELECT u.id, u.nick, u.last_seen FROM users u
+		`SELECT u.id, u.nick, u.last_seen
-		 INNER JOIN channel_members cm ON cm.user_id = u.id
+		 FROM users u
-		 WHERE cm.channel_id = ? ORDER BY u.nick`, channelID)
+		 INNER JOIN channel_members cm
+		   ON cm.user_id = u.id
+		 WHERE cm.channel_id = ?
+		 ORDER BY u.nick`, channelID)
 	if err != nil {
 		return nil, err
 	}
@@ -217,7 +293,7 @@ func (s *Database) ChannelMembers(
 	for rows.Next() {
 		var m MemberInfo
 
-		err := rows.Scan(&m.ID, &m.Nick, &m.LastSeen)
+		err = rows.Scan(&m.ID, &m.Nick, &m.LastSeen)
 		if err != nil {
 			return nil, err
 		}
@@ -237,407 +313,256 @@ func (s *Database) ChannelMembers(
 	return members, nil
 }
 
-// MessageInfo represents a chat message.
+// scanInt64s scans rows into an int64 slice.
-type MessageInfo struct {
+func scanInt64s(rows *sql.Rows) ([]int64, error) {
-	ID        int64     `json:"id"`
+	defer func() { _ = rows.Close() }()
-	Channel   string    `json:"channel,omitempty"`
+
-	Nick      string    `json:"nick"`
+	var ids []int64
-	Content   string    `json:"content"`
+
-	IsDM      bool      `json:"isDm,omitempty"`
+	for rows.Next() {
-	DMTarget  string    `json:"dmTarget,omitempty"`
+		var id int64
-	CreatedAt time.Time `json:"createdAt"`
+
+		err := rows.Scan(&id)
+		if err != nil {
+			return nil, err
+		}
+
+		ids = append(ids, id)
+	}
+
+	err := rows.Err()
+	if err != nil {
+		return nil, err
+	}
+
+	return ids, nil
 }
 
-// GetMessages returns messages for a channel, optionally
+// GetChannelMemberIDs returns user IDs in a channel.
-// after a given ID.
+func (s *Database) GetChannelMemberIDs(
-func (s *Database) GetMessages(
 	ctx context.Context,
 	channelID int64,
-	afterID int64,
+) ([]int64, error) {
-	limit int,
-) ([]MessageInfo, error) {
-	if limit <= 0 {
-		limit = defaultMessageLimit
-	}
-
 	rows, err := s.db.QueryContext(ctx,
-		`SELECT m.id, c.name, u.nick, m.content, m.created_at
+		`SELECT user_id FROM channel_members
-		 FROM messages m
+		 WHERE channel_id = ?`, channelID)
-		 INNER JOIN users u ON u.id = m.user_id
-		 INNER JOIN channels c ON c.id = m.channel_id
-		 WHERE m.channel_id = ? AND m.is_dm = 0 AND m.id > ?
-		 ORDER BY m.id ASC LIMIT ?`, channelID, afterID, limit)
 	if err != nil {
 		return nil, err
 	}
 
-	defer func() { _ = rows.Close() }()
+	return scanInt64s(rows)
-
-	var msgs []MessageInfo
-
-	for rows.Next() {
-		var m MessageInfo
-
-		err := rows.Scan(
-			&m.ID, &m.Channel, &m.Nick,
-			&m.Content, &m.CreatedAt,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		msgs = append(msgs, m)
-	}
-
-	err = rows.Err()
-	if err != nil {
-		return nil, err
-	}
-
-	if msgs == nil {
-		msgs = []MessageInfo{}
-	}
-
-	return msgs, nil
 }
 
-// SendMessage inserts a channel message.
+// GetUserChannelIDs returns channel IDs the user is in.
-func (s *Database) SendMessage(
+func (s *Database) GetUserChannelIDs(
-	ctx context.Context,
-	channelID, userID int64,
-	content string,
-) (int64, error) {
-	res, err := s.db.ExecContext(ctx,
-		"INSERT INTO messages (channel_id, user_id, content, is_dm, created_at) VALUES (?, ?, ?, 0, ?)",
-		channelID, userID, content, time.Now())
-	if err != nil {
-		return 0, err
-	}
-
-	return res.LastInsertId()
-}
-
-// SendDM inserts a direct message.
-func (s *Database) SendDM(
-	ctx context.Context,
-	fromID, toID int64,
-	content string,
-) (int64, error) {
-	res, err := s.db.ExecContext(ctx,
-		"INSERT INTO messages (user_id, content, is_dm, dm_target_id, created_at) VALUES (?, ?, 1, ?, ?)",
-		fromID, content, toID, time.Now())
-	if err != nil {
-		return 0, err
-	}
-
-	return res.LastInsertId()
-}
-
-// GetDMs returns direct messages between two users after a
-// given ID.
-func (s *Database) GetDMs(
-	ctx context.Context,
-	userA, userB int64,
-	afterID int64,
-	limit int,
-) ([]MessageInfo, error) {
-	if limit <= 0 {
-		limit = defaultMessageLimit
-	}
-
-	rows, err := s.db.QueryContext(ctx,
-		`SELECT m.id, u.nick, m.content, t.nick, m.created_at
-		 FROM messages m
-		 INNER JOIN users u ON u.id = m.user_id
-		 INNER JOIN users t ON t.id = m.dm_target_id
-		 WHERE m.is_dm = 1 AND m.id > ?
-		 AND ((m.user_id = ? AND m.dm_target_id = ?)
-		   OR (m.user_id = ? AND m.dm_target_id = ?))
-		 ORDER BY m.id ASC LIMIT ?`,
-		afterID, userA, userB, userB, userA, limit)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	var msgs []MessageInfo
-
-	for rows.Next() {
-		var m MessageInfo
-
-		err := rows.Scan(
-			&m.ID, &m.Nick, &m.Content,
-			&m.DMTarget, &m.CreatedAt,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		m.IsDM = true
-
-		msgs = append(msgs, m)
-	}
-
-	err = rows.Err()
-	if err != nil {
-		return nil, err
-	}
-
-	if msgs == nil {
-		msgs = []MessageInfo{}
-	}
-
-	return msgs, nil
-}
-
-// PollMessages returns all new messages (channel + DM) for
-// a user after a given ID.
-func (s *Database) PollMessages(
 	ctx context.Context,
 	userID int64,
-	afterID int64,
+) ([]int64, error) {
+	rows, err := s.db.QueryContext(ctx,
+		`SELECT channel_id FROM channel_members
+		 WHERE user_id = ?`, userID)
+	if err != nil {
+		return nil, err
+	}
+
+	return scanInt64s(rows)
+}
+
+// InsertMessage stores a message and returns its DB ID.
+func (s *Database) InsertMessage(
+	ctx context.Context,
+	command, from, to string,
+	body json.RawMessage,
+	meta json.RawMessage,
+) (int64, string, error) {
+	msgUUID := uuid.New().String()
+	now := time.Now().UTC()
+
+	if body == nil {
+		body = json.RawMessage("[]")
+	}
+
+	if meta == nil {
+		meta = json.RawMessage("{}")
+	}
+
+	res, err := s.db.ExecContext(ctx,
+		`INSERT INTO messages
+		 (uuid, command, msg_from, msg_to,
+		  body, meta, created_at)
+		 VALUES (?, ?, ?, ?, ?, ?, ?)`,
+		msgUUID, command, from, to,
+		string(body), string(meta), now)
+	if err != nil {
+		return 0, "", err
+	}
+
+	id, _ := res.LastInsertId()
+
+	return id, msgUUID, nil
+}
+
+// EnqueueMessage adds a message to a user's queue.
+func (s *Database) EnqueueMessage(
+	ctx context.Context,
+	userID, messageID int64,
+) error {
+	_, err := s.db.ExecContext(ctx,
+		`INSERT OR IGNORE INTO client_queues
+		 (user_id, message_id, created_at)
+		 VALUES (?, ?, ?)`,
+		userID, messageID, time.Now())
+
+	return err
+}
+
+// PollMessages returns queued messages for a user.
+func (s *Database) PollMessages(
+	ctx context.Context,
+	userID, afterQueueID int64,
 	limit int,
-) ([]MessageInfo, error) {
+) ([]IRCMessage, int64, error) {
 	if limit <= 0 {
 		limit = defaultPollLimit
 	}
 
 	rows, err := s.db.QueryContext(ctx,
-		`SELECT m.id, COALESCE(c.name, ''), u.nick, m.content,
+		`SELECT cq.id, m.uuid, m.command,
-		        m.is_dm, COALESCE(t.nick, ''), m.created_at
+		   m.msg_from, m.msg_to,
-		 FROM messages m
+		   m.body, m.meta, m.created_at
-		 INNER JOIN users u ON u.id = m.user_id
+		 FROM client_queues cq
-		 LEFT JOIN channels c ON c.id = m.channel_id
+		 INNER JOIN messages m
-		 LEFT JOIN users t ON t.id = m.dm_target_id
+		   ON m.id = cq.message_id
-		 WHERE m.id > ? AND (
+		 WHERE cq.user_id = ? AND cq.id > ?
-		   (m.is_dm = 0 AND m.channel_id IN
+		 ORDER BY cq.id ASC LIMIT ?`,
-		     (SELECT channel_id FROM channel_members
+		userID, afterQueueID, limit)
-		      WHERE user_id = ?))
+	if err != nil {
-		   OR (m.is_dm = 1
+		return nil, afterQueueID, err
-		     AND (m.user_id = ? OR m.dm_target_id = ?))
+	}
-		 )
+
-		 ORDER BY m.id ASC LIMIT ?`,
+	msgs, lastQID, scanErr := scanMessages(
-		afterID, userID, userID, userID, limit)
+		rows, afterQueueID,
+	)
+	if scanErr != nil {
+		return nil, afterQueueID, scanErr
+	}
+
+	return msgs, lastQID, nil
+}
+
+// GetHistory returns message history for a target.
+func (s *Database) GetHistory(
+	ctx context.Context,
+	target string,
+	beforeID int64,
+	limit int,
+) ([]IRCMessage, error) {
+	if limit <= 0 {
+		limit = defaultHistLimit
+	}
+
+	rows, err := s.queryHistory(
+		ctx, target, beforeID, limit,
+	)
 	if err != nil {
 		return nil, err
 	}
 
+	msgs, _, scanErr := scanMessages(rows, 0)
+	if scanErr != nil {
+		return nil, scanErr
+	}
+
+	if msgs == nil {
+		msgs = []IRCMessage{}
+	}
+
+	reverseMessages(msgs)
+
+	return msgs, nil
+}
+
+func (s *Database) queryHistory(
+	ctx context.Context,
+	target string,
+	beforeID int64,
+	limit int,
+) (*sql.Rows, error) {
+	if beforeID > 0 {
+		return s.db.QueryContext(ctx,
+			`SELECT id, uuid, command, msg_from,
+			   msg_to, body, meta, created_at
+			 FROM messages
+			 WHERE msg_to = ? AND id < ?
+			   AND command = 'PRIVMSG'
+			 ORDER BY id DESC LIMIT ?`,
+			target, beforeID, limit)
+	}
+
+	return s.db.QueryContext(ctx,
+		`SELECT id, uuid, command, msg_from,
+		   msg_to, body, meta, created_at
+		 FROM messages
+		 WHERE msg_to = ?
+		   AND command = 'PRIVMSG'
+		 ORDER BY id DESC LIMIT ?`,
+		target, limit)
+}
+
+func scanMessages(
+	rows *sql.Rows,
+	fallbackQID int64,
+) ([]IRCMessage, int64, error) {
 	defer func() { _ = rows.Close() }()
 
-	msgs := make([]MessageInfo, 0)
+	var msgs []IRCMessage
+
+	lastQID := fallbackQID
 
 	for rows.Next() {
 		var (
-			m    MessageInfo
+			m          IRCMessage
-			isDM int
+			qID        int64
+			body, meta string
+			ts         time.Time
 		)
 
 		err := rows.Scan(
-			&m.ID, &m.Channel, &m.Nick, &m.Content,
+			&qID, &m.ID, &m.Command,
-			&isDM, &m.DMTarget, &m.CreatedAt,
+			&m.From, &m.To,
+			&body, &meta, &ts,
 		)
 		if err != nil {
-			return nil, err
+			return nil, fallbackQID, err
 		}
 
-		m.IsDM = isDM == 1
+		m.Body = json.RawMessage(body)
-		msgs = append(msgs, m)
+		m.Meta = json.RawMessage(meta)
-	}
+		m.TS = ts.Format(time.RFC3339Nano)
-
+		m.DBID = qID
-	err = rows.Err()
+		lastQID = qID
-	if err != nil {
-		return nil, err
-	}
-
-	return msgs, nil
-}
-
-func scanChannelMessages(
-	rows *sql.Rows,
-) ([]MessageInfo, error) {
-	var msgs []MessageInfo
-
-	for rows.Next() {
-		var m MessageInfo
-
-		err := rows.Scan(
-			&m.ID, &m.Channel, &m.Nick,
-			&m.Content, &m.CreatedAt,
-		)
-		if err != nil {
-			return nil, err
-		}
 
 		msgs = append(msgs, m)
 	}
 
 	err := rows.Err()
 	if err != nil {
-		return nil, err
+		return nil, fallbackQID, err
 	}
 
 	if msgs == nil {
-		msgs = []MessageInfo{}
+		msgs = []IRCMessage{}
 	}
 
-	return msgs, nil
+	return msgs, lastQID, nil
 }
 
-func scanDMMessages(
+func reverseMessages(msgs []IRCMessage) {
-	rows *sql.Rows,
-) ([]MessageInfo, error) {
-	var msgs []MessageInfo
-
-	for rows.Next() {
-		var m MessageInfo
-
-		err := rows.Scan(
-			&m.ID, &m.Nick, &m.Content,
-			&m.DMTarget, &m.CreatedAt,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		m.IsDM = true
-
-		msgs = append(msgs, m)
-	}
-
-	err := rows.Err()
-	if err != nil {
-		return nil, err
-	}
-
-	if msgs == nil {
-		msgs = []MessageInfo{}
-	}
-
-	return msgs, nil
-}
-
-func reverseMessages(msgs []MessageInfo) {
 	for i, j := 0, len(msgs)-1; i < j; i, j = i+1, j-1 {
 		msgs[i], msgs[j] = msgs[j], msgs[i]
 	}
 }
 
-// GetMessagesBefore returns channel messages before a given
-// ID (for history scrollback).
-func (s *Database) GetMessagesBefore(
-	ctx context.Context,
-	channelID int64,
-	beforeID int64,
-	limit int,
-) ([]MessageInfo, error) {
-	if limit <= 0 {
-		limit = defaultMessageLimit
-	}
-
-	var query string
-
-	var args []any
-
-	if beforeID > 0 {
-		query = `SELECT m.id, c.name, u.nick, m.content,
-		                m.created_at
-			FROM messages m
-			INNER JOIN users u ON u.id = m.user_id
-			INNER JOIN channels c ON c.id = m.channel_id
-			WHERE m.channel_id = ? AND m.is_dm = 0
-			  AND m.id < ?
-			ORDER BY m.id DESC LIMIT ?`
-		args = []any{channelID, beforeID, limit}
-	} else {
-		query = `SELECT m.id, c.name, u.nick, m.content,
-		                m.created_at
-			FROM messages m
-			INNER JOIN users u ON u.id = m.user_id
-			INNER JOIN channels c ON c.id = m.channel_id
-			WHERE m.channel_id = ? AND m.is_dm = 0
-			ORDER BY m.id DESC LIMIT ?`
-		args = []any{channelID, limit}
-	}
-
-	rows, err := s.db.QueryContext(ctx, query, args...)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	msgs, scanErr := scanChannelMessages(rows)
-	if scanErr != nil {
-		return nil, scanErr
-	}
-
-	// Reverse to ascending order.
-	reverseMessages(msgs)
-
-	return msgs, nil
-}
-
-// GetDMsBefore returns DMs between two users before a given
-// ID (for history scrollback).
-func (s *Database) GetDMsBefore(
-	ctx context.Context,
-	userA, userB int64,
-	beforeID int64,
-	limit int,
-) ([]MessageInfo, error) {
-	if limit <= 0 {
-		limit = defaultMessageLimit
-	}
-
-	var query string
-
-	var args []any
-
-	if beforeID > 0 {
-		query = `SELECT m.id, u.nick, m.content, t.nick,
-		                m.created_at
-			FROM messages m
-			INNER JOIN users u ON u.id = m.user_id
-			INNER JOIN users t ON t.id = m.dm_target_id
-			WHERE m.is_dm = 1 AND m.id < ?
-			AND ((m.user_id = ? AND m.dm_target_id = ?)
-			  OR (m.user_id = ? AND m.dm_target_id = ?))
-			ORDER BY m.id DESC LIMIT ?`
-		args = []any{
-			beforeID, userA, userB, userB, userA, limit,
-		}
-	} else {
-		query = `SELECT m.id, u.nick, m.content, t.nick,
-		                m.created_at
-			FROM messages m
-			INNER JOIN users u ON u.id = m.user_id
-			INNER JOIN users t ON t.id = m.dm_target_id
-			WHERE m.is_dm = 1
-			AND ((m.user_id = ? AND m.dm_target_id = ?)
-			  OR (m.user_id = ? AND m.dm_target_id = ?))
-			ORDER BY m.id DESC LIMIT ?`
-		args = []any{userA, userB, userB, userA, limit}
-	}
-
-	rows, err := s.db.QueryContext(ctx, query, args...)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	msgs, scanErr := scanDMMessages(rows)
-	if scanErr != nil {
-		return nil, scanErr
-	}
-
-	// Reverse to ascending order.
-	reverseMessages(msgs)
-
-	return msgs, nil
-}
-
 // ChangeNick updates a user's nickname.
 func (s *Database) ChangeNick(
 	ctx context.Context,
@@ -654,58 +579,45 @@ func (s *Database) ChangeNick(
 // SetTopic sets the topic for a channel.
 func (s *Database) SetTopic(
 	ctx context.Context,
-	channelName string,
+	channelName, topic string,
-	_ int64,
-	topic string,
 ) error {
 	_, err := s.db.ExecContext(ctx,
-		"UPDATE channels SET topic = ? WHERE name = ?",
+		`UPDATE channels SET topic = ?,
-		topic, channelName)
+		 updated_at = ? WHERE name = ?`,
+		topic, time.Now(), channelName)
 
 	return err
 }
 
-// GetServerName returns the server name (unused, config
+// DeleteUser removes a user and all their data.
-// provides this).
+func (s *Database) DeleteUser(
-func (s *Database) GetServerName() string {
+	ctx context.Context,
-	return ""
+	userID int64,
+) error {
+	_, err := s.db.ExecContext(
+		ctx,
+		"DELETE FROM users WHERE id = ?",
+		userID,
+	)
+
+	return err
 }
 
-// ListAllChannels returns all channels.
+// GetAllChannelMembershipsForUser returns channels
-func (s *Database) ListAllChannels(
+// a user belongs to.
+func (s *Database) GetAllChannelMembershipsForUser(
 	ctx context.Context,
+	userID int64,
 ) ([]ChannelInfo, error) {
 	rows, err := s.db.QueryContext(ctx,
-		"SELECT id, name, topic FROM channels ORDER BY name")
+		`SELECT c.id, c.name, c.topic
+		 FROM channels c
+		 INNER JOIN channel_members cm
+		   ON cm.channel_id = c.id
+		 WHERE cm.user_id = ?`, userID)
 	if err != nil {
 		return nil, err
 	}
 
-	defer func() { _ = rows.Close() }()
+	return scanChannels(rows)
-
-	var channels []ChannelInfo
-
-	for rows.Next() {
-		var ch ChannelInfo
-
-		err := rows.Scan(
-			&ch.ID, &ch.Name, &ch.Topic,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		channels = append(channels, ch)
-	}
-
-	err = rows.Err()
-	if err != nil {
-		return nil, err
-	}
-
-	if channels == nil {
-		channels = []ChannelInfo{}
-	}
-
-	return channels, nil
 }

internal/db/queries_test.go

@@ -0,0 +1,549 @@
+package db_test
+
+import (
+	"encoding/json"
+	"testing"
+
+	"git.eeqj.de/sneak/chat/internal/db"
+
+	_ "modernc.org/sqlite"
+)
+
+func setupTestDB(t *testing.T) *db.Database {
+	t.Helper()
+
+	d, err := db.NewTestDatabase()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	t.Cleanup(func() {
+		closeErr := d.Close()
+		if closeErr != nil {
+			t.Logf("close db: %v", closeErr)
+		}
+	})
+
+	return d
+}
+
+func TestCreateUser(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	id, token, err := database.CreateUser(ctx, "alice")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if id == 0 || token == "" {
+		t.Fatal("expected valid id and token")
+	}
+
+	_, _, err = database.CreateUser(ctx, "alice")
+	if err == nil {
+		t.Fatal("expected error for duplicate nick")
+	}
+}
+
+func TestGetUserByToken(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	_, token, err := database.CreateUser(ctx, "bob")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	id, nick, err := database.GetUserByToken(ctx, token)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if nick != "bob" || id == 0 {
+		t.Fatalf("expected bob, got %s", nick)
+	}
+
+	_, _, err = database.GetUserByToken(ctx, "badtoken")
+	if err == nil {
+		t.Fatal("expected error for bad token")
+	}
+}
+
+func TestGetUserByNick(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	_, _, err := database.CreateUser(ctx, "charlie")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	id, err := database.GetUserByNick(ctx, "charlie")
+	if err != nil || id == 0 {
+		t.Fatal("expected to find charlie")
+	}
+
+	_, err = database.GetUserByNick(ctx, "nobody")
+	if err == nil {
+		t.Fatal("expected error for unknown nick")
+	}
+}
+
+func TestChannelOperations(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	chID, err := database.GetOrCreateChannel(ctx, "#test")
+	if err != nil || chID == 0 {
+		t.Fatal("expected channel id")
+	}
+
+	chID2, err := database.GetOrCreateChannel(ctx, "#test")
+	if err != nil || chID2 != chID {
+		t.Fatal("expected same channel id")
+	}
+
+	chID3, err := database.GetChannelByName(ctx, "#test")
+	if err != nil || chID3 != chID {
+		t.Fatal("expected same channel id")
+	}
+
+	_, err = database.GetChannelByName(ctx, "#nope")
+	if err == nil {
+		t.Fatal("expected error for nonexistent channel")
+	}
+}
+
+func TestJoinAndPart(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	uid, _, err := database.CreateUser(ctx, "user1")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	chID, err := database.GetOrCreateChannel(ctx, "#chan")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.JoinChannel(ctx, chID, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	ids, err := database.GetChannelMemberIDs(ctx, chID)
+	if err != nil || len(ids) != 1 || ids[0] != uid {
+		t.Fatal("expected user in channel")
+	}
+
+	err = database.JoinChannel(ctx, chID, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.PartChannel(ctx, chID, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	ids, _ = database.GetChannelMemberIDs(ctx, chID)
+	if len(ids) != 0 {
+		t.Fatal("expected empty channel")
+	}
+}
+
+func TestDeleteChannelIfEmpty(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	chID, err := database.GetOrCreateChannel(
+		ctx, "#empty",
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	uid, _, err := database.CreateUser(ctx, "temp")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.JoinChannel(ctx, chID, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.PartChannel(ctx, chID, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.DeleteChannelIfEmpty(ctx, chID)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = database.GetChannelByName(ctx, "#empty")
+	if err == nil {
+		t.Fatal("expected channel to be deleted")
+	}
+}
+
+func createUserWithChannels(
+	t *testing.T,
+	database *db.Database,
+	nick, ch1Name, ch2Name string,
+) (int64, int64, int64) {
+	t.Helper()
+
+	ctx := t.Context()
+
+	uid, _, err := database.CreateUser(ctx, nick)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	ch1, err := database.GetOrCreateChannel(
+		ctx, ch1Name,
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	ch2, err := database.GetOrCreateChannel(
+		ctx, ch2Name,
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.JoinChannel(ctx, ch1, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.JoinChannel(ctx, ch2, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	return uid, ch1, ch2
+}
+
+func TestListChannels(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	uid, _, _ := createUserWithChannels(
+		t, database, "lister", "#a", "#b",
+	)
+
+	channels, err := database.ListChannels(
+		t.Context(), uid,
+	)
+	if err != nil || len(channels) != 2 {
+		t.Fatalf(
+			"expected 2 channels, got %d",
+			len(channels),
+		)
+	}
+}
+
+func TestListAllChannels(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	_, err := database.GetOrCreateChannel(ctx, "#x")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = database.GetOrCreateChannel(ctx, "#y")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	channels, err := database.ListAllChannels(ctx)
+	if err != nil || len(channels) < 2 {
+		t.Fatalf(
+			"expected >= 2 channels, got %d",
+			len(channels),
+		)
+	}
+}
+
+func TestChangeNick(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	uid, token, err := database.CreateUser(ctx, "old")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.ChangeNick(ctx, uid, "new")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, nick, err := database.GetUserByToken(ctx, token)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if nick != "new" {
+		t.Fatalf("expected new, got %s", nick)
+	}
+}
+
+func TestSetTopic(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	_, err := database.GetOrCreateChannel(
+		ctx, "#topictest",
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.SetTopic(ctx, "#topictest", "Hello")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	channels, err := database.ListAllChannels(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	for _, ch := range channels {
+		if ch.Name == "#topictest" &&
+			ch.Topic != "Hello" {
+			t.Fatalf(
+				"expected topic Hello, got %s",
+				ch.Topic,
+			)
+		}
+	}
+}
+
+func TestInsertAndPollMessages(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	uid, _, err := database.CreateUser(ctx, "poller")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	body := json.RawMessage(`["hello"]`)
+
+	dbID, msgUUID, err := database.InsertMessage(
+		ctx, "PRIVMSG", "poller", "#test", body, nil,
+	)
+	if err != nil || dbID == 0 || msgUUID == "" {
+		t.Fatal("insert failed")
+	}
+
+	err = database.EnqueueMessage(ctx, uid, dbID)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	const batchSize = 10
+
+	msgs, lastQID, err := database.PollMessages(
+		ctx, uid, 0, batchSize,
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(msgs) != 1 {
+		t.Fatalf(
+			"expected 1 message, got %d", len(msgs),
+		)
+	}
+
+	if msgs[0].Command != "PRIVMSG" {
+		t.Fatalf(
+			"expected PRIVMSG, got %s", msgs[0].Command,
+		)
+	}
+
+	if lastQID == 0 {
+		t.Fatal("expected nonzero lastQID")
+	}
+
+	msgs, _, _ = database.PollMessages(
+		ctx, uid, lastQID, batchSize,
+	)
+
+	if len(msgs) != 0 {
+		t.Fatalf(
+			"expected 0 messages, got %d", len(msgs),
+		)
+	}
+}
+
+func TestGetHistory(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	const msgCount = 10
+
+	for range msgCount {
+		_, _, err := database.InsertMessage(
+			ctx, "PRIVMSG", "user", "#hist",
+			json.RawMessage(`["msg"]`), nil,
+		)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	const histLimit = 5
+
+	msgs, err := database.GetHistory(
+		ctx, "#hist", 0, histLimit,
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(msgs) != histLimit {
+		t.Fatalf("expected %d, got %d",
+			histLimit, len(msgs))
+	}
+
+	if msgs[0].DBID > msgs[histLimit-1].DBID {
+		t.Fatal("expected ascending order")
+	}
+}
+
+func TestDeleteUser(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	uid, _, err := database.CreateUser(ctx, "deleteme")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	chID, err := database.GetOrCreateChannel(
+		ctx, "#delchan",
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.JoinChannel(ctx, chID, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.DeleteUser(ctx, uid)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = database.GetUserByNick(ctx, "deleteme")
+	if err == nil {
+		t.Fatal("user should be deleted")
+	}
+
+	ids, _ := database.GetChannelMemberIDs(ctx, chID)
+	if len(ids) != 0 {
+		t.Fatal("expected no members after deletion")
+	}
+}
+
+func TestChannelMembers(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	ctx := t.Context()
+
+	uid1, _, err := database.CreateUser(ctx, "m1")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	uid2, _, err := database.CreateUser(ctx, "m2")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	chID, err := database.GetOrCreateChannel(
+		ctx, "#members",
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.JoinChannel(ctx, chID, uid1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = database.JoinChannel(ctx, chID, uid2)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	members, err := database.ChannelMembers(ctx, chID)
+	if err != nil || len(members) != 2 {
+		t.Fatalf(
+			"expected 2 members, got %d",
+			len(members),
+		)
+	}
+}
+
+func TestGetAllChannelMembershipsForUser(t *testing.T) {
+	t.Parallel()
+
+	database := setupTestDB(t)
+	uid, _, _ := createUserWithChannels(
+		t, database, "multi", "#m1", "#m2",
+	)
+
+	channels, err :=
+		database.GetAllChannelMembershipsForUser(
+			t.Context(), uid,
+		)
+	if err != nil || len(channels) != 2 {
+		t.Fatalf(
+			"expected 2 channels, got %d",
+			len(channels),
+		)
+	}
+}

internal/db/schema/001_initial.sql

@@ -1,4 +1,54 @@
-CREATE TABLE IF NOT EXISTS schema_migrations (
+-- Chat server schema (pre-1.0 consolidated)
-    version INTEGER PRIMARY KEY,
+PRAGMA foreign_keys = ON;
-    applied_at DATETIME DEFAULT CURRENT_TIMESTAMP
+
+-- Users: IRC-style sessions (no passwords, just nick + token)
+CREATE TABLE IF NOT EXISTS users (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    nick TEXT NOT NULL UNIQUE,
+    token TEXT NOT NULL UNIQUE,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    last_seen DATETIME DEFAULT CURRENT_TIMESTAMP
 );
+CREATE INDEX IF NOT EXISTS idx_users_token ON users(token);
+
+-- Channels
+CREATE TABLE IF NOT EXISTS channels (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT NOT NULL UNIQUE,
+    topic TEXT NOT NULL DEFAULT '',
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+-- Channel members
+CREATE TABLE IF NOT EXISTS channel_members (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    channel_id INTEGER NOT NULL REFERENCES channels(id) ON DELETE CASCADE,
+    user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
+    joined_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    UNIQUE(channel_id, user_id)
+);
+
+-- Messages: IRC envelope format
+CREATE TABLE IF NOT EXISTS messages (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    uuid TEXT NOT NULL UNIQUE,
+    command TEXT NOT NULL DEFAULT 'PRIVMSG',
+    msg_from TEXT NOT NULL DEFAULT '',
+    msg_to TEXT NOT NULL DEFAULT '',
+    body TEXT NOT NULL DEFAULT '[]',
+    meta TEXT NOT NULL DEFAULT '{}',
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+CREATE INDEX IF NOT EXISTS idx_messages_to_id ON messages(msg_to, id);
+CREATE INDEX IF NOT EXISTS idx_messages_created ON messages(created_at);
+
+-- Per-client message queues for fan-out delivery
+CREATE TABLE IF NOT EXISTS client_queues (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
+    message_id INTEGER NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    UNIQUE(user_id, message_id)
+);
+CREATE INDEX IF NOT EXISTS idx_client_queues_user ON client_queues(user_id, id);

internal/db/schema/002_schema.sql

@@ -1,89 +0,0 @@
--- All schema changes go into this file until 1.0.0 is tagged.
--- There will not be migrations during the early development phase.
--- After 1.0.0, new changes get their own numbered migration files.
-
--- Users: accounts and authentication
-CREATE TABLE IF NOT EXISTS users (
-    id TEXT PRIMARY KEY,                           -- UUID
-    nick TEXT NOT NULL UNIQUE,
-    password_hash TEXT NOT NULL,
-    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    last_seen_at DATETIME
-);
-
--- Auth tokens: one user can have multiple active tokens (multiple devices)
-CREATE TABLE IF NOT EXISTS auth_tokens (
-    token TEXT PRIMARY KEY,                        -- random token string
-    user_id TEXT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
-    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    expires_at DATETIME,                           -- NULL = no expiry
-    last_used_at DATETIME
-);
-CREATE INDEX IF NOT EXISTS idx_auth_tokens_user_id ON auth_tokens(user_id);
-
--- Channels: chat rooms
-CREATE TABLE IF NOT EXISTS channels (
-    id TEXT PRIMARY KEY,                           -- UUID
-    name TEXT NOT NULL UNIQUE,                     -- #general, etc.
-    topic TEXT NOT NULL DEFAULT '',
-    modes TEXT NOT NULL DEFAULT '',                -- +i, +m, +s, +t, +n
-    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
-);
-
--- Channel members: who is in which channel, with per-user modes
-CREATE TABLE IF NOT EXISTS channel_members (
-    channel_id TEXT NOT NULL REFERENCES channels(id) ON DELETE CASCADE,
-    user_id TEXT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
-    modes TEXT NOT NULL DEFAULT '',                -- +o (operator), +v (voice)
-    joined_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    PRIMARY KEY (channel_id, user_id)
-);
-CREATE INDEX IF NOT EXISTS idx_channel_members_user_id ON channel_members(user_id);
-
--- Messages: channel and DM history (rotated per MAX_HISTORY)
-CREATE TABLE IF NOT EXISTS messages (
-    id TEXT PRIMARY KEY,                           -- UUID
-    ts DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    from_user_id TEXT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
-    from_nick TEXT NOT NULL,                       -- denormalized for history
-    target TEXT NOT NULL,                          -- #channel name or user UUID for DMs
-    type TEXT NOT NULL DEFAULT 'message',          -- message, action, notice, join, part, quit, topic, mode, nick, system
-    body TEXT NOT NULL DEFAULT '',
-    meta TEXT NOT NULL DEFAULT '{}',               -- JSON extensible metadata
-    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
-);
-CREATE INDEX IF NOT EXISTS idx_messages_target_ts ON messages(target, ts);
-CREATE INDEX IF NOT EXISTS idx_messages_from_user ON messages(from_user_id);
-
--- Message queue: per-user pending delivery (unread messages)
-CREATE TABLE IF NOT EXISTS message_queue (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    user_id TEXT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
-    message_id TEXT NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
-    queued_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    UNIQUE(user_id, message_id)
-);
-CREATE INDEX IF NOT EXISTS idx_message_queue_user_id ON message_queue(user_id, queued_at);
-
--- Sessions: server-held session state
-CREATE TABLE IF NOT EXISTS sessions (
-    id TEXT PRIMARY KEY,                           -- UUID
-    user_id TEXT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
-    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    last_active_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    expires_at DATETIME                            -- idle timeout
-);
-CREATE INDEX IF NOT EXISTS idx_sessions_user_id ON sessions(user_id);
-
--- Server links: federation peer configuration
-CREATE TABLE IF NOT EXISTS server_links (
-    id TEXT PRIMARY KEY,                           -- UUID
-    name TEXT NOT NULL UNIQUE,                     -- human-readable peer name
-    url TEXT NOT NULL,                             -- base URL of peer server
-    shared_key_hash TEXT NOT NULL,                 -- hashed shared secret
-    is_active INTEGER NOT NULL DEFAULT 1,
-    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    last_seen_at DATETIME
-);

internal/db/schema/003_users.sql

@@ -1,53 +0,0 @@
--- Migration 003: Replace UUID-based tables with simple integer-keyed
--- tables for the HTTP API.  Drops the 002 tables and recreates them.
-
-PRAGMA foreign_keys = OFF;
-
-DROP TABLE IF EXISTS message_queue;
-DROP TABLE IF EXISTS sessions;
-DROP TABLE IF EXISTS server_links;
-DROP TABLE IF EXISTS messages;
-DROP TABLE IF EXISTS channel_members;
-DROP TABLE IF EXISTS auth_tokens;
-DROP TABLE IF EXISTS channels;
-DROP TABLE IF EXISTS users;
-
-CREATE TABLE users (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    nick TEXT NOT NULL UNIQUE,
-    token TEXT NOT NULL UNIQUE,
-    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-    last_seen DATETIME DEFAULT CURRENT_TIMESTAMP
-);
-
-CREATE TABLE channels (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    name TEXT NOT NULL UNIQUE,
-    topic TEXT NOT NULL DEFAULT '',
-    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
-);
-
-CREATE TABLE channel_members (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    channel_id INTEGER NOT NULL REFERENCES channels(id) ON DELETE CASCADE,
-    user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
-    joined_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-    UNIQUE(channel_id, user_id)
-);
-
-CREATE TABLE messages (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    channel_id INTEGER REFERENCES channels(id) ON DELETE CASCADE,
-    user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
-    content TEXT NOT NULL,
-    is_dm INTEGER NOT NULL DEFAULT 0,
-    dm_target_id INTEGER REFERENCES users(id) ON DELETE CASCADE,
-    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
-);
-
-CREATE INDEX idx_messages_channel ON messages(channel_id, created_at);
-CREATE INDEX idx_messages_dm ON messages(user_id, dm_target_id, created_at);
-CREATE INDEX idx_users_token ON users(token);
-
-PRAGMA foreign_keys = ON;

internal/handlers/handlers.go

@@ -4,9 +4,11 @@ package handlers
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"log/slog"
 	"net/http"
 
+	"git.eeqj.de/sneak/chat/internal/broker"
 	"git.eeqj.de/sneak/chat/internal/config"
 	"git.eeqj.de/sneak/chat/internal/db"
 	"git.eeqj.de/sneak/chat/internal/globals"
@@ -15,6 +17,8 @@ import (
 	"go.uber.org/fx"
 )
 
+var errUnauthorized = errors.New("unauthorized")
+
 // Params defines the dependencies for creating Handlers.
 type Params struct {
 	fx.In
@@ -31,14 +35,19 @@ type Handlers struct {
 	params *Params
 	log    *slog.Logger
 	hc     *healthcheck.Healthcheck
+	broker *broker.Broker
 }
 
 // New creates a new Handlers instance.
-func New(lc fx.Lifecycle, params Params) (*Handlers, error) {
+func New(
+	lc fx.Lifecycle,
+	params Params,
+) (*Handlers, error) {
 	s := new(Handlers)
 	s.params = &params
 	s.log = params.Logger.Get()
 	s.hc = params.Healthcheck
+	s.broker = broker.New()
 
 	lc.Append(fx.Hook{
 		OnStart: func(_ context.Context) error {
@@ -49,9 +58,17 @@ func New(lc fx.Lifecycle, params Params) (*Handlers, error) {
 	return s, nil
 }
 
-func (s *Handlers) respondJSON(w http.ResponseWriter, _ *http.Request, data any, status int) {
+func (s *Handlers) respondJSON(
+	w http.ResponseWriter,
+	_ *http.Request,
+	data any,
+	status int,
+) {
+	w.Header().Set(
+		"Content-Type",
+		"application/json; charset=utf-8",
+	)
 	w.WriteHeader(status)
-	w.Header().Set("Content-Type", "application/json")
 
 	if data != nil {
 		err := json.NewEncoder(w).Encode(data)

internal/models/auth_token.go

@@ -1,26 +0,0 @@
-package models
-
-import (
-	"context"
-	"time"
-)
-
-// AuthToken represents an authentication token for a user session.
-type AuthToken struct {
-	Base
-
-	Token      string     `json:"-"`
-	UserID     string     `json:"userId"`
-	CreatedAt  time.Time  `json:"createdAt"`
-	ExpiresAt  *time.Time `json:"expiresAt,omitempty"`
-	LastUsedAt *time.Time `json:"lastUsedAt,omitempty"`
-}
-
-// User returns the user who owns this token.
-func (t *AuthToken) User(ctx context.Context) (*User, error) {
-	if ul := t.GetUserLookup(); ul != nil {
-		return ul.GetUserByID(ctx, t.UserID)
-	}
-
-	return nil, ErrUserLookupNotAvailable
-}

internal/models/channel.go

@@ -1,96 +0,0 @@
-package models
-
-import (
-	"context"
-	"time"
-)
-
-// Channel represents a chat channel.
-type Channel struct {
-	Base
-
-	ID        string    `json:"id"`
-	Name      string    `json:"name"`
-	Topic     string    `json:"topic"`
-	Modes     string    `json:"modes"`
-	CreatedAt time.Time `json:"createdAt"`
-	UpdatedAt time.Time `json:"updatedAt"`
-}
-
-// Members returns all users who are members of this channel.
-func (c *Channel) Members(ctx context.Context) ([]*ChannelMember, error) {
-	rows, err := c.GetDB().QueryContext(ctx, `
-		SELECT cm.channel_id, cm.user_id, cm.modes, cm.joined_at,
-		       u.nick
-		FROM channel_members cm
-		JOIN users u ON u.id = cm.user_id
-		WHERE cm.channel_id = ?
-		ORDER BY cm.joined_at`,
-		c.ID,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	members := []*ChannelMember{}
-
-	for rows.Next() {
-		m := &ChannelMember{}
-		m.SetDB(c.db)
-
-		err = rows.Scan(
-			&m.ChannelID, &m.UserID, &m.Modes,
-			&m.JoinedAt, &m.Nick,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		members = append(members, m)
-	}
-
-	return members, rows.Err()
-}
-
-// RecentMessages returns the most recent messages in this channel.
-func (c *Channel) RecentMessages(
-	ctx context.Context,
-	limit int,
-) ([]*Message, error) {
-	rows, err := c.GetDB().QueryContext(ctx, `
-		SELECT id, ts, from_user_id, from_nick,
-		       target, type, body, meta, created_at
-		FROM messages
-		WHERE target = ?
-		ORDER BY ts DESC
-		LIMIT ?`,
-		c.Name, limit,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	messages := []*Message{}
-
-	for rows.Next() {
-		msg := &Message{}
-		msg.SetDB(c.db)
-
-		err = rows.Scan(
-			&msg.ID, &msg.Timestamp, &msg.FromUserID,
-			&msg.FromNick, &msg.Target, &msg.Type,
-			&msg.Body, &msg.Meta, &msg.CreatedAt,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		messages = append(messages, msg)
-	}
-
-	return messages, rows.Err()
-}

internal/models/channel_member.go

@@ -1,35 +0,0 @@
-package models
-
-import (
-	"context"
-	"time"
-)
-
-// ChannelMember represents a user's membership in a channel.
-type ChannelMember struct {
-	Base
-
-	ChannelID string    `json:"channelId"`
-	UserID    string    `json:"userId"`
-	Modes     string    `json:"modes"`
-	JoinedAt  time.Time `json:"joinedAt"`
-	Nick      string    `json:"nick"` // denormalized from users table
-}
-
-// User returns the full User for this membership.
-func (cm *ChannelMember) User(ctx context.Context) (*User, error) {
-	if ul := cm.GetUserLookup(); ul != nil {
-		return ul.GetUserByID(ctx, cm.UserID)
-	}
-
-	return nil, ErrUserLookupNotAvailable
-}
-
-// Channel returns the full Channel for this membership.
-func (cm *ChannelMember) Channel(ctx context.Context) (*Channel, error) {
-	if cl := cm.GetChannelLookup(); cl != nil {
-		return cl.GetChannelByID(ctx, cm.ChannelID)
-	}
-
-	return nil, ErrChannelLookupNotAvailable
-}

internal/models/message.go

@@ -1,20 +0,0 @@
-package models
-
-import (
-	"time"
-)
-
-// Message represents a chat message (channel or DM).
-type Message struct {
-	Base
-
-	ID         string    `json:"id"`
-	Timestamp  time.Time `json:"ts"`
-	FromUserID string    `json:"fromUserId"`
-	FromNick   string    `json:"from"`
-	Target     string    `json:"to"`
-	Type       string    `json:"type"`
-	Body       string    `json:"body"`
-	Meta       string    `json:"meta"`
-	CreatedAt  time.Time `json:"createdAt"`
-}

internal/models/message_queue.go

@@ -1,15 +0,0 @@
-package models
-
-import (
-	"time"
-)
-
-// MessageQueueEntry represents a pending message delivery for a user.
-type MessageQueueEntry struct {
-	Base
-
-	ID        int64     `json:"id"`
-	UserID    string    `json:"userId"`
-	MessageID string    `json:"messageId"`
-	QueuedAt  time.Time `json:"queuedAt"`
-}

internal/models/model.go

@@ -1,65 +0,0 @@
-// Package models defines the data models used by the chat application.
-// All model structs embed Base, which provides database access for
-// relation-fetching methods directly on model instances.
-package models
-
-import (
-	"context"
-	"database/sql"
-	"errors"
-)
-
-// DB is the interface that models use to query the database.
-// This avoids a circular import with the db package.
-type DB interface {
-	GetDB() *sql.DB
-}
-
-// UserLookup provides user lookup by ID without circular imports.
-type UserLookup interface {
-	GetUserByID(ctx context.Context, id string) (*User, error)
-}
-
-// ChannelLookup provides channel lookup by ID without circular imports.
-type ChannelLookup interface {
-	GetChannelByID(ctx context.Context, id string) (*Channel, error)
-}
-
-// Sentinel errors for model lookup methods.
-var (
-	ErrUserLookupNotAvailable    = errors.New("user lookup not available")
-	ErrChannelLookupNotAvailable = errors.New("channel lookup not available")
-)
-
-// Base is embedded in all model structs to provide database access.
-type Base struct {
-	db DB
-}
-
-// SetDB injects the database reference into a model.
-func (b *Base) SetDB(d DB) {
-	b.db = d
-}
-
-// GetDB returns the database interface for use in model methods.
-func (b *Base) GetDB() *sql.DB {
-	return b.db.GetDB()
-}
-
-// GetUserLookup returns the DB as a UserLookup if it implements the interface.
-func (b *Base) GetUserLookup() UserLookup { //nolint:ireturn
-	if ul, ok := b.db.(UserLookup); ok {
-		return ul
-	}
-
-	return nil
-}
-
-// GetChannelLookup returns the DB as a ChannelLookup if it implements the interface.
-func (b *Base) GetChannelLookup() ChannelLookup { //nolint:ireturn
-	if cl, ok := b.db.(ChannelLookup); ok {
-		return cl
-	}
-
-	return nil
-}

internal/models/server_link.go

@@ -1,18 +0,0 @@
-package models
-
-import (
-	"time"
-)
-
-// ServerLink represents a federation peer server configuration.
-type ServerLink struct {
-	Base
-
-	ID            string     `json:"id"`
-	Name          string     `json:"name"`
-	URL           string     `json:"url"`
-	SharedKeyHash string     `json:"-"`
-	IsActive      bool       `json:"isActive"`
-	CreatedAt     time.Time  `json:"createdAt"`
-	LastSeenAt    *time.Time `json:"lastSeenAt,omitempty"`
-}

internal/models/session.go

@@ -1,26 +0,0 @@
-package models
-
-import (
-	"context"
-	"time"
-)
-
-// Session represents a server-held user session.
-type Session struct {
-	Base
-
-	ID           string     `json:"id"`
-	UserID       string     `json:"userId"`
-	CreatedAt    time.Time  `json:"createdAt"`
-	LastActiveAt time.Time  `json:"lastActiveAt"`
-	ExpiresAt    *time.Time `json:"expiresAt,omitempty"`
-}
-
-// User returns the user who owns this session.
-func (s *Session) User(ctx context.Context) (*User, error) {
-	if ul := s.GetUserLookup(); ul != nil {
-		return ul.GetUserByID(ctx, s.UserID)
-	}
-
-	return nil, ErrUserLookupNotAvailable
-}

internal/models/user.go

@@ -1,92 +0,0 @@
-package models
-
-import (
-	"context"
-	"time"
-)
-
-// User represents a registered user account.
-type User struct {
-	Base
-
-	ID           string     `json:"id"`
-	Nick         string     `json:"nick"`
-	PasswordHash string     `json:"-"`
-	CreatedAt    time.Time  `json:"createdAt"`
-	UpdatedAt    time.Time  `json:"updatedAt"`
-	LastSeenAt   *time.Time `json:"lastSeenAt,omitempty"`
-}
-
-// Channels returns all channels the user is a member of.
-func (u *User) Channels(ctx context.Context) ([]*Channel, error) {
-	rows, err := u.GetDB().QueryContext(ctx, `
-		SELECT c.id, c.name, c.topic, c.modes, c.created_at, c.updated_at
-		FROM channels c
-		JOIN channel_members cm ON cm.channel_id = c.id
-		WHERE cm.user_id = ?
-		ORDER BY c.name`,
-		u.ID,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	channels := []*Channel{}
-
-	for rows.Next() {
-		c := &Channel{}
-		c.SetDB(u.db)
-
-		err = rows.Scan(
-			&c.ID, &c.Name, &c.Topic, &c.Modes,
-			&c.CreatedAt, &c.UpdatedAt,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		channels = append(channels, c)
-	}
-
-	return channels, rows.Err()
-}
-
-// QueuedMessages returns undelivered messages for this user.
-func (u *User) QueuedMessages(ctx context.Context) ([]*Message, error) {
-	rows, err := u.GetDB().QueryContext(ctx, `
-		SELECT m.id, m.ts, m.from_user_id, m.from_nick,
-		       m.target, m.type, m.body, m.meta, m.created_at
-		FROM messages m
-		JOIN message_queue mq ON mq.message_id = m.id
-		WHERE mq.user_id = ?
-		ORDER BY mq.queued_at ASC`,
-		u.ID,
-	)
-	if err != nil {
-		return nil, err
-	}
-
-	defer func() { _ = rows.Close() }()
-
-	messages := []*Message{}
-
-	for rows.Next() {
-		msg := &Message{}
-		msg.SetDB(u.db)
-
-		err = rows.Scan(
-			&msg.ID, &msg.Timestamp, &msg.FromUserID,
-			&msg.FromNick, &msg.Target, &msg.Type,
-			&msg.Body, &msg.Meta, &msg.CreatedAt,
-		)
-		if err != nil {
-			return nil, err
-		}
-
-		messages = append(messages, msg)
-	}
-
-	return messages, rows.Err()
-}

internal/server/http.go

@@ -4,6 +4,6 @@ import "time"
 
 const (
 	httpReadTimeout  = 10 * time.Second
-	httpWriteTimeout = 10 * time.Second
+	httpWriteTimeout = 60 * time.Second
 	maxHeaderBytes   = 1 << 20
 )

internal/server/routes.go

@@ -16,7 +16,7 @@ import (
 
 const routeTimeout = 60 * time.Second
 
-// SetupRoutes configures the HTTP routes and middleware chain.
+// SetupRoutes configures the HTTP routes and middleware.
 func (s *Server) SetupRoutes() {
 	s.router = chi.NewRouter()
 
@@ -39,13 +39,19 @@ func (s *Server) SetupRoutes() {
 	}
 
 	// Health check
-	s.router.Get("/.well-known/healthcheck.json", s.h.HandleHealthCheck())
+	s.router.Get(
+		"/.well-known/healthcheck.json",
+		s.h.HandleHealthCheck(),
+	)
 
 	// Protected metrics endpoint
 	if viper.GetString("METRICS_USERNAME") != "" {
 		s.router.Group(func(r chi.Router) {
 			r.Use(s.mw.MetricsAuth())
-			r.Get("/metrics", http.HandlerFunc(promhttp.Handler().ServeHTTP))
+			r.Get("/metrics",
+				http.HandlerFunc(
+					promhttp.Handler().ServeHTTP,
+				))
 		})
 	}
 
@@ -53,55 +59,66 @@ func (s *Server) SetupRoutes() {
 	s.router.Route("/api/v1", func(r chi.Router) {
 		r.Get("/server", s.h.HandleServerInfo())
 		r.Post("/session", s.h.HandleCreateSession())
-
-		// Unified state and message endpoints
 		r.Get("/state", s.h.HandleState())
 		r.Get("/messages", s.h.HandleGetMessages())
 		r.Post("/messages", s.h.HandleSendCommand())
 		r.Get("/history", s.h.HandleGetHistory())
-
-		// Channels
 		r.Get("/channels", s.h.HandleListAllChannels())
-		r.Get("/channels/{channel}/members", s.h.HandleChannelMembers())
+		r.Get(
+			"/channels/{channel}/members",
+			s.h.HandleChannelMembers(),
+		)
 	})
 
 	// Serve embedded SPA
+	s.setupSPA()
+}
+
+func (s *Server) setupSPA() {
 	distFS, err := fs.Sub(web.Dist, "dist")
 	if err != nil {
-		s.log.Error("failed to get web dist filesystem", "error", err)
+		s.log.Error(
-	} else {
+			"failed to get web dist filesystem",
-		fileServer := http.FileServer(http.FS(distFS))
+			"error", err,
-
+		)
-		s.router.Get("/*", func(w http.ResponseWriter, r *http.Request) {
-			s.serveSPA(distFS, fileServer, w, r)
-		})
-	}
-}
-
-func (s *Server) serveSPA(
-	distFS fs.FS,
-	fileServer http.Handler,
-	w http.ResponseWriter,
-	r *http.Request,
-) {
-	readFS, ok := distFS.(fs.ReadFileFS)
-	if !ok {
-		http.Error(w, "filesystem error", http.StatusInternalServerError)
 
 		return
 	}
 
-	// Try to serve the file; fall back to index.html for SPA routing.
+	fileServer := http.FileServer(http.FS(distFS))
-	f, err := readFS.ReadFile(r.URL.Path[1:])
-	if err != nil || len(f) == 0 {
-		indexHTML, _ := readFS.ReadFile("index.html")
 
-		w.Header().Set("Content-Type", "text/html; charset=utf-8")
+	s.router.Get("/*", func(
-		w.WriteHeader(http.StatusOK)
+		w http.ResponseWriter,
-		_, _ = w.Write(indexHTML)
+		r *http.Request,
+	) {
+		readFS, ok := distFS.(fs.ReadFileFS)
+		if !ok {
+			fileServer.ServeHTTP(w, r)
 
-		return
+			return
-	}
+		}
 
-	fileServer.ServeHTTP(w, r)
+		f, readErr := readFS.ReadFile(r.URL.Path[1:])
+		if readErr != nil || len(f) == 0 {
+			indexHTML, indexErr := readFS.ReadFile(
+				"index.html",
+			)
+			if indexErr != nil {
+				http.NotFound(w, r)
+
+				return
+			}
+
+			w.Header().Set(
+				"Content-Type",
+				"text/html; charset=utf-8",
+			)
+			w.WriteHeader(http.StatusOK)
+			_, _ = w.Write(indexHTML)
+
+			return
+		}
+
+		fileServer.ServeHTTP(w, r)
+	})
 }

internal/server/server.go

@@ -148,7 +148,9 @@ func (s *Server) cleanupForExit() {
 func (s *Server) cleanShutdown() {
 	s.exitCode = 0
 
-	ctxShutdown, shutdownCancel := context.WithTimeout(context.Background(), shutdownTimeout)
+	ctxShutdown, shutdownCancel := context.WithTimeout(
+		context.Background(), shutdownTimeout,
+	)
 
 	err := s.httpServer.Shutdown(ctxShutdown)
 	if err != nil {