Compare commits
1 Commits
fix/29-spl
...
15caf5c8d2
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
15caf5c8d2 |
@@ -1,9 +1,8 @@
|
|||||||
.git
|
bin/
|
||||||
*.md
|
|
||||||
!README.md
|
|
||||||
chatd
|
chatd
|
||||||
chat-cli
|
|
||||||
data.db
|
data.db
|
||||||
data.db-wal
|
|
||||||
data.db-shm
|
|
||||||
.env
|
.env
|
||||||
|
.git
|
||||||
|
*.test
|
||||||
|
*.out
|
||||||
|
debug.log
|
||||||
|
|||||||
@@ -1,12 +0,0 @@
|
|||||||
root = true
|
|
||||||
|
|
||||||
[*]
|
|
||||||
indent_style = space
|
|
||||||
indent_size = 4
|
|
||||||
end_of_line = lf
|
|
||||||
charset = utf-8
|
|
||||||
trim_trailing_whitespace = true
|
|
||||||
insert_final_newline = true
|
|
||||||
|
|
||||||
[Makefile]
|
|
||||||
indent_style = tab
|
|
||||||
@@ -1,9 +0,0 @@
|
|||||||
name: check
|
|
||||||
on: [push]
|
|
||||||
jobs:
|
|
||||||
check:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
# actions/checkout v4.2.2, 2026-02-22
|
|
||||||
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683
|
|
||||||
- run: docker build .
|
|
||||||
30
.gitignore
vendored
30
.gitignore
vendored
@@ -1,28 +1,7 @@
|
|||||||
# OS
|
|
||||||
.DS_Store
|
|
||||||
Thumbs.db
|
|
||||||
|
|
||||||
# Editors
|
|
||||||
*.swp
|
|
||||||
*.swo
|
|
||||||
*~
|
|
||||||
*.bak
|
|
||||||
.idea/
|
|
||||||
.vscode/
|
|
||||||
*.sublime-*
|
|
||||||
|
|
||||||
# Node
|
|
||||||
node_modules/
|
|
||||||
|
|
||||||
# Environment / secrets
|
|
||||||
.env
|
|
||||||
.env.*
|
|
||||||
*.pem
|
|
||||||
*.key
|
|
||||||
|
|
||||||
# Build artifacts
|
|
||||||
/chatd
|
/chatd
|
||||||
/bin/
|
/bin/
|
||||||
|
data.db
|
||||||
|
.env
|
||||||
*.exe
|
*.exe
|
||||||
*.dll
|
*.dll
|
||||||
*.so
|
*.so
|
||||||
@@ -30,9 +9,6 @@ node_modules/
|
|||||||
*.test
|
*.test
|
||||||
*.out
|
*.out
|
||||||
vendor/
|
vendor/
|
||||||
|
|
||||||
# Project
|
|
||||||
data.db
|
|
||||||
debug.log
|
debug.log
|
||||||
/chat-cli
|
|
||||||
web/node_modules/
|
web/node_modules/
|
||||||
|
chat-cli
|
||||||
|
|||||||
@@ -7,8 +7,15 @@ run:
|
|||||||
linters:
|
linters:
|
||||||
default: all
|
default: all
|
||||||
disable:
|
disable:
|
||||||
- wsl # Deprecated in v2, replaced by wsl_v5
|
# Genuinely incompatible with project patterns
|
||||||
settings:
|
- exhaustruct # Requires all struct fields
|
||||||
|
- depguard # Dependency allow/block lists
|
||||||
|
- godot # Requires comments to end with periods
|
||||||
|
- wsl # Deprecated, replaced by wsl_v5
|
||||||
|
- wrapcheck # Too verbose for internal packages
|
||||||
|
- varnamelen # Short names like db, id are idiomatic Go
|
||||||
|
|
||||||
|
linters-settings:
|
||||||
lll:
|
lll:
|
||||||
line-length: 88
|
line-length: 88
|
||||||
funlen:
|
funlen:
|
||||||
@@ -18,17 +25,6 @@ linters:
|
|||||||
max-complexity: 15
|
max-complexity: 15
|
||||||
dupl:
|
dupl:
|
||||||
threshold: 100
|
threshold: 100
|
||||||
gosec:
|
|
||||||
excludes:
|
|
||||||
- G704
|
|
||||||
depguard:
|
|
||||||
rules:
|
|
||||||
all:
|
|
||||||
deny:
|
|
||||||
- pkg: "io/ioutil"
|
|
||||||
desc: "Deprecated; use io and os packages."
|
|
||||||
- pkg: "math/rand$"
|
|
||||||
desc: "Use crypto/rand for security-sensitive code."
|
|
||||||
|
|
||||||
issues:
|
issues:
|
||||||
exclude-use-default: false
|
exclude-use-default: false
|
||||||
|
|||||||
40
Dockerfile
40
Dockerfile
@@ -1,42 +1,22 @@
|
|||||||
# Lint stage — fast feedback on formatting and lint issues
|
FROM golang:1.24-alpine AS builder
|
||||||
# golangci/golangci-lint:v2.1.6
|
|
||||||
FROM golangci/golangci-lint@sha256:568ee1c1c53493575fa9494e280e579ac9ca865787bafe4df3023ae59ecf299b AS lint
|
RUN apk add --no-cache git
|
||||||
|
|
||||||
WORKDIR /src
|
WORKDIR /src
|
||||||
COPY go.mod go.sum ./
|
|
||||||
RUN go mod download
|
|
||||||
COPY . .
|
|
||||||
RUN make fmt-check
|
|
||||||
RUN make lint
|
|
||||||
|
|
||||||
# Build stage — tests and compilation
|
|
||||||
# golang:1.24-alpine, 2026-02-26
|
|
||||||
FROM golang@sha256:8bee1901f1e530bfb4a7850aa7a479d17ae3a18beb6e09064ed54cfd245b7191 AS builder
|
|
||||||
WORKDIR /src
|
|
||||||
RUN apk add --no-cache git build-base make
|
|
||||||
|
|
||||||
# Force BuildKit to run the lint stage by creating a stage dependency
|
|
||||||
COPY --from=lint /src/go.sum /dev/null
|
|
||||||
|
|
||||||
COPY go.mod go.sum ./
|
COPY go.mod go.sum ./
|
||||||
RUN go mod download
|
RUN go mod download
|
||||||
|
|
||||||
COPY . .
|
COPY . .
|
||||||
|
|
||||||
RUN make test
|
|
||||||
|
|
||||||
# Build static binaries (no cgo needed at runtime — modernc.org/sqlite is pure Go)
|
|
||||||
ARG VERSION=dev
|
ARG VERSION=dev
|
||||||
RUN CGO_ENABLED=0 go build -trimpath -ldflags="-s -w -X main.Version=${VERSION}" -o /chatd ./cmd/chatd/
|
RUN go build -ldflags "-X main.Version=${VERSION}" -o /chatd ./cmd/chatd
|
||||||
RUN CGO_ENABLED=0 go build -trimpath -ldflags="-s -w" -o /chat-cli ./cmd/chat-cli/
|
|
||||||
|
|
||||||
# alpine:3.21, 2026-02-26
|
FROM alpine:3.21
|
||||||
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
|
COPY --from=builder /chatd /usr/local/bin/chatd
|
||||||
|
|
||||||
USER chat
|
WORKDIR /data
|
||||||
EXPOSE 8080
|
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"]
|
ENTRYPOINT ["chatd"]
|
||||||
|
|||||||
21
LICENSE
21
LICENSE
@@ -1,21 +0,0 @@
|
|||||||
MIT License
|
|
||||||
|
|
||||||
Copyright (c) 2025 sneak
|
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
||||||
of this software and associated documentation files (the "Software"), to deal
|
|
||||||
in the Software without restriction, including without limitation the rights
|
|
||||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
||||||
copies of the Software, and to permit persons to whom the Software is
|
|
||||||
furnished to do so, subject to the following conditions:
|
|
||||||
|
|
||||||
The above copyright notice and this permission notice shall be included in all
|
|
||||||
copies or substantial portions of the Software.
|
|
||||||
|
|
||||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
||||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
||||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
||||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
||||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
||||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
||||||
SOFTWARE.
|
|
||||||
26
Makefile
26
Makefile
@@ -1,4 +1,4 @@
|
|||||||
.PHONY: all build lint fmt fmt-check test check clean run debug docker hooks
|
.PHONY: all build lint fmt test check clean run debug
|
||||||
|
|
||||||
BINARY := chatd
|
BINARY := chatd
|
||||||
VERSION := $(shell git describe --tags --always --dirty 2>/dev/null || echo "dev")
|
VERSION := $(shell git describe --tags --always --dirty 2>/dev/null || echo "dev")
|
||||||
@@ -17,15 +17,18 @@ fmt:
|
|||||||
gofmt -s -w .
|
gofmt -s -w .
|
||||||
goimports -w .
|
goimports -w .
|
||||||
|
|
||||||
fmt-check:
|
|
||||||
@test -z "$$(gofmt -l .)" || (echo "Files not formatted:" && gofmt -l . && exit 1)
|
|
||||||
|
|
||||||
test:
|
test:
|
||||||
go test -timeout 30s -v -race -cover ./...
|
go test -v -race -cover ./...
|
||||||
|
|
||||||
# check runs all validation without making changes
|
# Check runs all validation without making changes
|
||||||
# Used by CI and Docker build — fails if anything is wrong
|
# Used by CI and Docker build — fails if anything is wrong
|
||||||
check: test lint fmt-check
|
check:
|
||||||
|
@echo "==> Checking formatting..."
|
||||||
|
@test -z "$$(gofmt -l .)" || (echo "Files not formatted:" && gofmt -l . && exit 1)
|
||||||
|
@echo "==> Running linter..."
|
||||||
|
golangci-lint run --config .golangci.yml ./...
|
||||||
|
@echo "==> Running tests..."
|
||||||
|
go test -v -race ./...
|
||||||
@echo "==> Building..."
|
@echo "==> Building..."
|
||||||
go build -ldflags "$(LDFLAGS)" -o /dev/null ./cmd/chatd
|
go build -ldflags "$(LDFLAGS)" -o /dev/null ./cmd/chatd
|
||||||
@echo "==> All checks passed!"
|
@echo "==> All checks passed!"
|
||||||
@@ -38,12 +41,3 @@ debug: build
|
|||||||
|
|
||||||
clean:
|
clean:
|
||||||
rm -rf bin/ chatd data.db
|
rm -rf bin/ chatd data.db
|
||||||
|
|
||||||
docker:
|
|
||||||
docker build -t chat .
|
|
||||||
|
|
||||||
hooks:
|
|
||||||
@printf '#!/bin/sh\nset -e\n' > .git/hooks/pre-commit
|
|
||||||
@printf 'go mod tidy\ngo fmt ./...\ngit diff --exit-code -- go.mod go.sum || { echo "go mod tidy changed files; please stage and retry"; exit 1; }\n' >> .git/hooks/pre-commit
|
|
||||||
@printf 'make check\n' >> .git/hooks/pre-commit
|
|
||||||
@chmod +x .git/hooks/pre-commit
|
|
||||||
|
|||||||
182
REPO_POLICIES.md
182
REPO_POLICIES.md
@@ -1,182 +0,0 @@
|
|||||||
---
|
|
||||||
title: Repository Policies
|
|
||||||
last_modified: 2026-02-22
|
|
||||||
---
|
|
||||||
|
|
||||||
This document covers repository structure, tooling, and workflow standards. Code
|
|
||||||
style conventions are in separate documents:
|
|
||||||
|
|
||||||
- [Code Styleguide](https://git.eeqj.de/sneak/prompts/raw/branch/main/prompts/CODE_STYLEGUIDE.md)
|
|
||||||
(general, bash, Docker)
|
|
||||||
- [Go](https://git.eeqj.de/sneak/prompts/raw/branch/main/prompts/CODE_STYLEGUIDE_GO.md)
|
|
||||||
- [JavaScript](https://git.eeqj.de/sneak/prompts/raw/branch/main/prompts/CODE_STYLEGUIDE_JS.md)
|
|
||||||
- [Python](https://git.eeqj.de/sneak/prompts/raw/branch/main/prompts/CODE_STYLEGUIDE_PYTHON.md)
|
|
||||||
- [Go HTTP Server Conventions](https://git.eeqj.de/sneak/prompts/raw/branch/main/prompts/GO_HTTP_SERVER_CONVENTIONS.md)
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
- Cross-project documentation (such as this file) must include
|
|
||||||
`last_modified: YYYY-MM-DD` in the YAML front matter so it can be kept in sync
|
|
||||||
with the authoritative source as policies evolve.
|
|
||||||
|
|
||||||
- **ALL external references must be pinned by cryptographic hash.** This
|
|
||||||
includes Docker base images, Go modules, npm packages, GitHub Actions, and
|
|
||||||
anything else fetched from a remote source. Version tags (`@v4`, `@latest`,
|
|
||||||
`:3.21`, etc.) are server-mutable and therefore remote code execution
|
|
||||||
vulnerabilities. The ONLY acceptable way to reference an external dependency
|
|
||||||
is by its content hash (Docker `@sha256:...`, Go module hash in `go.sum`, npm
|
|
||||||
integrity hash in lockfile, GitHub Actions `@<commit-sha>`). No exceptions.
|
|
||||||
This also means never `curl | bash` to install tools like pyenv, nvm, rustup,
|
|
||||||
etc. Instead, download a specific release archive from GitHub, verify its hash
|
|
||||||
(hardcoded in the Dockerfile or script), and only then install. Unverified
|
|
||||||
install scripts are arbitrary remote code execution. This is the single most
|
|
||||||
important rule in this document. Double-check every external reference in
|
|
||||||
every file before committing. There are zero exceptions to this rule.
|
|
||||||
|
|
||||||
- Every repo with software must have a root `Makefile` with these targets:
|
|
||||||
`make test`, `make lint`, `make fmt` (writes), `make fmt-check` (read-only),
|
|
||||||
`make check` (prereqs: `test`, `lint`, `fmt-check`), `make docker`, and
|
|
||||||
`make hooks` (installs pre-commit hook). A model Makefile is at
|
|
||||||
`https://git.eeqj.de/sneak/prompts/raw/branch/main/Makefile`.
|
|
||||||
|
|
||||||
- Always use Makefile targets (`make fmt`, `make test`, `make lint`, etc.)
|
|
||||||
instead of invoking the underlying tools directly. The Makefile is the single
|
|
||||||
source of truth for how these operations are run.
|
|
||||||
|
|
||||||
- The Makefile is authoritative documentation for how the repo is used. Beyond
|
|
||||||
the required targets above, it should have targets for every common operation:
|
|
||||||
running a local development server (`make run`, `make dev`), re-initializing
|
|
||||||
or migrating the database (`make db-reset`, `make migrate`), building
|
|
||||||
artifacts (`make build`), generating code, seeding data, or anything else a
|
|
||||||
developer would do regularly. If someone checks out the repo and types
|
|
||||||
`make<tab>`, they should see every meaningful operation available. A new
|
|
||||||
contributor should be able to understand the entire development workflow by
|
|
||||||
reading the Makefile.
|
|
||||||
|
|
||||||
- Every repo should have a `Dockerfile`. All Dockerfiles must run `make check`
|
|
||||||
as a build step so the build fails if the branch is not green. For non-server
|
|
||||||
repos, the Dockerfile should bring up a development environment and run
|
|
||||||
`make check`. For server repos, `make check` should run as an early build
|
|
||||||
stage before the final image is assembled.
|
|
||||||
|
|
||||||
- Every repo should have a Gitea Actions workflow (`.gitea/workflows/`) that
|
|
||||||
runs `docker build .` on push. Since the Dockerfile already runs `make check`,
|
|
||||||
a successful build implies all checks pass.
|
|
||||||
|
|
||||||
- Use platform-standard formatters: `black` for Python, `prettier` for
|
|
||||||
JS/CSS/Markdown/HTML, `go fmt` for Go. Always use default configuration with
|
|
||||||
two exceptions: four-space indents (except Go), and `proseWrap: always` for
|
|
||||||
Markdown (hard-wrap at 80 columns). Documentation and writing repos (Markdown,
|
|
||||||
HTML, CSS) should also have `.prettierrc` and `.prettierignore`.
|
|
||||||
|
|
||||||
- Pre-commit hook: `make check` if local testing is possible, otherwise
|
|
||||||
`make lint && make fmt-check`. The Makefile should provide a `make hooks`
|
|
||||||
target to install the pre-commit hook.
|
|
||||||
|
|
||||||
- All repos with software must have tests that run via the platform-standard
|
|
||||||
test framework (`go test`, `pytest`, `jest`/`vitest`, etc.). If no meaningful
|
|
||||||
tests exist yet, add the most minimal test possible — e.g. importing the
|
|
||||||
module under test to verify it compiles/parses. There is no excuse for
|
|
||||||
`make test` to be a no-op.
|
|
||||||
|
|
||||||
- `make test` must complete in under 20 seconds. Add a 30-second timeout in the
|
|
||||||
Makefile.
|
|
||||||
|
|
||||||
- Docker builds must complete in under 5 minutes.
|
|
||||||
|
|
||||||
- `make check` must not modify any files in the repo. Tests may use temporary
|
|
||||||
directories.
|
|
||||||
|
|
||||||
- `main` must always pass `make check`, no exceptions.
|
|
||||||
|
|
||||||
- Never commit secrets. `.env` files, credentials, API keys, and private keys
|
|
||||||
must be in `.gitignore`. No exceptions.
|
|
||||||
|
|
||||||
- `.gitignore` should be comprehensive from the start: OS files (`.DS_Store`),
|
|
||||||
editor files (`.swp`, `*~`), language build artifacts, and `node_modules/`.
|
|
||||||
Fetch the standard `.gitignore` from
|
|
||||||
`https://git.eeqj.de/sneak/prompts/raw/branch/main/.gitignore` when setting up
|
|
||||||
a new repo.
|
|
||||||
|
|
||||||
- Never use `git add -A` or `git add .`. Always stage files explicitly by name.
|
|
||||||
|
|
||||||
- Never force-push to `main`.
|
|
||||||
|
|
||||||
- Make all changes on a feature branch. You can do whatever you want on a
|
|
||||||
feature branch.
|
|
||||||
|
|
||||||
- `.golangci.yml` is standardized and must _NEVER_ be modified by an agent, only
|
|
||||||
manually by the user. Fetch from
|
|
||||||
`https://git.eeqj.de/sneak/prompts/raw/branch/main/.golangci.yml`.
|
|
||||||
|
|
||||||
- When pinning images or packages by hash, add a comment above the reference
|
|
||||||
with the version and date (YYYY-MM-DD).
|
|
||||||
|
|
||||||
- Use `yarn`, not `npm`.
|
|
||||||
|
|
||||||
- Write all dates as YYYY-MM-DD (ISO 8601).
|
|
||||||
|
|
||||||
- Simple projects should be configured with environment variables.
|
|
||||||
|
|
||||||
- Dockerized web services listen on port 8080 by default, overridable with
|
|
||||||
`PORT`.
|
|
||||||
|
|
||||||
- `README.md` is the primary documentation. Required sections:
|
|
||||||
- **Description**: First line must include the project name, purpose,
|
|
||||||
category (web server, SPA, CLI tool, etc.), license, and author. Example:
|
|
||||||
"µPaaS is an MIT-licensed Go web application by @sneak that receives
|
|
||||||
git-frontend webhooks and deploys applications via Docker in realtime."
|
|
||||||
- **Getting Started**: Copy-pasteable install/usage code block.
|
|
||||||
- **Rationale**: Why does this exist?
|
|
||||||
- **Design**: How is the program structured?
|
|
||||||
- **TODO**: Update meticulously, even between commits. When planning, put
|
|
||||||
the todo list in the README so a new agent can pick up where the last one
|
|
||||||
left off.
|
|
||||||
- **License**: MIT, GPL, or WTFPL. Ask the user for new projects. Include a
|
|
||||||
`LICENSE` file in the repo root and a License section in the README.
|
|
||||||
- **Author**: [@sneak](https://sneak.berlin).
|
|
||||||
|
|
||||||
- First commit of a new repo should contain only `README.md`.
|
|
||||||
|
|
||||||
- Go module root: `sneak.berlin/go/<name>`. Always run `go mod tidy` before
|
|
||||||
committing.
|
|
||||||
|
|
||||||
- Use SemVer.
|
|
||||||
|
|
||||||
- Database migrations live in `internal/db/migrations/` and must be embedded in
|
|
||||||
the binary. Pre-1.0.0: modify existing migrations (no installed base assumed).
|
|
||||||
Post-1.0.0: add new migration files.
|
|
||||||
|
|
||||||
- All repos should have an `.editorconfig` enforcing the project's indentation
|
|
||||||
settings.
|
|
||||||
|
|
||||||
- Avoid putting files in the repo root unless necessary. Root should contain
|
|
||||||
only project-level config files (`README.md`, `Makefile`, `Dockerfile`,
|
|
||||||
`LICENSE`, `.gitignore`, `.editorconfig`, `REPO_POLICIES.md`, and
|
|
||||||
language-specific config). Everything else goes in a subdirectory. Canonical
|
|
||||||
subdirectory names:
|
|
||||||
- `bin/` — executable scripts and tools
|
|
||||||
- `cmd/` — Go command entrypoints
|
|
||||||
- `configs/` — configuration templates and examples
|
|
||||||
- `deploy/` — deployment manifests (k8s, compose, terraform)
|
|
||||||
- `docs/` — documentation and markdown (README.md stays in root)
|
|
||||||
- `internal/` — Go internal packages
|
|
||||||
- `internal/db/migrations/` — database migrations
|
|
||||||
- `pkg/` — Go library packages
|
|
||||||
- `share/` — systemd units, data files
|
|
||||||
- `static/` — static assets (images, fonts, etc.)
|
|
||||||
- `web/` — web frontend source
|
|
||||||
|
|
||||||
- When setting up a new repo, files from the `prompts` repo may be used as
|
|
||||||
templates. Fetch them from
|
|
||||||
`https://git.eeqj.de/sneak/prompts/raw/branch/main/<path>`.
|
|
||||||
|
|
||||||
- New repos must contain at minimum:
|
|
||||||
- `README.md`, `.git`, `.gitignore`, `.editorconfig`
|
|
||||||
- `LICENSE`, `REPO_POLICIES.md` (copy from the `prompts` repo)
|
|
||||||
- `Makefile`
|
|
||||||
- `Dockerfile`, `.dockerignore`
|
|
||||||
- `.gitea/workflows/check.yml`
|
|
||||||
- Go: `go.mod`, `go.sum`, `.golangci.yml`
|
|
||||||
- JS: `package.json`, `yarn.lock`, `.prettierrc`, `.prettierignore`
|
|
||||||
- Python: `pyproject.toml`
|
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
// Package chatapi provides a client for the chat server API.
|
// Package chatapi provides a client for the chat server's HTTP API.
|
||||||
package chatapi
|
package chatapi
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@@ -11,17 +11,20 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
httpTimeout = 30 * time.Second
|
httpClientTimeout = 30
|
||||||
pollExtraTime = 5
|
httpErrorMinCode = 400
|
||||||
httpErrThreshold = 400
|
pollExtraTimeout = 5
|
||||||
)
|
)
|
||||||
|
|
||||||
var errHTTP = errors.New("HTTP error")
|
// ErrHTTPError is returned when the server responds with an error status.
|
||||||
|
var ErrHTTPError = errors.New("HTTP error")
|
||||||
|
|
||||||
|
// ErrUnexpectedFormat is returned when a response has an unexpected structure.
|
||||||
|
var ErrUnexpectedFormat = errors.New("unexpected format")
|
||||||
|
|
||||||
// Client wraps HTTP calls to the chat server API.
|
// Client wraps HTTP calls to the chat server API.
|
||||||
type Client struct {
|
type Client struct {
|
||||||
@@ -32,23 +35,17 @@ type Client struct {
|
|||||||
|
|
||||||
// NewClient creates a new API client.
|
// NewClient creates a new API client.
|
||||||
func NewClient(baseURL string) *Client {
|
func NewClient(baseURL string) *Client {
|
||||||
return &Client{ //nolint:exhaustruct // Token set after CreateSession
|
return &Client{
|
||||||
BaseURL: baseURL,
|
BaseURL: baseURL,
|
||||||
HTTPClient: &http.Client{ //nolint:exhaustruct // defaults fine
|
HTTPClient: &http.Client{
|
||||||
Timeout: httpTimeout,
|
Timeout: httpClientTimeout * time.Second,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// CreateSession creates a new session on the server.
|
// CreateSession creates a new session on the server.
|
||||||
func (client *Client) CreateSession(
|
func (c *Client) CreateSession(nick string) (*SessionResponse, error) {
|
||||||
nick string,
|
data, err := c.do("POST", "/api/v1/session", &SessionRequest{Nick: nick})
|
||||||
) (*SessionResponse, error) {
|
|
||||||
data, err := client.do(
|
|
||||||
http.MethodPost,
|
|
||||||
"/api/v1/session",
|
|
||||||
&SessionRequest{Nick: nick},
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -60,16 +57,14 @@ func (client *Client) CreateSession(
|
|||||||
return nil, fmt.Errorf("decode session: %w", err)
|
return nil, fmt.Errorf("decode session: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
client.Token = resp.Token
|
c.Token = resp.Token
|
||||||
|
|
||||||
return &resp, nil
|
return &resp, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetState returns the current user state.
|
// GetState returns the current user state.
|
||||||
func (client *Client) GetState() (*StateResponse, error) {
|
func (c *Client) GetState() (*StateResponse, error) {
|
||||||
data, err := client.do(
|
data, err := c.do("GET", "/api/v1/state", nil)
|
||||||
http.MethodGet, "/api/v1/state", nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -85,110 +80,84 @@ func (client *Client) GetState() (*StateResponse, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// SendMessage sends a message (any IRC command).
|
// SendMessage sends a message (any IRC command).
|
||||||
func (client *Client) SendMessage(msg *Message) error {
|
func (c *Client) SendMessage(msg *Message) error {
|
||||||
_, err := client.do(
|
_, err := c.do("POST", "/api/v1/messages", msg)
|
||||||
http.MethodPost, "/api/v1/messages", msg,
|
|
||||||
)
|
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// PollMessages long-polls for new messages.
|
// PollMessages long-polls for new messages.
|
||||||
func (client *Client) PollMessages(
|
func (c *Client) PollMessages(afterID string, timeout int) ([]Message, error) {
|
||||||
afterID int64,
|
// Use a longer HTTP timeout than the server long-poll timeout.
|
||||||
timeout int,
|
client := &http.Client{Timeout: time.Duration(timeout+pollExtraTimeout) * time.Second}
|
||||||
) (*PollResult, error) {
|
|
||||||
pollClient := &http.Client{ //nolint:exhaustruct // defaults fine
|
|
||||||
Timeout: time.Duration(
|
|
||||||
timeout+pollExtraTime,
|
|
||||||
) * time.Second,
|
|
||||||
}
|
|
||||||
|
|
||||||
params := url.Values{}
|
params := url.Values{}
|
||||||
if afterID > 0 {
|
if afterID != "" {
|
||||||
params.Set(
|
params.Set("after", afterID)
|
||||||
"after",
|
|
||||||
strconv.FormatInt(afterID, 10),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
params.Set("timeout", strconv.Itoa(timeout))
|
params.Set("timeout", strconv.Itoa(timeout))
|
||||||
|
|
||||||
path := "/api/v1/messages?" + params.Encode()
|
path := "/api/v1/messages"
|
||||||
|
if len(params) > 0 {
|
||||||
request, err := http.NewRequestWithContext(
|
path += "?" + params.Encode()
|
||||||
context.Background(),
|
|
||||||
http.MethodGet,
|
|
||||||
client.BaseURL+path,
|
|
||||||
nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("new request: %w", err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
request.Header.Set(
|
req, err := http.NewRequestWithContext(context.Background(), http.MethodGet, c.BaseURL+path, nil)
|
||||||
"Authorization", "Bearer "+client.Token,
|
|
||||||
)
|
|
||||||
|
|
||||||
resp, err := pollClient.Do(request)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("poll request: %w", err)
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
req.Header.Set("Authorization", "Bearer "+c.Token)
|
||||||
|
|
||||||
|
resp, err := client.Do(req) //nolint:gosec // URL is constructed from trusted base URL + API path, not user-tainted
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
defer func() { _ = resp.Body.Close() }()
|
defer func() { _ = resp.Body.Close() }()
|
||||||
|
|
||||||
data, err := io.ReadAll(resp.Body)
|
data, err := io.ReadAll(resp.Body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("read poll body: %w", err)
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if resp.StatusCode >= httpErrThreshold {
|
if resp.StatusCode >= httpErrorMinCode {
|
||||||
return nil, fmt.Errorf(
|
return nil, fmt.Errorf("%w: %d: %s", ErrHTTPError, resp.StatusCode, string(data))
|
||||||
"%w %d: %s",
|
|
||||||
errHTTP, resp.StatusCode, string(data),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// The server may return an array directly or wrapped.
|
||||||
|
var msgs []Message
|
||||||
|
|
||||||
|
err = json.Unmarshal(data, &msgs)
|
||||||
|
if err != nil {
|
||||||
|
// Try wrapped format.
|
||||||
var wrapped MessagesResponse
|
var wrapped MessagesResponse
|
||||||
|
|
||||||
err = json.Unmarshal(data, &wrapped)
|
err2 := json.Unmarshal(data, &wrapped)
|
||||||
if err != nil {
|
if err2 != nil {
|
||||||
return nil, fmt.Errorf(
|
return nil, fmt.Errorf("decode messages: %w (raw: %s)", err, string(data))
|
||||||
"decode messages: %w", err,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return &PollResult{
|
msgs = wrapped.Messages
|
||||||
Messages: wrapped.Messages,
|
}
|
||||||
LastID: wrapped.LastID,
|
|
||||||
}, nil
|
return msgs, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// JoinChannel joins a channel.
|
// JoinChannel joins a channel via the unified command endpoint.
|
||||||
func (client *Client) JoinChannel(channel string) error {
|
func (c *Client) JoinChannel(channel string) error {
|
||||||
return client.SendMessage(
|
return c.SendMessage(&Message{Command: "JOIN", To: channel})
|
||||||
&Message{ //nolint:exhaustruct // only command+to needed
|
|
||||||
Command: "JOIN", To: channel,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// PartChannel leaves a channel.
|
// PartChannel leaves a channel via the unified command endpoint.
|
||||||
func (client *Client) PartChannel(channel string) error {
|
func (c *Client) PartChannel(channel string) error {
|
||||||
return client.SendMessage(
|
return c.SendMessage(&Message{Command: "PART", To: channel})
|
||||||
&Message{ //nolint:exhaustruct // only command+to needed
|
|
||||||
Command: "PART", To: channel,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ListChannels returns all channels on the server.
|
// ListChannels returns all channels on the server.
|
||||||
func (client *Client) ListChannels() (
|
func (c *Client) ListChannels() ([]Channel, error) {
|
||||||
[]Channel, error,
|
data, err := c.do("GET", "/api/v1/channels", nil)
|
||||||
) {
|
|
||||||
data, err := client.do(
|
|
||||||
http.MethodGet, "/api/v1/channels", nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -197,26 +166,15 @@ func (client *Client) ListChannels() (
|
|||||||
|
|
||||||
err = json.Unmarshal(data, &channels)
|
err = json.Unmarshal(data, &channels)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf(
|
return nil, err
|
||||||
"decode channels: %w", err,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return channels, nil
|
return channels, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetMembers returns members of a channel.
|
// GetMembers returns members of a channel.
|
||||||
func (client *Client) GetMembers(
|
func (c *Client) GetMembers(channel string) ([]string, error) {
|
||||||
channel string,
|
data, err := c.do("GET", "/api/v1/channels/"+url.PathEscape(channel)+"/members", nil)
|
||||||
) ([]string, error) {
|
|
||||||
name := strings.TrimPrefix(channel, "#")
|
|
||||||
|
|
||||||
data, err := client.do(
|
|
||||||
http.MethodGet,
|
|
||||||
"/api/v1/channels/"+url.PathEscape(name)+
|
|
||||||
"/members",
|
|
||||||
nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -225,21 +183,24 @@ func (client *Client) GetMembers(
|
|||||||
|
|
||||||
err = json.Unmarshal(data, &members)
|
err = json.Unmarshal(data, &members)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf(
|
// Try object format.
|
||||||
"unexpected members format: %w", err,
|
var obj map[string]any
|
||||||
)
|
|
||||||
|
err2 := json.Unmarshal(data, &obj)
|
||||||
|
if err2 != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extract member names from whatever format.
|
||||||
|
return nil, fmt.Errorf("%w: members: %s", ErrUnexpectedFormat, string(data))
|
||||||
}
|
}
|
||||||
|
|
||||||
return members, nil
|
return members, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetServerInfo returns server info.
|
// GetServerInfo returns server info.
|
||||||
func (client *Client) GetServerInfo() (
|
func (c *Client) GetServerInfo() (*ServerInfo, error) {
|
||||||
*ServerInfo, error,
|
data, err := c.do("GET", "/api/v1/server", nil)
|
||||||
) {
|
|
||||||
data, err := client.do(
|
|
||||||
http.MethodGet, "/api/v1/server", nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -248,18 +209,13 @@ func (client *Client) GetServerInfo() (
|
|||||||
|
|
||||||
err = json.Unmarshal(data, &info)
|
err = json.Unmarshal(data, &info)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf(
|
return nil, err
|
||||||
"decode server info: %w", err,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return &info, nil
|
return &info, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (client *Client) do(
|
func (c *Client) do(method, path string, body any) ([]byte, error) {
|
||||||
method, path string,
|
|
||||||
body any,
|
|
||||||
) ([]byte, error) {
|
|
||||||
var bodyReader io.Reader
|
var bodyReader io.Reader
|
||||||
|
|
||||||
if body != nil {
|
if body != nil {
|
||||||
@@ -271,27 +227,19 @@ func (client *Client) do(
|
|||||||
bodyReader = bytes.NewReader(data)
|
bodyReader = bytes.NewReader(data)
|
||||||
}
|
}
|
||||||
|
|
||||||
request, err := http.NewRequestWithContext(
|
req, err := http.NewRequestWithContext(context.Background(), method, c.BaseURL+path, bodyReader)
|
||||||
context.Background(),
|
|
||||||
method,
|
|
||||||
client.BaseURL+path,
|
|
||||||
bodyReader,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("request: %w", err)
|
return nil, fmt.Errorf("request: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
request.Header.Set(
|
req.Header.Set("Content-Type", "application/json")
|
||||||
"Content-Type", "application/json",
|
|
||||||
)
|
|
||||||
|
|
||||||
if client.Token != "" {
|
if c.Token != "" {
|
||||||
request.Header.Set(
|
req.Header.Set("Authorization", "Bearer "+c.Token)
|
||||||
"Authorization", "Bearer "+client.Token,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
resp, err := client.HTTPClient.Do(request)
|
//nolint:gosec // URL built from trusted base + API path
|
||||||
|
resp, err := c.HTTPClient.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("http: %w", err)
|
return nil, fmt.Errorf("http: %w", err)
|
||||||
}
|
}
|
||||||
@@ -303,11 +251,8 @@ func (client *Client) do(
|
|||||||
return nil, fmt.Errorf("read body: %w", err)
|
return nil, fmt.Errorf("read body: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if resp.StatusCode >= httpErrThreshold {
|
if resp.StatusCode >= httpErrorMinCode {
|
||||||
return data, fmt.Errorf(
|
return data, fmt.Errorf("%w: %d: %s", ErrHTTPError, resp.StatusCode, string(data))
|
||||||
"%w %d: %s",
|
|
||||||
errHTTP, resp.StatusCode, string(data),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return data, nil
|
return data, nil
|
||||||
|
|||||||
@@ -7,16 +7,18 @@ type SessionRequest struct {
|
|||||||
Nick string `json:"nick"`
|
Nick string `json:"nick"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// SessionResponse is the response from session creation.
|
// SessionResponse is the response from POST /api/v1/session.
|
||||||
type SessionResponse struct {
|
type SessionResponse struct {
|
||||||
ID int64 `json:"id"`
|
SessionID string `json:"sessionId"`
|
||||||
|
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.
|
// StateResponse is the response from GET /api/v1/state.
|
||||||
type StateResponse struct {
|
type StateResponse struct {
|
||||||
ID int64 `json:"id"`
|
SessionID string `json:"sessionId"`
|
||||||
|
ClientID string `json:"clientId"`
|
||||||
Nick string `json:"nick"`
|
Nick string `json:"nick"`
|
||||||
Channels []string `json:"channels"`
|
Channels []string `json:"channels"`
|
||||||
}
|
}
|
||||||
@@ -33,21 +35,21 @@ type Message struct {
|
|||||||
Meta any `json:"meta,omitempty"`
|
Meta any `json:"meta,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// BodyLines returns the body as a string slice.
|
// BodyLines returns the body as a slice of strings (for text messages).
|
||||||
func (m *Message) BodyLines() []string {
|
func (m *Message) BodyLines() []string {
|
||||||
switch bodyVal := m.Body.(type) {
|
switch v := m.Body.(type) {
|
||||||
case []any:
|
case []any:
|
||||||
lines := make([]string, 0, len(bodyVal))
|
lines := make([]string, 0, len(v))
|
||||||
|
|
||||||
for _, item := range bodyVal {
|
for _, item := range v {
|
||||||
if str, ok := item.(string); ok {
|
if s, ok := item.(string); ok {
|
||||||
lines = append(lines, str)
|
lines = append(lines, s)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return lines
|
return lines
|
||||||
case []string:
|
case []string:
|
||||||
return bodyVal
|
return v
|
||||||
default:
|
default:
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -71,13 +73,6 @@ type ServerInfo struct {
|
|||||||
// MessagesResponse wraps polling results.
|
// MessagesResponse wraps polling results.
|
||||||
type MessagesResponse struct {
|
type MessagesResponse struct {
|
||||||
Messages []Message `json:"messages"`
|
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.
|
// ParseTS parses the message timestamp.
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
// Package main is the entry point for the chat-cli client.
|
// Package main provides a terminal-based IRC-style chat client.
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@@ -8,31 +8,32 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
api "git.eeqj.de/sneak/chat/cmd/chat-cli/api"
|
"git.eeqj.de/sneak/chat/cmd/chat-cli/api"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
splitParts = 2
|
maxNickLen = 32
|
||||||
pollTimeout = 15
|
pollTimeoutSec = 15
|
||||||
pollRetry = 2 * time.Second
|
pollRetrySec = 2
|
||||||
timeFormat = "15:04"
|
splitNParts = 2
|
||||||
|
commandSplitArgs = 2
|
||||||
)
|
)
|
||||||
|
|
||||||
// App holds the application state.
|
// App holds the application state.
|
||||||
type App struct {
|
type App struct {
|
||||||
ui *UI
|
ui *UI
|
||||||
client *api.Client
|
client *chatapi.Client
|
||||||
|
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
nick string
|
nick string
|
||||||
target string
|
target string // current target (#channel or nick for DM)
|
||||||
connected bool
|
connected bool
|
||||||
lastQID int64
|
lastMsgID string
|
||||||
stopPoll chan struct{}
|
stopPoll chan struct{}
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
app := &App{ //nolint:exhaustruct
|
app := &App{
|
||||||
ui: NewUI(),
|
ui: NewUI(),
|
||||||
nick: "guest",
|
nick: "guest",
|
||||||
}
|
}
|
||||||
@@ -40,14 +41,8 @@ func main() {
|
|||||||
app.ui.OnInput(app.handleInput)
|
app.ui.OnInput(app.handleInput)
|
||||||
app.ui.SetStatus(app.nick, "", "disconnected")
|
app.ui.SetStatus(app.nick, "", "disconnected")
|
||||||
|
|
||||||
app.ui.AddStatus(
|
app.ui.AddStatus("Welcome to chat-cli — an IRC-style client")
|
||||||
"Welcome to chat-cli — an IRC-style client",
|
app.ui.AddStatus("Type [yellow]/connect <server-url>[white] to begin, or [yellow]/help[white] for commands")
|
||||||
)
|
|
||||||
app.ui.AddStatus(
|
|
||||||
"Type [yellow]/connect <server-url>" +
|
|
||||||
"[white] to begin, " +
|
|
||||||
"or [yellow]/help[white] for commands",
|
|
||||||
)
|
|
||||||
|
|
||||||
err := app.ui.Run()
|
err := app.ui.Run()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -63,55 +58,51 @@ func (a *App) handleInput(text string) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Plain text → PRIVMSG to current target.
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
target := a.target
|
target := a.target
|
||||||
connected := a.connected
|
connected := a.connected
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
if !connected {
|
if !connected {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus("[red]Not connected. Use /connect <url>")
|
||||||
"[red]Not connected. Use /connect <url>",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if target == "" {
|
if target == "" {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus("[red]No target. Use /join #channel or /query nick")
|
||||||
"[red]No target. " +
|
|
||||||
"Use /join #channel or /query nick",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err := a.client.SendMessage(&api.Message{ //nolint:exhaustruct
|
err := a.client.SendMessage(&chatapi.Message{
|
||||||
Command: "PRIVMSG",
|
Command: "PRIVMSG",
|
||||||
To: target,
|
To: target,
|
||||||
Body: []string{text},
|
Body: []string{text},
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus(fmt.Sprintf("[red]Send error: %v", err))
|
||||||
"[red]Send error: " + err.Error(),
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
timestamp := time.Now().Format(timeFormat)
|
// Echo locally.
|
||||||
|
ts := time.Now().Format("15:04")
|
||||||
|
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
nick := a.nick
|
nick := a.nick
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
a.ui.AddLine(target, fmt.Sprintf(
|
a.ui.AddLine(target, fmt.Sprintf("[gray]%s [green]<%s>[white] %s", ts, nick, text))
|
||||||
"[gray]%s [green]<%s>[white] %s",
|
|
||||||
timestamp, nick, text,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handleCommand(text string) {
|
func (a *App) handleCommand(text string) {
|
||||||
parts := strings.SplitN(text, " ", splitParts)
|
a.dispatchCommand(text)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *App) dispatchCommand(text string) {
|
||||||
|
parts := strings.SplitN(text, " ", splitNParts)
|
||||||
cmd := strings.ToLower(parts[0])
|
cmd := strings.ToLower(parts[0])
|
||||||
|
|
||||||
args := ""
|
args := ""
|
||||||
@@ -119,66 +110,64 @@ func (a *App) handleCommand(text string) {
|
|||||||
args = parts[1]
|
args = parts[1]
|
||||||
}
|
}
|
||||||
|
|
||||||
a.dispatchCommand(cmd, args)
|
a.execCommand(cmd, args)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) dispatchCommand(cmd, args string) {
|
func (a *App) execCommand(cmd, args string) {
|
||||||
switch cmd {
|
commands := a.commandMap()
|
||||||
case "/connect":
|
|
||||||
a.cmdConnect(args)
|
handler, ok := commands[cmd]
|
||||||
case "/nick":
|
if !ok {
|
||||||
a.cmdNick(args)
|
a.ui.AddStatus("[red]Unknown command: " + cmd)
|
||||||
case "/join":
|
|
||||||
a.cmdJoin(args)
|
return
|
||||||
case "/part":
|
}
|
||||||
a.cmdPart(args)
|
|
||||||
case "/msg":
|
handler(args)
|
||||||
a.cmdMsg(args)
|
}
|
||||||
case "/query":
|
|
||||||
a.cmdQuery(args)
|
func (a *App) commandMap() map[string]func(string) {
|
||||||
case "/topic":
|
noArgs := func(fn func()) func(string) {
|
||||||
a.cmdTopic(args)
|
return func(_ string) { fn() }
|
||||||
case "/names":
|
}
|
||||||
a.cmdNames()
|
|
||||||
case "/list":
|
return map[string]func(string){
|
||||||
a.cmdList()
|
"/connect": a.cmdConnect,
|
||||||
case "/window", "/w":
|
"/nick": a.cmdNick,
|
||||||
a.cmdWindow(args)
|
"/join": a.cmdJoin,
|
||||||
case "/quit":
|
"/part": a.cmdPart,
|
||||||
a.cmdQuit()
|
"/msg": a.cmdMsg,
|
||||||
case "/help":
|
"/query": a.cmdQuery,
|
||||||
a.cmdHelp()
|
"/topic": a.cmdTopic,
|
||||||
default:
|
"/names": noArgs(a.cmdNames),
|
||||||
a.ui.AddStatus(
|
"/list": noArgs(a.cmdList),
|
||||||
"[red]Unknown command: " + cmd,
|
"/window": a.cmdWindow,
|
||||||
)
|
"/w": a.cmdWindow,
|
||||||
|
"/quit": noArgs(a.cmdQuit),
|
||||||
|
"/help": noArgs(a.cmdHelp),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) cmdConnect(serverURL string) {
|
func (a *App) cmdConnect(serverURL string) {
|
||||||
if serverURL == "" {
|
if serverURL == "" {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus("[red]Usage: /connect <server-url>")
|
||||||
"[red]Usage: /connect <server-url>",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
serverURL = strings.TrimRight(serverURL, "/")
|
serverURL = strings.TrimRight(serverURL, "/")
|
||||||
|
|
||||||
a.ui.AddStatus("Connecting to " + serverURL + "...")
|
a.ui.AddStatus(fmt.Sprintf("Connecting to %s...", serverURL))
|
||||||
|
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
nick := a.nick
|
nick := a.nick
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
client := api.NewClient(serverURL)
|
client := chatapi.NewClient(serverURL)
|
||||||
|
|
||||||
resp, err := client.CreateSession(nick)
|
resp, err := client.CreateSession(nick)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Connection failed: %v", err))
|
||||||
"[red]Connection failed: %v", err,
|
|
||||||
))
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -187,15 +176,13 @@ func (a *App) cmdConnect(serverURL string) {
|
|||||||
a.client = client
|
a.client = client
|
||||||
a.nick = resp.Nick
|
a.nick = resp.Nick
|
||||||
a.connected = true
|
a.connected = true
|
||||||
a.lastQID = 0
|
a.lastMsgID = ""
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[green]Connected! Nick: %s, Session: %s", resp.Nick, resp.SessionID))
|
||||||
"[green]Connected! Nick: %s, Session: %d",
|
|
||||||
resp.Nick, resp.ID,
|
|
||||||
))
|
|
||||||
a.ui.SetStatus(resp.Nick, "", "connected")
|
a.ui.SetStatus(resp.Nick, "", "connected")
|
||||||
|
|
||||||
|
// Start polling.
|
||||||
a.stopPoll = make(chan struct{})
|
a.stopPoll = make(chan struct{})
|
||||||
|
|
||||||
go a.pollLoop()
|
go a.pollLoop()
|
||||||
@@ -203,9 +190,7 @@ func (a *App) cmdConnect(serverURL string) {
|
|||||||
|
|
||||||
func (a *App) cmdNick(nick string) {
|
func (a *App) cmdNick(nick string) {
|
||||||
if nick == "" {
|
if nick == "" {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus("[red]Usage: /nick <name>")
|
||||||
"[red]Usage: /nick <name>",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -219,22 +204,17 @@ func (a *App) cmdNick(nick string) {
|
|||||||
a.nick = nick
|
a.nick = nick
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus(fmt.Sprintf("Nick set to %s (will be used on connect)", nick))
|
||||||
"Nick set to " + nick +
|
|
||||||
" (will be used on connect)",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err := a.client.SendMessage(&api.Message{ //nolint:exhaustruct
|
err := a.client.SendMessage(&chatapi.Message{
|
||||||
Command: "NICK",
|
Command: "NICK",
|
||||||
Body: []string{nick},
|
Body: []string{nick},
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Nick change failed: %v", err))
|
||||||
"[red]Nick change failed: %v", err,
|
|
||||||
))
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -250,9 +230,7 @@ func (a *App) cmdNick(nick string) {
|
|||||||
|
|
||||||
func (a *App) cmdJoin(channel string) {
|
func (a *App) cmdJoin(channel string) {
|
||||||
if channel == "" {
|
if channel == "" {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus("[red]Usage: /join #channel")
|
||||||
"[red]Usage: /join #channel",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -273,9 +251,7 @@ func (a *App) cmdJoin(channel string) {
|
|||||||
|
|
||||||
err := a.client.JoinChannel(channel)
|
err := a.client.JoinChannel(channel)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Join failed: %v", err))
|
||||||
"[red]Join failed: %v", err,
|
|
||||||
))
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -286,14 +262,13 @@ func (a *App) cmdJoin(channel string) {
|
|||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
a.ui.SwitchToBuffer(channel)
|
a.ui.SwitchToBuffer(channel)
|
||||||
a.ui.AddLine(channel,
|
a.ui.AddLine(channel, "[yellow]*** Joined "+channel)
|
||||||
"[yellow]*** Joined "+channel,
|
|
||||||
)
|
|
||||||
a.ui.SetStatus(nick, channel, "connected")
|
a.ui.SetStatus(nick, channel, "connected")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) cmdPart(channel string) {
|
func (a *App) cmdPart(channel string) {
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
|
|
||||||
if channel == "" {
|
if channel == "" {
|
||||||
channel = a.target
|
channel = a.target
|
||||||
}
|
}
|
||||||
@@ -301,8 +276,7 @@ func (a *App) cmdPart(channel string) {
|
|||||||
connected := a.connected
|
connected := a.connected
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
if channel == "" ||
|
if channel == "" || !strings.HasPrefix(channel, "#") {
|
||||||
!strings.HasPrefix(channel, "#") {
|
|
||||||
a.ui.AddStatus("[red]No channel to part")
|
a.ui.AddStatus("[red]No channel to part")
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -316,18 +290,15 @@ func (a *App) cmdPart(channel string) {
|
|||||||
|
|
||||||
err := a.client.PartChannel(channel)
|
err := a.client.PartChannel(channel)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Part failed: %v", err))
|
||||||
"[red]Part failed: %v", err,
|
|
||||||
))
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
a.ui.AddLine(channel,
|
a.ui.AddLine(channel, "[yellow]*** Left "+channel)
|
||||||
"[yellow]*** Left "+channel,
|
|
||||||
)
|
|
||||||
|
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
|
|
||||||
if a.target == channel {
|
if a.target == channel {
|
||||||
a.target = ""
|
a.target = ""
|
||||||
}
|
}
|
||||||
@@ -340,11 +311,10 @@ func (a *App) cmdPart(channel string) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) cmdMsg(args string) {
|
func (a *App) cmdMsg(args string) {
|
||||||
parts := strings.SplitN(args, " ", splitParts)
|
parts := strings.SplitN(args, " ", commandSplitArgs)
|
||||||
if len(parts) < splitParts {
|
|
||||||
a.ui.AddStatus(
|
if len(parts) < commandSplitArgs {
|
||||||
"[red]Usage: /msg <nick> <text>",
|
a.ui.AddStatus("[red]Usage: /msg <nick> <text>")
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -362,32 +332,24 @@ func (a *App) cmdMsg(args string) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err := a.client.SendMessage(&api.Message{ //nolint:exhaustruct
|
err := a.client.SendMessage(&chatapi.Message{
|
||||||
Command: "PRIVMSG",
|
Command: "PRIVMSG",
|
||||||
To: target,
|
To: target,
|
||||||
Body: []string{text},
|
Body: []string{text},
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Send failed: %v", err))
|
||||||
"[red]Send failed: %v", err,
|
|
||||||
))
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
timestamp := time.Now().Format(timeFormat)
|
ts := time.Now().Format("15:04")
|
||||||
|
a.ui.AddLine(target, fmt.Sprintf("[gray]%s [green]<%s>[white] %s", ts, nick, text))
|
||||||
a.ui.AddLine(target, fmt.Sprintf(
|
|
||||||
"[gray]%s [green]<%s>[white] %s",
|
|
||||||
timestamp, nick, text,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) cmdQuery(nick string) {
|
func (a *App) cmdQuery(nick string) {
|
||||||
if nick == "" {
|
if nick == "" {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus("[red]Usage: /query <nick>")
|
||||||
"[red]Usage: /query <nick>",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -420,28 +382,25 @@ func (a *App) cmdTopic(args string) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if args == "" {
|
if args == "" {
|
||||||
err := a.client.SendMessage(&api.Message{ //nolint:exhaustruct
|
// Query topic.
|
||||||
|
err := a.client.SendMessage(&chatapi.Message{
|
||||||
Command: "TOPIC",
|
Command: "TOPIC",
|
||||||
To: target,
|
To: target,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Topic query failed: %v", err))
|
||||||
"[red]Topic query failed: %v", err,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err := a.client.SendMessage(&api.Message{ //nolint:exhaustruct
|
err := a.client.SendMessage(&chatapi.Message{
|
||||||
Command: "TOPIC",
|
Command: "TOPIC",
|
||||||
To: target,
|
To: target,
|
||||||
Body: []string{args},
|
Body: []string{args},
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Topic set failed: %v", err))
|
||||||
"[red]Topic set failed: %v", err,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -465,17 +424,12 @@ func (a *App) cmdNames() {
|
|||||||
|
|
||||||
members, err := a.client.GetMembers(target)
|
members, err := a.client.GetMembers(target)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]Names failed: %v", err))
|
||||||
"[red]Names failed: %v", err,
|
|
||||||
))
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
a.ui.AddLine(target, fmt.Sprintf(
|
a.ui.AddLine(target, fmt.Sprintf("[cyan]*** Members of %s: %s", target, strings.Join(members, " ")))
|
||||||
"[cyan]*** Members of %s: %s",
|
|
||||||
target, strings.Join(members, " "),
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) cmdList() {
|
func (a *App) cmdList() {
|
||||||
@@ -491,9 +445,7 @@ func (a *App) cmdList() {
|
|||||||
|
|
||||||
channels, err := a.client.ListChannels()
|
channels, err := a.client.ListChannels()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[red]List failed: %v", err))
|
||||||
"[red]List failed: %v", err,
|
|
||||||
))
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -501,10 +453,7 @@ func (a *App) cmdList() {
|
|||||||
a.ui.AddStatus("[cyan]*** Channel list:")
|
a.ui.AddStatus("[cyan]*** Channel list:")
|
||||||
|
|
||||||
for _, ch := range channels {
|
for _, ch := range channels {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf(" %s (%d members) %s", ch.Name, ch.Members, ch.Topic))
|
||||||
" %s (%d members) %s",
|
|
||||||
ch.Name, ch.Members, ch.Topic,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
a.ui.AddStatus("[cyan]*** End of channel list")
|
a.ui.AddStatus("[cyan]*** End of channel list")
|
||||||
@@ -512,33 +461,30 @@ func (a *App) cmdList() {
|
|||||||
|
|
||||||
func (a *App) cmdWindow(args string) {
|
func (a *App) cmdWindow(args string) {
|
||||||
if args == "" {
|
if args == "" {
|
||||||
a.ui.AddStatus(
|
a.ui.AddStatus("[red]Usage: /window <number>")
|
||||||
"[red]Usage: /window <number>",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var bufIndex int
|
n := 0
|
||||||
|
_, _ = fmt.Sscanf(args, "%d", &n)
|
||||||
|
|
||||||
_, _ = fmt.Sscanf(args, "%d", &bufIndex)
|
a.ui.SwitchBuffer(n)
|
||||||
|
|
||||||
a.ui.SwitchBuffer(bufIndex)
|
|
||||||
|
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
nick := a.nick
|
nick := a.nick
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
if bufIndex >= 0 && bufIndex < a.ui.BufferCount() {
|
// Update target based on buffer.
|
||||||
buf := a.ui.buffers[bufIndex]
|
if n < a.ui.BufferCount() {
|
||||||
|
buf := a.ui.buffers[n]
|
||||||
|
|
||||||
if buf.Name != "(status)" {
|
if buf.Name != "(status)" {
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
a.target = buf.Name
|
a.target = buf.Name
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
a.ui.SetStatus(
|
a.ui.SetStatus(nick, buf.Name, "connected")
|
||||||
nick, buf.Name, "connected",
|
|
||||||
)
|
|
||||||
} else {
|
} else {
|
||||||
a.ui.SetStatus(nick, "", "connected")
|
a.ui.SetStatus(nick, "", "connected")
|
||||||
}
|
}
|
||||||
@@ -549,9 +495,7 @@ func (a *App) cmdQuit() {
|
|||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
|
|
||||||
if a.connected && a.client != nil {
|
if a.connected && a.client != nil {
|
||||||
_ = a.client.SendMessage(
|
_ = a.client.SendMessage(&chatapi.Message{Command: "QUIT"})
|
||||||
&api.Message{Command: "QUIT"}, //nolint:exhaustruct
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if a.stopPoll != nil {
|
if a.stopPoll != nil {
|
||||||
@@ -574,7 +518,7 @@ func (a *App) cmdHelp() {
|
|||||||
" /topic [text] — View/set topic",
|
" /topic [text] — View/set topic",
|
||||||
" /names — List channel members",
|
" /names — List channel members",
|
||||||
" /list — List channels",
|
" /list — List channels",
|
||||||
" /window <n> — Switch buffer",
|
" /window <n> — Switch buffer (Alt+0-9)",
|
||||||
" /quit — Disconnect and exit",
|
" /quit — Disconnect and exit",
|
||||||
" /help — This help",
|
" /help — This help",
|
||||||
" Plain text sends to current target.",
|
" Plain text sends to current target.",
|
||||||
@@ -596,36 +540,35 @@ func (a *App) pollLoop() {
|
|||||||
|
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
client := a.client
|
client := a.client
|
||||||
lastQID := a.lastQID
|
lastID := a.lastMsgID
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
if client == nil {
|
if client == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
result, err := client.PollMessages(
|
msgs, err := client.PollMessages(lastID, pollTimeoutSec)
|
||||||
lastQID, pollTimeout,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
time.Sleep(pollRetry)
|
// Transient error — retry after delay.
|
||||||
|
time.Sleep(pollRetrySec * time.Second)
|
||||||
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if result.LastID > 0 {
|
for _, msg := range msgs {
|
||||||
|
a.handleServerMessage(&msg)
|
||||||
|
|
||||||
|
if msg.ID != "" {
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
a.lastQID = result.LastID
|
a.lastMsgID = msg.ID
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
for i := range result.Messages {
|
|
||||||
a.handleServerMessage(&result.Messages[i])
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handleServerMessage(msg *api.Message) {
|
func (a *App) handleServerMessage(msg *chatapi.Message) {
|
||||||
timestamp := a.formatTS(msg)
|
ts := a.formatMessageTS(msg)
|
||||||
|
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
myNick := a.nick
|
myNick := a.nick
|
||||||
@@ -633,111 +576,88 @@ func (a *App) handleServerMessage(msg *api.Message) {
|
|||||||
|
|
||||||
switch msg.Command {
|
switch msg.Command {
|
||||||
case "PRIVMSG":
|
case "PRIVMSG":
|
||||||
a.handlePrivmsgEvent(msg, timestamp, myNick)
|
a.handlePrivmsg(msg, ts, myNick)
|
||||||
case "JOIN":
|
case "JOIN":
|
||||||
a.handleJoinEvent(msg, timestamp)
|
a.handleJoinMsg(msg, ts)
|
||||||
case "PART":
|
case "PART":
|
||||||
a.handlePartEvent(msg, timestamp)
|
a.handlePartMsg(msg, ts)
|
||||||
case "QUIT":
|
case "QUIT":
|
||||||
a.handleQuitEvent(msg, timestamp)
|
a.handleQuitMsg(msg, ts)
|
||||||
case "NICK":
|
case "NICK":
|
||||||
a.handleNickEvent(msg, timestamp, myNick)
|
a.handleNickMsg(msg, ts, myNick)
|
||||||
case "NOTICE":
|
case "NOTICE":
|
||||||
a.handleNoticeEvent(msg, timestamp)
|
a.handleNoticeMsg(msg, ts)
|
||||||
case "TOPIC":
|
case "TOPIC":
|
||||||
a.handleTopicEvent(msg, timestamp)
|
a.handleTopicMsg(msg, ts)
|
||||||
default:
|
default:
|
||||||
a.handleDefaultEvent(msg, timestamp)
|
a.handleDefaultMsg(msg, ts)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) formatTS(msg *api.Message) string {
|
func (a *App) formatMessageTS(msg *chatapi.Message) string {
|
||||||
if msg.TS != "" {
|
if msg.TS != "" {
|
||||||
return msg.ParseTS().UTC().Format(timeFormat)
|
t := msg.ParseTS()
|
||||||
|
|
||||||
|
return t.Local().Format("15:04") //nolint:gosmopolitan // Local time display is intentional for UI
|
||||||
}
|
}
|
||||||
|
|
||||||
return time.Now().Format(timeFormat)
|
return time.Now().Format("15:04")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handlePrivmsgEvent(
|
func (a *App) handlePrivmsg(msg *chatapi.Message, ts, myNick string) {
|
||||||
msg *api.Message, timestamp, myNick string,
|
|
||||||
) {
|
|
||||||
lines := msg.BodyLines()
|
lines := msg.BodyLines()
|
||||||
text := strings.Join(lines, " ")
|
text := strings.Join(lines, " ")
|
||||||
|
|
||||||
if msg.From == myNick {
|
if msg.From == myNick {
|
||||||
|
// Skip our own echoed messages (already displayed locally).
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
target := msg.To
|
target := msg.To
|
||||||
|
|
||||||
if !strings.HasPrefix(target, "#") {
|
if !strings.HasPrefix(target, "#") {
|
||||||
|
// DM — use sender's nick as buffer name.
|
||||||
target = msg.From
|
target = msg.From
|
||||||
}
|
}
|
||||||
|
|
||||||
a.ui.AddLine(target, fmt.Sprintf(
|
a.ui.AddLine(target, fmt.Sprintf("[gray]%s [green]<%s>[white] %s", ts, msg.From, text))
|
||||||
"[gray]%s [green]<%s>[white] %s",
|
|
||||||
timestamp, msg.From, text,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handleJoinEvent(
|
func (a *App) handleJoinMsg(msg *chatapi.Message, ts string) {
|
||||||
msg *api.Message, timestamp string,
|
target := msg.To
|
||||||
) {
|
if target != "" {
|
||||||
if msg.To == "" {
|
a.ui.AddLine(target, fmt.Sprintf("[gray]%s [yellow]*** %s has joined %s", ts, msg.From, target))
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
a.ui.AddLine(msg.To, fmt.Sprintf(
|
|
||||||
"[gray]%s [yellow]*** %s has joined %s",
|
|
||||||
timestamp, msg.From, msg.To,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handlePartEvent(
|
func (a *App) handlePartMsg(msg *chatapi.Message, ts string) {
|
||||||
msg *api.Message, timestamp string,
|
target := msg.To
|
||||||
) {
|
|
||||||
if msg.To == "" {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
lines := msg.BodyLines()
|
lines := msg.BodyLines()
|
||||||
|
|
||||||
|
reason := strings.Join(lines, " ")
|
||||||
|
|
||||||
|
if target != "" {
|
||||||
|
if reason != "" {
|
||||||
|
a.ui.AddLine(target, fmt.Sprintf("[gray]%s [yellow]*** %s has left %s (%s)", ts, msg.From, target, reason))
|
||||||
|
} else {
|
||||||
|
a.ui.AddLine(target, fmt.Sprintf("[gray]%s [yellow]*** %s has left %s", ts, msg.From, target))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *App) handleQuitMsg(msg *chatapi.Message, ts string) {
|
||||||
|
lines := msg.BodyLines()
|
||||||
|
|
||||||
reason := strings.Join(lines, " ")
|
reason := strings.Join(lines, " ")
|
||||||
|
|
||||||
if reason != "" {
|
if reason != "" {
|
||||||
a.ui.AddLine(msg.To, fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[gray]%s [yellow]*** %s has quit (%s)", ts, msg.From, reason))
|
||||||
"[gray]%s [yellow]*** %s has left %s (%s)",
|
|
||||||
timestamp, msg.From, msg.To, reason,
|
|
||||||
))
|
|
||||||
} else {
|
} else {
|
||||||
a.ui.AddLine(msg.To, fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[gray]%s [yellow]*** %s has quit", ts, msg.From))
|
||||||
"[gray]%s [yellow]*** %s has left %s",
|
|
||||||
timestamp, msg.From, msg.To,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handleQuitEvent(
|
func (a *App) handleNickMsg(msg *chatapi.Message, ts, myNick string) {
|
||||||
msg *api.Message, timestamp string,
|
|
||||||
) {
|
|
||||||
lines := msg.BodyLines()
|
|
||||||
reason := strings.Join(lines, " ")
|
|
||||||
|
|
||||||
if reason != "" {
|
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
|
||||||
"[gray]%s [yellow]*** %s has quit (%s)",
|
|
||||||
timestamp, msg.From, reason,
|
|
||||||
))
|
|
||||||
} else {
|
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
|
||||||
"[gray]%s [yellow]*** %s has quit",
|
|
||||||
timestamp, msg.From,
|
|
||||||
))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *App) handleNickEvent(
|
|
||||||
msg *api.Message, timestamp, myNick string,
|
|
||||||
) {
|
|
||||||
lines := msg.BodyLines()
|
lines := msg.BodyLines()
|
||||||
|
|
||||||
newNick := ""
|
newNick := ""
|
||||||
@@ -748,57 +668,39 @@ func (a *App) handleNickEvent(
|
|||||||
if msg.From == myNick && newNick != "" {
|
if msg.From == myNick && newNick != "" {
|
||||||
a.mu.Lock()
|
a.mu.Lock()
|
||||||
a.nick = newNick
|
a.nick = newNick
|
||||||
|
|
||||||
target := a.target
|
target := a.target
|
||||||
a.mu.Unlock()
|
a.mu.Unlock()
|
||||||
|
|
||||||
a.ui.SetStatus(newNick, target, "connected")
|
a.ui.SetStatus(newNick, target, "connected")
|
||||||
}
|
}
|
||||||
|
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[gray]%s [yellow]*** %s is now known as %s", ts, msg.From, newNick))
|
||||||
"[gray]%s [yellow]*** %s is now known as %s",
|
|
||||||
timestamp, msg.From, newNick,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handleNoticeEvent(
|
func (a *App) handleNoticeMsg(msg *chatapi.Message, ts string) {
|
||||||
msg *api.Message, timestamp string,
|
|
||||||
) {
|
|
||||||
lines := msg.BodyLines()
|
lines := msg.BodyLines()
|
||||||
|
|
||||||
text := strings.Join(lines, " ")
|
text := strings.Join(lines, " ")
|
||||||
|
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[gray]%s [magenta]--%s-- %s", ts, msg.From, text))
|
||||||
"[gray]%s [magenta]--%s-- %s",
|
|
||||||
timestamp, msg.From, text,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handleTopicEvent(
|
func (a *App) handleTopicMsg(msg *chatapi.Message, ts string) {
|
||||||
msg *api.Message, timestamp string,
|
lines := msg.BodyLines()
|
||||||
) {
|
|
||||||
if msg.To == "" {
|
text := strings.Join(lines, " ")
|
||||||
return
|
|
||||||
|
if msg.To != "" {
|
||||||
|
a.ui.AddLine(msg.To, fmt.Sprintf("[gray]%s [cyan]*** %s set topic: %s", ts, msg.From, text))
|
||||||
}
|
}
|
||||||
|
|
||||||
lines := msg.BodyLines()
|
|
||||||
text := strings.Join(lines, " ")
|
|
||||||
|
|
||||||
a.ui.AddLine(msg.To, fmt.Sprintf(
|
|
||||||
"[gray]%s [cyan]*** %s set topic: %s",
|
|
||||||
timestamp, msg.From, text,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) handleDefaultEvent(
|
func (a *App) handleDefaultMsg(msg *chatapi.Message, ts string) {
|
||||||
msg *api.Message, timestamp string,
|
|
||||||
) {
|
|
||||||
lines := msg.BodyLines()
|
lines := msg.BodyLines()
|
||||||
|
|
||||||
text := strings.Join(lines, " ")
|
text := strings.Join(lines, " ")
|
||||||
|
|
||||||
if text != "" {
|
if text != "" {
|
||||||
a.ui.AddStatus(fmt.Sprintf(
|
a.ui.AddStatus(fmt.Sprintf("[gray]%s [white][%s] %s", ts, msg.Command, text))
|
||||||
"[gray]%s [white][%s] %s",
|
|
||||||
timestamp, msg.Command, text,
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -32,38 +32,23 @@ type UI struct {
|
|||||||
|
|
||||||
// NewUI creates the tview-based IRC-like UI.
|
// NewUI creates the tview-based IRC-like UI.
|
||||||
func NewUI() *UI {
|
func NewUI() *UI {
|
||||||
ui := &UI{ //nolint:exhaustruct,varnamelen // fields set below; ui is idiomatic
|
ui := &UI{
|
||||||
app: tview.NewApplication(),
|
app: tview.NewApplication(),
|
||||||
buffers: []*Buffer{
|
buffers: []*Buffer{
|
||||||
{Name: "(status)", Lines: nil, Unread: 0},
|
{Name: "(status)", Lines: nil},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
ui.initMessages()
|
ui.setupWidgets()
|
||||||
ui.initStatusBar()
|
ui.setupInputCapture()
|
||||||
ui.initInput()
|
ui.setupLayout()
|
||||||
ui.initKeyCapture()
|
|
||||||
|
|
||||||
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
|
return ui
|
||||||
}
|
}
|
||||||
|
|
||||||
// Run starts the UI event loop (blocks).
|
// Run starts the UI event loop (blocks).
|
||||||
func (ui *UI) Run() error {
|
func (ui *UI) Run() error {
|
||||||
err := ui.app.Run()
|
return ui.app.Run()
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("run ui: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Stop stops the UI.
|
// Stop stops the UI.
|
||||||
@@ -77,43 +62,41 @@ func (ui *UI) OnInput(fn func(string)) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// AddLine adds a line to the specified buffer.
|
// 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() {
|
ui.app.QueueUpdateDraw(func() {
|
||||||
buf := ui.getOrCreateBuffer(bufferName)
|
buf := ui.getOrCreateBuffer(bufferName)
|
||||||
buf.Lines = append(buf.Lines, line)
|
buf.Lines = append(buf.Lines, line)
|
||||||
|
|
||||||
cur := ui.buffers[ui.currentBuffer]
|
// Mark unread if not currently viewing this buffer.
|
||||||
if cur != buf {
|
if ui.buffers[ui.currentBuffer] != buf {
|
||||||
buf.Unread++
|
buf.Unread++
|
||||||
|
|
||||||
ui.refreshStatusBar()
|
ui.refreshStatus()
|
||||||
}
|
}
|
||||||
|
|
||||||
if cur == buf {
|
// If viewing this buffer, append to display.
|
||||||
|
if ui.buffers[ui.currentBuffer] == buf {
|
||||||
_, _ = fmt.Fprintln(ui.messages, line)
|
_, _ = fmt.Fprintln(ui.messages, line)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// AddStatus adds a line to the status buffer.
|
// AddStatus adds a line to the status buffer (buffer 0).
|
||||||
func (ui *UI) AddStatus(line string) {
|
func (ui *UI) AddStatus(line string) {
|
||||||
ts := time.Now().Format("15:04")
|
ts := time.Now().Format("15:04")
|
||||||
ui.AddLine(
|
ui.AddLine("(status)", fmt.Sprintf("[gray]%s[white] %s", ts, line))
|
||||||
"(status)",
|
|
||||||
"[gray]"+ts+"[white] "+line,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// SwitchBuffer switches to the buffer at index n.
|
// SwitchBuffer switches to the buffer at index n.
|
||||||
func (ui *UI) SwitchBuffer(bufIndex int) {
|
func (ui *UI) SwitchBuffer(n int) {
|
||||||
ui.app.QueueUpdateDraw(func() {
|
ui.app.QueueUpdateDraw(func() {
|
||||||
if bufIndex < 0 || bufIndex >= len(ui.buffers) {
|
if n < 0 || n >= len(ui.buffers) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
ui.currentBuffer = bufIndex
|
ui.currentBuffer = n
|
||||||
|
buf := ui.buffers[n]
|
||||||
|
|
||||||
buf := ui.buffers[bufIndex]
|
|
||||||
buf.Unread = 0
|
buf.Unread = 0
|
||||||
|
|
||||||
ui.messages.Clear()
|
ui.messages.Clear()
|
||||||
@@ -123,12 +106,11 @@ func (ui *UI) SwitchBuffer(bufIndex int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
ui.messages.ScrollToEnd()
|
ui.messages.ScrollToEnd()
|
||||||
ui.refreshStatusBar()
|
ui.refreshStatus()
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// SwitchToBuffer switches to named buffer, creating if
|
// SwitchToBuffer switches to the named buffer, creating it if needed.
|
||||||
// needed.
|
|
||||||
func (ui *UI) SwitchToBuffer(name string) {
|
func (ui *UI) SwitchToBuffer(name string) {
|
||||||
ui.app.QueueUpdateDraw(func() {
|
ui.app.QueueUpdateDraw(func() {
|
||||||
buf := ui.getOrCreateBuffer(name)
|
buf := ui.getOrCreateBuffer(name)
|
||||||
@@ -150,16 +132,14 @@ func (ui *UI) SwitchToBuffer(name string) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
ui.messages.ScrollToEnd()
|
ui.messages.ScrollToEnd()
|
||||||
ui.refreshStatusBar()
|
ui.refreshStatus()
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetStatus updates the status bar text.
|
// SetStatus updates the status bar text.
|
||||||
func (ui *UI) SetStatus(
|
func (ui *UI) SetStatus(nick, target, connStatus string) {
|
||||||
nick, target, connStatus string,
|
|
||||||
) {
|
|
||||||
ui.app.QueueUpdateDraw(func() {
|
ui.app.QueueUpdateDraw(func() {
|
||||||
ui.renderStatusBar(nick, target, connStatus)
|
ui.refreshStatusWith(nick, target, connStatus)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -168,7 +148,7 @@ func (ui *UI) BufferCount() int {
|
|||||||
return len(ui.buffers)
|
return len(ui.buffers)
|
||||||
}
|
}
|
||||||
|
|
||||||
// BufferIndex returns the index of a named buffer.
|
// BufferIndex returns the index of a named buffer, or -1.
|
||||||
func (ui *UI) BufferIndex(name string) int {
|
func (ui *UI) BufferIndex(name string) int {
|
||||||
for i, buf := range ui.buffers {
|
for i, buf := range ui.buffers {
|
||||||
if buf.Name == name {
|
if buf.Name == name {
|
||||||
@@ -179,7 +159,7 @@ func (ui *UI) BufferIndex(name string) int {
|
|||||||
return -1
|
return -1
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ui *UI) initMessages() {
|
func (ui *UI) setupWidgets() {
|
||||||
ui.messages = tview.NewTextView().
|
ui.messages = tview.NewTextView().
|
||||||
SetDynamicColors(true).
|
SetDynamicColors(true).
|
||||||
SetScrollable(true).
|
SetScrollable(true).
|
||||||
@@ -188,25 +168,17 @@ func (ui *UI) initMessages() {
|
|||||||
ui.app.Draw()
|
ui.app.Draw()
|
||||||
})
|
})
|
||||||
ui.messages.SetBorder(false)
|
ui.messages.SetBorder(false)
|
||||||
}
|
|
||||||
|
|
||||||
func (ui *UI) initStatusBar() {
|
|
||||||
ui.statusBar = tview.NewTextView().
|
ui.statusBar = tview.NewTextView().
|
||||||
SetDynamicColors(true)
|
SetDynamicColors(true)
|
||||||
ui.statusBar.SetBackgroundColor(tcell.ColorNavy)
|
ui.statusBar.SetBackgroundColor(tcell.ColorNavy)
|
||||||
ui.statusBar.SetTextColor(tcell.ColorWhite)
|
ui.statusBar.SetTextColor(tcell.ColorWhite)
|
||||||
}
|
|
||||||
|
|
||||||
func (ui *UI) initInput() {
|
|
||||||
ui.input = tview.NewInputField().
|
ui.input = tview.NewInputField().
|
||||||
SetFieldBackgroundColor(tcell.ColorBlack).
|
SetFieldBackgroundColor(tcell.ColorBlack).
|
||||||
SetFieldTextColor(tcell.ColorWhite)
|
SetFieldTextColor(tcell.ColorWhite)
|
||||||
|
|
||||||
ui.input.SetDoneFunc(func(key tcell.Key) {
|
ui.input.SetDoneFunc(func(key tcell.Key) {
|
||||||
if key != tcell.KeyEnter {
|
if key == tcell.KeyEnter {
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
text := ui.input.GetText()
|
text := ui.input.GetText()
|
||||||
if text == "" {
|
if text == "" {
|
||||||
return
|
return
|
||||||
@@ -217,16 +189,13 @@ func (ui *UI) initInput() {
|
|||||||
if ui.onInput != nil {
|
if ui.onInput != nil {
|
||||||
ui.onInput(text)
|
ui.onInput(text)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ui *UI) initKeyCapture() {
|
func (ui *UI) setupInputCapture() {
|
||||||
ui.app.SetInputCapture(
|
ui.app.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
|
||||||
func(event *tcell.EventKey) *tcell.EventKey {
|
if event.Modifiers()&tcell.ModAlt != 0 {
|
||||||
if event.Modifiers()&tcell.ModAlt == 0 {
|
|
||||||
return event
|
|
||||||
}
|
|
||||||
|
|
||||||
r := event.Rune()
|
r := event.Rune()
|
||||||
if r >= '0' && r <= '9' {
|
if r >= '0' && r <= '9' {
|
||||||
idx := int(r - '0')
|
idx := int(r - '0')
|
||||||
@@ -234,50 +203,47 @@ func (ui *UI) initKeyCapture() {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return event
|
return event
|
||||||
},
|
})
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ui *UI) refreshStatusBar() {
|
func (ui *UI) setupLayout() {
|
||||||
// Placeholder; full refresh needs nick/target context.
|
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)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ui *UI) renderStatusBar(
|
func (ui *UI) refreshStatus() {
|
||||||
nick, target, connStatus string,
|
// Will be called from the main goroutine via QueueUpdateDraw parent.
|
||||||
) {
|
// Rebuild status from app state — caller must provide context.
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ui *UI) refreshStatusWith(nick, target, connStatus string) {
|
||||||
var unreadParts []string
|
var unreadParts []string
|
||||||
|
|
||||||
for i, buf := range ui.buffers {
|
for i, buf := range ui.buffers {
|
||||||
if buf.Unread > 0 {
|
if buf.Unread > 0 {
|
||||||
unreadParts = append(unreadParts,
|
unreadParts = append(unreadParts, fmt.Sprintf("%d:%s(%d)", i, buf.Name, buf.Unread))
|
||||||
fmt.Sprintf(
|
|
||||||
"%d:%s(%d)",
|
|
||||||
i, buf.Name, buf.Unread,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
unread := ""
|
unread := ""
|
||||||
if len(unreadParts) > 0 {
|
if len(unreadParts) > 0 {
|
||||||
unread = " [Act: " +
|
unread = " [Act: " + strings.Join(unreadParts, ",") + "]"
|
||||||
strings.Join(unreadParts, ",") + "]"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bufInfo := fmt.Sprintf(
|
bufInfo := fmt.Sprintf("[%d:%s]", ui.currentBuffer, ui.buffers[ui.currentBuffer].Name)
|
||||||
"[%d:%s]",
|
|
||||||
ui.currentBuffer,
|
|
||||||
ui.buffers[ui.currentBuffer].Name,
|
|
||||||
)
|
|
||||||
|
|
||||||
ui.statusBar.Clear()
|
ui.statusBar.Clear()
|
||||||
|
|
||||||
_, _ = fmt.Fprintf(ui.statusBar,
|
_, _ = fmt.Fprintf(ui.statusBar, " [%s] %s %s %s%s",
|
||||||
" [%s] %s %s %s%s",
|
connStatus, nick, bufInfo, target, unread)
|
||||||
connStatus, nick, bufInfo, target, unread,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ui *UI) getOrCreateBuffer(name string) *Buffer {
|
func (ui *UI) getOrCreateBuffer(name string) *Buffer {
|
||||||
@@ -287,7 +253,7 @@ func (ui *UI) getOrCreateBuffer(name string) *Buffer {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
buf := &Buffer{Name: name, Lines: nil, Unread: 0}
|
buf := &Buffer{Name: name}
|
||||||
ui.buffers = append(ui.buffers, buf)
|
ui.buffers = append(ui.buffers, buf)
|
||||||
|
|
||||||
return buf
|
return buf
|
||||||
|
|||||||
13
go.mod
13
go.mod
@@ -4,18 +4,14 @@ go 1.24.0
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/99designs/basicauth-go v0.0.0-20230316000542-bf6f9cbbf0f8
|
github.com/99designs/basicauth-go v0.0.0-20230316000542-bf6f9cbbf0f8
|
||||||
github.com/gdamore/tcell/v2 v2.13.8
|
|
||||||
github.com/getsentry/sentry-go v0.42.0
|
github.com/getsentry/sentry-go v0.42.0
|
||||||
github.com/go-chi/chi v1.5.5
|
github.com/go-chi/chi v1.5.5
|
||||||
github.com/go-chi/cors v1.2.2
|
github.com/go-chi/cors v1.2.2
|
||||||
github.com/google/uuid v1.6.0
|
|
||||||
github.com/joho/godotenv v1.5.1
|
github.com/joho/godotenv v1.5.1
|
||||||
github.com/prometheus/client_golang v1.23.2
|
github.com/prometheus/client_golang v1.23.2
|
||||||
github.com/rivo/tview v0.42.0
|
|
||||||
github.com/slok/go-http-metrics v0.13.0
|
github.com/slok/go-http-metrics v0.13.0
|
||||||
github.com/spf13/viper v1.21.0
|
github.com/spf13/viper v1.21.0
|
||||||
go.uber.org/fx v1.24.0
|
go.uber.org/fx v1.24.0
|
||||||
golang.org/x/crypto v0.48.0
|
|
||||||
modernc.org/sqlite v1.45.0
|
modernc.org/sqlite v1.45.0
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -25,7 +21,9 @@ require (
|
|||||||
github.com/dustin/go-humanize v1.0.1 // indirect
|
github.com/dustin/go-humanize v1.0.1 // indirect
|
||||||
github.com/fsnotify/fsnotify v1.9.0 // indirect
|
github.com/fsnotify/fsnotify v1.9.0 // indirect
|
||||||
github.com/gdamore/encoding v1.0.1 // indirect
|
github.com/gdamore/encoding v1.0.1 // indirect
|
||||||
|
github.com/gdamore/tcell/v2 v2.13.8 // indirect
|
||||||
github.com/go-viper/mapstructure/v2 v2.4.0 // indirect
|
github.com/go-viper/mapstructure/v2 v2.4.0 // indirect
|
||||||
|
github.com/google/uuid v1.6.0 // indirect
|
||||||
github.com/lucasb-eyer/go-colorful v1.3.0 // indirect
|
github.com/lucasb-eyer/go-colorful v1.3.0 // indirect
|
||||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
|
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
|
||||||
@@ -35,6 +33,7 @@ require (
|
|||||||
github.com/prometheus/common v0.66.1 // indirect
|
github.com/prometheus/common v0.66.1 // indirect
|
||||||
github.com/prometheus/procfs v0.16.1 // indirect
|
github.com/prometheus/procfs v0.16.1 // indirect
|
||||||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
|
||||||
|
github.com/rivo/tview v0.42.0 // indirect
|
||||||
github.com/rivo/uniseg v0.4.7 // indirect
|
github.com/rivo/uniseg v0.4.7 // indirect
|
||||||
github.com/sagikazarmark/locafero v0.11.0 // indirect
|
github.com/sagikazarmark/locafero v0.11.0 // indirect
|
||||||
github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8 // indirect
|
github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8 // indirect
|
||||||
@@ -48,9 +47,9 @@ require (
|
|||||||
go.yaml.in/yaml/v2 v2.4.2 // indirect
|
go.yaml.in/yaml/v2 v2.4.2 // indirect
|
||||||
go.yaml.in/yaml/v3 v3.0.4 // indirect
|
go.yaml.in/yaml/v3 v3.0.4 // indirect
|
||||||
golang.org/x/exp v0.0.0-20251023183803-a4bb9ffd2546 // indirect
|
golang.org/x/exp v0.0.0-20251023183803-a4bb9ffd2546 // indirect
|
||||||
golang.org/x/sys v0.41.0 // indirect
|
golang.org/x/sys v0.38.0 // indirect
|
||||||
golang.org/x/term v0.40.0 // indirect
|
golang.org/x/term v0.37.0 // indirect
|
||||||
golang.org/x/text v0.34.0 // indirect
|
golang.org/x/text v0.31.0 // indirect
|
||||||
google.golang.org/protobuf v1.36.8 // indirect
|
google.golang.org/protobuf v1.36.8 // indirect
|
||||||
modernc.org/libc v1.67.6 // indirect
|
modernc.org/libc v1.67.6 // indirect
|
||||||
modernc.org/mathutil v1.7.1 // indirect
|
modernc.org/mathutil v1.7.1 // indirect
|
||||||
|
|||||||
30
go.sum
30
go.sum
@@ -113,14 +113,12 @@ go.yaml.in/yaml/v3 v3.0.4 h1:tfq32ie2Jv2UxXFdLJdh3jXuOzWiL1fo0bu/FbuKpbc=
|
|||||||
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
|
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||||
golang.org/x/crypto v0.48.0 h1:/VRzVqiRSggnhY7gNRxPauEQ5Drw9haKdM0jqfcCFts=
|
|
||||||
golang.org/x/crypto v0.48.0/go.mod h1:r0kV5h3qnFPlQnBSrULhlsRfryS2pmewsg+XfMgkVos=
|
|
||||||
golang.org/x/exp v0.0.0-20251023183803-a4bb9ffd2546 h1:mgKeJMpvi0yx/sU5GsxQ7p6s2wtOnGAHZWCHUM4KGzY=
|
golang.org/x/exp v0.0.0-20251023183803-a4bb9ffd2546 h1:mgKeJMpvi0yx/sU5GsxQ7p6s2wtOnGAHZWCHUM4KGzY=
|
||||||
golang.org/x/exp v0.0.0-20251023183803-a4bb9ffd2546/go.mod h1:j/pmGrbnkbPtQfxEe5D0VQhZC6qKbfKifgD0oM7sR70=
|
golang.org/x/exp v0.0.0-20251023183803-a4bb9ffd2546/go.mod h1:j/pmGrbnkbPtQfxEe5D0VQhZC6qKbfKifgD0oM7sR70=
|
||||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||||
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
||||||
golang.org/x/mod v0.32.0 h1:9F4d3PHLljb6x//jOyokMv3eX+YDeepZSEo3mFJy93c=
|
golang.org/x/mod v0.29.0 h1:HV8lRxZC4l2cr3Zq1LvtOsi/ThTgWnUk/y64QSs8GwA=
|
||||||
golang.org/x/mod v0.32.0/go.mod h1:SgipZ/3h2Ci89DlEtEXWUk/HteuRin+HHhN+WbNhguU=
|
golang.org/x/mod v0.29.0/go.mod h1:NyhrlYXJ2H4eJiRy/WDBO6HMqZQ6q9nk4JzS3NuCK+w=
|
||||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
||||||
@@ -128,8 +126,8 @@ golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
|||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=
|
||||||
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
@@ -137,26 +135,30 @@ golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBc
|
|||||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.41.0 h1:Ivj+2Cp/ylzLiEU89QhWblYnOE9zerudt9Ftecq2C6k=
|
golang.org/x/sys v0.37.0 h1:fdNQudmxPjkdUTPnLn5mdQv7Zwvbvpaxqs831goi9kQ=
|
||||||
golang.org/x/sys v0.41.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
golang.org/x/sys v0.37.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
|
golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc=
|
||||||
|
golang.org/x/sys v0.38.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||||
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
|
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
|
||||||
golang.org/x/term v0.40.0 h1:36e4zGLqU4yhjlmxEaagx2KuYbJq3EwY8K943ZsHcvg=
|
golang.org/x/term v0.37.0 h1:8EGAD0qCmHYZg6J17DvsMy9/wJ7/D/4pV/wfnld5lTU=
|
||||||
golang.org/x/term v0.40.0/go.mod h1:w2P8uVp06p2iyKKuvXIm7N/y0UCRt3UfJTfZ7oOpglM=
|
golang.org/x/term v0.37.0/go.mod h1:5pB4lxRNYYVZuTLmy8oR2BH8dflOR+IbTYFD8fi3254=
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||||
golang.org/x/text v0.34.0 h1:oL/Qq0Kdaqxa1KbNeMKwQq0reLCCaFtqu2eNuSeNHbk=
|
golang.org/x/text v0.28.0 h1:rhazDwis8INMIwQ4tpjLDzUhx6RlXqZNPEM0huQojng=
|
||||||
golang.org/x/text v0.34.0/go.mod h1:homfLqTYRFyVYemLBFl5GgL/DWEiH5wcsQ5gSh1yziA=
|
golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU=
|
||||||
|
golang.org/x/text v0.31.0 h1:aC8ghyu4JhP8VojJ2lEHBnochRno1sgL6nEi9WGFGMM=
|
||||||
|
golang.org/x/text v0.31.0/go.mod h1:tKRAlv61yKIjGGHX/4tP1LTbc13YSec1pxVEWXzfoeM=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||||
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
||||||
golang.org/x/tools v0.41.0 h1:a9b8iMweWG+S0OBnlU36rzLp20z1Rp10w+IY2czHTQc=
|
golang.org/x/tools v0.38.0 h1:Hx2Xv8hISq8Lm16jvBZ2VQf+RLmbd7wVUsALibYI/IQ=
|
||||||
golang.org/x/tools v0.41.0/go.mod h1:XSY6eDqxVNiYgezAVqqCeihT4j1U2CCsqvH3WhQpnlg=
|
golang.org/x/tools v0.38.0/go.mod h1:yEsQ/d/YK8cjh0L6rZlY8tgtlKiBNTL14pGDJPJpYQs=
|
||||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/protobuf v1.36.8 h1:xHScyCOEuuwZEc6UtSOvPbAT4zRh0xcNRYekJwfqyMc=
|
google.golang.org/protobuf v1.36.8 h1:xHScyCOEuuwZEc6UtSOvPbAT4zRh0xcNRYekJwfqyMc=
|
||||||
google.golang.org/protobuf v1.36.8/go.mod h1:fuxRtAxBytpl4zzqUh6/eyUujkJdNiuEkXntxiD/uRU=
|
google.golang.org/protobuf v1.36.8/go.mod h1:fuxRtAxBytpl4zzqUh6/eyUujkJdNiuEkXntxiD/uRU=
|
||||||
|
|||||||
@@ -1,73 +0,0 @@
|
|||||||
// 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{}
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates a new Broker.
|
|
||||||
func New() *Broker {
|
|
||||||
return &Broker{ //nolint:exhaustruct // mu has zero-value default
|
|
||||||
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{} {
|
|
||||||
waitCh := make(chan struct{}, 1)
|
|
||||||
|
|
||||||
b.mu.Lock()
|
|
||||||
b.listeners[userID] = append(
|
|
||||||
b.listeners[userID], waitCh,
|
|
||||||
)
|
|
||||||
b.mu.Unlock()
|
|
||||||
|
|
||||||
return waitCh
|
|
||||||
}
|
|
||||||
|
|
||||||
// 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 _, waiter := range waiters {
|
|
||||||
select {
|
|
||||||
case waiter <- struct{}{}:
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Remove removes a specific wait channel (for cleanup on timeout).
|
|
||||||
func (b *Broker) Remove(
|
|
||||||
userID int64,
|
|
||||||
waitCh chan struct{},
|
|
||||||
) {
|
|
||||||
b.mu.Lock()
|
|
||||||
defer b.mu.Unlock()
|
|
||||||
|
|
||||||
waiters := b.listeners[userID]
|
|
||||||
|
|
||||||
for i, waiter := range waiters {
|
|
||||||
if waiter == waitCh {
|
|
||||||
b.listeners[userID] = append(
|
|
||||||
waiters[:i], waiters[i+1:]...,
|
|
||||||
)
|
|
||||||
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(b.listeners[userID]) == 0 {
|
|
||||||
delete(b.listeners, userID)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,121 +0,0 @@
|
|||||||
package broker_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"sync"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"git.eeqj.de/sneak/chat/internal/broker"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestNewBroker(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
brk := broker.New()
|
|
||||||
if brk == nil {
|
|
||||||
t.Fatal("expected non-nil broker")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWaitAndNotify(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
brk := broker.New()
|
|
||||||
waitCh := brk.Wait(1)
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
time.Sleep(10 * time.Millisecond)
|
|
||||||
brk.Notify(1)
|
|
||||||
}()
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-waitCh:
|
|
||||||
case <-time.After(2 * time.Second):
|
|
||||||
t.Fatal("timeout")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNotifyWithoutWaiters(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
brk := broker.New()
|
|
||||||
brk.Notify(42) // should not panic.
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRemove(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
brk := broker.New()
|
|
||||||
waitCh := brk.Wait(1)
|
|
||||||
|
|
||||||
brk.Remove(1, waitCh)
|
|
||||||
|
|
||||||
brk.Notify(1)
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-waitCh:
|
|
||||||
t.Fatal("should not receive after remove")
|
|
||||||
case <-time.After(50 * time.Millisecond):
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestMultipleWaiters(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
brk := broker.New()
|
|
||||||
waitCh1 := brk.Wait(1)
|
|
||||||
waitCh2 := brk.Wait(1)
|
|
||||||
|
|
||||||
brk.Notify(1)
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-waitCh1:
|
|
||||||
case <-time.After(time.Second):
|
|
||||||
t.Fatal("ch1 timeout")
|
|
||||||
}
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-waitCh2:
|
|
||||||
case <-time.After(time.Second):
|
|
||||||
t.Fatal("ch2 timeout")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestConcurrentWaitNotify(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
brk := broker.New()
|
|
||||||
|
|
||||||
var waitGroup sync.WaitGroup
|
|
||||||
|
|
||||||
const concurrency = 100
|
|
||||||
|
|
||||||
for idx := range concurrency {
|
|
||||||
waitGroup.Add(1)
|
|
||||||
|
|
||||||
go func(uid int64) {
|
|
||||||
defer waitGroup.Done()
|
|
||||||
|
|
||||||
waitCh := brk.Wait(uid)
|
|
||||||
|
|
||||||
brk.Notify(uid)
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-waitCh:
|
|
||||||
case <-time.After(time.Second):
|
|
||||||
t.Error("timeout")
|
|
||||||
}
|
|
||||||
}(int64(idx % 10))
|
|
||||||
}
|
|
||||||
|
|
||||||
waitGroup.Wait()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRemoveNonexistent(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
brk := broker.New()
|
|
||||||
waitCh := make(chan struct{}, 1)
|
|
||||||
|
|
||||||
brk.Remove(999, waitCh) // should not panic.
|
|
||||||
}
|
|
||||||
@@ -31,19 +31,17 @@ type Config struct {
|
|||||||
Port int
|
Port int
|
||||||
SentryDSN string
|
SentryDSN string
|
||||||
MaxHistory int
|
MaxHistory int
|
||||||
|
SessionTimeout int
|
||||||
MaxMessageSize int
|
MaxMessageSize int
|
||||||
MOTD string
|
MOTD string
|
||||||
ServerName string
|
ServerName string
|
||||||
FederationKey string
|
FederationKey string
|
||||||
SessionIdleTimeout string
|
|
||||||
params *Params
|
params *Params
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Config by reading from files and environment variables.
|
// New creates a new Config by reading from files and environment variables.
|
||||||
func New(
|
func New(_ fx.Lifecycle, params Params) (*Config, error) {
|
||||||
_ fx.Lifecycle, params Params,
|
|
||||||
) (*Config, error) {
|
|
||||||
log := params.Logger.Get()
|
log := params.Logger.Get()
|
||||||
name := params.Globals.Appname
|
name := params.Globals.Appname
|
||||||
|
|
||||||
@@ -61,11 +59,11 @@ func New(
|
|||||||
viper.SetDefault("METRICS_USERNAME", "")
|
viper.SetDefault("METRICS_USERNAME", "")
|
||||||
viper.SetDefault("METRICS_PASSWORD", "")
|
viper.SetDefault("METRICS_PASSWORD", "")
|
||||||
viper.SetDefault("MAX_HISTORY", "10000")
|
viper.SetDefault("MAX_HISTORY", "10000")
|
||||||
|
viper.SetDefault("SESSION_TIMEOUT", "86400")
|
||||||
viper.SetDefault("MAX_MESSAGE_SIZE", "4096")
|
viper.SetDefault("MAX_MESSAGE_SIZE", "4096")
|
||||||
viper.SetDefault("MOTD", "")
|
viper.SetDefault("MOTD", "")
|
||||||
viper.SetDefault("SERVER_NAME", "")
|
viper.SetDefault("SERVER_NAME", "")
|
||||||
viper.SetDefault("FEDERATION_KEY", "")
|
viper.SetDefault("FEDERATION_KEY", "")
|
||||||
viper.SetDefault("SESSION_IDLE_TIMEOUT", "24h")
|
|
||||||
|
|
||||||
err := viper.ReadInConfig()
|
err := viper.ReadInConfig()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -76,7 +74,7 @@ func New(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
cfg := &Config{
|
s := &Config{
|
||||||
DBURL: viper.GetString("DBURL"),
|
DBURL: viper.GetString("DBURL"),
|
||||||
Debug: viper.GetBool("DEBUG"),
|
Debug: viper.GetBool("DEBUG"),
|
||||||
Port: viper.GetInt("PORT"),
|
Port: viper.GetInt("PORT"),
|
||||||
@@ -85,19 +83,19 @@ func New(
|
|||||||
MetricsUsername: viper.GetString("METRICS_USERNAME"),
|
MetricsUsername: viper.GetString("METRICS_USERNAME"),
|
||||||
MetricsPassword: viper.GetString("METRICS_PASSWORD"),
|
MetricsPassword: viper.GetString("METRICS_PASSWORD"),
|
||||||
MaxHistory: viper.GetInt("MAX_HISTORY"),
|
MaxHistory: viper.GetInt("MAX_HISTORY"),
|
||||||
|
SessionTimeout: viper.GetInt("SESSION_TIMEOUT"),
|
||||||
MaxMessageSize: viper.GetInt("MAX_MESSAGE_SIZE"),
|
MaxMessageSize: viper.GetInt("MAX_MESSAGE_SIZE"),
|
||||||
MOTD: viper.GetString("MOTD"),
|
MOTD: viper.GetString("MOTD"),
|
||||||
ServerName: viper.GetString("SERVER_NAME"),
|
ServerName: viper.GetString("SERVER_NAME"),
|
||||||
FederationKey: viper.GetString("FEDERATION_KEY"),
|
FederationKey: viper.GetString("FEDERATION_KEY"),
|
||||||
SessionIdleTimeout: viper.GetString("SESSION_IDLE_TIMEOUT"),
|
|
||||||
log: log,
|
log: log,
|
||||||
params: ¶ms,
|
params: ¶ms,
|
||||||
}
|
}
|
||||||
|
|
||||||
if cfg.Debug {
|
if s.Debug {
|
||||||
params.Logger.EnableDebugLogging()
|
params.Logger.EnableDebugLogging()
|
||||||
cfg.log = params.Logger.Get()
|
s.log = params.Logger.Get()
|
||||||
}
|
}
|
||||||
|
|
||||||
return cfg, nil
|
return s, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,161 +0,0 @@
|
|||||||
package db
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
"golang.org/x/crypto/bcrypt"
|
|
||||||
)
|
|
||||||
|
|
||||||
const bcryptCost = bcrypt.DefaultCost
|
|
||||||
|
|
||||||
var errNoPassword = errors.New(
|
|
||||||
"account has no password set",
|
|
||||||
)
|
|
||||||
|
|
||||||
// RegisterUser creates a session with a hashed password
|
|
||||||
// and returns session ID, client ID, and token.
|
|
||||||
func (database *Database) RegisterUser(
|
|
||||||
ctx context.Context,
|
|
||||||
nick, password string,
|
|
||||||
) (int64, int64, string, error) {
|
|
||||||
hash, err := bcrypt.GenerateFromPassword(
|
|
||||||
[]byte(password), bcryptCost,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"hash password: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionUUID := uuid.New().String()
|
|
||||||
clientUUID := uuid.New().String()
|
|
||||||
|
|
||||||
token, err := generateToken()
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
now := time.Now()
|
|
||||||
|
|
||||||
transaction, err := database.conn.BeginTx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"begin tx: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
res, err := transaction.ExecContext(ctx,
|
|
||||||
`INSERT INTO sessions
|
|
||||||
(uuid, nick, password_hash,
|
|
||||||
created_at, last_seen)
|
|
||||||
VALUES (?, ?, ?, ?, ?)`,
|
|
||||||
sessionUUID, nick, string(hash), now, now)
|
|
||||||
if err != nil {
|
|
||||||
_ = transaction.Rollback()
|
|
||||||
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"create session: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionID, _ := res.LastInsertId()
|
|
||||||
|
|
||||||
clientRes, err := transaction.ExecContext(ctx,
|
|
||||||
`INSERT INTO clients
|
|
||||||
(uuid, session_id, token,
|
|
||||||
created_at, last_seen)
|
|
||||||
VALUES (?, ?, ?, ?, ?)`,
|
|
||||||
clientUUID, sessionID, token, now, now)
|
|
||||||
if err != nil {
|
|
||||||
_ = transaction.Rollback()
|
|
||||||
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"create client: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
clientID, _ := clientRes.LastInsertId()
|
|
||||||
|
|
||||||
err = transaction.Commit()
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"commit registration: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return sessionID, clientID, token, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// LoginUser verifies a nick/password and creates a new
|
|
||||||
// client token.
|
|
||||||
func (database *Database) LoginUser(
|
|
||||||
ctx context.Context,
|
|
||||||
nick, password string,
|
|
||||||
) (int64, int64, string, error) {
|
|
||||||
var (
|
|
||||||
sessionID int64
|
|
||||||
passwordHash string
|
|
||||||
)
|
|
||||||
|
|
||||||
err := database.conn.QueryRowContext(
|
|
||||||
ctx,
|
|
||||||
`SELECT id, password_hash
|
|
||||||
FROM sessions WHERE nick = ?`,
|
|
||||||
nick,
|
|
||||||
).Scan(&sessionID, &passwordHash)
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"get session for login: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
if passwordHash == "" {
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"login: %w", errNoPassword,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = bcrypt.CompareHashAndPassword(
|
|
||||||
[]byte(passwordHash), []byte(password),
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"verify password: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
clientUUID := uuid.New().String()
|
|
||||||
|
|
||||||
token, err := generateToken()
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
now := time.Now()
|
|
||||||
|
|
||||||
res, err := database.conn.ExecContext(ctx,
|
|
||||||
`INSERT INTO clients
|
|
||||||
(uuid, session_id, token,
|
|
||||||
created_at, last_seen)
|
|
||||||
VALUES (?, ?, ?, ?, ?)`,
|
|
||||||
clientUUID, sessionID, token, now, now)
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, "", fmt.Errorf(
|
|
||||||
"create login client: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
clientID, _ := res.LastInsertId()
|
|
||||||
|
|
||||||
_, _ = database.conn.ExecContext(
|
|
||||||
ctx,
|
|
||||||
"UPDATE sessions SET last_seen = ? WHERE id = ?",
|
|
||||||
now, sessionID,
|
|
||||||
)
|
|
||||||
|
|
||||||
return sessionID, clientID, token, nil
|
|
||||||
}
|
|
||||||
@@ -1,178 +0,0 @@
|
|||||||
package db_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
_ "modernc.org/sqlite"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestRegisterUser(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
sessionID, clientID, token, err :=
|
|
||||||
database.RegisterUser(ctx, "reguser", "password123")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if sessionID == 0 || clientID == 0 || token == "" {
|
|
||||||
t.Fatal("expected valid ids and token")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Verify session works via token lookup.
|
|
||||||
sid, cid, nick, err :=
|
|
||||||
database.GetSessionByToken(ctx, token)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if sid != sessionID || cid != clientID {
|
|
||||||
t.Fatal("session/client id mismatch")
|
|
||||||
}
|
|
||||||
|
|
||||||
if nick != "reguser" {
|
|
||||||
t.Fatalf("expected reguser, got %s", nick)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRegisterUserDuplicateNick(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
regSID, regCID, regToken, err :=
|
|
||||||
database.RegisterUser(ctx, "dupnick", "password123")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = regSID
|
|
||||||
_ = regCID
|
|
||||||
_ = regToken
|
|
||||||
|
|
||||||
dupSID, dupCID, dupToken, dupErr :=
|
|
||||||
database.RegisterUser(ctx, "dupnick", "other12345")
|
|
||||||
if dupErr == nil {
|
|
||||||
t.Fatal("expected error for duplicate nick")
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = dupSID
|
|
||||||
_ = dupCID
|
|
||||||
_ = dupToken
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoginUser(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
regSID, regCID, regToken, err :=
|
|
||||||
database.RegisterUser(ctx, "loginuser", "mypassword")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = regSID
|
|
||||||
_ = regCID
|
|
||||||
_ = regToken
|
|
||||||
|
|
||||||
sessionID, clientID, token, err :=
|
|
||||||
database.LoginUser(ctx, "loginuser", "mypassword")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if sessionID == 0 || clientID == 0 || token == "" {
|
|
||||||
t.Fatal("expected valid ids and token")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Verify the new token works.
|
|
||||||
_, _, nick, err :=
|
|
||||||
database.GetSessionByToken(ctx, token)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if nick != "loginuser" {
|
|
||||||
t.Fatalf("expected loginuser, got %s", nick)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoginUserWrongPassword(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
regSID, regCID, regToken, err :=
|
|
||||||
database.RegisterUser(ctx, "wrongpw", "correctpass")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = regSID
|
|
||||||
_ = regCID
|
|
||||||
_ = regToken
|
|
||||||
|
|
||||||
loginSID, loginCID, loginToken, loginErr :=
|
|
||||||
database.LoginUser(ctx, "wrongpw", "wrongpass12")
|
|
||||||
if loginErr == nil {
|
|
||||||
t.Fatal("expected error for wrong password")
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = loginSID
|
|
||||||
_ = loginCID
|
|
||||||
_ = loginToken
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoginUserNoPassword(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
// Create anonymous session (no password).
|
|
||||||
anonSID, anonCID, anonToken, err :=
|
|
||||||
database.CreateSession(ctx, "anon")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = anonSID
|
|
||||||
_ = anonCID
|
|
||||||
_ = anonToken
|
|
||||||
|
|
||||||
loginSID, loginCID, loginToken, loginErr :=
|
|
||||||
database.LoginUser(ctx, "anon", "anything1")
|
|
||||||
if loginErr == nil {
|
|
||||||
t.Fatal(
|
|
||||||
"expected error for login on passwordless account",
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = loginSID
|
|
||||||
_ = loginCID
|
|
||||||
_ = loginToken
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoginUserNonexistent(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
loginSID, loginCID, loginToken, err :=
|
|
||||||
database.LoginUser(ctx, "ghost", "password123")
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected error for nonexistent user")
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = loginSID
|
|
||||||
_ = loginCID
|
|
||||||
_ = loginToken
|
|
||||||
}
|
|
||||||
@@ -11,16 +11,20 @@ import (
|
|||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"git.eeqj.de/sneak/chat/internal/config"
|
"git.eeqj.de/sneak/chat/internal/config"
|
||||||
"git.eeqj.de/sneak/chat/internal/logger"
|
"git.eeqj.de/sneak/chat/internal/logger"
|
||||||
|
"git.eeqj.de/sneak/chat/internal/models"
|
||||||
"go.uber.org/fx"
|
"go.uber.org/fx"
|
||||||
|
|
||||||
_ "github.com/joho/godotenv/autoload" // .env
|
_ "github.com/joho/godotenv/autoload" // loads .env file
|
||||||
_ "modernc.org/sqlite" // driver
|
_ "modernc.org/sqlite" // SQLite driver
|
||||||
)
|
)
|
||||||
|
|
||||||
const minMigrationParts = 2
|
const (
|
||||||
|
minMigrationParts = 2
|
||||||
|
)
|
||||||
|
|
||||||
// SchemaFiles contains embedded SQL migration files.
|
// SchemaFiles contains embedded SQL migration files.
|
||||||
//
|
//
|
||||||
@@ -35,95 +39,527 @@ type Params struct {
|
|||||||
Config *config.Config
|
Config *config.Config
|
||||||
}
|
}
|
||||||
|
|
||||||
// Database manages the SQLite connection and migrations.
|
// Database manages the SQLite database connection and migrations.
|
||||||
type Database struct {
|
type Database struct {
|
||||||
conn *sql.DB
|
db *sql.DB
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
params *Params
|
params *Params
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Database and registers lifecycle hooks.
|
// New creates a new Database instance and registers lifecycle hooks.
|
||||||
func New(
|
func New(lc fx.Lifecycle, params Params) (*Database, error) {
|
||||||
lifecycle fx.Lifecycle,
|
s := new(Database)
|
||||||
params Params,
|
s.params = ¶ms
|
||||||
) (*Database, error) {
|
s.log = params.Logger.Get()
|
||||||
database := &Database{ //nolint:exhaustruct // conn set in OnStart
|
|
||||||
params: ¶ms,
|
|
||||||
log: params.Logger.Get(),
|
|
||||||
}
|
|
||||||
|
|
||||||
database.log.Info("Database instantiated")
|
s.log.Info("Database instantiated")
|
||||||
|
|
||||||
lifecycle.Append(fx.Hook{
|
lc.Append(fx.Hook{
|
||||||
OnStart: func(ctx context.Context) error {
|
OnStart: func(ctx context.Context) error {
|
||||||
database.log.Info("Database OnStart Hook")
|
s.log.Info("Database OnStart Hook")
|
||||||
|
|
||||||
return database.connect(ctx)
|
return s.connect(ctx)
|
||||||
},
|
},
|
||||||
OnStop: func(_ context.Context) error {
|
OnStop: func(_ context.Context) error {
|
||||||
database.log.Info("Database OnStop Hook")
|
s.log.Info("Database OnStop Hook")
|
||||||
|
|
||||||
if database.conn != nil {
|
if s.db != nil {
|
||||||
closeErr := database.conn.Close()
|
return s.db.Close()
|
||||||
if closeErr != nil {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"close db: %w", closeErr,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
return database, nil
|
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.ExecContext(context.Background(), "PRAGMA foreign_keys = ON")
|
||||||
|
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.
|
// GetDB returns the underlying sql.DB connection.
|
||||||
func (database *Database) GetDB() *sql.DB {
|
func (s *Database) GetDB() *sql.DB {
|
||||||
return database.conn
|
return s.db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (database *Database) connect(ctx context.Context) error {
|
// Hydrate injects the database reference into any model that
|
||||||
dbURL := database.params.Config.DBURL
|
// 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
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetUserByNick looks up a user by their nick.
|
||||||
|
func (s *Database) GetUserByNick(
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetUserByToken looks up a user by their auth token.
|
||||||
|
func (s *Database) GetUserByToken(
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateUser inserts a new user into the database.
|
||||||
|
func (s *Database) CreateUser(
|
||||||
|
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 // G201: placeholders are literal "?" strings, 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 == "" {
|
if dbURL == "" {
|
||||||
dbURL = "file:./data.db?_journal_mode=WAL&_busy_timeout=5000"
|
dbURL = "file:./data.db?_journal_mode=WAL"
|
||||||
}
|
}
|
||||||
|
|
||||||
database.log.Info(
|
s.log.Info("connecting to database", "url", dbURL)
|
||||||
"connecting to database", "url", dbURL,
|
|
||||||
)
|
|
||||||
|
|
||||||
conn, err := sql.Open("sqlite", dbURL)
|
d, err := sql.Open("sqlite", dbURL)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("open database: %w", err)
|
s.log.Error("failed to open database", "error", err)
|
||||||
|
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
err = conn.PingContext(ctx)
|
err = d.PingContext(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("ping database: %w", err)
|
s.log.Error("failed to ping database", "error", err)
|
||||||
|
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
conn.SetMaxOpenConns(1)
|
s.db = d
|
||||||
|
s.log.Info("database connected")
|
||||||
|
|
||||||
database.conn = conn
|
// Item 9: Enable foreign keys on every connection
|
||||||
database.log.Info("database connected")
|
_, err = s.db.ExecContext(ctx, "PRAGMA foreign_keys = ON")
|
||||||
|
|
||||||
_, err = database.conn.ExecContext(
|
|
||||||
ctx, "PRAGMA foreign_keys = ON",
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("enable foreign keys: %w", err)
|
return fmt.Errorf("enable foreign keys: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = database.conn.ExecContext(
|
return s.runMigrations(ctx)
|
||||||
ctx, "PRAGMA busy_timeout = 5000",
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("set busy timeout: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return database.runMigrations(ctx)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type migration struct {
|
type migration struct {
|
||||||
@@ -132,125 +568,51 @@ type migration struct {
|
|||||||
sql string
|
sql string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (database *Database) runMigrations(
|
func (s *Database) runMigrations(ctx context.Context) error {
|
||||||
|
err := s.bootstrapMigrationsTable(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
migrations, err := s.loadMigrations()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = s.applyMigrations(ctx, migrations)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
s.log.Info("database migrations complete")
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Database) bootstrapMigrationsTable(
|
||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
) error {
|
) error {
|
||||||
_, err := database.conn.ExecContext(ctx,
|
_, err := s.db.ExecContext(ctx,
|
||||||
`CREATE TABLE IF NOT EXISTS schema_migrations (
|
`CREATE TABLE IF NOT EXISTS schema_migrations (
|
||||||
version INTEGER PRIMARY KEY,
|
version INTEGER PRIMARY KEY,
|
||||||
applied_at DATETIME DEFAULT CURRENT_TIMESTAMP)`)
|
applied_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||||
|
)`)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf(
|
return fmt.Errorf(
|
||||||
"create schema_migrations: %w", err,
|
"create schema_migrations table: %w", err,
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
migrations, err := database.loadMigrations()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, mig := range migrations {
|
|
||||||
err = database.applyMigration(ctx, mig)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
database.log.Info("database migrations complete")
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (database *Database) applyMigration(
|
|
||||||
ctx context.Context,
|
|
||||||
mig migration,
|
|
||||||
) error {
|
|
||||||
var exists int
|
|
||||||
|
|
||||||
err := database.conn.QueryRowContext(ctx,
|
|
||||||
`SELECT COUNT(*) FROM schema_migrations
|
|
||||||
WHERE version = ?`,
|
|
||||||
mig.version,
|
|
||||||
).Scan(&exists)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"check migration %d: %w", mig.version, err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
if exists > 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
database.log.Info(
|
|
||||||
"applying migration",
|
|
||||||
"version", mig.version,
|
|
||||||
"name", mig.name,
|
|
||||||
)
|
|
||||||
|
|
||||||
return database.execMigration(ctx, mig)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (database *Database) execMigration(
|
|
||||||
ctx context.Context,
|
|
||||||
mig migration,
|
|
||||||
) error {
|
|
||||||
transaction, err := database.conn.BeginTx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"begin tx for migration %d: %w",
|
|
||||||
mig.version, err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = transaction.ExecContext(ctx, mig.sql)
|
|
||||||
if err != nil {
|
|
||||||
_ = transaction.Rollback()
|
|
||||||
|
|
||||||
return fmt.Errorf(
|
|
||||||
"apply migration %d (%s): %w",
|
|
||||||
mig.version, mig.name, err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = transaction.ExecContext(ctx,
|
|
||||||
`INSERT INTO schema_migrations (version)
|
|
||||||
VALUES (?)`,
|
|
||||||
mig.version,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
_ = transaction.Rollback()
|
|
||||||
|
|
||||||
return fmt.Errorf(
|
|
||||||
"record migration %d: %w",
|
|
||||||
mig.version, err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = transaction.Commit()
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"commit migration %d: %w",
|
|
||||||
mig.version, err,
|
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (database *Database) loadMigrations() (
|
func (s *Database) loadMigrations() ([]migration, error) {
|
||||||
[]migration,
|
|
||||||
error,
|
|
||||||
) {
|
|
||||||
entries, err := fs.ReadDir(SchemaFiles, "schema")
|
entries, err := fs.ReadDir(SchemaFiles, "schema")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf(
|
return nil, fmt.Errorf("read schema dir: %w", err)
|
||||||
"read schema dir: %w", err,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
migrations := make([]migration, 0, len(entries))
|
var migrations []migration
|
||||||
|
|
||||||
for _, entry := range entries {
|
for _, entry := range entries {
|
||||||
if entry.IsDir() ||
|
if entry.IsDir() ||
|
||||||
@@ -265,18 +627,17 @@ func (database *Database) loadMigrations() (
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
version, parseErr := strconv.Atoi(parts[0])
|
version, err := strconv.Atoi(parts[0])
|
||||||
if parseErr != nil {
|
if err != nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
content, readErr := SchemaFiles.ReadFile(
|
content, err := SchemaFiles.ReadFile(
|
||||||
"schema/" + entry.Name(),
|
"schema/" + entry.Name(),
|
||||||
)
|
)
|
||||||
if readErr != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf(
|
return nil, fmt.Errorf(
|
||||||
"read migration %s: %w",
|
"read migration %s: %w", entry.Name(), err,
|
||||||
entry.Name(), readErr,
|
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -293,3 +654,79 @@ func (database *Database) loadMigrations() (
|
|||||||
|
|
||||||
return migrations, nil
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
err = s.applySingleMigration(ctx, m)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Database) applySingleMigration(ctx context.Context, m migration) error {
|
||||||
|
s.log.Info(
|
||||||
|
"applying migration",
|
||||||
|
"version", m.version, "name", m.name,
|
||||||
|
)
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|||||||
494
internal/db/db_test.go
Normal file
494
internal/db/db_test.go
Normal file
@@ -0,0 +1,494 @@
|
|||||||
|
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()
|
||||||
|
|
||||||
|
u, err := d.CreateUser(ctx, "u1", nickAlice, "hash1")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateUser: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if u.ID != "u1" || u.Nick != nickAlice {
|
||||||
|
t.Errorf("got user %+v", u)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateAuthToken(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, err := d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateUser: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tok, err := d.CreateAuthToken(ctx, "tok1", "u1")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateAuthToken: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if tok.Token != "tok1" || tok.UserID != "u1" {
|
||||||
|
t.Errorf("unexpected token: %+v", tok)
|
||||||
|
}
|
||||||
|
|
||||||
|
u, err := tok.User(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("AuthToken.User: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if u.ID != "u1" || u.Nick != nickAlice {
|
||||||
|
t.Errorf("AuthToken.User got %+v", u)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateChannel(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
ch, err := d.CreateChannel(
|
||||||
|
ctx, "c1", "#general", "welcome", "+n",
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateChannel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if ch.ID != "c1" || ch.Name != "#general" {
|
||||||
|
t.Errorf("unexpected channel: %+v", ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAddChannelMember(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateChannel(ctx, "c1", "#general", "", "")
|
||||||
|
|
||||||
|
cm, err := d.AddChannelMember(ctx, "c1", "u1", "+o")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("AddChannelMember: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cm.ChannelID != "c1" || cm.Modes != "+o" {
|
||||||
|
t.Errorf("unexpected member: %+v", cm)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateMessage(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
|
||||||
|
msg, err := d.CreateMessage(
|
||||||
|
ctx, "m1", "u1", nickAlice,
|
||||||
|
"#general", "message", "hello",
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateMessage: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if msg.ID != "m1" || msg.Body != "hello" {
|
||||||
|
t.Errorf("unexpected message: %+v", msg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestQueueMessage(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateUser(ctx, "u2", nickBob, "h")
|
||||||
|
_, _ = d.CreateMessage(
|
||||||
|
ctx, "m1", "u1", nickAlice, "u2", "message", "hi",
|
||||||
|
)
|
||||||
|
|
||||||
|
mq, err := d.QueueMessage(ctx, "u2", "m1")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("QueueMessage: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if mq.UserID != "u2" || mq.MessageID != "m1" {
|
||||||
|
t.Errorf("unexpected queue entry: %+v", mq)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateSession(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
|
||||||
|
sess, err := d.CreateSession(ctx, "s1", "u1")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateSession: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if sess.ID != "s1" || sess.UserID != "u1" {
|
||||||
|
t.Errorf("unexpected session: %+v", sess)
|
||||||
|
}
|
||||||
|
|
||||||
|
u, err := sess.User(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Session.User: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if u.ID != "u1" {
|
||||||
|
t.Errorf("Session.User got %v, want u1", u.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateServerLink(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
sl, err := d.CreateServerLink(
|
||||||
|
ctx, "sl1", "peer1",
|
||||||
|
"https://peer.example.com", "keyhash", true,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateServerLink: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if sl.ID != "sl1" || !sl.IsActive {
|
||||||
|
t.Errorf("unexpected server link: %+v", sl)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUserChannels(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
u, _ := d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateChannel(ctx, "c1", "#alpha", "", "")
|
||||||
|
_, _ = d.CreateChannel(ctx, "c2", "#beta", "", "")
|
||||||
|
_, _ = d.AddChannelMember(ctx, "c1", "u1", "")
|
||||||
|
_, _ = d.AddChannelMember(ctx, "c2", "u1", "")
|
||||||
|
|
||||||
|
channels, err := u.Channels(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("User.Channels: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(channels) != 2 {
|
||||||
|
t.Fatalf("expected 2 channels, got %d", len(channels))
|
||||||
|
}
|
||||||
|
|
||||||
|
if channels[0].Name != "#alpha" {
|
||||||
|
t.Errorf("first channel: got %s", channels[0].Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
if channels[1].Name != "#beta" {
|
||||||
|
t.Errorf("second channel: got %s", channels[1].Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUserChannelsEmpty(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
u, _ := d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
|
||||||
|
channels, err := u.Channels(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("User.Channels: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(channels) != 0 {
|
||||||
|
t.Errorf("expected 0 channels, got %d", len(channels))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUserQueuedMessages(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
u, _ := d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateUser(ctx, "u2", nickBob, "h")
|
||||||
|
|
||||||
|
for i := range 3 {
|
||||||
|
id := fmt.Sprintf("m%d", i)
|
||||||
|
|
||||||
|
_, _ = d.CreateMessage(
|
||||||
|
ctx, id, "u2", nickBob, "u1",
|
||||||
|
"message", fmt.Sprintf("msg%d", i),
|
||||||
|
)
|
||||||
|
|
||||||
|
time.Sleep(10 * time.Millisecond)
|
||||||
|
|
||||||
|
_, _ = d.QueueMessage(ctx, "u1", id)
|
||||||
|
}
|
||||||
|
|
||||||
|
msgs, err := u.QueuedMessages(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("User.QueuedMessages: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(msgs) != 3 {
|
||||||
|
t.Fatalf("expected 3 messages, got %d", len(msgs))
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, msg := range msgs {
|
||||||
|
want := fmt.Sprintf("msg%d", i)
|
||||||
|
if msg.Body != want {
|
||||||
|
t.Errorf("msg %d: got %q, want %q", i, msg.Body, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUserQueuedMessagesEmpty(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
u, _ := d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
|
||||||
|
msgs, err := u.QueuedMessages(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("User.QueuedMessages: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(msgs) != 0 {
|
||||||
|
t.Errorf("expected 0 messages, got %d", len(msgs))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestChannelMembers(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
ch, _ := d.CreateChannel(ctx, "c1", "#general", "", "")
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateUser(ctx, "u2", nickBob, "h")
|
||||||
|
_, _ = d.CreateUser(ctx, "u3", nickCharlie, "h")
|
||||||
|
_, _ = d.AddChannelMember(ctx, "c1", "u1", "+o")
|
||||||
|
_, _ = d.AddChannelMember(ctx, "c1", "u2", "+v")
|
||||||
|
_, _ = d.AddChannelMember(ctx, "c1", "u3", "")
|
||||||
|
|
||||||
|
members, err := ch.Members(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Channel.Members: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(members) != 3 {
|
||||||
|
t.Fatalf("expected 3 members, got %d", len(members))
|
||||||
|
}
|
||||||
|
|
||||||
|
nicks := map[string]bool{}
|
||||||
|
for _, m := range members {
|
||||||
|
nicks[m.Nick] = true
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, want := range []string{
|
||||||
|
nickAlice, nickBob, nickCharlie,
|
||||||
|
} {
|
||||||
|
if !nicks[want] {
|
||||||
|
t.Errorf("missing nick %q", want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestChannelMembersEmpty(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
ch, _ := d.CreateChannel(ctx, "c1", "#empty", "", "")
|
||||||
|
|
||||||
|
members, err := ch.Members(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Channel.Members: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(members) != 0 {
|
||||||
|
t.Errorf("expected 0, got %d", len(members))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestChannelRecentMessages(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
ch, _ := d.CreateChannel(ctx, "c1", "#general", "", "")
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
|
||||||
|
for i := range 5 {
|
||||||
|
id := fmt.Sprintf("m%d", i)
|
||||||
|
|
||||||
|
_, _ = d.CreateMessage(
|
||||||
|
ctx, id, "u1", nickAlice, "#general",
|
||||||
|
"message", fmt.Sprintf("msg%d", i),
|
||||||
|
)
|
||||||
|
|
||||||
|
time.Sleep(10 * time.Millisecond)
|
||||||
|
}
|
||||||
|
|
||||||
|
msgs, err := ch.RecentMessages(ctx, 3)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("RecentMessages: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(msgs) != 3 {
|
||||||
|
t.Fatalf("expected 3, got %d", len(msgs))
|
||||||
|
}
|
||||||
|
|
||||||
|
if msgs[0].Body != "msg4" {
|
||||||
|
t.Errorf("first: got %q, want msg4", msgs[0].Body)
|
||||||
|
}
|
||||||
|
|
||||||
|
if msgs[2].Body != "msg2" {
|
||||||
|
t.Errorf("last: got %q, want msg2", msgs[2].Body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestChannelRecentMessagesLargeLimit(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
ch, _ := d.CreateChannel(ctx, "c1", "#general", "", "")
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateMessage(
|
||||||
|
ctx, "m1", "u1", nickAlice,
|
||||||
|
"#general", "message", "only",
|
||||||
|
)
|
||||||
|
|
||||||
|
msgs, err := ch.RecentMessages(ctx, 100)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("RecentMessages: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(msgs) != 1 {
|
||||||
|
t.Errorf("expected 1, got %d", len(msgs))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestChannelMemberUser(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateChannel(ctx, "c1", "#general", "", "")
|
||||||
|
|
||||||
|
cm, _ := d.AddChannelMember(ctx, "c1", "u1", "+o")
|
||||||
|
|
||||||
|
u, err := cm.User(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ChannelMember.User: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if u.ID != "u1" || u.Nick != nickAlice {
|
||||||
|
t.Errorf("got %+v", u)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestChannelMemberChannel(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateChannel(ctx, "c1", "#general", "topic", "+n")
|
||||||
|
|
||||||
|
cm, _ := d.AddChannelMember(ctx, "c1", "u1", "")
|
||||||
|
|
||||||
|
ch, err := cm.Channel(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ChannelMember.Channel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if ch.ID != "c1" || ch.Topic != "topic" {
|
||||||
|
t.Errorf("got %+v", ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDMMessage(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
d := setupTestDB(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
_, _ = d.CreateUser(ctx, "u1", nickAlice, "h")
|
||||||
|
_, _ = d.CreateUser(ctx, "u2", nickBob, "h")
|
||||||
|
|
||||||
|
msg, err := d.CreateMessage(
|
||||||
|
ctx, "m1", "u1", nickAlice, "u2", "message", "hey",
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateMessage DM: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if msg.Target != "u2" {
|
||||||
|
t.Errorf("target: got %q, want u2", msg.Target)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,60 +0,0 @@
|
|||||||
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) {
|
|
||||||
counter := testDBCounter.Add(1)
|
|
||||||
|
|
||||||
dsn := fmt.Sprintf(
|
|
||||||
"file:testdb%d?mode=memory"+
|
|
||||||
"&cache=shared&_pragma=foreign_keys(1)",
|
|
||||||
counter,
|
|
||||||
)
|
|
||||||
|
|
||||||
conn, err := sql.Open("sqlite", dsn)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("open test db: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
database := &Database{ //nolint:exhaustruct // test helper, params not needed
|
|
||||||
conn: conn,
|
|
||||||
log: slog.Default(),
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.runMigrations(context.Background())
|
|
||||||
if err != nil {
|
|
||||||
closeErr := conn.Close()
|
|
||||||
if closeErr != nil {
|
|
||||||
return nil, fmt.Errorf(
|
|
||||||
"close after migration failure: %w",
|
|
||||||
closeErr,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil, fmt.Errorf(
|
|
||||||
"run test migrations: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return database, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close closes the underlying database connection.
|
|
||||||
func (database *Database) Close() error {
|
|
||||||
err := database.conn.Close()
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("close database: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -1,653 +0,0 @@
|
|||||||
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()
|
|
||||||
|
|
||||||
database, err := db.NewTestDatabase()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
t.Cleanup(func() {
|
|
||||||
closeErr := database.Close()
|
|
||||||
if closeErr != nil {
|
|
||||||
t.Logf("close db: %v", closeErr)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
return database
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCreateSession(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
sessionID, _, token, err := database.CreateSession(
|
|
||||||
ctx, "alice",
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if sessionID == 0 || token == "" {
|
|
||||||
t.Fatal("expected valid id and token")
|
|
||||||
}
|
|
||||||
|
|
||||||
_, _, dupToken, dupErr := database.CreateSession(
|
|
||||||
ctx, "alice",
|
|
||||||
)
|
|
||||||
if dupErr == nil {
|
|
||||||
t.Fatal("expected error for duplicate nick")
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = dupToken
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetSessionByToken(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
_, _, token, err := database.CreateSession(ctx, "bob")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionID, clientID, nick, err :=
|
|
||||||
database.GetSessionByToken(ctx, token)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if nick != "bob" || sessionID == 0 || clientID == 0 {
|
|
||||||
t.Fatalf("expected bob, got %s", nick)
|
|
||||||
}
|
|
||||||
|
|
||||||
badSID, badCID, badNick, badErr :=
|
|
||||||
database.GetSessionByToken(ctx, "badtoken")
|
|
||||||
if badErr == nil {
|
|
||||||
t.Fatal("expected error for bad token")
|
|
||||||
}
|
|
||||||
|
|
||||||
if badSID != 0 || badCID != 0 || badNick != "" {
|
|
||||||
t.Fatal("expected zero values on error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetSessionByNick(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
charlieID, charlieClientID, charlieToken, err :=
|
|
||||||
database.CreateSession(ctx, "charlie")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if charlieID == 0 || charlieClientID == 0 {
|
|
||||||
t.Fatal("expected valid session/client IDs")
|
|
||||||
}
|
|
||||||
|
|
||||||
if charlieToken == "" {
|
|
||||||
t.Fatal("expected non-empty token")
|
|
||||||
}
|
|
||||||
|
|
||||||
id, err := database.GetSessionByNick(ctx, "charlie")
|
|
||||||
if err != nil || id == 0 {
|
|
||||||
t.Fatal("expected to find charlie")
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = database.GetSessionByNick(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()
|
|
||||||
|
|
||||||
sid, _, _, err := database.CreateSession(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, sid)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ids, err := database.GetChannelMemberIDs(ctx, chID)
|
|
||||||
if err != nil || len(ids) != 1 || ids[0] != sid {
|
|
||||||
t.Fatal("expected session in channel")
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.JoinChannel(ctx, chID, sid)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.PartChannel(ctx, chID, sid)
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
|
|
||||||
sid, _, _, err := database.CreateSession(ctx, "temp")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.JoinChannel(ctx, chID, sid)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.PartChannel(ctx, chID, sid)
|
|
||||||
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 createSessionWithChannels(
|
|
||||||
t *testing.T,
|
|
||||||
database *db.Database,
|
|
||||||
nick, ch1Name, ch2Name string,
|
|
||||||
) (int64, int64, int64) {
|
|
||||||
t.Helper()
|
|
||||||
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
sid, _, _, err := database.CreateSession(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, sid)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.JoinChannel(ctx, ch2, sid)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return sid, ch1, ch2
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestListChannels(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
sid, _, _ := createSessionWithChannels(
|
|
||||||
t, database, "lister", "#a", "#b",
|
|
||||||
)
|
|
||||||
|
|
||||||
channels, err := database.ListChannels(
|
|
||||||
t.Context(), sid,
|
|
||||||
)
|
|
||||||
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()
|
|
||||||
|
|
||||||
sid, _, token, err := database.CreateSession(
|
|
||||||
ctx, "old",
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.ChangeNick(ctx, sid, "new")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, _, nick, err := database.GetSessionByToken(
|
|
||||||
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 TestInsertMessage(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
body := json.RawMessage(`["hello"]`)
|
|
||||||
|
|
||||||
dbID, msgUUID, err := database.InsertMessage(
|
|
||||||
ctx, "PRIVMSG", "poller", "#test", body, nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if dbID == 0 || msgUUID == "" {
|
|
||||||
t.Fatal("expected valid id and uuid")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPollMessages(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
sid, _, token, err := database.CreateSession(
|
|
||||||
ctx, "poller",
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, clientID, _, err := database.GetSessionByToken(
|
|
||||||
ctx, token,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
body := json.RawMessage(`["hello"]`)
|
|
||||||
|
|
||||||
dbID, _, err := database.InsertMessage(
|
|
||||||
ctx, "PRIVMSG", "poller", "#test", body, nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.EnqueueToSession(ctx, sid, dbID)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
const batchSize = 10
|
|
||||||
|
|
||||||
msgs, lastQID, err := database.PollMessages(
|
|
||||||
ctx, clientID, 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, clientID, 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 TestDeleteSession(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
sid, _, _, err := database.CreateSession(
|
|
||||||
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, sid)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.DeleteSession(ctx, sid)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = database.GetSessionByNick(ctx, "deleteme")
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("session 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()
|
|
||||||
|
|
||||||
sid1, _, _, err := database.CreateSession(ctx, "m1")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
sid2, _, _, err := database.CreateSession(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, sid1)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.JoinChannel(ctx, chID, sid2)
|
|
||||||
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 TestGetSessionChannels(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
sid, _, _ := createSessionWithChannels(
|
|
||||||
t, database, "multi", "#m1", "#m2",
|
|
||||||
)
|
|
||||||
|
|
||||||
channels, err :=
|
|
||||||
database.GetSessionChannels(
|
|
||||||
t.Context(), sid,
|
|
||||||
)
|
|
||||||
if err != nil || len(channels) != 2 {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected 2 channels, got %d",
|
|
||||||
len(channels),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEnqueueToClient(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
database := setupTestDB(t)
|
|
||||||
ctx := t.Context()
|
|
||||||
|
|
||||||
_, _, token, err := database.CreateSession(
|
|
||||||
ctx, "enqclient",
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, clientID, _, err := database.GetSessionByToken(
|
|
||||||
ctx, token,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
body := json.RawMessage(`["test"]`)
|
|
||||||
|
|
||||||
dbID, _, err := database.InsertMessage(
|
|
||||||
ctx, "PRIVMSG", "sender", "#ch", body, nil,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = database.EnqueueToClient(ctx, clientID, dbID)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
const batchSize = 10
|
|
||||||
|
|
||||||
msgs, _, err := database.PollMessages(
|
|
||||||
ctx, clientID, 0, batchSize,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(msgs) != 1 {
|
|
||||||
t.Fatalf("expected 1, got %d", len(msgs))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,68 +1,4 @@
|
|||||||
-- Chat server schema (pre-1.0 consolidated)
|
CREATE TABLE IF NOT EXISTS schema_migrations (
|
||||||
PRAGMA foreign_keys = ON;
|
version INTEGER PRIMARY KEY,
|
||||||
|
applied_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||||
-- Sessions: each session is a user identity (nick + optional password + signing key)
|
|
||||||
CREATE TABLE IF NOT EXISTS sessions (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
uuid TEXT NOT NULL UNIQUE,
|
|
||||||
nick TEXT NOT NULL UNIQUE,
|
|
||||||
password_hash TEXT NOT NULL DEFAULT '',
|
|
||||||
signing_key TEXT NOT NULL DEFAULT '',
|
|
||||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
last_seen DATETIME DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
);
|
||||||
CREATE INDEX IF NOT EXISTS idx_sessions_uuid ON sessions(uuid);
|
|
||||||
|
|
||||||
-- Clients: each session can have multiple connected clients
|
|
||||||
CREATE TABLE IF NOT EXISTS clients (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
uuid TEXT NOT NULL UNIQUE,
|
|
||||||
session_id INTEGER NOT NULL REFERENCES sessions(id) ON DELETE CASCADE,
|
|
||||||
token TEXT NOT NULL UNIQUE,
|
|
||||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
last_seen DATETIME DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
CREATE INDEX IF NOT EXISTS idx_clients_token ON clients(token);
|
|
||||||
CREATE INDEX IF NOT EXISTS idx_clients_session ON clients(session_id);
|
|
||||||
|
|
||||||
-- 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,
|
|
||||||
session_id INTEGER NOT NULL REFERENCES sessions(id) ON DELETE CASCADE,
|
|
||||||
joined_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
UNIQUE(channel_id, session_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,
|
|
||||||
client_id INTEGER NOT NULL REFERENCES clients(id) ON DELETE CASCADE,
|
|
||||||
message_id INTEGER NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
|
|
||||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
UNIQUE(client_id, message_id)
|
|
||||||
);
|
|
||||||
CREATE INDEX IF NOT EXISTS idx_client_queues_client ON client_queues(client_id, id);
|
|
||||||
|
|||||||
89
internal/db/schema/002_schema.sql
Normal file
89
internal/db/schema/002_schema.sql
Normal file
@@ -0,0 +1,89 @@
|
|||||||
|
-- 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
|
||||||
|
);
|
||||||
16
internal/db/schema/003_users.sql
Normal file
16
internal/db/schema/003_users.sql
Normal file
@@ -0,0 +1,16 @@
|
|||||||
|
-- Migration 003: Add simple user auth columns.
|
||||||
|
-- This migration adds token-based auth support for the web client.
|
||||||
|
-- Tables created by 002 (with TEXT ids) take precedence via IF NOT EXISTS.
|
||||||
|
-- We only add columns/indexes that don't already exist.
|
||||||
|
|
||||||
|
-- Add token column to users table if it doesn't exist.
|
||||||
|
-- SQLite doesn't support IF NOT EXISTS for ALTER TABLE ADD COLUMN,
|
||||||
|
-- so we check via pragma first.
|
||||||
|
CREATE TABLE IF NOT EXISTS _migration_003_check (done INTEGER);
|
||||||
|
INSERT OR IGNORE INTO _migration_003_check VALUES (1);
|
||||||
|
|
||||||
|
-- The web chat client's simple tables are only created if migration 002
|
||||||
|
-- didn't already create them with the ORM schema.
|
||||||
|
-- Since 002 creates all needed tables, 003 is effectively a no-op
|
||||||
|
-- when run after 002.
|
||||||
|
DROP TABLE IF EXISTS _migration_003_check;
|
||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@@ -1,186 +0,0 @@
|
|||||||
package handlers
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"net/http"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
const minPasswordLength = 8
|
|
||||||
|
|
||||||
// HandleRegister creates a new user with a password.
|
|
||||||
func (hdlr *Handlers) HandleRegister() http.HandlerFunc {
|
|
||||||
return func(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
request.Body = http.MaxBytesReader(
|
|
||||||
writer, request.Body, hdlr.maxBodySize(),
|
|
||||||
)
|
|
||||||
|
|
||||||
hdlr.handleRegister(writer, request)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) handleRegister(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
type registerRequest struct {
|
|
||||||
Nick string `json:"nick"`
|
|
||||||
Password string `json:"password"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var payload registerRequest
|
|
||||||
|
|
||||||
err := json.NewDecoder(request.Body).Decode(&payload)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"invalid request body",
|
|
||||||
http.StatusBadRequest,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
payload.Nick = strings.TrimSpace(payload.Nick)
|
|
||||||
|
|
||||||
if !validNickRe.MatchString(payload.Nick) {
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"invalid nick format",
|
|
||||||
http.StatusBadRequest,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(payload.Password) < minPasswordLength {
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"password must be at least 8 characters",
|
|
||||||
http.StatusBadRequest,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionID, clientID, token, err :=
|
|
||||||
hdlr.params.Database.RegisterUser(
|
|
||||||
request.Context(),
|
|
||||||
payload.Nick,
|
|
||||||
payload.Password,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.handleRegisterError(
|
|
||||||
writer, request, err,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
hdlr.deliverMOTD(request, clientID, sessionID)
|
|
||||||
|
|
||||||
hdlr.respondJSON(writer, request, map[string]any{
|
|
||||||
"id": sessionID,
|
|
||||||
"nick": payload.Nick,
|
|
||||||
"token": token,
|
|
||||||
}, http.StatusCreated)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) handleRegisterError(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
err error,
|
|
||||||
) {
|
|
||||||
if strings.Contains(err.Error(), "UNIQUE") {
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"nick already taken",
|
|
||||||
http.StatusConflict,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
hdlr.log.Error(
|
|
||||||
"register user failed", "error", err,
|
|
||||||
)
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"internal error",
|
|
||||||
http.StatusInternalServerError,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// HandleLogin authenticates a user with nick and password.
|
|
||||||
func (hdlr *Handlers) HandleLogin() http.HandlerFunc {
|
|
||||||
return func(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
request.Body = http.MaxBytesReader(
|
|
||||||
writer, request.Body, hdlr.maxBodySize(),
|
|
||||||
)
|
|
||||||
|
|
||||||
hdlr.handleLogin(writer, request)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) handleLogin(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
type loginRequest struct {
|
|
||||||
Nick string `json:"nick"`
|
|
||||||
Password string `json:"password"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var payload loginRequest
|
|
||||||
|
|
||||||
err := json.NewDecoder(request.Body).Decode(&payload)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"invalid request body",
|
|
||||||
http.StatusBadRequest,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
payload.Nick = strings.TrimSpace(payload.Nick)
|
|
||||||
|
|
||||||
if payload.Nick == "" || payload.Password == "" {
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"nick and password required",
|
|
||||||
http.StatusBadRequest,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionID, _, token, err :=
|
|
||||||
hdlr.params.Database.LoginUser(
|
|
||||||
request.Context(),
|
|
||||||
payload.Nick,
|
|
||||||
payload.Password,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"invalid credentials",
|
|
||||||
http.StatusUnauthorized,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
hdlr.respondJSON(writer, request, map[string]any{
|
|
||||||
"id": sessionID,
|
|
||||||
"nick": payload.Nick,
|
|
||||||
"token": token,
|
|
||||||
}, http.StatusOK)
|
|
||||||
}
|
|
||||||
@@ -4,12 +4,9 @@ package handlers
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
|
||||||
|
|
||||||
"git.eeqj.de/sneak/chat/internal/broker"
|
|
||||||
"git.eeqj.de/sneak/chat/internal/config"
|
"git.eeqj.de/sneak/chat/internal/config"
|
||||||
"git.eeqj.de/sneak/chat/internal/db"
|
"git.eeqj.de/sneak/chat/internal/db"
|
||||||
"git.eeqj.de/sneak/chat/internal/globals"
|
"git.eeqj.de/sneak/chat/internal/globals"
|
||||||
@@ -18,8 +15,6 @@ import (
|
|||||||
"go.uber.org/fx"
|
"go.uber.org/fx"
|
||||||
)
|
)
|
||||||
|
|
||||||
var errUnauthorized = errors.New("unauthorized")
|
|
||||||
|
|
||||||
// Params defines the dependencies for creating Handlers.
|
// Params defines the dependencies for creating Handlers.
|
||||||
type Params struct {
|
type Params struct {
|
||||||
fx.In
|
fx.In
|
||||||
@@ -31,173 +26,37 @@ type Params struct {
|
|||||||
Healthcheck *healthcheck.Healthcheck
|
Healthcheck *healthcheck.Healthcheck
|
||||||
}
|
}
|
||||||
|
|
||||||
const defaultIdleTimeout = 24 * time.Hour
|
|
||||||
|
|
||||||
// Handlers manages HTTP request handling.
|
// Handlers manages HTTP request handling.
|
||||||
type Handlers struct {
|
type Handlers struct {
|
||||||
params *Params
|
params *Params
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
hc *healthcheck.Healthcheck
|
hc *healthcheck.Healthcheck
|
||||||
broker *broker.Broker
|
|
||||||
cancelCleanup context.CancelFunc
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Handlers instance.
|
// New creates a new Handlers instance.
|
||||||
func New(
|
func New(lc fx.Lifecycle, params Params) (*Handlers, error) {
|
||||||
lifecycle fx.Lifecycle,
|
s := new(Handlers)
|
||||||
params Params,
|
s.params = ¶ms
|
||||||
) (*Handlers, error) {
|
s.log = params.Logger.Get()
|
||||||
hdlr := &Handlers{ //nolint:exhaustruct // cancelCleanup set in startCleanup
|
s.hc = params.Healthcheck
|
||||||
params: ¶ms,
|
|
||||||
log: params.Logger.Get(),
|
|
||||||
hc: params.Healthcheck,
|
|
||||||
broker: broker.New(),
|
|
||||||
}
|
|
||||||
|
|
||||||
lifecycle.Append(fx.Hook{
|
|
||||||
OnStart: func(ctx context.Context) error {
|
|
||||||
hdlr.startCleanup(ctx)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
OnStop: func(_ context.Context) error {
|
|
||||||
hdlr.stopCleanup()
|
|
||||||
|
|
||||||
|
lc.Append(fx.Hook{
|
||||||
|
OnStart: func(_ context.Context) error {
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
return hdlr, nil
|
return s, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (hdlr *Handlers) respondJSON(
|
func (s *Handlers) respondJSON(w http.ResponseWriter, _ *http.Request, data any, status int) {
|
||||||
writer http.ResponseWriter,
|
w.WriteHeader(status)
|
||||||
_ *http.Request,
|
w.Header().Set("Content-Type", "application/json")
|
||||||
data any,
|
|
||||||
status int,
|
|
||||||
) {
|
|
||||||
writer.Header().Set(
|
|
||||||
"Content-Type",
|
|
||||||
"application/json; charset=utf-8",
|
|
||||||
)
|
|
||||||
writer.WriteHeader(status)
|
|
||||||
|
|
||||||
if data != nil {
|
if data != nil {
|
||||||
err := json.NewEncoder(writer).Encode(data)
|
err := json.NewEncoder(w).Encode(data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
hdlr.log.Error(
|
s.log.Error("json encode error", "error", err)
|
||||||
"json encode error", "error", err,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (hdlr *Handlers) respondError(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
msg string,
|
|
||||||
status int,
|
|
||||||
) {
|
|
||||||
hdlr.respondJSON(
|
|
||||||
writer, request,
|
|
||||||
map[string]string{"error": msg},
|
|
||||||
status,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) idleTimeout() time.Duration {
|
|
||||||
raw := hdlr.params.Config.SessionIdleTimeout
|
|
||||||
if raw == "" {
|
|
||||||
return defaultIdleTimeout
|
|
||||||
}
|
|
||||||
|
|
||||||
dur, err := time.ParseDuration(raw)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"invalid SESSION_IDLE_TIMEOUT, using default",
|
|
||||||
"value", raw, "error", err,
|
|
||||||
)
|
|
||||||
|
|
||||||
return defaultIdleTimeout
|
|
||||||
}
|
|
||||||
|
|
||||||
return dur
|
|
||||||
}
|
|
||||||
|
|
||||||
// startCleanup launches the idle-user cleanup goroutine.
|
|
||||||
// We use context.Background rather than the OnStart ctx
|
|
||||||
// because the OnStart context is startup-scoped and would
|
|
||||||
// cancel the goroutine once all start hooks complete.
|
|
||||||
//
|
|
||||||
//nolint:contextcheck // intentional Background ctx
|
|
||||||
func (hdlr *Handlers) startCleanup(_ context.Context) {
|
|
||||||
cleanupCtx, cancel := context.WithCancel(
|
|
||||||
context.Background(),
|
|
||||||
)
|
|
||||||
hdlr.cancelCleanup = cancel
|
|
||||||
|
|
||||||
go hdlr.cleanupLoop(cleanupCtx)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) stopCleanup() {
|
|
||||||
if hdlr.cancelCleanup != nil {
|
|
||||||
hdlr.cancelCleanup()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) cleanupLoop(ctx context.Context) {
|
|
||||||
timeout := hdlr.idleTimeout()
|
|
||||||
|
|
||||||
interval := max(timeout/2, time.Minute) //nolint:mnd // half the timeout
|
|
||||||
|
|
||||||
ticker := time.NewTicker(interval)
|
|
||||||
defer ticker.Stop()
|
|
||||||
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-ticker.C:
|
|
||||||
hdlr.runCleanup(ctx, timeout)
|
|
||||||
case <-ctx.Done():
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) runCleanup(
|
|
||||||
ctx context.Context,
|
|
||||||
timeout time.Duration,
|
|
||||||
) {
|
|
||||||
cutoff := time.Now().Add(-timeout)
|
|
||||||
|
|
||||||
// Find sessions that will be orphaned so we can send
|
|
||||||
// QUIT notifications before deleting anything.
|
|
||||||
stale, err := hdlr.params.Database.
|
|
||||||
GetStaleOrphanSessions(ctx, cutoff)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"stale session lookup failed", "error", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, ss := range stale {
|
|
||||||
hdlr.cleanupUser(ctx, ss.ID, ss.Nick)
|
|
||||||
}
|
|
||||||
|
|
||||||
deleted, err := hdlr.params.Database.DeleteStaleUsers(
|
|
||||||
ctx, cutoff,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"user cleanup failed", "error", err,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if deleted > 0 {
|
|
||||||
hdlr.log.Info(
|
|
||||||
"cleaned up stale users",
|
|
||||||
"deleted", deleted,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -7,12 +7,9 @@ import (
|
|||||||
const httpStatusOK = 200
|
const httpStatusOK = 200
|
||||||
|
|
||||||
// HandleHealthCheck returns an HTTP handler for the health check endpoint.
|
// HandleHealthCheck returns an HTTP handler for the health check endpoint.
|
||||||
func (hdlr *Handlers) HandleHealthCheck() http.HandlerFunc {
|
func (s *Handlers) HandleHealthCheck() http.HandlerFunc {
|
||||||
return func(
|
return func(w http.ResponseWriter, req *http.Request) {
|
||||||
writer http.ResponseWriter,
|
resp := s.hc.Healthcheck()
|
||||||
request *http.Request,
|
s.respondJSON(w, req, resp, httpStatusOK)
|
||||||
) {
|
|
||||||
resp := hdlr.hc.Healthcheck()
|
|
||||||
hdlr.respondJSON(writer, request, resp, httpStatusOK)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -33,17 +33,14 @@ type Healthcheck struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Healthcheck instance.
|
// New creates a new Healthcheck instance.
|
||||||
func New(
|
func New(lc fx.Lifecycle, params Params) (*Healthcheck, error) {
|
||||||
lifecycle fx.Lifecycle, params Params,
|
s := new(Healthcheck)
|
||||||
) (*Healthcheck, error) {
|
s.params = ¶ms
|
||||||
hcheck := &Healthcheck{ //nolint:exhaustruct // StartupTime set in OnStart
|
s.log = params.Logger.Get()
|
||||||
params: ¶ms,
|
|
||||||
log: params.Logger.Get(),
|
|
||||||
}
|
|
||||||
|
|
||||||
lifecycle.Append(fx.Hook{
|
lc.Append(fx.Hook{
|
||||||
OnStart: func(_ context.Context) error {
|
OnStart: func(_ context.Context) error {
|
||||||
hcheck.StartupTime = time.Now()
|
s.StartupTime = time.Now()
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
@@ -52,7 +49,7 @@ func New(
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
return hcheck, nil
|
return s, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Response is the JSON response returned by the health endpoint.
|
// Response is the JSON response returned by the health endpoint.
|
||||||
@@ -67,18 +64,19 @@ type Response struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Healthcheck returns the current health status of the server.
|
// Healthcheck returns the current health status of the server.
|
||||||
func (hcheck *Healthcheck) Healthcheck() *Response {
|
func (s *Healthcheck) Healthcheck() *Response {
|
||||||
return &Response{
|
resp := &Response{
|
||||||
Status: "ok",
|
Status: "ok",
|
||||||
Now: time.Now().UTC().Format(time.RFC3339Nano),
|
Now: time.Now().UTC().Format(time.RFC3339Nano),
|
||||||
UptimeSeconds: int64(hcheck.uptime().Seconds()),
|
UptimeSeconds: int64(s.uptime().Seconds()),
|
||||||
UptimeHuman: hcheck.uptime().String(),
|
UptimeHuman: s.uptime().String(),
|
||||||
Appname: hcheck.params.Globals.Appname,
|
Appname: s.params.Globals.Appname,
|
||||||
Version: hcheck.params.Globals.Version,
|
Version: s.params.Globals.Version,
|
||||||
Maintenance: hcheck.params.Config.MaintenanceMode,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return resp
|
||||||
}
|
}
|
||||||
|
|
||||||
func (hcheck *Healthcheck) uptime() time.Duration {
|
func (s *Healthcheck) uptime() time.Duration {
|
||||||
return time.Since(hcheck.StartupTime)
|
return time.Since(s.StartupTime)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,56 +23,51 @@ type Logger struct {
|
|||||||
params Params
|
params Params
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Logger with appropriate handler
|
// New creates a new Logger with appropriate handler based on terminal detection.
|
||||||
// based on terminal detection.
|
func New(_ fx.Lifecycle, params Params) (*Logger, error) {
|
||||||
func New(
|
l := new(Logger)
|
||||||
_ fx.Lifecycle, params Params,
|
l.level = new(slog.LevelVar)
|
||||||
) (*Logger, error) {
|
l.level.Set(slog.LevelInfo)
|
||||||
logger := new(Logger)
|
|
||||||
logger.level = new(slog.LevelVar)
|
|
||||||
logger.level.Set(slog.LevelInfo)
|
|
||||||
|
|
||||||
tty := false
|
tty := false
|
||||||
|
|
||||||
if fileInfo, _ := os.Stdout.Stat(); (fileInfo.Mode() & os.ModeCharDevice) != 0 {
|
if fileInfo, _ := os.Stdout.Stat(); (fileInfo.Mode() & os.ModeCharDevice) != 0 {
|
||||||
tty = true
|
tty = true
|
||||||
}
|
}
|
||||||
|
|
||||||
opts := &slog.HandlerOptions{ //nolint:exhaustruct // ReplaceAttr optional
|
|
||||||
Level: logger.level,
|
|
||||||
AddSource: true,
|
|
||||||
}
|
|
||||||
|
|
||||||
var handler slog.Handler
|
var handler slog.Handler
|
||||||
if tty {
|
if tty {
|
||||||
handler = slog.NewTextHandler(os.Stdout, opts)
|
handler = slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
|
||||||
|
Level: l.level,
|
||||||
|
AddSource: true,
|
||||||
|
})
|
||||||
} else {
|
} else {
|
||||||
handler = slog.NewJSONHandler(os.Stdout, opts)
|
handler = slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
|
||||||
|
Level: l.level,
|
||||||
|
AddSource: true,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.log = slog.New(handler)
|
l.log = slog.New(handler)
|
||||||
logger.params = params
|
l.params = params
|
||||||
|
|
||||||
return logger, nil
|
return l, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// EnableDebugLogging switches the log level to debug.
|
// EnableDebugLogging switches the log level to debug.
|
||||||
func (logger *Logger) EnableDebugLogging() {
|
func (l *Logger) EnableDebugLogging() {
|
||||||
logger.level.Set(slog.LevelDebug)
|
l.level.Set(slog.LevelDebug)
|
||||||
logger.log.Debug(
|
l.log.Debug("debug logging enabled", "debug", true)
|
||||||
"debug logging enabled", "debug", true,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get returns the underlying slog.Logger.
|
// Get returns the underlying slog.Logger.
|
||||||
func (logger *Logger) Get() *slog.Logger {
|
func (l *Logger) Get() *slog.Logger {
|
||||||
return logger.log
|
return l.log
|
||||||
}
|
}
|
||||||
|
|
||||||
// Identify logs the application name and version at startup.
|
// Identify logs the application name and version at startup.
|
||||||
func (logger *Logger) Identify() {
|
func (l *Logger) Identify() {
|
||||||
logger.log.Info("starting",
|
l.log.Info("starting",
|
||||||
"appname", logger.params.Globals.Appname,
|
"appname", l.params.Globals.Appname,
|
||||||
"version", logger.params.Globals.Version,
|
"version", l.params.Globals.Version,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ import (
|
|||||||
"git.eeqj.de/sneak/chat/internal/globals"
|
"git.eeqj.de/sneak/chat/internal/globals"
|
||||||
"git.eeqj.de/sneak/chat/internal/logger"
|
"git.eeqj.de/sneak/chat/internal/logger"
|
||||||
basicauth "github.com/99designs/basicauth-go"
|
basicauth "github.com/99designs/basicauth-go"
|
||||||
chimw "github.com/go-chi/chi/middleware"
|
"github.com/go-chi/chi/middleware"
|
||||||
"github.com/go-chi/cors"
|
"github.com/go-chi/cors"
|
||||||
metrics "github.com/slok/go-http-metrics/metrics/prometheus"
|
metrics "github.com/slok/go-http-metrics/metrics/prometheus"
|
||||||
ghmm "github.com/slok/go-http-metrics/middleware"
|
ghmm "github.com/slok/go-http-metrics/middleware"
|
||||||
@@ -38,28 +38,25 @@ type Middleware struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Middleware instance.
|
// New creates a new Middleware instance.
|
||||||
func New(
|
func New(_ fx.Lifecycle, params Params) (*Middleware, error) {
|
||||||
_ fx.Lifecycle, params Params,
|
s := new(Middleware)
|
||||||
) (*Middleware, error) {
|
s.params = ¶ms
|
||||||
mware := &Middleware{
|
s.log = params.Logger.Get()
|
||||||
params: ¶ms,
|
|
||||||
log: params.Logger.Get(),
|
|
||||||
}
|
|
||||||
|
|
||||||
return mware, nil
|
return s, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func ipFromHostPort(hostPort string) string {
|
func ipFromHostPort(hp string) string {
|
||||||
host, _, err := net.SplitHostPort(hostPort)
|
h, _, err := net.SplitHostPort(hp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(host) > 0 && host[0] == '[' {
|
if len(h) > 0 && h[0] == '[' {
|
||||||
return host[1 : len(host)-1]
|
return h[1 : len(h)-1]
|
||||||
}
|
}
|
||||||
|
|
||||||
return host
|
return h
|
||||||
}
|
}
|
||||||
|
|
||||||
type loggingResponseWriter struct {
|
type loggingResponseWriter struct {
|
||||||
@@ -68,15 +65,9 @@ type loggingResponseWriter struct {
|
|||||||
statusCode int
|
statusCode int
|
||||||
}
|
}
|
||||||
|
|
||||||
// newLoggingResponseWriter wraps a ResponseWriter
|
// newLoggingResponseWriter wraps a ResponseWriter to capture the status code.
|
||||||
// to capture the status code.
|
func newLoggingResponseWriter(w http.ResponseWriter) *loggingResponseWriter {
|
||||||
func newLoggingResponseWriter(
|
return &loggingResponseWriter{w, http.StatusOK}
|
||||||
writer http.ResponseWriter,
|
|
||||||
) *loggingResponseWriter {
|
|
||||||
return &loggingResponseWriter{
|
|
||||||
ResponseWriter: writer,
|
|
||||||
statusCode: http.StatusOK,
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (lrw *loggingResponseWriter) WriteHeader(code int) {
|
func (lrw *loggingResponseWriter) WriteHeader(code int) {
|
||||||
@@ -85,57 +76,43 @@ func (lrw *loggingResponseWriter) WriteHeader(code int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Logging returns middleware that logs each HTTP request.
|
// Logging returns middleware that logs each HTTP request.
|
||||||
func (mware *Middleware) Logging() func(http.Handler) http.Handler {
|
func (s *Middleware) Logging() func(http.Handler) http.Handler {
|
||||||
return func(next http.Handler) http.Handler {
|
return func(next http.Handler) http.Handler {
|
||||||
return http.HandlerFunc(
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
func(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
start := time.Now()
|
start := time.Now()
|
||||||
lrw := newLoggingResponseWriter(writer)
|
lrw := newLoggingResponseWriter(w)
|
||||||
ctx := request.Context()
|
ctx := r.Context()
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
latency := time.Since(start)
|
latency := time.Since(start)
|
||||||
|
|
||||||
reqID, _ := ctx.Value(
|
reqID, _ := ctx.Value(middleware.RequestIDKey).(string)
|
||||||
chimw.RequestIDKey,
|
|
||||||
).(string)
|
|
||||||
|
|
||||||
mware.log.InfoContext(
|
s.log.InfoContext(ctx, "request",
|
||||||
ctx, "request",
|
|
||||||
"request_start", start,
|
"request_start", start,
|
||||||
"method", request.Method,
|
"method", r.Method,
|
||||||
"url", request.URL.String(),
|
"url", r.URL.String(),
|
||||||
"useragent", request.UserAgent(),
|
"useragent", r.UserAgent(),
|
||||||
"request_id", reqID,
|
"request_id", reqID,
|
||||||
"referer", request.Referer(),
|
"referer", r.Referer(),
|
||||||
"proto", request.Proto,
|
"proto", r.Proto,
|
||||||
"remoteIP",
|
"remoteIP", ipFromHostPort(r.RemoteAddr),
|
||||||
ipFromHostPort(request.RemoteAddr),
|
|
||||||
"status", lrw.statusCode,
|
"status", lrw.statusCode,
|
||||||
"latency_ms",
|
"latency_ms", latency.Milliseconds(),
|
||||||
latency.Milliseconds(),
|
|
||||||
)
|
)
|
||||||
}()
|
}()
|
||||||
|
|
||||||
next.ServeHTTP(lrw, request)
|
next.ServeHTTP(lrw, r)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// CORS returns middleware that handles Cross-Origin Resource Sharing.
|
// CORS returns middleware that handles Cross-Origin Resource Sharing.
|
||||||
func (mware *Middleware) CORS() func(http.Handler) http.Handler {
|
func (s *Middleware) CORS() func(http.Handler) http.Handler {
|
||||||
return cors.Handler(cors.Options{ //nolint:exhaustruct // optional fields
|
return cors.Handler(cors.Options{
|
||||||
AllowedOrigins: []string{"*"},
|
AllowedOrigins: []string{"*"},
|
||||||
AllowedMethods: []string{
|
AllowedMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
||||||
"GET", "POST", "PUT", "DELETE", "OPTIONS",
|
AllowedHeaders: []string{"Accept", "Authorization", "Content-Type", "X-CSRF-Token"},
|
||||||
},
|
|
||||||
AllowedHeaders: []string{
|
|
||||||
"Accept", "Authorization",
|
|
||||||
"Content-Type", "X-CSRF-Token",
|
|
||||||
},
|
|
||||||
ExposedHeaders: []string{"Link"},
|
ExposedHeaders: []string{"Link"},
|
||||||
AllowCredentials: false,
|
AllowCredentials: false,
|
||||||
MaxAge: corsMaxAge,
|
MaxAge: corsMaxAge,
|
||||||
@@ -143,34 +120,28 @@ func (mware *Middleware) CORS() func(http.Handler) http.Handler {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Auth returns middleware that performs authentication.
|
// Auth returns middleware that performs authentication.
|
||||||
func (mware *Middleware) Auth() func(http.Handler) http.Handler {
|
func (s *Middleware) Auth() func(http.Handler) http.Handler {
|
||||||
return func(next http.Handler) http.Handler {
|
return func(next http.Handler) http.Handler {
|
||||||
return http.HandlerFunc(
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
func(
|
s.log.Info("AUTH: before request")
|
||||||
writer http.ResponseWriter,
|
next.ServeHTTP(w, r)
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
mware.log.Info("AUTH: before request")
|
|
||||||
next.ServeHTTP(writer, request)
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Metrics returns middleware that records HTTP metrics.
|
// Metrics returns middleware that records HTTP metrics.
|
||||||
func (mware *Middleware) Metrics() func(http.Handler) http.Handler {
|
func (s *Middleware) Metrics() func(http.Handler) http.Handler {
|
||||||
metricsMiddleware := ghmm.New(ghmm.Config{ //nolint:exhaustruct // optional fields
|
mdlw := ghmm.New(ghmm.Config{
|
||||||
Recorder: metrics.NewRecorder(
|
Recorder: metrics.NewRecorder(metrics.Config{}),
|
||||||
metrics.Config{}, //nolint:exhaustruct // defaults
|
|
||||||
),
|
|
||||||
})
|
})
|
||||||
|
|
||||||
return func(next http.Handler) http.Handler {
|
return func(next http.Handler) http.Handler {
|
||||||
return std.Handler("", metricsMiddleware, next)
|
return std.Handler("", mdlw, next)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MetricsAuth returns middleware that protects metrics with basic auth.
|
// MetricsAuth returns middleware that protects metrics with basic auth.
|
||||||
func (mware *Middleware) MetricsAuth() func(http.Handler) http.Handler {
|
func (s *Middleware) MetricsAuth() func(http.Handler) http.Handler {
|
||||||
return basicauth.New(
|
return basicauth.New(
|
||||||
"metrics",
|
"metrics",
|
||||||
map[string][]string{
|
map[string][]string{
|
||||||
|
|||||||
26
internal/models/auth_token.go
Normal file
26
internal/models/auth_token.go
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ErrUserLookupNotAvailable is returned when user lookup is not configured.
|
||||||
|
var ErrUserLookupNotAvailable = errors.New("user lookup not available")
|
||||||
|
|
||||||
|
// 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) {
|
||||||
|
return t.LookupUser(ctx, t.UserID)
|
||||||
|
}
|
||||||
96
internal/models/channel.go
Normal file
96
internal/models/channel.go
Normal file
@@ -0,0 +1,96 @@
|
|||||||
|
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()
|
||||||
|
}
|
||||||
31
internal/models/channel_member.go
Normal file
31
internal/models/channel_member.go
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ErrChannelLookupNotAvailable is returned when channel lookup is not configured.
|
||||||
|
var ErrChannelLookupNotAvailable = errors.New("channel lookup not available")
|
||||||
|
|
||||||
|
// 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) {
|
||||||
|
return cm.LookupUser(ctx, cm.UserID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Channel returns the full Channel for this membership.
|
||||||
|
func (cm *ChannelMember) Channel(ctx context.Context) (*Channel, error) {
|
||||||
|
return cm.LookupChannel(ctx, cm.ChannelID)
|
||||||
|
}
|
||||||
20
internal/models/message.go
Normal file
20
internal/models/message.go
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
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"`
|
||||||
|
}
|
||||||
15
internal/models/message_queue.go
Normal file
15
internal/models/message_queue.go
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
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"`
|
||||||
|
}
|
||||||
60
internal/models/model.go
Normal file
60
internal/models/model.go
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
// 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"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 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)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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()
|
||||||
|
}
|
||||||
|
|
||||||
|
// LookupUser looks up a user by ID if the database supports it.
|
||||||
|
func (b *Base) LookupUser(ctx context.Context, id string) (*User, error) {
|
||||||
|
ul, ok := b.db.(UserLookup)
|
||||||
|
if !ok {
|
||||||
|
return nil, ErrUserLookupNotAvailable
|
||||||
|
}
|
||||||
|
|
||||||
|
return ul.GetUserByID(ctx, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
// LookupChannel looks up a channel by ID if the database supports it.
|
||||||
|
func (b *Base) LookupChannel(ctx context.Context, id string) (*Channel, error) {
|
||||||
|
cl, ok := b.db.(ChannelLookup)
|
||||||
|
if !ok {
|
||||||
|
return nil, ErrChannelLookupNotAvailable
|
||||||
|
}
|
||||||
|
|
||||||
|
return cl.GetChannelByID(ctx, id)
|
||||||
|
}
|
||||||
18
internal/models/server_link.go
Normal file
18
internal/models/server_link.go
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
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"`
|
||||||
|
}
|
||||||
22
internal/models/session.go
Normal file
22
internal/models/session.go
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
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) {
|
||||||
|
return s.LookupUser(ctx, s.UserID)
|
||||||
|
}
|
||||||
92
internal/models/user.go
Normal file
92
internal/models/user.go
Normal file
@@ -0,0 +1,92 @@
|
|||||||
|
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()
|
||||||
|
}
|
||||||
@@ -4,6 +4,6 @@ import "time"
|
|||||||
|
|
||||||
const (
|
const (
|
||||||
httpReadTimeout = 10 * time.Second
|
httpReadTimeout = 10 * time.Second
|
||||||
httpWriteTimeout = 60 * time.Second
|
httpWriteTimeout = 10 * time.Second
|
||||||
maxHeaderBytes = 1 << 20
|
maxHeaderBytes = 1 << 20
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -16,153 +16,91 @@ import (
|
|||||||
|
|
||||||
const routeTimeout = 60 * time.Second
|
const routeTimeout = 60 * time.Second
|
||||||
|
|
||||||
// SetupRoutes configures the HTTP routes and middleware.
|
// SetupRoutes configures the HTTP routes and middleware chain.
|
||||||
func (srv *Server) SetupRoutes() {
|
func (s *Server) SetupRoutes() {
|
||||||
srv.router = chi.NewRouter()
|
s.router = chi.NewRouter()
|
||||||
|
|
||||||
srv.router.Use(middleware.Recoverer)
|
s.setupMiddleware()
|
||||||
srv.router.Use(middleware.RequestID)
|
s.setupHealthAndMetrics()
|
||||||
srv.router.Use(srv.mw.Logging())
|
s.setupAPIRoutes()
|
||||||
|
s.setupSPA()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Server) setupMiddleware() {
|
||||||
|
s.router.Use(middleware.Recoverer)
|
||||||
|
s.router.Use(middleware.RequestID)
|
||||||
|
s.router.Use(s.mw.Logging())
|
||||||
|
|
||||||
if viper.GetString("METRICS_USERNAME") != "" {
|
if viper.GetString("METRICS_USERNAME") != "" {
|
||||||
srv.router.Use(srv.mw.Metrics())
|
s.router.Use(s.mw.Metrics())
|
||||||
}
|
}
|
||||||
|
|
||||||
srv.router.Use(srv.mw.CORS())
|
s.router.Use(s.mw.CORS())
|
||||||
srv.router.Use(middleware.Timeout(routeTimeout))
|
s.router.Use(middleware.Timeout(routeTimeout))
|
||||||
|
|
||||||
if srv.sentryEnabled {
|
if s.sentryEnabled {
|
||||||
sentryHandler := sentryhttp.New(
|
sentryHandler := sentryhttp.New(sentryhttp.Options{
|
||||||
sentryhttp.Options{ //nolint:exhaustruct // optional fields
|
|
||||||
Repanic: true,
|
Repanic: true,
|
||||||
},
|
})
|
||||||
)
|
s.router.Use(sentryHandler.Handle)
|
||||||
|
|
||||||
srv.router.Use(sentryHandler.Handle)
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Health check.
|
func (s *Server) setupHealthAndMetrics() {
|
||||||
srv.router.Get(
|
s.router.Get("/.well-known/healthcheck.json", s.h.HandleHealthCheck())
|
||||||
"/.well-known/healthcheck.json",
|
|
||||||
srv.handlers.HandleHealthCheck(),
|
|
||||||
)
|
|
||||||
|
|
||||||
// Protected metrics endpoint.
|
|
||||||
if viper.GetString("METRICS_USERNAME") != "" {
|
if viper.GetString("METRICS_USERNAME") != "" {
|
||||||
srv.router.Group(func(router chi.Router) {
|
s.router.Group(func(r chi.Router) {
|
||||||
router.Use(srv.mw.MetricsAuth())
|
r.Use(s.mw.MetricsAuth())
|
||||||
router.Get("/metrics",
|
r.Get("/metrics", http.HandlerFunc(promhttp.Handler().ServeHTTP))
|
||||||
http.HandlerFunc(
|
|
||||||
promhttp.Handler().ServeHTTP,
|
|
||||||
))
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// API v1.
|
|
||||||
srv.router.Route("/api/v1", srv.setupAPIv1)
|
|
||||||
|
|
||||||
// Serve embedded SPA.
|
|
||||||
srv.setupSPA()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) setupAPIv1(router chi.Router) {
|
func (s *Server) setupAPIRoutes() {
|
||||||
router.Get(
|
s.router.Route("/api/v1", func(r chi.Router) {
|
||||||
"/server",
|
r.Get("/server", s.h.HandleServerInfo())
|
||||||
srv.handlers.HandleServerInfo(),
|
r.Post("/session", s.h.HandleCreateSession())
|
||||||
)
|
|
||||||
router.Post(
|
r.Get("/state", s.h.HandleState())
|
||||||
"/session",
|
r.Get("/messages", s.h.HandleGetMessages())
|
||||||
srv.handlers.HandleCreateSession(),
|
r.Post("/messages", s.h.HandleSendCommand())
|
||||||
)
|
r.Get("/history", s.h.HandleGetHistory())
|
||||||
router.Post(
|
|
||||||
"/register",
|
r.Get("/channels", s.h.HandleListAllChannels())
|
||||||
srv.handlers.HandleRegister(),
|
r.Get("/channels/{channel}/members", s.h.HandleChannelMembers())
|
||||||
)
|
})
|
||||||
router.Post(
|
|
||||||
"/login",
|
|
||||||
srv.handlers.HandleLogin(),
|
|
||||||
)
|
|
||||||
router.Get(
|
|
||||||
"/state",
|
|
||||||
srv.handlers.HandleState(),
|
|
||||||
)
|
|
||||||
router.Post(
|
|
||||||
"/logout",
|
|
||||||
srv.handlers.HandleLogout(),
|
|
||||||
)
|
|
||||||
router.Get(
|
|
||||||
"/users/me",
|
|
||||||
srv.handlers.HandleUsersMe(),
|
|
||||||
)
|
|
||||||
router.Get(
|
|
||||||
"/messages",
|
|
||||||
srv.handlers.HandleGetMessages(),
|
|
||||||
)
|
|
||||||
router.Post(
|
|
||||||
"/messages",
|
|
||||||
srv.handlers.HandleSendCommand(),
|
|
||||||
)
|
|
||||||
router.Get(
|
|
||||||
"/history",
|
|
||||||
srv.handlers.HandleGetHistory(),
|
|
||||||
)
|
|
||||||
router.Get(
|
|
||||||
"/channels",
|
|
||||||
srv.handlers.HandleListAllChannels(),
|
|
||||||
)
|
|
||||||
router.Get(
|
|
||||||
"/channels/{channel}/members",
|
|
||||||
srv.handlers.HandleChannelMembers(),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) setupSPA() {
|
func (s *Server) setupSPA() {
|
||||||
distFS, err := fs.Sub(web.Dist, "dist")
|
distFS, err := fs.Sub(web.Dist, "dist")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
srv.log.Error(
|
s.log.Error("failed to get web dist filesystem", "error", err)
|
||||||
"failed to get web dist filesystem",
|
|
||||||
"error", err,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
fileServer := http.FileServer(http.FS(distFS))
|
fileServer := http.FileServer(http.FS(distFS))
|
||||||
|
|
||||||
srv.router.Get("/*", func(
|
s.router.Get("/*", func(w http.ResponseWriter, r *http.Request) {
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
readFS, ok := distFS.(fs.ReadFileFS)
|
readFS, ok := distFS.(fs.ReadFileFS)
|
||||||
if !ok {
|
if !ok {
|
||||||
fileServer.ServeHTTP(writer, request)
|
http.NotFound(w, r)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
fileData, readErr := readFS.ReadFile(
|
f, readErr := readFS.ReadFile(r.URL.Path[1:])
|
||||||
request.URL.Path[1:],
|
if readErr != nil || len(f) == 0 {
|
||||||
)
|
indexHTML, _ := readFS.ReadFile("index.html")
|
||||||
if readErr != nil || len(fileData) == 0 {
|
|
||||||
indexHTML, indexErr := readFS.ReadFile(
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||||||
"index.html",
|
w.WriteHeader(http.StatusOK)
|
||||||
)
|
_, _ = w.Write(indexHTML)
|
||||||
if indexErr != nil {
|
|
||||||
http.NotFound(writer, request)
|
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
writer.Header().Set(
|
fileServer.ServeHTTP(w, r)
|
||||||
"Content-Type",
|
|
||||||
"text/html; charset=utf-8",
|
|
||||||
)
|
|
||||||
writer.WriteHeader(http.StatusOK)
|
|
||||||
_, _ = writer.Write(indexHTML)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
fileServer.ServeHTTP(writer, request)
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -41,8 +41,7 @@ type Params struct {
|
|||||||
Handlers *handlers.Handlers
|
Handlers *handlers.Handlers
|
||||||
}
|
}
|
||||||
|
|
||||||
// Server is the main HTTP server.
|
// Server is the main HTTP server. It manages routing, middleware, and lifecycle.
|
||||||
// It manages routing, middleware, and lifecycle.
|
|
||||||
type Server struct {
|
type Server struct {
|
||||||
startupTime time.Time
|
startupTime time.Time
|
||||||
exitCode int
|
exitCode int
|
||||||
@@ -54,24 +53,21 @@ type Server struct {
|
|||||||
router *chi.Mux
|
router *chi.Mux
|
||||||
params Params
|
params Params
|
||||||
mw *middleware.Middleware
|
mw *middleware.Middleware
|
||||||
handlers *handlers.Handlers
|
h *handlers.Handlers
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Server and registers its lifecycle hooks.
|
// New creates a new Server and registers its lifecycle hooks.
|
||||||
func New(
|
func New(lc fx.Lifecycle, params Params) (*Server, error) {
|
||||||
lifecycle fx.Lifecycle, params Params,
|
s := new(Server)
|
||||||
) (*Server, error) {
|
s.params = params
|
||||||
srv := &Server{ //nolint:exhaustruct // fields set during lifecycle
|
s.mw = params.Middleware
|
||||||
params: params,
|
s.h = params.Handlers
|
||||||
mw: params.Middleware,
|
s.log = params.Logger.Get()
|
||||||
handlers: params.Handlers,
|
|
||||||
log: params.Logger.Get(),
|
|
||||||
}
|
|
||||||
|
|
||||||
lifecycle.Append(fx.Hook{
|
lc.Append(fx.Hook{
|
||||||
OnStart: func(_ context.Context) error {
|
OnStart: func(_ context.Context) error {
|
||||||
srv.startupTime = time.Now()
|
s.startupTime = time.Now()
|
||||||
go srv.Run() //nolint:contextcheck
|
go s.Run() //nolint:contextcheck
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
@@ -80,140 +76,120 @@ func New(
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
return srv, nil
|
return s, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Run starts the server configuration, Sentry, and begins serving.
|
// Run starts the server configuration, Sentry, and begins serving.
|
||||||
func (srv *Server) Run() {
|
func (s *Server) Run() {
|
||||||
srv.configure()
|
s.configure()
|
||||||
srv.enableSentry()
|
s.enableSentry()
|
||||||
srv.serve()
|
s.serve()
|
||||||
}
|
}
|
||||||
|
|
||||||
// ServeHTTP delegates to the chi router.
|
// ServeHTTP delegates to the chi router.
|
||||||
func (srv *Server) ServeHTTP(
|
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
writer http.ResponseWriter,
|
s.router.ServeHTTP(w, r)
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
srv.router.ServeHTTP(writer, request)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// MaintenanceMode reports whether the server is in maintenance mode.
|
// MaintenanceMode reports whether the server is in maintenance mode.
|
||||||
func (srv *Server) MaintenanceMode() bool {
|
func (s *Server) MaintenanceMode() bool {
|
||||||
return srv.params.Config.MaintenanceMode
|
return s.params.Config.MaintenanceMode
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) enableSentry() {
|
func (s *Server) enableSentry() {
|
||||||
srv.sentryEnabled = false
|
s.sentryEnabled = false
|
||||||
|
|
||||||
if srv.params.Config.SentryDSN == "" {
|
if s.params.Config.SentryDSN == "" {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err := sentry.Init(sentry.ClientOptions{ //nolint:exhaustruct // only essential fields
|
err := sentry.Init(sentry.ClientOptions{
|
||||||
Dsn: srv.params.Config.SentryDSN,
|
Dsn: s.params.Config.SentryDSN,
|
||||||
Release: fmt.Sprintf(
|
Release: fmt.Sprintf("%s-%s", s.params.Globals.Appname, s.params.Globals.Version),
|
||||||
"%s-%s",
|
|
||||||
srv.params.Globals.Appname,
|
|
||||||
srv.params.Globals.Version,
|
|
||||||
),
|
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
srv.log.Error("sentry init failure", "error", err)
|
s.log.Error("sentry init failure", "error", err)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
srv.log.Info("sentry error reporting activated")
|
s.log.Info("sentry error reporting activated")
|
||||||
srv.sentryEnabled = true
|
s.sentryEnabled = true
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) serve() int {
|
func (s *Server) serve() int {
|
||||||
srv.ctx, srv.cancelFunc = context.WithCancel(
|
s.ctx, s.cancelFunc = context.WithCancel(context.Background())
|
||||||
context.Background(),
|
|
||||||
)
|
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
sigCh := make(chan os.Signal, 1)
|
c := make(chan os.Signal, 1)
|
||||||
|
|
||||||
signal.Ignore(syscall.SIGPIPE)
|
signal.Ignore(syscall.SIGPIPE)
|
||||||
signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
||||||
|
sig := <-c
|
||||||
|
s.log.Info("signal received", "signal", sig)
|
||||||
|
|
||||||
sig := <-sigCh
|
if s.cancelFunc != nil {
|
||||||
|
s.cancelFunc()
|
||||||
srv.log.Info("signal received", "signal", sig)
|
|
||||||
|
|
||||||
if srv.cancelFunc != nil {
|
|
||||||
srv.cancelFunc()
|
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
go srv.serveUntilShutdown()
|
go s.serveUntilShutdown()
|
||||||
|
|
||||||
<-srv.ctx.Done()
|
<-s.ctx.Done()
|
||||||
|
|
||||||
srv.cleanShutdown()
|
s.cleanShutdown()
|
||||||
|
|
||||||
return srv.exitCode
|
return s.exitCode
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) cleanupForExit() {
|
func (s *Server) cleanupForExit() {
|
||||||
srv.log.Info("cleaning up")
|
s.log.Info("cleaning up")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) cleanShutdown() {
|
func (s *Server) cleanShutdown() {
|
||||||
srv.exitCode = 0
|
s.exitCode = 0
|
||||||
|
|
||||||
ctxShutdown, shutdownCancel := context.WithTimeout(
|
ctxShutdown, shutdownCancel := context.WithTimeout(context.Background(), shutdownTimeout)
|
||||||
context.Background(), shutdownTimeout,
|
|
||||||
)
|
|
||||||
|
|
||||||
err := srv.httpServer.Shutdown(ctxShutdown)
|
err := s.httpServer.Shutdown(ctxShutdown)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
srv.log.Error(
|
s.log.Error("server clean shutdown failed", "error", err)
|
||||||
"server clean shutdown failed", "error", err,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if shutdownCancel != nil {
|
if shutdownCancel != nil {
|
||||||
shutdownCancel()
|
shutdownCancel()
|
||||||
}
|
}
|
||||||
|
|
||||||
srv.cleanupForExit()
|
s.cleanupForExit()
|
||||||
|
|
||||||
if srv.sentryEnabled {
|
if s.sentryEnabled {
|
||||||
sentry.Flush(sentryFlushTime)
|
sentry.Flush(sentryFlushTime)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) configure() {
|
func (s *Server) configure() {
|
||||||
// Server configuration placeholder.
|
// server configuration placeholder
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *Server) serveUntilShutdown() {
|
func (s *Server) serveUntilShutdown() {
|
||||||
listenAddr := fmt.Sprintf(
|
listenAddr := fmt.Sprintf(":%d", s.params.Config.Port)
|
||||||
":%d", srv.params.Config.Port,
|
s.httpServer = &http.Server{
|
||||||
)
|
|
||||||
|
|
||||||
srv.httpServer = &http.Server{ //nolint:exhaustruct // optional fields
|
|
||||||
Addr: listenAddr,
|
Addr: listenAddr,
|
||||||
ReadTimeout: httpReadTimeout,
|
ReadTimeout: httpReadTimeout,
|
||||||
WriteTimeout: httpWriteTimeout,
|
WriteTimeout: httpWriteTimeout,
|
||||||
MaxHeaderBytes: maxHeaderBytes,
|
MaxHeaderBytes: maxHeaderBytes,
|
||||||
Handler: srv,
|
Handler: s,
|
||||||
}
|
}
|
||||||
|
|
||||||
srv.SetupRoutes()
|
s.SetupRoutes()
|
||||||
|
|
||||||
srv.log.Info(
|
s.log.Info("http begin listen", "listenaddr", listenAddr)
|
||||||
"http begin listen", "listenaddr", listenAddr,
|
|
||||||
)
|
|
||||||
|
|
||||||
err := srv.httpServer.ListenAndServe()
|
err := s.httpServer.ListenAndServe()
|
||||||
if err != nil && !errors.Is(err, http.ErrServerClosed) {
|
if err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||||
srv.log.Error("listen error", "error", err)
|
s.log.Error("listen error", "error", err)
|
||||||
|
|
||||||
if srv.cancelFunc != nil {
|
if s.cancelFunc != nil {
|
||||||
srv.cancelFunc()
|
s.cancelFunc()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
3
web/dist/app.js
vendored
3
web/dist/app.js
vendored
File diff suppressed because one or more lines are too long
43
web/dist/style.css
vendored
43
web/dist/style.css
vendored
@@ -14,9 +14,6 @@
|
|||||||
--tab-active: #e94560;
|
--tab-active: #e94560;
|
||||||
--tab-bg: #16213e;
|
--tab-bg: #16213e;
|
||||||
--tab-hover: #1a1a3e;
|
--tab-hover: #1a1a3e;
|
||||||
--topic-bg: #121a30;
|
|
||||||
--unread-bg: #e94560;
|
|
||||||
--warn: #f0ad4e;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
html, body, #root {
|
html, body, #root {
|
||||||
@@ -89,7 +86,6 @@ html, body, #root {
|
|||||||
border-bottom: 1px solid var(--border);
|
border-bottom: 1px solid var(--border);
|
||||||
overflow-x: auto;
|
overflow-x: auto;
|
||||||
flex-shrink: 0;
|
flex-shrink: 0;
|
||||||
align-items: center;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.tab {
|
.tab {
|
||||||
@@ -99,7 +95,6 @@ html, body, #root {
|
|||||||
white-space: nowrap;
|
white-space: nowrap;
|
||||||
color: var(--text-muted);
|
color: var(--text-muted);
|
||||||
user-select: none;
|
user-select: none;
|
||||||
position: relative;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.tab:hover {
|
.tab:hover {
|
||||||
@@ -121,43 +116,6 @@ html, body, #root {
|
|||||||
color: var(--accent);
|
color: var(--accent);
|
||||||
}
|
}
|
||||||
|
|
||||||
.tab .unread-badge {
|
|
||||||
display: inline-block;
|
|
||||||
background: var(--unread-bg);
|
|
||||||
color: white;
|
|
||||||
font-size: 10px;
|
|
||||||
font-weight: bold;
|
|
||||||
padding: 1px 5px;
|
|
||||||
border-radius: 8px;
|
|
||||||
margin-left: 6px;
|
|
||||||
min-width: 16px;
|
|
||||||
text-align: center;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Connection status */
|
|
||||||
.connection-status {
|
|
||||||
padding: 4px 12px;
|
|
||||||
background: var(--warn);
|
|
||||||
color: #1a1a2e;
|
|
||||||
font-size: 12px;
|
|
||||||
font-weight: bold;
|
|
||||||
white-space: nowrap;
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Topic bar */
|
|
||||||
.topic-bar {
|
|
||||||
padding: 6px 12px;
|
|
||||||
background: var(--topic-bg);
|
|
||||||
border-bottom: 1px solid var(--border);
|
|
||||||
color: var(--text-muted);
|
|
||||||
font-size: 12px;
|
|
||||||
white-space: nowrap;
|
|
||||||
overflow: hidden;
|
|
||||||
text-overflow: ellipsis;
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Content area */
|
/* Content area */
|
||||||
.content {
|
.content {
|
||||||
display: flex;
|
display: flex;
|
||||||
@@ -285,7 +243,6 @@ html, body, #root {
|
|||||||
gap: 8px;
|
gap: 8px;
|
||||||
background: var(--bg-secondary);
|
background: var(--bg-secondary);
|
||||||
border-bottom: 1px solid var(--border);
|
border-bottom: 1px solid var(--border);
|
||||||
margin-left: auto;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.join-dialog input {
|
.join-dialog input {
|
||||||
|
|||||||
350
web/src/app.jsx
350
web/src/app.jsx
@@ -1,17 +1,13 @@
|
|||||||
import { h, render } from 'preact';
|
import { h, render, Component } from 'preact';
|
||||||
import { useState, useEffect, useRef, useCallback } from 'preact/hooks';
|
import { useState, useEffect, useRef, useCallback } from 'preact/hooks';
|
||||||
|
|
||||||
const API = '/api/v1';
|
const API = '/api/v1';
|
||||||
const POLL_TIMEOUT = 15;
|
|
||||||
const RECONNECT_DELAY = 3000;
|
|
||||||
const MEMBER_REFRESH_INTERVAL = 10000;
|
|
||||||
|
|
||||||
function api(path, opts = {}) {
|
function api(path, opts = {}) {
|
||||||
const token = localStorage.getItem('chat_token');
|
const token = localStorage.getItem('chat_token');
|
||||||
const headers = { 'Content-Type': 'application/json', ...(opts.headers || {}) };
|
const headers = { 'Content-Type': 'application/json', ...(opts.headers || {}) };
|
||||||
if (token) headers['Authorization'] = `Bearer ${token}`;
|
if (token) headers['Authorization'] = `Bearer ${token}`;
|
||||||
const { signal, ...rest } = opts;
|
return fetch(API + path, { ...opts, headers }).then(async r => {
|
||||||
return fetch(API + path, { ...rest, headers, signal }).then(async r => {
|
|
||||||
const data = await r.json().catch(() => null);
|
const data = await r.json().catch(() => null);
|
||||||
if (!r.ok) throw { status: r.status, data };
|
if (!r.ok) throw { status: r.status, data };
|
||||||
return data;
|
return data;
|
||||||
@@ -23,6 +19,7 @@ function formatTime(ts) {
|
|||||||
return d.toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', second: '2-digit' });
|
return d.toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', second: '2-digit' });
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Nick color hashing
|
||||||
function nickColor(nick) {
|
function nickColor(nick) {
|
||||||
let h = 0;
|
let h = 0;
|
||||||
for (let i = 0; i < nick.length; i++) h = nick.charCodeAt(i) + ((h << 5) - h);
|
for (let i = 0; i < nick.length; i++) h = nick.charCodeAt(i) + ((h << 5) - h);
|
||||||
@@ -42,9 +39,10 @@ function LoginScreen({ onLogin }) {
|
|||||||
if (s.name) setServerName(s.name);
|
if (s.name) setServerName(s.name);
|
||||||
if (s.motd) setMotd(s.motd);
|
if (s.motd) setMotd(s.motd);
|
||||||
}).catch(() => {});
|
}).catch(() => {});
|
||||||
|
// Check for saved token
|
||||||
const saved = localStorage.getItem('chat_token');
|
const saved = localStorage.getItem('chat_token');
|
||||||
if (saved) {
|
if (saved) {
|
||||||
api('/state').then(u => onLogin(u.nick)).catch(() => localStorage.removeItem('chat_token'));
|
api('/state').then(u => onLogin(u.nick, saved)).catch(() => localStorage.removeItem('chat_token'));
|
||||||
}
|
}
|
||||||
inputRef.current?.focus();
|
inputRef.current?.focus();
|
||||||
}, []);
|
}, []);
|
||||||
@@ -58,7 +56,7 @@ function LoginScreen({ onLogin }) {
|
|||||||
body: JSON.stringify({ nick: nick.trim() })
|
body: JSON.stringify({ nick: nick.trim() })
|
||||||
});
|
});
|
||||||
localStorage.setItem('chat_token', res.token);
|
localStorage.setItem('chat_token', res.token);
|
||||||
onLogin(res.nick);
|
onLogin(res.nick, res.token);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
setError(err.data?.error || 'Connection failed');
|
setError(err.data?.error || 'Connection failed');
|
||||||
}
|
}
|
||||||
@@ -86,19 +84,11 @@ function LoginScreen({ onLogin }) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
function Message({ msg }) {
|
function Message({ msg }) {
|
||||||
if (msg.system) {
|
|
||||||
return (
|
return (
|
||||||
<div class="message system">
|
<div class={`message ${msg.system ? 'system' : ''}`}>
|
||||||
<span class="timestamp">{formatTime(msg.ts)}</span>
|
<span class="timestamp">{formatTime(msg.createdAt)}</span>
|
||||||
<span class="content">{msg.text}</span>
|
<span class="nick" style={{ color: msg.system ? undefined : nickColor(msg.nick) }}>{msg.nick}</span>
|
||||||
</div>
|
<span class="content">{msg.content}</span>
|
||||||
);
|
|
||||||
}
|
|
||||||
return (
|
|
||||||
<div class="message">
|
|
||||||
<span class="timestamp">{formatTime(msg.ts)}</span>
|
|
||||||
<span class="nick" style={{ color: nickColor(msg.from) }}>{msg.from}</span>
|
|
||||||
<span class="content">{msg.text}</span>
|
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -108,194 +98,93 @@ function App() {
|
|||||||
const [nick, setNick] = useState('');
|
const [nick, setNick] = useState('');
|
||||||
const [tabs, setTabs] = useState([{ type: 'server', name: 'Server' }]);
|
const [tabs, setTabs] = useState([{ type: 'server', name: 'Server' }]);
|
||||||
const [activeTab, setActiveTab] = useState(0);
|
const [activeTab, setActiveTab] = useState(0);
|
||||||
const [messages, setMessages] = useState({ Server: [] });
|
const [messages, setMessages] = useState({ server: [] }); // keyed by tab name
|
||||||
const [members, setMembers] = useState({});
|
const [members, setMembers] = useState({}); // keyed by channel name
|
||||||
const [topics, setTopics] = useState({});
|
|
||||||
const [unread, setUnread] = useState({});
|
|
||||||
const [input, setInput] = useState('');
|
const [input, setInput] = useState('');
|
||||||
const [joinInput, setJoinInput] = useState('');
|
const [joinInput, setJoinInput] = useState('');
|
||||||
const [connected, setConnected] = useState(true);
|
const [lastMsgId, setLastMsgId] = useState(0);
|
||||||
|
|
||||||
const lastIdRef = useRef(0);
|
|
||||||
const seenIdsRef = useRef(new Set());
|
|
||||||
const pollAbortRef = useRef(null);
|
|
||||||
const tabsRef = useRef(tabs);
|
|
||||||
const activeTabRef = useRef(activeTab);
|
|
||||||
const nickRef = useRef(nick);
|
|
||||||
const messagesEndRef = useRef();
|
const messagesEndRef = useRef();
|
||||||
const inputRef = useRef();
|
const inputRef = useRef();
|
||||||
|
const pollRef = useRef();
|
||||||
useEffect(() => { tabsRef.current = tabs; }, [tabs]);
|
|
||||||
useEffect(() => { activeTabRef.current = activeTab; }, [activeTab]);
|
|
||||||
useEffect(() => { nickRef.current = nick; }, [nick]);
|
|
||||||
|
|
||||||
// Persist joined channels
|
|
||||||
useEffect(() => {
|
|
||||||
const channels = tabs.filter(t => t.type === 'channel').map(t => t.name);
|
|
||||||
localStorage.setItem('chat_channels', JSON.stringify(channels));
|
|
||||||
}, [tabs]);
|
|
||||||
|
|
||||||
// Clear unread on tab switch
|
|
||||||
useEffect(() => {
|
|
||||||
const tab = tabs[activeTab];
|
|
||||||
if (tab) setUnread(prev => ({ ...prev, [tab.name]: 0 }));
|
|
||||||
}, [activeTab, tabs]);
|
|
||||||
|
|
||||||
const addMessage = useCallback((tabName, msg) => {
|
const addMessage = useCallback((tabName, msg) => {
|
||||||
if (msg.id && seenIdsRef.current.has(msg.id)) return;
|
|
||||||
if (msg.id) seenIdsRef.current.add(msg.id);
|
|
||||||
setMessages(prev => ({
|
setMessages(prev => ({
|
||||||
...prev,
|
...prev,
|
||||||
[tabName]: [...(prev[tabName] || []), msg]
|
[tabName]: [...(prev[tabName] || []), msg]
|
||||||
}));
|
}));
|
||||||
const currentTab = tabsRef.current[activeTabRef.current];
|
|
||||||
if (!currentTab || currentTab.name !== tabName) {
|
|
||||||
setUnread(prev => ({ ...prev, [tabName]: (prev[tabName] || 0) + 1 }));
|
|
||||||
}
|
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
const addSystemMessage = useCallback((tabName, text) => {
|
const addSystemMessage = useCallback((tabName, text) => {
|
||||||
setMessages(prev => ({
|
addMessage(tabName, {
|
||||||
...prev,
|
id: Date.now(),
|
||||||
[tabName]: [...(prev[tabName] || []), {
|
nick: '*',
|
||||||
id: 'sys-' + Date.now() + '-' + Math.random(),
|
content: text,
|
||||||
ts: new Date().toISOString(),
|
createdAt: new Date().toISOString(),
|
||||||
text,
|
|
||||||
system: true
|
system: true
|
||||||
}]
|
});
|
||||||
}));
|
}, [addMessage]);
|
||||||
}, []);
|
|
||||||
|
|
||||||
const refreshMembers = useCallback((channel) => {
|
const onLogin = useCallback((userNick, token) => {
|
||||||
const chName = channel.replace('#', '');
|
setNick(userNick);
|
||||||
api(`/channels/${chName}/members`).then(m => {
|
setLoggedIn(true);
|
||||||
setMembers(prev => ({ ...prev, [channel]: m }));
|
addSystemMessage('server', `Connected as ${userNick}`);
|
||||||
|
// Fetch server info
|
||||||
|
api('/server').then(s => {
|
||||||
|
if (s.motd) addSystemMessage('server', `MOTD: ${s.motd}`);
|
||||||
}).catch(() => {});
|
}).catch(() => {});
|
||||||
}, []);
|
}, [addSystemMessage]);
|
||||||
|
|
||||||
const processMessage = useCallback((msg) => {
|
// Poll for new messages
|
||||||
const body = Array.isArray(msg.body) ? msg.body.join('\n') : '';
|
|
||||||
const base = { id: msg.id, ts: msg.ts, from: msg.from, to: msg.to, command: msg.command };
|
|
||||||
|
|
||||||
switch (msg.command) {
|
|
||||||
case 'PRIVMSG':
|
|
||||||
case 'NOTICE': {
|
|
||||||
const parsed = { ...base, text: body, system: false };
|
|
||||||
const target = msg.to;
|
|
||||||
if (target && target.startsWith('#')) {
|
|
||||||
addMessage(target, parsed);
|
|
||||||
} else {
|
|
||||||
const dmPeer = msg.from === nickRef.current ? msg.to : msg.from;
|
|
||||||
setTabs(prev => {
|
|
||||||
if (!prev.find(t => t.type === 'dm' && t.name === dmPeer)) {
|
|
||||||
return [...prev, { type: 'dm', name: dmPeer }];
|
|
||||||
}
|
|
||||||
return prev;
|
|
||||||
});
|
|
||||||
addMessage(dmPeer, parsed);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case 'JOIN': {
|
|
||||||
const text = `${msg.from} has joined ${msg.to}`;
|
|
||||||
if (msg.to) addMessage(msg.to, { ...base, text, system: true });
|
|
||||||
if (msg.to && msg.to.startsWith('#')) refreshMembers(msg.to);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case 'PART': {
|
|
||||||
const reason = body ? ': ' + body : '';
|
|
||||||
const text = `${msg.from} has left ${msg.to}${reason}`;
|
|
||||||
if (msg.to) addMessage(msg.to, { ...base, text, system: true });
|
|
||||||
if (msg.to && msg.to.startsWith('#')) refreshMembers(msg.to);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case 'QUIT': {
|
|
||||||
const reason = body ? ': ' + body : '';
|
|
||||||
const text = `${msg.from} has quit${reason}`;
|
|
||||||
tabsRef.current.forEach(tab => {
|
|
||||||
if (tab.type === 'channel') {
|
|
||||||
addMessage(tab.name, { ...base, text, system: true });
|
|
||||||
}
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case 'NICK': {
|
|
||||||
const newNick = Array.isArray(msg.body) ? msg.body[0] : body;
|
|
||||||
const text = `${msg.from} is now known as ${newNick}`;
|
|
||||||
tabsRef.current.forEach(tab => {
|
|
||||||
if (tab.type === 'channel') {
|
|
||||||
addMessage(tab.name, { ...base, text, system: true });
|
|
||||||
}
|
|
||||||
});
|
|
||||||
if (msg.from === nickRef.current && newNick) setNick(newNick);
|
|
||||||
// Refresh members in all channels
|
|
||||||
tabsRef.current.forEach(tab => {
|
|
||||||
if (tab.type === 'channel') refreshMembers(tab.name);
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case 'TOPIC': {
|
|
||||||
const text = `${msg.from} set the topic: ${body}`;
|
|
||||||
if (msg.to) {
|
|
||||||
addMessage(msg.to, { ...base, text, system: true });
|
|
||||||
setTopics(prev => ({ ...prev, [msg.to]: body }));
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case '375':
|
|
||||||
case '372':
|
|
||||||
case '376':
|
|
||||||
addMessage('Server', { ...base, text: body, system: true });
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
addMessage('Server', { ...base, text: body || msg.command, system: true });
|
|
||||||
}
|
|
||||||
}, [addMessage, refreshMembers]);
|
|
||||||
|
|
||||||
// Long-poll loop
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (!loggedIn) return;
|
if (!loggedIn) return;
|
||||||
let alive = true;
|
let alive = true;
|
||||||
|
|
||||||
const poll = async () => {
|
const poll = async () => {
|
||||||
while (alive) {
|
|
||||||
try {
|
try {
|
||||||
const controller = new AbortController();
|
const msgs = await api(`/messages?after=${lastMsgId}`);
|
||||||
pollAbortRef.current = controller;
|
if (!alive) return;
|
||||||
const result = await api(
|
let maxId = lastMsgId;
|
||||||
`/messages?after=${lastIdRef.current}&timeout=${POLL_TIMEOUT}`,
|
for (const msg of msgs) {
|
||||||
{ signal: controller.signal }
|
if (msg.id > maxId) maxId = msg.id;
|
||||||
);
|
if (msg.isDm) {
|
||||||
if (!alive) break;
|
const dmTab = msg.nick === nick ? msg.dmTarget : msg.nick;
|
||||||
setConnected(true);
|
// Ensure DM tab exists
|
||||||
if (result.messages) {
|
setTabs(prev => {
|
||||||
for (const m of result.messages) processMessage(m);
|
if (!prev.find(t => t.type === 'dm' && t.name === dmTab)) {
|
||||||
|
return [...prev, { type: 'dm', name: dmTab }];
|
||||||
}
|
}
|
||||||
if (result.last_id > lastIdRef.current) {
|
return prev;
|
||||||
lastIdRef.current = result.last_id;
|
});
|
||||||
|
addMessage(dmTab, msg);
|
||||||
|
} else if (msg.channel) {
|
||||||
|
addMessage(msg.channel, msg);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
if (maxId > lastMsgId) setLastMsgId(maxId);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
if (!alive) break;
|
// silent
|
||||||
if (err.name === 'AbortError') continue;
|
|
||||||
setConnected(false);
|
|
||||||
await new Promise(r => setTimeout(r, RECONNECT_DELAY));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
pollRef.current = setInterval(poll, 1500);
|
||||||
poll();
|
poll();
|
||||||
return () => { alive = false; pollAbortRef.current?.abort(); };
|
return () => { alive = false; clearInterval(pollRef.current); };
|
||||||
}, [loggedIn, processMessage]);
|
}, [loggedIn, lastMsgId, nick, addMessage]);
|
||||||
|
|
||||||
// Refresh members for active channel
|
// Fetch members for active channel tab
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (!loggedIn) return;
|
if (!loggedIn) return;
|
||||||
const tab = tabs[activeTab];
|
const tab = tabs[activeTab];
|
||||||
if (!tab || tab.type !== 'channel') return;
|
if (!tab || tab.type !== 'channel') return;
|
||||||
refreshMembers(tab.name);
|
const chName = tab.name.replace('#', '');
|
||||||
const iv = setInterval(() => refreshMembers(tab.name), MEMBER_REFRESH_INTERVAL);
|
api(`/channels/${chName}/members`).then(m => {
|
||||||
|
setMembers(prev => ({ ...prev, [tab.name]: m }));
|
||||||
|
}).catch(() => {});
|
||||||
|
const iv = setInterval(() => {
|
||||||
|
api(`/channels/${chName}/members`).then(m => {
|
||||||
|
setMembers(prev => ({ ...prev, [tab.name]: m }));
|
||||||
|
}).catch(() => {});
|
||||||
|
}, 5000);
|
||||||
return () => clearInterval(iv);
|
return () => clearInterval(iv);
|
||||||
}, [loggedIn, activeTab, tabs, refreshMembers]);
|
}, [loggedIn, activeTab, tabs]);
|
||||||
|
|
||||||
// Auto-scroll
|
// Auto-scroll
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -303,37 +192,9 @@ function App() {
|
|||||||
}, [messages, activeTab]);
|
}, [messages, activeTab]);
|
||||||
|
|
||||||
// Focus input on tab change
|
// Focus input on tab change
|
||||||
useEffect(() => { inputRef.current?.focus(); }, [activeTab]);
|
|
||||||
|
|
||||||
// Fetch topic for active channel
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (!loggedIn) return;
|
inputRef.current?.focus();
|
||||||
const tab = tabs[activeTab];
|
}, [activeTab]);
|
||||||
if (!tab || tab.type !== 'channel') return;
|
|
||||||
api('/channels').then(channels => {
|
|
||||||
const ch = channels.find(c => c.name === tab.name);
|
|
||||||
if (ch && ch.topic) setTopics(prev => ({ ...prev, [tab.name]: ch.topic }));
|
|
||||||
}).catch(() => {});
|
|
||||||
}, [loggedIn, activeTab, tabs]);
|
|
||||||
|
|
||||||
const onLogin = useCallback(async (userNick) => {
|
|
||||||
setNick(userNick);
|
|
||||||
setLoggedIn(true);
|
|
||||||
addSystemMessage('Server', `Connected as ${userNick}`);
|
|
||||||
// Auto-rejoin saved channels
|
|
||||||
const saved = JSON.parse(localStorage.getItem('chat_channels') || '[]');
|
|
||||||
for (const ch of saved) {
|
|
||||||
try {
|
|
||||||
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'JOIN', to: ch }) });
|
|
||||||
setTabs(prev => {
|
|
||||||
if (prev.find(t => t.type === 'channel' && t.name === ch)) return prev;
|
|
||||||
return [...prev, { type: 'channel', name: ch }];
|
|
||||||
});
|
|
||||||
} catch (e) {
|
|
||||||
// Channel may not exist anymore
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}, [addSystemMessage]);
|
|
||||||
|
|
||||||
const joinChannel = async (name) => {
|
const joinChannel = async (name) => {
|
||||||
if (!name) return;
|
if (!name) return;
|
||||||
@@ -345,29 +206,22 @@ function App() {
|
|||||||
if (prev.find(t => t.type === 'channel' && t.name === name)) return prev;
|
if (prev.find(t => t.type === 'channel' && t.name === name)) return prev;
|
||||||
return [...prev, { type: 'channel', name }];
|
return [...prev, { type: 'channel', name }];
|
||||||
});
|
});
|
||||||
setActiveTab(tabs.length);
|
setActiveTab(tabs.length); // switch to new tab
|
||||||
// Load history
|
addSystemMessage(name, `Joined ${name}`);
|
||||||
try {
|
|
||||||
const hist = await api(`/history?target=${encodeURIComponent(name)}&limit=50`);
|
|
||||||
if (Array.isArray(hist)) {
|
|
||||||
for (const m of hist) processMessage(m);
|
|
||||||
}
|
|
||||||
} catch (e) {
|
|
||||||
// History may be empty
|
|
||||||
}
|
|
||||||
setJoinInput('');
|
setJoinInput('');
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
addSystemMessage('Server', `Failed to join ${name}: ${err.data?.error || 'error'}`);
|
addSystemMessage('server', `Failed to join ${name}: ${err.data?.error || 'error'}`);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
const partChannel = async (name) => {
|
const partChannel = async (name) => {
|
||||||
try {
|
try {
|
||||||
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'PART', to: name }) });
|
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'PART', to: name }) });
|
||||||
} catch (e) {
|
} catch (err) { /* ignore */ }
|
||||||
// Ignore
|
setTabs(prev => {
|
||||||
}
|
const next = prev.filter(t => !(t.type === 'channel' && t.name === name));
|
||||||
setTabs(prev => prev.filter(t => !(t.type === 'channel' && t.name === name)));
|
return next;
|
||||||
|
});
|
||||||
setActiveTab(0);
|
setActiveTab(0);
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -386,8 +240,7 @@ function App() {
|
|||||||
if (prev.find(t => t.type === 'dm' && t.name === targetNick)) return prev;
|
if (prev.find(t => t.type === 'dm' && t.name === targetNick)) return prev;
|
||||||
return [...prev, { type: 'dm', name: targetNick }];
|
return [...prev, { type: 'dm', name: targetNick }];
|
||||||
});
|
});
|
||||||
const idx = tabs.findIndex(t => t.type === 'dm' && t.name === targetNick);
|
setActiveTab(tabs.findIndex(t => t.type === 'dm' && t.name === targetNick) || tabs.length);
|
||||||
setActiveTab(idx >= 0 ? idx : tabs.length);
|
|
||||||
};
|
};
|
||||||
|
|
||||||
const sendMessage = async () => {
|
const sendMessage = async () => {
|
||||||
@@ -397,45 +250,46 @@ function App() {
|
|||||||
const tab = tabs[activeTab];
|
const tab = tabs[activeTab];
|
||||||
if (!tab || tab.type === 'server') return;
|
if (!tab || tab.type === 'server') return;
|
||||||
|
|
||||||
|
// Handle /commands
|
||||||
if (text.startsWith('/')) {
|
if (text.startsWith('/')) {
|
||||||
const parts = text.split(' ');
|
const parts = text.split(' ');
|
||||||
const cmd = parts[0].toLowerCase();
|
const cmd = parts[0].toLowerCase();
|
||||||
if (cmd === '/join' && parts[1]) { joinChannel(parts[1]); return; }
|
if (cmd === '/join' && parts[1]) {
|
||||||
if (cmd === '/part') { if (tab.type === 'channel') partChannel(tab.name); return; }
|
joinChannel(parts[1]);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (cmd === '/part') {
|
||||||
|
if (tab.type === 'channel') partChannel(tab.name);
|
||||||
|
return;
|
||||||
|
}
|
||||||
if (cmd === '/msg' && parts[1] && parts.slice(2).join(' ')) {
|
if (cmd === '/msg' && parts[1] && parts.slice(2).join(' ')) {
|
||||||
const target = parts[1];
|
const target = parts[1];
|
||||||
const body = parts.slice(2).join(' ');
|
const msg = parts.slice(2).join(' ');
|
||||||
try {
|
try {
|
||||||
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'PRIVMSG', to: target, body: [body] }) });
|
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'PRIVMSG', to: target, body: [msg] }) });
|
||||||
openDM(target);
|
openDM(target);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
addSystemMessage('Server', `DM failed: ${err.data?.error || 'error'}`);
|
addSystemMessage('server', `Failed to send DM: ${err.data?.error || 'error'}`);
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (cmd === '/nick' && parts[1]) {
|
if (cmd === '/nick' && parts[1]) {
|
||||||
try {
|
try {
|
||||||
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'NICK', body: [parts[1]] }) });
|
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'NICK', body: [parts[1]] }) });
|
||||||
|
setNick(parts[1]);
|
||||||
|
addSystemMessage('server', `Nick changed to ${parts[1]}`);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
addSystemMessage('Server', `Nick change failed: ${err.data?.error || 'error'}`);
|
addSystemMessage('server', `Nick change failed: ${err.data?.error || 'error'}`);
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (cmd === '/topic' && tab.type === 'channel') {
|
addSystemMessage('server', `Unknown command: ${cmd}`);
|
||||||
const topicText = parts.slice(1).join(' ');
|
|
||||||
try {
|
|
||||||
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'TOPIC', to: tab.name, body: [topicText] }) });
|
|
||||||
} catch (err) {
|
|
||||||
addSystemMessage('Server', `Topic failed: ${err.data?.error || 'error'}`);
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
addSystemMessage('Server', `Unknown command: ${cmd}`);
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const to = tab.type === 'channel' ? tab.name : tab.name;
|
||||||
try {
|
try {
|
||||||
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'PRIVMSG', to: tab.name, body: [text] }) });
|
await api('/messages', { method: 'POST', body: JSON.stringify({ command: 'PRIVMSG', to, body: [text] }) });
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
addSystemMessage(tab.name, `Send failed: ${err.data?.error || 'error'}`);
|
addSystemMessage(tab.name, `Send failed: ${err.data?.error || 'error'}`);
|
||||||
}
|
}
|
||||||
@@ -446,21 +300,16 @@ function App() {
|
|||||||
const currentTab = tabs[activeTab] || tabs[0];
|
const currentTab = tabs[activeTab] || tabs[0];
|
||||||
const currentMessages = messages[currentTab.name] || [];
|
const currentMessages = messages[currentTab.name] || [];
|
||||||
const currentMembers = members[currentTab.name] || [];
|
const currentMembers = members[currentTab.name] || [];
|
||||||
const currentTopic = topics[currentTab.name] || '';
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div class="app">
|
<div class="app">
|
||||||
<div class="tab-bar">
|
<div class="tab-bar">
|
||||||
{!connected && <div class="connection-status">⚠ Reconnecting...</div>}
|
|
||||||
{tabs.map((tab, i) => (
|
{tabs.map((tab, i) => (
|
||||||
<div
|
<div
|
||||||
class={`tab ${i === activeTab ? 'active' : ''}`}
|
class={`tab ${i === activeTab ? 'active' : ''}`}
|
||||||
onClick={() => setActiveTab(i)}
|
onClick={() => setActiveTab(i)}
|
||||||
>
|
>
|
||||||
{tab.type === 'dm' ? `→${tab.name}` : tab.name}
|
{tab.type === 'dm' ? `→${tab.name}` : tab.name}
|
||||||
{unread[tab.name] > 0 && i !== activeTab && (
|
|
||||||
<span class="unread-badge">{unread[tab.name]}</span>
|
|
||||||
)}
|
|
||||||
{tab.type !== 'server' && (
|
{tab.type !== 'server' && (
|
||||||
<span class="close-btn" onClick={(e) => { e.stopPropagation(); closeTab(i); }}>×</span>
|
<span class="close-btn" onClick={(e) => { e.stopPropagation(); closeTab(i); }}>×</span>
|
||||||
)}
|
)}
|
||||||
@@ -477,17 +326,19 @@ function App() {
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{currentTab.type === 'channel' && currentTopic && (
|
|
||||||
<div class="topic-bar" title={currentTopic}>{currentTopic}</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
<div class="content">
|
<div class="content">
|
||||||
<div class="messages-pane">
|
<div class="messages-pane">
|
||||||
<div class={currentTab.type === 'server' ? 'server-messages' : 'messages'}>
|
{currentTab.type === 'server' ? (
|
||||||
|
<div class="server-messages">
|
||||||
|
{currentMessages.map(m => <Message msg={m} />)}
|
||||||
|
<div ref={messagesEndRef} />
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
|
<div class="messages">
|
||||||
{currentMessages.map(m => <Message msg={m} />)}
|
{currentMessages.map(m => <Message msg={m} />)}
|
||||||
<div ref={messagesEndRef} />
|
<div ref={messagesEndRef} />
|
||||||
</div>
|
</div>
|
||||||
{currentTab.type !== 'server' && (
|
|
||||||
<div class="input-bar">
|
<div class="input-bar">
|
||||||
<input
|
<input
|
||||||
ref={inputRef}
|
ref={inputRef}
|
||||||
@@ -498,6 +349,7 @@ function App() {
|
|||||||
/>
|
/>
|
||||||
<button onClick={sendMessage}>Send</button>
|
<button onClick={sendMessage}>Send</button>
|
||||||
</div>
|
</div>
|
||||||
|
</>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
|||||||
@@ -14,9 +14,6 @@
|
|||||||
--tab-active: #e94560;
|
--tab-active: #e94560;
|
||||||
--tab-bg: #16213e;
|
--tab-bg: #16213e;
|
||||||
--tab-hover: #1a1a3e;
|
--tab-hover: #1a1a3e;
|
||||||
--topic-bg: #121a30;
|
|
||||||
--unread-bg: #e94560;
|
|
||||||
--warn: #f0ad4e;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
html, body, #root {
|
html, body, #root {
|
||||||
@@ -89,7 +86,6 @@ html, body, #root {
|
|||||||
border-bottom: 1px solid var(--border);
|
border-bottom: 1px solid var(--border);
|
||||||
overflow-x: auto;
|
overflow-x: auto;
|
||||||
flex-shrink: 0;
|
flex-shrink: 0;
|
||||||
align-items: center;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.tab {
|
.tab {
|
||||||
@@ -99,7 +95,6 @@ html, body, #root {
|
|||||||
white-space: nowrap;
|
white-space: nowrap;
|
||||||
color: var(--text-muted);
|
color: var(--text-muted);
|
||||||
user-select: none;
|
user-select: none;
|
||||||
position: relative;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.tab:hover {
|
.tab:hover {
|
||||||
@@ -121,43 +116,6 @@ html, body, #root {
|
|||||||
color: var(--accent);
|
color: var(--accent);
|
||||||
}
|
}
|
||||||
|
|
||||||
.tab .unread-badge {
|
|
||||||
display: inline-block;
|
|
||||||
background: var(--unread-bg);
|
|
||||||
color: white;
|
|
||||||
font-size: 10px;
|
|
||||||
font-weight: bold;
|
|
||||||
padding: 1px 5px;
|
|
||||||
border-radius: 8px;
|
|
||||||
margin-left: 6px;
|
|
||||||
min-width: 16px;
|
|
||||||
text-align: center;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Connection status */
|
|
||||||
.connection-status {
|
|
||||||
padding: 4px 12px;
|
|
||||||
background: var(--warn);
|
|
||||||
color: #1a1a2e;
|
|
||||||
font-size: 12px;
|
|
||||||
font-weight: bold;
|
|
||||||
white-space: nowrap;
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Topic bar */
|
|
||||||
.topic-bar {
|
|
||||||
padding: 6px 12px;
|
|
||||||
background: var(--topic-bg);
|
|
||||||
border-bottom: 1px solid var(--border);
|
|
||||||
color: var(--text-muted);
|
|
||||||
font-size: 12px;
|
|
||||||
white-space: nowrap;
|
|
||||||
overflow: hidden;
|
|
||||||
text-overflow: ellipsis;
|
|
||||||
flex-shrink: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Content area */
|
/* Content area */
|
||||||
.content {
|
.content {
|
||||||
display: flex;
|
display: flex;
|
||||||
@@ -285,7 +243,6 @@ html, body, #root {
|
|||||||
gap: 8px;
|
gap: 8px;
|
||||||
background: var(--bg-secondary);
|
background: var(--bg-secondary);
|
||||||
border-bottom: 1px solid var(--border);
|
border-bottom: 1px solid var(--border);
|
||||||
margin-left: auto;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.join-dialog input {
|
.join-dialog input {
|
||||||
|
|||||||
Reference in New Issue
Block a user