Compare commits
1 Commits
feat/proje
...
feat/ms21-
| Author | SHA1 | Date | |
|---|---|---|---|
| e4a56ab850 |
@@ -34,9 +34,3 @@ CVE-2026-26996 # HIGH: minimatch DoS via specially crafted glob patterns (needs
|
|||||||
# OpenBao 2.5.0 compiled with Go 1.25.6, fix needs Go >= 1.25.7.
|
# OpenBao 2.5.0 compiled with Go 1.25.6, fix needs Go >= 1.25.7.
|
||||||
# Cannot build OpenBao from source (large project). Waiting for upstream release.
|
# Cannot build OpenBao from source (large project). Waiting for upstream release.
|
||||||
CVE-2025-68121 # CRITICAL: crypto/tls session resumption
|
CVE-2025-68121 # CRITICAL: crypto/tls session resumption
|
||||||
|
|
||||||
# === multer CVEs (upstream via @nestjs/platform-express) ===
|
|
||||||
# multer <2.1.0 — waiting on NestJS to update their dependency
|
|
||||||
# These are DoS vulnerabilities in file upload handling
|
|
||||||
GHSA-xf7r-hgr6-v32p # HIGH: DoS via incomplete cleanup
|
|
||||||
GHSA-v52c-386h-88mc # HIGH: DoS via resource exhaustion
|
|
||||||
|
|||||||
232
.woodpecker/api.yml
Normal file
232
.woodpecker/api.yml
Normal file
@@ -0,0 +1,232 @@
|
|||||||
|
# API Pipeline - Mosaic Stack
|
||||||
|
# Quality gates, build, and Docker publish for @mosaic/api
|
||||||
|
#
|
||||||
|
# Triggers on: apps/api/**, packages/**, root configs
|
||||||
|
# Security chain: source audit + Trivy container scan
|
||||||
|
|
||||||
|
when:
|
||||||
|
- event: [push, pull_request, manual]
|
||||||
|
path:
|
||||||
|
include:
|
||||||
|
- "apps/api/**"
|
||||||
|
- "packages/**"
|
||||||
|
- "pnpm-lock.yaml"
|
||||||
|
- "pnpm-workspace.yaml"
|
||||||
|
- "turbo.json"
|
||||||
|
- "package.json"
|
||||||
|
- ".woodpecker/api.yml"
|
||||||
|
- ".trivyignore"
|
||||||
|
|
||||||
|
variables:
|
||||||
|
- &node_image "node:24-alpine"
|
||||||
|
- &install_deps |
|
||||||
|
corepack enable
|
||||||
|
pnpm install --frozen-lockfile
|
||||||
|
- &use_deps |
|
||||||
|
corepack enable
|
||||||
|
- &turbo_env
|
||||||
|
TURBO_API:
|
||||||
|
from_secret: turbo_api
|
||||||
|
TURBO_TOKEN:
|
||||||
|
from_secret: turbo_token
|
||||||
|
TURBO_TEAM:
|
||||||
|
from_secret: turbo_team
|
||||||
|
- &kaniko_setup |
|
||||||
|
mkdir -p /kaniko/.docker
|
||||||
|
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$GITEA_USER\",\"password\":\"$GITEA_TOKEN\"}}}" > /kaniko/.docker/config.json
|
||||||
|
|
||||||
|
services:
|
||||||
|
postgres:
|
||||||
|
image: postgres:17.7-alpine3.22
|
||||||
|
environment:
|
||||||
|
POSTGRES_DB: test_db
|
||||||
|
POSTGRES_USER: test_user
|
||||||
|
POSTGRES_PASSWORD: test_password
|
||||||
|
|
||||||
|
steps:
|
||||||
|
# === Quality Gates ===
|
||||||
|
|
||||||
|
install:
|
||||||
|
image: *node_image
|
||||||
|
commands:
|
||||||
|
- *install_deps
|
||||||
|
|
||||||
|
security-audit:
|
||||||
|
image: *node_image
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm audit --audit-level=high
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
prisma-generate:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm --filter "@mosaic/api" prisma:generate
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
lint:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo lint --filter=@mosaic/api
|
||||||
|
depends_on:
|
||||||
|
- prisma-generate
|
||||||
|
|
||||||
|
typecheck:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo typecheck --filter=@mosaic/api
|
||||||
|
depends_on:
|
||||||
|
- prisma-generate
|
||||||
|
|
||||||
|
prisma-migrate:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
DATABASE_URL: "postgresql://test_user:test_password@postgres:5432/test_db?schema=public"
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm --filter "@mosaic/api" prisma migrate deploy
|
||||||
|
depends_on:
|
||||||
|
- prisma-generate
|
||||||
|
|
||||||
|
test:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
DATABASE_URL: "postgresql://test_user:test_password@postgres:5432/test_db?schema=public"
|
||||||
|
ENCRYPTION_KEY: "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm --filter "@mosaic/api" exec vitest run --exclude 'src/auth/auth-rls.integration.spec.ts' --exclude 'src/credentials/user-credential.model.spec.ts' --exclude 'src/job-events/job-events.performance.spec.ts' --exclude 'src/knowledge/services/fulltext-search.spec.ts' --exclude 'src/mosaic-telemetry/mosaic-telemetry.module.spec.ts'
|
||||||
|
depends_on:
|
||||||
|
- prisma-migrate
|
||||||
|
|
||||||
|
# === Build ===
|
||||||
|
|
||||||
|
build:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
NODE_ENV: "production"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo build --filter=@mosaic/api
|
||||||
|
depends_on:
|
||||||
|
- lint
|
||||||
|
- typecheck
|
||||||
|
- test
|
||||||
|
- security-audit
|
||||||
|
|
||||||
|
# === Docker Build & Push ===
|
||||||
|
|
||||||
|
docker-build-api:
|
||||||
|
image: gcr.io/kaniko-project/executor:debug
|
||||||
|
environment:
|
||||||
|
GITEA_USER:
|
||||||
|
from_secret: gitea_username
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
||||||
|
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
||||||
|
commands:
|
||||||
|
- *kaniko_setup
|
||||||
|
- |
|
||||||
|
DESTINATIONS=""
|
||||||
|
if [ -n "$CI_COMMIT_TAG" ]; then
|
||||||
|
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-api:$CI_COMMIT_TAG"
|
||||||
|
elif [ "$CI_COMMIT_BRANCH" = "main" ]; then
|
||||||
|
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-api:latest"
|
||||||
|
fi
|
||||||
|
/kaniko/executor --context . --dockerfile apps/api/Dockerfile --snapshot-mode=redo $DESTINATIONS
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- build
|
||||||
|
|
||||||
|
# === Container Security Scan ===
|
||||||
|
|
||||||
|
security-trivy-api:
|
||||||
|
image: aquasec/trivy:latest
|
||||||
|
environment:
|
||||||
|
GITEA_USER:
|
||||||
|
from_secret: gitea_username
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
||||||
|
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
||||||
|
commands:
|
||||||
|
- |
|
||||||
|
if [ -n "$$CI_COMMIT_TAG" ]; then
|
||||||
|
SCAN_TAG="$$CI_COMMIT_TAG"
|
||||||
|
elif [ "$$CI_COMMIT_BRANCH" = "main" ]; then
|
||||||
|
SCAN_TAG="latest"
|
||||||
|
else
|
||||||
|
SCAN_TAG="latest"
|
||||||
|
fi
|
||||||
|
mkdir -p ~/.docker
|
||||||
|
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$$GITEA_USER\",\"password\":\"$$GITEA_TOKEN\"}}}" > ~/.docker/config.json
|
||||||
|
trivy image --exit-code 1 --severity HIGH,CRITICAL --ignore-unfixed \
|
||||||
|
--ignorefile .trivyignore \
|
||||||
|
git.mosaicstack.dev/mosaic/stack-api:$$SCAN_TAG
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- docker-build-api
|
||||||
|
|
||||||
|
# === Package Linking ===
|
||||||
|
|
||||||
|
link-packages:
|
||||||
|
image: alpine:3
|
||||||
|
environment:
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
commands:
|
||||||
|
- apk add --no-cache curl
|
||||||
|
- sleep 10
|
||||||
|
- |
|
||||||
|
set -e
|
||||||
|
link_package() {
|
||||||
|
PKG="$$1"
|
||||||
|
echo "Linking $$PKG..."
|
||||||
|
for attempt in 1 2 3; do
|
||||||
|
STATUS=$$(curl -s -o /tmp/link-response.txt -w "%{http_code}" -X POST \
|
||||||
|
-H "Authorization: token $$GITEA_TOKEN" \
|
||||||
|
"https://git.mosaicstack.dev/api/v1/packages/mosaic/container/$$PKG/-/link/stack")
|
||||||
|
if [ "$$STATUS" = "201" ] || [ "$$STATUS" = "204" ]; then
|
||||||
|
echo " Linked $$PKG"
|
||||||
|
return 0
|
||||||
|
elif [ "$$STATUS" = "400" ]; then
|
||||||
|
echo " $$PKG already linked"
|
||||||
|
return 0
|
||||||
|
elif [ "$$STATUS" = "404" ] && [ $$attempt -lt 3 ]; then
|
||||||
|
echo " $$PKG not found yet, retrying in 5s (attempt $$attempt/3)..."
|
||||||
|
sleep 5
|
||||||
|
else
|
||||||
|
echo " FAILED: $$PKG status $$STATUS"
|
||||||
|
cat /tmp/link-response.txt
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
link_package "stack-api"
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- security-trivy-api
|
||||||
@@ -1,337 +0,0 @@
|
|||||||
# Unified CI Pipeline - Mosaic Stack
|
|
||||||
# Single install, parallel quality gates, sequential deploy
|
|
||||||
#
|
|
||||||
# Replaces: api.yml, orchestrator.yml, web.yml
|
|
||||||
# Keeps: coordinator.yml (Python), infra.yml (separate concerns)
|
|
||||||
#
|
|
||||||
# Flow:
|
|
||||||
# install → security-audit
|
|
||||||
# → prisma-generate → lint + typecheck (parallel)
|
|
||||||
# → prisma-migrate → test
|
|
||||||
# → build (after all gates pass)
|
|
||||||
# → docker builds (main only, parallel)
|
|
||||||
# → trivy scans (main only, parallel)
|
|
||||||
# → package linking (main only)
|
|
||||||
|
|
||||||
when:
|
|
||||||
- event: [push, pull_request, manual]
|
|
||||||
path:
|
|
||||||
include:
|
|
||||||
- "apps/api/**"
|
|
||||||
- "apps/orchestrator/**"
|
|
||||||
- "apps/web/**"
|
|
||||||
- "packages/**"
|
|
||||||
- "pnpm-lock.yaml"
|
|
||||||
- "pnpm-workspace.yaml"
|
|
||||||
- "turbo.json"
|
|
||||||
- "package.json"
|
|
||||||
- ".woodpecker/ci.yml"
|
|
||||||
- ".trivyignore"
|
|
||||||
|
|
||||||
variables:
|
|
||||||
- &node_image "node:24-alpine"
|
|
||||||
- &install_deps |
|
|
||||||
corepack enable
|
|
||||||
pnpm install --frozen-lockfile
|
|
||||||
- &use_deps |
|
|
||||||
corepack enable
|
|
||||||
- &turbo_env
|
|
||||||
TURBO_API:
|
|
||||||
from_secret: turbo_api
|
|
||||||
TURBO_TOKEN:
|
|
||||||
from_secret: turbo_token
|
|
||||||
TURBO_TEAM:
|
|
||||||
from_secret: turbo_team
|
|
||||||
- &kaniko_setup |
|
|
||||||
mkdir -p /kaniko/.docker
|
|
||||||
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$GITEA_USER\",\"password\":\"$GITEA_TOKEN\"}}}" > /kaniko/.docker/config.json
|
|
||||||
|
|
||||||
services:
|
|
||||||
postgres:
|
|
||||||
image: postgres:17.7-alpine3.22
|
|
||||||
environment:
|
|
||||||
POSTGRES_DB: test_db
|
|
||||||
POSTGRES_USER: test_user
|
|
||||||
POSTGRES_PASSWORD: test_password
|
|
||||||
|
|
||||||
steps:
|
|
||||||
# ─── Install (once) ─────────────────────────────────────────
|
|
||||||
install:
|
|
||||||
image: *node_image
|
|
||||||
commands:
|
|
||||||
- *install_deps
|
|
||||||
|
|
||||||
# ─── Security Audit (once) ──────────────────────────────────
|
|
||||||
security-audit:
|
|
||||||
image: *node_image
|
|
||||||
commands:
|
|
||||||
- *use_deps
|
|
||||||
- pnpm audit --audit-level=high
|
|
||||||
depends_on:
|
|
||||||
- install
|
|
||||||
|
|
||||||
# ─── Prisma Generate ────────────────────────────────────────
|
|
||||||
prisma-generate:
|
|
||||||
image: *node_image
|
|
||||||
environment:
|
|
||||||
SKIP_ENV_VALIDATION: "true"
|
|
||||||
commands:
|
|
||||||
- *use_deps
|
|
||||||
- pnpm --filter "@mosaic/api" prisma:generate
|
|
||||||
depends_on:
|
|
||||||
- install
|
|
||||||
|
|
||||||
# ─── Lint (all packages) ────────────────────────────────────
|
|
||||||
lint:
|
|
||||||
image: *node_image
|
|
||||||
environment:
|
|
||||||
SKIP_ENV_VALIDATION: "true"
|
|
||||||
<<: *turbo_env
|
|
||||||
commands:
|
|
||||||
- *use_deps
|
|
||||||
- pnpm turbo lint
|
|
||||||
depends_on:
|
|
||||||
- prisma-generate
|
|
||||||
|
|
||||||
# ─── Typecheck (all packages, parallel with lint) ───────────
|
|
||||||
typecheck:
|
|
||||||
image: *node_image
|
|
||||||
environment:
|
|
||||||
SKIP_ENV_VALIDATION: "true"
|
|
||||||
<<: *turbo_env
|
|
||||||
commands:
|
|
||||||
- *use_deps
|
|
||||||
- pnpm turbo typecheck
|
|
||||||
depends_on:
|
|
||||||
- prisma-generate
|
|
||||||
|
|
||||||
# ─── Prisma Migrate (test DB) ──────────────────────────────
|
|
||||||
prisma-migrate:
|
|
||||||
image: *node_image
|
|
||||||
environment:
|
|
||||||
SKIP_ENV_VALIDATION: "true"
|
|
||||||
DATABASE_URL: "postgresql://test_user:test_password@postgres:5432/test_db?schema=public"
|
|
||||||
commands:
|
|
||||||
- *use_deps
|
|
||||||
- pnpm --filter "@mosaic/api" prisma migrate deploy
|
|
||||||
depends_on:
|
|
||||||
- prisma-generate
|
|
||||||
|
|
||||||
# ─── Test (all packages) ───────────────────────────────────
|
|
||||||
test:
|
|
||||||
image: *node_image
|
|
||||||
environment:
|
|
||||||
SKIP_ENV_VALIDATION: "true"
|
|
||||||
DATABASE_URL: "postgresql://test_user:test_password@postgres:5432/test_db?schema=public"
|
|
||||||
ENCRYPTION_KEY: "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
|
|
||||||
<<: *turbo_env
|
|
||||||
commands:
|
|
||||||
- *use_deps
|
|
||||||
- pnpm --filter "@mosaic/api" exec vitest run --exclude 'src/auth/auth-rls.integration.spec.ts' --exclude 'src/credentials/user-credential.model.spec.ts' --exclude 'src/job-events/job-events.performance.spec.ts' --exclude 'src/knowledge/services/fulltext-search.spec.ts' --exclude 'src/mosaic-telemetry/mosaic-telemetry.module.spec.ts'
|
|
||||||
- pnpm turbo test --filter=@mosaic/orchestrator --filter=@mosaic/web
|
|
||||||
depends_on:
|
|
||||||
- prisma-migrate
|
|
||||||
|
|
||||||
# ─── Build (all packages) ──────────────────────────────────
|
|
||||||
build:
|
|
||||||
image: *node_image
|
|
||||||
environment:
|
|
||||||
SKIP_ENV_VALIDATION: "true"
|
|
||||||
NODE_ENV: "production"
|
|
||||||
<<: *turbo_env
|
|
||||||
commands:
|
|
||||||
- *use_deps
|
|
||||||
- pnpm turbo build
|
|
||||||
depends_on:
|
|
||||||
- lint
|
|
||||||
- typecheck
|
|
||||||
- test
|
|
||||||
- security-audit
|
|
||||||
|
|
||||||
# ─── Docker Builds (main only, parallel) ───────────────────
|
|
||||||
|
|
||||||
docker-build-api:
|
|
||||||
image: gcr.io/kaniko-project/executor:debug
|
|
||||||
environment:
|
|
||||||
GITEA_USER:
|
|
||||||
from_secret: gitea_username
|
|
||||||
GITEA_TOKEN:
|
|
||||||
from_secret: gitea_token
|
|
||||||
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
|
||||||
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
|
||||||
commands:
|
|
||||||
- *kaniko_setup
|
|
||||||
- |
|
|
||||||
DESTINATIONS=""
|
|
||||||
if [ -n "$CI_COMMIT_TAG" ]; then
|
|
||||||
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-api:$CI_COMMIT_TAG"
|
|
||||||
elif [ "$CI_COMMIT_BRANCH" = "main" ]; then
|
|
||||||
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-api:latest"
|
|
||||||
fi
|
|
||||||
/kaniko/executor --context . --dockerfile apps/api/Dockerfile --snapshot-mode=redo $DESTINATIONS
|
|
||||||
when:
|
|
||||||
- branch: [main]
|
|
||||||
event: [push, manual, tag]
|
|
||||||
depends_on:
|
|
||||||
- build
|
|
||||||
|
|
||||||
docker-build-orchestrator:
|
|
||||||
image: gcr.io/kaniko-project/executor:debug
|
|
||||||
environment:
|
|
||||||
GITEA_USER:
|
|
||||||
from_secret: gitea_username
|
|
||||||
GITEA_TOKEN:
|
|
||||||
from_secret: gitea_token
|
|
||||||
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
|
||||||
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
|
||||||
commands:
|
|
||||||
- *kaniko_setup
|
|
||||||
- |
|
|
||||||
DESTINATIONS=""
|
|
||||||
if [ -n "$CI_COMMIT_TAG" ]; then
|
|
||||||
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-orchestrator:$CI_COMMIT_TAG"
|
|
||||||
elif [ "$CI_COMMIT_BRANCH" = "main" ]; then
|
|
||||||
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-orchestrator:latest"
|
|
||||||
fi
|
|
||||||
/kaniko/executor --context . --dockerfile apps/orchestrator/Dockerfile --snapshot-mode=redo $DESTINATIONS
|
|
||||||
when:
|
|
||||||
- branch: [main]
|
|
||||||
event: [push, manual, tag]
|
|
||||||
depends_on:
|
|
||||||
- build
|
|
||||||
|
|
||||||
docker-build-web:
|
|
||||||
image: gcr.io/kaniko-project/executor:debug
|
|
||||||
environment:
|
|
||||||
GITEA_USER:
|
|
||||||
from_secret: gitea_username
|
|
||||||
GITEA_TOKEN:
|
|
||||||
from_secret: gitea_token
|
|
||||||
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
|
||||||
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
|
||||||
commands:
|
|
||||||
- *kaniko_setup
|
|
||||||
- |
|
|
||||||
DESTINATIONS=""
|
|
||||||
if [ -n "$CI_COMMIT_TAG" ]; then
|
|
||||||
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-web:$CI_COMMIT_TAG"
|
|
||||||
elif [ "$CI_COMMIT_BRANCH" = "main" ]; then
|
|
||||||
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-web:latest"
|
|
||||||
fi
|
|
||||||
/kaniko/executor --context . --dockerfile apps/web/Dockerfile --snapshot-mode=redo --build-arg NEXT_PUBLIC_API_URL=https://api.mosaicstack.dev $DESTINATIONS
|
|
||||||
when:
|
|
||||||
- branch: [main]
|
|
||||||
event: [push, manual, tag]
|
|
||||||
depends_on:
|
|
||||||
- build
|
|
||||||
|
|
||||||
# ─── Container Security Scans (main only) ──────────────────
|
|
||||||
|
|
||||||
security-trivy-api:
|
|
||||||
image: aquasec/trivy:latest
|
|
||||||
environment:
|
|
||||||
GITEA_USER:
|
|
||||||
from_secret: gitea_username
|
|
||||||
GITEA_TOKEN:
|
|
||||||
from_secret: gitea_token
|
|
||||||
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
|
||||||
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
|
||||||
commands:
|
|
||||||
- |
|
|
||||||
if [ -n "$$CI_COMMIT_TAG" ]; then SCAN_TAG="$$CI_COMMIT_TAG"; else SCAN_TAG="latest"; fi
|
|
||||||
mkdir -p ~/.docker
|
|
||||||
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$$GITEA_USER\",\"password\":\"$$GITEA_TOKEN\"}}}" > ~/.docker/config.json
|
|
||||||
trivy image --exit-code 1 --severity HIGH,CRITICAL --ignore-unfixed --ignorefile .trivyignore git.mosaicstack.dev/mosaic/stack-api:$$SCAN_TAG
|
|
||||||
when:
|
|
||||||
- branch: [main]
|
|
||||||
event: [push, manual, tag]
|
|
||||||
depends_on:
|
|
||||||
- docker-build-api
|
|
||||||
|
|
||||||
security-trivy-orchestrator:
|
|
||||||
image: aquasec/trivy:latest
|
|
||||||
environment:
|
|
||||||
GITEA_USER:
|
|
||||||
from_secret: gitea_username
|
|
||||||
GITEA_TOKEN:
|
|
||||||
from_secret: gitea_token
|
|
||||||
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
|
||||||
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
|
||||||
commands:
|
|
||||||
- |
|
|
||||||
if [ -n "$$CI_COMMIT_TAG" ]; then SCAN_TAG="$$CI_COMMIT_TAG"; else SCAN_TAG="latest"; fi
|
|
||||||
mkdir -p ~/.docker
|
|
||||||
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$$GITEA_USER\",\"password\":\"$$GITEA_TOKEN\"}}}" > ~/.docker/config.json
|
|
||||||
trivy image --exit-code 1 --severity HIGH,CRITICAL --ignore-unfixed --ignorefile .trivyignore git.mosaicstack.dev/mosaic/stack-orchestrator:$$SCAN_TAG
|
|
||||||
when:
|
|
||||||
- branch: [main]
|
|
||||||
event: [push, manual, tag]
|
|
||||||
depends_on:
|
|
||||||
- docker-build-orchestrator
|
|
||||||
|
|
||||||
security-trivy-web:
|
|
||||||
image: aquasec/trivy:latest
|
|
||||||
environment:
|
|
||||||
GITEA_USER:
|
|
||||||
from_secret: gitea_username
|
|
||||||
GITEA_TOKEN:
|
|
||||||
from_secret: gitea_token
|
|
||||||
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
|
||||||
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
|
||||||
commands:
|
|
||||||
- |
|
|
||||||
if [ -n "$$CI_COMMIT_TAG" ]; then SCAN_TAG="$$CI_COMMIT_TAG"; else SCAN_TAG="latest"; fi
|
|
||||||
mkdir -p ~/.docker
|
|
||||||
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$$GITEA_USER\",\"password\":\"$$GITEA_TOKEN\"}}}" > ~/.docker/config.json
|
|
||||||
trivy image --exit-code 1 --severity HIGH,CRITICAL --ignore-unfixed --ignorefile .trivyignore git.mosaicstack.dev/mosaic/stack-web:$$SCAN_TAG
|
|
||||||
when:
|
|
||||||
- branch: [main]
|
|
||||||
event: [push, manual, tag]
|
|
||||||
depends_on:
|
|
||||||
- docker-build-web
|
|
||||||
|
|
||||||
# ─── Package Linking (main only, once) ─────────────────────
|
|
||||||
|
|
||||||
link-packages:
|
|
||||||
image: alpine:3
|
|
||||||
environment:
|
|
||||||
GITEA_TOKEN:
|
|
||||||
from_secret: gitea_token
|
|
||||||
commands:
|
|
||||||
- apk add --no-cache curl
|
|
||||||
- sleep 10
|
|
||||||
- |
|
|
||||||
set -e
|
|
||||||
link_package() {
|
|
||||||
PKG="$$1"
|
|
||||||
echo "Linking $$PKG..."
|
|
||||||
for attempt in 1 2 3; do
|
|
||||||
STATUS=$$(curl -s -o /tmp/link-response.txt -w "%{http_code}" -X POST \
|
|
||||||
-H "Authorization: token $$GITEA_TOKEN" \
|
|
||||||
"https://git.mosaicstack.dev/api/v1/packages/mosaic/container/$$PKG/-/link/stack")
|
|
||||||
if [ "$$STATUS" = "201" ] || [ "$$STATUS" = "204" ]; then
|
|
||||||
echo " Linked $$PKG"
|
|
||||||
return 0
|
|
||||||
elif [ "$$STATUS" = "400" ]; then
|
|
||||||
echo " $$PKG already linked"
|
|
||||||
return 0
|
|
||||||
elif [ "$$STATUS" = "404" ] && [ $$attempt -lt 3 ]; then
|
|
||||||
echo " $$PKG not found yet, retrying in 5s (attempt $$attempt/3)..."
|
|
||||||
sleep 5
|
|
||||||
else
|
|
||||||
echo " FAILED: $$PKG status $$STATUS"
|
|
||||||
cat /tmp/link-response.txt
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
}
|
|
||||||
link_package "stack-api"
|
|
||||||
link_package "stack-orchestrator"
|
|
||||||
link_package "stack-web"
|
|
||||||
when:
|
|
||||||
- branch: [main]
|
|
||||||
event: [push, manual, tag]
|
|
||||||
depends_on:
|
|
||||||
- security-trivy-api
|
|
||||||
- security-trivy-orchestrator
|
|
||||||
- security-trivy-web
|
|
||||||
202
.woodpecker/orchestrator.yml
Normal file
202
.woodpecker/orchestrator.yml
Normal file
@@ -0,0 +1,202 @@
|
|||||||
|
# Orchestrator Pipeline - Mosaic Stack
|
||||||
|
# Quality gates, build, and Docker publish for @mosaic/orchestrator
|
||||||
|
#
|
||||||
|
# Triggers on: apps/orchestrator/**, packages/**, root configs
|
||||||
|
# Security chain: source audit + Trivy container scan
|
||||||
|
|
||||||
|
when:
|
||||||
|
- event: [push, pull_request, manual]
|
||||||
|
path:
|
||||||
|
include:
|
||||||
|
- "apps/orchestrator/**"
|
||||||
|
- "packages/**"
|
||||||
|
- "pnpm-lock.yaml"
|
||||||
|
- "pnpm-workspace.yaml"
|
||||||
|
- "turbo.json"
|
||||||
|
- "package.json"
|
||||||
|
- ".woodpecker/orchestrator.yml"
|
||||||
|
- ".trivyignore"
|
||||||
|
|
||||||
|
variables:
|
||||||
|
- &node_image "node:24-alpine"
|
||||||
|
- &install_deps |
|
||||||
|
corepack enable
|
||||||
|
pnpm install --frozen-lockfile
|
||||||
|
- &use_deps |
|
||||||
|
corepack enable
|
||||||
|
- &turbo_env
|
||||||
|
TURBO_API:
|
||||||
|
from_secret: turbo_api
|
||||||
|
TURBO_TOKEN:
|
||||||
|
from_secret: turbo_token
|
||||||
|
TURBO_TEAM:
|
||||||
|
from_secret: turbo_team
|
||||||
|
- &kaniko_setup |
|
||||||
|
mkdir -p /kaniko/.docker
|
||||||
|
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$GITEA_USER\",\"password\":\"$GITEA_TOKEN\"}}}" > /kaniko/.docker/config.json
|
||||||
|
|
||||||
|
steps:
|
||||||
|
# === Quality Gates ===
|
||||||
|
|
||||||
|
install:
|
||||||
|
image: *node_image
|
||||||
|
commands:
|
||||||
|
- *install_deps
|
||||||
|
|
||||||
|
security-audit:
|
||||||
|
image: *node_image
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm audit --audit-level=high
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
lint:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo lint --filter=@mosaic/orchestrator
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
typecheck:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo typecheck --filter=@mosaic/orchestrator
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
test:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo test --filter=@mosaic/orchestrator
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
# === Build ===
|
||||||
|
|
||||||
|
build:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
NODE_ENV: "production"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo build --filter=@mosaic/orchestrator
|
||||||
|
depends_on:
|
||||||
|
- lint
|
||||||
|
- typecheck
|
||||||
|
- test
|
||||||
|
- security-audit
|
||||||
|
|
||||||
|
# === Docker Build & Push ===
|
||||||
|
|
||||||
|
docker-build-orchestrator:
|
||||||
|
image: gcr.io/kaniko-project/executor:debug
|
||||||
|
environment:
|
||||||
|
GITEA_USER:
|
||||||
|
from_secret: gitea_username
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
||||||
|
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
||||||
|
commands:
|
||||||
|
- *kaniko_setup
|
||||||
|
- |
|
||||||
|
DESTINATIONS=""
|
||||||
|
if [ -n "$CI_COMMIT_TAG" ]; then
|
||||||
|
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-orchestrator:$CI_COMMIT_TAG"
|
||||||
|
elif [ "$CI_COMMIT_BRANCH" = "main" ]; then
|
||||||
|
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-orchestrator:latest"
|
||||||
|
fi
|
||||||
|
/kaniko/executor --context . --dockerfile apps/orchestrator/Dockerfile --snapshot-mode=redo $DESTINATIONS
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- build
|
||||||
|
|
||||||
|
# === Container Security Scan ===
|
||||||
|
|
||||||
|
security-trivy-orchestrator:
|
||||||
|
image: aquasec/trivy:latest
|
||||||
|
environment:
|
||||||
|
GITEA_USER:
|
||||||
|
from_secret: gitea_username
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
||||||
|
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
||||||
|
commands:
|
||||||
|
- |
|
||||||
|
if [ -n "$$CI_COMMIT_TAG" ]; then
|
||||||
|
SCAN_TAG="$$CI_COMMIT_TAG"
|
||||||
|
elif [ "$$CI_COMMIT_BRANCH" = "main" ]; then
|
||||||
|
SCAN_TAG="latest"
|
||||||
|
else
|
||||||
|
SCAN_TAG="latest"
|
||||||
|
fi
|
||||||
|
mkdir -p ~/.docker
|
||||||
|
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$$GITEA_USER\",\"password\":\"$$GITEA_TOKEN\"}}}" > ~/.docker/config.json
|
||||||
|
trivy image --exit-code 1 --severity HIGH,CRITICAL --ignore-unfixed \
|
||||||
|
--ignorefile .trivyignore \
|
||||||
|
git.mosaicstack.dev/mosaic/stack-orchestrator:$$SCAN_TAG
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- docker-build-orchestrator
|
||||||
|
|
||||||
|
# === Package Linking ===
|
||||||
|
|
||||||
|
link-packages:
|
||||||
|
image: alpine:3
|
||||||
|
environment:
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
commands:
|
||||||
|
- apk add --no-cache curl
|
||||||
|
- sleep 10
|
||||||
|
- |
|
||||||
|
set -e
|
||||||
|
link_package() {
|
||||||
|
PKG="$$1"
|
||||||
|
echo "Linking $$PKG..."
|
||||||
|
for attempt in 1 2 3; do
|
||||||
|
STATUS=$$(curl -s -o /tmp/link-response.txt -w "%{http_code}" -X POST \
|
||||||
|
-H "Authorization: token $$GITEA_TOKEN" \
|
||||||
|
"https://git.mosaicstack.dev/api/v1/packages/mosaic/container/$$PKG/-/link/stack")
|
||||||
|
if [ "$$STATUS" = "201" ] || [ "$$STATUS" = "204" ]; then
|
||||||
|
echo " Linked $$PKG"
|
||||||
|
return 0
|
||||||
|
elif [ "$$STATUS" = "400" ]; then
|
||||||
|
echo " $$PKG already linked"
|
||||||
|
return 0
|
||||||
|
elif [ "$$STATUS" = "404" ] && [ $$attempt -lt 3 ]; then
|
||||||
|
echo " $$PKG not found yet, retrying in 5s (attempt $$attempt/3)..."
|
||||||
|
sleep 5
|
||||||
|
else
|
||||||
|
echo " FAILED: $$PKG status $$STATUS"
|
||||||
|
cat /tmp/link-response.txt
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
link_package "stack-orchestrator"
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- security-trivy-orchestrator
|
||||||
202
.woodpecker/web.yml
Normal file
202
.woodpecker/web.yml
Normal file
@@ -0,0 +1,202 @@
|
|||||||
|
# Web Pipeline - Mosaic Stack
|
||||||
|
# Quality gates, build, and Docker publish for @mosaic/web
|
||||||
|
#
|
||||||
|
# Triggers on: apps/web/**, packages/**, root configs
|
||||||
|
# Security chain: source audit + Trivy container scan
|
||||||
|
|
||||||
|
when:
|
||||||
|
- event: [push, pull_request, manual]
|
||||||
|
path:
|
||||||
|
include:
|
||||||
|
- "apps/web/**"
|
||||||
|
- "packages/**"
|
||||||
|
- "pnpm-lock.yaml"
|
||||||
|
- "pnpm-workspace.yaml"
|
||||||
|
- "turbo.json"
|
||||||
|
- "package.json"
|
||||||
|
- ".woodpecker/web.yml"
|
||||||
|
- ".trivyignore"
|
||||||
|
|
||||||
|
variables:
|
||||||
|
- &node_image "node:24-alpine"
|
||||||
|
- &install_deps |
|
||||||
|
corepack enable
|
||||||
|
pnpm install --frozen-lockfile
|
||||||
|
- &use_deps |
|
||||||
|
corepack enable
|
||||||
|
- &turbo_env
|
||||||
|
TURBO_API:
|
||||||
|
from_secret: turbo_api
|
||||||
|
TURBO_TOKEN:
|
||||||
|
from_secret: turbo_token
|
||||||
|
TURBO_TEAM:
|
||||||
|
from_secret: turbo_team
|
||||||
|
- &kaniko_setup |
|
||||||
|
mkdir -p /kaniko/.docker
|
||||||
|
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$GITEA_USER\",\"password\":\"$GITEA_TOKEN\"}}}" > /kaniko/.docker/config.json
|
||||||
|
|
||||||
|
steps:
|
||||||
|
# === Quality Gates ===
|
||||||
|
|
||||||
|
install:
|
||||||
|
image: *node_image
|
||||||
|
commands:
|
||||||
|
- *install_deps
|
||||||
|
|
||||||
|
security-audit:
|
||||||
|
image: *node_image
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm audit --audit-level=high
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
lint:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo lint --filter=@mosaic/web
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
typecheck:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo typecheck --filter=@mosaic/web
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
test:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo test --filter=@mosaic/web
|
||||||
|
depends_on:
|
||||||
|
- install
|
||||||
|
|
||||||
|
# === Build ===
|
||||||
|
|
||||||
|
build:
|
||||||
|
image: *node_image
|
||||||
|
environment:
|
||||||
|
SKIP_ENV_VALIDATION: "true"
|
||||||
|
NODE_ENV: "production"
|
||||||
|
<<: *turbo_env
|
||||||
|
commands:
|
||||||
|
- *use_deps
|
||||||
|
- pnpm turbo build --filter=@mosaic/web
|
||||||
|
depends_on:
|
||||||
|
- lint
|
||||||
|
- typecheck
|
||||||
|
- test
|
||||||
|
- security-audit
|
||||||
|
|
||||||
|
# === Docker Build & Push ===
|
||||||
|
|
||||||
|
docker-build-web:
|
||||||
|
image: gcr.io/kaniko-project/executor:debug
|
||||||
|
environment:
|
||||||
|
GITEA_USER:
|
||||||
|
from_secret: gitea_username
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
||||||
|
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
||||||
|
commands:
|
||||||
|
- *kaniko_setup
|
||||||
|
- |
|
||||||
|
DESTINATIONS=""
|
||||||
|
if [ -n "$CI_COMMIT_TAG" ]; then
|
||||||
|
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-web:$CI_COMMIT_TAG"
|
||||||
|
elif [ "$CI_COMMIT_BRANCH" = "main" ]; then
|
||||||
|
DESTINATIONS="--destination git.mosaicstack.dev/mosaic/stack-web:latest"
|
||||||
|
fi
|
||||||
|
/kaniko/executor --context . --dockerfile apps/web/Dockerfile --snapshot-mode=redo --build-arg NEXT_PUBLIC_API_URL=https://api.mosaicstack.dev $DESTINATIONS
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- build
|
||||||
|
|
||||||
|
# === Container Security Scan ===
|
||||||
|
|
||||||
|
security-trivy-web:
|
||||||
|
image: aquasec/trivy:latest
|
||||||
|
environment:
|
||||||
|
GITEA_USER:
|
||||||
|
from_secret: gitea_username
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
CI_COMMIT_BRANCH: ${CI_COMMIT_BRANCH}
|
||||||
|
CI_COMMIT_TAG: ${CI_COMMIT_TAG}
|
||||||
|
commands:
|
||||||
|
- |
|
||||||
|
if [ -n "$$CI_COMMIT_TAG" ]; then
|
||||||
|
SCAN_TAG="$$CI_COMMIT_TAG"
|
||||||
|
elif [ "$$CI_COMMIT_BRANCH" = "main" ]; then
|
||||||
|
SCAN_TAG="latest"
|
||||||
|
else
|
||||||
|
SCAN_TAG="latest"
|
||||||
|
fi
|
||||||
|
mkdir -p ~/.docker
|
||||||
|
echo "{\"auths\":{\"git.mosaicstack.dev\":{\"username\":\"$$GITEA_USER\",\"password\":\"$$GITEA_TOKEN\"}}}" > ~/.docker/config.json
|
||||||
|
trivy image --exit-code 1 --severity HIGH,CRITICAL --ignore-unfixed \
|
||||||
|
--ignorefile .trivyignore \
|
||||||
|
git.mosaicstack.dev/mosaic/stack-web:$$SCAN_TAG
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- docker-build-web
|
||||||
|
|
||||||
|
# === Package Linking ===
|
||||||
|
|
||||||
|
link-packages:
|
||||||
|
image: alpine:3
|
||||||
|
environment:
|
||||||
|
GITEA_TOKEN:
|
||||||
|
from_secret: gitea_token
|
||||||
|
commands:
|
||||||
|
- apk add --no-cache curl
|
||||||
|
- sleep 10
|
||||||
|
- |
|
||||||
|
set -e
|
||||||
|
link_package() {
|
||||||
|
PKG="$$1"
|
||||||
|
echo "Linking $$PKG..."
|
||||||
|
for attempt in 1 2 3; do
|
||||||
|
STATUS=$$(curl -s -o /tmp/link-response.txt -w "%{http_code}" -X POST \
|
||||||
|
-H "Authorization: token $$GITEA_TOKEN" \
|
||||||
|
"https://git.mosaicstack.dev/api/v1/packages/mosaic/container/$$PKG/-/link/stack")
|
||||||
|
if [ "$$STATUS" = "201" ] || [ "$$STATUS" = "204" ]; then
|
||||||
|
echo " Linked $$PKG"
|
||||||
|
return 0
|
||||||
|
elif [ "$$STATUS" = "400" ]; then
|
||||||
|
echo " $$PKG already linked"
|
||||||
|
return 0
|
||||||
|
elif [ "$$STATUS" = "404" ] && [ $$attempt -lt 3 ]; then
|
||||||
|
echo " $$PKG not found yet, retrying in 5s (attempt $$attempt/3)..."
|
||||||
|
sleep 5
|
||||||
|
else
|
||||||
|
echo " FAILED: $$PKG status $$STATUS"
|
||||||
|
cat /tmp/link-response.txt
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
link_package "stack-web"
|
||||||
|
when:
|
||||||
|
- branch: [main]
|
||||||
|
event: [push, manual, tag]
|
||||||
|
depends_on:
|
||||||
|
- security-trivy-web
|
||||||
@@ -36,7 +36,6 @@
|
|||||||
"@nestjs/mapped-types": "^2.1.0",
|
"@nestjs/mapped-types": "^2.1.0",
|
||||||
"@nestjs/platform-express": "^11.1.12",
|
"@nestjs/platform-express": "^11.1.12",
|
||||||
"@nestjs/platform-socket.io": "^11.1.12",
|
"@nestjs/platform-socket.io": "^11.1.12",
|
||||||
"@nestjs/schedule": "^6.1.1",
|
|
||||||
"@nestjs/throttler": "^6.5.0",
|
"@nestjs/throttler": "^6.5.0",
|
||||||
"@nestjs/websockets": "^11.1.12",
|
"@nestjs/websockets": "^11.1.12",
|
||||||
"@opentelemetry/api": "^1.9.0",
|
"@opentelemetry/api": "^1.9.0",
|
||||||
@@ -60,7 +59,6 @@
|
|||||||
"class-validator": "^0.14.3",
|
"class-validator": "^0.14.3",
|
||||||
"cookie-parser": "^1.4.7",
|
"cookie-parser": "^1.4.7",
|
||||||
"discord.js": "^14.25.1",
|
"discord.js": "^14.25.1",
|
||||||
"dockerode": "^4.0.9",
|
|
||||||
"gray-matter": "^4.0.3",
|
"gray-matter": "^4.0.3",
|
||||||
"highlight.js": "^11.11.1",
|
"highlight.js": "^11.11.1",
|
||||||
"ioredis": "^5.9.2",
|
"ioredis": "^5.9.2",
|
||||||
@@ -90,7 +88,6 @@
|
|||||||
"@types/archiver": "^7.0.0",
|
"@types/archiver": "^7.0.0",
|
||||||
"@types/bcryptjs": "^3.0.0",
|
"@types/bcryptjs": "^3.0.0",
|
||||||
"@types/cookie-parser": "^1.4.10",
|
"@types/cookie-parser": "^1.4.10",
|
||||||
"@types/dockerode": "^3.3.47",
|
|
||||||
"@types/express": "^5.0.1",
|
"@types/express": "^5.0.1",
|
||||||
"@types/highlight.js": "^10.1.0",
|
"@types/highlight.js": "^10.1.0",
|
||||||
"@types/node": "^22.13.4",
|
"@types/node": "^22.13.4",
|
||||||
|
|||||||
@@ -1,24 +0,0 @@
|
|||||||
-- CreateTable
|
|
||||||
CREATE TABLE "agent_memories" (
|
|
||||||
"id" UUID NOT NULL,
|
|
||||||
"workspace_id" UUID NOT NULL,
|
|
||||||
"agent_id" TEXT NOT NULL,
|
|
||||||
"key" TEXT NOT NULL,
|
|
||||||
"value" JSONB NOT NULL,
|
|
||||||
"created_at" TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updated_at" TIMESTAMPTZ NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "agent_memories_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "agent_memories_workspace_id_agent_id_key_key" ON "agent_memories"("workspace_id", "agent_id", "key");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "agent_memories_workspace_id_idx" ON "agent_memories"("workspace_id");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "agent_memories_agent_id_idx" ON "agent_memories"("agent_id");
|
|
||||||
|
|
||||||
-- AddForeignKey
|
|
||||||
ALTER TABLE "agent_memories" ADD CONSTRAINT "agent_memories_workspace_id_fkey" FOREIGN KEY ("workspace_id") REFERENCES "workspaces"("id") ON DELETE CASCADE ON UPDATE CASCADE;
|
|
||||||
@@ -1,33 +0,0 @@
|
|||||||
-- CreateTable
|
|
||||||
CREATE TABLE "conversation_archives" (
|
|
||||||
"id" UUID NOT NULL,
|
|
||||||
"workspace_id" UUID NOT NULL,
|
|
||||||
"session_id" TEXT NOT NULL,
|
|
||||||
"agent_id" TEXT NOT NULL,
|
|
||||||
"messages" JSONB NOT NULL,
|
|
||||||
"message_count" INTEGER NOT NULL,
|
|
||||||
"summary" TEXT NOT NULL,
|
|
||||||
"embedding" vector(1536),
|
|
||||||
"started_at" TIMESTAMPTZ NOT NULL,
|
|
||||||
"ended_at" TIMESTAMPTZ,
|
|
||||||
"metadata" JSONB NOT NULL DEFAULT '{}',
|
|
||||||
"created_at" TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updated_at" TIMESTAMPTZ NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "conversation_archives_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "conversation_archives_workspace_id_session_id_key" ON "conversation_archives"("workspace_id", "session_id");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "conversation_archives_workspace_id_idx" ON "conversation_archives"("workspace_id");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "conversation_archives_agent_id_idx" ON "conversation_archives"("agent_id");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "conversation_archives_started_at_idx" ON "conversation_archives"("started_at");
|
|
||||||
|
|
||||||
-- AddForeignKey
|
|
||||||
ALTER TABLE "conversation_archives" ADD CONSTRAINT "conversation_archives_workspace_id_fkey" FOREIGN KEY ("workspace_id") REFERENCES "workspaces"("id") ON DELETE CASCADE ON UPDATE CASCADE;
|
|
||||||
@@ -1,109 +0,0 @@
|
|||||||
-- CreateTable
|
|
||||||
CREATE TABLE "SystemConfig" (
|
|
||||||
"id" TEXT NOT NULL,
|
|
||||||
"key" TEXT NOT NULL,
|
|
||||||
"value" TEXT NOT NULL,
|
|
||||||
"encrypted" BOOLEAN NOT NULL DEFAULT false,
|
|
||||||
"updatedAt" TIMESTAMP(3) NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "SystemConfig_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateTable
|
|
||||||
CREATE TABLE "BreakglassUser" (
|
|
||||||
"id" TEXT NOT NULL,
|
|
||||||
"username" TEXT NOT NULL,
|
|
||||||
"passwordHash" TEXT NOT NULL,
|
|
||||||
"isActive" BOOLEAN NOT NULL DEFAULT true,
|
|
||||||
"createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updatedAt" TIMESTAMP(3) NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "BreakglassUser_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateTable
|
|
||||||
CREATE TABLE "LlmProvider" (
|
|
||||||
"id" TEXT NOT NULL,
|
|
||||||
"userId" TEXT NOT NULL,
|
|
||||||
"name" TEXT NOT NULL,
|
|
||||||
"displayName" TEXT NOT NULL,
|
|
||||||
"type" TEXT NOT NULL,
|
|
||||||
"baseUrl" TEXT,
|
|
||||||
"apiKey" TEXT,
|
|
||||||
"apiType" TEXT NOT NULL DEFAULT 'openai-completions',
|
|
||||||
"models" JSONB NOT NULL DEFAULT '[]',
|
|
||||||
"isActive" BOOLEAN NOT NULL DEFAULT true,
|
|
||||||
"createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updatedAt" TIMESTAMP(3) NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "LlmProvider_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateTable
|
|
||||||
CREATE TABLE "UserContainer" (
|
|
||||||
"id" TEXT NOT NULL,
|
|
||||||
"userId" TEXT NOT NULL,
|
|
||||||
"containerId" TEXT,
|
|
||||||
"containerName" TEXT NOT NULL,
|
|
||||||
"gatewayPort" INTEGER,
|
|
||||||
"gatewayToken" TEXT NOT NULL,
|
|
||||||
"status" TEXT NOT NULL DEFAULT 'stopped',
|
|
||||||
"lastActiveAt" TIMESTAMP(3),
|
|
||||||
"idleTimeoutMin" INTEGER NOT NULL DEFAULT 30,
|
|
||||||
"config" JSONB NOT NULL DEFAULT '{}',
|
|
||||||
"createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updatedAt" TIMESTAMP(3) NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "UserContainer_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateTable
|
|
||||||
CREATE TABLE "SystemContainer" (
|
|
||||||
"id" TEXT NOT NULL,
|
|
||||||
"name" TEXT NOT NULL,
|
|
||||||
"role" TEXT NOT NULL,
|
|
||||||
"containerId" TEXT,
|
|
||||||
"gatewayPort" INTEGER,
|
|
||||||
"gatewayToken" TEXT NOT NULL,
|
|
||||||
"status" TEXT NOT NULL DEFAULT 'stopped',
|
|
||||||
"primaryModel" TEXT NOT NULL,
|
|
||||||
"isActive" BOOLEAN NOT NULL DEFAULT true,
|
|
||||||
"createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updatedAt" TIMESTAMP(3) NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "SystemContainer_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateTable
|
|
||||||
CREATE TABLE "UserAgentConfig" (
|
|
||||||
"id" TEXT NOT NULL,
|
|
||||||
"userId" TEXT NOT NULL,
|
|
||||||
"primaryModel" TEXT,
|
|
||||||
"fallbackModels" JSONB NOT NULL DEFAULT '[]',
|
|
||||||
"personality" TEXT,
|
|
||||||
"createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updatedAt" TIMESTAMP(3) NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "UserAgentConfig_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "SystemConfig_key_key" ON "SystemConfig"("key");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "BreakglassUser_username_key" ON "BreakglassUser"("username");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "LlmProvider_userId_idx" ON "LlmProvider"("userId");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "LlmProvider_userId_name_key" ON "LlmProvider"("userId", "name");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "UserContainer_userId_key" ON "UserContainer"("userId");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "SystemContainer_name_key" ON "SystemContainer"("name");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "UserAgentConfig_userId_key" ON "UserAgentConfig"("userId");
|
|
||||||
@@ -1,37 +0,0 @@
|
|||||||
-- CreateTable
|
|
||||||
CREATE TABLE "findings" (
|
|
||||||
"id" UUID NOT NULL,
|
|
||||||
"workspace_id" UUID NOT NULL,
|
|
||||||
"task_id" UUID,
|
|
||||||
"agent_id" TEXT NOT NULL,
|
|
||||||
"type" TEXT NOT NULL,
|
|
||||||
"title" TEXT NOT NULL,
|
|
||||||
"data" JSONB NOT NULL,
|
|
||||||
"summary" TEXT NOT NULL,
|
|
||||||
"embedding" vector(1536),
|
|
||||||
"created_at" TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
"updated_at" TIMESTAMPTZ NOT NULL,
|
|
||||||
|
|
||||||
CONSTRAINT "findings_pkey" PRIMARY KEY ("id")
|
|
||||||
);
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE UNIQUE INDEX "findings_id_workspace_id_key" ON "findings"("id", "workspace_id");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "findings_workspace_id_idx" ON "findings"("workspace_id");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "findings_agent_id_idx" ON "findings"("agent_id");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "findings_type_idx" ON "findings"("type");
|
|
||||||
|
|
||||||
-- CreateIndex
|
|
||||||
CREATE INDEX "findings_task_id_idx" ON "findings"("task_id");
|
|
||||||
|
|
||||||
-- AddForeignKey
|
|
||||||
ALTER TABLE "findings" ADD CONSTRAINT "findings_workspace_id_fkey" FOREIGN KEY ("workspace_id") REFERENCES "workspaces"("id") ON DELETE CASCADE ON UPDATE CASCADE;
|
|
||||||
|
|
||||||
-- AddForeignKey
|
|
||||||
ALTER TABLE "findings" ADD CONSTRAINT "findings_task_id_fkey" FOREIGN KEY ("task_id") REFERENCES "agent_tasks"("id") ON DELETE SET NULL ON UPDATE CASCADE;
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
-- AlterTable
|
|
||||||
ALTER TABLE "tasks" ADD COLUMN "assigned_agent" TEXT;
|
|
||||||
@@ -298,8 +298,6 @@ model Workspace {
|
|||||||
agents Agent[]
|
agents Agent[]
|
||||||
agentSessions AgentSession[]
|
agentSessions AgentSession[]
|
||||||
agentTasks AgentTask[]
|
agentTasks AgentTask[]
|
||||||
findings Finding[]
|
|
||||||
agentMemories AgentMemory[]
|
|
||||||
userLayouts UserLayout[]
|
userLayouts UserLayout[]
|
||||||
knowledgeEntries KnowledgeEntry[]
|
knowledgeEntries KnowledgeEntry[]
|
||||||
knowledgeTags KnowledgeTag[]
|
knowledgeTags KnowledgeTag[]
|
||||||
@@ -314,7 +312,6 @@ model Workspace {
|
|||||||
llmUsageLogs LlmUsageLog[]
|
llmUsageLogs LlmUsageLog[]
|
||||||
userCredentials UserCredential[]
|
userCredentials UserCredential[]
|
||||||
terminalSessions TerminalSession[]
|
terminalSessions TerminalSession[]
|
||||||
conversationArchives ConversationArchive[]
|
|
||||||
|
|
||||||
@@index([ownerId])
|
@@index([ownerId])
|
||||||
@@map("workspaces")
|
@@map("workspaces")
|
||||||
@@ -379,7 +376,6 @@ model Task {
|
|||||||
creatorId String @map("creator_id") @db.Uuid
|
creatorId String @map("creator_id") @db.Uuid
|
||||||
projectId String? @map("project_id") @db.Uuid
|
projectId String? @map("project_id") @db.Uuid
|
||||||
parentId String? @map("parent_id") @db.Uuid
|
parentId String? @map("parent_id") @db.Uuid
|
||||||
assignedAgent String? @map("assigned_agent")
|
|
||||||
domainId String? @map("domain_id") @db.Uuid
|
domainId String? @map("domain_id") @db.Uuid
|
||||||
sortOrder Int @default(0) @map("sort_order")
|
sortOrder Int @default(0) @map("sort_order")
|
||||||
metadata Json @default("{}")
|
metadata Json @default("{}")
|
||||||
@@ -693,7 +689,6 @@ model AgentTask {
|
|||||||
createdBy User @relation("AgentTaskCreator", fields: [createdById], references: [id], onDelete: Cascade)
|
createdBy User @relation("AgentTaskCreator", fields: [createdById], references: [id], onDelete: Cascade)
|
||||||
createdById String @map("created_by_id") @db.Uuid
|
createdById String @map("created_by_id") @db.Uuid
|
||||||
runnerJobs RunnerJob[]
|
runnerJobs RunnerJob[]
|
||||||
findings Finding[]
|
|
||||||
|
|
||||||
@@unique([id, workspaceId])
|
@@unique([id, workspaceId])
|
||||||
@@index([workspaceId])
|
@@index([workspaceId])
|
||||||
@@ -703,33 +698,6 @@ model AgentTask {
|
|||||||
@@map("agent_tasks")
|
@@map("agent_tasks")
|
||||||
}
|
}
|
||||||
|
|
||||||
model Finding {
|
|
||||||
id String @id @default(uuid()) @db.Uuid
|
|
||||||
workspaceId String @map("workspace_id") @db.Uuid
|
|
||||||
taskId String? @map("task_id") @db.Uuid
|
|
||||||
|
|
||||||
agentId String @map("agent_id")
|
|
||||||
type String
|
|
||||||
title String
|
|
||||||
data Json
|
|
||||||
summary String @db.Text
|
|
||||||
// Note: vector dimension (1536) must match EMBEDDING_DIMENSION constant in @mosaic/shared
|
|
||||||
embedding Unsupported("vector(1536)")?
|
|
||||||
|
|
||||||
createdAt DateTime @default(now()) @map("created_at") @db.Timestamptz
|
|
||||||
updatedAt DateTime @updatedAt @map("updated_at") @db.Timestamptz
|
|
||||||
|
|
||||||
workspace Workspace @relation(fields: [workspaceId], references: [id], onDelete: Cascade)
|
|
||||||
task AgentTask? @relation(fields: [taskId], references: [id], onDelete: SetNull)
|
|
||||||
|
|
||||||
@@unique([id, workspaceId])
|
|
||||||
@@index([workspaceId])
|
|
||||||
@@index([agentId])
|
|
||||||
@@index([type])
|
|
||||||
@@index([taskId])
|
|
||||||
@@map("findings")
|
|
||||||
}
|
|
||||||
|
|
||||||
model AgentSession {
|
model AgentSession {
|
||||||
id String @id @default(uuid()) @db.Uuid
|
id String @id @default(uuid()) @db.Uuid
|
||||||
workspaceId String @map("workspace_id") @db.Uuid
|
workspaceId String @map("workspace_id") @db.Uuid
|
||||||
@@ -767,23 +735,6 @@ model AgentSession {
|
|||||||
@@map("agent_sessions")
|
@@map("agent_sessions")
|
||||||
}
|
}
|
||||||
|
|
||||||
model AgentMemory {
|
|
||||||
id String @id @default(uuid()) @db.Uuid
|
|
||||||
workspaceId String @map("workspace_id") @db.Uuid
|
|
||||||
agentId String @map("agent_id")
|
|
||||||
key String
|
|
||||||
value Json
|
|
||||||
createdAt DateTime @default(now()) @map("created_at") @db.Timestamptz
|
|
||||||
updatedAt DateTime @updatedAt @map("updated_at") @db.Timestamptz
|
|
||||||
|
|
||||||
workspace Workspace @relation(fields: [workspaceId], references: [id], onDelete: Cascade)
|
|
||||||
|
|
||||||
@@unique([workspaceId, agentId, key])
|
|
||||||
@@index([workspaceId])
|
|
||||||
@@index([agentId])
|
|
||||||
@@map("agent_memories")
|
|
||||||
}
|
|
||||||
|
|
||||||
model WidgetDefinition {
|
model WidgetDefinition {
|
||||||
id String @id @default(uuid()) @db.Uuid
|
id String @id @default(uuid()) @db.Uuid
|
||||||
|
|
||||||
@@ -1595,111 +1546,3 @@ model TerminalSession {
|
|||||||
@@index([workspaceId, status])
|
@@index([workspaceId, status])
|
||||||
@@map("terminal_sessions")
|
@@map("terminal_sessions")
|
||||||
}
|
}
|
||||||
|
|
||||||
// ============================================
|
|
||||||
// CONVERSATION ARCHIVE MODULE
|
|
||||||
// ============================================
|
|
||||||
|
|
||||||
model ConversationArchive {
|
|
||||||
id String @id @default(uuid()) @db.Uuid
|
|
||||||
workspaceId String @map("workspace_id") @db.Uuid
|
|
||||||
sessionId String @map("session_id")
|
|
||||||
agentId String @map("agent_id")
|
|
||||||
messages Json
|
|
||||||
messageCount Int @map("message_count")
|
|
||||||
summary String @db.Text
|
|
||||||
// Note: vector dimension (1536) must match EMBEDDING_DIMENSION constant in @mosaic/shared
|
|
||||||
embedding Unsupported("vector(1536)")?
|
|
||||||
startedAt DateTime @map("started_at") @db.Timestamptz
|
|
||||||
endedAt DateTime? @map("ended_at") @db.Timestamptz
|
|
||||||
metadata Json @default("{}")
|
|
||||||
createdAt DateTime @default(now()) @map("created_at") @db.Timestamptz
|
|
||||||
updatedAt DateTime @updatedAt @map("updated_at") @db.Timestamptz
|
|
||||||
|
|
||||||
// Relations
|
|
||||||
workspace Workspace @relation(fields: [workspaceId], references: [id], onDelete: Cascade)
|
|
||||||
|
|
||||||
@@unique([workspaceId, sessionId])
|
|
||||||
@@index([workspaceId])
|
|
||||||
@@index([agentId])
|
|
||||||
@@index([startedAt])
|
|
||||||
@@map("conversation_archives")
|
|
||||||
}
|
|
||||||
|
|
||||||
// ============================================
|
|
||||||
// AGENT FLEET MODULE
|
|
||||||
// ============================================
|
|
||||||
|
|
||||||
model SystemConfig {
|
|
||||||
id String @id @default(cuid())
|
|
||||||
key String @unique
|
|
||||||
value String
|
|
||||||
encrypted Boolean @default(false)
|
|
||||||
updatedAt DateTime @updatedAt
|
|
||||||
}
|
|
||||||
|
|
||||||
model BreakglassUser {
|
|
||||||
id String @id @default(cuid())
|
|
||||||
username String @unique
|
|
||||||
passwordHash String
|
|
||||||
isActive Boolean @default(true)
|
|
||||||
createdAt DateTime @default(now())
|
|
||||||
updatedAt DateTime @updatedAt
|
|
||||||
}
|
|
||||||
|
|
||||||
model LlmProvider {
|
|
||||||
id String @id @default(cuid())
|
|
||||||
userId String
|
|
||||||
name String
|
|
||||||
displayName String
|
|
||||||
type String
|
|
||||||
baseUrl String?
|
|
||||||
apiKey String?
|
|
||||||
apiType String @default("openai-completions")
|
|
||||||
models Json @default("[]")
|
|
||||||
isActive Boolean @default(true)
|
|
||||||
createdAt DateTime @default(now())
|
|
||||||
updatedAt DateTime @updatedAt
|
|
||||||
|
|
||||||
@@unique([userId, name])
|
|
||||||
@@index([userId])
|
|
||||||
}
|
|
||||||
|
|
||||||
model UserContainer {
|
|
||||||
id String @id @default(cuid())
|
|
||||||
userId String @unique
|
|
||||||
containerId String?
|
|
||||||
containerName String
|
|
||||||
gatewayPort Int?
|
|
||||||
gatewayToken String
|
|
||||||
status String @default("stopped")
|
|
||||||
lastActiveAt DateTime?
|
|
||||||
idleTimeoutMin Int @default(30)
|
|
||||||
config Json @default("{}")
|
|
||||||
createdAt DateTime @default(now())
|
|
||||||
updatedAt DateTime @updatedAt
|
|
||||||
}
|
|
||||||
|
|
||||||
model SystemContainer {
|
|
||||||
id String @id @default(cuid())
|
|
||||||
name String @unique
|
|
||||||
role String
|
|
||||||
containerId String?
|
|
||||||
gatewayPort Int?
|
|
||||||
gatewayToken String
|
|
||||||
status String @default("stopped")
|
|
||||||
primaryModel String
|
|
||||||
isActive Boolean @default(true)
|
|
||||||
createdAt DateTime @default(now())
|
|
||||||
updatedAt DateTime @updatedAt
|
|
||||||
}
|
|
||||||
|
|
||||||
model UserAgentConfig {
|
|
||||||
id String @id @default(cuid())
|
|
||||||
userId String @unique
|
|
||||||
primaryModel String?
|
|
||||||
fallbackModels Json @default("[]")
|
|
||||||
personality String?
|
|
||||||
createdAt DateTime @default(now())
|
|
||||||
updatedAt DateTime @updatedAt
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,40 +0,0 @@
|
|||||||
import {
|
|
||||||
Controller,
|
|
||||||
ForbiddenException,
|
|
||||||
Get,
|
|
||||||
Param,
|
|
||||||
Req,
|
|
||||||
UnauthorizedException,
|
|
||||||
UseGuards,
|
|
||||||
} from "@nestjs/common";
|
|
||||||
import { AgentConfigService } from "./agent-config.service";
|
|
||||||
import { AgentConfigGuard, type AgentConfigRequest } from "./agent-config.guard";
|
|
||||||
|
|
||||||
@Controller("internal")
|
|
||||||
@UseGuards(AgentConfigGuard)
|
|
||||||
export class AgentConfigController {
|
|
||||||
constructor(private readonly agentConfigService: AgentConfigService) {}
|
|
||||||
|
|
||||||
// GET /api/internal/agent-config/:id
|
|
||||||
// Auth: Bearer token (validated against UserContainer.gatewayToken or SystemContainer.gatewayToken)
|
|
||||||
// Returns: assembled openclaw.json
|
|
||||||
//
|
|
||||||
// The :id param is the container record ID (cuid)
|
|
||||||
// Token must match the container requesting its own config
|
|
||||||
@Get("agent-config/:id")
|
|
||||||
async getAgentConfig(
|
|
||||||
@Param("id") id: string,
|
|
||||||
@Req() request: AgentConfigRequest
|
|
||||||
): Promise<object> {
|
|
||||||
const containerAuth = request.containerAuth;
|
|
||||||
if (!containerAuth) {
|
|
||||||
throw new UnauthorizedException("Missing container authentication context");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (containerAuth.id !== id) {
|
|
||||||
throw new ForbiddenException("Token is not authorized for the requested container");
|
|
||||||
}
|
|
||||||
|
|
||||||
return this.agentConfigService.generateConfigForContainer(containerAuth.type, id);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,43 +0,0 @@
|
|||||||
import { CanActivate, ExecutionContext, Injectable, UnauthorizedException } from "@nestjs/common";
|
|
||||||
import type { Request } from "express";
|
|
||||||
import { AgentConfigService, type ContainerTokenValidation } from "./agent-config.service";
|
|
||||||
|
|
||||||
export interface AgentConfigRequest extends Request {
|
|
||||||
containerAuth?: ContainerTokenValidation;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class AgentConfigGuard implements CanActivate {
|
|
||||||
constructor(private readonly agentConfigService: AgentConfigService) {}
|
|
||||||
|
|
||||||
async canActivate(context: ExecutionContext): Promise<boolean> {
|
|
||||||
const request = context.switchToHttp().getRequest<AgentConfigRequest>();
|
|
||||||
const token = this.extractBearerToken(request.headers.authorization);
|
|
||||||
|
|
||||||
if (!token) {
|
|
||||||
throw new UnauthorizedException("Missing Bearer token");
|
|
||||||
}
|
|
||||||
|
|
||||||
const containerAuth = await this.agentConfigService.validateContainerToken(token);
|
|
||||||
if (!containerAuth) {
|
|
||||||
throw new UnauthorizedException("Invalid container token");
|
|
||||||
}
|
|
||||||
|
|
||||||
request.containerAuth = containerAuth;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
private extractBearerToken(headerValue: string | string[] | undefined): string | null {
|
|
||||||
const normalizedHeader = Array.isArray(headerValue) ? headerValue[0] : headerValue;
|
|
||||||
if (!normalizedHeader) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
const [scheme, token] = normalizedHeader.split(" ");
|
|
||||||
if (!scheme || !token || scheme.toLowerCase() !== "bearer") {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
return token;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,14 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { CryptoModule } from "../crypto/crypto.module";
|
|
||||||
import { AgentConfigController } from "./agent-config.controller";
|
|
||||||
import { AgentConfigService } from "./agent-config.service";
|
|
||||||
import { AgentConfigGuard } from "./agent-config.guard";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [PrismaModule, CryptoModule],
|
|
||||||
controllers: [AgentConfigController],
|
|
||||||
providers: [AgentConfigService, AgentConfigGuard],
|
|
||||||
exports: [AgentConfigService],
|
|
||||||
})
|
|
||||||
export class AgentConfigModule {}
|
|
||||||
@@ -1,215 +0,0 @@
|
|||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
import { AgentConfigService } from "./agent-config.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { CryptoService } from "../crypto/crypto.service";
|
|
||||||
|
|
||||||
describe("AgentConfigService", () => {
|
|
||||||
let service: AgentConfigService;
|
|
||||||
|
|
||||||
const mockPrismaService = {
|
|
||||||
userAgentConfig: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
},
|
|
||||||
llmProvider: {
|
|
||||||
findMany: vi.fn(),
|
|
||||||
},
|
|
||||||
userContainer: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
findMany: vi.fn(),
|
|
||||||
},
|
|
||||||
systemContainer: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
findMany: vi.fn(),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockCryptoService = {
|
|
||||||
isEncrypted: vi.fn((value: string) => value.startsWith("enc:")),
|
|
||||||
decrypt: vi.fn((value: string) => value.replace(/^enc:/, "")),
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
|
|
||||||
service = new AgentConfigService(
|
|
||||||
mockPrismaService as unknown as PrismaService,
|
|
||||||
mockCryptoService as unknown as CryptoService
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("generateUserConfig returns valid openclaw.json structure", async () => {
|
|
||||||
mockPrismaService.userAgentConfig.findUnique.mockResolvedValue({
|
|
||||||
id: "cfg-1",
|
|
||||||
userId: "user-1",
|
|
||||||
primaryModel: "my-zai/glm-5",
|
|
||||||
});
|
|
||||||
|
|
||||||
mockPrismaService.userContainer.findUnique.mockResolvedValue({
|
|
||||||
id: "container-1",
|
|
||||||
userId: "user-1",
|
|
||||||
gatewayPort: 19001,
|
|
||||||
});
|
|
||||||
|
|
||||||
mockPrismaService.llmProvider.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "provider-1",
|
|
||||||
userId: "user-1",
|
|
||||||
name: "my-zai",
|
|
||||||
displayName: "Z.ai",
|
|
||||||
type: "zai",
|
|
||||||
baseUrl: "https://api.z.ai/v1",
|
|
||||||
apiKey: "enc:secret-zai-key",
|
|
||||||
apiType: "openai-completions",
|
|
||||||
models: [{ id: "glm-5" }],
|
|
||||||
isActive: true,
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await service.generateUserConfig("user-1");
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
gateway: {
|
|
||||||
mode: "local",
|
|
||||||
port: 19001,
|
|
||||||
bind: "lan",
|
|
||||||
auth: { mode: "token" },
|
|
||||||
http: {
|
|
||||||
endpoints: {
|
|
||||||
chatCompletions: { enabled: true },
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
agents: {
|
|
||||||
defaults: {
|
|
||||||
model: {
|
|
||||||
primary: "my-zai/glm-5",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
models: {
|
|
||||||
providers: {
|
|
||||||
"my-zai": {
|
|
||||||
apiKey: "secret-zai-key",
|
|
||||||
baseUrl: "https://api.z.ai/v1",
|
|
||||||
models: {
|
|
||||||
"glm-5": {},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("generateUserConfig decrypts API keys correctly", async () => {
|
|
||||||
mockPrismaService.userAgentConfig.findUnique.mockResolvedValue({
|
|
||||||
id: "cfg-1",
|
|
||||||
userId: "user-1",
|
|
||||||
primaryModel: "openai-work/gpt-4.1",
|
|
||||||
});
|
|
||||||
|
|
||||||
mockPrismaService.userContainer.findUnique.mockResolvedValue({
|
|
||||||
id: "container-1",
|
|
||||||
userId: "user-1",
|
|
||||||
gatewayPort: 18789,
|
|
||||||
});
|
|
||||||
|
|
||||||
mockPrismaService.llmProvider.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "provider-1",
|
|
||||||
userId: "user-1",
|
|
||||||
name: "openai-work",
|
|
||||||
displayName: "OpenAI Work",
|
|
||||||
type: "openai",
|
|
||||||
baseUrl: "https://api.openai.com/v1",
|
|
||||||
apiKey: "enc:encrypted-openai-key",
|
|
||||||
apiType: "openai-completions",
|
|
||||||
models: [{ id: "gpt-4.1" }],
|
|
||||||
isActive: true,
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await service.generateUserConfig("user-1");
|
|
||||||
|
|
||||||
expect(mockCryptoService.decrypt).toHaveBeenCalledWith("enc:encrypted-openai-key");
|
|
||||||
expect(result.models.providers["openai-work"]?.apiKey).toBe("encrypted-openai-key");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("generateUserConfig handles user with no providers", async () => {
|
|
||||||
mockPrismaService.userAgentConfig.findUnique.mockResolvedValue({
|
|
||||||
id: "cfg-1",
|
|
||||||
userId: "user-2",
|
|
||||||
primaryModel: "openai/gpt-4o-mini",
|
|
||||||
});
|
|
||||||
|
|
||||||
mockPrismaService.userContainer.findUnique.mockResolvedValue({
|
|
||||||
id: "container-2",
|
|
||||||
userId: "user-2",
|
|
||||||
gatewayPort: null,
|
|
||||||
});
|
|
||||||
|
|
||||||
mockPrismaService.llmProvider.findMany.mockResolvedValue([]);
|
|
||||||
|
|
||||||
const result = await service.generateUserConfig("user-2");
|
|
||||||
|
|
||||||
expect(result.models.providers).toEqual({});
|
|
||||||
expect(result.gateway.port).toBe(18789);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("validateContainerToken returns correct type for user container", async () => {
|
|
||||||
mockPrismaService.userContainer.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "user-container-1",
|
|
||||||
gatewayToken: "enc:user-token-1",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
mockPrismaService.systemContainer.findMany.mockResolvedValue([]);
|
|
||||||
|
|
||||||
const result = await service.validateContainerToken("user-token-1");
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
type: "user",
|
|
||||||
id: "user-container-1",
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("validateContainerToken returns correct type for system container", async () => {
|
|
||||||
mockPrismaService.userContainer.findMany.mockResolvedValue([]);
|
|
||||||
mockPrismaService.systemContainer.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "system-container-1",
|
|
||||||
gatewayToken: "enc:system-token-1",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await service.validateContainerToken("system-token-1");
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
type: "system",
|
|
||||||
id: "system-container-1",
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("validateContainerToken returns null for invalid token", async () => {
|
|
||||||
mockPrismaService.userContainer.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "user-container-1",
|
|
||||||
gatewayToken: "enc:user-token-1",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
mockPrismaService.systemContainer.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "system-container-1",
|
|
||||||
gatewayToken: "enc:system-token-1",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await service.validateContainerToken("no-match");
|
|
||||||
|
|
||||||
expect(result).toBeNull();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,285 +0,0 @@
|
|||||||
import { Injectable, NotFoundException } from "@nestjs/common";
|
|
||||||
import type { LlmProvider } from "@prisma/client";
|
|
||||||
import { createHash, timingSafeEqual } from "node:crypto";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { CryptoService } from "../crypto/crypto.service";
|
|
||||||
|
|
||||||
const DEFAULT_GATEWAY_PORT = 18789;
|
|
||||||
const DEFAULT_PRIMARY_MODEL = "openai/gpt-4o-mini";
|
|
||||||
|
|
||||||
type ContainerType = "user" | "system";
|
|
||||||
|
|
||||||
export interface ContainerTokenValidation {
|
|
||||||
type: ContainerType;
|
|
||||||
id: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
type OpenClawModelMap = Record<string, Record<string, never>>;
|
|
||||||
|
|
||||||
interface OpenClawProviderConfig {
|
|
||||||
apiKey?: string;
|
|
||||||
baseUrl?: string;
|
|
||||||
models: OpenClawModelMap;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface OpenClawConfig {
|
|
||||||
gateway: {
|
|
||||||
mode: "local";
|
|
||||||
port: number;
|
|
||||||
bind: "lan";
|
|
||||||
auth: { mode: "token" };
|
|
||||||
http: {
|
|
||||||
endpoints: {
|
|
||||||
chatCompletions: { enabled: true };
|
|
||||||
};
|
|
||||||
};
|
|
||||||
};
|
|
||||||
agents: {
|
|
||||||
defaults: {
|
|
||||||
model: {
|
|
||||||
primary: string;
|
|
||||||
};
|
|
||||||
};
|
|
||||||
};
|
|
||||||
models: {
|
|
||||||
providers: Record<string, OpenClawProviderConfig>;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class AgentConfigService {
|
|
||||||
constructor(
|
|
||||||
private readonly prisma: PrismaService,
|
|
||||||
private readonly crypto: CryptoService
|
|
||||||
) {}
|
|
||||||
|
|
||||||
// Generate complete openclaw.json for a user container
|
|
||||||
async generateUserConfig(userId: string): Promise<OpenClawConfig> {
|
|
||||||
const [userAgentConfig, providers, userContainer] = await Promise.all([
|
|
||||||
this.prisma.userAgentConfig.findUnique({
|
|
||||||
where: { userId },
|
|
||||||
}),
|
|
||||||
this.prisma.llmProvider.findMany({
|
|
||||||
where: {
|
|
||||||
userId,
|
|
||||||
isActive: true,
|
|
||||||
},
|
|
||||||
orderBy: {
|
|
||||||
createdAt: "asc",
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
this.prisma.userContainer.findUnique({
|
|
||||||
where: { userId },
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
|
|
||||||
if (!userContainer) {
|
|
||||||
throw new NotFoundException(`User container not found for user ${userId}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
const primaryModel =
|
|
||||||
userAgentConfig?.primaryModel ??
|
|
||||||
this.resolvePrimaryModelFromProviders(providers) ??
|
|
||||||
DEFAULT_PRIMARY_MODEL;
|
|
||||||
|
|
||||||
return this.buildOpenClawConfig(primaryModel, userContainer.gatewayPort, providers);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate config for a system container
|
|
||||||
async generateSystemConfig(containerId: string): Promise<OpenClawConfig> {
|
|
||||||
const systemContainer = await this.prisma.systemContainer.findUnique({
|
|
||||||
where: { id: containerId },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!systemContainer) {
|
|
||||||
throw new NotFoundException(`System container ${containerId} not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return this.buildOpenClawConfig(
|
|
||||||
systemContainer.primaryModel || DEFAULT_PRIMARY_MODEL,
|
|
||||||
systemContainer.gatewayPort,
|
|
||||||
[]
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
async generateConfigForContainer(
|
|
||||||
type: ContainerType,
|
|
||||||
containerId: string
|
|
||||||
): Promise<OpenClawConfig> {
|
|
||||||
if (type === "system") {
|
|
||||||
return this.generateSystemConfig(containerId);
|
|
||||||
}
|
|
||||||
|
|
||||||
const userContainer = await this.prisma.userContainer.findUnique({
|
|
||||||
where: { id: containerId },
|
|
||||||
select: { userId: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!userContainer) {
|
|
||||||
throw new NotFoundException(`User container ${containerId} not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return this.generateUserConfig(userContainer.userId);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Validate a container's bearer token
|
|
||||||
async validateContainerToken(token: string): Promise<ContainerTokenValidation | null> {
|
|
||||||
if (!token) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
const [userContainers, systemContainers] = await Promise.all([
|
|
||||||
this.prisma.userContainer.findMany({
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
gatewayToken: true,
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
this.prisma.systemContainer.findMany({
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
gatewayToken: true,
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
|
|
||||||
let match: ContainerTokenValidation | null = null;
|
|
||||||
|
|
||||||
for (const container of userContainers) {
|
|
||||||
const storedToken = this.decryptContainerToken(container.gatewayToken);
|
|
||||||
if (!match && storedToken && this.tokensEqual(storedToken, token)) {
|
|
||||||
match = { type: "user", id: container.id };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const container of systemContainers) {
|
|
||||||
const storedToken = this.decryptContainerToken(container.gatewayToken);
|
|
||||||
if (!match && storedToken && this.tokensEqual(storedToken, token)) {
|
|
||||||
match = { type: "system", id: container.id };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return match;
|
|
||||||
}
|
|
||||||
|
|
||||||
private buildOpenClawConfig(
|
|
||||||
primaryModel: string,
|
|
||||||
gatewayPort: number | null,
|
|
||||||
providers: LlmProvider[]
|
|
||||||
): OpenClawConfig {
|
|
||||||
return {
|
|
||||||
gateway: {
|
|
||||||
mode: "local",
|
|
||||||
port: gatewayPort ?? DEFAULT_GATEWAY_PORT,
|
|
||||||
bind: "lan",
|
|
||||||
auth: { mode: "token" },
|
|
||||||
http: {
|
|
||||||
endpoints: {
|
|
||||||
chatCompletions: { enabled: true },
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
agents: {
|
|
||||||
defaults: {
|
|
||||||
model: {
|
|
||||||
primary: primaryModel,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
models: {
|
|
||||||
providers: this.buildProviderConfig(providers),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
private buildProviderConfig(providers: LlmProvider[]): Record<string, OpenClawProviderConfig> {
|
|
||||||
const providerConfig: Record<string, OpenClawProviderConfig> = {};
|
|
||||||
|
|
||||||
for (const provider of providers) {
|
|
||||||
const config: OpenClawProviderConfig = {
|
|
||||||
models: this.extractModels(provider.models),
|
|
||||||
};
|
|
||||||
|
|
||||||
const apiKey = this.decryptIfNeeded(provider.apiKey);
|
|
||||||
if (apiKey) {
|
|
||||||
config.apiKey = apiKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (provider.baseUrl) {
|
|
||||||
config.baseUrl = provider.baseUrl;
|
|
||||||
}
|
|
||||||
|
|
||||||
providerConfig[provider.name] = config;
|
|
||||||
}
|
|
||||||
|
|
||||||
return providerConfig;
|
|
||||||
}
|
|
||||||
|
|
||||||
private extractModels(models: unknown): OpenClawModelMap {
|
|
||||||
const modelMap: OpenClawModelMap = {};
|
|
||||||
|
|
||||||
if (!Array.isArray(models)) {
|
|
||||||
return modelMap;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const modelEntry of models) {
|
|
||||||
if (typeof modelEntry === "string") {
|
|
||||||
modelMap[modelEntry] = {};
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (this.hasModelId(modelEntry)) {
|
|
||||||
modelMap[modelEntry.id] = {};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return modelMap;
|
|
||||||
}
|
|
||||||
|
|
||||||
private resolvePrimaryModelFromProviders(providers: LlmProvider[]): string | null {
|
|
||||||
for (const provider of providers) {
|
|
||||||
const modelIds = Object.keys(this.extractModels(provider.models));
|
|
||||||
const firstModelId = modelIds[0];
|
|
||||||
|
|
||||||
if (firstModelId) {
|
|
||||||
return `${provider.name}/${firstModelId}`;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
private decryptIfNeeded(value: string | null | undefined): string | undefined {
|
|
||||||
if (!value) {
|
|
||||||
return undefined;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (this.crypto.isEncrypted(value)) {
|
|
||||||
return this.crypto.decrypt(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
|
|
||||||
private decryptContainerToken(value: string): string | null {
|
|
||||||
try {
|
|
||||||
return this.decryptIfNeeded(value) ?? null;
|
|
||||||
} catch {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private tokensEqual(left: string, right: string): boolean {
|
|
||||||
const leftDigest = createHash("sha256").update(left, "utf8").digest();
|
|
||||||
const rightDigest = createHash("sha256").update(right, "utf8").digest();
|
|
||||||
return timingSafeEqual(leftDigest, rightDigest);
|
|
||||||
}
|
|
||||||
|
|
||||||
private hasModelId(modelEntry: unknown): modelEntry is { id: string } {
|
|
||||||
if (typeof modelEntry !== "object" || modelEntry === null || !("id" in modelEntry)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return typeof (modelEntry as { id?: unknown }).id === "string";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,102 +0,0 @@
|
|||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { AgentMemoryController } from "./agent-memory.controller";
|
|
||||||
import { AgentMemoryService } from "./agent-memory.service";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
|
|
||||||
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
||||||
|
|
||||||
describe("AgentMemoryController", () => {
|
|
||||||
let controller: AgentMemoryController;
|
|
||||||
|
|
||||||
const mockAgentMemoryService = {
|
|
||||||
upsert: vi.fn(),
|
|
||||||
findAll: vi.fn(),
|
|
||||||
findOne: vi.fn(),
|
|
||||||
remove: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockGuard = { canActivate: vi.fn(() => true) };
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
const module: TestingModule = await Test.createTestingModule({
|
|
||||||
controllers: [AgentMemoryController],
|
|
||||||
providers: [
|
|
||||||
{
|
|
||||||
provide: AgentMemoryService,
|
|
||||||
useValue: mockAgentMemoryService,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
})
|
|
||||||
.overrideGuard(AuthGuard)
|
|
||||||
.useValue(mockGuard)
|
|
||||||
.overrideGuard(WorkspaceGuard)
|
|
||||||
.useValue(mockGuard)
|
|
||||||
.overrideGuard(PermissionGuard)
|
|
||||||
.useValue(mockGuard)
|
|
||||||
.compile();
|
|
||||||
|
|
||||||
controller = module.get<AgentMemoryController>(AgentMemoryController);
|
|
||||||
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
const workspaceId = "workspace-1";
|
|
||||||
const agentId = "agent-1";
|
|
||||||
const key = "context";
|
|
||||||
|
|
||||||
describe("upsert", () => {
|
|
||||||
it("should upsert a memory entry", async () => {
|
|
||||||
const dto = { value: { foo: "bar" } };
|
|
||||||
const mockEntry = { id: "mem-1", workspaceId, agentId, key, value: dto.value };
|
|
||||||
|
|
||||||
mockAgentMemoryService.upsert.mockResolvedValue(mockEntry);
|
|
||||||
|
|
||||||
const result = await controller.upsert(agentId, key, dto, workspaceId);
|
|
||||||
|
|
||||||
expect(mockAgentMemoryService.upsert).toHaveBeenCalledWith(workspaceId, agentId, key, dto);
|
|
||||||
expect(result).toEqual(mockEntry);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findAll", () => {
|
|
||||||
it("should list all memory entries for an agent", async () => {
|
|
||||||
const mockEntries = [
|
|
||||||
{ id: "mem-1", key: "a", value: 1 },
|
|
||||||
{ id: "mem-2", key: "b", value: 2 },
|
|
||||||
];
|
|
||||||
|
|
||||||
mockAgentMemoryService.findAll.mockResolvedValue(mockEntries);
|
|
||||||
|
|
||||||
const result = await controller.findAll(agentId, workspaceId);
|
|
||||||
|
|
||||||
expect(mockAgentMemoryService.findAll).toHaveBeenCalledWith(workspaceId, agentId);
|
|
||||||
expect(result).toEqual(mockEntries);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findOne", () => {
|
|
||||||
it("should get a single memory entry", async () => {
|
|
||||||
const mockEntry = { id: "mem-1", key, value: "v" };
|
|
||||||
|
|
||||||
mockAgentMemoryService.findOne.mockResolvedValue(mockEntry);
|
|
||||||
|
|
||||||
const result = await controller.findOne(agentId, key, workspaceId);
|
|
||||||
|
|
||||||
expect(mockAgentMemoryService.findOne).toHaveBeenCalledWith(workspaceId, agentId, key);
|
|
||||||
expect(result).toEqual(mockEntry);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("remove", () => {
|
|
||||||
it("should delete a memory entry", async () => {
|
|
||||||
const mockResponse = { message: "Memory entry deleted successfully" };
|
|
||||||
|
|
||||||
mockAgentMemoryService.remove.mockResolvedValue(mockResponse);
|
|
||||||
|
|
||||||
const result = await controller.remove(agentId, key, workspaceId);
|
|
||||||
|
|
||||||
expect(mockAgentMemoryService.remove).toHaveBeenCalledWith(workspaceId, agentId, key);
|
|
||||||
expect(result).toEqual(mockResponse);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,89 +0,0 @@
|
|||||||
import {
|
|
||||||
Controller,
|
|
||||||
Get,
|
|
||||||
Put,
|
|
||||||
Delete,
|
|
||||||
Body,
|
|
||||||
Param,
|
|
||||||
UseGuards,
|
|
||||||
HttpCode,
|
|
||||||
HttpStatus,
|
|
||||||
} from "@nestjs/common";
|
|
||||||
import { AgentMemoryService } from "./agent-memory.service";
|
|
||||||
import { UpsertAgentMemoryDto } from "./dto";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
|
|
||||||
import { Workspace, Permission, RequirePermission } from "../common/decorators";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Controller for per-agent key/value memory endpoints.
|
|
||||||
* All endpoints require authentication and workspace context.
|
|
||||||
*
|
|
||||||
* Guards are applied in order:
|
|
||||||
* 1. AuthGuard - Verifies user authentication
|
|
||||||
* 2. WorkspaceGuard - Validates workspace access
|
|
||||||
* 3. PermissionGuard - Checks role-based permissions
|
|
||||||
*/
|
|
||||||
@Controller("agents/:agentId/memory")
|
|
||||||
@UseGuards(AuthGuard, WorkspaceGuard, PermissionGuard)
|
|
||||||
export class AgentMemoryController {
|
|
||||||
constructor(private readonly agentMemoryService: AgentMemoryService) {}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* PUT /api/agents/:agentId/memory/:key
|
|
||||||
* Upsert a memory entry for an agent
|
|
||||||
* Requires: MEMBER role or higher
|
|
||||||
*/
|
|
||||||
@Put(":key")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_MEMBER)
|
|
||||||
async upsert(
|
|
||||||
@Param("agentId") agentId: string,
|
|
||||||
@Param("key") key: string,
|
|
||||||
@Body() dto: UpsertAgentMemoryDto,
|
|
||||||
@Workspace() workspaceId: string
|
|
||||||
) {
|
|
||||||
return this.agentMemoryService.upsert(workspaceId, agentId, key, dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* GET /api/agents/:agentId/memory
|
|
||||||
* List all memory entries for an agent
|
|
||||||
* Requires: Any workspace member (including GUEST)
|
|
||||||
*/
|
|
||||||
@Get()
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async findAll(@Param("agentId") agentId: string, @Workspace() workspaceId: string) {
|
|
||||||
return this.agentMemoryService.findAll(workspaceId, agentId);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* GET /api/agents/:agentId/memory/:key
|
|
||||||
* Get a single memory entry by key
|
|
||||||
* Requires: Any workspace member (including GUEST)
|
|
||||||
*/
|
|
||||||
@Get(":key")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async findOne(
|
|
||||||
@Param("agentId") agentId: string,
|
|
||||||
@Param("key") key: string,
|
|
||||||
@Workspace() workspaceId: string
|
|
||||||
) {
|
|
||||||
return this.agentMemoryService.findOne(workspaceId, agentId, key);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DELETE /api/agents/:agentId/memory/:key
|
|
||||||
* Remove a memory entry
|
|
||||||
* Requires: MEMBER role or higher
|
|
||||||
*/
|
|
||||||
@Delete(":key")
|
|
||||||
@HttpCode(HttpStatus.OK)
|
|
||||||
@RequirePermission(Permission.WORKSPACE_MEMBER)
|
|
||||||
async remove(
|
|
||||||
@Param("agentId") agentId: string,
|
|
||||||
@Param("key") key: string,
|
|
||||||
@Workspace() workspaceId: string
|
|
||||||
) {
|
|
||||||
return this.agentMemoryService.remove(workspaceId, agentId, key);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,198 +0,0 @@
|
|||||||
import { beforeAll, beforeEach, describe, expect, it, afterAll } from "vitest";
|
|
||||||
import { randomUUID as uuid } from "crypto";
|
|
||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { NotFoundException } from "@nestjs/common";
|
|
||||||
import { PrismaClient } from "@prisma/client";
|
|
||||||
import { AgentMemoryService } from "./agent-memory.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
|
|
||||||
const shouldRunDbIntegrationTests =
|
|
||||||
process.env.RUN_DB_TESTS === "true" && Boolean(process.env.DATABASE_URL);
|
|
||||||
const describeFn = shouldRunDbIntegrationTests ? describe : describe.skip;
|
|
||||||
|
|
||||||
async function createWorkspace(
|
|
||||||
prisma: PrismaClient,
|
|
||||||
label: string
|
|
||||||
): Promise<{ workspaceId: string; ownerId: string }> {
|
|
||||||
const workspace = await prisma.workspace.create({
|
|
||||||
data: {
|
|
||||||
name: `${label} ${Date.now()}`,
|
|
||||||
owner: {
|
|
||||||
create: {
|
|
||||||
email: `${label.toLowerCase().replace(/\s+/g, "-")}-${Date.now()}@example.com`,
|
|
||||||
name: `${label} Owner`,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
return {
|
|
||||||
workspaceId: workspace.id,
|
|
||||||
ownerId: workspace.ownerId,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
describeFn("AgentMemoryService Integration", () => {
|
|
||||||
let moduleRef: TestingModule;
|
|
||||||
let prisma: PrismaClient;
|
|
||||||
let service: AgentMemoryService;
|
|
||||||
let setupComplete = false;
|
|
||||||
|
|
||||||
let workspaceAId: string;
|
|
||||||
let workspaceAOwnerId: string;
|
|
||||||
let workspaceBId: string;
|
|
||||||
let workspaceBOwnerId: string;
|
|
||||||
|
|
||||||
beforeAll(async () => {
|
|
||||||
prisma = new PrismaClient();
|
|
||||||
await prisma.$connect();
|
|
||||||
|
|
||||||
const workspaceA = await createWorkspace(prisma, "Agent Memory Integration A");
|
|
||||||
workspaceAId = workspaceA.workspaceId;
|
|
||||||
workspaceAOwnerId = workspaceA.ownerId;
|
|
||||||
|
|
||||||
const workspaceB = await createWorkspace(prisma, "Agent Memory Integration B");
|
|
||||||
workspaceBId = workspaceB.workspaceId;
|
|
||||||
workspaceBOwnerId = workspaceB.ownerId;
|
|
||||||
|
|
||||||
moduleRef = await Test.createTestingModule({
|
|
||||||
providers: [
|
|
||||||
AgentMemoryService,
|
|
||||||
{
|
|
||||||
provide: PrismaService,
|
|
||||||
useValue: prisma,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}).compile();
|
|
||||||
|
|
||||||
service = moduleRef.get<AgentMemoryService>(AgentMemoryService);
|
|
||||||
setupComplete = true;
|
|
||||||
});
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
await prisma.agentMemory.deleteMany({
|
|
||||||
where: {
|
|
||||||
workspaceId: {
|
|
||||||
in: [workspaceAId, workspaceBId],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
afterAll(async () => {
|
|
||||||
if (!prisma) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const workspaceIds = [workspaceAId, workspaceBId].filter(
|
|
||||||
(id): id is string => typeof id === "string"
|
|
||||||
);
|
|
||||||
const ownerIds = [workspaceAOwnerId, workspaceBOwnerId].filter(
|
|
||||||
(id): id is string => typeof id === "string"
|
|
||||||
);
|
|
||||||
|
|
||||||
if (workspaceIds.length > 0) {
|
|
||||||
await prisma.agentMemory.deleteMany({
|
|
||||||
where: {
|
|
||||||
workspaceId: {
|
|
||||||
in: workspaceIds,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
await prisma.workspace.deleteMany({ where: { id: { in: workspaceIds } } });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (ownerIds.length > 0) {
|
|
||||||
await prisma.user.deleteMany({ where: { id: { in: ownerIds } } });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (moduleRef) {
|
|
||||||
await moduleRef.close();
|
|
||||||
}
|
|
||||||
await prisma.$disconnect();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("upserts and lists memory entries", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const agentId = `agent-${uuid()}`;
|
|
||||||
|
|
||||||
const entry = await service.upsert(workspaceAId, agentId, "session-context", {
|
|
||||||
value: { intent: "create-tests", depth: "integration" },
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(entry.workspaceId).toBe(workspaceAId);
|
|
||||||
expect(entry.agentId).toBe(agentId);
|
|
||||||
expect(entry.key).toBe("session-context");
|
|
||||||
|
|
||||||
const listed = await service.findAll(workspaceAId, agentId);
|
|
||||||
|
|
||||||
expect(listed).toHaveLength(1);
|
|
||||||
expect(listed[0]?.id).toBe(entry.id);
|
|
||||||
expect(listed[0]?.value).toMatchObject({ intent: "create-tests" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("updates existing key via upsert without creating duplicates", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const agentId = `agent-${uuid()}`;
|
|
||||||
|
|
||||||
const first = await service.upsert(workspaceAId, agentId, "preferences", {
|
|
||||||
value: { model: "fast" },
|
|
||||||
});
|
|
||||||
|
|
||||||
const second = await service.upsert(workspaceAId, agentId, "preferences", {
|
|
||||||
value: { model: "accurate" },
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(second.id).toBe(first.id);
|
|
||||||
expect(second.value).toMatchObject({ model: "accurate" });
|
|
||||||
|
|
||||||
const rowCount = await prisma.agentMemory.count({
|
|
||||||
where: {
|
|
||||||
workspaceId: workspaceAId,
|
|
||||||
agentId,
|
|
||||||
key: "preferences",
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(rowCount).toBe(1);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("lists keys in sorted order and isolates by workspace", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const agentId = `agent-${uuid()}`;
|
|
||||||
|
|
||||||
await service.upsert(workspaceAId, agentId, "beta", { value: { v: 2 } });
|
|
||||||
await service.upsert(workspaceAId, agentId, "alpha", { value: { v: 1 } });
|
|
||||||
await service.upsert(workspaceBId, agentId, "alpha", { value: { v: 99 } });
|
|
||||||
|
|
||||||
const workspaceAEntries = await service.findAll(workspaceAId, agentId);
|
|
||||||
const workspaceBEntries = await service.findAll(workspaceBId, agentId);
|
|
||||||
|
|
||||||
expect(workspaceAEntries.map((row) => row.key)).toEqual(["alpha", "beta"]);
|
|
||||||
expect(workspaceBEntries).toHaveLength(1);
|
|
||||||
expect(workspaceBEntries[0]?.value).toMatchObject({ v: 99 });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("throws NotFoundException when requesting unknown key", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
await expect(service.findOne(workspaceAId, `agent-${uuid()}`, "missing")).rejects.toThrow(
|
|
||||||
NotFoundException
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,13 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { AgentMemoryController } from "./agent-memory.controller";
|
|
||||||
import { AgentMemoryService } from "./agent-memory.service";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { AuthModule } from "../auth/auth.module";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [PrismaModule, AuthModule],
|
|
||||||
controllers: [AgentMemoryController],
|
|
||||||
providers: [AgentMemoryService],
|
|
||||||
exports: [AgentMemoryService],
|
|
||||||
})
|
|
||||||
export class AgentMemoryModule {}
|
|
||||||
@@ -1,126 +0,0 @@
|
|||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { AgentMemoryService } from "./agent-memory.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { NotFoundException } from "@nestjs/common";
|
|
||||||
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
||||||
|
|
||||||
describe("AgentMemoryService", () => {
|
|
||||||
let service: AgentMemoryService;
|
|
||||||
|
|
||||||
const mockPrismaService = {
|
|
||||||
agentMemory: {
|
|
||||||
upsert: vi.fn(),
|
|
||||||
findMany: vi.fn(),
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
delete: vi.fn(),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
const module: TestingModule = await Test.createTestingModule({
|
|
||||||
providers: [
|
|
||||||
AgentMemoryService,
|
|
||||||
{
|
|
||||||
provide: PrismaService,
|
|
||||||
useValue: mockPrismaService,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}).compile();
|
|
||||||
|
|
||||||
service = module.get<AgentMemoryService>(AgentMemoryService);
|
|
||||||
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
const workspaceId = "workspace-1";
|
|
||||||
const agentId = "agent-1";
|
|
||||||
const key = "session-context";
|
|
||||||
|
|
||||||
describe("upsert", () => {
|
|
||||||
it("should upsert a memory entry", async () => {
|
|
||||||
const dto = { value: { data: "some context" } };
|
|
||||||
const mockEntry = {
|
|
||||||
id: "mem-1",
|
|
||||||
workspaceId,
|
|
||||||
agentId,
|
|
||||||
key,
|
|
||||||
value: dto.value,
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
|
|
||||||
mockPrismaService.agentMemory.upsert.mockResolvedValue(mockEntry);
|
|
||||||
|
|
||||||
const result = await service.upsert(workspaceId, agentId, key, dto);
|
|
||||||
|
|
||||||
expect(mockPrismaService.agentMemory.upsert).toHaveBeenCalledWith({
|
|
||||||
where: { workspaceId_agentId_key: { workspaceId, agentId, key } },
|
|
||||||
create: { workspaceId, agentId, key, value: dto.value },
|
|
||||||
update: { value: dto.value },
|
|
||||||
});
|
|
||||||
expect(result).toEqual(mockEntry);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findAll", () => {
|
|
||||||
it("should return all memory entries for an agent", async () => {
|
|
||||||
const mockEntries = [
|
|
||||||
{ id: "mem-1", key: "a", value: 1 },
|
|
||||||
{ id: "mem-2", key: "b", value: 2 },
|
|
||||||
];
|
|
||||||
|
|
||||||
mockPrismaService.agentMemory.findMany.mockResolvedValue(mockEntries);
|
|
||||||
|
|
||||||
const result = await service.findAll(workspaceId, agentId);
|
|
||||||
|
|
||||||
expect(mockPrismaService.agentMemory.findMany).toHaveBeenCalledWith({
|
|
||||||
where: { workspaceId, agentId },
|
|
||||||
orderBy: { key: "asc" },
|
|
||||||
});
|
|
||||||
expect(result).toEqual(mockEntries);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findOne", () => {
|
|
||||||
it("should return a memory entry by key", async () => {
|
|
||||||
const mockEntry = { id: "mem-1", workspaceId, agentId, key, value: "ctx" };
|
|
||||||
|
|
||||||
mockPrismaService.agentMemory.findUnique.mockResolvedValue(mockEntry);
|
|
||||||
|
|
||||||
const result = await service.findOne(workspaceId, agentId, key);
|
|
||||||
|
|
||||||
expect(mockPrismaService.agentMemory.findUnique).toHaveBeenCalledWith({
|
|
||||||
where: { workspaceId_agentId_key: { workspaceId, agentId, key } },
|
|
||||||
});
|
|
||||||
expect(result).toEqual(mockEntry);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should throw NotFoundException when key not found", async () => {
|
|
||||||
mockPrismaService.agentMemory.findUnique.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(service.findOne(workspaceId, agentId, key)).rejects.toThrow(NotFoundException);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("remove", () => {
|
|
||||||
it("should delete a memory entry", async () => {
|
|
||||||
const mockEntry = { id: "mem-1", workspaceId, agentId, key, value: "x" };
|
|
||||||
|
|
||||||
mockPrismaService.agentMemory.findUnique.mockResolvedValue(mockEntry);
|
|
||||||
mockPrismaService.agentMemory.delete.mockResolvedValue(mockEntry);
|
|
||||||
|
|
||||||
const result = await service.remove(workspaceId, agentId, key);
|
|
||||||
|
|
||||||
expect(mockPrismaService.agentMemory.delete).toHaveBeenCalledWith({
|
|
||||||
where: { workspaceId_agentId_key: { workspaceId, agentId, key } },
|
|
||||||
});
|
|
||||||
expect(result).toEqual({ message: "Memory entry deleted successfully" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should throw NotFoundException when key not found", async () => {
|
|
||||||
mockPrismaService.agentMemory.findUnique.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(service.remove(workspaceId, agentId, key)).rejects.toThrow(NotFoundException);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,79 +0,0 @@
|
|||||||
import { Injectable, NotFoundException } from "@nestjs/common";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { Prisma } from "@prisma/client";
|
|
||||||
import type { UpsertAgentMemoryDto } from "./dto";
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class AgentMemoryService {
|
|
||||||
constructor(private readonly prisma: PrismaService) {}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Upsert a memory entry for an agent.
|
|
||||||
*/
|
|
||||||
async upsert(workspaceId: string, agentId: string, key: string, dto: UpsertAgentMemoryDto) {
|
|
||||||
return this.prisma.agentMemory.upsert({
|
|
||||||
where: {
|
|
||||||
workspaceId_agentId_key: { workspaceId, agentId, key },
|
|
||||||
},
|
|
||||||
create: {
|
|
||||||
workspaceId,
|
|
||||||
agentId,
|
|
||||||
key,
|
|
||||||
value: dto.value as Prisma.InputJsonValue,
|
|
||||||
},
|
|
||||||
update: {
|
|
||||||
value: dto.value as Prisma.InputJsonValue,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* List all memory entries for an agent in a workspace.
|
|
||||||
*/
|
|
||||||
async findAll(workspaceId: string, agentId: string) {
|
|
||||||
return this.prisma.agentMemory.findMany({
|
|
||||||
where: { workspaceId, agentId },
|
|
||||||
orderBy: { key: "asc" },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Get a single memory entry by key.
|
|
||||||
*/
|
|
||||||
async findOne(workspaceId: string, agentId: string, key: string) {
|
|
||||||
const entry = await this.prisma.agentMemory.findUnique({
|
|
||||||
where: {
|
|
||||||
workspaceId_agentId_key: { workspaceId, agentId, key },
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!entry) {
|
|
||||||
throw new NotFoundException(`Memory key "${key}" not found for agent "${agentId}"`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return entry;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Delete a memory entry by key.
|
|
||||||
*/
|
|
||||||
async remove(workspaceId: string, agentId: string, key: string) {
|
|
||||||
const entry = await this.prisma.agentMemory.findUnique({
|
|
||||||
where: {
|
|
||||||
workspaceId_agentId_key: { workspaceId, agentId, key },
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!entry) {
|
|
||||||
throw new NotFoundException(`Memory key "${key}" not found for agent "${agentId}"`);
|
|
||||||
}
|
|
||||||
|
|
||||||
await this.prisma.agentMemory.delete({
|
|
||||||
where: {
|
|
||||||
workspaceId_agentId_key: { workspaceId, agentId, key },
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
return { message: "Memory entry deleted successfully" };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
export * from "./upsert-agent-memory.dto";
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
import { IsNotEmpty } from "class-validator";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DTO for upserting an agent memory entry.
|
|
||||||
* The value accepts any JSON-serializable data.
|
|
||||||
*/
|
|
||||||
export class UpsertAgentMemoryDto {
|
|
||||||
@IsNotEmpty({ message: "value must not be empty" })
|
|
||||||
value!: unknown;
|
|
||||||
}
|
|
||||||
@@ -2,7 +2,6 @@ import { Module } from "@nestjs/common";
|
|||||||
import { APP_INTERCEPTOR, APP_GUARD } from "@nestjs/core";
|
import { APP_INTERCEPTOR, APP_GUARD } from "@nestjs/core";
|
||||||
import { ThrottlerModule } from "@nestjs/throttler";
|
import { ThrottlerModule } from "@nestjs/throttler";
|
||||||
import { BullModule } from "@nestjs/bullmq";
|
import { BullModule } from "@nestjs/bullmq";
|
||||||
import { ScheduleModule } from "@nestjs/schedule";
|
|
||||||
import { ThrottlerValkeyStorageService, ThrottlerApiKeyGuard } from "./common/throttler";
|
import { ThrottlerValkeyStorageService, ThrottlerApiKeyGuard } from "./common/throttler";
|
||||||
import { CsrfGuard } from "./common/guards/csrf.guard";
|
import { CsrfGuard } from "./common/guards/csrf.guard";
|
||||||
import { CsrfService } from "./common/services/csrf.service";
|
import { CsrfService } from "./common/services/csrf.service";
|
||||||
@@ -28,8 +27,6 @@ import { LlmUsageModule } from "./llm-usage/llm-usage.module";
|
|||||||
import { BrainModule } from "./brain/brain.module";
|
import { BrainModule } from "./brain/brain.module";
|
||||||
import { CronModule } from "./cron/cron.module";
|
import { CronModule } from "./cron/cron.module";
|
||||||
import { AgentTasksModule } from "./agent-tasks/agent-tasks.module";
|
import { AgentTasksModule } from "./agent-tasks/agent-tasks.module";
|
||||||
import { FindingsModule } from "./findings/findings.module";
|
|
||||||
import { AgentMemoryModule } from "./agent-memory/agent-memory.module";
|
|
||||||
import { ValkeyModule } from "./valkey/valkey.module";
|
import { ValkeyModule } from "./valkey/valkey.module";
|
||||||
import { BullMqModule } from "./bullmq/bullmq.module";
|
import { BullMqModule } from "./bullmq/bullmq.module";
|
||||||
import { StitcherModule } from "./stitcher/stitcher.module";
|
import { StitcherModule } from "./stitcher/stitcher.module";
|
||||||
@@ -40,7 +37,6 @@ import { JobStepsModule } from "./job-steps/job-steps.module";
|
|||||||
import { CoordinatorIntegrationModule } from "./coordinator-integration/coordinator-integration.module";
|
import { CoordinatorIntegrationModule } from "./coordinator-integration/coordinator-integration.module";
|
||||||
import { FederationModule } from "./federation/federation.module";
|
import { FederationModule } from "./federation/federation.module";
|
||||||
import { CredentialsModule } from "./credentials/credentials.module";
|
import { CredentialsModule } from "./credentials/credentials.module";
|
||||||
import { CryptoModule } from "./crypto/crypto.module";
|
|
||||||
import { MosaicTelemetryModule } from "./mosaic-telemetry";
|
import { MosaicTelemetryModule } from "./mosaic-telemetry";
|
||||||
import { SpeechModule } from "./speech/speech.module";
|
import { SpeechModule } from "./speech/speech.module";
|
||||||
import { DashboardModule } from "./dashboard/dashboard.module";
|
import { DashboardModule } from "./dashboard/dashboard.module";
|
||||||
@@ -50,15 +46,7 @@ import { WorkspacesModule } from "./workspaces/workspaces.module";
|
|||||||
import { AdminModule } from "./admin/admin.module";
|
import { AdminModule } from "./admin/admin.module";
|
||||||
import { TeamsModule } from "./teams/teams.module";
|
import { TeamsModule } from "./teams/teams.module";
|
||||||
import { ImportModule } from "./import/import.module";
|
import { ImportModule } from "./import/import.module";
|
||||||
import { ConversationArchiveModule } from "./conversation-archive/conversation-archive.module";
|
|
||||||
import { RlsContextInterceptor } from "./common/interceptors/rls-context.interceptor";
|
import { RlsContextInterceptor } from "./common/interceptors/rls-context.interceptor";
|
||||||
import { AgentConfigModule } from "./agent-config/agent-config.module";
|
|
||||||
import { ContainerLifecycleModule } from "./container-lifecycle/container-lifecycle.module";
|
|
||||||
import { ContainerReaperModule } from "./container-reaper/container-reaper.module";
|
|
||||||
import { FleetSettingsModule } from "./fleet-settings/fleet-settings.module";
|
|
||||||
import { OnboardingModule } from "./onboarding/onboarding.module";
|
|
||||||
import { ChatProxyModule } from "./chat-proxy/chat-proxy.module";
|
|
||||||
import { OrchestratorModule } from "./orchestrator/orchestrator.module";
|
|
||||||
|
|
||||||
@Module({
|
@Module({
|
||||||
imports: [
|
imports: [
|
||||||
@@ -89,7 +77,6 @@ import { OrchestratorModule } from "./orchestrator/orchestrator.module";
|
|||||||
};
|
};
|
||||||
})(),
|
})(),
|
||||||
}),
|
}),
|
||||||
ScheduleModule.forRoot(),
|
|
||||||
TelemetryModule,
|
TelemetryModule,
|
||||||
PrismaModule,
|
PrismaModule,
|
||||||
DatabaseModule,
|
DatabaseModule,
|
||||||
@@ -113,15 +100,12 @@ import { OrchestratorModule } from "./orchestrator/orchestrator.module";
|
|||||||
BrainModule,
|
BrainModule,
|
||||||
CronModule,
|
CronModule,
|
||||||
AgentTasksModule,
|
AgentTasksModule,
|
||||||
FindingsModule,
|
|
||||||
AgentMemoryModule,
|
|
||||||
RunnerJobsModule,
|
RunnerJobsModule,
|
||||||
JobEventsModule,
|
JobEventsModule,
|
||||||
JobStepsModule,
|
JobStepsModule,
|
||||||
CoordinatorIntegrationModule,
|
CoordinatorIntegrationModule,
|
||||||
FederationModule,
|
FederationModule,
|
||||||
CredentialsModule,
|
CredentialsModule,
|
||||||
CryptoModule,
|
|
||||||
MosaicTelemetryModule,
|
MosaicTelemetryModule,
|
||||||
SpeechModule,
|
SpeechModule,
|
||||||
DashboardModule,
|
DashboardModule,
|
||||||
@@ -131,14 +115,6 @@ import { OrchestratorModule } from "./orchestrator/orchestrator.module";
|
|||||||
AdminModule,
|
AdminModule,
|
||||||
TeamsModule,
|
TeamsModule,
|
||||||
ImportModule,
|
ImportModule,
|
||||||
ConversationArchiveModule,
|
|
||||||
AgentConfigModule,
|
|
||||||
ContainerLifecycleModule,
|
|
||||||
ContainerReaperModule,
|
|
||||||
FleetSettingsModule,
|
|
||||||
OnboardingModule,
|
|
||||||
ChatProxyModule,
|
|
||||||
OrchestratorModule,
|
|
||||||
],
|
],
|
||||||
controllers: [AppController, CsrfController],
|
controllers: [AppController, CsrfController],
|
||||||
providers: [
|
providers: [
|
||||||
|
|||||||
@@ -1,102 +0,0 @@
|
|||||||
import {
|
|
||||||
Body,
|
|
||||||
Controller,
|
|
||||||
HttpException,
|
|
||||||
Logger,
|
|
||||||
Post,
|
|
||||||
Req,
|
|
||||||
Res,
|
|
||||||
UnauthorizedException,
|
|
||||||
UseGuards,
|
|
||||||
} from "@nestjs/common";
|
|
||||||
import type { Response } from "express";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import type { MaybeAuthenticatedRequest } from "../auth/types/better-auth-request.interface";
|
|
||||||
import { ChatStreamDto } from "./chat-proxy.dto";
|
|
||||||
import { ChatProxyService } from "./chat-proxy.service";
|
|
||||||
|
|
||||||
@Controller("chat")
|
|
||||||
@UseGuards(AuthGuard)
|
|
||||||
export class ChatProxyController {
|
|
||||||
private readonly logger = new Logger(ChatProxyController.name);
|
|
||||||
|
|
||||||
constructor(private readonly chatProxyService: ChatProxyService) {}
|
|
||||||
|
|
||||||
// POST /api/chat/stream
|
|
||||||
// Request: { messages: Array<{role, content}> }
|
|
||||||
// Response: SSE stream of chat completion events
|
|
||||||
@Post("stream")
|
|
||||||
async streamChat(
|
|
||||||
@Body() body: ChatStreamDto,
|
|
||||||
@Req() req: MaybeAuthenticatedRequest,
|
|
||||||
@Res() res: Response
|
|
||||||
): Promise<void> {
|
|
||||||
const userId = req.user?.id;
|
|
||||||
if (!userId) {
|
|
||||||
throw new UnauthorizedException("No authenticated user found on request");
|
|
||||||
}
|
|
||||||
|
|
||||||
const abortController = new AbortController();
|
|
||||||
req.once("close", () => {
|
|
||||||
abortController.abort();
|
|
||||||
});
|
|
||||||
|
|
||||||
res.setHeader("Content-Type", "text/event-stream");
|
|
||||||
res.setHeader("Cache-Control", "no-cache");
|
|
||||||
res.setHeader("Connection", "keep-alive");
|
|
||||||
res.setHeader("X-Accel-Buffering", "no");
|
|
||||||
|
|
||||||
try {
|
|
||||||
const upstreamResponse = await this.chatProxyService.proxyChat(
|
|
||||||
userId,
|
|
||||||
body.messages,
|
|
||||||
abortController.signal
|
|
||||||
);
|
|
||||||
|
|
||||||
const upstreamContentType = upstreamResponse.headers.get("content-type");
|
|
||||||
if (upstreamContentType) {
|
|
||||||
res.setHeader("Content-Type", upstreamContentType);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!upstreamResponse.body) {
|
|
||||||
throw new Error("OpenClaw response did not include a stream body");
|
|
||||||
}
|
|
||||||
|
|
||||||
for await (const chunk of upstreamResponse.body as unknown as AsyncIterable<Uint8Array>) {
|
|
||||||
if (res.writableEnded || res.destroyed) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
res.write(Buffer.from(chunk));
|
|
||||||
}
|
|
||||||
} catch (error: unknown) {
|
|
||||||
this.logStreamError(error);
|
|
||||||
|
|
||||||
if (!res.writableEnded && !res.destroyed) {
|
|
||||||
res.write("event: error\n");
|
|
||||||
res.write(`data: ${JSON.stringify({ error: this.toSafeClientMessage(error) })}\n\n`);
|
|
||||||
}
|
|
||||||
} finally {
|
|
||||||
if (!res.writableEnded && !res.destroyed) {
|
|
||||||
res.end();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private toSafeClientMessage(error: unknown): string {
|
|
||||||
if (error instanceof HttpException && error.getStatus() < 500) {
|
|
||||||
return "Chat request was rejected";
|
|
||||||
}
|
|
||||||
|
|
||||||
return "Chat stream failed";
|
|
||||||
}
|
|
||||||
|
|
||||||
private logStreamError(error: unknown): void {
|
|
||||||
if (error instanceof Error) {
|
|
||||||
this.logger.warn(`Chat stream failed: ${error.message}`);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
this.logger.warn(`Chat stream failed: ${String(error)}`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,25 +0,0 @@
|
|||||||
import { Type } from "class-transformer";
|
|
||||||
import { ArrayMinSize, IsArray, IsNotEmpty, IsString, ValidateNested } from "class-validator";
|
|
||||||
|
|
||||||
export interface ChatMessage {
|
|
||||||
role: string;
|
|
||||||
content: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export class ChatMessageDto implements ChatMessage {
|
|
||||||
@IsString({ message: "role must be a string" })
|
|
||||||
@IsNotEmpty({ message: "role is required" })
|
|
||||||
role!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "content must be a string" })
|
|
||||||
@IsNotEmpty({ message: "content is required" })
|
|
||||||
content!: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export class ChatStreamDto {
|
|
||||||
@IsArray({ message: "messages must be an array" })
|
|
||||||
@ArrayMinSize(1, { message: "messages must contain at least one message" })
|
|
||||||
@ValidateNested({ each: true })
|
|
||||||
@Type(() => ChatMessageDto)
|
|
||||||
messages!: ChatMessageDto[];
|
|
||||||
}
|
|
||||||
@@ -1,15 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { AuthModule } from "../auth/auth.module";
|
|
||||||
import { AgentConfigModule } from "../agent-config/agent-config.module";
|
|
||||||
import { ContainerLifecycleModule } from "../container-lifecycle/container-lifecycle.module";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { ChatProxyController } from "./chat-proxy.controller";
|
|
||||||
import { ChatProxyService } from "./chat-proxy.service";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [AuthModule, PrismaModule, ContainerLifecycleModule, AgentConfigModule],
|
|
||||||
controllers: [ChatProxyController],
|
|
||||||
providers: [ChatProxyService],
|
|
||||||
exports: [ChatProxyService],
|
|
||||||
})
|
|
||||||
export class ChatProxyModule {}
|
|
||||||
@@ -1,108 +0,0 @@
|
|||||||
import { ServiceUnavailableException } from "@nestjs/common";
|
|
||||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
import { ChatProxyService } from "./chat-proxy.service";
|
|
||||||
|
|
||||||
describe("ChatProxyService", () => {
|
|
||||||
const userId = "user-123";
|
|
||||||
|
|
||||||
const prisma = {
|
|
||||||
userAgentConfig: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const containerLifecycle = {
|
|
||||||
ensureRunning: vi.fn(),
|
|
||||||
touch: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
let service: ChatProxyService;
|
|
||||||
let fetchMock: ReturnType<typeof vi.fn>;
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
fetchMock = vi.fn();
|
|
||||||
vi.stubGlobal("fetch", fetchMock);
|
|
||||||
service = new ChatProxyService(prisma as never, containerLifecycle as never);
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(() => {
|
|
||||||
vi.unstubAllGlobals();
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("getContainerUrl", () => {
|
|
||||||
it("calls ensureRunning and touch for the user", async () => {
|
|
||||||
containerLifecycle.ensureRunning.mockResolvedValue({
|
|
||||||
url: "http://mosaic-user-user-123:19000",
|
|
||||||
token: "gateway-token",
|
|
||||||
});
|
|
||||||
containerLifecycle.touch.mockResolvedValue(undefined);
|
|
||||||
|
|
||||||
const url = await service.getContainerUrl(userId);
|
|
||||||
|
|
||||||
expect(url).toBe("http://mosaic-user-user-123:19000");
|
|
||||||
expect(containerLifecycle.ensureRunning).toHaveBeenCalledWith(userId);
|
|
||||||
expect(containerLifecycle.touch).toHaveBeenCalledWith(userId);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("proxyChat", () => {
|
|
||||||
it("forwards the request to the user's OpenClaw container", async () => {
|
|
||||||
containerLifecycle.ensureRunning.mockResolvedValue({
|
|
||||||
url: "http://mosaic-user-user-123:19000",
|
|
||||||
token: "gateway-token",
|
|
||||||
});
|
|
||||||
containerLifecycle.touch.mockResolvedValue(undefined);
|
|
||||||
fetchMock.mockResolvedValue(new Response("event: token\ndata: hello\n\n"));
|
|
||||||
|
|
||||||
const messages = [{ role: "user", content: "Hello from Mosaic" }];
|
|
||||||
const response = await service.proxyChat(userId, messages);
|
|
||||||
|
|
||||||
expect(response).toBeInstanceOf(Response);
|
|
||||||
expect(fetchMock).toHaveBeenCalledWith(
|
|
||||||
"http://mosaic-user-user-123:19000/v1/chat/completions",
|
|
||||||
expect.objectContaining({
|
|
||||||
method: "POST",
|
|
||||||
headers: {
|
|
||||||
Authorization: "Bearer gateway-token",
|
|
||||||
"Content-Type": "application/json",
|
|
||||||
},
|
|
||||||
})
|
|
||||||
);
|
|
||||||
|
|
||||||
const [, request] = fetchMock.mock.calls[0] as [string, RequestInit];
|
|
||||||
const parsedBody = JSON.parse(String(request.body));
|
|
||||||
expect(parsedBody).toEqual({
|
|
||||||
messages,
|
|
||||||
model: "openclaw:default",
|
|
||||||
stream: true,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("throws ServiceUnavailableException on connection refused errors", async () => {
|
|
||||||
containerLifecycle.ensureRunning.mockResolvedValue({
|
|
||||||
url: "http://mosaic-user-user-123:19000",
|
|
||||||
token: "gateway-token",
|
|
||||||
});
|
|
||||||
containerLifecycle.touch.mockResolvedValue(undefined);
|
|
||||||
fetchMock.mockRejectedValue(new Error("connect ECONNREFUSED 127.0.0.1:19000"));
|
|
||||||
|
|
||||||
await expect(service.proxyChat(userId, [])).rejects.toBeInstanceOf(
|
|
||||||
ServiceUnavailableException
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("throws ServiceUnavailableException on timeout errors", async () => {
|
|
||||||
containerLifecycle.ensureRunning.mockResolvedValue({
|
|
||||||
url: "http://mosaic-user-user-123:19000",
|
|
||||||
token: "gateway-token",
|
|
||||||
});
|
|
||||||
containerLifecycle.touch.mockResolvedValue(undefined);
|
|
||||||
fetchMock.mockRejectedValue(new Error("The operation was aborted due to timeout"));
|
|
||||||
|
|
||||||
await expect(service.proxyChat(userId, [])).rejects.toBeInstanceOf(
|
|
||||||
ServiceUnavailableException
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,110 +0,0 @@
|
|||||||
import {
|
|
||||||
BadGatewayException,
|
|
||||||
Injectable,
|
|
||||||
Logger,
|
|
||||||
ServiceUnavailableException,
|
|
||||||
} from "@nestjs/common";
|
|
||||||
import { ContainerLifecycleService } from "../container-lifecycle/container-lifecycle.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import type { ChatMessage } from "./chat-proxy.dto";
|
|
||||||
|
|
||||||
const DEFAULT_OPENCLAW_MODEL = "openclaw:default";
|
|
||||||
|
|
||||||
interface ContainerConnection {
|
|
||||||
url: string;
|
|
||||||
token: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class ChatProxyService {
|
|
||||||
private readonly logger = new Logger(ChatProxyService.name);
|
|
||||||
|
|
||||||
constructor(
|
|
||||||
private readonly prisma: PrismaService,
|
|
||||||
private readonly containerLifecycle: ContainerLifecycleService
|
|
||||||
) {}
|
|
||||||
|
|
||||||
// Get the user's OpenClaw container URL and mark it active.
|
|
||||||
async getContainerUrl(userId: string): Promise<string> {
|
|
||||||
const { url } = await this.getContainerConnection(userId);
|
|
||||||
return url;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Proxy chat request to OpenClaw.
|
|
||||||
async proxyChat(
|
|
||||||
userId: string,
|
|
||||||
messages: ChatMessage[],
|
|
||||||
signal?: AbortSignal
|
|
||||||
): Promise<Response> {
|
|
||||||
const { url: containerUrl, token: gatewayToken } = await this.getContainerConnection(userId);
|
|
||||||
const model = await this.getPreferredModel(userId);
|
|
||||||
const requestInit: RequestInit = {
|
|
||||||
method: "POST",
|
|
||||||
headers: {
|
|
||||||
"Content-Type": "application/json",
|
|
||||||
Authorization: `Bearer ${gatewayToken}`,
|
|
||||||
},
|
|
||||||
body: JSON.stringify({
|
|
||||||
messages,
|
|
||||||
model,
|
|
||||||
stream: true,
|
|
||||||
}),
|
|
||||||
};
|
|
||||||
|
|
||||||
if (signal) {
|
|
||||||
requestInit.signal = signal;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const response = await fetch(`${containerUrl}/v1/chat/completions`, requestInit);
|
|
||||||
|
|
||||||
if (!response.ok) {
|
|
||||||
const detail = await this.readResponseText(response);
|
|
||||||
const status = `${String(response.status)} ${response.statusText}`.trim();
|
|
||||||
this.logger.warn(
|
|
||||||
detail ? `OpenClaw returned ${status}: ${detail}` : `OpenClaw returned ${status}`
|
|
||||||
);
|
|
||||||
throw new BadGatewayException(`OpenClaw returned ${status}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return response;
|
|
||||||
} catch (error: unknown) {
|
|
||||||
if (error instanceof BadGatewayException) {
|
|
||||||
throw error;
|
|
||||||
}
|
|
||||||
|
|
||||||
const message = error instanceof Error ? error.message : String(error);
|
|
||||||
this.logger.warn(`Failed to proxy chat request: ${message}`);
|
|
||||||
throw new ServiceUnavailableException("Failed to proxy chat to OpenClaw");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private async getContainerConnection(userId: string): Promise<ContainerConnection> {
|
|
||||||
const connection = await this.containerLifecycle.ensureRunning(userId);
|
|
||||||
await this.containerLifecycle.touch(userId);
|
|
||||||
return connection;
|
|
||||||
}
|
|
||||||
|
|
||||||
private async getPreferredModel(userId: string): Promise<string> {
|
|
||||||
const config = await this.prisma.userAgentConfig.findUnique({
|
|
||||||
where: { userId },
|
|
||||||
select: { primaryModel: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
const primaryModel = config?.primaryModel?.trim();
|
|
||||||
if (!primaryModel) {
|
|
||||||
return DEFAULT_OPENCLAW_MODEL;
|
|
||||||
}
|
|
||||||
|
|
||||||
return primaryModel;
|
|
||||||
}
|
|
||||||
|
|
||||||
private async readResponseText(response: Response): Promise<string | null> {
|
|
||||||
try {
|
|
||||||
const text = (await response.text()).trim();
|
|
||||||
return text.length > 0 ? text : null;
|
|
||||||
} catch {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -87,17 +87,6 @@ describe("CsrfGuard", () => {
|
|||||||
});
|
});
|
||||||
|
|
||||||
describe("State-changing methods requiring CSRF", () => {
|
describe("State-changing methods requiring CSRF", () => {
|
||||||
it("should allow POST with Bearer auth without CSRF token", () => {
|
|
||||||
const context = createContext(
|
|
||||||
"POST",
|
|
||||||
{},
|
|
||||||
{ authorization: "Bearer api-token" },
|
|
||||||
false,
|
|
||||||
"user-123"
|
|
||||||
);
|
|
||||||
expect(guard.canActivate(context)).toBe(true);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should reject POST without CSRF token", () => {
|
it("should reject POST without CSRF token", () => {
|
||||||
const context = createContext("POST", {}, {}, false, "user-123");
|
const context = createContext("POST", {}, {}, false, "user-123");
|
||||||
expect(() => guard.canActivate(context)).toThrow(ForbiddenException);
|
expect(() => guard.canActivate(context)).toThrow(ForbiddenException);
|
||||||
|
|||||||
@@ -57,11 +57,6 @@ export class CsrfGuard implements CanActivate {
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
const authHeader = request.headers.authorization;
|
|
||||||
if (typeof authHeader === "string" && authHeader.startsWith("Bearer ")) {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get CSRF token from cookie and header
|
// Get CSRF token from cookie and header
|
||||||
const cookies = request.cookies as Record<string, string> | undefined;
|
const cookies = request.cookies as Record<string, string> | undefined;
|
||||||
const cookieToken = cookies?.["csrf-token"];
|
const cookieToken = cookies?.["csrf-token"];
|
||||||
|
|||||||
@@ -1,12 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { ConfigModule } from "@nestjs/config";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { CryptoModule } from "../crypto/crypto.module";
|
|
||||||
import { ContainerLifecycleService } from "./container-lifecycle.service";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [ConfigModule, PrismaModule, CryptoModule],
|
|
||||||
providers: [ContainerLifecycleService],
|
|
||||||
exports: [ContainerLifecycleService],
|
|
||||||
})
|
|
||||||
export class ContainerLifecycleModule {}
|
|
||||||
@@ -1,593 +0,0 @@
|
|||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
import type { ConfigService } from "@nestjs/config";
|
|
||||||
import type { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import type { CryptoService } from "../crypto/crypto.service";
|
|
||||||
|
|
||||||
interface MockUserContainerRecord {
|
|
||||||
id: string;
|
|
||||||
userId: string;
|
|
||||||
containerId: string | null;
|
|
||||||
containerName: string;
|
|
||||||
gatewayPort: number | null;
|
|
||||||
gatewayToken: string;
|
|
||||||
status: string;
|
|
||||||
lastActiveAt: Date | null;
|
|
||||||
idleTimeoutMin: number;
|
|
||||||
config: Record<string, unknown>;
|
|
||||||
createdAt: Date;
|
|
||||||
updatedAt: Date;
|
|
||||||
}
|
|
||||||
|
|
||||||
const dockerMock = vi.hoisted(() => {
|
|
||||||
interface MockDockerContainerState {
|
|
||||||
id: string;
|
|
||||||
name: string;
|
|
||||||
running: boolean;
|
|
||||||
port: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
const containers = new Map<string, MockDockerContainerState>();
|
|
||||||
const handles = new Map<
|
|
||||||
string,
|
|
||||||
{
|
|
||||||
inspect: ReturnType<typeof vi.fn>;
|
|
||||||
start: ReturnType<typeof vi.fn>;
|
|
||||||
stop: ReturnType<typeof vi.fn>;
|
|
||||||
}
|
|
||||||
>();
|
|
||||||
|
|
||||||
const ensureHandle = (id: string) => {
|
|
||||||
const existing = handles.get(id);
|
|
||||||
if (existing) {
|
|
||||||
return existing;
|
|
||||||
}
|
|
||||||
|
|
||||||
const handle = {
|
|
||||||
inspect: vi.fn(async () => {
|
|
||||||
const container = containers.get(id);
|
|
||||||
if (!container) {
|
|
||||||
throw { statusCode: 404 };
|
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
|
||||||
Id: container.id,
|
|
||||||
State: {
|
|
||||||
Running: container.running,
|
|
||||||
},
|
|
||||||
NetworkSettings: {
|
|
||||||
Ports: {
|
|
||||||
"18789/tcp": [{ HostPort: String(container.port) }],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
};
|
|
||||||
}),
|
|
||||||
start: vi.fn(async () => {
|
|
||||||
const container = containers.get(id);
|
|
||||||
if (!container) {
|
|
||||||
throw { statusCode: 404 };
|
|
||||||
}
|
|
||||||
container.running = true;
|
|
||||||
}),
|
|
||||||
stop: vi.fn(async () => {
|
|
||||||
const container = containers.get(id);
|
|
||||||
if (!container) {
|
|
||||||
throw { statusCode: 404 };
|
|
||||||
}
|
|
||||||
container.running = false;
|
|
||||||
}),
|
|
||||||
};
|
|
||||||
|
|
||||||
handles.set(id, handle);
|
|
||||||
return handle;
|
|
||||||
};
|
|
||||||
|
|
||||||
const listContainers = vi.fn(
|
|
||||||
async (options?: { all?: boolean; filters?: { name?: string[] } }) => {
|
|
||||||
const nameFilter = options?.filters?.name?.[0];
|
|
||||||
return [...containers.values()]
|
|
||||||
.filter((container) => (nameFilter ? container.name.includes(nameFilter) : true))
|
|
||||||
.map((container) => ({
|
|
||||||
Id: container.id,
|
|
||||||
Names: [`/${container.name}`],
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
);
|
|
||||||
|
|
||||||
const getContainer = vi.fn((id: string) => ensureHandle(id));
|
|
||||||
|
|
||||||
const createContainer = vi.fn(
|
|
||||||
async (options: {
|
|
||||||
name?: string;
|
|
||||||
HostConfig?: { PortBindings?: Record<string, Array<{ HostPort?: string }>> };
|
|
||||||
}) => {
|
|
||||||
const id = `ctr-${containers.size + 1}`;
|
|
||||||
const name = options.name ?? id;
|
|
||||||
const hostPort = options.HostConfig?.PortBindings?.["18789/tcp"]?.[0]?.HostPort;
|
|
||||||
const port = hostPort ? Number.parseInt(hostPort, 10) : 0;
|
|
||||||
|
|
||||||
containers.set(id, {
|
|
||||||
id,
|
|
||||||
name,
|
|
||||||
running: false,
|
|
||||||
port,
|
|
||||||
});
|
|
||||||
|
|
||||||
return ensureHandle(id);
|
|
||||||
}
|
|
||||||
);
|
|
||||||
|
|
||||||
const dockerInstance = {
|
|
||||||
listContainers,
|
|
||||||
getContainer,
|
|
||||||
createContainer,
|
|
||||||
};
|
|
||||||
|
|
||||||
const constructorSpy = vi.fn();
|
|
||||||
class DockerConstructorMock {
|
|
||||||
constructor(options?: unknown) {
|
|
||||||
constructorSpy(options);
|
|
||||||
return dockerInstance;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const registerContainer = (container: MockDockerContainerState) => {
|
|
||||||
containers.set(container.id, { ...container });
|
|
||||||
ensureHandle(container.id);
|
|
||||||
};
|
|
||||||
|
|
||||||
const reset = () => {
|
|
||||||
containers.clear();
|
|
||||||
handles.clear();
|
|
||||||
constructorSpy.mockClear();
|
|
||||||
listContainers.mockClear();
|
|
||||||
getContainer.mockClear();
|
|
||||||
createContainer.mockClear();
|
|
||||||
};
|
|
||||||
|
|
||||||
return {
|
|
||||||
DockerConstructorMock,
|
|
||||||
constructorSpy,
|
|
||||||
createContainer,
|
|
||||||
handles,
|
|
||||||
registerContainer,
|
|
||||||
reset,
|
|
||||||
};
|
|
||||||
});
|
|
||||||
|
|
||||||
vi.mock("dockerode", () => ({
|
|
||||||
default: dockerMock.DockerConstructorMock,
|
|
||||||
}));
|
|
||||||
|
|
||||||
import { ContainerLifecycleService } from "./container-lifecycle.service";
|
|
||||||
|
|
||||||
function createConfigMock(values: Record<string, string> = {}) {
|
|
||||||
return {
|
|
||||||
get: vi.fn((key: string) => values[key]),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function createCryptoMock() {
|
|
||||||
return {
|
|
||||||
generateToken: vi.fn(() => "generated-token"),
|
|
||||||
encrypt: vi.fn((value: string) => `enc:${value}`),
|
|
||||||
decrypt: vi.fn((value: string) => value.replace(/^enc:/, "")),
|
|
||||||
isEncrypted: vi.fn((value: string) => value.startsWith("enc:")),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function projectRecord(
|
|
||||||
record: MockUserContainerRecord,
|
|
||||||
select?: Record<string, boolean>
|
|
||||||
): Partial<MockUserContainerRecord> {
|
|
||||||
if (!select) {
|
|
||||||
return { ...record };
|
|
||||||
}
|
|
||||||
|
|
||||||
const projection: Partial<MockUserContainerRecord> = {};
|
|
||||||
for (const [field, enabled] of Object.entries(select)) {
|
|
||||||
if (enabled) {
|
|
||||||
const key = field as keyof MockUserContainerRecord;
|
|
||||||
projection[key] = record[key];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return projection;
|
|
||||||
}
|
|
||||||
|
|
||||||
function createPrismaMock(initialRecords: MockUserContainerRecord[] = []) {
|
|
||||||
const records = new Map<string, MockUserContainerRecord>();
|
|
||||||
for (const record of initialRecords) {
|
|
||||||
records.set(record.userId, { ...record });
|
|
||||||
}
|
|
||||||
|
|
||||||
const userContainer = {
|
|
||||||
findUnique: vi.fn(
|
|
||||||
async (args: {
|
|
||||||
where: { userId?: string; id?: string };
|
|
||||||
select?: Record<string, boolean>;
|
|
||||||
}) => {
|
|
||||||
let record: MockUserContainerRecord | undefined;
|
|
||||||
if (args.where.userId) {
|
|
||||||
record = records.get(args.where.userId);
|
|
||||||
} else if (args.where.id) {
|
|
||||||
record = [...records.values()].find((entry) => entry.id === args.where.id);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!record) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
return projectRecord(record, args.select);
|
|
||||||
}
|
|
||||||
),
|
|
||||||
create: vi.fn(
|
|
||||||
async (args: {
|
|
||||||
data: Partial<MockUserContainerRecord> & {
|
|
||||||
userId: string;
|
|
||||||
containerName: string;
|
|
||||||
gatewayToken: string;
|
|
||||||
};
|
|
||||||
}) => {
|
|
||||||
const now = new Date();
|
|
||||||
const next: MockUserContainerRecord = {
|
|
||||||
id: args.data.id ?? `uc-${records.size + 1}`,
|
|
||||||
userId: args.data.userId,
|
|
||||||
containerId: args.data.containerId ?? null,
|
|
||||||
containerName: args.data.containerName,
|
|
||||||
gatewayPort: args.data.gatewayPort ?? null,
|
|
||||||
gatewayToken: args.data.gatewayToken,
|
|
||||||
status: args.data.status ?? "stopped",
|
|
||||||
lastActiveAt: args.data.lastActiveAt ?? null,
|
|
||||||
idleTimeoutMin: args.data.idleTimeoutMin ?? 30,
|
|
||||||
config: args.data.config ?? {},
|
|
||||||
createdAt: now,
|
|
||||||
updatedAt: now,
|
|
||||||
};
|
|
||||||
|
|
||||||
records.set(next.userId, next);
|
|
||||||
return { ...next };
|
|
||||||
}
|
|
||||||
),
|
|
||||||
update: vi.fn(
|
|
||||||
async (args: { where: { userId: string }; data: Partial<MockUserContainerRecord> }) => {
|
|
||||||
const record = records.get(args.where.userId);
|
|
||||||
if (!record) {
|
|
||||||
throw new Error(`Record ${args.where.userId} not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
const updated: MockUserContainerRecord = {
|
|
||||||
...record,
|
|
||||||
...args.data,
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
records.set(updated.userId, updated);
|
|
||||||
return { ...updated };
|
|
||||||
}
|
|
||||||
),
|
|
||||||
updateMany: vi.fn(
|
|
||||||
async (args: { where: { userId: string }; data: Partial<MockUserContainerRecord> }) => {
|
|
||||||
const record = records.get(args.where.userId);
|
|
||||||
if (!record) {
|
|
||||||
return { count: 0 };
|
|
||||||
}
|
|
||||||
|
|
||||||
const updated: MockUserContainerRecord = {
|
|
||||||
...record,
|
|
||||||
...args.data,
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
records.set(updated.userId, updated);
|
|
||||||
return { count: 1 };
|
|
||||||
}
|
|
||||||
),
|
|
||||||
findMany: vi.fn(
|
|
||||||
async (args?: {
|
|
||||||
where?: {
|
|
||||||
status?: string;
|
|
||||||
lastActiveAt?: { not: null };
|
|
||||||
gatewayPort?: { not: null };
|
|
||||||
};
|
|
||||||
select?: Record<string, boolean>;
|
|
||||||
}) => {
|
|
||||||
let rows = [...records.values()];
|
|
||||||
|
|
||||||
if (args?.where?.status) {
|
|
||||||
rows = rows.filter((record) => record.status === args.where?.status);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (args?.where?.lastActiveAt?.not === null) {
|
|
||||||
rows = rows.filter((record) => record.lastActiveAt !== null);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (args?.where?.gatewayPort?.not === null) {
|
|
||||||
rows = rows.filter((record) => record.gatewayPort !== null);
|
|
||||||
}
|
|
||||||
|
|
||||||
return rows.map((record) => projectRecord(record, args?.select));
|
|
||||||
}
|
|
||||||
),
|
|
||||||
};
|
|
||||||
|
|
||||||
return {
|
|
||||||
prisma: {
|
|
||||||
userContainer,
|
|
||||||
},
|
|
||||||
records,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function createRecord(overrides: Partial<MockUserContainerRecord>): MockUserContainerRecord {
|
|
||||||
const now = new Date();
|
|
||||||
return {
|
|
||||||
id: overrides.id ?? "uc-default",
|
|
||||||
userId: overrides.userId ?? "user-default",
|
|
||||||
containerId: overrides.containerId ?? null,
|
|
||||||
containerName: overrides.containerName ?? "mosaic-user-user-default",
|
|
||||||
gatewayPort: overrides.gatewayPort ?? null,
|
|
||||||
gatewayToken: overrides.gatewayToken ?? "enc:token-default",
|
|
||||||
status: overrides.status ?? "stopped",
|
|
||||||
lastActiveAt: overrides.lastActiveAt ?? null,
|
|
||||||
idleTimeoutMin: overrides.idleTimeoutMin ?? 30,
|
|
||||||
config: overrides.config ?? {},
|
|
||||||
createdAt: overrides.createdAt ?? now,
|
|
||||||
updatedAt: overrides.updatedAt ?? now,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
describe("ContainerLifecycleService", () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
dockerMock.reset();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("ensureRunning creates container when none exists", async () => {
|
|
||||||
const { prisma, records } = createPrismaMock();
|
|
||||||
const crypto = createCryptoMock();
|
|
||||||
const config = createConfigMock();
|
|
||||||
const service = new ContainerLifecycleService(
|
|
||||||
prisma as unknown as PrismaService,
|
|
||||||
crypto as unknown as CryptoService,
|
|
||||||
config as unknown as ConfigService
|
|
||||||
);
|
|
||||||
|
|
||||||
const result = await service.ensureRunning("user-1");
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
url: "http://mosaic-user-user-1:19000",
|
|
||||||
token: "generated-token",
|
|
||||||
});
|
|
||||||
|
|
||||||
const updatedRecord = records.get("user-1");
|
|
||||||
expect(updatedRecord?.status).toBe("running");
|
|
||||||
expect(updatedRecord?.containerId).toBe("ctr-1");
|
|
||||||
expect(updatedRecord?.gatewayPort).toBe(19000);
|
|
||||||
expect(updatedRecord?.gatewayToken).toBe("enc:generated-token");
|
|
||||||
|
|
||||||
expect(dockerMock.createContainer).toHaveBeenCalledTimes(1);
|
|
||||||
const [createCall] = dockerMock.createContainer.mock.calls[0] as [
|
|
||||||
{
|
|
||||||
name: string;
|
|
||||||
Image: string;
|
|
||||||
Env: string[];
|
|
||||||
HostConfig: { Binds: string[]; NetworkMode: string };
|
|
||||||
},
|
|
||||||
];
|
|
||||||
expect(createCall.name).toBe("mosaic-user-user-1");
|
|
||||||
expect(createCall.Image).toBe("alpine/openclaw:latest");
|
|
||||||
expect(createCall.HostConfig.Binds).toEqual(["mosaic-user-user-1-state:/home/node/.openclaw"]);
|
|
||||||
expect(createCall.HostConfig.NetworkMode).toBe("mosaic-internal");
|
|
||||||
expect(createCall.Env).toContain("AGENT_TOKEN=generated-token");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("ensureRunning starts existing stopped container", async () => {
|
|
||||||
const { prisma, records } = createPrismaMock([
|
|
||||||
createRecord({
|
|
||||||
id: "uc-1",
|
|
||||||
userId: "user-2",
|
|
||||||
containerId: "ctr-stopped",
|
|
||||||
containerName: "mosaic-user-user-2",
|
|
||||||
gatewayToken: "enc:existing-token",
|
|
||||||
status: "stopped",
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
const crypto = createCryptoMock();
|
|
||||||
const config = createConfigMock();
|
|
||||||
const service = new ContainerLifecycleService(
|
|
||||||
prisma as unknown as PrismaService,
|
|
||||||
crypto as unknown as CryptoService,
|
|
||||||
config as unknown as ConfigService
|
|
||||||
);
|
|
||||||
|
|
||||||
dockerMock.registerContainer({
|
|
||||||
id: "ctr-stopped",
|
|
||||||
name: "mosaic-user-user-2",
|
|
||||||
running: false,
|
|
||||||
port: 19042,
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await service.ensureRunning("user-2");
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
url: "http://mosaic-user-user-2:19042",
|
|
||||||
token: "existing-token",
|
|
||||||
});
|
|
||||||
|
|
||||||
const handle = dockerMock.handles.get("ctr-stopped");
|
|
||||||
expect(handle?.start).toHaveBeenCalledTimes(1);
|
|
||||||
expect(records.get("user-2")?.status).toBe("running");
|
|
||||||
expect(records.get("user-2")?.gatewayPort).toBe(19042);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("ensureRunning returns existing running container", async () => {
|
|
||||||
const { prisma } = createPrismaMock([
|
|
||||||
createRecord({
|
|
||||||
id: "uc-2",
|
|
||||||
userId: "user-3",
|
|
||||||
containerId: "ctr-running",
|
|
||||||
containerName: "mosaic-user-user-3",
|
|
||||||
gatewayPort: 19043,
|
|
||||||
gatewayToken: "enc:running-token",
|
|
||||||
status: "running",
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
const crypto = createCryptoMock();
|
|
||||||
const config = createConfigMock();
|
|
||||||
const service = new ContainerLifecycleService(
|
|
||||||
prisma as unknown as PrismaService,
|
|
||||||
crypto as unknown as CryptoService,
|
|
||||||
config as unknown as ConfigService
|
|
||||||
);
|
|
||||||
|
|
||||||
dockerMock.registerContainer({
|
|
||||||
id: "ctr-running",
|
|
||||||
name: "mosaic-user-user-3",
|
|
||||||
running: true,
|
|
||||||
port: 19043,
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await service.ensureRunning("user-3");
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
url: "http://mosaic-user-user-3:19043",
|
|
||||||
token: "running-token",
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(dockerMock.createContainer).not.toHaveBeenCalled();
|
|
||||||
const handle = dockerMock.handles.get("ctr-running");
|
|
||||||
expect(handle?.start).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("stop gracefully stops container and updates DB", async () => {
|
|
||||||
const { prisma, records } = createPrismaMock([
|
|
||||||
createRecord({
|
|
||||||
id: "uc-stop",
|
|
||||||
userId: "user-stop",
|
|
||||||
containerId: "ctr-stop",
|
|
||||||
containerName: "mosaic-user-user-stop",
|
|
||||||
gatewayPort: 19044,
|
|
||||||
status: "running",
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
const crypto = createCryptoMock();
|
|
||||||
const config = createConfigMock();
|
|
||||||
const service = new ContainerLifecycleService(
|
|
||||||
prisma as unknown as PrismaService,
|
|
||||||
crypto as unknown as CryptoService,
|
|
||||||
config as unknown as ConfigService
|
|
||||||
);
|
|
||||||
|
|
||||||
dockerMock.registerContainer({
|
|
||||||
id: "ctr-stop",
|
|
||||||
name: "mosaic-user-user-stop",
|
|
||||||
running: true,
|
|
||||||
port: 19044,
|
|
||||||
});
|
|
||||||
|
|
||||||
await service.stop("user-stop");
|
|
||||||
|
|
||||||
const handle = dockerMock.handles.get("ctr-stop");
|
|
||||||
expect(handle?.stop).toHaveBeenCalledWith({ t: 10 });
|
|
||||||
|
|
||||||
const updatedRecord = records.get("user-stop");
|
|
||||||
expect(updatedRecord?.status).toBe("stopped");
|
|
||||||
expect(updatedRecord?.containerId).toBeNull();
|
|
||||||
expect(updatedRecord?.gatewayPort).toBeNull();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("reapIdle stops only containers past their idle timeout", async () => {
|
|
||||||
const now = Date.now();
|
|
||||||
const { prisma, records } = createPrismaMock([
|
|
||||||
createRecord({
|
|
||||||
id: "uc-old",
|
|
||||||
userId: "user-old",
|
|
||||||
containerId: "ctr-old",
|
|
||||||
containerName: "mosaic-user-user-old",
|
|
||||||
gatewayPort: 19045,
|
|
||||||
status: "running",
|
|
||||||
lastActiveAt: new Date(now - 60 * 60 * 1000),
|
|
||||||
idleTimeoutMin: 30,
|
|
||||||
}),
|
|
||||||
createRecord({
|
|
||||||
id: "uc-fresh",
|
|
||||||
userId: "user-fresh",
|
|
||||||
containerId: "ctr-fresh",
|
|
||||||
containerName: "mosaic-user-user-fresh",
|
|
||||||
gatewayPort: 19046,
|
|
||||||
status: "running",
|
|
||||||
lastActiveAt: new Date(now - 5 * 60 * 1000),
|
|
||||||
idleTimeoutMin: 30,
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
const crypto = createCryptoMock();
|
|
||||||
const config = createConfigMock();
|
|
||||||
const service = new ContainerLifecycleService(
|
|
||||||
prisma as unknown as PrismaService,
|
|
||||||
crypto as unknown as CryptoService,
|
|
||||||
config as unknown as ConfigService
|
|
||||||
);
|
|
||||||
|
|
||||||
dockerMock.registerContainer({
|
|
||||||
id: "ctr-old",
|
|
||||||
name: "mosaic-user-user-old",
|
|
||||||
running: true,
|
|
||||||
port: 19045,
|
|
||||||
});
|
|
||||||
dockerMock.registerContainer({
|
|
||||||
id: "ctr-fresh",
|
|
||||||
name: "mosaic-user-user-fresh",
|
|
||||||
running: true,
|
|
||||||
port: 19046,
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await service.reapIdle();
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
stopped: ["user-old"],
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(records.get("user-old")?.status).toBe("stopped");
|
|
||||||
expect(records.get("user-fresh")?.status).toBe("running");
|
|
||||||
|
|
||||||
const oldHandle = dockerMock.handles.get("ctr-old");
|
|
||||||
const freshHandle = dockerMock.handles.get("ctr-fresh");
|
|
||||||
expect(oldHandle?.stop).toHaveBeenCalledTimes(1);
|
|
||||||
expect(freshHandle?.stop).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("touch updates lastActiveAt", async () => {
|
|
||||||
const { prisma, records } = createPrismaMock([
|
|
||||||
createRecord({
|
|
||||||
id: "uc-touch",
|
|
||||||
userId: "user-touch",
|
|
||||||
containerName: "mosaic-user-user-touch",
|
|
||||||
lastActiveAt: null,
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
const crypto = createCryptoMock();
|
|
||||||
const config = createConfigMock();
|
|
||||||
const service = new ContainerLifecycleService(
|
|
||||||
prisma as unknown as PrismaService,
|
|
||||||
crypto as unknown as CryptoService,
|
|
||||||
config as unknown as ConfigService
|
|
||||||
);
|
|
||||||
|
|
||||||
await service.touch("user-touch");
|
|
||||||
|
|
||||||
const updatedRecord = records.get("user-touch");
|
|
||||||
expect(updatedRecord?.lastActiveAt).toBeInstanceOf(Date);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("getStatus returns null for unknown user", async () => {
|
|
||||||
const { prisma } = createPrismaMock();
|
|
||||||
const crypto = createCryptoMock();
|
|
||||||
const config = createConfigMock();
|
|
||||||
const service = new ContainerLifecycleService(
|
|
||||||
prisma as unknown as PrismaService,
|
|
||||||
crypto as unknown as CryptoService,
|
|
||||||
config as unknown as ConfigService
|
|
||||||
);
|
|
||||||
|
|
||||||
const status = await service.getStatus("missing-user");
|
|
||||||
|
|
||||||
expect(status).toBeNull();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,532 +0,0 @@
|
|||||||
import { Injectable, Logger } from "@nestjs/common";
|
|
||||||
import { ConfigService } from "@nestjs/config";
|
|
||||||
import Docker from "dockerode";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { CryptoService } from "../crypto/crypto.service";
|
|
||||||
|
|
||||||
const DEFAULT_DOCKER_SOCKET_PATH = "/var/run/docker.sock";
|
|
||||||
const DEFAULT_DOCKER_TCP_PORT = 2375;
|
|
||||||
const DEFAULT_OPENCLAW_IMAGE = "alpine/openclaw:latest";
|
|
||||||
const DEFAULT_OPENCLAW_NETWORK = "mosaic-internal";
|
|
||||||
const DEFAULT_OPENCLAW_PORT_RANGE_START = 19000;
|
|
||||||
const DEFAULT_MOSAIC_API_URL = "http://mosaic-api:3000/api";
|
|
||||||
const OPENCLAW_GATEWAY_PORT_KEY = "18789/tcp";
|
|
||||||
const OPENCLAW_STATE_PATH = "/home/node/.openclaw";
|
|
||||||
const CONTAINER_STOP_TIMEOUT_SECONDS = 10;
|
|
||||||
|
|
||||||
interface ContainerHandle {
|
|
||||||
inspect(): Promise<DockerInspect>;
|
|
||||||
start(): Promise<void>;
|
|
||||||
stop(options?: { t?: number }): Promise<void>;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface DockerInspect {
|
|
||||||
Id?: string;
|
|
||||||
State?: {
|
|
||||||
Running?: boolean;
|
|
||||||
Health?: {
|
|
||||||
Status?: string;
|
|
||||||
};
|
|
||||||
};
|
|
||||||
NetworkSettings?: {
|
|
||||||
Ports?: Record<string, { HostPort?: string }[] | null>;
|
|
||||||
};
|
|
||||||
HostConfig?: {
|
|
||||||
PortBindings?: Record<string, { HostPort?: string }[] | null>;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
interface UserContainerRecord {
|
|
||||||
id: string;
|
|
||||||
userId: string;
|
|
||||||
containerId: string | null;
|
|
||||||
containerName: string;
|
|
||||||
gatewayPort: number | null;
|
|
||||||
gatewayToken: string;
|
|
||||||
status: string;
|
|
||||||
lastActiveAt: Date | null;
|
|
||||||
idleTimeoutMin: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface ContainerLookup {
|
|
||||||
containerId: string | null;
|
|
||||||
containerName: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class ContainerLifecycleService {
|
|
||||||
private readonly logger = new Logger(ContainerLifecycleService.name);
|
|
||||||
private readonly docker: Docker;
|
|
||||||
|
|
||||||
constructor(
|
|
||||||
private readonly prisma: PrismaService,
|
|
||||||
private readonly crypto: CryptoService,
|
|
||||||
private readonly config: ConfigService
|
|
||||||
) {
|
|
||||||
const dockerHost = this.config.get<string>("DOCKER_HOST");
|
|
||||||
this.docker = this.createDockerClient(dockerHost);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Ensure a user's container is running. Creates if needed, starts if stopped.
|
|
||||||
// Returns the container's internal URL and gateway token.
|
|
||||||
async ensureRunning(userId: string): Promise<{ url: string; token: string }> {
|
|
||||||
const containerRecord = await this.getOrCreateContainerRecord(userId);
|
|
||||||
const token = this.getGatewayToken(containerRecord.gatewayToken);
|
|
||||||
const existingContainer = await this.resolveContainer(containerRecord);
|
|
||||||
|
|
||||||
let container: ContainerHandle;
|
|
||||||
if (existingContainer) {
|
|
||||||
container = existingContainer;
|
|
||||||
const inspect = await container.inspect();
|
|
||||||
if (!inspect.State?.Running) {
|
|
||||||
await container.start();
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
const port = await this.findAvailableGatewayPort();
|
|
||||||
container = await this.createContainer(containerRecord, token, port);
|
|
||||||
await container.start();
|
|
||||||
}
|
|
||||||
|
|
||||||
const inspect = await container.inspect();
|
|
||||||
const containerId = inspect.Id;
|
|
||||||
if (!containerId) {
|
|
||||||
throw new Error(
|
|
||||||
`Docker inspect did not return container ID for ${containerRecord.containerName}`
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
const gatewayPort = this.extractGatewayPort(inspect);
|
|
||||||
if (!gatewayPort) {
|
|
||||||
throw new Error(`Could not determine gateway port for ${containerRecord.containerName}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
const now = new Date();
|
|
||||||
await this.prisma.userContainer.update({
|
|
||||||
where: { userId },
|
|
||||||
data: {
|
|
||||||
containerId,
|
|
||||||
gatewayPort,
|
|
||||||
status: "running",
|
|
||||||
lastActiveAt: now,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
return {
|
|
||||||
url: `http://${containerRecord.containerName}:${String(gatewayPort)}`,
|
|
||||||
token,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
// Stop a user's container
|
|
||||||
async stop(userId: string): Promise<void> {
|
|
||||||
const containerRecord = await this.prisma.userContainer.findUnique({
|
|
||||||
where: { userId },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!containerRecord) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const container = await this.resolveContainer(containerRecord);
|
|
||||||
if (container) {
|
|
||||||
try {
|
|
||||||
await container.stop({ t: CONTAINER_STOP_TIMEOUT_SECONDS });
|
|
||||||
} catch (error) {
|
|
||||||
if (!this.isDockerNotFound(error) && !this.isAlreadyStopped(error)) {
|
|
||||||
throw error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
await this.prisma.userContainer.update({
|
|
||||||
where: { userId },
|
|
||||||
data: {
|
|
||||||
status: "stopped",
|
|
||||||
containerId: null,
|
|
||||||
gatewayPort: null,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Stop idle containers (called by cron/scheduler)
|
|
||||||
async reapIdle(): Promise<{ stopped: string[] }> {
|
|
||||||
const now = Date.now();
|
|
||||||
const runningContainers = await this.prisma.userContainer.findMany({
|
|
||||||
where: {
|
|
||||||
status: "running",
|
|
||||||
lastActiveAt: { not: null },
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
userId: true,
|
|
||||||
lastActiveAt: true,
|
|
||||||
idleTimeoutMin: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const stopped: string[] = [];
|
|
||||||
for (const container of runningContainers) {
|
|
||||||
const lastActiveAt = container.lastActiveAt;
|
|
||||||
if (!lastActiveAt) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
const idleLimitMs = container.idleTimeoutMin * 60 * 1000;
|
|
||||||
if (now - lastActiveAt.getTime() < idleLimitMs) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
await this.stop(container.userId);
|
|
||||||
stopped.push(container.userId);
|
|
||||||
} catch (error) {
|
|
||||||
this.logger.warn(
|
|
||||||
`Failed to stop idle container for user ${container.userId}: ${this.getErrorMessage(error)}`
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return { stopped };
|
|
||||||
}
|
|
||||||
|
|
||||||
// Health check all running containers
|
|
||||||
async healthCheckAll(): Promise<{ userId: string; healthy: boolean; error?: string }[]> {
|
|
||||||
const runningContainers = await this.prisma.userContainer.findMany({
|
|
||||||
where: {
|
|
||||||
status: "running",
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
userId: true,
|
|
||||||
containerId: true,
|
|
||||||
containerName: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const results: { userId: string; healthy: boolean; error?: string }[] = [];
|
|
||||||
for (const containerRecord of runningContainers) {
|
|
||||||
const container = await this.resolveContainer(containerRecord);
|
|
||||||
if (!container) {
|
|
||||||
results.push({
|
|
||||||
userId: containerRecord.userId,
|
|
||||||
healthy: false,
|
|
||||||
error: "Container not found",
|
|
||||||
});
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const inspect = await container.inspect();
|
|
||||||
const isRunning = inspect.State?.Running === true;
|
|
||||||
const healthState = inspect.State?.Health?.Status;
|
|
||||||
const healthy = isRunning && healthState !== "unhealthy";
|
|
||||||
|
|
||||||
if (healthy) {
|
|
||||||
results.push({
|
|
||||||
userId: containerRecord.userId,
|
|
||||||
healthy: true,
|
|
||||||
});
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
results.push({
|
|
||||||
userId: containerRecord.userId,
|
|
||||||
healthy: false,
|
|
||||||
error:
|
|
||||||
healthState === "unhealthy" ? "Container healthcheck failed" : "Container not running",
|
|
||||||
});
|
|
||||||
} catch (error) {
|
|
||||||
results.push({
|
|
||||||
userId: containerRecord.userId,
|
|
||||||
healthy: false,
|
|
||||||
error: this.getErrorMessage(error),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return results;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Restart a container with fresh config (for config updates)
|
|
||||||
async restart(userId: string): Promise<void> {
|
|
||||||
await this.stop(userId);
|
|
||||||
await this.ensureRunning(userId);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Update lastActiveAt timestamp (called on each chat request)
|
|
||||||
async touch(userId: string): Promise<void> {
|
|
||||||
await this.prisma.userContainer.updateMany({
|
|
||||||
where: { userId },
|
|
||||||
data: {
|
|
||||||
lastActiveAt: new Date(),
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get container status for a user
|
|
||||||
async getStatus(
|
|
||||||
userId: string
|
|
||||||
): Promise<{ status: string; port?: number; lastActive?: Date } | null> {
|
|
||||||
const container = await this.prisma.userContainer.findUnique({
|
|
||||||
where: { userId },
|
|
||||||
select: {
|
|
||||||
status: true,
|
|
||||||
gatewayPort: true,
|
|
||||||
lastActiveAt: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!container) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
const status: { status: string; port?: number; lastActive?: Date } = {
|
|
||||||
status: container.status,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (container.gatewayPort !== null) {
|
|
||||||
status.port = container.gatewayPort;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (container.lastActiveAt !== null) {
|
|
||||||
status.lastActive = container.lastActiveAt;
|
|
||||||
}
|
|
||||||
|
|
||||||
return status;
|
|
||||||
}
|
|
||||||
|
|
||||||
private createDockerClient(dockerHost?: string): Docker {
|
|
||||||
if (!dockerHost || dockerHost.trim().length === 0) {
|
|
||||||
return new Docker({ socketPath: DEFAULT_DOCKER_SOCKET_PATH });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (dockerHost.startsWith("unix://")) {
|
|
||||||
return new Docker({ socketPath: dockerHost.slice("unix://".length) });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (dockerHost.startsWith("tcp://")) {
|
|
||||||
const parsed = new URL(dockerHost.replace("tcp://", "http://"));
|
|
||||||
return new Docker({
|
|
||||||
host: parsed.hostname,
|
|
||||||
port: this.parseInteger(parsed.port, DEFAULT_DOCKER_TCP_PORT),
|
|
||||||
protocol: "http",
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
if (dockerHost.startsWith("http://") || dockerHost.startsWith("https://")) {
|
|
||||||
const parsed = new URL(dockerHost);
|
|
||||||
const protocol = parsed.protocol.replace(":", "");
|
|
||||||
return new Docker({
|
|
||||||
host: parsed.hostname,
|
|
||||||
port: this.parseInteger(parsed.port, DEFAULT_DOCKER_TCP_PORT),
|
|
||||||
protocol: protocol === "https" ? "https" : "http",
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return new Docker({ socketPath: dockerHost });
|
|
||||||
}
|
|
||||||
|
|
||||||
private async getOrCreateContainerRecord(userId: string): Promise<UserContainerRecord> {
|
|
||||||
const existingContainer = await this.prisma.userContainer.findUnique({
|
|
||||||
where: { userId },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (existingContainer) {
|
|
||||||
return existingContainer;
|
|
||||||
}
|
|
||||||
|
|
||||||
const token = this.crypto.generateToken();
|
|
||||||
const containerName = this.getContainerName(userId);
|
|
||||||
return this.prisma.userContainer.create({
|
|
||||||
data: {
|
|
||||||
userId,
|
|
||||||
containerName,
|
|
||||||
gatewayToken: this.crypto.encrypt(token),
|
|
||||||
status: "stopped",
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
private getContainerName(userId: string): string {
|
|
||||||
return `mosaic-user-${userId}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getVolumeName(userId: string): string {
|
|
||||||
return `mosaic-user-${userId}-state`;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getOpenClawImage(): string {
|
|
||||||
return this.config.get<string>("OPENCLAW_IMAGE") ?? DEFAULT_OPENCLAW_IMAGE;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getOpenClawNetwork(): string {
|
|
||||||
return this.config.get<string>("OPENCLAW_NETWORK") ?? DEFAULT_OPENCLAW_NETWORK;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getMosaicApiUrl(): string {
|
|
||||||
return this.config.get<string>("MOSAIC_API_URL") ?? DEFAULT_MOSAIC_API_URL;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getPortRangeStart(): number {
|
|
||||||
return this.parseInteger(
|
|
||||||
this.config.get<string>("OPENCLAW_PORT_RANGE_START"),
|
|
||||||
DEFAULT_OPENCLAW_PORT_RANGE_START
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
private async resolveContainer(record: ContainerLookup): Promise<ContainerHandle | null> {
|
|
||||||
if (record.containerId) {
|
|
||||||
const byId = this.docker.getContainer(record.containerId) as unknown as ContainerHandle;
|
|
||||||
if (await this.containerExists(byId)) {
|
|
||||||
return byId;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const byName = await this.findContainerByName(record.containerName);
|
|
||||||
if (byName) {
|
|
||||||
return byName;
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
private async findContainerByName(containerName: string): Promise<ContainerHandle | null> {
|
|
||||||
const containers = await this.docker.listContainers({
|
|
||||||
all: true,
|
|
||||||
filters: {
|
|
||||||
name: [containerName],
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const match = containers.find((container) => {
|
|
||||||
const names = container.Names;
|
|
||||||
return names.some((name) => name === `/${containerName}` || name.includes(containerName));
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!match?.Id) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
return this.docker.getContainer(match.Id) as unknown as ContainerHandle;
|
|
||||||
}
|
|
||||||
|
|
||||||
private async containerExists(container: ContainerHandle): Promise<boolean> {
|
|
||||||
try {
|
|
||||||
await container.inspect();
|
|
||||||
return true;
|
|
||||||
} catch (error) {
|
|
||||||
if (this.isDockerNotFound(error)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
throw error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private async createContainer(
|
|
||||||
containerRecord: UserContainerRecord,
|
|
||||||
token: string,
|
|
||||||
gatewayPort: number
|
|
||||||
): Promise<ContainerHandle> {
|
|
||||||
const container = await this.docker.createContainer({
|
|
||||||
name: containerRecord.containerName,
|
|
||||||
Image: this.getOpenClawImage(),
|
|
||||||
Env: [
|
|
||||||
`MOSAIC_API_URL=${this.getMosaicApiUrl()}`,
|
|
||||||
`AGENT_TOKEN=${token}`,
|
|
||||||
`AGENT_ID=${containerRecord.id}`,
|
|
||||||
],
|
|
||||||
ExposedPorts: {
|
|
||||||
[OPENCLAW_GATEWAY_PORT_KEY]: {},
|
|
||||||
},
|
|
||||||
HostConfig: {
|
|
||||||
Binds: [`${this.getVolumeName(containerRecord.userId)}:${OPENCLAW_STATE_PATH}`],
|
|
||||||
PortBindings: {
|
|
||||||
[OPENCLAW_GATEWAY_PORT_KEY]: [{ HostPort: String(gatewayPort) }],
|
|
||||||
},
|
|
||||||
NetworkMode: this.getOpenClawNetwork(),
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
return container as unknown as ContainerHandle;
|
|
||||||
}
|
|
||||||
|
|
||||||
private extractGatewayPort(inspect: DockerInspect): number | null {
|
|
||||||
const networkPort = inspect.NetworkSettings?.Ports?.[OPENCLAW_GATEWAY_PORT_KEY]?.[0]?.HostPort;
|
|
||||||
if (networkPort) {
|
|
||||||
return this.parseInteger(networkPort, 0) || null;
|
|
||||||
}
|
|
||||||
|
|
||||||
const hostPort = inspect.HostConfig?.PortBindings?.[OPENCLAW_GATEWAY_PORT_KEY]?.[0]?.HostPort;
|
|
||||||
if (hostPort) {
|
|
||||||
return this.parseInteger(hostPort, 0) || null;
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
private async findAvailableGatewayPort(): Promise<number> {
|
|
||||||
const usedPorts = await this.prisma.userContainer.findMany({
|
|
||||||
where: {
|
|
||||||
gatewayPort: { not: null },
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
gatewayPort: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const takenPorts = new Set<number>();
|
|
||||||
for (const entry of usedPorts) {
|
|
||||||
if (entry.gatewayPort !== null) {
|
|
||||||
takenPorts.add(entry.gatewayPort);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let candidate = this.getPortRangeStart();
|
|
||||||
while (takenPorts.has(candidate)) {
|
|
||||||
candidate += 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return candidate;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getGatewayToken(storedToken: string): string {
|
|
||||||
if (this.crypto.isEncrypted(storedToken)) {
|
|
||||||
return this.crypto.decrypt(storedToken);
|
|
||||||
}
|
|
||||||
|
|
||||||
return storedToken;
|
|
||||||
}
|
|
||||||
|
|
||||||
private parseInteger(value: string | undefined, fallback: number): number {
|
|
||||||
if (!value) {
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
const parsed = Number.parseInt(value, 10);
|
|
||||||
return Number.isFinite(parsed) ? parsed : fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
private isDockerNotFound(error: unknown): boolean {
|
|
||||||
return this.getDockerStatusCode(error) === 404;
|
|
||||||
}
|
|
||||||
|
|
||||||
private isAlreadyStopped(error: unknown): boolean {
|
|
||||||
return this.getDockerStatusCode(error) === 304;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getDockerStatusCode(error: unknown): number | null {
|
|
||||||
if (typeof error !== "object" || error === null || !("statusCode" in error)) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
const statusCode = error.statusCode;
|
|
||||||
return typeof statusCode === "number" ? statusCode : null;
|
|
||||||
}
|
|
||||||
|
|
||||||
private getErrorMessage(error: unknown): string {
|
|
||||||
if (error instanceof Error) {
|
|
||||||
return error.message;
|
|
||||||
}
|
|
||||||
|
|
||||||
return "Unknown error";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { ScheduleModule } from "@nestjs/schedule";
|
|
||||||
import { ContainerLifecycleModule } from "../container-lifecycle/container-lifecycle.module";
|
|
||||||
import { ContainerReaperService } from "./container-reaper.service";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [ScheduleModule, ContainerLifecycleModule],
|
|
||||||
providers: [ContainerReaperService],
|
|
||||||
})
|
|
||||||
export class ContainerReaperModule {}
|
|
||||||
@@ -1,45 +0,0 @@
|
|||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
import type { ContainerLifecycleService } from "../container-lifecycle/container-lifecycle.service";
|
|
||||||
import { ContainerReaperService } from "./container-reaper.service";
|
|
||||||
|
|
||||||
describe("ContainerReaperService", () => {
|
|
||||||
let service: ContainerReaperService;
|
|
||||||
let containerLifecycle: Pick<ContainerLifecycleService, "reapIdle">;
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
containerLifecycle = {
|
|
||||||
reapIdle: vi.fn(),
|
|
||||||
};
|
|
||||||
service = new ContainerReaperService(containerLifecycle as ContainerLifecycleService);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("reapIdleContainers calls containerLifecycle.reapIdle()", async () => {
|
|
||||||
vi.mocked(containerLifecycle.reapIdle).mockResolvedValue({ stopped: [] });
|
|
||||||
|
|
||||||
await service.reapIdleContainers();
|
|
||||||
|
|
||||||
expect(containerLifecycle.reapIdle).toHaveBeenCalledTimes(1);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("reapIdleContainers handles errors gracefully", async () => {
|
|
||||||
const error = new Error("reap failure");
|
|
||||||
vi.mocked(containerLifecycle.reapIdle).mockRejectedValue(error);
|
|
||||||
const loggerError = vi.spyOn(service["logger"], "error").mockImplementation(() => {});
|
|
||||||
|
|
||||||
await expect(service.reapIdleContainers()).resolves.toBeUndefined();
|
|
||||||
|
|
||||||
expect(loggerError).toHaveBeenCalledWith(
|
|
||||||
"Failed to reap idle containers",
|
|
||||||
expect.stringContaining("reap failure")
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("reapIdleContainers logs stopped container count", async () => {
|
|
||||||
vi.mocked(containerLifecycle.reapIdle).mockResolvedValue({ stopped: ["user-1", "user-2"] });
|
|
||||||
const loggerLog = vi.spyOn(service["logger"], "log").mockImplementation(() => {});
|
|
||||||
|
|
||||||
await service.reapIdleContainers();
|
|
||||||
|
|
||||||
expect(loggerLog).toHaveBeenCalledWith("Stopped 2 idle containers: user-1, user-2");
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,30 +0,0 @@
|
|||||||
import { Injectable, Logger } from "@nestjs/common";
|
|
||||||
import { Cron, CronExpression } from "@nestjs/schedule";
|
|
||||||
import { ContainerLifecycleService } from "../container-lifecycle/container-lifecycle.service";
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class ContainerReaperService {
|
|
||||||
private readonly logger = new Logger(ContainerReaperService.name);
|
|
||||||
|
|
||||||
constructor(private readonly containerLifecycle: ContainerLifecycleService) {}
|
|
||||||
|
|
||||||
@Cron(CronExpression.EVERY_5_MINUTES)
|
|
||||||
async reapIdleContainers(): Promise<void> {
|
|
||||||
this.logger.log("Running idle container reap cycle...");
|
|
||||||
try {
|
|
||||||
const result = await this.containerLifecycle.reapIdle();
|
|
||||||
if (result.stopped.length > 0) {
|
|
||||||
this.logger.log(
|
|
||||||
`Stopped ${String(result.stopped.length)} idle containers: ${result.stopped.join(", ")}`
|
|
||||||
);
|
|
||||||
} else {
|
|
||||||
this.logger.debug("No idle containers to stop");
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
this.logger.error(
|
|
||||||
"Failed to reap idle containers",
|
|
||||||
error instanceof Error ? error.stack : String(error)
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,69 +0,0 @@
|
|||||||
import { Controller, Post, Get, Body, Param, Query, UseGuards } from "@nestjs/common";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
|
|
||||||
import { Workspace, RequirePermission, Permission } from "../common/decorators";
|
|
||||||
import { ConversationArchiveService } from "./conversation-archive.service";
|
|
||||||
import { IngestConversationDto, SearchConversationDto, ListConversationsDto } from "./dto";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Controller for conversation archive endpoints.
|
|
||||||
* All endpoints require workspace membership.
|
|
||||||
*/
|
|
||||||
@Controller("conversations")
|
|
||||||
@UseGuards(AuthGuard, WorkspaceGuard, PermissionGuard)
|
|
||||||
export class ConversationArchiveController {
|
|
||||||
constructor(private readonly service: ConversationArchiveService) {}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* POST /api/conversations/ingest
|
|
||||||
* Ingest a conversation session log and auto-embed for semantic search.
|
|
||||||
* Requires: MEMBER or higher
|
|
||||||
*/
|
|
||||||
@Post("ingest")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_MEMBER)
|
|
||||||
async ingest(
|
|
||||||
@Workspace() workspaceId: string,
|
|
||||||
@Body() dto: IngestConversationDto
|
|
||||||
): Promise<{ id: string }> {
|
|
||||||
return this.service.ingest(workspaceId, dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* POST /api/conversations/search
|
|
||||||
* Vector similarity search across archived conversations.
|
|
||||||
* Requires: Any workspace member
|
|
||||||
*/
|
|
||||||
@Post("search")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async search(
|
|
||||||
@Workspace() workspaceId: string,
|
|
||||||
@Body() dto: SearchConversationDto
|
|
||||||
): Promise<unknown> {
|
|
||||||
return this.service.search(workspaceId, dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* GET /api/conversations
|
|
||||||
* List conversation archives with filtering and pagination.
|
|
||||||
* Requires: Any workspace member
|
|
||||||
*/
|
|
||||||
@Get()
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async findAll(
|
|
||||||
@Workspace() workspaceId: string,
|
|
||||||
@Query() query: ListConversationsDto
|
|
||||||
): Promise<unknown> {
|
|
||||||
return this.service.findAll(workspaceId, query);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* GET /api/conversations/:id
|
|
||||||
* Get a single conversation archive by ID (includes full messages).
|
|
||||||
* Requires: Any workspace member
|
|
||||||
*/
|
|
||||||
@Get(":id")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async findOne(@Workspace() workspaceId: string, @Param("id") id: string): Promise<unknown> {
|
|
||||||
return this.service.findOne(workspaceId, id);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,239 +0,0 @@
|
|||||||
import { beforeAll, beforeEach, describe, expect, it, afterAll, vi } from "vitest";
|
|
||||||
import { randomUUID as uuid } from "crypto";
|
|
||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { ConflictException } from "@nestjs/common";
|
|
||||||
import { PrismaClient, Prisma } from "@prisma/client";
|
|
||||||
import { EMBEDDING_DIMENSION } from "@mosaic/shared";
|
|
||||||
import { ConversationArchiveService } from "./conversation-archive.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { EmbeddingService } from "../knowledge/services/embedding.service";
|
|
||||||
|
|
||||||
const shouldRunDbIntegrationTests =
|
|
||||||
process.env.RUN_DB_TESTS === "true" && Boolean(process.env.DATABASE_URL);
|
|
||||||
const describeFn = shouldRunDbIntegrationTests ? describe : describe.skip;
|
|
||||||
|
|
||||||
function vector(value: number): number[] {
|
|
||||||
return Array.from({ length: EMBEDDING_DIMENSION }, () => value);
|
|
||||||
}
|
|
||||||
|
|
||||||
function toVectorLiteral(input: number[]): string {
|
|
||||||
return `[${input.join(",")}]`;
|
|
||||||
}
|
|
||||||
|
|
||||||
describeFn("ConversationArchiveService Integration", () => {
|
|
||||||
let moduleRef: TestingModule;
|
|
||||||
let prisma: PrismaClient;
|
|
||||||
let service: ConversationArchiveService;
|
|
||||||
let workspaceId: string;
|
|
||||||
let ownerId: string;
|
|
||||||
let setupComplete = false;
|
|
||||||
|
|
||||||
const embeddingServiceMock = {
|
|
||||||
isConfigured: vi.fn(),
|
|
||||||
generateEmbedding: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeAll(async () => {
|
|
||||||
prisma = new PrismaClient();
|
|
||||||
await prisma.$connect();
|
|
||||||
|
|
||||||
const workspace = await prisma.workspace.create({
|
|
||||||
data: {
|
|
||||||
name: `Conversation Archive Integration ${Date.now()}`,
|
|
||||||
owner: {
|
|
||||||
create: {
|
|
||||||
email: `conversation-archive-integration-${Date.now()}@example.com`,
|
|
||||||
name: "Conversation Archive Integration Owner",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
workspaceId = workspace.id;
|
|
||||||
ownerId = workspace.ownerId;
|
|
||||||
|
|
||||||
moduleRef = await Test.createTestingModule({
|
|
||||||
providers: [
|
|
||||||
ConversationArchiveService,
|
|
||||||
{
|
|
||||||
provide: PrismaService,
|
|
||||||
useValue: prisma,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
provide: EmbeddingService,
|
|
||||||
useValue: embeddingServiceMock,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}).compile();
|
|
||||||
|
|
||||||
service = moduleRef.get<ConversationArchiveService>(ConversationArchiveService);
|
|
||||||
setupComplete = true;
|
|
||||||
});
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
embeddingServiceMock.isConfigured.mockReturnValue(false);
|
|
||||||
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
await prisma.conversationArchive.deleteMany({ where: { workspaceId } });
|
|
||||||
});
|
|
||||||
|
|
||||||
afterAll(async () => {
|
|
||||||
if (!prisma) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (workspaceId) {
|
|
||||||
await prisma.conversationArchive.deleteMany({ where: { workspaceId } });
|
|
||||||
await prisma.workspace.deleteMany({ where: { id: workspaceId } });
|
|
||||||
}
|
|
||||||
if (ownerId) {
|
|
||||||
await prisma.user.deleteMany({ where: { id: ownerId } });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (moduleRef) {
|
|
||||||
await moduleRef.close();
|
|
||||||
}
|
|
||||||
await prisma.$disconnect();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("ingests a conversation log", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const sessionId = `session-${uuid()}`;
|
|
||||||
|
|
||||||
const result = await service.ingest(workspaceId, {
|
|
||||||
sessionId,
|
|
||||||
agentId: "agent-conversation-ingest",
|
|
||||||
messages: [
|
|
||||||
{ role: "user", content: "Can you summarize deployment issues?" },
|
|
||||||
{ role: "assistant", content: "Yes, three retries timed out in staging." },
|
|
||||||
],
|
|
||||||
summary: "Deployment retry failures discussed",
|
|
||||||
startedAt: "2026-02-28T21:00:00.000Z",
|
|
||||||
endedAt: "2026-02-28T21:05:00.000Z",
|
|
||||||
metadata: { source: "integration-test" },
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.id).toBeDefined();
|
|
||||||
|
|
||||||
const stored = await prisma.conversationArchive.findUnique({
|
|
||||||
where: {
|
|
||||||
id: result.id,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(stored).toBeTruthy();
|
|
||||||
expect(stored?.workspaceId).toBe(workspaceId);
|
|
||||||
expect(stored?.sessionId).toBe(sessionId);
|
|
||||||
expect(stored?.messageCount).toBe(2);
|
|
||||||
expect(stored?.summary).toBe("Deployment retry failures discussed");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("rejects duplicate session ingest per workspace", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const sessionId = `session-${uuid()}`;
|
|
||||||
const dto = {
|
|
||||||
sessionId,
|
|
||||||
agentId: "agent-conversation-duplicate",
|
|
||||||
messages: [{ role: "user", content: "hello" }],
|
|
||||||
summary: "simple conversation",
|
|
||||||
startedAt: "2026-02-28T22:00:00.000Z",
|
|
||||||
};
|
|
||||||
|
|
||||||
await service.ingest(workspaceId, dto);
|
|
||||||
|
|
||||||
await expect(service.ingest(workspaceId, dto)).rejects.toThrow(ConflictException);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("rejects semantic search when embeddings are disabled", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
embeddingServiceMock.isConfigured.mockReturnValue(false);
|
|
||||||
|
|
||||||
await expect(
|
|
||||||
service.search(workspaceId, {
|
|
||||||
query: "deployment retries",
|
|
||||||
})
|
|
||||||
).rejects.toThrow(ConflictException);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("searches archived conversations by vector similarity", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const near = vector(0.02);
|
|
||||||
const far = vector(0.85);
|
|
||||||
|
|
||||||
const matching = await prisma.conversationArchive.create({
|
|
||||||
data: {
|
|
||||||
workspaceId,
|
|
||||||
sessionId: `session-search-${uuid()}`,
|
|
||||||
agentId: "agent-conversation-search-a",
|
|
||||||
messages: [
|
|
||||||
{ role: "user", content: "What caused deployment retries?" },
|
|
||||||
{ role: "assistant", content: "A connection pool timeout." },
|
|
||||||
] as unknown as Prisma.InputJsonValue,
|
|
||||||
messageCount: 2,
|
|
||||||
summary: "Deployment retries caused by connection pool timeout",
|
|
||||||
startedAt: new Date("2026-02-28T23:00:00.000Z"),
|
|
||||||
metadata: { channel: "cli" } as Prisma.InputJsonValue,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const nonMatching = await prisma.conversationArchive.create({
|
|
||||||
data: {
|
|
||||||
workspaceId,
|
|
||||||
sessionId: `session-search-${uuid()}`,
|
|
||||||
agentId: "agent-conversation-search-b",
|
|
||||||
messages: [
|
|
||||||
{ role: "user", content: "How is billing configured?" },
|
|
||||||
] as unknown as Prisma.InputJsonValue,
|
|
||||||
messageCount: 1,
|
|
||||||
summary: "Billing and quotas conversation",
|
|
||||||
startedAt: new Date("2026-02-28T23:10:00.000Z"),
|
|
||||||
metadata: { channel: "cli" } as Prisma.InputJsonValue,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
await prisma.$executeRaw`
|
|
||||||
UPDATE conversation_archives
|
|
||||||
SET embedding = ${toVectorLiteral(near)}::vector(${EMBEDDING_DIMENSION})
|
|
||||||
WHERE id = ${matching.id}::uuid
|
|
||||||
`;
|
|
||||||
await prisma.$executeRaw`
|
|
||||||
UPDATE conversation_archives
|
|
||||||
SET embedding = ${toVectorLiteral(far)}::vector(${EMBEDDING_DIMENSION})
|
|
||||||
WHERE id = ${nonMatching.id}::uuid
|
|
||||||
`;
|
|
||||||
|
|
||||||
embeddingServiceMock.isConfigured.mockReturnValue(true);
|
|
||||||
embeddingServiceMock.generateEmbedding.mockResolvedValue(near);
|
|
||||||
|
|
||||||
const result = await service.search(workspaceId, {
|
|
||||||
query: "deployment retries timeout",
|
|
||||||
agentId: "agent-conversation-search-a",
|
|
||||||
similarityThreshold: 0,
|
|
||||||
limit: 10,
|
|
||||||
});
|
|
||||||
|
|
||||||
const rows = result.data as Array<{ id: string; agent_id: string; similarity: number }>;
|
|
||||||
|
|
||||||
expect(result.pagination.total).toBe(1);
|
|
||||||
expect(rows).toHaveLength(1);
|
|
||||||
expect(rows[0]?.id).toBe(matching.id);
|
|
||||||
expect(rows[0]?.agent_id).toBe("agent-conversation-search-a");
|
|
||||||
expect(rows[0]?.similarity).toBeGreaterThan(0);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,14 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { AuthModule } from "../auth/auth.module";
|
|
||||||
import { KnowledgeModule } from "../knowledge/knowledge.module";
|
|
||||||
import { ConversationArchiveService } from "./conversation-archive.service";
|
|
||||||
import { ConversationArchiveController } from "./conversation-archive.controller";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [PrismaModule, AuthModule, KnowledgeModule],
|
|
||||||
controllers: [ConversationArchiveController],
|
|
||||||
providers: [ConversationArchiveService],
|
|
||||||
exports: [ConversationArchiveService],
|
|
||||||
})
|
|
||||||
export class ConversationArchiveModule {}
|
|
||||||
@@ -1,149 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { ConflictException, NotFoundException } from "@nestjs/common";
|
|
||||||
import { ConversationArchiveService } from "./conversation-archive.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { EmbeddingService } from "../knowledge/services/embedding.service";
|
|
||||||
|
|
||||||
const mockPrisma = {
|
|
||||||
conversationArchive: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
create: vi.fn(),
|
|
||||||
count: vi.fn(),
|
|
||||||
findMany: vi.fn(),
|
|
||||||
findFirst: vi.fn(),
|
|
||||||
},
|
|
||||||
$queryRaw: vi.fn(),
|
|
||||||
$executeRaw: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockEmbedding = {
|
|
||||||
isConfigured: vi.fn(),
|
|
||||||
generateEmbedding: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
describe("ConversationArchiveService", () => {
|
|
||||||
let service: ConversationArchiveService;
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
|
|
||||||
const module: TestingModule = await Test.createTestingModule({
|
|
||||||
providers: [
|
|
||||||
ConversationArchiveService,
|
|
||||||
{ provide: PrismaService, useValue: mockPrisma },
|
|
||||||
{ provide: EmbeddingService, useValue: mockEmbedding },
|
|
||||||
],
|
|
||||||
}).compile();
|
|
||||||
|
|
||||||
service = module.get<ConversationArchiveService>(ConversationArchiveService);
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("ingest", () => {
|
|
||||||
const workspaceId = "ws-1";
|
|
||||||
const dto = {
|
|
||||||
sessionId: "sess-abc",
|
|
||||||
agentId: "agent-xyz",
|
|
||||||
messages: [
|
|
||||||
{ role: "user", content: "Hello" },
|
|
||||||
{ role: "assistant", content: "Hi there!" },
|
|
||||||
],
|
|
||||||
summary: "A greeting conversation",
|
|
||||||
startedAt: "2026-02-28T10:00:00Z",
|
|
||||||
};
|
|
||||||
|
|
||||||
it("creates a conversation archive and returns id", async () => {
|
|
||||||
mockPrisma.conversationArchive.findUnique.mockResolvedValue(null);
|
|
||||||
mockPrisma.conversationArchive.create.mockResolvedValue({ id: "conv-1" });
|
|
||||||
mockEmbedding.isConfigured.mockReturnValue(false);
|
|
||||||
|
|
||||||
const result = await service.ingest(workspaceId, dto);
|
|
||||||
|
|
||||||
expect(result).toEqual({ id: "conv-1" });
|
|
||||||
expect(mockPrisma.conversationArchive.create).toHaveBeenCalledWith(
|
|
||||||
expect.objectContaining({
|
|
||||||
data: expect.objectContaining({
|
|
||||||
workspaceId,
|
|
||||||
sessionId: dto.sessionId,
|
|
||||||
agentId: dto.agentId,
|
|
||||||
messageCount: 2,
|
|
||||||
}),
|
|
||||||
})
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("throws ConflictException when session already exists", async () => {
|
|
||||||
mockPrisma.conversationArchive.findUnique.mockResolvedValue({ id: "existing" });
|
|
||||||
|
|
||||||
await expect(service.ingest(workspaceId, dto)).rejects.toThrow(ConflictException);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findAll", () => {
|
|
||||||
const workspaceId = "ws-1";
|
|
||||||
|
|
||||||
it("returns paginated list", async () => {
|
|
||||||
mockPrisma.conversationArchive.count.mockResolvedValue(5);
|
|
||||||
mockPrisma.conversationArchive.findMany.mockResolvedValue([
|
|
||||||
{ id: "conv-1", sessionId: "sess-1" },
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await service.findAll(workspaceId, { page: 1, limit: 10 });
|
|
||||||
|
|
||||||
expect(result.pagination.total).toBe(5);
|
|
||||||
expect(result.data).toHaveLength(1);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("uses default pagination when not provided", async () => {
|
|
||||||
mockPrisma.conversationArchive.count.mockResolvedValue(0);
|
|
||||||
mockPrisma.conversationArchive.findMany.mockResolvedValue([]);
|
|
||||||
|
|
||||||
const result = await service.findAll(workspaceId, {});
|
|
||||||
|
|
||||||
expect(result.pagination.page).toBe(1);
|
|
||||||
expect(result.pagination.limit).toBe(20);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findOne", () => {
|
|
||||||
const workspaceId = "ws-1";
|
|
||||||
|
|
||||||
it("returns record when found", async () => {
|
|
||||||
const record = { id: "conv-1", workspaceId, sessionId: "sess-1" };
|
|
||||||
mockPrisma.conversationArchive.findFirst.mockResolvedValue(record);
|
|
||||||
|
|
||||||
const result = await service.findOne(workspaceId, "conv-1");
|
|
||||||
|
|
||||||
expect(result).toEqual(record);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("throws NotFoundException when record does not exist", async () => {
|
|
||||||
mockPrisma.conversationArchive.findFirst.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(service.findOne(workspaceId, "missing")).rejects.toThrow(NotFoundException);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("search", () => {
|
|
||||||
it("throws ConflictException when embedding is not configured", async () => {
|
|
||||||
mockEmbedding.isConfigured.mockReturnValue(false);
|
|
||||||
|
|
||||||
await expect(service.search("ws-1", { query: "test query" })).rejects.toThrow(
|
|
||||||
ConflictException
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("performs vector search when configured", async () => {
|
|
||||||
mockEmbedding.isConfigured.mockReturnValue(true);
|
|
||||||
mockEmbedding.generateEmbedding.mockResolvedValue(new Array(1536).fill(0.1));
|
|
||||||
mockPrisma.$queryRaw
|
|
||||||
.mockResolvedValueOnce([{ id: "conv-1", similarity: 0.9 }])
|
|
||||||
.mockResolvedValueOnce([{ count: BigInt(1) }]);
|
|
||||||
|
|
||||||
const result = await service.search("ws-1", { query: "greetings" });
|
|
||||||
|
|
||||||
expect(result.data).toHaveLength(1);
|
|
||||||
expect(result.pagination.total).toBe(1);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,277 +0,0 @@
|
|||||||
import { Injectable, Logger, NotFoundException, ConflictException } from "@nestjs/common";
|
|
||||||
import { Prisma } from "@prisma/client";
|
|
||||||
import { EMBEDDING_DIMENSION } from "@mosaic/shared";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { EmbeddingService } from "../knowledge/services/embedding.service";
|
|
||||||
import type { IngestConversationDto, SearchConversationDto, ListConversationsDto } from "./dto";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Shape of a raw conversation archive row from $queryRaw vector search
|
|
||||||
*/
|
|
||||||
interface RawConversationResult {
|
|
||||||
id: string;
|
|
||||||
workspace_id: string;
|
|
||||||
session_id: string;
|
|
||||||
agent_id: string;
|
|
||||||
messages: unknown;
|
|
||||||
message_count: number;
|
|
||||||
summary: string;
|
|
||||||
started_at: Date;
|
|
||||||
ended_at: Date | null;
|
|
||||||
metadata: unknown;
|
|
||||||
created_at: Date;
|
|
||||||
updated_at: Date;
|
|
||||||
similarity: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Paginated response wrapper
|
|
||||||
*/
|
|
||||||
export interface PaginatedConversations<T> {
|
|
||||||
data: T[];
|
|
||||||
pagination: {
|
|
||||||
page: number;
|
|
||||||
limit: number;
|
|
||||||
total: number;
|
|
||||||
totalPages: number;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class ConversationArchiveService {
|
|
||||||
private readonly logger = new Logger(ConversationArchiveService.name);
|
|
||||||
private readonly defaultSimilarityThreshold = 0.5;
|
|
||||||
|
|
||||||
constructor(
|
|
||||||
private readonly prisma: PrismaService,
|
|
||||||
private readonly embedding: EmbeddingService
|
|
||||||
) {}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Ingest a conversation session log.
|
|
||||||
* Generates a vector embedding from the summary + message content and stores it alongside the record.
|
|
||||||
*/
|
|
||||||
async ingest(workspaceId: string, dto: IngestConversationDto): Promise<{ id: string }> {
|
|
||||||
// Verify no duplicate session in this workspace
|
|
||||||
const existing = await this.prisma.conversationArchive.findUnique({
|
|
||||||
where: { workspaceId_sessionId: { workspaceId, sessionId: dto.sessionId } },
|
|
||||||
select: { id: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (existing) {
|
|
||||||
throw new ConflictException(
|
|
||||||
`Conversation session '${dto.sessionId}' already exists in this workspace`
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
const messageCount = dto.messages.length;
|
|
||||||
|
|
||||||
// Create record first to get ID for embedding
|
|
||||||
const record = await this.prisma.conversationArchive.create({
|
|
||||||
data: {
|
|
||||||
workspaceId,
|
|
||||||
sessionId: dto.sessionId,
|
|
||||||
agentId: dto.agentId,
|
|
||||||
messages: dto.messages as unknown as Prisma.InputJsonValue,
|
|
||||||
messageCount,
|
|
||||||
summary: dto.summary,
|
|
||||||
startedAt: new Date(dto.startedAt),
|
|
||||||
endedAt: dto.endedAt ? new Date(dto.endedAt) : null,
|
|
||||||
metadata: (dto.metadata ?? {}) as Prisma.InputJsonValue,
|
|
||||||
},
|
|
||||||
select: { id: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
// Generate and store embedding asynchronously (non-blocking for ingest)
|
|
||||||
if (this.embedding.isConfigured()) {
|
|
||||||
const textForEmbedding = this.buildEmbeddingText(dto.summary, dto.messages);
|
|
||||||
this.storeEmbedding(record.id, textForEmbedding).catch((err: unknown) => {
|
|
||||||
this.logger.error(`Failed to store embedding for conversation ${record.id}`, err);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
this.logger.log(`Ingested conversation ${record.id} (session: ${dto.sessionId})`);
|
|
||||||
return { id: record.id };
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Semantic vector search across conversation archives in a workspace.
|
|
||||||
*/
|
|
||||||
async search(
|
|
||||||
workspaceId: string,
|
|
||||||
dto: SearchConversationDto
|
|
||||||
): Promise<PaginatedConversations<RawConversationResult>> {
|
|
||||||
if (!this.embedding.isConfigured()) {
|
|
||||||
throw new ConflictException("Semantic search requires OpenAI API key to be configured");
|
|
||||||
}
|
|
||||||
|
|
||||||
const limit = dto.limit ?? 20;
|
|
||||||
const threshold = dto.similarityThreshold ?? this.defaultSimilarityThreshold;
|
|
||||||
const distanceThreshold = 1 - threshold;
|
|
||||||
|
|
||||||
const queryEmbedding = await this.embedding.generateEmbedding(dto.query);
|
|
||||||
const embeddingStr = `[${queryEmbedding.join(",")}]`;
|
|
||||||
|
|
||||||
const agentFilter = dto.agentId ? Prisma.sql`AND ca.agent_id = ${dto.agentId}` : Prisma.sql``;
|
|
||||||
|
|
||||||
const rows = await this.prisma.$queryRaw<RawConversationResult[]>`
|
|
||||||
SELECT
|
|
||||||
ca.id,
|
|
||||||
ca.workspace_id,
|
|
||||||
ca.session_id,
|
|
||||||
ca.agent_id,
|
|
||||||
ca.messages,
|
|
||||||
ca.message_count,
|
|
||||||
ca.summary,
|
|
||||||
ca.started_at,
|
|
||||||
ca.ended_at,
|
|
||||||
ca.metadata,
|
|
||||||
ca.created_at,
|
|
||||||
ca.updated_at,
|
|
||||||
(1 - (ca.embedding <=> ${embeddingStr}::vector(${EMBEDDING_DIMENSION}))) AS similarity
|
|
||||||
FROM conversation_archives ca
|
|
||||||
WHERE ca.workspace_id = ${workspaceId}::uuid
|
|
||||||
AND ca.embedding IS NOT NULL
|
|
||||||
AND (ca.embedding <=> ${embeddingStr}::vector(${EMBEDDING_DIMENSION})) <= ${distanceThreshold}
|
|
||||||
${agentFilter}
|
|
||||||
ORDER BY ca.embedding <=> ${embeddingStr}::vector(${EMBEDDING_DIMENSION})
|
|
||||||
LIMIT ${limit}
|
|
||||||
`;
|
|
||||||
|
|
||||||
const countResult = await this.prisma.$queryRaw<[{ count: bigint }]>`
|
|
||||||
SELECT COUNT(*) AS count
|
|
||||||
FROM conversation_archives ca
|
|
||||||
WHERE ca.workspace_id = ${workspaceId}::uuid
|
|
||||||
AND ca.embedding IS NOT NULL
|
|
||||||
AND (ca.embedding <=> ${embeddingStr}::vector(${EMBEDDING_DIMENSION})) <= ${distanceThreshold}
|
|
||||||
${agentFilter}
|
|
||||||
`;
|
|
||||||
|
|
||||||
const total = Number(countResult[0].count);
|
|
||||||
|
|
||||||
return {
|
|
||||||
data: rows,
|
|
||||||
pagination: {
|
|
||||||
page: 1,
|
|
||||||
limit,
|
|
||||||
total,
|
|
||||||
totalPages: Math.ceil(total / limit),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* List conversation archives with filtering and pagination.
|
|
||||||
*/
|
|
||||||
async findAll(
|
|
||||||
workspaceId: string,
|
|
||||||
query: ListConversationsDto
|
|
||||||
): Promise<PaginatedConversations<object>> {
|
|
||||||
const page = query.page ?? 1;
|
|
||||||
const limit = query.limit ?? 20;
|
|
||||||
const skip = (page - 1) * limit;
|
|
||||||
|
|
||||||
const where: Prisma.ConversationArchiveWhereInput = {
|
|
||||||
workspaceId,
|
|
||||||
...(query.agentId ? { agentId: query.agentId } : {}),
|
|
||||||
...(query.startedAfter || query.startedBefore
|
|
||||||
? {
|
|
||||||
startedAt: {
|
|
||||||
...(query.startedAfter ? { gte: new Date(query.startedAfter) } : {}),
|
|
||||||
...(query.startedBefore ? { lte: new Date(query.startedBefore) } : {}),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
: {}),
|
|
||||||
};
|
|
||||||
|
|
||||||
const [total, records] = await Promise.all([
|
|
||||||
this.prisma.conversationArchive.count({ where }),
|
|
||||||
this.prisma.conversationArchive.findMany({
|
|
||||||
where,
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
workspaceId: true,
|
|
||||||
sessionId: true,
|
|
||||||
agentId: true,
|
|
||||||
messageCount: true,
|
|
||||||
summary: true,
|
|
||||||
startedAt: true,
|
|
||||||
endedAt: true,
|
|
||||||
metadata: true,
|
|
||||||
createdAt: true,
|
|
||||||
updatedAt: true,
|
|
||||||
},
|
|
||||||
orderBy: { startedAt: "desc" },
|
|
||||||
skip,
|
|
||||||
take: limit,
|
|
||||||
}),
|
|
||||||
]);
|
|
||||||
|
|
||||||
return {
|
|
||||||
data: records,
|
|
||||||
pagination: {
|
|
||||||
page,
|
|
||||||
limit,
|
|
||||||
total,
|
|
||||||
totalPages: Math.ceil(total / limit),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Get a single conversation archive by ID.
|
|
||||||
*/
|
|
||||||
async findOne(workspaceId: string, id: string): Promise<object> {
|
|
||||||
const record = await this.prisma.conversationArchive.findFirst({
|
|
||||||
where: { id, workspaceId },
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
workspaceId: true,
|
|
||||||
sessionId: true,
|
|
||||||
agentId: true,
|
|
||||||
messages: true,
|
|
||||||
messageCount: true,
|
|
||||||
summary: true,
|
|
||||||
startedAt: true,
|
|
||||||
endedAt: true,
|
|
||||||
metadata: true,
|
|
||||||
createdAt: true,
|
|
||||||
updatedAt: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!record) {
|
|
||||||
throw new NotFoundException(`Conversation archive '${id}' not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return record;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Build text content for embedding from summary and messages.
|
|
||||||
*/
|
|
||||||
private buildEmbeddingText(
|
|
||||||
summary: string,
|
|
||||||
messages: { role: string; content: string }[]
|
|
||||||
): string {
|
|
||||||
const messageText = messages.map((m) => `${m.role}: ${m.content}`).join("\n");
|
|
||||||
return `${summary}\n\n${messageText}`.trim();
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Generate embedding and store it on the conversation_archives row.
|
|
||||||
*/
|
|
||||||
private async storeEmbedding(id: string, text: string): Promise<void> {
|
|
||||||
const vector = await this.embedding.generateEmbedding(text);
|
|
||||||
const embeddingStr = `[${vector.join(",")}]`;
|
|
||||||
|
|
||||||
await this.prisma.$executeRaw`
|
|
||||||
UPDATE conversation_archives
|
|
||||||
SET embedding = ${embeddingStr}::vector(${EMBEDDING_DIMENSION}),
|
|
||||||
updated_at = NOW()
|
|
||||||
WHERE id = ${id}::uuid
|
|
||||||
`;
|
|
||||||
|
|
||||||
this.logger.log(`Stored embedding for conversation ${id}`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,3 +0,0 @@
|
|||||||
export { IngestConversationDto, ConversationMessageDto } from "./ingest-conversation.dto";
|
|
||||||
export { SearchConversationDto } from "./search-conversation.dto";
|
|
||||||
export { ListConversationsDto } from "./list-conversations.dto";
|
|
||||||
@@ -1,64 +0,0 @@
|
|||||||
import {
|
|
||||||
IsString,
|
|
||||||
IsArray,
|
|
||||||
IsOptional,
|
|
||||||
IsDateString,
|
|
||||||
MinLength,
|
|
||||||
MaxLength,
|
|
||||||
IsObject,
|
|
||||||
ValidateNested,
|
|
||||||
ArrayMinSize,
|
|
||||||
} from "class-validator";
|
|
||||||
import { Type } from "class-transformer";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Represents a single message in a conversation session
|
|
||||||
*/
|
|
||||||
export class ConversationMessageDto {
|
|
||||||
@IsString()
|
|
||||||
role!: string;
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
@MinLength(1)
|
|
||||||
content!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsDateString()
|
|
||||||
timestamp?: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DTO for ingesting a conversation session log
|
|
||||||
*/
|
|
||||||
export class IngestConversationDto {
|
|
||||||
@IsString()
|
|
||||||
@MinLength(1)
|
|
||||||
@MaxLength(500)
|
|
||||||
sessionId!: string;
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
@MinLength(1)
|
|
||||||
@MaxLength(500)
|
|
||||||
agentId!: string;
|
|
||||||
|
|
||||||
@IsArray()
|
|
||||||
@ArrayMinSize(1)
|
|
||||||
@ValidateNested({ each: true })
|
|
||||||
@Type(() => ConversationMessageDto)
|
|
||||||
messages!: ConversationMessageDto[];
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
@MinLength(1)
|
|
||||||
summary!: string;
|
|
||||||
|
|
||||||
@IsDateString()
|
|
||||||
startedAt!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsDateString()
|
|
||||||
endedAt?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsObject()
|
|
||||||
metadata?: Record<string, unknown>;
|
|
||||||
}
|
|
||||||
@@ -1,33 +0,0 @@
|
|||||||
import { IsString, IsOptional, MaxLength, IsInt, Min, Max, IsDateString } from "class-validator";
|
|
||||||
import { Type } from "class-transformer";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DTO for listing/filtering conversation archives
|
|
||||||
*/
|
|
||||||
export class ListConversationsDto {
|
|
||||||
@IsOptional()
|
|
||||||
@IsString()
|
|
||||||
@MaxLength(500)
|
|
||||||
agentId?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsDateString()
|
|
||||||
startedAfter?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsDateString()
|
|
||||||
startedBefore?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsInt()
|
|
||||||
@Min(1)
|
|
||||||
page?: number;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsInt()
|
|
||||||
@Min(1)
|
|
||||||
@Max(100)
|
|
||||||
limit?: number;
|
|
||||||
}
|
|
||||||
@@ -1,40 +0,0 @@
|
|||||||
import {
|
|
||||||
IsString,
|
|
||||||
IsOptional,
|
|
||||||
MinLength,
|
|
||||||
MaxLength,
|
|
||||||
IsInt,
|
|
||||||
Min,
|
|
||||||
Max,
|
|
||||||
IsNumber,
|
|
||||||
} from "class-validator";
|
|
||||||
import { Type } from "class-transformer";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DTO for semantic search across conversation archives
|
|
||||||
*/
|
|
||||||
export class SearchConversationDto {
|
|
||||||
@IsString()
|
|
||||||
@MinLength(1)
|
|
||||||
@MaxLength(1000)
|
|
||||||
query!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString()
|
|
||||||
@MaxLength(500)
|
|
||||||
agentId?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsInt()
|
|
||||||
@Min(1)
|
|
||||||
@Max(100)
|
|
||||||
limit?: number;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsNumber()
|
|
||||||
@Min(0)
|
|
||||||
@Max(1)
|
|
||||||
similarityThreshold?: number;
|
|
||||||
}
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { ConfigModule } from "@nestjs/config";
|
|
||||||
import { CryptoService } from "./crypto.service";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [ConfigModule],
|
|
||||||
providers: [CryptoService],
|
|
||||||
exports: [CryptoService],
|
|
||||||
})
|
|
||||||
export class CryptoModule {}
|
|
||||||
@@ -1,71 +0,0 @@
|
|||||||
import { describe, it, expect, beforeEach } from "vitest";
|
|
||||||
import { ConfigService } from "@nestjs/config";
|
|
||||||
import { CryptoService } from "./crypto.service";
|
|
||||||
|
|
||||||
function createConfigService(secret?: string): ConfigService {
|
|
||||||
return {
|
|
||||||
get: (key: string) => {
|
|
||||||
if (key === "MOSAIC_SECRET_KEY") {
|
|
||||||
return secret;
|
|
||||||
}
|
|
||||||
return undefined;
|
|
||||||
},
|
|
||||||
} as unknown as ConfigService;
|
|
||||||
}
|
|
||||||
|
|
||||||
describe("CryptoService", () => {
|
|
||||||
let service: CryptoService;
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
service = new CryptoService(createConfigService("this-is-a-test-secret-key-with-32+chars"));
|
|
||||||
});
|
|
||||||
|
|
||||||
it("encrypt -> decrypt roundtrip", () => {
|
|
||||||
const plaintext = "my-secret-api-key";
|
|
||||||
|
|
||||||
const encrypted = service.encrypt(plaintext);
|
|
||||||
const decrypted = service.decrypt(encrypted);
|
|
||||||
|
|
||||||
expect(encrypted.startsWith("enc:")).toBe(true);
|
|
||||||
expect(decrypted).toBe(plaintext);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("decrypt rejects tampered ciphertext", () => {
|
|
||||||
const encrypted = service.encrypt("sensitive-token");
|
|
||||||
const payload = encrypted.slice(4);
|
|
||||||
const bytes = Buffer.from(payload, "base64");
|
|
||||||
|
|
||||||
bytes[bytes.length - 1] = bytes[bytes.length - 1]! ^ 0xff;
|
|
||||||
|
|
||||||
const tampered = `enc:${bytes.toString("base64")}`;
|
|
||||||
|
|
||||||
expect(() => service.decrypt(tampered)).toThrow();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("decrypt rejects non-encrypted string", () => {
|
|
||||||
expect(() => service.decrypt("plain-text-value")).toThrow();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("isEncrypted detects prefix correctly", () => {
|
|
||||||
expect(service.isEncrypted("enc:abc")).toBe(true);
|
|
||||||
expect(service.isEncrypted("ENC:abc")).toBe(false);
|
|
||||||
expect(service.isEncrypted("plain-text")).toBe(false);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("generateToken returns 64-char hex string", () => {
|
|
||||||
const token = service.generateToken();
|
|
||||||
|
|
||||||
expect(token).toMatch(/^[0-9a-f]{64}$/);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("different plaintexts produce different ciphertexts (random IV)", () => {
|
|
||||||
const encryptedA = service.encrypt("value-a");
|
|
||||||
const encryptedB = service.encrypt("value-b");
|
|
||||||
|
|
||||||
expect(encryptedA).not.toBe(encryptedB);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("missing MOSAIC_SECRET_KEY throws on construction", () => {
|
|
||||||
expect(() => new CryptoService(createConfigService(undefined))).toThrow();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,82 +0,0 @@
|
|||||||
import { Injectable } from "@nestjs/common";
|
|
||||||
import { ConfigService } from "@nestjs/config";
|
|
||||||
import { createCipheriv, createDecipheriv, hkdfSync, randomBytes } from "crypto";
|
|
||||||
|
|
||||||
const ALGORITHM = "aes-256-gcm";
|
|
||||||
const ENCRYPTED_PREFIX = "enc:";
|
|
||||||
const IV_LENGTH = 12;
|
|
||||||
const AUTH_TAG_LENGTH = 16;
|
|
||||||
const DERIVED_KEY_LENGTH = 32;
|
|
||||||
const HKDF_SALT = "mosaic.crypto.v1";
|
|
||||||
const HKDF_INFO = "mosaic-db-secret-encryption";
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class CryptoService {
|
|
||||||
private readonly key: Buffer;
|
|
||||||
|
|
||||||
constructor(private readonly config: ConfigService) {
|
|
||||||
const secret = this.config.get<string>("MOSAIC_SECRET_KEY");
|
|
||||||
|
|
||||||
if (!secret) {
|
|
||||||
throw new Error("MOSAIC_SECRET_KEY environment variable is required");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (secret.length < 32) {
|
|
||||||
throw new Error("MOSAIC_SECRET_KEY must be at least 32 characters");
|
|
||||||
}
|
|
||||||
|
|
||||||
this.key = Buffer.from(
|
|
||||||
hkdfSync(
|
|
||||||
"sha256",
|
|
||||||
Buffer.from(secret, "utf8"),
|
|
||||||
Buffer.from(HKDF_SALT, "utf8"),
|
|
||||||
Buffer.from(HKDF_INFO, "utf8"),
|
|
||||||
DERIVED_KEY_LENGTH
|
|
||||||
)
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
encrypt(plaintext: string): string {
|
|
||||||
const iv = randomBytes(IV_LENGTH);
|
|
||||||
const cipher = createCipheriv(ALGORITHM, this.key, iv);
|
|
||||||
const ciphertext = Buffer.concat([cipher.update(plaintext, "utf8"), cipher.final()]);
|
|
||||||
const authTag = cipher.getAuthTag();
|
|
||||||
const payload = Buffer.concat([iv, ciphertext, authTag]).toString("base64");
|
|
||||||
|
|
||||||
return `${ENCRYPTED_PREFIX}${payload}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
decrypt(encrypted: string): string {
|
|
||||||
if (!this.isEncrypted(encrypted)) {
|
|
||||||
throw new Error("Value is not encrypted");
|
|
||||||
}
|
|
||||||
|
|
||||||
const payloadBase64 = encrypted.slice(ENCRYPTED_PREFIX.length);
|
|
||||||
|
|
||||||
try {
|
|
||||||
const payload = Buffer.from(payloadBase64, "base64");
|
|
||||||
if (payload.length < IV_LENGTH + AUTH_TAG_LENGTH) {
|
|
||||||
throw new Error("Encrypted payload is too short");
|
|
||||||
}
|
|
||||||
|
|
||||||
const iv = payload.subarray(0, IV_LENGTH);
|
|
||||||
const authTag = payload.subarray(payload.length - AUTH_TAG_LENGTH);
|
|
||||||
const ciphertext = payload.subarray(IV_LENGTH, payload.length - AUTH_TAG_LENGTH);
|
|
||||||
|
|
||||||
const decipher = createDecipheriv(ALGORITHM, this.key, iv);
|
|
||||||
decipher.setAuthTag(authTag);
|
|
||||||
|
|
||||||
return Buffer.concat([decipher.update(ciphertext), decipher.final()]).toString("utf8");
|
|
||||||
} catch {
|
|
||||||
throw new Error("Failed to decrypt value");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
isEncrypted(value: string): boolean {
|
|
||||||
return value.startsWith(ENCRYPTED_PREFIX);
|
|
||||||
}
|
|
||||||
|
|
||||||
generateToken(): string {
|
|
||||||
return randomBytes(32).toString("hex");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,33 +0,0 @@
|
|||||||
import { IsObject, IsOptional, IsString, IsUUID, MaxLength, MinLength } from "class-validator";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DTO for creating a finding
|
|
||||||
*/
|
|
||||||
export class CreateFindingDto {
|
|
||||||
@IsOptional()
|
|
||||||
@IsUUID("4", { message: "taskId must be a valid UUID" })
|
|
||||||
taskId?: string;
|
|
||||||
|
|
||||||
@IsString({ message: "agentId must be a string" })
|
|
||||||
@MinLength(1, { message: "agentId must not be empty" })
|
|
||||||
@MaxLength(255, { message: "agentId must not exceed 255 characters" })
|
|
||||||
agentId!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "type must be a string" })
|
|
||||||
@MinLength(1, { message: "type must not be empty" })
|
|
||||||
@MaxLength(100, { message: "type must not exceed 100 characters" })
|
|
||||||
type!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "title must be a string" })
|
|
||||||
@MinLength(1, { message: "title must not be empty" })
|
|
||||||
@MaxLength(255, { message: "title must not exceed 255 characters" })
|
|
||||||
title!: string;
|
|
||||||
|
|
||||||
@IsObject({ message: "data must be an object" })
|
|
||||||
data!: Record<string, unknown>;
|
|
||||||
|
|
||||||
@IsString({ message: "summary must be a string" })
|
|
||||||
@MinLength(1, { message: "summary must not be empty" })
|
|
||||||
@MaxLength(20000, { message: "summary must not exceed 20000 characters" })
|
|
||||||
summary!: string;
|
|
||||||
}
|
|
||||||
@@ -1,3 +0,0 @@
|
|||||||
export { CreateFindingDto } from "./create-finding.dto";
|
|
||||||
export { QueryFindingsDto } from "./query-findings.dto";
|
|
||||||
export { SearchFindingsDto } from "./search-findings.dto";
|
|
||||||
@@ -1,32 +0,0 @@
|
|||||||
import { Type } from "class-transformer";
|
|
||||||
import { IsInt, IsOptional, IsString, IsUUID, Max, Min } from "class-validator";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DTO for querying findings with filters and pagination
|
|
||||||
*/
|
|
||||||
export class QueryFindingsDto {
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsInt({ message: "page must be an integer" })
|
|
||||||
@Min(1, { message: "page must be at least 1" })
|
|
||||||
page?: number;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsInt({ message: "limit must be an integer" })
|
|
||||||
@Min(1, { message: "limit must be at least 1" })
|
|
||||||
@Max(100, { message: "limit must not exceed 100" })
|
|
||||||
limit?: number;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "agentId must be a string" })
|
|
||||||
agentId?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "type must be a string" })
|
|
||||||
type?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsUUID("4", { message: "taskId must be a valid UUID" })
|
|
||||||
taskId?: string;
|
|
||||||
}
|
|
||||||
@@ -1,52 +0,0 @@
|
|||||||
import { Type } from "class-transformer";
|
|
||||||
import {
|
|
||||||
IsInt,
|
|
||||||
IsNumber,
|
|
||||||
IsOptional,
|
|
||||||
IsString,
|
|
||||||
IsUUID,
|
|
||||||
Max,
|
|
||||||
MaxLength,
|
|
||||||
Min,
|
|
||||||
} from "class-validator";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DTO for finding semantic similarity search
|
|
||||||
*/
|
|
||||||
export class SearchFindingsDto {
|
|
||||||
@IsString({ message: "query must be a string" })
|
|
||||||
@MaxLength(1000, { message: "query must not exceed 1000 characters" })
|
|
||||||
query!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsInt({ message: "page must be an integer" })
|
|
||||||
@Min(1, { message: "page must be at least 1" })
|
|
||||||
page?: number;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsInt({ message: "limit must be an integer" })
|
|
||||||
@Min(1, { message: "limit must be at least 1" })
|
|
||||||
@Max(100, { message: "limit must not exceed 100" })
|
|
||||||
limit?: number;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@Type(() => Number)
|
|
||||||
@IsNumber({}, { message: "similarityThreshold must be a number" })
|
|
||||||
@Min(0, { message: "similarityThreshold must be at least 0" })
|
|
||||||
@Max(1, { message: "similarityThreshold must not exceed 1" })
|
|
||||||
similarityThreshold?: number;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "agentId must be a string" })
|
|
||||||
agentId?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "type must be a string" })
|
|
||||||
type?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsUUID("4", { message: "taskId must be a valid UUID" })
|
|
||||||
taskId?: string;
|
|
||||||
}
|
|
||||||
@@ -1,195 +0,0 @@
|
|||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
||||||
import { FindingsController } from "./findings.controller";
|
|
||||||
import { FindingsService } from "./findings.service";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
|
|
||||||
import { CreateFindingDto, QueryFindingsDto, SearchFindingsDto } from "./dto";
|
|
||||||
|
|
||||||
describe("FindingsController", () => {
|
|
||||||
let controller: FindingsController;
|
|
||||||
let service: FindingsService;
|
|
||||||
|
|
||||||
const mockFindingsService = {
|
|
||||||
create: vi.fn(),
|
|
||||||
findAll: vi.fn(),
|
|
||||||
findOne: vi.fn(),
|
|
||||||
search: vi.fn(),
|
|
||||||
remove: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockAuthGuard = {
|
|
||||||
canActivate: vi.fn(() => true),
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockWorkspaceGuard = {
|
|
||||||
canActivate: vi.fn(() => true),
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockPermissionGuard = {
|
|
||||||
canActivate: vi.fn(() => true),
|
|
||||||
};
|
|
||||||
|
|
||||||
const workspaceId = "550e8400-e29b-41d4-a716-446655440001";
|
|
||||||
const findingId = "550e8400-e29b-41d4-a716-446655440002";
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
const module: TestingModule = await Test.createTestingModule({
|
|
||||||
controllers: [FindingsController],
|
|
||||||
providers: [
|
|
||||||
{
|
|
||||||
provide: FindingsService,
|
|
||||||
useValue: mockFindingsService,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
})
|
|
||||||
.overrideGuard(AuthGuard)
|
|
||||||
.useValue(mockAuthGuard)
|
|
||||||
.overrideGuard(WorkspaceGuard)
|
|
||||||
.useValue(mockWorkspaceGuard)
|
|
||||||
.overrideGuard(PermissionGuard)
|
|
||||||
.useValue(mockPermissionGuard)
|
|
||||||
.compile();
|
|
||||||
|
|
||||||
controller = module.get<FindingsController>(FindingsController);
|
|
||||||
service = module.get<FindingsService>(FindingsService);
|
|
||||||
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should be defined", () => {
|
|
||||||
expect(controller).toBeDefined();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("create", () => {
|
|
||||||
it("should create a finding", async () => {
|
|
||||||
const createDto: CreateFindingDto = {
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
};
|
|
||||||
|
|
||||||
const createdFinding = {
|
|
||||||
id: findingId,
|
|
||||||
workspaceId,
|
|
||||||
taskId: null,
|
|
||||||
...createDto,
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
|
|
||||||
mockFindingsService.create.mockResolvedValue(createdFinding);
|
|
||||||
|
|
||||||
const result = await controller.create(createDto, workspaceId);
|
|
||||||
|
|
||||||
expect(result).toEqual(createdFinding);
|
|
||||||
expect(service.create).toHaveBeenCalledWith(workspaceId, createDto);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findAll", () => {
|
|
||||||
it("should return paginated findings", async () => {
|
|
||||||
const query: QueryFindingsDto = {
|
|
||||||
page: 1,
|
|
||||||
limit: 10,
|
|
||||||
type: "security",
|
|
||||||
};
|
|
||||||
|
|
||||||
const response = {
|
|
||||||
data: [],
|
|
||||||
meta: {
|
|
||||||
total: 0,
|
|
||||||
page: 1,
|
|
||||||
limit: 10,
|
|
||||||
totalPages: 0,
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
mockFindingsService.findAll.mockResolvedValue(response);
|
|
||||||
|
|
||||||
const result = await controller.findAll(query, workspaceId);
|
|
||||||
|
|
||||||
expect(result).toEqual(response);
|
|
||||||
expect(service.findAll).toHaveBeenCalledWith(workspaceId, query);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findOne", () => {
|
|
||||||
it("should return a finding", async () => {
|
|
||||||
const finding = {
|
|
||||||
id: findingId,
|
|
||||||
workspaceId,
|
|
||||||
taskId: null,
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
|
|
||||||
mockFindingsService.findOne.mockResolvedValue(finding);
|
|
||||||
|
|
||||||
const result = await controller.findOne(findingId, workspaceId);
|
|
||||||
|
|
||||||
expect(result).toEqual(finding);
|
|
||||||
expect(service.findOne).toHaveBeenCalledWith(findingId, workspaceId);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("search", () => {
|
|
||||||
it("should perform semantic search", async () => {
|
|
||||||
const searchDto: SearchFindingsDto = {
|
|
||||||
query: "sql injection",
|
|
||||||
limit: 5,
|
|
||||||
};
|
|
||||||
|
|
||||||
const response = {
|
|
||||||
data: [
|
|
||||||
{
|
|
||||||
id: findingId,
|
|
||||||
workspaceId,
|
|
||||||
taskId: null,
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
score: 0.91,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
meta: {
|
|
||||||
total: 1,
|
|
||||||
page: 1,
|
|
||||||
limit: 5,
|
|
||||||
totalPages: 1,
|
|
||||||
},
|
|
||||||
query: "sql injection",
|
|
||||||
};
|
|
||||||
|
|
||||||
mockFindingsService.search.mockResolvedValue(response);
|
|
||||||
|
|
||||||
const result = await controller.search(searchDto, workspaceId);
|
|
||||||
|
|
||||||
expect(result).toEqual(response);
|
|
||||||
expect(service.search).toHaveBeenCalledWith(workspaceId, searchDto);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("remove", () => {
|
|
||||||
it("should delete a finding", async () => {
|
|
||||||
const response = { message: "Finding deleted successfully" };
|
|
||||||
mockFindingsService.remove.mockResolvedValue(response);
|
|
||||||
|
|
||||||
const result = await controller.remove(findingId, workspaceId);
|
|
||||||
|
|
||||||
expect(result).toEqual(response);
|
|
||||||
expect(service.remove).toHaveBeenCalledWith(findingId, workspaceId);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,81 +0,0 @@
|
|||||||
import { Body, Controller, Delete, Get, Param, Post, Query, UseGuards } from "@nestjs/common";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
|
|
||||||
import { Workspace, Permission, RequirePermission } from "../common/decorators";
|
|
||||||
import { CreateFindingDto, QueryFindingsDto, SearchFindingsDto } from "./dto";
|
|
||||||
import {
|
|
||||||
FindingsService,
|
|
||||||
FindingsSearchResponse,
|
|
||||||
PaginatedFindingsResponse,
|
|
||||||
} from "./findings.service";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Controller for findings endpoints
|
|
||||||
* All endpoints require authentication and workspace context
|
|
||||||
*/
|
|
||||||
@Controller("findings")
|
|
||||||
@UseGuards(AuthGuard, WorkspaceGuard, PermissionGuard)
|
|
||||||
export class FindingsController {
|
|
||||||
constructor(private readonly findingsService: FindingsService) {}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* POST /api/findings
|
|
||||||
* Create a new finding and embed its summary
|
|
||||||
* Requires: MEMBER role or higher
|
|
||||||
*/
|
|
||||||
@Post()
|
|
||||||
@RequirePermission(Permission.WORKSPACE_MEMBER)
|
|
||||||
async create(@Body() createFindingDto: CreateFindingDto, @Workspace() workspaceId: string) {
|
|
||||||
return this.findingsService.create(workspaceId, createFindingDto);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* GET /api/findings
|
|
||||||
* Get paginated findings with optional filters
|
|
||||||
* Requires: Any workspace member
|
|
||||||
*/
|
|
||||||
@Get()
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async findAll(
|
|
||||||
@Query() query: QueryFindingsDto,
|
|
||||||
@Workspace() workspaceId: string
|
|
||||||
): Promise<PaginatedFindingsResponse> {
|
|
||||||
return this.findingsService.findAll(workspaceId, query);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* GET /api/findings/:id
|
|
||||||
* Get a single finding by ID
|
|
||||||
* Requires: Any workspace member
|
|
||||||
*/
|
|
||||||
@Get(":id")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async findOne(@Param("id") id: string, @Workspace() workspaceId: string) {
|
|
||||||
return this.findingsService.findOne(id, workspaceId);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* POST /api/findings/search
|
|
||||||
* Semantic search findings by vector similarity
|
|
||||||
* Requires: Any workspace member
|
|
||||||
*/
|
|
||||||
@Post("search")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
||||||
async search(
|
|
||||||
@Body() searchDto: SearchFindingsDto,
|
|
||||||
@Workspace() workspaceId: string
|
|
||||||
): Promise<FindingsSearchResponse> {
|
|
||||||
return this.findingsService.search(workspaceId, searchDto);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* DELETE /api/findings/:id
|
|
||||||
* Delete a finding
|
|
||||||
* Requires: ADMIN role or higher
|
|
||||||
*/
|
|
||||||
@Delete(":id")
|
|
||||||
@RequirePermission(Permission.WORKSPACE_ADMIN)
|
|
||||||
async remove(@Param("id") id: string, @Workspace() workspaceId: string) {
|
|
||||||
return this.findingsService.remove(id, workspaceId);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,226 +0,0 @@
|
|||||||
import { beforeAll, beforeEach, describe, expect, it, afterAll, vi } from "vitest";
|
|
||||||
import { randomUUID as uuid } from "crypto";
|
|
||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { BadRequestException, NotFoundException } from "@nestjs/common";
|
|
||||||
import { PrismaClient, Prisma } from "@prisma/client";
|
|
||||||
import { FindingsService } from "./findings.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { EmbeddingService } from "../knowledge/services/embedding.service";
|
|
||||||
|
|
||||||
const shouldRunDbIntegrationTests =
|
|
||||||
process.env.RUN_DB_TESTS === "true" && Boolean(process.env.DATABASE_URL);
|
|
||||||
const describeFn = shouldRunDbIntegrationTests ? describe : describe.skip;
|
|
||||||
|
|
||||||
const EMBEDDING_DIMENSION = 1536;
|
|
||||||
|
|
||||||
function vector(value: number): number[] {
|
|
||||||
return Array.from({ length: EMBEDDING_DIMENSION }, () => value);
|
|
||||||
}
|
|
||||||
|
|
||||||
function toVectorLiteral(input: number[]): string {
|
|
||||||
return `[${input.join(",")}]`;
|
|
||||||
}
|
|
||||||
|
|
||||||
describeFn("FindingsService Integration", () => {
|
|
||||||
let moduleRef: TestingModule;
|
|
||||||
let prisma: PrismaClient;
|
|
||||||
let service: FindingsService;
|
|
||||||
let workspaceId: string;
|
|
||||||
let ownerId: string;
|
|
||||||
let setupComplete = false;
|
|
||||||
|
|
||||||
const embeddingServiceMock = {
|
|
||||||
isConfigured: vi.fn(),
|
|
||||||
generateEmbedding: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeAll(async () => {
|
|
||||||
prisma = new PrismaClient();
|
|
||||||
await prisma.$connect();
|
|
||||||
|
|
||||||
const workspace = await prisma.workspace.create({
|
|
||||||
data: {
|
|
||||||
name: `Findings Integration ${Date.now()}`,
|
|
||||||
owner: {
|
|
||||||
create: {
|
|
||||||
email: `findings-integration-${Date.now()}@example.com`,
|
|
||||||
name: "Findings Integration Owner",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
workspaceId = workspace.id;
|
|
||||||
ownerId = workspace.ownerId;
|
|
||||||
|
|
||||||
moduleRef = await Test.createTestingModule({
|
|
||||||
providers: [
|
|
||||||
FindingsService,
|
|
||||||
{
|
|
||||||
provide: PrismaService,
|
|
||||||
useValue: prisma,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
provide: EmbeddingService,
|
|
||||||
useValue: embeddingServiceMock,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}).compile();
|
|
||||||
|
|
||||||
service = moduleRef.get<FindingsService>(FindingsService);
|
|
||||||
setupComplete = true;
|
|
||||||
});
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
embeddingServiceMock.isConfigured.mockReturnValue(false);
|
|
||||||
});
|
|
||||||
|
|
||||||
afterAll(async () => {
|
|
||||||
if (!prisma) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (workspaceId) {
|
|
||||||
await prisma.finding.deleteMany({ where: { workspaceId } });
|
|
||||||
await prisma.workspace.deleteMany({ where: { id: workspaceId } });
|
|
||||||
}
|
|
||||||
if (ownerId) {
|
|
||||||
await prisma.user.deleteMany({ where: { id: ownerId } });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (moduleRef) {
|
|
||||||
await moduleRef.close();
|
|
||||||
}
|
|
||||||
await prisma.$disconnect();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("creates, lists, fetches, and deletes findings", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const created = await service.create(workspaceId, {
|
|
||||||
agentId: "agent-findings-crud",
|
|
||||||
type: "security",
|
|
||||||
title: "Unescaped SQL fragment",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential injection risk in dynamic query path.",
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(created.id).toBeDefined();
|
|
||||||
expect(created.workspaceId).toBe(workspaceId);
|
|
||||||
expect(created.taskId).toBeNull();
|
|
||||||
|
|
||||||
const listed = await service.findAll(workspaceId, {
|
|
||||||
page: 1,
|
|
||||||
limit: 10,
|
|
||||||
agentId: "agent-findings-crud",
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(listed.meta.total).toBeGreaterThanOrEqual(1);
|
|
||||||
expect(listed.data.some((row) => row.id === created.id)).toBe(true);
|
|
||||||
|
|
||||||
const found = await service.findOne(created.id, workspaceId);
|
|
||||||
expect(found.id).toBe(created.id);
|
|
||||||
expect(found.title).toBe("Unescaped SQL fragment");
|
|
||||||
|
|
||||||
await expect(service.findOne(created.id, uuid())).rejects.toThrow(NotFoundException);
|
|
||||||
|
|
||||||
await expect(service.remove(created.id, workspaceId)).resolves.toEqual({
|
|
||||||
message: "Finding deleted successfully",
|
|
||||||
});
|
|
||||||
|
|
||||||
await expect(service.findOne(created.id, workspaceId)).rejects.toThrow(NotFoundException);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("rejects create when taskId does not exist in workspace", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
await expect(
|
|
||||||
service.create(workspaceId, {
|
|
||||||
taskId: uuid(),
|
|
||||||
agentId: "agent-findings-missing-task",
|
|
||||||
type: "bug",
|
|
||||||
title: "Invalid task id",
|
|
||||||
data: { source: "integration-test" },
|
|
||||||
summary: "Should fail when task relation is not found.",
|
|
||||||
})
|
|
||||||
).rejects.toThrow(NotFoundException);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("rejects vector search when embeddings are disabled", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
embeddingServiceMock.isConfigured.mockReturnValue(false);
|
|
||||||
|
|
||||||
await expect(
|
|
||||||
service.search(workspaceId, {
|
|
||||||
query: "security issue",
|
|
||||||
})
|
|
||||||
).rejects.toThrow(BadRequestException);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("searches findings by vector similarity with filters", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const near = vector(0.01);
|
|
||||||
const far = vector(0.9);
|
|
||||||
|
|
||||||
const matchedFinding = await prisma.finding.create({
|
|
||||||
data: {
|
|
||||||
workspaceId,
|
|
||||||
agentId: "agent-findings-search-a",
|
|
||||||
type: "incident",
|
|
||||||
title: "Authentication bypass",
|
|
||||||
data: { score: 0.9 } as Prisma.InputJsonValue,
|
|
||||||
summary: "Bypass risk found in login checks.",
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const otherFinding = await prisma.finding.create({
|
|
||||||
data: {
|
|
||||||
workspaceId,
|
|
||||||
agentId: "agent-findings-search-b",
|
|
||||||
type: "incident",
|
|
||||||
title: "Retry timeout",
|
|
||||||
data: { score: 0.2 } as Prisma.InputJsonValue,
|
|
||||||
summary: "Timeout issue in downstream retries.",
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
await prisma.$executeRaw`
|
|
||||||
UPDATE findings
|
|
||||||
SET embedding = ${toVectorLiteral(near)}::vector(1536)
|
|
||||||
WHERE id = ${matchedFinding.id}::uuid
|
|
||||||
`;
|
|
||||||
await prisma.$executeRaw`
|
|
||||||
UPDATE findings
|
|
||||||
SET embedding = ${toVectorLiteral(far)}::vector(1536)
|
|
||||||
WHERE id = ${otherFinding.id}::uuid
|
|
||||||
`;
|
|
||||||
|
|
||||||
embeddingServiceMock.isConfigured.mockReturnValue(true);
|
|
||||||
embeddingServiceMock.generateEmbedding.mockResolvedValue(near);
|
|
||||||
|
|
||||||
const result = await service.search(workspaceId, {
|
|
||||||
query: "authentication bypass risk",
|
|
||||||
agentId: "agent-findings-search-a",
|
|
||||||
limit: 10,
|
|
||||||
similarityThreshold: 0,
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.query).toBe("authentication bypass risk");
|
|
||||||
expect(result.meta.total).toBe(1);
|
|
||||||
expect(result.data).toHaveLength(1);
|
|
||||||
expect(result.data[0]?.id).toBe(matchedFinding.id);
|
|
||||||
expect(result.data[0]?.agentId).toBe("agent-findings-search-a");
|
|
||||||
expect(result.data.find((row) => row.id === otherFinding.id)).toBeUndefined();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,14 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { AuthModule } from "../auth/auth.module";
|
|
||||||
import { KnowledgeModule } from "../knowledge/knowledge.module";
|
|
||||||
import { FindingsController } from "./findings.controller";
|
|
||||||
import { FindingsService } from "./findings.service";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [PrismaModule, AuthModule, KnowledgeModule],
|
|
||||||
controllers: [FindingsController],
|
|
||||||
providers: [FindingsService],
|
|
||||||
exports: [FindingsService],
|
|
||||||
})
|
|
||||||
export class FindingsModule {}
|
|
||||||
@@ -1,300 +0,0 @@
|
|||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
||||||
import { BadRequestException, NotFoundException } from "@nestjs/common";
|
|
||||||
import { FindingsService } from "./findings.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { EmbeddingService } from "../knowledge/services/embedding.service";
|
|
||||||
|
|
||||||
describe("FindingsService", () => {
|
|
||||||
let service: FindingsService;
|
|
||||||
let prisma: PrismaService;
|
|
||||||
let embeddingService: EmbeddingService;
|
|
||||||
|
|
||||||
const mockWorkspaceId = "550e8400-e29b-41d4-a716-446655440001";
|
|
||||||
const mockFindingId = "550e8400-e29b-41d4-a716-446655440002";
|
|
||||||
const mockTaskId = "550e8400-e29b-41d4-a716-446655440003";
|
|
||||||
|
|
||||||
const mockPrismaService = {
|
|
||||||
finding: {
|
|
||||||
create: vi.fn(),
|
|
||||||
findMany: vi.fn(),
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
count: vi.fn(),
|
|
||||||
delete: vi.fn(),
|
|
||||||
},
|
|
||||||
agentTask: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
},
|
|
||||||
$queryRaw: vi.fn(),
|
|
||||||
$executeRaw: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockEmbeddingService = {
|
|
||||||
isConfigured: vi.fn(),
|
|
||||||
generateEmbedding: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
const module: TestingModule = await Test.createTestingModule({
|
|
||||||
providers: [
|
|
||||||
FindingsService,
|
|
||||||
{
|
|
||||||
provide: PrismaService,
|
|
||||||
useValue: mockPrismaService,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
provide: EmbeddingService,
|
|
||||||
useValue: mockEmbeddingService,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}).compile();
|
|
||||||
|
|
||||||
service = module.get<FindingsService>(FindingsService);
|
|
||||||
prisma = module.get<PrismaService>(PrismaService);
|
|
||||||
embeddingService = module.get<EmbeddingService>(EmbeddingService);
|
|
||||||
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should be defined", () => {
|
|
||||||
expect(service).toBeDefined();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("create", () => {
|
|
||||||
it("should create a finding and store embedding when configured", async () => {
|
|
||||||
const createDto = {
|
|
||||||
taskId: mockTaskId,
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
};
|
|
||||||
|
|
||||||
const createdFinding = {
|
|
||||||
id: mockFindingId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
...createDto,
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
|
|
||||||
mockPrismaService.agentTask.findUnique.mockResolvedValue({
|
|
||||||
id: mockTaskId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
});
|
|
||||||
mockPrismaService.finding.create.mockResolvedValue(createdFinding);
|
|
||||||
mockPrismaService.finding.findUnique.mockResolvedValue(createdFinding);
|
|
||||||
mockEmbeddingService.isConfigured.mockReturnValue(true);
|
|
||||||
mockEmbeddingService.generateEmbedding.mockResolvedValue([0.1, 0.2, 0.3]);
|
|
||||||
mockPrismaService.$executeRaw.mockResolvedValue(1);
|
|
||||||
|
|
||||||
const result = await service.create(mockWorkspaceId, createDto);
|
|
||||||
|
|
||||||
expect(result).toEqual(createdFinding);
|
|
||||||
expect(prisma.finding.create).toHaveBeenCalledWith({
|
|
||||||
data: expect.objectContaining({
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
taskId: mockTaskId,
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
}),
|
|
||||||
select: expect.any(Object),
|
|
||||||
});
|
|
||||||
expect(embeddingService.generateEmbedding).toHaveBeenCalledWith(createDto.summary);
|
|
||||||
expect(prisma.$executeRaw).toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should create a finding without embedding when not configured", async () => {
|
|
||||||
const createDto = {
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
};
|
|
||||||
|
|
||||||
const createdFinding = {
|
|
||||||
id: mockFindingId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
taskId: null,
|
|
||||||
...createDto,
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
|
|
||||||
mockPrismaService.finding.create.mockResolvedValue(createdFinding);
|
|
||||||
mockEmbeddingService.isConfigured.mockReturnValue(false);
|
|
||||||
|
|
||||||
const result = await service.create(mockWorkspaceId, createDto);
|
|
||||||
|
|
||||||
expect(result).toEqual(createdFinding);
|
|
||||||
expect(embeddingService.generateEmbedding).not.toHaveBeenCalled();
|
|
||||||
expect(prisma.$executeRaw).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findAll", () => {
|
|
||||||
it("should return paginated findings with filters", async () => {
|
|
||||||
const findings = [
|
|
||||||
{
|
|
||||||
id: mockFindingId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
taskId: null,
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
mockPrismaService.finding.findMany.mockResolvedValue(findings);
|
|
||||||
mockPrismaService.finding.count.mockResolvedValue(1);
|
|
||||||
|
|
||||||
const result = await service.findAll(mockWorkspaceId, {
|
|
||||||
page: 1,
|
|
||||||
limit: 10,
|
|
||||||
type: "security",
|
|
||||||
agentId: "research-agent",
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
data: findings,
|
|
||||||
meta: {
|
|
||||||
total: 1,
|
|
||||||
page: 1,
|
|
||||||
limit: 10,
|
|
||||||
totalPages: 1,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
expect(prisma.finding.findMany).toHaveBeenCalledWith(
|
|
||||||
expect.objectContaining({
|
|
||||||
where: {
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
type: "security",
|
|
||||||
agentId: "research-agent",
|
|
||||||
},
|
|
||||||
})
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findOne", () => {
|
|
||||||
it("should return a finding", async () => {
|
|
||||||
const finding = {
|
|
||||||
id: mockFindingId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
taskId: null,
|
|
||||||
agentId: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
createdAt: new Date(),
|
|
||||||
updatedAt: new Date(),
|
|
||||||
};
|
|
||||||
|
|
||||||
mockPrismaService.finding.findUnique.mockResolvedValue(finding);
|
|
||||||
|
|
||||||
const result = await service.findOne(mockFindingId, mockWorkspaceId);
|
|
||||||
|
|
||||||
expect(result).toEqual(finding);
|
|
||||||
expect(prisma.finding.findUnique).toHaveBeenCalledWith({
|
|
||||||
where: {
|
|
||||||
id: mockFindingId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
},
|
|
||||||
select: expect.any(Object),
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should throw when finding does not exist", async () => {
|
|
||||||
mockPrismaService.finding.findUnique.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(service.findOne(mockFindingId, mockWorkspaceId)).rejects.toThrow(
|
|
||||||
NotFoundException
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("search", () => {
|
|
||||||
it("should throw BadRequestException when embeddings are not configured", async () => {
|
|
||||||
mockEmbeddingService.isConfigured.mockReturnValue(false);
|
|
||||||
|
|
||||||
await expect(
|
|
||||||
service.search(mockWorkspaceId, {
|
|
||||||
query: "sql injection",
|
|
||||||
})
|
|
||||||
).rejects.toThrow(BadRequestException);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should return similarity-ranked search results", async () => {
|
|
||||||
mockEmbeddingService.isConfigured.mockReturnValue(true);
|
|
||||||
mockEmbeddingService.generateEmbedding.mockResolvedValue([0.1, 0.2, 0.3]);
|
|
||||||
mockPrismaService.$queryRaw
|
|
||||||
.mockResolvedValueOnce([
|
|
||||||
{
|
|
||||||
id: mockFindingId,
|
|
||||||
workspace_id: mockWorkspaceId,
|
|
||||||
task_id: null,
|
|
||||||
agent_id: "research-agent",
|
|
||||||
type: "security",
|
|
||||||
title: "SQL injection risk",
|
|
||||||
data: { severity: "high" },
|
|
||||||
summary: "Potential SQL injection in search endpoint.",
|
|
||||||
created_at: new Date(),
|
|
||||||
updated_at: new Date(),
|
|
||||||
score: 0.91,
|
|
||||||
},
|
|
||||||
])
|
|
||||||
.mockResolvedValueOnce([{ count: BigInt(1) }]);
|
|
||||||
|
|
||||||
const result = await service.search(mockWorkspaceId, {
|
|
||||||
query: "sql injection",
|
|
||||||
page: 1,
|
|
||||||
limit: 5,
|
|
||||||
similarityThreshold: 0.5,
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(result.query).toBe("sql injection");
|
|
||||||
expect(result.data).toHaveLength(1);
|
|
||||||
expect(result.data[0].score).toBe(0.91);
|
|
||||||
expect(result.meta.total).toBe(1);
|
|
||||||
expect(prisma.$queryRaw).toHaveBeenCalledTimes(2);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("remove", () => {
|
|
||||||
it("should delete a finding", async () => {
|
|
||||||
mockPrismaService.finding.findUnique.mockResolvedValue({
|
|
||||||
id: mockFindingId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
});
|
|
||||||
mockPrismaService.finding.delete.mockResolvedValue({
|
|
||||||
id: mockFindingId,
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await service.remove(mockFindingId, mockWorkspaceId);
|
|
||||||
|
|
||||||
expect(result).toEqual({ message: "Finding deleted successfully" });
|
|
||||||
expect(prisma.finding.delete).toHaveBeenCalledWith({
|
|
||||||
where: {
|
|
||||||
id: mockFindingId,
|
|
||||||
workspaceId: mockWorkspaceId,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("should throw when finding does not exist", async () => {
|
|
||||||
mockPrismaService.finding.findUnique.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(service.remove(mockFindingId, mockWorkspaceId)).rejects.toThrow(
|
|
||||||
NotFoundException
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,337 +0,0 @@
|
|||||||
import { BadRequestException, Injectable, Logger, NotFoundException } from "@nestjs/common";
|
|
||||||
import { Prisma } from "@prisma/client";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { EmbeddingService } from "../knowledge/services/embedding.service";
|
|
||||||
import type { CreateFindingDto, QueryFindingsDto, SearchFindingsDto } from "./dto";
|
|
||||||
|
|
||||||
const findingSelect = {
|
|
||||||
id: true,
|
|
||||||
workspaceId: true,
|
|
||||||
taskId: true,
|
|
||||||
agentId: true,
|
|
||||||
type: true,
|
|
||||||
title: true,
|
|
||||||
data: true,
|
|
||||||
summary: true,
|
|
||||||
createdAt: true,
|
|
||||||
updatedAt: true,
|
|
||||||
} satisfies Prisma.FindingSelect;
|
|
||||||
|
|
||||||
type FindingRecord = Prisma.FindingGetPayload<{ select: typeof findingSelect }>;
|
|
||||||
|
|
||||||
interface RawFindingSearchResult {
|
|
||||||
id: string;
|
|
||||||
workspace_id: string;
|
|
||||||
task_id: string | null;
|
|
||||||
agent_id: string;
|
|
||||||
type: string;
|
|
||||||
title: string;
|
|
||||||
data: Prisma.JsonValue;
|
|
||||||
summary: string;
|
|
||||||
created_at: Date;
|
|
||||||
updated_at: Date;
|
|
||||||
score: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface FindingSearchResult extends FindingRecord {
|
|
||||||
score: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface PaginatedMeta {
|
|
||||||
total: number;
|
|
||||||
page: number;
|
|
||||||
limit: number;
|
|
||||||
totalPages: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface PaginatedFindingsResponse {
|
|
||||||
data: FindingRecord[];
|
|
||||||
meta: PaginatedMeta;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface FindingsSearchResponse {
|
|
||||||
data: FindingSearchResult[];
|
|
||||||
meta: PaginatedMeta;
|
|
||||||
query: string;
|
|
||||||
similarityThreshold: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Service for managing structured findings with vector search support
|
|
||||||
*/
|
|
||||||
@Injectable()
|
|
||||||
export class FindingsService {
|
|
||||||
private readonly logger = new Logger(FindingsService.name);
|
|
||||||
private readonly defaultSimilarityThreshold: number;
|
|
||||||
|
|
||||||
constructor(
|
|
||||||
private readonly prisma: PrismaService,
|
|
||||||
private readonly embeddingService: EmbeddingService
|
|
||||||
) {
|
|
||||||
const parsedThreshold = Number.parseFloat(process.env.FINDINGS_SIMILARITY_THRESHOLD ?? "0.5");
|
|
||||||
|
|
||||||
this.defaultSimilarityThreshold =
|
|
||||||
Number.isFinite(parsedThreshold) && parsedThreshold >= 0 && parsedThreshold <= 1
|
|
||||||
? parsedThreshold
|
|
||||||
: 0.5;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Create a finding and generate its embedding from the summary when available
|
|
||||||
*/
|
|
||||||
async create(workspaceId: string, createFindingDto: CreateFindingDto): Promise<FindingRecord> {
|
|
||||||
if (createFindingDto.taskId) {
|
|
||||||
const task = await this.prisma.agentTask.findUnique({
|
|
||||||
where: {
|
|
||||||
id: createFindingDto.taskId,
|
|
||||||
workspaceId,
|
|
||||||
},
|
|
||||||
select: { id: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!task) {
|
|
||||||
throw new NotFoundException(`Agent task with ID ${createFindingDto.taskId} not found`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const createInput: Prisma.FindingUncheckedCreateInput = {
|
|
||||||
workspaceId,
|
|
||||||
agentId: createFindingDto.agentId,
|
|
||||||
type: createFindingDto.type,
|
|
||||||
title: createFindingDto.title,
|
|
||||||
data: createFindingDto.data as Prisma.InputJsonValue,
|
|
||||||
summary: createFindingDto.summary,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (createFindingDto.taskId) {
|
|
||||||
createInput.taskId = createFindingDto.taskId;
|
|
||||||
}
|
|
||||||
|
|
||||||
const finding = await this.prisma.finding.create({
|
|
||||||
data: createInput,
|
|
||||||
select: findingSelect,
|
|
||||||
});
|
|
||||||
|
|
||||||
await this.generateAndStoreEmbedding(finding.id, workspaceId, finding.summary);
|
|
||||||
|
|
||||||
if (this.embeddingService.isConfigured()) {
|
|
||||||
return this.findOne(finding.id, workspaceId);
|
|
||||||
}
|
|
||||||
|
|
||||||
return finding;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Get paginated findings with optional filters
|
|
||||||
*/
|
|
||||||
async findAll(workspaceId: string, query: QueryFindingsDto): Promise<PaginatedFindingsResponse> {
|
|
||||||
const page = query.page ?? 1;
|
|
||||||
const limit = query.limit ?? 50;
|
|
||||||
const skip = (page - 1) * limit;
|
|
||||||
|
|
||||||
const where: Prisma.FindingWhereInput = {
|
|
||||||
workspaceId,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (query.agentId) {
|
|
||||||
where.agentId = query.agentId;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (query.type) {
|
|
||||||
where.type = query.type;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (query.taskId) {
|
|
||||||
where.taskId = query.taskId;
|
|
||||||
}
|
|
||||||
|
|
||||||
const [data, total] = await Promise.all([
|
|
||||||
this.prisma.finding.findMany({
|
|
||||||
where,
|
|
||||||
select: findingSelect,
|
|
||||||
orderBy: {
|
|
||||||
createdAt: "desc",
|
|
||||||
},
|
|
||||||
skip,
|
|
||||||
take: limit,
|
|
||||||
}),
|
|
||||||
this.prisma.finding.count({ where }),
|
|
||||||
]);
|
|
||||||
|
|
||||||
return {
|
|
||||||
data,
|
|
||||||
meta: {
|
|
||||||
total,
|
|
||||||
page,
|
|
||||||
limit,
|
|
||||||
totalPages: Math.ceil(total / limit),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Get a single finding by ID
|
|
||||||
*/
|
|
||||||
async findOne(id: string, workspaceId: string): Promise<FindingRecord> {
|
|
||||||
const finding = await this.prisma.finding.findUnique({
|
|
||||||
where: {
|
|
||||||
id,
|
|
||||||
workspaceId,
|
|
||||||
},
|
|
||||||
select: findingSelect,
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!finding) {
|
|
||||||
throw new NotFoundException(`Finding with ID ${id} not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return finding;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Semantic search findings using vector similarity
|
|
||||||
*/
|
|
||||||
async search(workspaceId: string, searchDto: SearchFindingsDto): Promise<FindingsSearchResponse> {
|
|
||||||
if (!this.embeddingService.isConfigured()) {
|
|
||||||
throw new BadRequestException(
|
|
||||||
"Finding vector search requires OPENAI_API_KEY to be configured"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
const page = searchDto.page ?? 1;
|
|
||||||
const limit = searchDto.limit ?? 20;
|
|
||||||
const offset = (page - 1) * limit;
|
|
||||||
const similarityThreshold = searchDto.similarityThreshold ?? this.defaultSimilarityThreshold;
|
|
||||||
const distanceThreshold = 1 - similarityThreshold;
|
|
||||||
|
|
||||||
const queryEmbedding = await this.embeddingService.generateEmbedding(searchDto.query);
|
|
||||||
const embeddingString = `[${queryEmbedding.join(",")}]`;
|
|
||||||
|
|
||||||
const agentFilter = searchDto.agentId
|
|
||||||
? Prisma.sql`AND f.agent_id = ${searchDto.agentId}`
|
|
||||||
: Prisma.sql``;
|
|
||||||
const typeFilter = searchDto.type ? Prisma.sql`AND f.type = ${searchDto.type}` : Prisma.sql``;
|
|
||||||
const taskFilter = searchDto.taskId
|
|
||||||
? Prisma.sql`AND f.task_id = ${searchDto.taskId}::uuid`
|
|
||||||
: Prisma.sql``;
|
|
||||||
|
|
||||||
const searchResults = await this.prisma.$queryRaw<RawFindingSearchResult[]>`
|
|
||||||
SELECT
|
|
||||||
f.id,
|
|
||||||
f.workspace_id,
|
|
||||||
f.task_id,
|
|
||||||
f.agent_id,
|
|
||||||
f.type,
|
|
||||||
f.title,
|
|
||||||
f.data,
|
|
||||||
f.summary,
|
|
||||||
f.created_at,
|
|
||||||
f.updated_at,
|
|
||||||
(1 - (f.embedding <=> ${embeddingString}::vector)) AS score
|
|
||||||
FROM findings f
|
|
||||||
WHERE f.workspace_id = ${workspaceId}::uuid
|
|
||||||
AND f.embedding IS NOT NULL
|
|
||||||
${agentFilter}
|
|
||||||
${typeFilter}
|
|
||||||
${taskFilter}
|
|
||||||
AND (f.embedding <=> ${embeddingString}::vector) <= ${distanceThreshold}
|
|
||||||
ORDER BY f.embedding <=> ${embeddingString}::vector
|
|
||||||
LIMIT ${limit}
|
|
||||||
OFFSET ${offset}
|
|
||||||
`;
|
|
||||||
|
|
||||||
const countResult = await this.prisma.$queryRaw<[{ count: bigint }]>`
|
|
||||||
SELECT COUNT(*) as count
|
|
||||||
FROM findings f
|
|
||||||
WHERE f.workspace_id = ${workspaceId}::uuid
|
|
||||||
AND f.embedding IS NOT NULL
|
|
||||||
${agentFilter}
|
|
||||||
${typeFilter}
|
|
||||||
${taskFilter}
|
|
||||||
AND (f.embedding <=> ${embeddingString}::vector) <= ${distanceThreshold}
|
|
||||||
`;
|
|
||||||
|
|
||||||
const total = Number(countResult[0].count);
|
|
||||||
|
|
||||||
const data: FindingSearchResult[] = searchResults.map((row) => ({
|
|
||||||
id: row.id,
|
|
||||||
workspaceId: row.workspace_id,
|
|
||||||
taskId: row.task_id,
|
|
||||||
agentId: row.agent_id,
|
|
||||||
type: row.type,
|
|
||||||
title: row.title,
|
|
||||||
data: row.data,
|
|
||||||
summary: row.summary,
|
|
||||||
createdAt: row.created_at,
|
|
||||||
updatedAt: row.updated_at,
|
|
||||||
score: row.score,
|
|
||||||
}));
|
|
||||||
|
|
||||||
return {
|
|
||||||
data,
|
|
||||||
meta: {
|
|
||||||
total,
|
|
||||||
page,
|
|
||||||
limit,
|
|
||||||
totalPages: Math.ceil(total / limit),
|
|
||||||
},
|
|
||||||
query: searchDto.query,
|
|
||||||
similarityThreshold,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Delete a finding
|
|
||||||
*/
|
|
||||||
async remove(id: string, workspaceId: string): Promise<{ message: string }> {
|
|
||||||
const existingFinding = await this.prisma.finding.findUnique({
|
|
||||||
where: {
|
|
||||||
id,
|
|
||||||
workspaceId,
|
|
||||||
},
|
|
||||||
select: { id: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!existingFinding) {
|
|
||||||
throw new NotFoundException(`Finding with ID ${id} not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
await this.prisma.finding.delete({
|
|
||||||
where: {
|
|
||||||
id,
|
|
||||||
workspaceId,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
return { message: "Finding deleted successfully" };
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Generate and persist embedding for a finding summary
|
|
||||||
*/
|
|
||||||
private async generateAndStoreEmbedding(
|
|
||||||
findingId: string,
|
|
||||||
workspaceId: string,
|
|
||||||
summary: string
|
|
||||||
): Promise<void> {
|
|
||||||
if (!this.embeddingService.isConfigured()) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const embedding = await this.embeddingService.generateEmbedding(summary);
|
|
||||||
const embeddingString = `[${embedding.join(",")}]`;
|
|
||||||
|
|
||||||
await this.prisma.$executeRaw`
|
|
||||||
UPDATE findings
|
|
||||||
SET embedding = ${embeddingString}::vector,
|
|
||||||
updated_at = NOW()
|
|
||||||
WHERE id = ${findingId}::uuid
|
|
||||||
AND workspace_id = ${workspaceId}::uuid
|
|
||||||
`;
|
|
||||||
} catch (error) {
|
|
||||||
const message = error instanceof Error ? error.message : String(error);
|
|
||||||
this.logger.warn(`Failed to generate embedding for finding ${findingId}: ${message}`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,115 +0,0 @@
|
|||||||
import {
|
|
||||||
Body,
|
|
||||||
Controller,
|
|
||||||
Delete,
|
|
||||||
Get,
|
|
||||||
HttpCode,
|
|
||||||
HttpStatus,
|
|
||||||
Param,
|
|
||||||
Patch,
|
|
||||||
Post,
|
|
||||||
Put,
|
|
||||||
UseGuards,
|
|
||||||
} from "@nestjs/common";
|
|
||||||
import type { AuthUser } from "@mosaic/shared";
|
|
||||||
import { CurrentUser } from "../auth/decorators/current-user.decorator";
|
|
||||||
import { AdminGuard } from "../auth/guards/admin.guard";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import type {
|
|
||||||
CreateProviderDto,
|
|
||||||
ResetPasswordDto,
|
|
||||||
UpdateAgentConfigDto,
|
|
||||||
UpdateOidcDto,
|
|
||||||
UpdateProviderDto,
|
|
||||||
} from "./fleet-settings.dto";
|
|
||||||
import { FleetSettingsService } from "./fleet-settings.service";
|
|
||||||
|
|
||||||
@Controller("fleet-settings")
|
|
||||||
@UseGuards(AuthGuard)
|
|
||||||
export class FleetSettingsController {
|
|
||||||
constructor(private readonly fleetSettingsService: FleetSettingsService) {}
|
|
||||||
|
|
||||||
// --- Provider endpoints (user-scoped) ---
|
|
||||||
// GET /api/fleet-settings/providers — list user's providers
|
|
||||||
@Get("providers")
|
|
||||||
async listProviders(@CurrentUser() user: AuthUser) {
|
|
||||||
return this.fleetSettingsService.listProviders(user.id);
|
|
||||||
}
|
|
||||||
|
|
||||||
// GET /api/fleet-settings/providers/:id — get single provider
|
|
||||||
@Get("providers/:id")
|
|
||||||
async getProvider(@CurrentUser() user: AuthUser, @Param("id") id: string) {
|
|
||||||
return this.fleetSettingsService.getProvider(user.id, id);
|
|
||||||
}
|
|
||||||
|
|
||||||
// POST /api/fleet-settings/providers — create provider
|
|
||||||
@Post("providers")
|
|
||||||
async createProvider(@CurrentUser() user: AuthUser, @Body() dto: CreateProviderDto) {
|
|
||||||
return this.fleetSettingsService.createProvider(user.id, dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
// PATCH /api/fleet-settings/providers/:id — update provider
|
|
||||||
@Patch("providers/:id")
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async updateProvider(
|
|
||||||
@CurrentUser() user: AuthUser,
|
|
||||||
@Param("id") id: string,
|
|
||||||
@Body() dto: UpdateProviderDto
|
|
||||||
) {
|
|
||||||
await this.fleetSettingsService.updateProvider(user.id, id, dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
// DELETE /api/fleet-settings/providers/:id — delete provider
|
|
||||||
@Delete("providers/:id")
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async deleteProvider(@CurrentUser() user: AuthUser, @Param("id") id: string) {
|
|
||||||
await this.fleetSettingsService.deleteProvider(user.id, id);
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- Agent config endpoints (user-scoped) ---
|
|
||||||
// GET /api/fleet-settings/agent-config — get user's agent config
|
|
||||||
@Get("agent-config")
|
|
||||||
async getAgentConfig(@CurrentUser() user: AuthUser) {
|
|
||||||
return this.fleetSettingsService.getAgentConfig(user.id);
|
|
||||||
}
|
|
||||||
|
|
||||||
// PATCH /api/fleet-settings/agent-config — update user's agent config
|
|
||||||
@Patch("agent-config")
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async updateAgentConfig(@CurrentUser() user: AuthUser, @Body() dto: UpdateAgentConfigDto) {
|
|
||||||
await this.fleetSettingsService.updateAgentConfig(user.id, dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- OIDC endpoints (admin only — use AdminGuard) ---
|
|
||||||
// GET /api/fleet-settings/oidc — get OIDC config
|
|
||||||
@Get("oidc")
|
|
||||||
@UseGuards(AdminGuard)
|
|
||||||
async getOidcConfig() {
|
|
||||||
return this.fleetSettingsService.getOidcConfig();
|
|
||||||
}
|
|
||||||
|
|
||||||
// PUT /api/fleet-settings/oidc — update OIDC config
|
|
||||||
@Put("oidc")
|
|
||||||
@UseGuards(AdminGuard)
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async updateOidcConfig(@Body() dto: UpdateOidcDto) {
|
|
||||||
await this.fleetSettingsService.updateOidcConfig(dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
// DELETE /api/fleet-settings/oidc — remove OIDC config
|
|
||||||
@Delete("oidc")
|
|
||||||
@UseGuards(AdminGuard)
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async deleteOidcConfig() {
|
|
||||||
await this.fleetSettingsService.deleteOidcConfig();
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- Breakglass endpoints (admin only) ---
|
|
||||||
// POST /api/fleet-settings/breakglass/reset-password — reset admin password
|
|
||||||
@Post("breakglass/reset-password")
|
|
||||||
@UseGuards(AdminGuard)
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async resetBreakglassPassword(@Body() dto: ResetPasswordDto) {
|
|
||||||
await this.fleetSettingsService.resetBreakglassPassword(dto.username, dto.newPassword);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,122 +0,0 @@
|
|||||||
import {
|
|
||||||
ArrayNotEmpty,
|
|
||||||
IsArray,
|
|
||||||
IsBoolean,
|
|
||||||
IsNotEmpty,
|
|
||||||
IsObject,
|
|
||||||
IsOptional,
|
|
||||||
IsString,
|
|
||||||
IsUrl,
|
|
||||||
MaxLength,
|
|
||||||
MinLength,
|
|
||||||
} from "class-validator";
|
|
||||||
|
|
||||||
export class CreateProviderDto {
|
|
||||||
@IsString({ message: "name must be a string" })
|
|
||||||
@IsNotEmpty({ message: "name is required" })
|
|
||||||
@MaxLength(100, { message: "name must not exceed 100 characters" })
|
|
||||||
name!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "displayName must be a string" })
|
|
||||||
@IsNotEmpty({ message: "displayName is required" })
|
|
||||||
@MaxLength(255, { message: "displayName must not exceed 255 characters" })
|
|
||||||
displayName!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "type must be a string" })
|
|
||||||
@IsNotEmpty({ message: "type is required" })
|
|
||||||
@MaxLength(100, { message: "type must not exceed 100 characters" })
|
|
||||||
type!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsUrl(
|
|
||||||
{ require_tld: false },
|
|
||||||
{ message: "baseUrl must be a valid URL (for example: https://api.example.com/v1)" }
|
|
||||||
)
|
|
||||||
baseUrl?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "apiKey must be a string" })
|
|
||||||
apiKey?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "apiType must be a string" })
|
|
||||||
@MaxLength(100, { message: "apiType must not exceed 100 characters" })
|
|
||||||
apiType?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsArray({ message: "models must be an array" })
|
|
||||||
@IsObject({ each: true, message: "each model must be an object" })
|
|
||||||
models?: Record<string, unknown>[];
|
|
||||||
}
|
|
||||||
|
|
||||||
export class UpdateProviderDto {
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "displayName must be a string" })
|
|
||||||
@MaxLength(255, { message: "displayName must not exceed 255 characters" })
|
|
||||||
displayName?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsUrl(
|
|
||||||
{ require_tld: false },
|
|
||||||
{ message: "baseUrl must be a valid URL (for example: https://api.example.com/v1)" }
|
|
||||||
)
|
|
||||||
baseUrl?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "apiKey must be a string" })
|
|
||||||
apiKey?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsBoolean({ message: "isActive must be a boolean" })
|
|
||||||
isActive?: boolean;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsArray({ message: "models must be an array" })
|
|
||||||
@IsObject({ each: true, message: "each model must be an object" })
|
|
||||||
models?: Record<string, unknown>[];
|
|
||||||
}
|
|
||||||
|
|
||||||
export class UpdateAgentConfigDto {
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "primaryModel must be a string" })
|
|
||||||
@MaxLength(255, { message: "primaryModel must not exceed 255 characters" })
|
|
||||||
primaryModel?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsArray({ message: "fallbackModels must be an array" })
|
|
||||||
@ArrayNotEmpty({ message: "fallbackModels cannot be empty" })
|
|
||||||
@IsString({ each: true, message: "each fallback model must be a string" })
|
|
||||||
fallbackModels?: string[];
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "personality must be a string" })
|
|
||||||
personality?: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export class UpdateOidcDto {
|
|
||||||
@IsString({ message: "issuerUrl must be a string" })
|
|
||||||
@IsNotEmpty({ message: "issuerUrl is required" })
|
|
||||||
@IsUrl(
|
|
||||||
{ require_tld: false },
|
|
||||||
{ message: "issuerUrl must be a valid URL (for example: https://issuer.example.com)" }
|
|
||||||
)
|
|
||||||
issuerUrl!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "clientId must be a string" })
|
|
||||||
@IsNotEmpty({ message: "clientId is required" })
|
|
||||||
clientId!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "clientSecret must be a string" })
|
|
||||||
@IsNotEmpty({ message: "clientSecret is required" })
|
|
||||||
clientSecret!: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export class ResetPasswordDto {
|
|
||||||
@IsString({ message: "username must be a string" })
|
|
||||||
@IsNotEmpty({ message: "username is required" })
|
|
||||||
username!: string;
|
|
||||||
|
|
||||||
@IsString({ message: "newPassword must be a string" })
|
|
||||||
@MinLength(8, { message: "newPassword must be at least 8 characters" })
|
|
||||||
newPassword!: string;
|
|
||||||
}
|
|
||||||
@@ -1,14 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { AuthModule } from "../auth/auth.module";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { CryptoModule } from "../crypto/crypto.module";
|
|
||||||
import { FleetSettingsController } from "./fleet-settings.controller";
|
|
||||||
import { FleetSettingsService } from "./fleet-settings.service";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [AuthModule, PrismaModule, CryptoModule],
|
|
||||||
controllers: [FleetSettingsController],
|
|
||||||
providers: [FleetSettingsService],
|
|
||||||
exports: [FleetSettingsService],
|
|
||||||
})
|
|
||||||
export class FleetSettingsModule {}
|
|
||||||
@@ -1,200 +0,0 @@
|
|||||||
import { NotFoundException } from "@nestjs/common";
|
|
||||||
import { compare } from "bcryptjs";
|
|
||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
import { FleetSettingsService } from "./fleet-settings.service";
|
|
||||||
import type { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import type { CryptoService } from "../crypto/crypto.service";
|
|
||||||
|
|
||||||
describe("FleetSettingsService", () => {
|
|
||||||
let service: FleetSettingsService;
|
|
||||||
|
|
||||||
const mockPrisma = {
|
|
||||||
llmProvider: {
|
|
||||||
findMany: vi.fn(),
|
|
||||||
findFirst: vi.fn(),
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
create: vi.fn(),
|
|
||||||
update: vi.fn(),
|
|
||||||
delete: vi.fn(),
|
|
||||||
},
|
|
||||||
userAgentConfig: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
upsert: vi.fn(),
|
|
||||||
},
|
|
||||||
systemConfig: {
|
|
||||||
findMany: vi.fn(),
|
|
||||||
upsert: vi.fn(),
|
|
||||||
deleteMany: vi.fn(),
|
|
||||||
},
|
|
||||||
breakglassUser: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
update: vi.fn(),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockCrypto = {
|
|
||||||
encrypt: vi.fn((value: string) => `enc:${value}`),
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
service = new FleetSettingsService(
|
|
||||||
mockPrisma as unknown as PrismaService,
|
|
||||||
mockCrypto as unknown as CryptoService
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("listProviders returns only providers for the given userId", async () => {
|
|
||||||
mockPrisma.llmProvider.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "prov-1",
|
|
||||||
name: "openai-main",
|
|
||||||
displayName: "OpenAI",
|
|
||||||
type: "openai",
|
|
||||||
baseUrl: "https://api.openai.com/v1",
|
|
||||||
isActive: true,
|
|
||||||
models: [{ id: "gpt-4.1" }],
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await service.listProviders("user-1");
|
|
||||||
|
|
||||||
expect(mockPrisma.llmProvider.findMany).toHaveBeenCalledWith({
|
|
||||||
where: { userId: "user-1" },
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
name: true,
|
|
||||||
displayName: true,
|
|
||||||
type: true,
|
|
||||||
baseUrl: true,
|
|
||||||
isActive: true,
|
|
||||||
models: true,
|
|
||||||
},
|
|
||||||
orderBy: { createdAt: "asc" },
|
|
||||||
});
|
|
||||||
expect(result).toEqual([
|
|
||||||
{
|
|
||||||
id: "prov-1",
|
|
||||||
name: "openai-main",
|
|
||||||
displayName: "OpenAI",
|
|
||||||
type: "openai",
|
|
||||||
baseUrl: "https://api.openai.com/v1",
|
|
||||||
isActive: true,
|
|
||||||
models: [{ id: "gpt-4.1" }],
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("createProvider encrypts apiKey", async () => {
|
|
||||||
mockPrisma.llmProvider.create.mockResolvedValue({
|
|
||||||
id: "prov-2",
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await service.createProvider("user-1", {
|
|
||||||
name: "zai-main",
|
|
||||||
displayName: "Z.ai",
|
|
||||||
type: "zai",
|
|
||||||
apiKey: "plaintext-key",
|
|
||||||
models: [],
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(mockCrypto.encrypt).toHaveBeenCalledWith("plaintext-key");
|
|
||||||
expect(mockPrisma.llmProvider.create).toHaveBeenCalledWith({
|
|
||||||
data: {
|
|
||||||
userId: "user-1",
|
|
||||||
name: "zai-main",
|
|
||||||
displayName: "Z.ai",
|
|
||||||
type: "zai",
|
|
||||||
baseUrl: null,
|
|
||||||
apiKey: "enc:plaintext-key",
|
|
||||||
apiType: "openai-completions",
|
|
||||||
models: [],
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
expect(result).toEqual({ id: "prov-2" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("updateProvider rejects if not owned by user", async () => {
|
|
||||||
mockPrisma.llmProvider.findFirst.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(
|
|
||||||
service.updateProvider("user-1", "provider-1", {
|
|
||||||
displayName: "New Name",
|
|
||||||
})
|
|
||||||
).rejects.toBeInstanceOf(NotFoundException);
|
|
||||||
|
|
||||||
expect(mockPrisma.llmProvider.update).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("deleteProvider rejects if not owned by user", async () => {
|
|
||||||
mockPrisma.llmProvider.findFirst.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(service.deleteProvider("user-1", "provider-1")).rejects.toBeInstanceOf(
|
|
||||||
NotFoundException
|
|
||||||
);
|
|
||||||
|
|
||||||
expect(mockPrisma.llmProvider.delete).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("getOidcConfig never returns clientSecret", async () => {
|
|
||||||
mockPrisma.systemConfig.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
key: "oidc.issuerUrl",
|
|
||||||
value: "https://issuer.example.com",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
key: "oidc.clientId",
|
|
||||||
value: "client-id-1",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
key: "oidc.clientSecret",
|
|
||||||
value: "enc:very-secret",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await service.getOidcConfig();
|
|
||||||
|
|
||||||
expect(result).toEqual({
|
|
||||||
issuerUrl: "https://issuer.example.com",
|
|
||||||
clientId: "client-id-1",
|
|
||||||
configured: true,
|
|
||||||
});
|
|
||||||
expect(result).not.toHaveProperty("clientSecret");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("updateOidcConfig encrypts clientSecret", async () => {
|
|
||||||
await service.updateOidcConfig({
|
|
||||||
issuerUrl: "https://issuer.example.com",
|
|
||||||
clientId: "client-id-1",
|
|
||||||
clientSecret: "super-secret",
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(mockCrypto.encrypt).toHaveBeenCalledWith("super-secret");
|
|
||||||
expect(mockPrisma.systemConfig.upsert).toHaveBeenCalledTimes(3);
|
|
||||||
expect(mockPrisma.systemConfig.upsert).toHaveBeenCalledWith({
|
|
||||||
where: { key: "oidc.clientSecret" },
|
|
||||||
update: { value: "enc:super-secret", encrypted: true },
|
|
||||||
create: { key: "oidc.clientSecret", value: "enc:super-secret", encrypted: true },
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("resetBreakglassPassword hashes new password", async () => {
|
|
||||||
mockPrisma.breakglassUser.findUnique.mockResolvedValue({
|
|
||||||
id: "bg-1",
|
|
||||||
username: "admin",
|
|
||||||
passwordHash: "old-hash",
|
|
||||||
});
|
|
||||||
|
|
||||||
await service.resetBreakglassPassword("admin", "new-password-123");
|
|
||||||
|
|
||||||
expect(mockPrisma.breakglassUser.update).toHaveBeenCalledOnce();
|
|
||||||
const updateCall = mockPrisma.breakglassUser.update.mock.calls[0]?.[0];
|
|
||||||
const newHash = updateCall?.data?.passwordHash;
|
|
||||||
expect(newHash).toBeTypeOf("string");
|
|
||||||
expect(newHash).not.toBe("new-password-123");
|
|
||||||
expect(await compare("new-password-123", newHash as string)).toBe(true);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,296 +0,0 @@
|
|||||||
import { Injectable, NotFoundException } from "@nestjs/common";
|
|
||||||
import { hash } from "bcryptjs";
|
|
||||||
import type { Prisma } from "@prisma/client";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { CryptoService } from "../crypto/crypto.service";
|
|
||||||
import type {
|
|
||||||
CreateProviderDto,
|
|
||||||
ResetPasswordDto,
|
|
||||||
UpdateAgentConfigDto,
|
|
||||||
UpdateOidcDto,
|
|
||||||
UpdateProviderDto,
|
|
||||||
} from "./fleet-settings.dto";
|
|
||||||
|
|
||||||
const BCRYPT_ROUNDS = 12;
|
|
||||||
const DEFAULT_PROVIDER_API_TYPE = "openai-completions";
|
|
||||||
const OIDC_ISSUER_KEY = "oidc.issuerUrl";
|
|
||||||
const OIDC_CLIENT_ID_KEY = "oidc.clientId";
|
|
||||||
const OIDC_CLIENT_SECRET_KEY = "oidc.clientSecret";
|
|
||||||
const OIDC_KEYS = [OIDC_ISSUER_KEY, OIDC_CLIENT_ID_KEY, OIDC_CLIENT_SECRET_KEY] as const;
|
|
||||||
|
|
||||||
export interface FleetProviderResponse {
|
|
||||||
id: string;
|
|
||||||
name: string;
|
|
||||||
displayName: string;
|
|
||||||
type: string;
|
|
||||||
baseUrl: string | null;
|
|
||||||
isActive: boolean;
|
|
||||||
models: unknown;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface FleetAgentConfigResponse {
|
|
||||||
primaryModel: string | null;
|
|
||||||
fallbackModels: unknown[];
|
|
||||||
personality: string | null;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface OidcConfigResponse {
|
|
||||||
issuerUrl?: string;
|
|
||||||
clientId?: string;
|
|
||||||
configured: boolean;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class FleetSettingsService {
|
|
||||||
constructor(
|
|
||||||
private readonly prisma: PrismaService,
|
|
||||||
private readonly crypto: CryptoService
|
|
||||||
) {}
|
|
||||||
|
|
||||||
// --- LLM Provider CRUD (per-user scoped) ---
|
|
||||||
|
|
||||||
async listProviders(userId: string): Promise<FleetProviderResponse[]> {
|
|
||||||
return this.prisma.llmProvider.findMany({
|
|
||||||
where: { userId },
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
name: true,
|
|
||||||
displayName: true,
|
|
||||||
type: true,
|
|
||||||
baseUrl: true,
|
|
||||||
isActive: true,
|
|
||||||
models: true,
|
|
||||||
},
|
|
||||||
orderBy: { createdAt: "asc" },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async getProvider(userId: string, providerId: string): Promise<FleetProviderResponse> {
|
|
||||||
const provider = await this.prisma.llmProvider.findFirst({
|
|
||||||
where: {
|
|
||||||
id: providerId,
|
|
||||||
userId,
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
name: true,
|
|
||||||
displayName: true,
|
|
||||||
type: true,
|
|
||||||
baseUrl: true,
|
|
||||||
isActive: true,
|
|
||||||
models: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!provider) {
|
|
||||||
throw new NotFoundException(`Provider ${providerId} not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
return provider;
|
|
||||||
}
|
|
||||||
|
|
||||||
async createProvider(userId: string, data: CreateProviderDto): Promise<{ id: string }> {
|
|
||||||
const provider = await this.prisma.llmProvider.create({
|
|
||||||
data: {
|
|
||||||
userId,
|
|
||||||
name: data.name,
|
|
||||||
displayName: data.displayName,
|
|
||||||
type: data.type,
|
|
||||||
baseUrl: data.baseUrl ?? null,
|
|
||||||
apiKey: data.apiKey ? this.crypto.encrypt(data.apiKey) : null,
|
|
||||||
apiType: data.apiType ?? DEFAULT_PROVIDER_API_TYPE,
|
|
||||||
models: (data.models ?? []) as Prisma.InputJsonValue,
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
return provider;
|
|
||||||
}
|
|
||||||
|
|
||||||
async updateProvider(userId: string, providerId: string, data: UpdateProviderDto): Promise<void> {
|
|
||||||
await this.assertProviderOwnership(userId, providerId);
|
|
||||||
|
|
||||||
const updateData: Prisma.LlmProviderUpdateInput = {};
|
|
||||||
if (data.displayName !== undefined) {
|
|
||||||
updateData.displayName = data.displayName;
|
|
||||||
}
|
|
||||||
if (data.baseUrl !== undefined) {
|
|
||||||
updateData.baseUrl = data.baseUrl;
|
|
||||||
}
|
|
||||||
if (data.isActive !== undefined) {
|
|
||||||
updateData.isActive = data.isActive;
|
|
||||||
}
|
|
||||||
if (data.models !== undefined) {
|
|
||||||
updateData.models = data.models as Prisma.InputJsonValue;
|
|
||||||
}
|
|
||||||
if (data.apiKey !== undefined) {
|
|
||||||
updateData.apiKey = data.apiKey.length > 0 ? this.crypto.encrypt(data.apiKey) : null;
|
|
||||||
}
|
|
||||||
|
|
||||||
await this.prisma.llmProvider.update({
|
|
||||||
where: { id: providerId },
|
|
||||||
data: updateData,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async deleteProvider(userId: string, providerId: string): Promise<void> {
|
|
||||||
await this.assertProviderOwnership(userId, providerId);
|
|
||||||
|
|
||||||
await this.prisma.llmProvider.delete({
|
|
||||||
where: { id: providerId },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- User Agent Config ---
|
|
||||||
|
|
||||||
async getAgentConfig(userId: string): Promise<FleetAgentConfigResponse> {
|
|
||||||
const config = await this.prisma.userAgentConfig.findUnique({
|
|
||||||
where: { userId },
|
|
||||||
select: {
|
|
||||||
primaryModel: true,
|
|
||||||
fallbackModels: true,
|
|
||||||
personality: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!config) {
|
|
||||||
return {
|
|
||||||
primaryModel: null,
|
|
||||||
fallbackModels: [],
|
|
||||||
personality: null,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
|
||||||
primaryModel: config.primaryModel,
|
|
||||||
fallbackModels: this.normalizeJsonArray(config.fallbackModels),
|
|
||||||
personality: config.personality,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
async updateAgentConfig(userId: string, data: UpdateAgentConfigDto): Promise<void> {
|
|
||||||
const updateData: Prisma.UserAgentConfigUpdateInput = {};
|
|
||||||
if (data.primaryModel !== undefined) {
|
|
||||||
updateData.primaryModel = data.primaryModel;
|
|
||||||
}
|
|
||||||
if (data.personality !== undefined) {
|
|
||||||
updateData.personality = data.personality;
|
|
||||||
}
|
|
||||||
if (data.fallbackModels !== undefined) {
|
|
||||||
updateData.fallbackModels = data.fallbackModels as Prisma.InputJsonValue;
|
|
||||||
}
|
|
||||||
|
|
||||||
const createData: Prisma.UserAgentConfigCreateInput = {
|
|
||||||
userId,
|
|
||||||
fallbackModels: (data.fallbackModels ?? []) as Prisma.InputJsonValue,
|
|
||||||
...(data.primaryModel !== undefined ? { primaryModel: data.primaryModel } : {}),
|
|
||||||
...(data.personality !== undefined ? { personality: data.personality } : {}),
|
|
||||||
};
|
|
||||||
|
|
||||||
await this.prisma.userAgentConfig.upsert({
|
|
||||||
where: { userId },
|
|
||||||
create: createData,
|
|
||||||
update: updateData,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- OIDC Config (admin only) ---
|
|
||||||
|
|
||||||
async getOidcConfig(): Promise<OidcConfigResponse> {
|
|
||||||
const entries = await this.prisma.systemConfig.findMany({
|
|
||||||
where: {
|
|
||||||
key: {
|
|
||||||
in: [...OIDC_KEYS],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
key: true,
|
|
||||||
value: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const byKey = new Map(entries.map((entry) => [entry.key, entry.value]));
|
|
||||||
const issuerUrl = byKey.get(OIDC_ISSUER_KEY);
|
|
||||||
const clientId = byKey.get(OIDC_CLIENT_ID_KEY);
|
|
||||||
const hasSecret = byKey.has(OIDC_CLIENT_SECRET_KEY);
|
|
||||||
|
|
||||||
return {
|
|
||||||
...(issuerUrl ? { issuerUrl } : {}),
|
|
||||||
...(clientId ? { clientId } : {}),
|
|
||||||
configured: Boolean(issuerUrl && clientId && hasSecret),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
async updateOidcConfig(data: UpdateOidcDto): Promise<void> {
|
|
||||||
const encryptedSecret = this.crypto.encrypt(data.clientSecret);
|
|
||||||
|
|
||||||
await Promise.all([
|
|
||||||
this.upsertSystemConfig(OIDC_ISSUER_KEY, data.issuerUrl, false),
|
|
||||||
this.upsertSystemConfig(OIDC_CLIENT_ID_KEY, data.clientId, false),
|
|
||||||
this.upsertSystemConfig(OIDC_CLIENT_SECRET_KEY, encryptedSecret, true),
|
|
||||||
]);
|
|
||||||
}
|
|
||||||
|
|
||||||
async deleteOidcConfig(): Promise<void> {
|
|
||||||
await this.prisma.systemConfig.deleteMany({
|
|
||||||
where: {
|
|
||||||
key: {
|
|
||||||
in: [...OIDC_KEYS],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- Breakglass (admin only) ---
|
|
||||||
|
|
||||||
async resetBreakglassPassword(
|
|
||||||
username: ResetPasswordDto["username"],
|
|
||||||
newPassword: ResetPasswordDto["newPassword"]
|
|
||||||
): Promise<void> {
|
|
||||||
const user = await this.prisma.breakglassUser.findUnique({
|
|
||||||
where: { username },
|
|
||||||
select: { id: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!user) {
|
|
||||||
throw new NotFoundException(`Breakglass user ${username} not found`);
|
|
||||||
}
|
|
||||||
|
|
||||||
const passwordHash = await hash(newPassword, BCRYPT_ROUNDS);
|
|
||||||
|
|
||||||
await this.prisma.breakglassUser.update({
|
|
||||||
where: { id: user.id },
|
|
||||||
data: { passwordHash },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
private async assertProviderOwnership(userId: string, providerId: string): Promise<void> {
|
|
||||||
const provider = await this.prisma.llmProvider.findFirst({
|
|
||||||
where: {
|
|
||||||
id: providerId,
|
|
||||||
userId,
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!provider) {
|
|
||||||
throw new NotFoundException(`Provider ${providerId} not found`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private async upsertSystemConfig(key: string, value: string, encrypted: boolean): Promise<void> {
|
|
||||||
await this.prisma.systemConfig.upsert({
|
|
||||||
where: { key },
|
|
||||||
update: { value, encrypted },
|
|
||||||
create: { key, value, encrypted },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
private normalizeJsonArray(value: unknown): unknown[] {
|
|
||||||
return Array.isArray(value) ? value : [];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,63 +0,0 @@
|
|||||||
import { Body, Controller, Get, HttpCode, HttpStatus, Post, UseGuards } from "@nestjs/common";
|
|
||||||
import {
|
|
||||||
AddProviderDto,
|
|
||||||
ConfigureOidcDto,
|
|
||||||
CreateBreakglassDto,
|
|
||||||
TestProviderDto,
|
|
||||||
} from "./onboarding.dto";
|
|
||||||
import { OnboardingGuard } from "./onboarding.guard";
|
|
||||||
import { OnboardingService } from "./onboarding.service";
|
|
||||||
|
|
||||||
@Controller("onboarding")
|
|
||||||
export class OnboardingController {
|
|
||||||
constructor(private readonly onboardingService: OnboardingService) {}
|
|
||||||
|
|
||||||
// GET /api/onboarding/status — returns { completed: boolean }
|
|
||||||
@Get("status")
|
|
||||||
async status(): Promise<{ completed: boolean }> {
|
|
||||||
return {
|
|
||||||
completed: await this.onboardingService.isCompleted(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
// POST /api/onboarding/breakglass — body: { username, password } → create admin
|
|
||||||
@Post("breakglass")
|
|
||||||
@UseGuards(OnboardingGuard)
|
|
||||||
async createBreakglass(
|
|
||||||
@Body() body: CreateBreakglassDto
|
|
||||||
): Promise<{ id: string; username: string }> {
|
|
||||||
return this.onboardingService.createBreakglassUser(body.username, body.password);
|
|
||||||
}
|
|
||||||
|
|
||||||
// POST /api/onboarding/oidc — body: { issuerUrl, clientId, clientSecret } → save OIDC
|
|
||||||
@Post("oidc")
|
|
||||||
@UseGuards(OnboardingGuard)
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async configureOidc(@Body() body: ConfigureOidcDto): Promise<void> {
|
|
||||||
await this.onboardingService.configureOidc(body.issuerUrl, body.clientId, body.clientSecret);
|
|
||||||
}
|
|
||||||
|
|
||||||
// POST /api/onboarding/provider — body: { name, displayName, type, baseUrl?, apiKey?, models? } → add provider
|
|
||||||
@Post("provider")
|
|
||||||
@UseGuards(OnboardingGuard)
|
|
||||||
async addProvider(@Body() body: AddProviderDto): Promise<{ id: string }> {
|
|
||||||
const userId = await this.onboardingService.getBreakglassUserId();
|
|
||||||
|
|
||||||
return this.onboardingService.addProvider(userId, body);
|
|
||||||
}
|
|
||||||
|
|
||||||
// POST /api/onboarding/provider/test — body: { type, baseUrl?, apiKey? } → test connection
|
|
||||||
@Post("provider/test")
|
|
||||||
@UseGuards(OnboardingGuard)
|
|
||||||
async testProvider(@Body() body: TestProviderDto): Promise<{ success: boolean; error?: string }> {
|
|
||||||
return this.onboardingService.testProvider(body.type, body.baseUrl, body.apiKey);
|
|
||||||
}
|
|
||||||
|
|
||||||
// POST /api/onboarding/complete — mark done
|
|
||||||
@Post("complete")
|
|
||||||
@UseGuards(OnboardingGuard)
|
|
||||||
@HttpCode(HttpStatus.NO_CONTENT)
|
|
||||||
async complete(): Promise<void> {
|
|
||||||
await this.onboardingService.complete();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,71 +0,0 @@
|
|||||||
import { Type } from "class-transformer";
|
|
||||||
import { IsArray, IsOptional, IsString, IsUrl, MinLength, ValidateNested } from "class-validator";
|
|
||||||
|
|
||||||
export class CreateBreakglassDto {
|
|
||||||
@IsString()
|
|
||||||
@MinLength(3)
|
|
||||||
username!: string;
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
@MinLength(8)
|
|
||||||
password!: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export class ConfigureOidcDto {
|
|
||||||
@IsString()
|
|
||||||
@IsUrl({ require_tld: false })
|
|
||||||
issuerUrl!: string;
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
clientId!: string;
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
clientSecret!: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export class ProviderModelDto {
|
|
||||||
@IsString()
|
|
||||||
id!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString()
|
|
||||||
name?: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export class AddProviderDto {
|
|
||||||
@IsString()
|
|
||||||
name!: string;
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
displayName!: string;
|
|
||||||
|
|
||||||
@IsString()
|
|
||||||
type!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString()
|
|
||||||
baseUrl?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString()
|
|
||||||
apiKey?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsArray()
|
|
||||||
@ValidateNested({ each: true })
|
|
||||||
@Type(() => ProviderModelDto)
|
|
||||||
models?: ProviderModelDto[];
|
|
||||||
}
|
|
||||||
|
|
||||||
export class TestProviderDto {
|
|
||||||
@IsString()
|
|
||||||
type!: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString()
|
|
||||||
baseUrl?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString()
|
|
||||||
apiKey?: string;
|
|
||||||
}
|
|
||||||
@@ -1,17 +0,0 @@
|
|||||||
import { CanActivate, ExecutionContext, ForbiddenException, Injectable } from "@nestjs/common";
|
|
||||||
import { OnboardingService } from "./onboarding.service";
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class OnboardingGuard implements CanActivate {
|
|
||||||
constructor(private readonly onboardingService: OnboardingService) {}
|
|
||||||
|
|
||||||
async canActivate(_context: ExecutionContext): Promise<boolean> {
|
|
||||||
const completed = await this.onboardingService.isCompleted();
|
|
||||||
|
|
||||||
if (completed) {
|
|
||||||
throw new ForbiddenException("Onboarding already completed");
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,15 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { ConfigModule } from "@nestjs/config";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { CryptoModule } from "../crypto/crypto.module";
|
|
||||||
import { OnboardingController } from "./onboarding.controller";
|
|
||||||
import { OnboardingService } from "./onboarding.service";
|
|
||||||
import { OnboardingGuard } from "./onboarding.guard";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [PrismaModule, CryptoModule, ConfigModule],
|
|
||||||
controllers: [OnboardingController],
|
|
||||||
providers: [OnboardingService, OnboardingGuard],
|
|
||||||
exports: [OnboardingService],
|
|
||||||
})
|
|
||||||
export class OnboardingModule {}
|
|
||||||
@@ -1,206 +0,0 @@
|
|||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
import { hash } from "bcryptjs";
|
|
||||||
import { OnboardingService } from "./onboarding.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { CryptoService } from "../crypto/crypto.service";
|
|
||||||
|
|
||||||
vi.mock("bcryptjs", () => ({
|
|
||||||
hash: vi.fn(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
describe("OnboardingService", () => {
|
|
||||||
let service: OnboardingService;
|
|
||||||
|
|
||||||
const mockPrismaService = {
|
|
||||||
systemConfig: {
|
|
||||||
findUnique: vi.fn(),
|
|
||||||
upsert: vi.fn(),
|
|
||||||
},
|
|
||||||
breakglassUser: {
|
|
||||||
count: vi.fn(),
|
|
||||||
create: vi.fn(),
|
|
||||||
findFirst: vi.fn(),
|
|
||||||
},
|
|
||||||
llmProvider: {
|
|
||||||
create: vi.fn(),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const mockCryptoService = {
|
|
||||||
encrypt: vi.fn(),
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
|
|
||||||
service = new OnboardingService(
|
|
||||||
mockPrismaService as unknown as PrismaService,
|
|
||||||
mockCryptoService as unknown as CryptoService
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("isCompleted returns false when no config exists", async () => {
|
|
||||||
mockPrismaService.systemConfig.findUnique.mockResolvedValue(null);
|
|
||||||
|
|
||||||
await expect(service.isCompleted()).resolves.toBe(false);
|
|
||||||
expect(mockPrismaService.systemConfig.findUnique).toHaveBeenCalledWith({
|
|
||||||
where: { key: "onboarding.completed" },
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("isCompleted returns true when completed", async () => {
|
|
||||||
mockPrismaService.systemConfig.findUnique.mockResolvedValue({
|
|
||||||
id: "cfg-1",
|
|
||||||
key: "onboarding.completed",
|
|
||||||
value: "true",
|
|
||||||
encrypted: false,
|
|
||||||
updatedAt: new Date(),
|
|
||||||
});
|
|
||||||
|
|
||||||
await expect(service.isCompleted()).resolves.toBe(true);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("createBreakglassUser hashes password and creates record", async () => {
|
|
||||||
const mockedHash = vi.mocked(hash);
|
|
||||||
mockedHash.mockResolvedValue("hashed-password");
|
|
||||||
|
|
||||||
mockPrismaService.breakglassUser.count.mockResolvedValue(0);
|
|
||||||
mockPrismaService.breakglassUser.create.mockResolvedValue({
|
|
||||||
id: "breakglass-1",
|
|
||||||
username: "admin",
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await service.createBreakglassUser("admin", "supersecret123");
|
|
||||||
|
|
||||||
expect(mockedHash).toHaveBeenCalledWith("supersecret123", 12);
|
|
||||||
expect(mockPrismaService.breakglassUser.create).toHaveBeenCalledWith({
|
|
||||||
data: {
|
|
||||||
username: "admin",
|
|
||||||
passwordHash: "hashed-password",
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
username: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
expect(result).toEqual({ id: "breakglass-1", username: "admin" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("createBreakglassUser rejects if user already exists", async () => {
|
|
||||||
mockPrismaService.breakglassUser.count.mockResolvedValue(1);
|
|
||||||
|
|
||||||
await expect(service.createBreakglassUser("admin", "supersecret123")).rejects.toThrow(
|
|
||||||
"Breakglass user already exists"
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("configureOidc encrypts secret and saves to SystemConfig", async () => {
|
|
||||||
mockCryptoService.encrypt.mockReturnValue("enc:oidc-secret");
|
|
||||||
mockPrismaService.systemConfig.upsert.mockResolvedValue({
|
|
||||||
id: "cfg",
|
|
||||||
key: "oidc.clientSecret",
|
|
||||||
value: "enc:oidc-secret",
|
|
||||||
encrypted: true,
|
|
||||||
updatedAt: new Date(),
|
|
||||||
});
|
|
||||||
|
|
||||||
await service.configureOidc("https://auth.example.com", "client-id", "client-secret");
|
|
||||||
|
|
||||||
expect(mockCryptoService.encrypt).toHaveBeenCalledWith("client-secret");
|
|
||||||
expect(mockPrismaService.systemConfig.upsert).toHaveBeenCalledTimes(3);
|
|
||||||
expect(mockPrismaService.systemConfig.upsert).toHaveBeenCalledWith({
|
|
||||||
where: { key: "oidc.issuerUrl" },
|
|
||||||
create: {
|
|
||||||
key: "oidc.issuerUrl",
|
|
||||||
value: "https://auth.example.com",
|
|
||||||
encrypted: false,
|
|
||||||
},
|
|
||||||
update: {
|
|
||||||
value: "https://auth.example.com",
|
|
||||||
encrypted: false,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
expect(mockPrismaService.systemConfig.upsert).toHaveBeenCalledWith({
|
|
||||||
where: { key: "oidc.clientId" },
|
|
||||||
create: {
|
|
||||||
key: "oidc.clientId",
|
|
||||||
value: "client-id",
|
|
||||||
encrypted: false,
|
|
||||||
},
|
|
||||||
update: {
|
|
||||||
value: "client-id",
|
|
||||||
encrypted: false,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
expect(mockPrismaService.systemConfig.upsert).toHaveBeenCalledWith({
|
|
||||||
where: { key: "oidc.clientSecret" },
|
|
||||||
create: {
|
|
||||||
key: "oidc.clientSecret",
|
|
||||||
value: "enc:oidc-secret",
|
|
||||||
encrypted: true,
|
|
||||||
},
|
|
||||||
update: {
|
|
||||||
value: "enc:oidc-secret",
|
|
||||||
encrypted: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("addProvider encrypts apiKey and creates LlmProvider", async () => {
|
|
||||||
mockCryptoService.encrypt.mockReturnValue("enc:api-key");
|
|
||||||
mockPrismaService.llmProvider.create.mockResolvedValue({
|
|
||||||
id: "provider-1",
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await service.addProvider("breakglass-1", {
|
|
||||||
name: "my-openai",
|
|
||||||
displayName: "OpenAI",
|
|
||||||
type: "openai",
|
|
||||||
baseUrl: "https://api.openai.com/v1",
|
|
||||||
apiKey: "sk-test",
|
|
||||||
models: [{ id: "gpt-4o-mini", name: "GPT-4o Mini" }],
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(mockCryptoService.encrypt).toHaveBeenCalledWith("sk-test");
|
|
||||||
expect(mockPrismaService.llmProvider.create).toHaveBeenCalledWith({
|
|
||||||
data: {
|
|
||||||
userId: "breakglass-1",
|
|
||||||
name: "my-openai",
|
|
||||||
displayName: "OpenAI",
|
|
||||||
type: "openai",
|
|
||||||
baseUrl: "https://api.openai.com/v1",
|
|
||||||
apiKey: "enc:api-key",
|
|
||||||
models: [{ id: "gpt-4o-mini", name: "GPT-4o Mini" }],
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
expect(result).toEqual({ id: "provider-1" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("complete sets SystemConfig flag", async () => {
|
|
||||||
mockPrismaService.systemConfig.upsert.mockResolvedValue({
|
|
||||||
id: "cfg-1",
|
|
||||||
key: "onboarding.completed",
|
|
||||||
value: "true",
|
|
||||||
encrypted: false,
|
|
||||||
updatedAt: new Date(),
|
|
||||||
});
|
|
||||||
|
|
||||||
await service.complete();
|
|
||||||
|
|
||||||
expect(mockPrismaService.systemConfig.upsert).toHaveBeenCalledWith({
|
|
||||||
where: { key: "onboarding.completed" },
|
|
||||||
create: {
|
|
||||||
key: "onboarding.completed",
|
|
||||||
value: "true",
|
|
||||||
encrypted: false,
|
|
||||||
},
|
|
||||||
update: {
|
|
||||||
value: "true",
|
|
||||||
encrypted: false,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,191 +0,0 @@
|
|||||||
import { BadRequestException, ConflictException, Injectable } from "@nestjs/common";
|
|
||||||
import type { InputJsonValue } from "@prisma/client/runtime/library";
|
|
||||||
import { hash } from "bcryptjs";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { CryptoService } from "../crypto/crypto.service";
|
|
||||||
|
|
||||||
const BCRYPT_ROUNDS = 12;
|
|
||||||
const TEST_PROVIDER_TIMEOUT_MS = 8000;
|
|
||||||
|
|
||||||
const ONBOARDING_COMPLETED_KEY = "onboarding.completed";
|
|
||||||
const OIDC_ISSUER_URL_KEY = "oidc.issuerUrl";
|
|
||||||
const OIDC_CLIENT_ID_KEY = "oidc.clientId";
|
|
||||||
const OIDC_CLIENT_SECRET_KEY = "oidc.clientSecret";
|
|
||||||
|
|
||||||
interface ProviderModelInput {
|
|
||||||
id: string;
|
|
||||||
name?: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface AddProviderInput {
|
|
||||||
name: string;
|
|
||||||
displayName: string;
|
|
||||||
type: string;
|
|
||||||
baseUrl?: string;
|
|
||||||
apiKey?: string;
|
|
||||||
models?: ProviderModelInput[];
|
|
||||||
}
|
|
||||||
|
|
||||||
@Injectable()
|
|
||||||
export class OnboardingService {
|
|
||||||
constructor(
|
|
||||||
private readonly prisma: PrismaService,
|
|
||||||
private readonly crypto: CryptoService
|
|
||||||
) {}
|
|
||||||
|
|
||||||
// Check if onboarding is completed
|
|
||||||
async isCompleted(): Promise<boolean> {
|
|
||||||
const completedFlag = await this.prisma.systemConfig.findUnique({
|
|
||||||
where: { key: ONBOARDING_COMPLETED_KEY },
|
|
||||||
});
|
|
||||||
|
|
||||||
return completedFlag?.value === "true";
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 1: Create breakglass admin user
|
|
||||||
async createBreakglassUser(
|
|
||||||
username: string,
|
|
||||||
password: string
|
|
||||||
): Promise<{ id: string; username: string }> {
|
|
||||||
const breakglassCount = await this.prisma.breakglassUser.count();
|
|
||||||
if (breakglassCount > 0) {
|
|
||||||
throw new ConflictException("Breakglass user already exists");
|
|
||||||
}
|
|
||||||
|
|
||||||
const passwordHash = await hash(password, BCRYPT_ROUNDS);
|
|
||||||
|
|
||||||
return this.prisma.breakglassUser.create({
|
|
||||||
data: {
|
|
||||||
username,
|
|
||||||
passwordHash,
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
username: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 2: Configure OIDC provider (optional)
|
|
||||||
async configureOidc(issuerUrl: string, clientId: string, clientSecret: string): Promise<void> {
|
|
||||||
const encryptedSecret = this.crypto.encrypt(clientSecret);
|
|
||||||
|
|
||||||
await Promise.all([
|
|
||||||
this.upsertSystemConfig(OIDC_ISSUER_URL_KEY, issuerUrl, false),
|
|
||||||
this.upsertSystemConfig(OIDC_CLIENT_ID_KEY, clientId, false),
|
|
||||||
this.upsertSystemConfig(OIDC_CLIENT_SECRET_KEY, encryptedSecret, true),
|
|
||||||
]);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 3: Add first LLM provider
|
|
||||||
async addProvider(userId: string, data: AddProviderInput): Promise<{ id: string }> {
|
|
||||||
const encryptedApiKey = data.apiKey ? this.crypto.encrypt(data.apiKey) : undefined;
|
|
||||||
|
|
||||||
return this.prisma.llmProvider.create({
|
|
||||||
data: {
|
|
||||||
userId,
|
|
||||||
name: data.name,
|
|
||||||
displayName: data.displayName,
|
|
||||||
type: data.type,
|
|
||||||
baseUrl: data.baseUrl ?? null,
|
|
||||||
apiKey: encryptedApiKey ?? null,
|
|
||||||
models: (data.models ?? []) as unknown as InputJsonValue,
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 3b: Test LLM provider connection
|
|
||||||
async testProvider(
|
|
||||||
type: string,
|
|
||||||
baseUrl?: string,
|
|
||||||
apiKey?: string
|
|
||||||
): Promise<{ success: boolean; error?: string }> {
|
|
||||||
const normalizedType = type.trim().toLowerCase();
|
|
||||||
if (!normalizedType) {
|
|
||||||
return { success: false, error: "Provider type is required" };
|
|
||||||
}
|
|
||||||
|
|
||||||
let probeUrl: string;
|
|
||||||
try {
|
|
||||||
probeUrl = this.buildProbeUrl(normalizedType, baseUrl);
|
|
||||||
} catch (error: unknown) {
|
|
||||||
const message = error instanceof Error ? error.message : String(error);
|
|
||||||
return { success: false, error: message };
|
|
||||||
}
|
|
||||||
|
|
||||||
const headers: Record<string, string> = {
|
|
||||||
Accept: "application/json",
|
|
||||||
};
|
|
||||||
if (apiKey) {
|
|
||||||
headers.Authorization = `Bearer ${apiKey}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const response = await fetch(probeUrl, {
|
|
||||||
method: "GET",
|
|
||||||
headers,
|
|
||||||
signal: AbortSignal.timeout(TEST_PROVIDER_TIMEOUT_MS),
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!response.ok) {
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
error: `Provider returned ${String(response.status)} ${response.statusText}`.trim(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
return { success: true };
|
|
||||||
} catch (error: unknown) {
|
|
||||||
const message = error instanceof Error ? error.message : String(error);
|
|
||||||
return { success: false, error: message };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 4: Mark onboarding complete
|
|
||||||
async complete(): Promise<void> {
|
|
||||||
await this.upsertSystemConfig(ONBOARDING_COMPLETED_KEY, "true", false);
|
|
||||||
}
|
|
||||||
|
|
||||||
async getBreakglassUserId(): Promise<string> {
|
|
||||||
const user = await this.prisma.breakglassUser.findFirst({
|
|
||||||
where: { isActive: true },
|
|
||||||
orderBy: { createdAt: "asc" },
|
|
||||||
select: { id: true },
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!user) {
|
|
||||||
throw new BadRequestException("Create a breakglass user before adding a provider");
|
|
||||||
}
|
|
||||||
|
|
||||||
return user.id;
|
|
||||||
}
|
|
||||||
|
|
||||||
private async upsertSystemConfig(key: string, value: string, encrypted: boolean): Promise<void> {
|
|
||||||
await this.prisma.systemConfig.upsert({
|
|
||||||
where: { key },
|
|
||||||
create: { key, value, encrypted },
|
|
||||||
update: { value, encrypted },
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
private buildProbeUrl(type: string, baseUrl?: string): string {
|
|
||||||
const resolvedBaseUrl = baseUrl ?? this.getDefaultProviderBaseUrl(type);
|
|
||||||
const normalizedBaseUrl = resolvedBaseUrl.endsWith("/")
|
|
||||||
? resolvedBaseUrl
|
|
||||||
: `${resolvedBaseUrl}/`;
|
|
||||||
const endpointPath = type === "ollama" ? "api/tags" : "models";
|
|
||||||
|
|
||||||
return new URL(endpointPath, normalizedBaseUrl).toString();
|
|
||||||
}
|
|
||||||
|
|
||||||
private getDefaultProviderBaseUrl(type: string): string {
|
|
||||||
if (type === "ollama") {
|
|
||||||
return "http://localhost:11434";
|
|
||||||
}
|
|
||||||
|
|
||||||
return "https://api.openai.com/v1";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,194 +0,0 @@
|
|||||||
import { beforeEach, describe, expect, it, vi, afterEach } from "vitest";
|
|
||||||
import type { Response } from "express";
|
|
||||||
import { AgentStatus } from "@prisma/client";
|
|
||||||
import { OrchestratorController } from "./orchestrator.controller";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
|
|
||||||
describe("OrchestratorController", () => {
|
|
||||||
const mockPrismaService = {
|
|
||||||
agent: {
|
|
||||||
findMany: vi.fn(),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
let controller: OrchestratorController;
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
controller = new OrchestratorController(mockPrismaService as unknown as PrismaService);
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(() => {
|
|
||||||
vi.useRealTimers();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("getAgents", () => {
|
|
||||||
it("returns active agents with API widget shape", async () => {
|
|
||||||
mockPrismaService.agent.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "agent-1",
|
|
||||||
name: "Planner",
|
|
||||||
status: AgentStatus.WORKING,
|
|
||||||
role: "planner",
|
|
||||||
createdAt: new Date("2026-02-28T10:00:00.000Z"),
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await controller.getAgents();
|
|
||||||
|
|
||||||
expect(result).toEqual([
|
|
||||||
{
|
|
||||||
id: "agent-1",
|
|
||||||
name: "Planner",
|
|
||||||
status: AgentStatus.WORKING,
|
|
||||||
type: "planner",
|
|
||||||
createdAt: new Date("2026-02-28T10:00:00.000Z"),
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
expect(mockPrismaService.agent.findMany).toHaveBeenCalledWith({
|
|
||||||
where: {
|
|
||||||
status: {
|
|
||||||
not: AgentStatus.TERMINATED,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
orderBy: {
|
|
||||||
createdAt: "desc",
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
name: true,
|
|
||||||
status: true,
|
|
||||||
role: true,
|
|
||||||
createdAt: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("falls back to type=agent when role is missing", async () => {
|
|
||||||
mockPrismaService.agent.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "agent-2",
|
|
||||||
name: null,
|
|
||||||
status: AgentStatus.IDLE,
|
|
||||||
role: null,
|
|
||||||
createdAt: new Date("2026-02-28T11:00:00.000Z"),
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
const result = await controller.getAgents();
|
|
||||||
|
|
||||||
expect(result[0]).toMatchObject({
|
|
||||||
id: "agent-2",
|
|
||||||
type: "agent",
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("streamEvents", () => {
|
|
||||||
it("sets SSE headers and writes initial data payload", async () => {
|
|
||||||
const onHandlers: Record<string, (() => void) | undefined> = {};
|
|
||||||
const mockRes = {
|
|
||||||
setHeader: vi.fn(),
|
|
||||||
write: vi.fn(),
|
|
||||||
end: vi.fn(),
|
|
||||||
on: vi.fn((event: string, handler: () => void) => {
|
|
||||||
onHandlers[event] = handler;
|
|
||||||
return mockRes;
|
|
||||||
}),
|
|
||||||
} as unknown as Response;
|
|
||||||
|
|
||||||
mockPrismaService.agent.findMany.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "agent-1",
|
|
||||||
name: "Worker",
|
|
||||||
status: AgentStatus.WORKING,
|
|
||||||
role: "worker",
|
|
||||||
createdAt: new Date("2026-02-28T12:00:00.000Z"),
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
await controller.streamEvents(mockRes);
|
|
||||||
|
|
||||||
expect(mockRes.setHeader).toHaveBeenCalledWith("Content-Type", "text/event-stream");
|
|
||||||
expect(mockRes.setHeader).toHaveBeenCalledWith("Cache-Control", "no-cache");
|
|
||||||
expect(mockRes.setHeader).toHaveBeenCalledWith("Connection", "keep-alive");
|
|
||||||
expect(mockRes.setHeader).toHaveBeenCalledWith("X-Accel-Buffering", "no");
|
|
||||||
|
|
||||||
expect(mockRes.write).toHaveBeenCalledWith(
|
|
||||||
expect.stringContaining('"type":"agents:updated"')
|
|
||||||
);
|
|
||||||
expect(typeof onHandlers.close).toBe("function");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("polls every 5 seconds and only emits when payload changes", async () => {
|
|
||||||
vi.useFakeTimers();
|
|
||||||
|
|
||||||
const onHandlers: Record<string, (() => void) | undefined> = {};
|
|
||||||
const mockRes = {
|
|
||||||
setHeader: vi.fn(),
|
|
||||||
write: vi.fn(),
|
|
||||||
end: vi.fn(),
|
|
||||||
on: vi.fn((event: string, handler: () => void) => {
|
|
||||||
onHandlers[event] = handler;
|
|
||||||
return mockRes;
|
|
||||||
}),
|
|
||||||
} as unknown as Response;
|
|
||||||
|
|
||||||
const firstPayload = [
|
|
||||||
{
|
|
||||||
id: "agent-1",
|
|
||||||
name: "Worker",
|
|
||||||
status: AgentStatus.WORKING,
|
|
||||||
role: "worker",
|
|
||||||
createdAt: new Date("2026-02-28T12:00:00.000Z"),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
const secondPayload = [
|
|
||||||
{
|
|
||||||
id: "agent-1",
|
|
||||||
name: "Worker",
|
|
||||||
status: AgentStatus.WAITING,
|
|
||||||
role: "worker",
|
|
||||||
createdAt: new Date("2026-02-28T12:00:00.000Z"),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
mockPrismaService.agent.findMany
|
|
||||||
.mockResolvedValueOnce(firstPayload)
|
|
||||||
.mockResolvedValueOnce(firstPayload)
|
|
||||||
.mockResolvedValueOnce(secondPayload);
|
|
||||||
|
|
||||||
await controller.streamEvents(mockRes);
|
|
||||||
|
|
||||||
// 1 initial data event
|
|
||||||
const getDataEventCalls = () =>
|
|
||||||
mockRes.write.mock.calls.filter(
|
|
||||||
(call) => typeof call[0] === "string" && call[0].startsWith("data: ")
|
|
||||||
);
|
|
||||||
|
|
||||||
expect(getDataEventCalls()).toHaveLength(1);
|
|
||||||
|
|
||||||
// No change after first poll => no new data event
|
|
||||||
await vi.advanceTimersByTimeAsync(5000);
|
|
||||||
expect(getDataEventCalls()).toHaveLength(1);
|
|
||||||
|
|
||||||
// Status changed on second poll => emits new data event
|
|
||||||
await vi.advanceTimersByTimeAsync(5000);
|
|
||||||
expect(getDataEventCalls()).toHaveLength(2);
|
|
||||||
|
|
||||||
onHandlers.close?.();
|
|
||||||
expect(mockRes.end).toHaveBeenCalledTimes(1);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("security", () => {
|
|
||||||
it("uses AuthGuard at the controller level", () => {
|
|
||||||
const guards = Reflect.getMetadata("__guards__", OrchestratorController) as unknown[];
|
|
||||||
const guardClasses = guards.map((guard) => guard);
|
|
||||||
|
|
||||||
expect(guardClasses).toContain(AuthGuard);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,115 +0,0 @@
|
|||||||
import { Controller, Get, Res, UseGuards } from "@nestjs/common";
|
|
||||||
import { AgentStatus } from "@prisma/client";
|
|
||||||
import type { Response } from "express";
|
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
|
|
||||||
const AGENT_POLL_INTERVAL_MS = 5_000;
|
|
||||||
const SSE_HEARTBEAT_MS = 15_000;
|
|
||||||
|
|
||||||
interface OrchestratorAgentDto {
|
|
||||||
id: string;
|
|
||||||
name: string | null;
|
|
||||||
status: AgentStatus;
|
|
||||||
type: string;
|
|
||||||
createdAt: Date;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Controller("orchestrator")
|
|
||||||
@UseGuards(AuthGuard)
|
|
||||||
export class OrchestratorController {
|
|
||||||
constructor(private readonly prisma: PrismaService) {}
|
|
||||||
|
|
||||||
@Get("agents")
|
|
||||||
async getAgents(): Promise<OrchestratorAgentDto[]> {
|
|
||||||
return this.fetchActiveAgents();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Get("events")
|
|
||||||
async streamEvents(@Res() res: Response): Promise<void> {
|
|
||||||
res.setHeader("Content-Type", "text/event-stream");
|
|
||||||
res.setHeader("Cache-Control", "no-cache");
|
|
||||||
res.setHeader("Connection", "keep-alive");
|
|
||||||
res.setHeader("X-Accel-Buffering", "no");
|
|
||||||
|
|
||||||
if (typeof res.flushHeaders === "function") {
|
|
||||||
res.flushHeaders();
|
|
||||||
}
|
|
||||||
|
|
||||||
let isClosed = false;
|
|
||||||
let previousSnapshot = "";
|
|
||||||
|
|
||||||
const emitSnapshotIfChanged = async (): Promise<void> => {
|
|
||||||
if (isClosed) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const agents = await this.fetchActiveAgents();
|
|
||||||
const snapshot = JSON.stringify(agents);
|
|
||||||
|
|
||||||
if (snapshot !== previousSnapshot) {
|
|
||||||
previousSnapshot = snapshot;
|
|
||||||
res.write(
|
|
||||||
`data: ${JSON.stringify({
|
|
||||||
type: "agents:updated",
|
|
||||||
agents,
|
|
||||||
timestamp: new Date().toISOString(),
|
|
||||||
})}\n\n`
|
|
||||||
);
|
|
||||||
}
|
|
||||||
} catch (error: unknown) {
|
|
||||||
const message = error instanceof Error ? error.message : String(error);
|
|
||||||
res.write(`event: error\n`);
|
|
||||||
res.write(`data: ${JSON.stringify({ error: message })}\n\n`);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
await emitSnapshotIfChanged();
|
|
||||||
|
|
||||||
const pollInterval = setInterval(() => {
|
|
||||||
void emitSnapshotIfChanged();
|
|
||||||
}, AGENT_POLL_INTERVAL_MS);
|
|
||||||
|
|
||||||
const heartbeatInterval = setInterval(() => {
|
|
||||||
if (!isClosed) {
|
|
||||||
res.write(": keepalive\n\n");
|
|
||||||
}
|
|
||||||
}, SSE_HEARTBEAT_MS);
|
|
||||||
|
|
||||||
res.on("close", () => {
|
|
||||||
isClosed = true;
|
|
||||||
clearInterval(pollInterval);
|
|
||||||
clearInterval(heartbeatInterval);
|
|
||||||
res.end();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
private async fetchActiveAgents(): Promise<OrchestratorAgentDto[]> {
|
|
||||||
const agents = await this.prisma.agent.findMany({
|
|
||||||
where: {
|
|
||||||
status: {
|
|
||||||
not: AgentStatus.TERMINATED,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
orderBy: {
|
|
||||||
createdAt: "desc",
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
name: true,
|
|
||||||
status: true,
|
|
||||||
role: true,
|
|
||||||
createdAt: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
return agents.map((agent) => ({
|
|
||||||
id: agent.id,
|
|
||||||
name: agent.name,
|
|
||||||
status: agent.status,
|
|
||||||
type: agent.role ?? "agent",
|
|
||||||
createdAt: agent.createdAt,
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
import { Module } from "@nestjs/common";
|
|
||||||
import { AuthModule } from "../auth/auth.module";
|
|
||||||
import { PrismaModule } from "../prisma/prisma.module";
|
|
||||||
import { OrchestratorController } from "./orchestrator.controller";
|
|
||||||
|
|
||||||
@Module({
|
|
||||||
imports: [AuthModule, PrismaModule],
|
|
||||||
controllers: [OrchestratorController],
|
|
||||||
})
|
|
||||||
export class OrchestratorModule {}
|
|
||||||
@@ -50,12 +50,6 @@ export class CreateTaskDto {
|
|||||||
@IsUUID("4", { message: "parentId must be a valid UUID" })
|
@IsUUID("4", { message: "parentId must be a valid UUID" })
|
||||||
parentId?: string;
|
parentId?: string;
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "assignedAgent must be a string" })
|
|
||||||
@MinLength(1, { message: "assignedAgent must not be empty" })
|
|
||||||
@MaxLength(255, { message: "assignedAgent must not exceed 255 characters" })
|
|
||||||
assignedAgent?: string;
|
|
||||||
|
|
||||||
@IsOptional()
|
@IsOptional()
|
||||||
@IsInt({ message: "sortOrder must be an integer" })
|
@IsInt({ message: "sortOrder must be an integer" })
|
||||||
@Min(0, { message: "sortOrder must be at least 0" })
|
@Min(0, { message: "sortOrder must be at least 0" })
|
||||||
|
|||||||
@@ -52,12 +52,6 @@ export class UpdateTaskDto {
|
|||||||
@IsUUID("4", { message: "parentId must be a valid UUID" })
|
@IsUUID("4", { message: "parentId must be a valid UUID" })
|
||||||
parentId?: string | null;
|
parentId?: string | null;
|
||||||
|
|
||||||
@IsOptional()
|
|
||||||
@IsString({ message: "assignedAgent must be a string" })
|
|
||||||
@MinLength(1, { message: "assignedAgent must not be empty" })
|
|
||||||
@MaxLength(255, { message: "assignedAgent must not exceed 255 characters" })
|
|
||||||
assignedAgent?: string | null;
|
|
||||||
|
|
||||||
@IsOptional()
|
@IsOptional()
|
||||||
@IsInt({ message: "sortOrder must be an integer" })
|
@IsInt({ message: "sortOrder must be an integer" })
|
||||||
@Min(0, { message: "sortOrder must be at least 0" })
|
@Min(0, { message: "sortOrder must be at least 0" })
|
||||||
|
|||||||
@@ -1,162 +0,0 @@
|
|||||||
import { beforeAll, beforeEach, describe, expect, it, afterAll, vi } from "vitest";
|
|
||||||
import { randomUUID as uuid } from "crypto";
|
|
||||||
import { Test, TestingModule } from "@nestjs/testing";
|
|
||||||
import { PrismaClient } from "@prisma/client";
|
|
||||||
import { TasksService } from "./tasks.service";
|
|
||||||
import { PrismaService } from "../prisma/prisma.service";
|
|
||||||
import { ActivityService } from "../activity/activity.service";
|
|
||||||
|
|
||||||
const shouldRunDbIntegrationTests =
|
|
||||||
process.env.RUN_DB_TESTS === "true" && Boolean(process.env.DATABASE_URL);
|
|
||||||
const describeFn = shouldRunDbIntegrationTests ? describe : describe.skip;
|
|
||||||
|
|
||||||
describeFn("TasksService assignedAgent Integration", () => {
|
|
||||||
let moduleRef: TestingModule;
|
|
||||||
let prisma: PrismaClient;
|
|
||||||
let service: TasksService;
|
|
||||||
let workspaceId: string;
|
|
||||||
let ownerId: string;
|
|
||||||
let setupComplete = false;
|
|
||||||
|
|
||||||
const activityServiceMock = {
|
|
||||||
logTaskCreated: vi.fn().mockResolvedValue(undefined),
|
|
||||||
logTaskUpdated: vi.fn().mockResolvedValue(undefined),
|
|
||||||
logTaskDeleted: vi.fn().mockResolvedValue(undefined),
|
|
||||||
logTaskCompleted: vi.fn().mockResolvedValue(undefined),
|
|
||||||
logTaskAssigned: vi.fn().mockResolvedValue(undefined),
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeAll(async () => {
|
|
||||||
prisma = new PrismaClient();
|
|
||||||
await prisma.$connect();
|
|
||||||
|
|
||||||
const workspace = await prisma.workspace.create({
|
|
||||||
data: {
|
|
||||||
name: `Tasks Assigned Agent Integration ${Date.now()}`,
|
|
||||||
owner: {
|
|
||||||
create: {
|
|
||||||
email: `tasks-assigned-agent-integration-${Date.now()}@example.com`,
|
|
||||||
name: "Tasks Assigned Agent Integration Owner",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
workspaceId = workspace.id;
|
|
||||||
ownerId = workspace.ownerId;
|
|
||||||
|
|
||||||
moduleRef = await Test.createTestingModule({
|
|
||||||
providers: [
|
|
||||||
TasksService,
|
|
||||||
{
|
|
||||||
provide: PrismaService,
|
|
||||||
useValue: prisma,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
provide: ActivityService,
|
|
||||||
useValue: activityServiceMock,
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}).compile();
|
|
||||||
|
|
||||||
service = moduleRef.get<TasksService>(TasksService);
|
|
||||||
setupComplete = true;
|
|
||||||
});
|
|
||||||
|
|
||||||
beforeEach(async () => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
await prisma.task.deleteMany({ where: { workspaceId } });
|
|
||||||
});
|
|
||||||
|
|
||||||
afterAll(async () => {
|
|
||||||
if (!prisma) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (workspaceId) {
|
|
||||||
await prisma.task.deleteMany({ where: { workspaceId } });
|
|
||||||
await prisma.workspace.deleteMany({ where: { id: workspaceId } });
|
|
||||||
}
|
|
||||||
if (ownerId) {
|
|
||||||
await prisma.user.deleteMany({ where: { id: ownerId } });
|
|
||||||
}
|
|
||||||
|
|
||||||
if (moduleRef) {
|
|
||||||
await moduleRef.close();
|
|
||||||
}
|
|
||||||
await prisma.$disconnect();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("persists assignedAgent on create", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const task = await service.create(workspaceId, ownerId, {
|
|
||||||
title: `Assigned agent create ${uuid()}`,
|
|
||||||
assignedAgent: "fleet-worker-1",
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(task.assignedAgent).toBe("fleet-worker-1");
|
|
||||||
|
|
||||||
const stored = await prisma.task.findUnique({
|
|
||||||
where: {
|
|
||||||
id: task.id,
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
id: true,
|
|
||||||
assignedAgent: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(stored).toMatchObject({
|
|
||||||
id: task.id,
|
|
||||||
assignedAgent: "fleet-worker-1",
|
|
||||||
});
|
|
||||||
|
|
||||||
const listed = await service.findAll({ workspaceId, page: 1, limit: 10 }, ownerId);
|
|
||||||
const listedTask = listed.data.find((row) => row.id === task.id);
|
|
||||||
|
|
||||||
expect(listedTask?.assignedAgent).toBe("fleet-worker-1");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("updates and clears assignedAgent", async () => {
|
|
||||||
if (!setupComplete) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const created = await service.create(workspaceId, ownerId, {
|
|
||||||
title: `Assigned agent update ${uuid()}`,
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(created.assignedAgent).toBeNull();
|
|
||||||
|
|
||||||
const updated = await service.update(created.id, workspaceId, ownerId, {
|
|
||||||
assignedAgent: "fleet-worker-2",
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(updated.assignedAgent).toBe("fleet-worker-2");
|
|
||||||
|
|
||||||
const cleared = await service.update(created.id, workspaceId, ownerId, {
|
|
||||||
assignedAgent: null,
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(cleared.assignedAgent).toBeNull();
|
|
||||||
|
|
||||||
const stored = await prisma.task.findUnique({
|
|
||||||
where: {
|
|
||||||
id: created.id,
|
|
||||||
},
|
|
||||||
select: {
|
|
||||||
assignedAgent: true,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(stored?.assignedAgent).toBeNull();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -48,7 +48,6 @@ describe("TasksService", () => {
|
|||||||
creatorId: mockUserId,
|
creatorId: mockUserId,
|
||||||
projectId: null,
|
projectId: null,
|
||||||
parentId: null,
|
parentId: null,
|
||||||
assignedAgent: null,
|
|
||||||
sortOrder: 0,
|
sortOrder: 0,
|
||||||
metadata: {},
|
metadata: {},
|
||||||
createdAt: new Date(),
|
createdAt: new Date(),
|
||||||
@@ -159,28 +158,6 @@ describe("TasksService", () => {
|
|||||||
})
|
})
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it("should include assignedAgent when provided", async () => {
|
|
||||||
const createDto = {
|
|
||||||
title: "Agent-owned Task",
|
|
||||||
assignedAgent: "fleet-worker-1",
|
|
||||||
};
|
|
||||||
|
|
||||||
mockPrismaService.task.create.mockResolvedValue({
|
|
||||||
...mockTask,
|
|
||||||
assignedAgent: createDto.assignedAgent,
|
|
||||||
});
|
|
||||||
|
|
||||||
await service.create(mockWorkspaceId, mockUserId, createDto);
|
|
||||||
|
|
||||||
expect(prisma.task.create).toHaveBeenCalledWith(
|
|
||||||
expect.objectContaining({
|
|
||||||
data: expect.objectContaining({
|
|
||||||
assignedAgent: createDto.assignedAgent,
|
|
||||||
}),
|
|
||||||
})
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|
||||||
describe("findAll", () => {
|
describe("findAll", () => {
|
||||||
@@ -492,26 +469,6 @@ describe("TasksService", () => {
|
|||||||
service.update(mockTaskId, mockWorkspaceId, mockUserId, { title: "Test" })
|
service.update(mockTaskId, mockWorkspaceId, mockUserId, { title: "Test" })
|
||||||
).rejects.toThrow(NotFoundException);
|
).rejects.toThrow(NotFoundException);
|
||||||
});
|
});
|
||||||
|
|
||||||
it("should update assignedAgent when provided", async () => {
|
|
||||||
const updateDto = { assignedAgent: "fleet-worker-2" };
|
|
||||||
|
|
||||||
mockPrismaService.task.findUnique.mockResolvedValue(mockTask);
|
|
||||||
mockPrismaService.task.update.mockResolvedValue({
|
|
||||||
...mockTask,
|
|
||||||
assignedAgent: updateDto.assignedAgent,
|
|
||||||
});
|
|
||||||
|
|
||||||
await service.update(mockTaskId, mockWorkspaceId, mockUserId, updateDto);
|
|
||||||
|
|
||||||
expect(prisma.task.update).toHaveBeenCalledWith(
|
|
||||||
expect.objectContaining({
|
|
||||||
data: expect.objectContaining({
|
|
||||||
assignedAgent: updateDto.assignedAgent,
|
|
||||||
}),
|
|
||||||
})
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|
||||||
describe("remove", () => {
|
describe("remove", () => {
|
||||||
|
|||||||
@@ -67,9 +67,6 @@ export class TasksService {
|
|||||||
metadata: createTaskDto.metadata
|
metadata: createTaskDto.metadata
|
||||||
? (createTaskDto.metadata as unknown as Prisma.InputJsonValue)
|
? (createTaskDto.metadata as unknown as Prisma.InputJsonValue)
|
||||||
: {},
|
: {},
|
||||||
...(createTaskDto.assignedAgent !== undefined && {
|
|
||||||
assignedAgent: createTaskDto.assignedAgent,
|
|
||||||
}),
|
|
||||||
...(assigneeConnection && { assignee: assigneeConnection }),
|
...(assigneeConnection && { assignee: assigneeConnection }),
|
||||||
...(projectConnection && { project: projectConnection }),
|
...(projectConnection && { project: projectConnection }),
|
||||||
...(parentConnection && { parent: parentConnection }),
|
...(parentConnection && { parent: parentConnection }),
|
||||||
@@ -294,9 +291,6 @@ export class TasksService {
|
|||||||
if (updateTaskDto.parentId !== undefined && updateTaskDto.parentId !== null) {
|
if (updateTaskDto.parentId !== undefined && updateTaskDto.parentId !== null) {
|
||||||
data.parent = { connect: { id: updateTaskDto.parentId } };
|
data.parent = { connect: { id: updateTaskDto.parentId } };
|
||||||
}
|
}
|
||||||
if (updateTaskDto.assignedAgent !== undefined) {
|
|
||||||
data.assignedAgent = updateTaskDto.assignedAgent;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle completedAt based on status changes
|
// Handle completedAt based on status changes
|
||||||
if (updateTaskDto.status) {
|
if (updateTaskDto.status) {
|
||||||
|
|||||||
@@ -1,31 +0,0 @@
|
|||||||
import { describe, expect, it } from "vitest";
|
|
||||||
import { WidgetsController } from "./widgets.controller";
|
|
||||||
|
|
||||||
const THROTTLER_SKIP_DEFAULT_KEY = "THROTTLER:SKIPdefault";
|
|
||||||
|
|
||||||
describe("WidgetsController throttler metadata", () => {
|
|
||||||
it("marks widget data polling endpoints to skip throttling", () => {
|
|
||||||
const pollingHandlers = [
|
|
||||||
WidgetsController.prototype.getStatCardData,
|
|
||||||
WidgetsController.prototype.getChartData,
|
|
||||||
WidgetsController.prototype.getListData,
|
|
||||||
WidgetsController.prototype.getCalendarPreviewData,
|
|
||||||
WidgetsController.prototype.getActiveProjectsData,
|
|
||||||
WidgetsController.prototype.getAgentChainsData,
|
|
||||||
];
|
|
||||||
|
|
||||||
for (const handler of pollingHandlers) {
|
|
||||||
expect(Reflect.getMetadata(THROTTLER_SKIP_DEFAULT_KEY, handler)).toBe(true);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
it("does not skip throttling for non-polling widget routes", () => {
|
|
||||||
expect(
|
|
||||||
Reflect.getMetadata(THROTTLER_SKIP_DEFAULT_KEY, WidgetsController.prototype.findAll)
|
|
||||||
).toBe(undefined);
|
|
||||||
|
|
||||||
expect(
|
|
||||||
Reflect.getMetadata(THROTTLER_SKIP_DEFAULT_KEY, WidgetsController.prototype.findByName)
|
|
||||||
).toBe(undefined);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,5 +1,4 @@
|
|||||||
import { Controller, Get, Post, Body, Param, UseGuards, Request } from "@nestjs/common";
|
import { Controller, Get, Post, Body, Param, UseGuards, Request } from "@nestjs/common";
|
||||||
import { SkipThrottle as SkipThrottler } from "@nestjs/throttler";
|
|
||||||
import { WidgetsService } from "./widgets.service";
|
import { WidgetsService } from "./widgets.service";
|
||||||
import { WidgetDataService } from "./widget-data.service";
|
import { WidgetDataService } from "./widget-data.service";
|
||||||
import { AuthGuard } from "../auth/guards/auth.guard";
|
import { AuthGuard } from "../auth/guards/auth.guard";
|
||||||
@@ -44,7 +43,6 @@ export class WidgetsController {
|
|||||||
* Get stat card widget data
|
* Get stat card widget data
|
||||||
*/
|
*/
|
||||||
@Post("data/stat-card")
|
@Post("data/stat-card")
|
||||||
@SkipThrottler()
|
|
||||||
@UseGuards(WorkspaceGuard)
|
@UseGuards(WorkspaceGuard)
|
||||||
async getStatCardData(@Request() req: RequestWithWorkspace, @Body() query: StatCardQueryDto) {
|
async getStatCardData(@Request() req: RequestWithWorkspace, @Body() query: StatCardQueryDto) {
|
||||||
return this.widgetDataService.getStatCardData(req.workspace.id, query);
|
return this.widgetDataService.getStatCardData(req.workspace.id, query);
|
||||||
@@ -55,7 +53,6 @@ export class WidgetsController {
|
|||||||
* Get chart widget data
|
* Get chart widget data
|
||||||
*/
|
*/
|
||||||
@Post("data/chart")
|
@Post("data/chart")
|
||||||
@SkipThrottler()
|
|
||||||
@UseGuards(WorkspaceGuard)
|
@UseGuards(WorkspaceGuard)
|
||||||
async getChartData(@Request() req: RequestWithWorkspace, @Body() query: ChartQueryDto) {
|
async getChartData(@Request() req: RequestWithWorkspace, @Body() query: ChartQueryDto) {
|
||||||
return this.widgetDataService.getChartData(req.workspace.id, query);
|
return this.widgetDataService.getChartData(req.workspace.id, query);
|
||||||
@@ -66,7 +63,6 @@ export class WidgetsController {
|
|||||||
* Get list widget data
|
* Get list widget data
|
||||||
*/
|
*/
|
||||||
@Post("data/list")
|
@Post("data/list")
|
||||||
@SkipThrottler()
|
|
||||||
@UseGuards(WorkspaceGuard)
|
@UseGuards(WorkspaceGuard)
|
||||||
async getListData(@Request() req: RequestWithWorkspace, @Body() query: ListQueryDto) {
|
async getListData(@Request() req: RequestWithWorkspace, @Body() query: ListQueryDto) {
|
||||||
return this.widgetDataService.getListData(req.workspace.id, query);
|
return this.widgetDataService.getListData(req.workspace.id, query);
|
||||||
@@ -77,7 +73,6 @@ export class WidgetsController {
|
|||||||
* Get calendar preview widget data
|
* Get calendar preview widget data
|
||||||
*/
|
*/
|
||||||
@Post("data/calendar-preview")
|
@Post("data/calendar-preview")
|
||||||
@SkipThrottler()
|
|
||||||
@UseGuards(WorkspaceGuard)
|
@UseGuards(WorkspaceGuard)
|
||||||
async getCalendarPreviewData(
|
async getCalendarPreviewData(
|
||||||
@Request() req: RequestWithWorkspace,
|
@Request() req: RequestWithWorkspace,
|
||||||
@@ -91,7 +86,6 @@ export class WidgetsController {
|
|||||||
* Get active projects widget data
|
* Get active projects widget data
|
||||||
*/
|
*/
|
||||||
@Post("data/active-projects")
|
@Post("data/active-projects")
|
||||||
@SkipThrottler()
|
|
||||||
@UseGuards(WorkspaceGuard)
|
@UseGuards(WorkspaceGuard)
|
||||||
async getActiveProjectsData(@Request() req: RequestWithWorkspace) {
|
async getActiveProjectsData(@Request() req: RequestWithWorkspace) {
|
||||||
return this.widgetDataService.getActiveProjectsData(req.workspace.id);
|
return this.widgetDataService.getActiveProjectsData(req.workspace.id);
|
||||||
@@ -102,7 +96,6 @@ export class WidgetsController {
|
|||||||
* Get agent chains widget data (active agent sessions)
|
* Get agent chains widget data (active agent sessions)
|
||||||
*/
|
*/
|
||||||
@Post("data/agent-chains")
|
@Post("data/agent-chains")
|
||||||
@SkipThrottler()
|
|
||||||
@UseGuards(WorkspaceGuard)
|
@UseGuards(WorkspaceGuard)
|
||||||
async getAgentChainsData(@Request() req: RequestWithWorkspace) {
|
async getAgentChainsData(@Request() req: RequestWithWorkspace) {
|
||||||
return this.widgetDataService.getAgentChainsData(req.workspace.id);
|
return this.widgetDataService.getAgentChainsData(req.workspace.id);
|
||||||
|
|||||||
@@ -1,491 +0,0 @@
|
|||||||
"use client";
|
|
||||||
|
|
||||||
import { useState, useEffect, useCallback } from "react";
|
|
||||||
import type { ReactElement } from "react";
|
|
||||||
import { useParams, useRouter } from "next/navigation";
|
|
||||||
import { ArrowLeft } from "lucide-react";
|
|
||||||
|
|
||||||
import { MosaicSpinner } from "@/components/ui/MosaicSpinner";
|
|
||||||
import { fetchProject, type ProjectDetail } from "@/lib/api/projects";
|
|
||||||
import { useWorkspaceId } from "@/lib/hooks";
|
|
||||||
|
|
||||||
interface BadgeStyle {
|
|
||||||
label: string;
|
|
||||||
bg: string;
|
|
||||||
color: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface StatusBadgeProps {
|
|
||||||
style: BadgeStyle;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface MetaItemProps {
|
|
||||||
label: string;
|
|
||||||
value: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
function getProjectStatusStyle(status: string): BadgeStyle {
|
|
||||||
switch (status) {
|
|
||||||
case "PLANNING":
|
|
||||||
return { label: "Planning", bg: "rgba(47,128,255,0.15)", color: "var(--primary)" };
|
|
||||||
case "ACTIVE":
|
|
||||||
return { label: "Active", bg: "rgba(20,184,166,0.15)", color: "var(--success)" };
|
|
||||||
case "PAUSED":
|
|
||||||
return { label: "Paused", bg: "rgba(245,158,11,0.15)", color: "var(--warn)" };
|
|
||||||
case "COMPLETED":
|
|
||||||
return { label: "Completed", bg: "rgba(139,92,246,0.15)", color: "var(--purple)" };
|
|
||||||
case "ARCHIVED":
|
|
||||||
return { label: "Archived", bg: "rgba(143,157,183,0.15)", color: "var(--muted)" };
|
|
||||||
default:
|
|
||||||
return { label: status, bg: "rgba(143,157,183,0.15)", color: "var(--muted)" };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function getPriorityStyle(priority: string | null | undefined): BadgeStyle {
|
|
||||||
switch (priority) {
|
|
||||||
case "HIGH":
|
|
||||||
return { label: "High", bg: "rgba(229,72,77,0.15)", color: "var(--danger)" };
|
|
||||||
case "MEDIUM":
|
|
||||||
return { label: "Medium", bg: "rgba(245,158,11,0.15)", color: "var(--warn)" };
|
|
||||||
case "LOW":
|
|
||||||
return { label: "Low", bg: "rgba(143,157,183,0.15)", color: "var(--muted)" };
|
|
||||||
default:
|
|
||||||
return { label: "Unspecified", bg: "rgba(143,157,183,0.15)", color: "var(--muted)" };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function getTaskStatusStyle(status: string): BadgeStyle {
|
|
||||||
switch (status) {
|
|
||||||
case "NOT_STARTED":
|
|
||||||
return { label: "Not Started", bg: "rgba(47,128,255,0.15)", color: "var(--primary)" };
|
|
||||||
case "IN_PROGRESS":
|
|
||||||
return { label: "In Progress", bg: "rgba(245,158,11,0.15)", color: "var(--warn)" };
|
|
||||||
case "PAUSED":
|
|
||||||
return { label: "Paused", bg: "rgba(143,157,183,0.15)", color: "var(--muted)" };
|
|
||||||
case "COMPLETED":
|
|
||||||
return { label: "Completed", bg: "rgba(20,184,166,0.15)", color: "var(--success)" };
|
|
||||||
case "ARCHIVED":
|
|
||||||
return { label: "Archived", bg: "rgba(143,157,183,0.15)", color: "var(--muted)" };
|
|
||||||
default:
|
|
||||||
return { label: status, bg: "rgba(143,157,183,0.15)", color: "var(--muted)" };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function formatDate(iso: string | null | undefined): string {
|
|
||||||
if (!iso) return "Not set";
|
|
||||||
|
|
||||||
try {
|
|
||||||
return new Date(iso).toLocaleDateString("en-US", {
|
|
||||||
month: "short",
|
|
||||||
day: "numeric",
|
|
||||||
year: "numeric",
|
|
||||||
});
|
|
||||||
} catch {
|
|
||||||
return iso;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function formatDateTime(iso: string | null | undefined): string {
|
|
||||||
if (!iso) return "Not set";
|
|
||||||
|
|
||||||
try {
|
|
||||||
return new Date(iso).toLocaleString("en-US", {
|
|
||||||
month: "short",
|
|
||||||
day: "numeric",
|
|
||||||
year: "numeric",
|
|
||||||
hour: "numeric",
|
|
||||||
minute: "2-digit",
|
|
||||||
});
|
|
||||||
} catch {
|
|
||||||
return iso;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function toFriendlyErrorMessage(error: unknown): string {
|
|
||||||
const fallback = "We had trouble loading this project. Please try again when you're ready.";
|
|
||||||
|
|
||||||
if (!(error instanceof Error)) {
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
const message = error.message.trim();
|
|
||||||
if (message.toLowerCase().includes("not found")) {
|
|
||||||
return "Project not found. It may have been deleted or you may not have access to it.";
|
|
||||||
}
|
|
||||||
|
|
||||||
return message || fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
function StatusBadge({ style: statusStyle }: StatusBadgeProps): ReactElement {
|
|
||||||
return (
|
|
||||||
<span
|
|
||||||
style={{
|
|
||||||
display: "inline-flex",
|
|
||||||
alignItems: "center",
|
|
||||||
padding: "2px 10px",
|
|
||||||
borderRadius: "var(--r)",
|
|
||||||
background: statusStyle.bg,
|
|
||||||
color: statusStyle.color,
|
|
||||||
fontSize: "0.75rem",
|
|
||||||
fontWeight: 500,
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
{statusStyle.label}
|
|
||||||
</span>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
function MetaItem({ label, value }: MetaItemProps): ReactElement {
|
|
||||||
return (
|
|
||||||
<div
|
|
||||||
style={{
|
|
||||||
background: "var(--bg)",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
borderRadius: "var(--r)",
|
|
||||||
padding: "10px 12px",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<p style={{ margin: "0 0 4px", fontSize: "0.75rem", color: "var(--muted)" }}>{label}</p>
|
|
||||||
<p style={{ margin: 0, fontSize: "0.85rem", color: "var(--text)" }}>{value}</p>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
export default function ProjectDetailPage(): ReactElement {
|
|
||||||
const router = useRouter();
|
|
||||||
const params = useParams<{ id: string | string[] }>();
|
|
||||||
const workspaceId = useWorkspaceId();
|
|
||||||
const rawProjectId = params.id;
|
|
||||||
const projectId = Array.isArray(rawProjectId) ? (rawProjectId[0] ?? null) : rawProjectId;
|
|
||||||
|
|
||||||
const [project, setProject] = useState<ProjectDetail | null>(null);
|
|
||||||
const [isLoading, setIsLoading] = useState(true);
|
|
||||||
const [error, setError] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const loadProject = useCallback(async (id: string, wsId: string): Promise<void> => {
|
|
||||||
try {
|
|
||||||
setIsLoading(true);
|
|
||||||
setError(null);
|
|
||||||
const data = await fetchProject(id, wsId);
|
|
||||||
setProject(data);
|
|
||||||
} catch (err: unknown) {
|
|
||||||
console.error("[ProjectDetail] Failed to fetch project:", err);
|
|
||||||
setProject(null);
|
|
||||||
setError(toFriendlyErrorMessage(err));
|
|
||||||
} finally {
|
|
||||||
setIsLoading(false);
|
|
||||||
}
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
if (!projectId) {
|
|
||||||
setProject(null);
|
|
||||||
setError("The project link is invalid. Please return to the projects page.");
|
|
||||||
setIsLoading(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!workspaceId) {
|
|
||||||
setProject(null);
|
|
||||||
setError("Select a workspace to view this project.");
|
|
||||||
setIsLoading(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const id = projectId;
|
|
||||||
const wsId = workspaceId;
|
|
||||||
let cancelled = false;
|
|
||||||
|
|
||||||
async function load(): Promise<void> {
|
|
||||||
try {
|
|
||||||
setIsLoading(true);
|
|
||||||
setError(null);
|
|
||||||
const data = await fetchProject(id, wsId);
|
|
||||||
if (!cancelled) {
|
|
||||||
setProject(data);
|
|
||||||
}
|
|
||||||
} catch (err: unknown) {
|
|
||||||
console.error("[ProjectDetail] Failed to fetch project:", err);
|
|
||||||
if (!cancelled) {
|
|
||||||
setProject(null);
|
|
||||||
setError(toFriendlyErrorMessage(err));
|
|
||||||
}
|
|
||||||
} finally {
|
|
||||||
if (!cancelled) {
|
|
||||||
setIsLoading(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void load();
|
|
||||||
|
|
||||||
return (): void => {
|
|
||||||
cancelled = true;
|
|
||||||
};
|
|
||||||
}, [projectId, workspaceId]);
|
|
||||||
|
|
||||||
function handleRetry(): void {
|
|
||||||
if (!projectId || !workspaceId) return;
|
|
||||||
void loadProject(projectId, workspaceId);
|
|
||||||
}
|
|
||||||
|
|
||||||
function handleBack(): void {
|
|
||||||
router.push("/projects");
|
|
||||||
}
|
|
||||||
|
|
||||||
const projectStatus = project ? getProjectStatusStyle(project.status) : null;
|
|
||||||
const projectPriority = project ? getPriorityStyle(project.priority) : null;
|
|
||||||
const dueDate = project?.dueDate ?? project?.endDate;
|
|
||||||
const creator =
|
|
||||||
project?.creator.name && project.creator.name.trim().length > 0
|
|
||||||
? `${project.creator.name} (${project.creator.email})`
|
|
||||||
: (project?.creator.email ?? "Unknown");
|
|
||||||
|
|
||||||
return (
|
|
||||||
<main className="container mx-auto px-4 py-8" style={{ maxWidth: 960 }}>
|
|
||||||
<button
|
|
||||||
onClick={handleBack}
|
|
||||||
style={{
|
|
||||||
display: "inline-flex",
|
|
||||||
alignItems: "center",
|
|
||||||
gap: 8,
|
|
||||||
marginBottom: 20,
|
|
||||||
padding: "8px 12px",
|
|
||||||
borderRadius: "var(--r)",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
background: "var(--surface)",
|
|
||||||
color: "var(--text-2)",
|
|
||||||
fontSize: "0.85rem",
|
|
||||||
fontWeight: 500,
|
|
||||||
cursor: "pointer",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<ArrowLeft size={16} />
|
|
||||||
Back to projects
|
|
||||||
</button>
|
|
||||||
|
|
||||||
{isLoading ? (
|
|
||||||
<div className="flex justify-center py-16">
|
|
||||||
<MosaicSpinner label="Loading project..." />
|
|
||||||
</div>
|
|
||||||
) : error !== null ? (
|
|
||||||
<div
|
|
||||||
style={{
|
|
||||||
background: "var(--surface)",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
borderRadius: "var(--r-lg)",
|
|
||||||
padding: 32,
|
|
||||||
textAlign: "center",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<p style={{ color: "var(--danger)", margin: "0 0 20px" }}>{error}</p>
|
|
||||||
<div style={{ display: "flex", gap: 12, justifyContent: "center", flexWrap: "wrap" }}>
|
|
||||||
<button
|
|
||||||
onClick={handleBack}
|
|
||||||
style={{
|
|
||||||
padding: "8px 16px",
|
|
||||||
background: "transparent",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
borderRadius: "var(--r)",
|
|
||||||
color: "var(--text-2)",
|
|
||||||
fontSize: "0.85rem",
|
|
||||||
cursor: "pointer",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
Back to projects
|
|
||||||
</button>
|
|
||||||
<button
|
|
||||||
onClick={handleRetry}
|
|
||||||
style={{
|
|
||||||
padding: "8px 16px",
|
|
||||||
background: "var(--danger)",
|
|
||||||
border: "none",
|
|
||||||
borderRadius: "var(--r)",
|
|
||||||
color: "#fff",
|
|
||||||
fontSize: "0.85rem",
|
|
||||||
fontWeight: 500,
|
|
||||||
cursor: "pointer",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
Try again
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
) : project === null ? (
|
|
||||||
<div
|
|
||||||
style={{
|
|
||||||
background: "var(--surface)",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
borderRadius: "var(--r-lg)",
|
|
||||||
padding: 32,
|
|
||||||
textAlign: "center",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<p style={{ color: "var(--muted)", margin: 0 }}>Project details are not available.</p>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<div style={{ display: "flex", flexDirection: "column", gap: 16 }}>
|
|
||||||
<section
|
|
||||||
style={{
|
|
||||||
background: "var(--surface)",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
borderRadius: "var(--r-lg)",
|
|
||||||
padding: 24,
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<div
|
|
||||||
style={{
|
|
||||||
display: "flex",
|
|
||||||
justifyContent: "space-between",
|
|
||||||
alignItems: "flex-start",
|
|
||||||
gap: 12,
|
|
||||||
flexWrap: "wrap",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<div style={{ minWidth: 0 }}>
|
|
||||||
<h1
|
|
||||||
style={{ margin: 0, fontSize: "1.875rem", fontWeight: 700, color: "var(--text)" }}
|
|
||||||
>
|
|
||||||
{project.name}
|
|
||||||
</h1>
|
|
||||||
</div>
|
|
||||||
<div style={{ display: "flex", gap: 8, flexWrap: "wrap" }}>
|
|
||||||
{projectStatus && <StatusBadge style={projectStatus} />}
|
|
||||||
{projectPriority && <StatusBadge style={projectPriority} />}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{project.description ? (
|
|
||||||
<p
|
|
||||||
style={{
|
|
||||||
margin: "14px 0 0",
|
|
||||||
color: "var(--muted)",
|
|
||||||
fontSize: "0.9rem",
|
|
||||||
lineHeight: 1.6,
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
{project.description}
|
|
||||||
</p>
|
|
||||||
) : (
|
|
||||||
<p
|
|
||||||
style={{
|
|
||||||
margin: "14px 0 0",
|
|
||||||
color: "var(--muted)",
|
|
||||||
fontSize: "0.9rem",
|
|
||||||
lineHeight: 1.6,
|
|
||||||
fontStyle: "italic",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
No description provided.
|
|
||||||
</p>
|
|
||||||
)}
|
|
||||||
|
|
||||||
<div className="grid grid-cols-1 md:grid-cols-2 gap-3" style={{ marginTop: 18 }}>
|
|
||||||
<MetaItem label="Start date" value={formatDate(project.startDate)} />
|
|
||||||
<MetaItem label="Due date" value={formatDate(dueDate)} />
|
|
||||||
<MetaItem label="Created" value={formatDateTime(project.createdAt)} />
|
|
||||||
<MetaItem label="Updated" value={formatDateTime(project.updatedAt)} />
|
|
||||||
<MetaItem label="Creator" value={creator} />
|
|
||||||
<MetaItem
|
|
||||||
label="Work items"
|
|
||||||
value={`${String(project._count.tasks)} tasks · ${String(project._count.events)} events`}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
</section>
|
|
||||||
|
|
||||||
<section
|
|
||||||
style={{
|
|
||||||
background: "var(--surface)",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
borderRadius: "var(--r-lg)",
|
|
||||||
padding: 24,
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<h2 style={{ margin: "0 0 12px", fontSize: "1.1rem", color: "var(--text)" }}>
|
|
||||||
Tasks ({String(project._count.tasks)})
|
|
||||||
</h2>
|
|
||||||
|
|
||||||
{project.tasks.length === 0 ? (
|
|
||||||
<p style={{ margin: 0, color: "var(--muted)", fontSize: "0.9rem" }}>
|
|
||||||
No tasks yet for this project.
|
|
||||||
</p>
|
|
||||||
) : (
|
|
||||||
<div>
|
|
||||||
{project.tasks.map((task, index) => (
|
|
||||||
<div
|
|
||||||
key={task.id}
|
|
||||||
style={{
|
|
||||||
padding: "12px 0",
|
|
||||||
borderTop: index === 0 ? "none" : "1px solid var(--border)",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<div
|
|
||||||
style={{
|
|
||||||
display: "flex",
|
|
||||||
alignItems: "flex-start",
|
|
||||||
justifyContent: "space-between",
|
|
||||||
gap: 12,
|
|
||||||
flexWrap: "wrap",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<div style={{ minWidth: 0 }}>
|
|
||||||
<p style={{ margin: 0, color: "var(--text)", fontWeight: 500 }}>
|
|
||||||
{task.title}
|
|
||||||
</p>
|
|
||||||
<p style={{ margin: "4px 0 0", color: "var(--muted)", fontSize: "0.8rem" }}>
|
|
||||||
Due: {formatDate(task.dueDate)}
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
<div style={{ display: "flex", gap: 8, flexWrap: "wrap" }}>
|
|
||||||
<StatusBadge style={getTaskStatusStyle(task.status)} />
|
|
||||||
<StatusBadge style={getPriorityStyle(task.priority)} />
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</section>
|
|
||||||
|
|
||||||
<section
|
|
||||||
style={{
|
|
||||||
background: "var(--surface)",
|
|
||||||
border: "1px solid var(--border)",
|
|
||||||
borderRadius: "var(--r-lg)",
|
|
||||||
padding: 24,
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<h2 style={{ margin: "0 0 12px", fontSize: "1.1rem", color: "var(--text)" }}>
|
|
||||||
Events ({String(project._count.events)})
|
|
||||||
</h2>
|
|
||||||
|
|
||||||
{project.events.length === 0 ? (
|
|
||||||
<p style={{ margin: 0, color: "var(--muted)", fontSize: "0.9rem" }}>
|
|
||||||
No events scheduled for this project.
|
|
||||||
</p>
|
|
||||||
) : (
|
|
||||||
<div>
|
|
||||||
{project.events.map((event, index) => (
|
|
||||||
<div
|
|
||||||
key={event.id}
|
|
||||||
style={{
|
|
||||||
padding: "12px 0",
|
|
||||||
borderTop: index === 0 ? "none" : "1px solid var(--border)",
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<p style={{ margin: 0, color: "var(--text)", fontWeight: 500 }}>
|
|
||||||
{event.title}
|
|
||||||
</p>
|
|
||||||
<p style={{ margin: "4px 0 0", color: "var(--muted)", fontSize: "0.8rem" }}>
|
|
||||||
{formatDateTime(event.startTime)} - {formatDateTime(event.endTime)}
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</section>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</main>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -1,356 +0,0 @@
|
|||||||
"use client";
|
|
||||||
|
|
||||||
import {
|
|
||||||
useCallback,
|
|
||||||
useEffect,
|
|
||||||
useMemo,
|
|
||||||
useState,
|
|
||||||
type ChangeEvent,
|
|
||||||
type ReactElement,
|
|
||||||
type SyntheticEvent,
|
|
||||||
} from "react";
|
|
||||||
import { FleetSettingsNav } from "@/components/settings/FleetSettingsNav";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import {
|
|
||||||
Select,
|
|
||||||
SelectContent,
|
|
||||||
SelectItem,
|
|
||||||
SelectTrigger,
|
|
||||||
SelectValue,
|
|
||||||
} from "@/components/ui/select";
|
|
||||||
import { Textarea } from "@/components/ui/textarea";
|
|
||||||
import {
|
|
||||||
fetchFleetAgentConfig,
|
|
||||||
fetchFleetProviders,
|
|
||||||
updateFleetAgentConfig,
|
|
||||||
type FleetProvider,
|
|
||||||
type FleetProviderModel,
|
|
||||||
type UpdateFleetAgentConfigRequest,
|
|
||||||
} from "@/lib/api/fleet-settings";
|
|
||||||
|
|
||||||
function getErrorMessage(error: unknown, fallback: string): string {
|
|
||||||
if (error instanceof Error && error.message.trim().length > 0) {
|
|
||||||
return error.message;
|
|
||||||
}
|
|
||||||
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
function normalizeProviderModels(models: unknown): FleetProviderModel[] {
|
|
||||||
if (!Array.isArray(models)) {
|
|
||||||
return [];
|
|
||||||
}
|
|
||||||
|
|
||||||
const parsed: FleetProviderModel[] = [];
|
|
||||||
|
|
||||||
models.forEach((entry) => {
|
|
||||||
if (typeof entry === "string" && entry.trim().length > 0) {
|
|
||||||
parsed.push({ id: entry.trim(), name: entry.trim() });
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (entry && typeof entry === "object") {
|
|
||||||
const record = entry as Record<string, unknown>;
|
|
||||||
const id =
|
|
||||||
typeof record.id === "string"
|
|
||||||
? record.id.trim()
|
|
||||||
: typeof record.name === "string"
|
|
||||||
? record.name.trim()
|
|
||||||
: "";
|
|
||||||
|
|
||||||
if (id.length > 0) {
|
|
||||||
parsed.push({ id, name: id });
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
const seen = new Set<string>();
|
|
||||||
return parsed.filter((model) => {
|
|
||||||
if (seen.has(model.id)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
seen.add(model.id);
|
|
||||||
return true;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function parseModelList(value: string): string[] {
|
|
||||||
const seen = new Set<string>();
|
|
||||||
|
|
||||||
return value
|
|
||||||
.split(/\n|,/g)
|
|
||||||
.map((segment) => segment.trim())
|
|
||||||
.filter((segment) => segment.length > 0)
|
|
||||||
.filter((segment) => {
|
|
||||||
if (seen.has(segment)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
seen.add(segment);
|
|
||||||
return true;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function deriveAvailableModels(providers: FleetProvider[]): string[] {
|
|
||||||
const seen = new Set<string>();
|
|
||||||
const models: string[] = [];
|
|
||||||
|
|
||||||
providers.forEach((provider) => {
|
|
||||||
normalizeProviderModels(provider.models).forEach((model) => {
|
|
||||||
if (seen.has(model.id)) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
seen.add(model.id);
|
|
||||||
models.push(model.id);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
return models.sort((left, right) => left.localeCompare(right));
|
|
||||||
}
|
|
||||||
|
|
||||||
export default function AgentConfigSettingsPage(): ReactElement {
|
|
||||||
const [providers, setProviders] = useState<FleetProvider[]>([]);
|
|
||||||
const [primaryModel, setPrimaryModel] = useState<string>("");
|
|
||||||
const [fallbackModelsText, setFallbackModelsText] = useState<string>("");
|
|
||||||
const [personality, setPersonality] = useState<string>("");
|
|
||||||
|
|
||||||
const [isLoading, setIsLoading] = useState<boolean>(true);
|
|
||||||
const [isSaving, setIsSaving] = useState<boolean>(false);
|
|
||||||
const [error, setError] = useState<string | null>(null);
|
|
||||||
const [successMessage, setSuccessMessage] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const availableModels = useMemo(() => deriveAvailableModels(providers), [providers]);
|
|
||||||
const fallbackModels = useMemo(() => parseModelList(fallbackModelsText), [fallbackModelsText]);
|
|
||||||
|
|
||||||
const modelSelectOptions = useMemo(() => {
|
|
||||||
if (primaryModel.length > 0 && !availableModels.includes(primaryModel)) {
|
|
||||||
return [primaryModel, ...availableModels];
|
|
||||||
}
|
|
||||||
|
|
||||||
return availableModels;
|
|
||||||
}, [availableModels, primaryModel]);
|
|
||||||
|
|
||||||
const loadSettings = useCallback(async (): Promise<void> => {
|
|
||||||
setIsLoading(true);
|
|
||||||
|
|
||||||
try {
|
|
||||||
const [providerData, agentConfig] = await Promise.all([
|
|
||||||
fetchFleetProviders(),
|
|
||||||
fetchFleetAgentConfig(),
|
|
||||||
]);
|
|
||||||
setProviders(providerData);
|
|
||||||
setPrimaryModel(agentConfig.primaryModel ?? "");
|
|
||||||
setFallbackModelsText(agentConfig.fallbackModels.join("\n"));
|
|
||||||
setPersonality(agentConfig.personality ?? "");
|
|
||||||
setError(null);
|
|
||||||
} catch (loadError: unknown) {
|
|
||||||
setError(getErrorMessage(loadError, "Failed to load agent configuration."));
|
|
||||||
} finally {
|
|
||||||
setIsLoading(false);
|
|
||||||
}
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
void loadSettings();
|
|
||||||
}, [loadSettings]);
|
|
||||||
|
|
||||||
function appendFallbackModel(model: string): void {
|
|
||||||
const current = parseModelList(fallbackModelsText);
|
|
||||||
if (current.includes(model)) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const next = [...current, model];
|
|
||||||
setFallbackModelsText(next.join("\n"));
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleSave(event: SyntheticEvent): Promise<void> {
|
|
||||||
event.preventDefault();
|
|
||||||
setError(null);
|
|
||||||
setSuccessMessage(null);
|
|
||||||
|
|
||||||
const updatePayload: UpdateFleetAgentConfigRequest = {
|
|
||||||
personality: personality.trim(),
|
|
||||||
};
|
|
||||||
|
|
||||||
if (primaryModel.trim().length > 0) {
|
|
||||||
updatePayload.primaryModel = primaryModel.trim();
|
|
||||||
}
|
|
||||||
|
|
||||||
const parsedFallbacks = parseModelList(fallbackModelsText).filter(
|
|
||||||
(model) => model !== primaryModel.trim()
|
|
||||||
);
|
|
||||||
if (parsedFallbacks.length > 0) {
|
|
||||||
updatePayload.fallbackModels = parsedFallbacks;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsSaving(true);
|
|
||||||
await updateFleetAgentConfig(updatePayload);
|
|
||||||
setSuccessMessage("Agent configuration saved.");
|
|
||||||
await loadSettings();
|
|
||||||
} catch (saveError: unknown) {
|
|
||||||
setError(getErrorMessage(saveError, "Failed to save agent configuration."));
|
|
||||||
} finally {
|
|
||||||
setIsSaving(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div className="max-w-6xl mx-auto p-6 space-y-6">
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div>
|
|
||||||
<h1 className="text-3xl font-bold">Agent Configuration</h1>
|
|
||||||
<p className="text-muted-foreground mt-1">
|
|
||||||
Assign primary and fallback models for your agent runtime behavior.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
<FleetSettingsNav />
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Card>
|
|
||||||
<CardHeader>
|
|
||||||
<CardTitle>Current Assignment</CardTitle>
|
|
||||||
<CardDescription>
|
|
||||||
Snapshot of your currently saved model routing configuration.
|
|
||||||
</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent className="space-y-4">
|
|
||||||
{isLoading ? (
|
|
||||||
<p className="text-sm text-muted-foreground">Loading configuration...</p>
|
|
||||||
) : (
|
|
||||||
<>
|
|
||||||
<div>
|
|
||||||
<p className="text-sm font-medium">Primary Model</p>
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
{primaryModel.length > 0 ? primaryModel : "No primary model configured"}
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div>
|
|
||||||
<p className="text-sm font-medium">Fallback Models</p>
|
|
||||||
{fallbackModels.length === 0 ? (
|
|
||||||
<p className="text-sm text-muted-foreground">No fallback models configured</p>
|
|
||||||
) : (
|
|
||||||
<div className="flex flex-wrap gap-2 mt-2">
|
|
||||||
{fallbackModels.map((model) => (
|
|
||||||
<Badge key={`current-${model}`} variant="outline">
|
|
||||||
{model}
|
|
||||||
</Badge>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
</>
|
|
||||||
)}
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
|
|
||||||
<Card>
|
|
||||||
<CardHeader>
|
|
||||||
<CardTitle>Update Agent Config</CardTitle>
|
|
||||||
<CardDescription>
|
|
||||||
Select a primary model and define fallback ordering. Models come from your provider
|
|
||||||
settings.
|
|
||||||
</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent>
|
|
||||||
<form onSubmit={(event) => void handleSave(event)} className="space-y-5">
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="primary-model">Primary Model</Label>
|
|
||||||
<Select
|
|
||||||
value={primaryModel.length > 0 ? primaryModel : "__none__"}
|
|
||||||
onValueChange={(value) => {
|
|
||||||
setPrimaryModel(value === "__none__" ? "" : value);
|
|
||||||
}}
|
|
||||||
disabled={isLoading || isSaving}
|
|
||||||
>
|
|
||||||
<SelectTrigger id="primary-model">
|
|
||||||
<SelectValue placeholder="Select a primary model" />
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent>
|
|
||||||
<SelectItem value="__none__">No primary model selected</SelectItem>
|
|
||||||
{modelSelectOptions.map((model) => (
|
|
||||||
<SelectItem key={model} value={model}>
|
|
||||||
{model}
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
{availableModels.length === 0 ? (
|
|
||||||
<p className="text-xs text-muted-foreground">
|
|
||||||
No models available yet. Add provider models first in Providers settings.
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="fallback-models">Fallback Models</Label>
|
|
||||||
<Textarea
|
|
||||||
id="fallback-models"
|
|
||||||
value={fallbackModelsText}
|
|
||||||
onChange={(event: ChangeEvent<HTMLTextAreaElement>) => {
|
|
||||||
setFallbackModelsText(event.target.value);
|
|
||||||
}}
|
|
||||||
rows={4}
|
|
||||||
placeholder={"One model per line\nExample: gpt-4.1-mini"}
|
|
||||||
disabled={isLoading || isSaving}
|
|
||||||
/>
|
|
||||||
{availableModels.length > 0 ? (
|
|
||||||
<div className="flex flex-wrap gap-2">
|
|
||||||
{availableModels
|
|
||||||
.filter((model) => model !== primaryModel)
|
|
||||||
.map((model) => (
|
|
||||||
<Button
|
|
||||||
key={`suggest-${model}`}
|
|
||||||
type="button"
|
|
||||||
variant="outline"
|
|
||||||
size="sm"
|
|
||||||
onClick={() => {
|
|
||||||
appendFallbackModel(model);
|
|
||||||
}}
|
|
||||||
disabled={fallbackModels.includes(model) || isSaving}
|
|
||||||
>
|
|
||||||
{fallbackModels.includes(model) ? `Added: ${model}` : `Add ${model}`}
|
|
||||||
</Button>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
) : null}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="agent-personality">Personality / SOUL</Label>
|
|
||||||
<Textarea
|
|
||||||
id="agent-personality"
|
|
||||||
value={personality}
|
|
||||||
onChange={(event: ChangeEvent<HTMLTextAreaElement>) => {
|
|
||||||
setPersonality(event.target.value);
|
|
||||||
}}
|
|
||||||
rows={8}
|
|
||||||
placeholder="Optional system personality instructions..."
|
|
||||||
disabled={isLoading || isSaving}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{error ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{error}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
{successMessage ? <p className="text-sm text-emerald-600">{successMessage}</p> : null}
|
|
||||||
|
|
||||||
<Button type="submit" disabled={isLoading || isSaving}>
|
|
||||||
{isSaving ? "Saving..." : "Save Agent Config"}
|
|
||||||
</Button>
|
|
||||||
</form>
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -1,492 +0,0 @@
|
|||||||
"use client";
|
|
||||||
|
|
||||||
import {
|
|
||||||
useCallback,
|
|
||||||
useEffect,
|
|
||||||
useState,
|
|
||||||
type ChangeEvent,
|
|
||||||
type ReactElement,
|
|
||||||
type SyntheticEvent,
|
|
||||||
} from "react";
|
|
||||||
import { FleetSettingsNav } from "@/components/settings/FleetSettingsNav";
|
|
||||||
import { SettingsAccessDenied } from "@/components/settings/SettingsAccessDenied";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
|
||||||
import {
|
|
||||||
AlertDialog,
|
|
||||||
AlertDialogAction,
|
|
||||||
AlertDialogCancel,
|
|
||||||
AlertDialogContent,
|
|
||||||
AlertDialogDescription,
|
|
||||||
AlertDialogFooter,
|
|
||||||
AlertDialogHeader,
|
|
||||||
AlertDialogTitle,
|
|
||||||
} from "@/components/ui/alert-dialog";
|
|
||||||
import { Input } from "@/components/ui/input";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import {
|
|
||||||
deleteFleetOidcConfig,
|
|
||||||
fetchFleetOidcConfig,
|
|
||||||
resetBreakglassAdminPassword,
|
|
||||||
updateFleetOidcConfig,
|
|
||||||
type FleetOidcConfig,
|
|
||||||
} from "@/lib/api/fleet-settings";
|
|
||||||
import { fetchOnboardingStatus } from "@/lib/api/onboarding";
|
|
||||||
|
|
||||||
interface OidcFormState {
|
|
||||||
issuerUrl: string;
|
|
||||||
clientId: string;
|
|
||||||
clientSecret: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface BreakglassFormState {
|
|
||||||
username: string;
|
|
||||||
currentPassword: string;
|
|
||||||
newPassword: string;
|
|
||||||
confirmPassword: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
const INITIAL_OIDC_FORM: OidcFormState = {
|
|
||||||
issuerUrl: "",
|
|
||||||
clientId: "",
|
|
||||||
clientSecret: "",
|
|
||||||
};
|
|
||||||
|
|
||||||
const INITIAL_BREAKGLASS_FORM: BreakglassFormState = {
|
|
||||||
username: "",
|
|
||||||
currentPassword: "",
|
|
||||||
newPassword: "",
|
|
||||||
confirmPassword: "",
|
|
||||||
};
|
|
||||||
|
|
||||||
function getErrorMessage(error: unknown, fallback: string): string {
|
|
||||||
if (error instanceof Error && error.message.trim().length > 0) {
|
|
||||||
return error.message;
|
|
||||||
}
|
|
||||||
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
function isAdminGuardError(error: unknown): boolean {
|
|
||||||
if (!(error instanceof Error)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
const normalized = error.message.toLowerCase();
|
|
||||||
return (
|
|
||||||
normalized.includes("requires system administrator") ||
|
|
||||||
normalized.includes("forbidden") ||
|
|
||||||
normalized.includes("403")
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
export default function AuthSettingsPage(): ReactElement {
|
|
||||||
const [oidcConfig, setOidcConfig] = useState<FleetOidcConfig | null>(null);
|
|
||||||
const [oidcForm, setOidcForm] = useState<OidcFormState>(INITIAL_OIDC_FORM);
|
|
||||||
const [isLoading, setIsLoading] = useState<boolean>(true);
|
|
||||||
const [isSavingOidc, setIsSavingOidc] = useState<boolean>(false);
|
|
||||||
const [isDeletingOidc, setIsDeletingOidc] = useState<boolean>(false);
|
|
||||||
const [oidcError, setOidcError] = useState<string | null>(null);
|
|
||||||
const [oidcSuccessMessage, setOidcSuccessMessage] = useState<string | null>(null);
|
|
||||||
const [showRemoveOidcDialog, setShowRemoveOidcDialog] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const [breakglassForm, setBreakglassForm] =
|
|
||||||
useState<BreakglassFormState>(INITIAL_BREAKGLASS_FORM);
|
|
||||||
const [breakglassStatus, setBreakglassStatus] = useState<"active" | "inactive">("inactive");
|
|
||||||
const [isResettingPassword, setIsResettingPassword] = useState<boolean>(false);
|
|
||||||
const [breakglassError, setBreakglassError] = useState<string | null>(null);
|
|
||||||
const [breakglassSuccessMessage, setBreakglassSuccessMessage] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const [isAccessDenied, setIsAccessDenied] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const loadAuthSettings = useCallback(async (): Promise<void> => {
|
|
||||||
setIsLoading(true);
|
|
||||||
|
|
||||||
try {
|
|
||||||
const [oidcResponse, onboardingStatus] = await Promise.all([
|
|
||||||
fetchFleetOidcConfig(),
|
|
||||||
fetchOnboardingStatus().catch(() => ({ completed: false })),
|
|
||||||
]);
|
|
||||||
|
|
||||||
setOidcConfig(oidcResponse);
|
|
||||||
setOidcForm({
|
|
||||||
issuerUrl: oidcResponse.issuerUrl ?? "",
|
|
||||||
clientId: oidcResponse.clientId ?? "",
|
|
||||||
clientSecret: "",
|
|
||||||
});
|
|
||||||
setBreakglassStatus(onboardingStatus.completed ? "active" : "inactive");
|
|
||||||
setIsAccessDenied(false);
|
|
||||||
setOidcError(null);
|
|
||||||
} catch (loadError: unknown) {
|
|
||||||
if (isAdminGuardError(loadError)) {
|
|
||||||
setIsAccessDenied(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setOidcError(getErrorMessage(loadError, "Failed to load authentication settings."));
|
|
||||||
} finally {
|
|
||||||
setIsLoading(false);
|
|
||||||
}
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
void loadAuthSettings();
|
|
||||||
}, [loadAuthSettings]);
|
|
||||||
|
|
||||||
async function handleSaveOidc(event: SyntheticEvent): Promise<void> {
|
|
||||||
event.preventDefault();
|
|
||||||
setOidcError(null);
|
|
||||||
setOidcSuccessMessage(null);
|
|
||||||
|
|
||||||
const issuerUrl = oidcForm.issuerUrl.trim();
|
|
||||||
const clientId = oidcForm.clientId.trim();
|
|
||||||
const clientSecret = oidcForm.clientSecret.trim();
|
|
||||||
|
|
||||||
if (issuerUrl.length === 0 || clientId.length === 0 || clientSecret.length === 0) {
|
|
||||||
setOidcError("Issuer URL, client ID, and client secret are required.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsSavingOidc(true);
|
|
||||||
await updateFleetOidcConfig({
|
|
||||||
issuerUrl,
|
|
||||||
clientId,
|
|
||||||
clientSecret,
|
|
||||||
});
|
|
||||||
setOidcSuccessMessage("OIDC configuration updated.");
|
|
||||||
await loadAuthSettings();
|
|
||||||
} catch (saveError: unknown) {
|
|
||||||
setOidcError(getErrorMessage(saveError, "Failed to update OIDC configuration."));
|
|
||||||
} finally {
|
|
||||||
setIsSavingOidc(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleRemoveOidc(): Promise<void> {
|
|
||||||
try {
|
|
||||||
setIsDeletingOidc(true);
|
|
||||||
await deleteFleetOidcConfig();
|
|
||||||
setOidcSuccessMessage("OIDC configuration removed.");
|
|
||||||
setShowRemoveOidcDialog(false);
|
|
||||||
await loadAuthSettings();
|
|
||||||
} catch (deleteError: unknown) {
|
|
||||||
setOidcError(getErrorMessage(deleteError, "Failed to remove OIDC configuration."));
|
|
||||||
} finally {
|
|
||||||
setIsDeletingOidc(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleResetBreakglassPassword(event: SyntheticEvent): Promise<void> {
|
|
||||||
event.preventDefault();
|
|
||||||
setBreakglassError(null);
|
|
||||||
setBreakglassSuccessMessage(null);
|
|
||||||
|
|
||||||
const username = breakglassForm.username.trim();
|
|
||||||
const newPassword = breakglassForm.newPassword;
|
|
||||||
const confirmPassword = breakglassForm.confirmPassword;
|
|
||||||
|
|
||||||
if (username.length === 0) {
|
|
||||||
setBreakglassError("Username is required.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (newPassword.length < 8) {
|
|
||||||
setBreakglassError("New password must be at least 8 characters.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (newPassword !== confirmPassword) {
|
|
||||||
setBreakglassError("Password confirmation does not match.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsResettingPassword(true);
|
|
||||||
await resetBreakglassAdminPassword({
|
|
||||||
username,
|
|
||||||
newPassword,
|
|
||||||
});
|
|
||||||
setBreakglassSuccessMessage(`Password reset for "${username}".`);
|
|
||||||
setBreakglassStatus("active");
|
|
||||||
setBreakglassForm((previous) => ({
|
|
||||||
...previous,
|
|
||||||
currentPassword: "",
|
|
||||||
newPassword: "",
|
|
||||||
confirmPassword: "",
|
|
||||||
}));
|
|
||||||
} catch (resetError: unknown) {
|
|
||||||
setBreakglassError(getErrorMessage(resetError, "Failed to reset breakglass password."));
|
|
||||||
} finally {
|
|
||||||
setIsResettingPassword(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div className="max-w-6xl mx-auto p-6 space-y-6">
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div>
|
|
||||||
<h1 className="text-3xl font-bold">Authentication Settings</h1>
|
|
||||||
<p className="text-muted-foreground mt-1">
|
|
||||||
Configure OIDC and breakglass admin recovery credentials.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
<FleetSettingsNav />
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{isLoading ? (
|
|
||||||
<Card>
|
|
||||||
<CardContent className="py-8 text-sm text-muted-foreground">
|
|
||||||
Loading authentication settings...
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
{!isLoading && isAccessDenied ? (
|
|
||||||
<SettingsAccessDenied message="Authentication settings require system administrator privileges." />
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
{!isLoading && !isAccessDenied ? (
|
|
||||||
<>
|
|
||||||
<Card>
|
|
||||||
<CardHeader>
|
|
||||||
<CardTitle>OIDC Provider</CardTitle>
|
|
||||||
<CardDescription>
|
|
||||||
Manage your OpenID Connect issuer and OAuth client credentials.
|
|
||||||
</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent className="space-y-5">
|
|
||||||
<div className="rounded-lg border p-4 space-y-2">
|
|
||||||
<div className="flex items-center gap-2">
|
|
||||||
<p className="font-medium">Configured</p>
|
|
||||||
<Badge variant={oidcConfig?.configured ? "default" : "secondary"}>
|
|
||||||
{oidcConfig?.configured ? "Yes" : "No"}
|
|
||||||
</Badge>
|
|
||||||
</div>
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
Issuer URL: {oidcConfig?.issuerUrl ?? "Not configured"}
|
|
||||||
</p>
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
Client ID: {oidcConfig?.clientId ?? "Not configured"}
|
|
||||||
</p>
|
|
||||||
<p className="text-sm text-muted-foreground">Client secret: hidden</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<form onSubmit={(event) => void handleSaveOidc(event)} className="space-y-4">
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="oidc-issuer-url">Issuer URL</Label>
|
|
||||||
<Input
|
|
||||||
id="oidc-issuer-url"
|
|
||||||
value={oidcForm.issuerUrl}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setOidcForm((previous) => ({ ...previous, issuerUrl: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="https://issuer.example.com"
|
|
||||||
disabled={isSavingOidc}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="oidc-client-id">Client ID</Label>
|
|
||||||
<Input
|
|
||||||
id="oidc-client-id"
|
|
||||||
value={oidcForm.clientId}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setOidcForm((previous) => ({ ...previous, clientId: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="mosaic-web"
|
|
||||||
disabled={isSavingOidc}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="oidc-client-secret">Client Secret</Label>
|
|
||||||
<Input
|
|
||||||
id="oidc-client-secret"
|
|
||||||
type="password"
|
|
||||||
value={oidcForm.clientSecret}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setOidcForm((previous) => ({
|
|
||||||
...previous,
|
|
||||||
clientSecret: event.target.value,
|
|
||||||
}));
|
|
||||||
}}
|
|
||||||
placeholder="Enter new secret"
|
|
||||||
autoComplete="new-password"
|
|
||||||
disabled={isSavingOidc}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
<p className="text-xs text-muted-foreground">
|
|
||||||
The secret is encrypted on save and never returned to the UI.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{oidcError ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{oidcError}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
{oidcSuccessMessage ? (
|
|
||||||
<p className="text-sm text-emerald-600">{oidcSuccessMessage}</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<div className="flex items-center gap-2">
|
|
||||||
<Button type="submit" disabled={isSavingOidc}>
|
|
||||||
{isSavingOidc ? "Saving..." : "Save OIDC"}
|
|
||||||
</Button>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="destructive"
|
|
||||||
onClick={() => {
|
|
||||||
setShowRemoveOidcDialog(true);
|
|
||||||
}}
|
|
||||||
disabled={isDeletingOidc || !oidcConfig?.configured}
|
|
||||||
>
|
|
||||||
Remove OIDC
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
|
|
||||||
<Card>
|
|
||||||
<CardHeader>
|
|
||||||
<CardTitle>Breakglass Admin</CardTitle>
|
|
||||||
<CardDescription>
|
|
||||||
Reset breakglass credentials for emergency local access.
|
|
||||||
</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent className="space-y-5">
|
|
||||||
<div className="flex items-center gap-2">
|
|
||||||
<p className="font-medium">Status</p>
|
|
||||||
<Badge variant={breakglassStatus === "active" ? "default" : "secondary"}>
|
|
||||||
{breakglassStatus}
|
|
||||||
</Badge>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<form
|
|
||||||
onSubmit={(event) => void handleResetBreakglassPassword(event)}
|
|
||||||
className="space-y-4"
|
|
||||||
>
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="breakglass-username">Username</Label>
|
|
||||||
<Input
|
|
||||||
id="breakglass-username"
|
|
||||||
value={breakglassForm.username}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setBreakglassForm((previous) => ({
|
|
||||||
...previous,
|
|
||||||
username: event.target.value,
|
|
||||||
}));
|
|
||||||
}}
|
|
||||||
placeholder="admin"
|
|
||||||
disabled={isResettingPassword}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="breakglass-current-password">Current Password (optional)</Label>
|
|
||||||
<Input
|
|
||||||
id="breakglass-current-password"
|
|
||||||
type="password"
|
|
||||||
value={breakglassForm.currentPassword}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setBreakglassForm((previous) => ({
|
|
||||||
...previous,
|
|
||||||
currentPassword: event.target.value,
|
|
||||||
}));
|
|
||||||
}}
|
|
||||||
placeholder="Optional operator confirmation"
|
|
||||||
autoComplete="current-password"
|
|
||||||
disabled={isResettingPassword}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="breakglass-new-password">New Password</Label>
|
|
||||||
<Input
|
|
||||||
id="breakglass-new-password"
|
|
||||||
type="password"
|
|
||||||
value={breakglassForm.newPassword}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setBreakglassForm((previous) => ({
|
|
||||||
...previous,
|
|
||||||
newPassword: event.target.value,
|
|
||||||
}));
|
|
||||||
}}
|
|
||||||
placeholder="At least 8 characters"
|
|
||||||
autoComplete="new-password"
|
|
||||||
disabled={isResettingPassword}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="breakglass-confirm-password">Confirm Password</Label>
|
|
||||||
<Input
|
|
||||||
id="breakglass-confirm-password"
|
|
||||||
type="password"
|
|
||||||
value={breakglassForm.confirmPassword}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setBreakglassForm((previous) => ({
|
|
||||||
...previous,
|
|
||||||
confirmPassword: event.target.value,
|
|
||||||
}));
|
|
||||||
}}
|
|
||||||
placeholder="Re-enter password"
|
|
||||||
autoComplete="new-password"
|
|
||||||
disabled={isResettingPassword}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{breakglassError ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{breakglassError}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
{breakglassSuccessMessage ? (
|
|
||||||
<p className="text-sm text-emerald-600">{breakglassSuccessMessage}</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<Button type="submit" disabled={isResettingPassword}>
|
|
||||||
{isResettingPassword ? "Resetting..." : "Reset Password"}
|
|
||||||
</Button>
|
|
||||||
</form>
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
</>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<AlertDialog
|
|
||||||
open={showRemoveOidcDialog}
|
|
||||||
onOpenChange={(open) => {
|
|
||||||
if (!open && !isDeletingOidc) {
|
|
||||||
setShowRemoveOidcDialog(false);
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<AlertDialogContent>
|
|
||||||
<AlertDialogHeader>
|
|
||||||
<AlertDialogTitle>Remove OIDC Configuration</AlertDialogTitle>
|
|
||||||
<AlertDialogDescription>
|
|
||||||
This will remove issuer URL, client ID, and client secret from system configuration.
|
|
||||||
</AlertDialogDescription>
|
|
||||||
</AlertDialogHeader>
|
|
||||||
<AlertDialogFooter>
|
|
||||||
<AlertDialogCancel disabled={isDeletingOidc}>Cancel</AlertDialogCancel>
|
|
||||||
<AlertDialogAction onClick={handleRemoveOidc} disabled={isDeletingOidc}>
|
|
||||||
{isDeletingOidc ? "Removing..." : "Remove OIDC"}
|
|
||||||
</AlertDialogAction>
|
|
||||||
</AlertDialogFooter>
|
|
||||||
</AlertDialogContent>
|
|
||||||
</AlertDialog>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -200,82 +200,6 @@ const categories: CategoryConfig[] = [
|
|||||||
</svg>
|
</svg>
|
||||||
),
|
),
|
||||||
},
|
},
|
||||||
{
|
|
||||||
title: "LLM Providers",
|
|
||||||
description:
|
|
||||||
"Add and manage LLM providers, encrypted API keys, base URLs, and model inventories.",
|
|
||||||
href: "/settings/providers",
|
|
||||||
accent: "var(--ms-blue-400)",
|
|
||||||
iconBg: "rgba(47, 128, 255, 0.12)",
|
|
||||||
icon: (
|
|
||||||
<svg
|
|
||||||
width="20"
|
|
||||||
height="20"
|
|
||||||
viewBox="0 0 20 20"
|
|
||||||
fill="none"
|
|
||||||
stroke="currentColor"
|
|
||||||
strokeWidth="1.5"
|
|
||||||
strokeLinecap="round"
|
|
||||||
strokeLinejoin="round"
|
|
||||||
aria-hidden="true"
|
|
||||||
>
|
|
||||||
<rect x="2.5" y="4" width="15" height="12" rx="2" />
|
|
||||||
<path d="M2.5 8h15" />
|
|
||||||
<circle cx="6" cy="12" r="1" />
|
|
||||||
<circle cx="10" cy="12" r="1" />
|
|
||||||
<circle cx="14" cy="12" r="1" />
|
|
||||||
</svg>
|
|
||||||
),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
title: "Agent Config",
|
|
||||||
description: "Choose primary and fallback models, plus optional personality/SOUL instructions.",
|
|
||||||
href: "/settings/agent-config",
|
|
||||||
accent: "var(--ms-teal-400)",
|
|
||||||
iconBg: "rgba(20, 184, 166, 0.12)",
|
|
||||||
icon: (
|
|
||||||
<svg
|
|
||||||
width="20"
|
|
||||||
height="20"
|
|
||||||
viewBox="0 0 20 20"
|
|
||||||
fill="none"
|
|
||||||
stroke="currentColor"
|
|
||||||
strokeWidth="1.5"
|
|
||||||
strokeLinecap="round"
|
|
||||||
strokeLinejoin="round"
|
|
||||||
aria-hidden="true"
|
|
||||||
>
|
|
||||||
<path d="M4 5h12" />
|
|
||||||
<path d="M4 10h12" />
|
|
||||||
<path d="M4 15h7" />
|
|
||||||
<circle cx="14.5" cy="15" r="1.5" />
|
|
||||||
</svg>
|
|
||||||
),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
title: "Authentication",
|
|
||||||
description: "Manage OIDC provider settings and breakglass admin password recovery.",
|
|
||||||
href: "/settings/auth",
|
|
||||||
accent: "var(--ms-amber-400)",
|
|
||||||
iconBg: "rgba(245, 158, 11, 0.12)",
|
|
||||||
icon: (
|
|
||||||
<svg
|
|
||||||
width="20"
|
|
||||||
height="20"
|
|
||||||
viewBox="0 0 20 20"
|
|
||||||
fill="none"
|
|
||||||
stroke="currentColor"
|
|
||||||
strokeWidth="1.5"
|
|
||||||
strokeLinecap="round"
|
|
||||||
strokeLinejoin="round"
|
|
||||||
aria-hidden="true"
|
|
||||||
>
|
|
||||||
<rect x="5" y="8" width="10" height="8" rx="1.5" />
|
|
||||||
<path d="M7 8V6a3 3 0 0 1 6 0v2" />
|
|
||||||
<circle cx="10" cy="12" r="1" />
|
|
||||||
</svg>
|
|
||||||
),
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
title: "Users",
|
title: "Users",
|
||||||
description: "Invite, manage roles, and deactivate users across your workspaces.",
|
description: "Invite, manage roles, and deactivate users across your workspaces.",
|
||||||
@@ -306,7 +230,6 @@ const categories: CategoryConfig[] = [
|
|||||||
title: "Teams",
|
title: "Teams",
|
||||||
description: "Create and manage teams within your active workspace.",
|
description: "Create and manage teams within your active workspace.",
|
||||||
href: "/settings/teams",
|
href: "/settings/teams",
|
||||||
adminOnly: true,
|
|
||||||
accent: "var(--ms-blue-400)",
|
accent: "var(--ms-blue-400)",
|
||||||
iconBg: "rgba(47, 128, 255, 0.12)",
|
iconBg: "rgba(47, 128, 255, 0.12)",
|
||||||
icon: (
|
icon: (
|
||||||
|
|||||||
@@ -1,634 +0,0 @@
|
|||||||
"use client";
|
|
||||||
|
|
||||||
import {
|
|
||||||
useCallback,
|
|
||||||
useEffect,
|
|
||||||
useMemo,
|
|
||||||
useState,
|
|
||||||
type ChangeEvent,
|
|
||||||
type ReactElement,
|
|
||||||
type SyntheticEvent,
|
|
||||||
} from "react";
|
|
||||||
import { Settings, Trash2 } from "lucide-react";
|
|
||||||
import { FleetSettingsNav } from "@/components/settings/FleetSettingsNav";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
|
||||||
import {
|
|
||||||
AlertDialog,
|
|
||||||
AlertDialogAction,
|
|
||||||
AlertDialogCancel,
|
|
||||||
AlertDialogContent,
|
|
||||||
AlertDialogDescription,
|
|
||||||
AlertDialogFooter,
|
|
||||||
AlertDialogHeader,
|
|
||||||
AlertDialogTitle,
|
|
||||||
} from "@/components/ui/alert-dialog";
|
|
||||||
import {
|
|
||||||
Dialog,
|
|
||||||
DialogContent,
|
|
||||||
DialogDescription,
|
|
||||||
DialogFooter,
|
|
||||||
DialogHeader,
|
|
||||||
DialogTitle,
|
|
||||||
} from "@/components/ui/dialog";
|
|
||||||
import { Input } from "@/components/ui/input";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import {
|
|
||||||
Select,
|
|
||||||
SelectContent,
|
|
||||||
SelectItem,
|
|
||||||
SelectTrigger,
|
|
||||||
SelectValue,
|
|
||||||
} from "@/components/ui/select";
|
|
||||||
import { Switch } from "@/components/ui/switch";
|
|
||||||
import { Textarea } from "@/components/ui/textarea";
|
|
||||||
import {
|
|
||||||
createFleetProvider,
|
|
||||||
deleteFleetProvider,
|
|
||||||
fetchFleetProviders,
|
|
||||||
updateFleetProvider,
|
|
||||||
type CreateFleetProviderRequest,
|
|
||||||
type FleetProvider,
|
|
||||||
type FleetProviderModel,
|
|
||||||
type UpdateFleetProviderRequest,
|
|
||||||
} from "@/lib/api/fleet-settings";
|
|
||||||
|
|
||||||
interface ProviderTypeOption {
|
|
||||||
value: string;
|
|
||||||
label: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface ProviderFormState {
|
|
||||||
type: string;
|
|
||||||
displayName: string;
|
|
||||||
apiKey: string;
|
|
||||||
baseUrl: string;
|
|
||||||
modelsText: string;
|
|
||||||
isActive: boolean;
|
|
||||||
}
|
|
||||||
|
|
||||||
const PROVIDER_TYPE_OPTIONS: ProviderTypeOption[] = [
|
|
||||||
{ value: "openai", label: "OpenAI Compatible" },
|
|
||||||
{ value: "claude", label: "Claude / Anthropic" },
|
|
||||||
{ value: "ollama", label: "Ollama" },
|
|
||||||
{ value: "zai", label: "Z.ai" },
|
|
||||||
{ value: "custom", label: "Custom" },
|
|
||||||
];
|
|
||||||
|
|
||||||
const INITIAL_FORM: ProviderFormState = {
|
|
||||||
type: "openai",
|
|
||||||
displayName: "",
|
|
||||||
apiKey: "",
|
|
||||||
baseUrl: "",
|
|
||||||
modelsText: "",
|
|
||||||
isActive: true,
|
|
||||||
};
|
|
||||||
|
|
||||||
function buildProviderName(displayName: string, type: string): string {
|
|
||||||
const slug = displayName
|
|
||||||
.trim()
|
|
||||||
.toLowerCase()
|
|
||||||
.replace(/[^a-z0-9]+/g, "-")
|
|
||||||
.replace(/^-+/, "")
|
|
||||||
.replace(/-+$/, "");
|
|
||||||
|
|
||||||
const candidate = `${type}-${slug.length > 0 ? slug : "provider"}`;
|
|
||||||
return candidate.slice(0, 100);
|
|
||||||
}
|
|
||||||
|
|
||||||
function getErrorMessage(error: unknown, fallback: string): string {
|
|
||||||
if (error instanceof Error && error.message.trim().length > 0) {
|
|
||||||
return error.message;
|
|
||||||
}
|
|
||||||
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
function normalizeProviderModels(models: unknown): FleetProviderModel[] {
|
|
||||||
if (!Array.isArray(models)) {
|
|
||||||
return [];
|
|
||||||
}
|
|
||||||
|
|
||||||
const normalized: FleetProviderModel[] = [];
|
|
||||||
|
|
||||||
models.forEach((entry) => {
|
|
||||||
if (typeof entry === "string" && entry.trim().length > 0) {
|
|
||||||
normalized.push({ id: entry.trim(), name: entry.trim() });
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (entry && typeof entry === "object") {
|
|
||||||
const record = entry as Record<string, unknown>;
|
|
||||||
const id =
|
|
||||||
typeof record.id === "string"
|
|
||||||
? record.id.trim()
|
|
||||||
: typeof record.name === "string"
|
|
||||||
? record.name.trim()
|
|
||||||
: "";
|
|
||||||
|
|
||||||
if (id.length > 0) {
|
|
||||||
const name =
|
|
||||||
typeof record.name === "string" && record.name.trim().length > 0
|
|
||||||
? record.name.trim()
|
|
||||||
: id;
|
|
||||||
normalized.push({ id, name });
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
const seen = new Set<string>();
|
|
||||||
return normalized.filter((model) => {
|
|
||||||
if (seen.has(model.id)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
seen.add(model.id);
|
|
||||||
return true;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function modelsToEditorText(models: unknown): string {
|
|
||||||
return normalizeProviderModels(models)
|
|
||||||
.map((model) => model.id)
|
|
||||||
.join("\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
function parseModelsText(value: string): string[] {
|
|
||||||
const seen = new Set<string>();
|
|
||||||
|
|
||||||
return value
|
|
||||||
.split(/\r?\n/g)
|
|
||||||
.map((segment) => segment.trim())
|
|
||||||
.filter((segment) => segment.length > 0)
|
|
||||||
.filter((segment) => {
|
|
||||||
if (seen.has(segment)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
seen.add(segment);
|
|
||||||
return true;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function maskApiKey(value: string): string {
|
|
||||||
if (value.length === 0) {
|
|
||||||
return "Not set";
|
|
||||||
}
|
|
||||||
|
|
||||||
if (value.length <= 7) {
|
|
||||||
return "*".repeat(Math.max(4, value.length));
|
|
||||||
}
|
|
||||||
|
|
||||||
return `${value.slice(0, 3)}****...${value.slice(-4)}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
export default function ProvidersSettingsPage(): ReactElement {
|
|
||||||
const [providers, setProviders] = useState<FleetProvider[]>([]);
|
|
||||||
const [isLoading, setIsLoading] = useState<boolean>(true);
|
|
||||||
const [isRefreshing, setIsRefreshing] = useState<boolean>(false);
|
|
||||||
const [error, setError] = useState<string | null>(null);
|
|
||||||
const [successMessage, setSuccessMessage] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const [isDialogOpen, setIsDialogOpen] = useState<boolean>(false);
|
|
||||||
const [editingProvider, setEditingProvider] = useState<FleetProvider | null>(null);
|
|
||||||
const [form, setForm] = useState<ProviderFormState>(INITIAL_FORM);
|
|
||||||
const [formError, setFormError] = useState<string | null>(null);
|
|
||||||
const [isSaving, setIsSaving] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const [deleteTarget, setDeleteTarget] = useState<FleetProvider | null>(null);
|
|
||||||
const [isDeleting, setIsDeleting] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const loadProviders = useCallback(async (showLoadingState: boolean): Promise<void> => {
|
|
||||||
if (showLoadingState) {
|
|
||||||
setIsLoading(true);
|
|
||||||
} else {
|
|
||||||
setIsRefreshing(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const data = await fetchFleetProviders();
|
|
||||||
setProviders(data);
|
|
||||||
setError(null);
|
|
||||||
} catch (loadError: unknown) {
|
|
||||||
setError(getErrorMessage(loadError, "Failed to load providers."));
|
|
||||||
} finally {
|
|
||||||
setIsLoading(false);
|
|
||||||
setIsRefreshing(false);
|
|
||||||
}
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
void loadProviders(true);
|
|
||||||
}, [loadProviders]);
|
|
||||||
|
|
||||||
const apiKeyHint = useMemo(() => {
|
|
||||||
const enteredKey = form.apiKey.trim();
|
|
||||||
|
|
||||||
if (enteredKey.length > 0) {
|
|
||||||
return `Masked preview: ${maskApiKey(enteredKey)}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (editingProvider) {
|
|
||||||
return "Stored API key remains encrypted and hidden. Enter a new key only when rotating.";
|
|
||||||
}
|
|
||||||
|
|
||||||
return "API keys are never shown decrypted. Only masked previews are displayed while typing.";
|
|
||||||
}, [editingProvider, form.apiKey]);
|
|
||||||
|
|
||||||
function openCreateDialog(): void {
|
|
||||||
setEditingProvider(null);
|
|
||||||
setForm(INITIAL_FORM);
|
|
||||||
setFormError(null);
|
|
||||||
setIsDialogOpen(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
function openEditDialog(provider: FleetProvider): void {
|
|
||||||
setEditingProvider(provider);
|
|
||||||
setForm({
|
|
||||||
type: provider.type,
|
|
||||||
displayName: provider.displayName,
|
|
||||||
apiKey: "",
|
|
||||||
baseUrl: provider.baseUrl ?? "",
|
|
||||||
modelsText: modelsToEditorText(provider.models),
|
|
||||||
isActive: provider.isActive,
|
|
||||||
});
|
|
||||||
setFormError(null);
|
|
||||||
setIsDialogOpen(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
function closeDialog(): void {
|
|
||||||
if (isSaving) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsDialogOpen(false);
|
|
||||||
setEditingProvider(null);
|
|
||||||
setForm(INITIAL_FORM);
|
|
||||||
setFormError(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleSubmit(event: SyntheticEvent): Promise<void> {
|
|
||||||
event.preventDefault();
|
|
||||||
setFormError(null);
|
|
||||||
setSuccessMessage(null);
|
|
||||||
|
|
||||||
const displayName = form.displayName.trim();
|
|
||||||
if (displayName.length === 0) {
|
|
||||||
setFormError("Display name is required.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const models = parseModelsText(form.modelsText);
|
|
||||||
const providerModels = models.map((id) => ({ id, name: id }));
|
|
||||||
const baseUrl = form.baseUrl.trim();
|
|
||||||
const apiKey = form.apiKey.trim();
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsSaving(true);
|
|
||||||
|
|
||||||
if (editingProvider) {
|
|
||||||
const updatePayload: UpdateFleetProviderRequest = {
|
|
||||||
displayName,
|
|
||||||
isActive: form.isActive,
|
|
||||||
models: providerModels,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (baseUrl.length > 0) {
|
|
||||||
updatePayload.baseUrl = baseUrl;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (apiKey.length > 0) {
|
|
||||||
updatePayload.apiKey = apiKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
await updateFleetProvider(editingProvider.id, updatePayload);
|
|
||||||
setSuccessMessage(`Updated provider "${displayName}".`);
|
|
||||||
} else {
|
|
||||||
const createPayload: CreateFleetProviderRequest = {
|
|
||||||
name: buildProviderName(displayName, form.type),
|
|
||||||
displayName,
|
|
||||||
type: form.type,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (baseUrl.length > 0) {
|
|
||||||
createPayload.baseUrl = baseUrl;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (apiKey.length > 0) {
|
|
||||||
createPayload.apiKey = apiKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (providerModels.length > 0) {
|
|
||||||
createPayload.models = providerModels;
|
|
||||||
}
|
|
||||||
|
|
||||||
await createFleetProvider(createPayload);
|
|
||||||
setSuccessMessage(`Added provider "${displayName}".`);
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsDialogOpen(false);
|
|
||||||
setEditingProvider(null);
|
|
||||||
setForm(INITIAL_FORM);
|
|
||||||
await loadProviders(false);
|
|
||||||
} catch (saveError: unknown) {
|
|
||||||
setFormError(getErrorMessage(saveError, "Unable to save provider."));
|
|
||||||
} finally {
|
|
||||||
setIsSaving(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleDeleteProvider(): Promise<void> {
|
|
||||||
if (!deleteTarget) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsDeleting(true);
|
|
||||||
await deleteFleetProvider(deleteTarget.id);
|
|
||||||
setSuccessMessage(`Deleted provider "${deleteTarget.displayName}".`);
|
|
||||||
setDeleteTarget(null);
|
|
||||||
await loadProviders(false);
|
|
||||||
} catch (deleteError: unknown) {
|
|
||||||
setError(getErrorMessage(deleteError, "Failed to delete provider."));
|
|
||||||
} finally {
|
|
||||||
setIsDeleting(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div className="max-w-6xl mx-auto p-6 space-y-6">
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div>
|
|
||||||
<h1 className="text-3xl font-bold">LLM Providers</h1>
|
|
||||||
<p className="text-muted-foreground mt-1">
|
|
||||||
Manage provider endpoints, model inventories, and encrypted API credentials.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
<FleetSettingsNav />
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Card>
|
|
||||||
<CardHeader className="flex flex-col gap-3 sm:flex-row sm:items-center sm:justify-between">
|
|
||||||
<div>
|
|
||||||
<CardTitle>Provider Directory</CardTitle>
|
|
||||||
<CardDescription>
|
|
||||||
API keys are always encrypted in storage and never displayed in plaintext.
|
|
||||||
</CardDescription>
|
|
||||||
</div>
|
|
||||||
<div className="flex items-center gap-2">
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
onClick={() => {
|
|
||||||
void loadProviders(false);
|
|
||||||
}}
|
|
||||||
disabled={isLoading || isRefreshing}
|
|
||||||
>
|
|
||||||
{isRefreshing ? "Refreshing..." : "Refresh"}
|
|
||||||
</Button>
|
|
||||||
<Button onClick={openCreateDialog}>Add Provider</Button>
|
|
||||||
</div>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent className="space-y-3">
|
|
||||||
{error ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{error}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
{successMessage ? <p className="text-sm text-emerald-600">{successMessage}</p> : null}
|
|
||||||
|
|
||||||
{isLoading ? (
|
|
||||||
<p className="text-sm text-muted-foreground">Loading providers...</p>
|
|
||||||
) : providers.length === 0 ? (
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
No providers configured yet. Add one to make models available for agent assignment.
|
|
||||||
</p>
|
|
||||||
) : (
|
|
||||||
providers.map((provider) => {
|
|
||||||
const providerModels = normalizeProviderModels(provider.models);
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div
|
|
||||||
key={provider.id}
|
|
||||||
className="rounded-lg border p-4 flex flex-col gap-4 md:flex-row md:items-start md:justify-between"
|
|
||||||
>
|
|
||||||
<div className="space-y-2 min-w-0">
|
|
||||||
<div className="flex items-center gap-2 flex-wrap">
|
|
||||||
<p className="font-semibold truncate">{provider.displayName}</p>
|
|
||||||
<Badge variant={provider.isActive ? "default" : "secondary"}>
|
|
||||||
{provider.isActive ? "Active" : "Inactive"}
|
|
||||||
</Badge>
|
|
||||||
<Badge variant="outline">{provider.type}</Badge>
|
|
||||||
</div>
|
|
||||||
<p className="text-sm text-muted-foreground">Name: {provider.name}</p>
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
Base URL: {provider.baseUrl ?? "Provider default"}
|
|
||||||
</p>
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
API Key: encrypted and hidden (never returned decrypted)
|
|
||||||
</p>
|
|
||||||
<div className="flex flex-wrap gap-2">
|
|
||||||
{providerModels.length === 0 ? (
|
|
||||||
<Badge variant="secondary">No models configured</Badge>
|
|
||||||
) : (
|
|
||||||
providerModels.map((model) => (
|
|
||||||
<Badge key={`${provider.id}-${model.id}`} variant="outline">
|
|
||||||
{model.id}
|
|
||||||
</Badge>
|
|
||||||
))
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="flex items-center gap-2">
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
size="sm"
|
|
||||||
onClick={() => {
|
|
||||||
openEditDialog(provider);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Settings className="h-4 w-4 mr-2" />
|
|
||||||
Edit
|
|
||||||
</Button>
|
|
||||||
<Button
|
|
||||||
variant="destructive"
|
|
||||||
size="sm"
|
|
||||||
onClick={() => {
|
|
||||||
setDeleteTarget(provider);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Trash2 className="h-4 w-4 mr-2" />
|
|
||||||
Delete
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
})
|
|
||||||
)}
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
|
|
||||||
<Dialog
|
|
||||||
open={isDialogOpen}
|
|
||||||
onOpenChange={(nextOpen) => {
|
|
||||||
if (!nextOpen) {
|
|
||||||
closeDialog();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsDialogOpen(true);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<DialogContent>
|
|
||||||
<DialogHeader>
|
|
||||||
<DialogTitle>{editingProvider ? "Edit Provider" : "Add Provider"}</DialogTitle>
|
|
||||||
<DialogDescription>
|
|
||||||
Configure connection details and model IDs. API keys are masked in the UI.
|
|
||||||
</DialogDescription>
|
|
||||||
</DialogHeader>
|
|
||||||
|
|
||||||
<form onSubmit={(event) => void handleSubmit(event)} className="space-y-4">
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="provider-type">Type</Label>
|
|
||||||
<Select
|
|
||||||
value={form.type}
|
|
||||||
onValueChange={(value) => {
|
|
||||||
setForm((previous) => ({ ...previous, type: value }));
|
|
||||||
}}
|
|
||||||
disabled={Boolean(editingProvider)}
|
|
||||||
>
|
|
||||||
<SelectTrigger id="provider-type">
|
|
||||||
<SelectValue placeholder="Select provider type" />
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent>
|
|
||||||
{PROVIDER_TYPE_OPTIONS.map((option) => (
|
|
||||||
<SelectItem key={option.value} value={option.value}>
|
|
||||||
{option.label}
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="provider-display-name">Display Name</Label>
|
|
||||||
<Input
|
|
||||||
id="provider-display-name"
|
|
||||||
value={form.displayName}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setForm((previous) => ({ ...previous, displayName: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="OpenAI Primary"
|
|
||||||
maxLength={255}
|
|
||||||
disabled={isSaving}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="provider-api-key">API Key</Label>
|
|
||||||
<Input
|
|
||||||
id="provider-api-key"
|
|
||||||
type="password"
|
|
||||||
value={form.apiKey}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setForm((previous) => ({ ...previous, apiKey: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder={editingProvider ? "Enter new key to rotate" : "sk-..."}
|
|
||||||
autoComplete="new-password"
|
|
||||||
disabled={isSaving}
|
|
||||||
/>
|
|
||||||
<p className="text-xs text-muted-foreground">{apiKeyHint}</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="provider-base-url">Base URL</Label>
|
|
||||||
<Input
|
|
||||||
id="provider-base-url"
|
|
||||||
value={form.baseUrl}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setForm((previous) => ({ ...previous, baseUrl: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="https://api.provider.com/v1"
|
|
||||||
disabled={isSaving}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="provider-models">Models</Label>
|
|
||||||
<Textarea
|
|
||||||
id="provider-models"
|
|
||||||
value={form.modelsText}
|
|
||||||
onChange={(event: ChangeEvent<HTMLTextAreaElement>) => {
|
|
||||||
setForm((previous) => ({ ...previous, modelsText: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder={"One model ID per line\nExample: gpt-4.1-mini"}
|
|
||||||
rows={5}
|
|
||||||
disabled={isSaving}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{editingProvider ? (
|
|
||||||
<div className="flex items-center justify-between rounded-md border px-3 py-2">
|
|
||||||
<div>
|
|
||||||
<Label htmlFor="provider-active">Provider Status</Label>
|
|
||||||
<p className="text-xs text-muted-foreground">
|
|
||||||
Disable to keep configuration without using this provider.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
<Switch
|
|
||||||
id="provider-active"
|
|
||||||
checked={form.isActive}
|
|
||||||
onCheckedChange={(checked) => {
|
|
||||||
setForm((previous) => ({ ...previous, isActive: checked }));
|
|
||||||
}}
|
|
||||||
disabled={isSaving}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
{formError ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{formError}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<DialogFooter>
|
|
||||||
<Button type="button" variant="outline" onClick={closeDialog} disabled={isSaving}>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
<Button type="submit" disabled={isSaving}>
|
|
||||||
{isSaving ? "Saving..." : editingProvider ? "Save Changes" : "Create Provider"}
|
|
||||||
</Button>
|
|
||||||
</DialogFooter>
|
|
||||||
</form>
|
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
|
||||||
|
|
||||||
<AlertDialog
|
|
||||||
open={deleteTarget !== null}
|
|
||||||
onOpenChange={(open) => {
|
|
||||||
if (!open && !isDeleting) {
|
|
||||||
setDeleteTarget(null);
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<AlertDialogContent>
|
|
||||||
<AlertDialogHeader>
|
|
||||||
<AlertDialogTitle>Delete Provider</AlertDialogTitle>
|
|
||||||
<AlertDialogDescription>
|
|
||||||
Delete provider "{deleteTarget?.displayName}"? This removes its configuration and
|
|
||||||
model mappings.
|
|
||||||
</AlertDialogDescription>
|
|
||||||
</AlertDialogHeader>
|
|
||||||
<AlertDialogFooter>
|
|
||||||
<AlertDialogCancel disabled={isDeleting}>Cancel</AlertDialogCancel>
|
|
||||||
<AlertDialogAction onClick={handleDeleteProvider} disabled={isDeleting}>
|
|
||||||
{isDeleting ? "Deleting..." : "Delete Provider"}
|
|
||||||
</AlertDialogAction>
|
|
||||||
</AlertDialogFooter>
|
|
||||||
</AlertDialogContent>
|
|
||||||
</AlertDialog>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -1,12 +1,9 @@
|
|||||||
import type { ReactElement, ReactNode } from "react";
|
import type { ReactElement, ReactNode } from "react";
|
||||||
import type { TeamRecord } from "@/lib/api/teams";
|
import type { TeamRecord } from "@/lib/api/teams";
|
||||||
|
|
||||||
import { WorkspaceMemberRole } from "@mosaic/shared";
|
import { render, screen } from "@testing-library/react";
|
||||||
import { render, screen, waitFor } from "@testing-library/react";
|
|
||||||
import userEvent from "@testing-library/user-event";
|
|
||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||||
import { createTeam, deleteTeam, fetchTeams, updateTeam } from "@/lib/api/teams";
|
import { fetchTeams } from "@/lib/api/teams";
|
||||||
import { fetchUserWorkspaces } from "@/lib/api/workspaces";
|
|
||||||
|
|
||||||
import TeamsSettingsPage from "./page";
|
import TeamsSettingsPage from "./page";
|
||||||
|
|
||||||
@@ -25,19 +22,9 @@ vi.mock("next/link", () => ({
|
|||||||
vi.mock("@/lib/api/teams", () => ({
|
vi.mock("@/lib/api/teams", () => ({
|
||||||
fetchTeams: vi.fn(),
|
fetchTeams: vi.fn(),
|
||||||
createTeam: vi.fn(),
|
createTeam: vi.fn(),
|
||||||
updateTeam: vi.fn(),
|
|
||||||
deleteTeam: vi.fn(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock("@/lib/api/workspaces", () => ({
|
|
||||||
fetchUserWorkspaces: vi.fn(),
|
|
||||||
}));
|
}));
|
||||||
|
|
||||||
const fetchTeamsMock = vi.mocked(fetchTeams);
|
const fetchTeamsMock = vi.mocked(fetchTeams);
|
||||||
const createTeamMock = vi.mocked(createTeam);
|
|
||||||
const updateTeamMock = vi.mocked(updateTeam);
|
|
||||||
const deleteTeamMock = vi.mocked(deleteTeam);
|
|
||||||
const fetchUserWorkspacesMock = vi.mocked(fetchUserWorkspaces);
|
|
||||||
|
|
||||||
const baseTeam: TeamRecord = {
|
const baseTeam: TeamRecord = {
|
||||||
id: "team-1",
|
id: "team-1",
|
||||||
@@ -55,33 +42,6 @@ const baseTeam: TeamRecord = {
|
|||||||
describe("TeamsSettingsPage", () => {
|
describe("TeamsSettingsPage", () => {
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
vi.clearAllMocks();
|
vi.clearAllMocks();
|
||||||
fetchTeamsMock.mockResolvedValue([]);
|
|
||||||
fetchUserWorkspacesMock.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "workspace-1",
|
|
||||||
name: "Personal Workspace",
|
|
||||||
ownerId: "owner-1",
|
|
||||||
role: WorkspaceMemberRole.OWNER,
|
|
||||||
createdAt: "2026-01-01T00:00:00.000Z",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("shows access denied to non-admin users", async () => {
|
|
||||||
fetchUserWorkspacesMock.mockResolvedValueOnce([
|
|
||||||
{
|
|
||||||
id: "workspace-1",
|
|
||||||
name: "Personal Workspace",
|
|
||||||
ownerId: "owner-1",
|
|
||||||
role: WorkspaceMemberRole.MEMBER,
|
|
||||||
createdAt: "2026-01-01T00:00:00.000Z",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
render(<TeamsSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("Access Denied")).toBeInTheDocument();
|
|
||||||
expect(fetchTeamsMock).not.toHaveBeenCalled();
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it("loads and renders teams from the API", async () => {
|
it("loads and renders teams from the API", async () => {
|
||||||
@@ -89,7 +49,9 @@ describe("TeamsSettingsPage", () => {
|
|||||||
|
|
||||||
render(<TeamsSettingsPage />);
|
render(<TeamsSettingsPage />);
|
||||||
|
|
||||||
expect(await screen.findByText("Team Directory")).toBeInTheDocument();
|
expect(screen.getByText("Loading teams...")).toBeInTheDocument();
|
||||||
|
|
||||||
|
expect(await screen.findByText("Your Teams (1)")).toBeInTheDocument();
|
||||||
expect(screen.getByText("Platform Team")).toBeInTheDocument();
|
expect(screen.getByText("Platform Team")).toBeInTheDocument();
|
||||||
expect(fetchTeamsMock).toHaveBeenCalledTimes(1);
|
expect(fetchTeamsMock).toHaveBeenCalledTimes(1);
|
||||||
});
|
});
|
||||||
@@ -99,8 +61,8 @@ describe("TeamsSettingsPage", () => {
|
|||||||
|
|
||||||
render(<TeamsSettingsPage />);
|
render(<TeamsSettingsPage />);
|
||||||
|
|
||||||
expect(await screen.findByText("No Teams Yet")).toBeInTheDocument();
|
expect(await screen.findByText("Your Teams (0)")).toBeInTheDocument();
|
||||||
expect(screen.getByText("Create the first team to get started.")).toBeInTheDocument();
|
expect(screen.getByText("No teams yet")).toBeInTheDocument();
|
||||||
});
|
});
|
||||||
|
|
||||||
it("shows error state and does not show empty state", async () => {
|
it("shows error state and does not show empty state", async () => {
|
||||||
@@ -109,82 +71,6 @@ describe("TeamsSettingsPage", () => {
|
|||||||
render(<TeamsSettingsPage />);
|
render(<TeamsSettingsPage />);
|
||||||
|
|
||||||
expect(await screen.findByText("Unable to load teams")).toBeInTheDocument();
|
expect(await screen.findByText("Unable to load teams")).toBeInTheDocument();
|
||||||
});
|
expect(screen.queryByText("No teams yet")).not.toBeInTheDocument();
|
||||||
|
|
||||||
it("creates a team from the create dialog", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
fetchTeamsMock.mockResolvedValue([baseTeam]);
|
|
||||||
createTeamMock.mockResolvedValue({
|
|
||||||
...baseTeam,
|
|
||||||
id: "team-2",
|
|
||||||
name: "Design Team",
|
|
||||||
description: "Owns design quality",
|
|
||||||
});
|
|
||||||
|
|
||||||
render(<TeamsSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("Platform Team")).toBeInTheDocument();
|
|
||||||
|
|
||||||
const triggerButton = screen.getByRole("button", { name: "Create Team" });
|
|
||||||
await user.click(triggerButton);
|
|
||||||
|
|
||||||
await user.type(screen.getByLabelText("Name"), "Design Team");
|
|
||||||
await user.type(screen.getByLabelText("Description"), "Owns design quality");
|
|
||||||
|
|
||||||
const submitButton = screen.getAllByRole("button", { name: "Create Team" })[1];
|
|
||||||
if (!submitButton) {
|
|
||||||
throw new Error("Expected create-team submit button to be rendered");
|
|
||||||
}
|
|
||||||
await user.click(submitButton);
|
|
||||||
|
|
||||||
await waitFor(() => {
|
|
||||||
expect(createTeamMock).toHaveBeenCalledWith({
|
|
||||||
name: "Design Team",
|
|
||||||
description: "Owns design quality",
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("opens team details and updates name", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
fetchTeamsMock.mockResolvedValue([baseTeam]);
|
|
||||||
updateTeamMock.mockResolvedValue({
|
|
||||||
...baseTeam,
|
|
||||||
name: "Platform Engineering",
|
|
||||||
});
|
|
||||||
|
|
||||||
render(<TeamsSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("Platform Team")).toBeInTheDocument();
|
|
||||||
|
|
||||||
await user.click(screen.getByText("Platform Team"));
|
|
||||||
|
|
||||||
const nameInput = await screen.findByLabelText("Name");
|
|
||||||
await user.clear(nameInput);
|
|
||||||
await user.type(nameInput, "Platform Engineering");
|
|
||||||
await user.click(screen.getByRole("button", { name: "Save Changes" }));
|
|
||||||
|
|
||||||
await waitFor(() => {
|
|
||||||
expect(updateTeamMock).toHaveBeenCalledWith("team-1", {
|
|
||||||
name: "Platform Engineering",
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("deletes a team from the confirmation dialog", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
fetchTeamsMock.mockResolvedValue([baseTeam]);
|
|
||||||
deleteTeamMock.mockResolvedValue();
|
|
||||||
|
|
||||||
render(<TeamsSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("Platform Team")).toBeInTheDocument();
|
|
||||||
|
|
||||||
await user.click(screen.getByRole("button", { name: "Delete" }));
|
|
||||||
await user.click(screen.getByRole("button", { name: "Delete Team" }));
|
|
||||||
|
|
||||||
await waitFor(() => {
|
|
||||||
expect(deleteTeamMock).toHaveBeenCalledWith("team-1");
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -1,582 +1,244 @@
|
|||||||
"use client";
|
"use client";
|
||||||
|
|
||||||
import {
|
import type { ReactElement, SyntheticEvent } from "react";
|
||||||
useCallback,
|
|
||||||
useEffect,
|
import { useCallback, useEffect, useState } from "react";
|
||||||
useState,
|
|
||||||
type ChangeEvent,
|
|
||||||
type KeyboardEvent,
|
|
||||||
type ReactElement,
|
|
||||||
type SyntheticEvent,
|
|
||||||
} from "react";
|
|
||||||
import Link from "next/link";
|
import Link from "next/link";
|
||||||
import { Plus, Trash2, Users } from "lucide-react";
|
import { createTeam, fetchTeams, type CreateTeamDto, type TeamRecord } from "@/lib/api/teams";
|
||||||
import { WorkspaceMemberRole } from "@mosaic/shared";
|
|
||||||
import { SettingsAccessDenied } from "@/components/settings/SettingsAccessDenied";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
|
||||||
import {
|
|
||||||
Dialog,
|
|
||||||
DialogContent,
|
|
||||||
DialogDescription,
|
|
||||||
DialogFooter,
|
|
||||||
DialogHeader,
|
|
||||||
DialogTitle,
|
|
||||||
DialogTrigger,
|
|
||||||
} from "@/components/ui/dialog";
|
|
||||||
import { Input } from "@/components/ui/input";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import { Textarea } from "@/components/ui/textarea";
|
|
||||||
import {
|
|
||||||
AlertDialog,
|
|
||||||
AlertDialogAction,
|
|
||||||
AlertDialogCancel,
|
|
||||||
AlertDialogContent,
|
|
||||||
AlertDialogDescription,
|
|
||||||
AlertDialogFooter,
|
|
||||||
AlertDialogHeader,
|
|
||||||
AlertDialogTitle,
|
|
||||||
} from "@/components/ui/alert-dialog";
|
|
||||||
import {
|
|
||||||
createTeam,
|
|
||||||
deleteTeam,
|
|
||||||
fetchTeams,
|
|
||||||
updateTeam,
|
|
||||||
type CreateTeamDto,
|
|
||||||
type TeamRecord,
|
|
||||||
type UpdateTeamDto,
|
|
||||||
} from "@/lib/api/teams";
|
|
||||||
import { fetchUserWorkspaces } from "@/lib/api/workspaces";
|
|
||||||
|
|
||||||
const INITIAL_CREATE_FORM = {
|
function getErrorMessage(error: unknown, fallback: string): string {
|
||||||
name: "",
|
if (error instanceof Error) {
|
||||||
description: "",
|
return error.message;
|
||||||
};
|
}
|
||||||
|
|
||||||
const INITIAL_DETAIL_FORM = {
|
return fallback;
|
||||||
name: "",
|
|
||||||
description: "",
|
|
||||||
};
|
|
||||||
|
|
||||||
interface DetailInitialState {
|
|
||||||
name: string;
|
|
||||||
description: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
function toMemberLabel(count: number): string {
|
|
||||||
return `${String(count)} member${count === 1 ? "" : "s"}`;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export default function TeamsSettingsPage(): ReactElement {
|
export default function TeamsSettingsPage(): ReactElement {
|
||||||
const [teams, setTeams] = useState<TeamRecord[]>([]);
|
const [teams, setTeams] = useState<TeamRecord[]>([]);
|
||||||
const [isLoading, setIsLoading] = useState<boolean>(true);
|
const [isLoading, setIsLoading] = useState(true);
|
||||||
const [isRefreshing, setIsRefreshing] = useState<boolean>(false);
|
const [loadError, setLoadError] = useState<string | null>(null);
|
||||||
const [error, setError] = useState<string | null>(null);
|
const [isCreateDialogOpen, setIsCreateDialogOpen] = useState(false);
|
||||||
const [isAdmin, setIsAdmin] = useState<boolean | null>(null);
|
const [isCreating, setIsCreating] = useState(false);
|
||||||
|
const [newTeamName, setNewTeamName] = useState("");
|
||||||
const [isCreateOpen, setIsCreateOpen] = useState<boolean>(false);
|
const [newTeamDescription, setNewTeamDescription] = useState("");
|
||||||
const [createForm, setCreateForm] = useState(INITIAL_CREATE_FORM);
|
|
||||||
const [createError, setCreateError] = useState<string | null>(null);
|
const [createError, setCreateError] = useState<string | null>(null);
|
||||||
const [isCreating, setIsCreating] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const [detailTarget, setDetailTarget] = useState<TeamRecord | null>(null);
|
const loadTeams = useCallback(async (): Promise<void> => {
|
||||||
const [detailForm, setDetailForm] = useState(INITIAL_DETAIL_FORM);
|
setIsLoading(true);
|
||||||
const [detailInitial, setDetailInitial] = useState<DetailInitialState | null>(null);
|
|
||||||
const [detailError, setDetailError] = useState<string | null>(null);
|
|
||||||
const [isSavingDetails, setIsSavingDetails] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const [deleteTarget, setDeleteTarget] = useState<TeamRecord | null>(null);
|
|
||||||
const [isDeleting, setIsDeleting] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const loadTeams = useCallback(async (showLoadingState: boolean): Promise<void> => {
|
|
||||||
try {
|
try {
|
||||||
if (showLoadingState) {
|
|
||||||
setIsLoading(true);
|
|
||||||
} else {
|
|
||||||
setIsRefreshing(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
const data = await fetchTeams();
|
const data = await fetchTeams();
|
||||||
setTeams(data);
|
setTeams(data);
|
||||||
setError(null);
|
setLoadError(null);
|
||||||
} catch (err: unknown) {
|
} catch (error) {
|
||||||
setError(err instanceof Error ? err.message : "Failed to load teams");
|
setLoadError(getErrorMessage(error, "Failed to load teams"));
|
||||||
} finally {
|
} finally {
|
||||||
setIsLoading(false);
|
setIsLoading(false);
|
||||||
setIsRefreshing(false);
|
|
||||||
}
|
}
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
fetchUserWorkspaces()
|
void loadTeams();
|
||||||
.then((workspaces) => {
|
}, [loadTeams]);
|
||||||
const adminRoles: WorkspaceMemberRole[] = [
|
|
||||||
WorkspaceMemberRole.OWNER,
|
|
||||||
WorkspaceMemberRole.ADMIN,
|
|
||||||
];
|
|
||||||
|
|
||||||
setIsAdmin(workspaces.some((workspace) => adminRoles.includes(workspace.role)));
|
const handleCreateTeam = async (e: SyntheticEvent<HTMLFormElement>): Promise<void> => {
|
||||||
})
|
e.preventDefault();
|
||||||
.catch(() => {
|
|
||||||
setIsAdmin(true); // fail open
|
|
||||||
});
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
useEffect(() => {
|
const teamName = newTeamName.trim();
|
||||||
if (isAdmin !== true) {
|
if (!teamName) return;
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
void loadTeams(true);
|
setIsCreating(true);
|
||||||
}, [isAdmin, loadTeams]);
|
|
||||||
|
|
||||||
function resetCreateForm(): void {
|
|
||||||
setCreateForm(INITIAL_CREATE_FORM);
|
|
||||||
setCreateError(null);
|
setCreateError(null);
|
||||||
}
|
|
||||||
|
|
||||||
function openTeamDetails(team: TeamRecord): void {
|
|
||||||
const nextDetailForm = {
|
|
||||||
name: team.name,
|
|
||||||
description: team.description ?? "",
|
|
||||||
};
|
|
||||||
|
|
||||||
setDetailTarget(team);
|
|
||||||
setDetailForm(nextDetailForm);
|
|
||||||
setDetailInitial({
|
|
||||||
name: nextDetailForm.name,
|
|
||||||
description: nextDetailForm.description,
|
|
||||||
});
|
|
||||||
setDetailError(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
function resetTeamDetails(): void {
|
|
||||||
setDetailTarget(null);
|
|
||||||
setDetailForm(INITIAL_DETAIL_FORM);
|
|
||||||
setDetailInitial(null);
|
|
||||||
setDetailError(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
function handleTeamRowKeyDown(event: KeyboardEvent<HTMLDivElement>, team: TeamRecord): void {
|
|
||||||
if (event.key === "Enter" || event.key === " ") {
|
|
||||||
event.preventDefault();
|
|
||||||
openTeamDetails(team);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleCreateSubmit(event: SyntheticEvent): Promise<void> {
|
|
||||||
event.preventDefault();
|
|
||||||
setCreateError(null);
|
|
||||||
|
|
||||||
const name = createForm.name.trim();
|
|
||||||
if (!name) {
|
|
||||||
setCreateError("Name is required.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const description = createForm.description.trim();
|
|
||||||
const dto: CreateTeamDto = { name };
|
|
||||||
if (description) {
|
|
||||||
dto.description = description;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
setIsCreating(true);
|
const description = newTeamDescription.trim();
|
||||||
|
const dto: CreateTeamDto = { name: teamName };
|
||||||
|
if (description.length > 0) {
|
||||||
|
dto.description = description;
|
||||||
|
}
|
||||||
|
|
||||||
await createTeam(dto);
|
await createTeam(dto);
|
||||||
setIsCreateOpen(false);
|
setNewTeamName("");
|
||||||
resetCreateForm();
|
setNewTeamDescription("");
|
||||||
await loadTeams(false);
|
setIsCreateDialogOpen(false);
|
||||||
} catch (err: unknown) {
|
await loadTeams();
|
||||||
setCreateError(err instanceof Error ? err.message : "Failed to create team");
|
} catch (error) {
|
||||||
|
setCreateError(getErrorMessage(error, "Failed to create team"));
|
||||||
} finally {
|
} finally {
|
||||||
setIsCreating(false);
|
setIsCreating(false);
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
async function handleDetailSubmit(event: SyntheticEvent): Promise<void> {
|
|
||||||
event.preventDefault();
|
|
||||||
|
|
||||||
if (detailTarget === null || detailInitial === null) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const name = detailForm.name.trim();
|
|
||||||
if (!name) {
|
|
||||||
setDetailError("Name is required.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const nextDescription = detailForm.description.trim();
|
|
||||||
const normalizedNextDescription = nextDescription.length > 0 ? nextDescription : null;
|
|
||||||
const normalizedInitialDescription =
|
|
||||||
detailInitial.description.trim().length > 0 ? detailInitial.description.trim() : null;
|
|
||||||
|
|
||||||
const dto: UpdateTeamDto = {};
|
|
||||||
|
|
||||||
if (name !== detailInitial.name) {
|
|
||||||
dto.name = name;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (normalizedNextDescription !== normalizedInitialDescription) {
|
|
||||||
dto.description = normalizedNextDescription;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (Object.keys(dto).length === 0) {
|
|
||||||
resetTeamDetails();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsSavingDetails(true);
|
|
||||||
setDetailError(null);
|
|
||||||
await updateTeam(detailTarget.id, dto);
|
|
||||||
resetTeamDetails();
|
|
||||||
await loadTeams(false);
|
|
||||||
} catch (err: unknown) {
|
|
||||||
setDetailError(err instanceof Error ? err.message : "Failed to update team");
|
|
||||||
} finally {
|
|
||||||
setIsSavingDetails(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function confirmDelete(): Promise<void> {
|
|
||||||
if (!deleteTarget) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsDeleting(true);
|
|
||||||
await deleteTeam(deleteTarget.id);
|
|
||||||
setDeleteTarget(null);
|
|
||||||
await loadTeams(false);
|
|
||||||
setError(null);
|
|
||||||
} catch (err: unknown) {
|
|
||||||
setError(err instanceof Error ? err.message : "Failed to delete team");
|
|
||||||
} finally {
|
|
||||||
setIsDeleting(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (isAdmin === null) {
|
|
||||||
return (
|
|
||||||
<Card className="max-w-2xl mx-auto mt-8">
|
|
||||||
<CardContent className="py-12 text-center text-muted-foreground">
|
|
||||||
Checking permissions...
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!isAdmin) {
|
|
||||||
return <SettingsAccessDenied message="You need Admin or Owner role to manage teams." />;
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="max-w-6xl mx-auto p-6 space-y-6">
|
<main className="container mx-auto px-4 py-8 max-w-5xl">
|
||||||
<div className="flex items-start justify-between gap-4">
|
<div className="mb-8">
|
||||||
<div>
|
<div className="flex items-center justify-between mb-2">
|
||||||
<div className="flex items-center gap-3">
|
<h1 className="text-3xl font-bold text-gray-900">Teams</h1>
|
||||||
<h1 className="text-3xl font-bold">Teams</h1>
|
<Link href="/settings" className="text-sm text-blue-600 hover:text-blue-700">
|
||||||
<Badge variant="outline">{teams.length} total</Badge>
|
{"<-"} Back to Settings
|
||||||
</div>
|
</Link>
|
||||||
<p className="text-muted-foreground mt-1">Create and manage workspace teams</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="flex items-center gap-2">
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
onClick={() => {
|
|
||||||
void loadTeams(false);
|
|
||||||
}}
|
|
||||||
disabled={isLoading || isRefreshing}
|
|
||||||
>
|
|
||||||
{isRefreshing ? "Refreshing..." : "Refresh"}
|
|
||||||
</Button>
|
|
||||||
|
|
||||||
<Dialog
|
|
||||||
open={isCreateOpen}
|
|
||||||
onOpenChange={(open) => {
|
|
||||||
if (!open && !isCreating) {
|
|
||||||
resetCreateForm();
|
|
||||||
}
|
|
||||||
setIsCreateOpen(open);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<DialogTrigger asChild>
|
|
||||||
<Button>
|
|
||||||
<Plus className="h-4 w-4 mr-2" />
|
|
||||||
Create Team
|
|
||||||
</Button>
|
|
||||||
</DialogTrigger>
|
|
||||||
<DialogContent>
|
|
||||||
<DialogHeader>
|
|
||||||
<DialogTitle>Create Team</DialogTitle>
|
|
||||||
<DialogDescription>
|
|
||||||
Create a team in the active workspace to organize members and permissions.
|
|
||||||
</DialogDescription>
|
|
||||||
</DialogHeader>
|
|
||||||
|
|
||||||
<form
|
|
||||||
onSubmit={(event) => {
|
|
||||||
void handleCreateSubmit(event);
|
|
||||||
}}
|
|
||||||
className="space-y-4"
|
|
||||||
>
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="create-name">Name</Label>
|
|
||||||
<Input
|
|
||||||
id="create-name"
|
|
||||||
value={createForm.name}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setCreateForm((prev) => ({ ...prev, name: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="Platform Team"
|
|
||||||
maxLength={100}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="create-description">Description</Label>
|
|
||||||
<Textarea
|
|
||||||
id="create-description"
|
|
||||||
value={createForm.description}
|
|
||||||
onChange={(event: ChangeEvent<HTMLTextAreaElement>) => {
|
|
||||||
setCreateForm((prev) => ({ ...prev, description: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="Owns platform services and infrastructure"
|
|
||||||
maxLength={500}
|
|
||||||
rows={4}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{createError ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{createError}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<DialogFooter>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="outline"
|
|
||||||
onClick={() => {
|
|
||||||
if (!isCreating) {
|
|
||||||
setIsCreateOpen(false);
|
|
||||||
resetCreateForm();
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
disabled={isCreating}
|
|
||||||
>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
<Button type="submit" disabled={isCreating}>
|
|
||||||
{isCreating ? "Creating..." : "Create Team"}
|
|
||||||
</Button>
|
|
||||||
</DialogFooter>
|
|
||||||
</form>
|
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
|
||||||
</div>
|
</div>
|
||||||
|
<p className="text-gray-600">Manage teams in your active workspace</p>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div>
|
<div className="bg-white rounded-lg shadow-sm border border-gray-200 p-6 mb-6">
|
||||||
<Link href="/settings" className="text-sm text-blue-600 hover:text-blue-700">
|
<div className="flex items-center justify-between gap-4">
|
||||||
← Back to Settings
|
<div>
|
||||||
</Link>
|
<h2 className="text-lg font-semibold text-gray-900">Create New Team</h2>
|
||||||
</div>
|
<p className="text-sm text-gray-600 mt-1">
|
||||||
|
Add a team to organize members and permissions.
|
||||||
{error ? (
|
|
||||||
<Card>
|
|
||||||
<CardContent className="py-4">
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{error}
|
|
||||||
</p>
|
</p>
|
||||||
</CardContent>
|
</div>
|
||||||
</Card>
|
<button
|
||||||
) : null}
|
type="button"
|
||||||
|
onClick={() => {
|
||||||
|
setCreateError(null);
|
||||||
|
setIsCreateDialogOpen(true);
|
||||||
|
}}
|
||||||
|
className="px-6 py-2 bg-blue-600 text-white rounded-lg hover:bg-blue-700 font-medium"
|
||||||
|
>
|
||||||
|
Create Team
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
{isLoading ? (
|
{isCreateDialogOpen && (
|
||||||
<Card>
|
<div
|
||||||
<CardContent className="py-12 text-center text-muted-foreground">
|
className="fixed inset-0 z-50 flex items-center justify-center bg-black/40 px-4"
|
||||||
Loading teams...
|
role="dialog"
|
||||||
</CardContent>
|
>
|
||||||
</Card>
|
<div className="w-full max-w-lg rounded-lg border border-gray-200 bg-white p-6 shadow-xl">
|
||||||
) : teams.length === 0 ? (
|
<h3 className="text-lg font-semibold text-gray-900">Create New Team</h3>
|
||||||
<Card>
|
<p className="mt-1 text-sm text-gray-600">
|
||||||
<CardHeader>
|
Enter a team name and optional description.
|
||||||
<CardTitle>No Teams Yet</CardTitle>
|
</p>
|
||||||
<CardDescription>Create the first team to get started.</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
</Card>
|
|
||||||
) : (
|
|
||||||
<Card>
|
|
||||||
<CardHeader>
|
|
||||||
<CardTitle>Team Directory</CardTitle>
|
|
||||||
<CardDescription>
|
|
||||||
Click a team to view details or edit name and description.
|
|
||||||
</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent className="space-y-3">
|
|
||||||
{teams.map((team) => {
|
|
||||||
const memberCount = team._count?.members ?? 0;
|
|
||||||
const description = team.description?.trim();
|
|
||||||
|
|
||||||
return (
|
<form onSubmit={handleCreateTeam} className="mt-4 space-y-4">
|
||||||
<div
|
<div>
|
||||||
key={team.id}
|
<label htmlFor="team-name" className="mb-1 block text-sm font-medium text-gray-700">
|
||||||
className="rounded-md border p-4 flex flex-col gap-3 md:flex-row md:items-center md:justify-between cursor-pointer hover:bg-muted/30"
|
Team Name
|
||||||
role="button"
|
</label>
|
||||||
tabIndex={0}
|
<input
|
||||||
onClick={() => {
|
id="team-name"
|
||||||
openTeamDetails(team);
|
type="text"
|
||||||
}}
|
value={newTeamName}
|
||||||
onKeyDown={(event) => {
|
onChange={(e) => {
|
||||||
handleTeamRowKeyDown(event, team);
|
setNewTeamName(e.target.value);
|
||||||
}}
|
}}
|
||||||
|
placeholder="Enter team name..."
|
||||||
|
disabled={isCreating}
|
||||||
|
className="w-full rounded-lg border border-gray-300 px-4 py-2 focus:border-transparent focus:ring-2 focus:ring-blue-500 disabled:bg-gray-100"
|
||||||
|
autoFocus
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div>
|
||||||
|
<label
|
||||||
|
htmlFor="team-description"
|
||||||
|
className="mb-1 block text-sm font-medium text-gray-700"
|
||||||
>
|
>
|
||||||
<div className="space-y-1 min-w-0">
|
Description (optional)
|
||||||
<p className="font-semibold truncate">{team.name}</p>
|
</label>
|
||||||
<p className="text-sm text-muted-foreground truncate">
|
<textarea
|
||||||
{description && description.length > 0 ? description : "No description"}
|
id="team-description"
|
||||||
</p>
|
value={newTeamDescription}
|
||||||
</div>
|
onChange={(e) => {
|
||||||
|
setNewTeamDescription(e.target.value);
|
||||||
|
}}
|
||||||
|
placeholder="Describe this team's purpose..."
|
||||||
|
disabled={isCreating}
|
||||||
|
rows={3}
|
||||||
|
className="w-full rounded-lg border border-gray-300 px-4 py-2 focus:border-transparent focus:ring-2 focus:ring-blue-500 disabled:bg-gray-100"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
<div className="flex items-center gap-2 flex-wrap md:justify-end">
|
{createError !== null && (
|
||||||
<Badge variant="outline">
|
<div className="rounded-md border border-red-200 bg-red-50 px-3 py-2 text-sm text-red-700">
|
||||||
<Users className="h-3.5 w-3.5 mr-1" />
|
{createError}
|
||||||
{toMemberLabel(memberCount)}
|
|
||||||
</Badge>
|
|
||||||
<Badge variant="secondary">
|
|
||||||
Created {new Date(team.createdAt).toLocaleDateString()}
|
|
||||||
</Badge>
|
|
||||||
<Button
|
|
||||||
variant="destructive"
|
|
||||||
size="sm"
|
|
||||||
onClick={(event) => {
|
|
||||||
event.stopPropagation();
|
|
||||||
setDeleteTarget(team);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Trash2 className="h-4 w-4 mr-2" />
|
|
||||||
Delete
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</div>
|
</div>
|
||||||
);
|
)}
|
||||||
})}
|
|
||||||
</CardContent>
|
<div className="flex justify-end gap-3">
|
||||||
</Card>
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={() => {
|
||||||
|
if (!isCreating) {
|
||||||
|
setIsCreateDialogOpen(false);
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
disabled={isCreating}
|
||||||
|
className="px-4 py-2 rounded-lg border border-gray-300 text-gray-700 hover:bg-gray-50 disabled:cursor-not-allowed"
|
||||||
|
>
|
||||||
|
Cancel
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="submit"
|
||||||
|
disabled={isCreating || !newTeamName.trim()}
|
||||||
|
className="px-5 py-2 rounded-lg bg-blue-600 text-white font-medium hover:bg-blue-700 disabled:cursor-not-allowed disabled:opacity-50"
|
||||||
|
>
|
||||||
|
{isCreating ? "Creating..." : "Create Team"}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
<Dialog
|
<div className="space-y-4">
|
||||||
open={detailTarget !== null}
|
<h2 className="text-xl font-semibold text-gray-900">
|
||||||
onOpenChange={(open) => {
|
Your Teams ({isLoading ? "..." : teams.length})
|
||||||
if (!open && !isSavingDetails) {
|
</h2>
|
||||||
resetTeamDetails();
|
{loadError !== null ? (
|
||||||
}
|
<div className="rounded-md border border-red-200 bg-red-50 px-4 py-3 text-red-700">
|
||||||
}}
|
{loadError}
|
||||||
>
|
</div>
|
||||||
<DialogContent>
|
) : isLoading ? (
|
||||||
<DialogHeader>
|
<div className="bg-white rounded-lg shadow-sm border border-gray-200 p-12 text-center text-gray-600">
|
||||||
<DialogTitle>Team Details</DialogTitle>
|
Loading teams...
|
||||||
<DialogDescription>
|
</div>
|
||||||
Edit team details for {detailTarget?.name ?? "selected team"}.
|
) : teams.length === 0 ? (
|
||||||
</DialogDescription>
|
<div className="bg-white rounded-lg shadow-sm border border-gray-200 p-12 text-center">
|
||||||
</DialogHeader>
|
<svg
|
||||||
|
className="mx-auto h-12 w-12 text-gray-400 mb-4"
|
||||||
<form
|
fill="none"
|
||||||
onSubmit={(event) => {
|
stroke="currentColor"
|
||||||
void handleDetailSubmit(event);
|
viewBox="0 0 24 24"
|
||||||
}}
|
|
||||||
className="space-y-4"
|
|
||||||
>
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="detail-name">Name</Label>
|
|
||||||
<Input
|
|
||||||
id="detail-name"
|
|
||||||
value={detailForm.name}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setDetailForm((prev) => ({ ...prev, name: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="Team name"
|
|
||||||
maxLength={100}
|
|
||||||
disabled={isSavingDetails}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="detail-description">Description</Label>
|
|
||||||
<Textarea
|
|
||||||
id="detail-description"
|
|
||||||
value={detailForm.description}
|
|
||||||
onChange={(event: ChangeEvent<HTMLTextAreaElement>) => {
|
|
||||||
setDetailForm((prev) => ({ ...prev, description: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="Describe this team"
|
|
||||||
maxLength={500}
|
|
||||||
rows={4}
|
|
||||||
disabled={isSavingDetails}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{detailError !== null ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{detailError}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<DialogFooter>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="outline"
|
|
||||||
onClick={() => {
|
|
||||||
if (!isSavingDetails) {
|
|
||||||
resetTeamDetails();
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
disabled={isSavingDetails}
|
|
||||||
>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
<Button type="submit" disabled={isSavingDetails}>
|
|
||||||
{isSavingDetails ? "Saving..." : "Save Changes"}
|
|
||||||
</Button>
|
|
||||||
</DialogFooter>
|
|
||||||
</form>
|
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
|
||||||
|
|
||||||
<AlertDialog
|
|
||||||
open={deleteTarget !== null}
|
|
||||||
onOpenChange={(open) => {
|
|
||||||
if (!open && !isDeleting) {
|
|
||||||
setDeleteTarget(null);
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<AlertDialogContent>
|
|
||||||
<AlertDialogHeader>
|
|
||||||
<AlertDialogTitle>Delete Team</AlertDialogTitle>
|
|
||||||
<AlertDialogDescription>
|
|
||||||
Delete {deleteTarget?.name}? Team members will be removed from this team assignment.
|
|
||||||
</AlertDialogDescription>
|
|
||||||
</AlertDialogHeader>
|
|
||||||
<AlertDialogFooter>
|
|
||||||
<AlertDialogCancel disabled={isDeleting}>Cancel</AlertDialogCancel>
|
|
||||||
<AlertDialogAction
|
|
||||||
disabled={isDeleting}
|
|
||||||
onClick={() => {
|
|
||||||
void confirmDelete();
|
|
||||||
}}
|
|
||||||
>
|
>
|
||||||
{isDeleting ? "Deleting..." : "Delete Team"}
|
<path
|
||||||
</AlertDialogAction>
|
strokeLinecap="round"
|
||||||
</AlertDialogFooter>
|
strokeLinejoin="round"
|
||||||
</AlertDialogContent>
|
strokeWidth={2}
|
||||||
</AlertDialog>
|
d="M17 20h5V8H2v12h5m10 0v-4a3 3 0 10-6 0v4m6 0H7"
|
||||||
</div>
|
/>
|
||||||
|
</svg>
|
||||||
|
<h3 className="text-lg font-medium text-gray-900 mb-2">No teams yet</h3>
|
||||||
|
<p className="text-gray-600">Create your first team to get started</p>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="grid grid-cols-1 md:grid-cols-2 gap-4">
|
||||||
|
{teams.map((team) => (
|
||||||
|
<article
|
||||||
|
key={team.id}
|
||||||
|
className="rounded-lg border border-gray-200 bg-white p-5 shadow-sm"
|
||||||
|
data-testid="team-card"
|
||||||
|
>
|
||||||
|
<h3 className="text-lg font-semibold text-gray-900">{team.name}</h3>
|
||||||
|
{team.description ? (
|
||||||
|
<p className="mt-1 text-sm text-gray-600">{team.description}</p>
|
||||||
|
) : (
|
||||||
|
<p className="mt-1 text-sm text-gray-400 italic">No description</p>
|
||||||
|
)}
|
||||||
|
<div className="mt-4 flex items-center gap-3 text-xs text-gray-500">
|
||||||
|
<span>{team._count?.members ?? 0} members</span>
|
||||||
|
<span>|</span>
|
||||||
|
<span>Created {new Date(team.createdAt).toLocaleDateString()}</span>
|
||||||
|
</div>
|
||||||
|
</article>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</main>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,352 +0,0 @@
|
|||||||
import type { ReactElement, ReactNode } from "react";
|
|
||||||
|
|
||||||
import { WorkspaceMemberRole } from "@mosaic/shared";
|
|
||||||
import { render, screen, waitFor, within } from "@testing-library/react";
|
|
||||||
import userEvent from "@testing-library/user-event";
|
|
||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
|
|
||||||
import {
|
|
||||||
type AdminUser,
|
|
||||||
deactivateUser,
|
|
||||||
fetchAdminUsers,
|
|
||||||
inviteUser,
|
|
||||||
updateUser,
|
|
||||||
type AdminUsersResponse,
|
|
||||||
} from "@/lib/api/admin";
|
|
||||||
import { useAuth } from "@/lib/auth/auth-context";
|
|
||||||
import { fetchUserWorkspaces, updateWorkspaceMemberRole } from "@/lib/api/workspaces";
|
|
||||||
import UsersSettingsPage from "./page";
|
|
||||||
|
|
||||||
vi.mock("next/link", () => ({
|
|
||||||
default: function LinkMock({
|
|
||||||
children,
|
|
||||||
href,
|
|
||||||
}: {
|
|
||||||
children: ReactNode;
|
|
||||||
href: string;
|
|
||||||
}): ReactElement {
|
|
||||||
return <a href={href}>{children}</a>;
|
|
||||||
},
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock("@/lib/api/admin", () => ({
|
|
||||||
fetchAdminUsers: vi.fn(),
|
|
||||||
inviteUser: vi.fn(),
|
|
||||||
updateUser: vi.fn(),
|
|
||||||
deactivateUser: vi.fn(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock("@/lib/api/workspaces", () => ({
|
|
||||||
fetchUserWorkspaces: vi.fn(),
|
|
||||||
updateWorkspaceMemberRole: vi.fn(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock("@/lib/auth/auth-context", () => ({
|
|
||||||
useAuth: vi.fn(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
const fetchAdminUsersMock = vi.mocked(fetchAdminUsers);
|
|
||||||
const inviteUserMock = vi.mocked(inviteUser);
|
|
||||||
const updateUserMock = vi.mocked(updateUser);
|
|
||||||
const deactivateUserMock = vi.mocked(deactivateUser);
|
|
||||||
const fetchUserWorkspacesMock = vi.mocked(fetchUserWorkspaces);
|
|
||||||
const updateWorkspaceMemberRoleMock = vi.mocked(updateWorkspaceMemberRole);
|
|
||||||
const useAuthMock = vi.mocked(useAuth);
|
|
||||||
|
|
||||||
function makeAdminUser(overrides?: Partial<AdminUser>): AdminUser {
|
|
||||||
return {
|
|
||||||
id: "user-1",
|
|
||||||
name: "Alice",
|
|
||||||
email: "alice@example.com",
|
|
||||||
emailVerified: true,
|
|
||||||
image: null,
|
|
||||||
createdAt: "2026-01-01T00:00:00.000Z",
|
|
||||||
deactivatedAt: null,
|
|
||||||
isLocalAuth: false,
|
|
||||||
invitedAt: null,
|
|
||||||
invitedBy: null,
|
|
||||||
workspaceMemberships: [
|
|
||||||
{
|
|
||||||
workspaceId: "workspace-1",
|
|
||||||
workspaceName: "Personal Workspace",
|
|
||||||
role: WorkspaceMemberRole.ADMIN,
|
|
||||||
joinedAt: "2026-01-01T00:00:00.000Z",
|
|
||||||
},
|
|
||||||
],
|
|
||||||
...overrides,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function makeAdminUsersResponse(options?: {
|
|
||||||
data?: AdminUser[];
|
|
||||||
page?: number;
|
|
||||||
totalPages?: number;
|
|
||||||
total?: number;
|
|
||||||
limit?: number;
|
|
||||||
}): AdminUsersResponse {
|
|
||||||
const data = options?.data ?? [makeAdminUser()];
|
|
||||||
return {
|
|
||||||
data,
|
|
||||||
meta: {
|
|
||||||
total: options?.total ?? data.length,
|
|
||||||
page: options?.page ?? 1,
|
|
||||||
limit: options?.limit ?? 50,
|
|
||||||
totalPages: options?.totalPages ?? 1,
|
|
||||||
},
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function makeAuthState(userId: string): ReturnType<typeof useAuth> {
|
|
||||||
return {
|
|
||||||
user: { id: userId, email: `${userId}@example.com`, name: "Current User" },
|
|
||||||
isLoading: false,
|
|
||||||
isAuthenticated: true,
|
|
||||||
authError: null,
|
|
||||||
sessionExpiring: false,
|
|
||||||
sessionMinutesRemaining: 0,
|
|
||||||
signOut: vi.fn(() => Promise.resolve()),
|
|
||||||
refreshSession: vi.fn(() => Promise.resolve()),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
describe("UsersSettingsPage", () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
|
|
||||||
const adminUsersResponse = makeAdminUsersResponse();
|
|
||||||
|
|
||||||
fetchAdminUsersMock.mockResolvedValue(adminUsersResponse);
|
|
||||||
fetchUserWorkspacesMock.mockResolvedValue([
|
|
||||||
{
|
|
||||||
id: "workspace-1",
|
|
||||||
name: "Personal Workspace",
|
|
||||||
ownerId: "owner-1",
|
|
||||||
role: WorkspaceMemberRole.OWNER,
|
|
||||||
createdAt: "2026-01-01T00:00:00.000Z",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
inviteUserMock.mockResolvedValue({
|
|
||||||
userId: "user-2",
|
|
||||||
invitationToken: "token-1",
|
|
||||||
email: "new@example.com",
|
|
||||||
invitedAt: "2026-01-02T00:00:00.000Z",
|
|
||||||
});
|
|
||||||
const firstUser = adminUsersResponse.data[0] ?? makeAdminUser();
|
|
||||||
|
|
||||||
updateUserMock.mockResolvedValue(firstUser);
|
|
||||||
deactivateUserMock.mockResolvedValue(firstUser);
|
|
||||||
updateWorkspaceMemberRoleMock.mockResolvedValue({
|
|
||||||
workspaceId: "workspace-1",
|
|
||||||
userId: "user-1",
|
|
||||||
role: WorkspaceMemberRole.ADMIN,
|
|
||||||
joinedAt: "2026-01-01T00:00:00.000Z",
|
|
||||||
user: {
|
|
||||||
id: "user-1",
|
|
||||||
email: "alice@example.com",
|
|
||||||
name: "Alice",
|
|
||||||
image: null,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
useAuthMock.mockReturnValue(makeAuthState("user-current"));
|
|
||||||
});
|
|
||||||
|
|
||||||
it("shows access denied to non-admin users", async () => {
|
|
||||||
fetchUserWorkspacesMock.mockResolvedValueOnce([
|
|
||||||
{
|
|
||||||
id: "workspace-1",
|
|
||||||
name: "Personal Workspace",
|
|
||||||
ownerId: "owner-1",
|
|
||||||
role: WorkspaceMemberRole.MEMBER,
|
|
||||||
createdAt: "2026-01-01T00:00:00.000Z",
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
render(<UsersSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("Access Denied")).toBeInTheDocument();
|
|
||||||
expect(fetchAdminUsersMock).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("invites a user with email and role from the dialog", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
render(<UsersSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("User Directory")).toBeInTheDocument();
|
|
||||||
|
|
||||||
await user.click(screen.getByRole("button", { name: "Invite User" }));
|
|
||||||
await user.type(screen.getByLabelText("Email"), "new@example.com");
|
|
||||||
await user.click(screen.getByRole("button", { name: "Send Invite" }));
|
|
||||||
|
|
||||||
await waitFor(() => {
|
|
||||||
expect(inviteUserMock).toHaveBeenCalledWith({
|
|
||||||
email: "new@example.com",
|
|
||||||
role: WorkspaceMemberRole.MEMBER,
|
|
||||||
workspaceId: "workspace-1",
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it("opens user detail dialog from row click and saves edited profile fields", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
render(<UsersSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("alice@example.com")).toBeInTheDocument();
|
|
||||||
|
|
||||||
await user.click(screen.getByText("Alice"));
|
|
||||||
|
|
||||||
const nameInput = await screen.findByLabelText("Name");
|
|
||||||
await user.clear(nameInput);
|
|
||||||
await user.type(nameInput, "Alice Updated");
|
|
||||||
|
|
||||||
await user.click(screen.getByRole("button", { name: "Save Changes" }));
|
|
||||||
|
|
||||||
await waitFor(() => {
|
|
||||||
expect(updateUserMock).toHaveBeenCalledWith("user-1", { name: "Alice Updated" });
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(updateWorkspaceMemberRoleMock).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("caps pagination to the last valid page after deactivation shrinks the dataset", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
const pageOneUser = makeAdminUser({
|
|
||||||
id: "user-1",
|
|
||||||
name: "Alice",
|
|
||||||
email: "alice@example.com",
|
|
||||||
});
|
|
||||||
const pageTwoUser = makeAdminUser({
|
|
||||||
id: "user-2",
|
|
||||||
name: "Bob",
|
|
||||||
email: "bob@example.com",
|
|
||||||
});
|
|
||||||
|
|
||||||
fetchAdminUsersMock.mockReset();
|
|
||||||
const responses = [
|
|
||||||
{
|
|
||||||
expectedPage: 1,
|
|
||||||
response: makeAdminUsersResponse({
|
|
||||||
data: [pageOneUser],
|
|
||||||
page: 1,
|
|
||||||
totalPages: 2,
|
|
||||||
total: 2,
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
expectedPage: 2,
|
|
||||||
response: makeAdminUsersResponse({
|
|
||||||
data: [pageTwoUser],
|
|
||||||
page: 2,
|
|
||||||
totalPages: 2,
|
|
||||||
total: 2,
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
expectedPage: 2,
|
|
||||||
response: makeAdminUsersResponse({
|
|
||||||
data: [],
|
|
||||||
page: 2,
|
|
||||||
totalPages: 1,
|
|
||||||
total: 1,
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
expectedPage: 1,
|
|
||||||
response: makeAdminUsersResponse({
|
|
||||||
data: [pageOneUser],
|
|
||||||
page: 1,
|
|
||||||
totalPages: 1,
|
|
||||||
total: 1,
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
fetchAdminUsersMock.mockImplementation((page = 1) => {
|
|
||||||
const next = responses.shift();
|
|
||||||
if (!next) {
|
|
||||||
throw new Error("Unexpected fetchAdminUsers call in pagination-cap test");
|
|
||||||
}
|
|
||||||
|
|
||||||
expect(page).toBe(next.expectedPage);
|
|
||||||
return Promise.resolve(next.response);
|
|
||||||
});
|
|
||||||
|
|
||||||
render(<UsersSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("alice@example.com")).toBeInTheDocument();
|
|
||||||
|
|
||||||
await user.click(screen.getByRole("button", { name: "Next" }));
|
|
||||||
expect(await screen.findByText("bob@example.com")).toBeInTheDocument();
|
|
||||||
|
|
||||||
const pageTwoRow = screen.getByText("bob@example.com").closest('[role="button"]');
|
|
||||||
if (!(pageTwoRow instanceof HTMLElement)) {
|
|
||||||
throw new Error("Expected Bob's row to exist");
|
|
||||||
}
|
|
||||||
|
|
||||||
await user.click(within(pageTwoRow).getByRole("button", { name: "Deactivate" }));
|
|
||||||
const deactivateButtons = await screen.findAllByRole("button", { name: "Deactivate" });
|
|
||||||
const confirmDeactivateButton = deactivateButtons[deactivateButtons.length - 1];
|
|
||||||
if (!confirmDeactivateButton) {
|
|
||||||
throw new Error("Expected confirmation deactivate button to be rendered");
|
|
||||||
}
|
|
||||||
await user.click(confirmDeactivateButton);
|
|
||||||
|
|
||||||
expect(await screen.findByText("alice@example.com")).toBeInTheDocument();
|
|
||||||
expect(screen.queryByText("No Users Yet")).not.toBeInTheDocument();
|
|
||||||
expect(deactivateUserMock).toHaveBeenCalledWith("user-2");
|
|
||||||
const requestedPages = fetchAdminUsersMock.mock.calls.map(([requestedPage]) => requestedPage);
|
|
||||||
expect(requestedPages.slice(-2)).toEqual([2, 1]);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("shows the API error state without rendering the empty-state message", async () => {
|
|
||||||
fetchAdminUsersMock.mockRejectedValueOnce(new Error("Unable to load users"));
|
|
||||||
|
|
||||||
render(<UsersSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("Unable to load users")).toBeInTheDocument();
|
|
||||||
expect(screen.queryByText("No Users Yet")).not.toBeInTheDocument();
|
|
||||||
expect(screen.queryByText("Invite the first user to get started.")).not.toBeInTheDocument();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("prevents the current user from deactivating their own account", async () => {
|
|
||||||
useAuthMock.mockReturnValue(makeAuthState("user-1"));
|
|
||||||
|
|
||||||
const selfUser = makeAdminUser({
|
|
||||||
id: "user-1",
|
|
||||||
name: "Alice",
|
|
||||||
email: "alice@example.com",
|
|
||||||
});
|
|
||||||
const otherUser = makeAdminUser({
|
|
||||||
id: "user-2",
|
|
||||||
name: "Bob",
|
|
||||||
email: "bob@example.com",
|
|
||||||
});
|
|
||||||
|
|
||||||
fetchAdminUsersMock.mockResolvedValueOnce(
|
|
||||||
makeAdminUsersResponse({
|
|
||||||
data: [selfUser, otherUser],
|
|
||||||
page: 1,
|
|
||||||
totalPages: 1,
|
|
||||||
total: 2,
|
|
||||||
})
|
|
||||||
);
|
|
||||||
|
|
||||||
render(<UsersSettingsPage />);
|
|
||||||
|
|
||||||
expect(await screen.findByText("alice@example.com")).toBeInTheDocument();
|
|
||||||
expect(screen.getByText("bob@example.com")).toBeInTheDocument();
|
|
||||||
|
|
||||||
const selfRow = screen.getByText("alice@example.com").closest('[role="button"]');
|
|
||||||
if (!(selfRow instanceof HTMLElement)) {
|
|
||||||
throw new Error("Expected current-user row to exist");
|
|
||||||
}
|
|
||||||
expect(within(selfRow).queryByRole("button", { name: "Deactivate" })).not.toBeInTheDocument();
|
|
||||||
|
|
||||||
const otherRow = screen.getByText("bob@example.com").closest('[role="button"]');
|
|
||||||
if (!(otherRow instanceof HTMLElement)) {
|
|
||||||
throw new Error("Expected other-user row to exist");
|
|
||||||
}
|
|
||||||
expect(within(otherRow).getByRole("button", { name: "Deactivate" })).toBeInTheDocument();
|
|
||||||
expect(deactivateUserMock).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -5,14 +5,12 @@ import {
|
|||||||
useEffect,
|
useEffect,
|
||||||
useState,
|
useState,
|
||||||
type ChangeEvent,
|
type ChangeEvent,
|
||||||
type KeyboardEvent,
|
|
||||||
type ReactElement,
|
type ReactElement,
|
||||||
type SyntheticEvent,
|
type SyntheticEvent,
|
||||||
} from "react";
|
} from "react";
|
||||||
import Link from "next/link";
|
import Link from "next/link";
|
||||||
import { UserPlus, UserX } from "lucide-react";
|
import { Pencil, UserPlus, UserX } from "lucide-react";
|
||||||
import { WorkspaceMemberRole } from "@mosaic/shared";
|
import { WorkspaceMemberRole } from "@mosaic/shared";
|
||||||
import { isValidEmail } from "@/components/workspace/validation";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
import { Badge } from "@/components/ui/badge";
|
||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
||||||
@@ -51,13 +49,9 @@ import {
|
|||||||
updateUser,
|
updateUser,
|
||||||
type AdminUser,
|
type AdminUser,
|
||||||
type AdminUsersResponse,
|
type AdminUsersResponse,
|
||||||
type AdminWorkspaceMembership,
|
|
||||||
type InviteUserDto,
|
type InviteUserDto,
|
||||||
type UpdateUserDto,
|
type UpdateUserDto,
|
||||||
} from "@/lib/api/admin";
|
} from "@/lib/api/admin";
|
||||||
import { useAuth } from "@/lib/auth/auth-context";
|
|
||||||
import { fetchUserWorkspaces, updateWorkspaceMemberRole } from "@/lib/api/workspaces";
|
|
||||||
import { SettingsAccessDenied } from "@/components/settings/SettingsAccessDenied";
|
|
||||||
|
|
||||||
const ROLE_PRIORITY: Record<WorkspaceMemberRole, number> = {
|
const ROLE_PRIORITY: Record<WorkspaceMemberRole, number> = {
|
||||||
[WorkspaceMemberRole.OWNER]: 4,
|
[WorkspaceMemberRole.OWNER]: 4,
|
||||||
@@ -68,168 +62,87 @@ const ROLE_PRIORITY: Record<WorkspaceMemberRole, number> = {
|
|||||||
|
|
||||||
const INITIAL_INVITE_FORM = {
|
const INITIAL_INVITE_FORM = {
|
||||||
email: "",
|
email: "",
|
||||||
role: WorkspaceMemberRole.MEMBER,
|
|
||||||
};
|
|
||||||
|
|
||||||
const INITIAL_DETAIL_FORM = {
|
|
||||||
name: "",
|
name: "",
|
||||||
email: "",
|
workspaceId: "",
|
||||||
role: WorkspaceMemberRole.MEMBER,
|
role: WorkspaceMemberRole.MEMBER,
|
||||||
workspaceId: null as string | null,
|
|
||||||
workspaceName: null as string | null,
|
|
||||||
};
|
};
|
||||||
const USERS_PAGE_SIZE = 50;
|
|
||||||
|
|
||||||
interface DetailInitialState {
|
|
||||||
name: string;
|
|
||||||
email: string;
|
|
||||||
role: WorkspaceMemberRole;
|
|
||||||
workspaceId: string | null;
|
|
||||||
}
|
|
||||||
|
|
||||||
function toRoleLabel(role: WorkspaceMemberRole): string {
|
function toRoleLabel(role: WorkspaceMemberRole): string {
|
||||||
return `${role.charAt(0)}${role.slice(1).toLowerCase()}`;
|
return `${role.charAt(0)}${role.slice(1).toLowerCase()}`;
|
||||||
}
|
}
|
||||||
|
|
||||||
function getPrimaryMembership(user: AdminUser): AdminWorkspaceMembership | null {
|
function getPrimaryRole(user: AdminUser): WorkspaceMemberRole | null {
|
||||||
const [firstMembership, ...restMemberships] = user.workspaceMemberships;
|
const [firstMembership, ...restMemberships] = user.workspaceMemberships;
|
||||||
if (!firstMembership) {
|
if (!firstMembership) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
return restMemberships.reduce((highest, membership) => {
|
return restMemberships.reduce((highest, membership) => {
|
||||||
if (ROLE_PRIORITY[membership.role] > ROLE_PRIORITY[highest.role]) {
|
if (ROLE_PRIORITY[membership.role] > ROLE_PRIORITY[highest]) {
|
||||||
return membership;
|
return membership.role;
|
||||||
}
|
}
|
||||||
return highest;
|
return highest;
|
||||||
}, firstMembership);
|
}, firstMembership.role);
|
||||||
}
|
}
|
||||||
|
|
||||||
export default function UsersSettingsPage(): ReactElement {
|
export default function UsersSettingsPage(): ReactElement {
|
||||||
const { user: authUser } = useAuth();
|
|
||||||
|
|
||||||
const [users, setUsers] = useState<AdminUser[]>([]);
|
const [users, setUsers] = useState<AdminUser[]>([]);
|
||||||
const [meta, setMeta] = useState<AdminUsersResponse["meta"] | null>(null);
|
const [meta, setMeta] = useState<AdminUsersResponse["meta"] | null>(null);
|
||||||
const [page, setPage] = useState<number>(1);
|
|
||||||
const [isLoading, setIsLoading] = useState<boolean>(true);
|
const [isLoading, setIsLoading] = useState<boolean>(true);
|
||||||
const [isRefreshing, setIsRefreshing] = useState<boolean>(false);
|
const [isRefreshing, setIsRefreshing] = useState<boolean>(false);
|
||||||
const [error, setError] = useState<string | null>(null);
|
const [error, setError] = useState<string | null>(null);
|
||||||
|
|
||||||
const [defaultWorkspaceId, setDefaultWorkspaceId] = useState<string | null>(null);
|
|
||||||
const [isAdmin, setIsAdmin] = useState<boolean | null>(null);
|
|
||||||
|
|
||||||
const [isInviteOpen, setIsInviteOpen] = useState<boolean>(false);
|
const [isInviteOpen, setIsInviteOpen] = useState<boolean>(false);
|
||||||
const [inviteForm, setInviteForm] = useState(INITIAL_INVITE_FORM);
|
const [inviteForm, setInviteForm] = useState(INITIAL_INVITE_FORM);
|
||||||
const [inviteError, setInviteError] = useState<string | null>(null);
|
const [inviteError, setInviteError] = useState<string | null>(null);
|
||||||
const [isInviting, setIsInviting] = useState<boolean>(false);
|
const [isInviting, setIsInviting] = useState<boolean>(false);
|
||||||
|
|
||||||
const [detailTarget, setDetailTarget] = useState<AdminUser | null>(null);
|
|
||||||
const [detailForm, setDetailForm] = useState(INITIAL_DETAIL_FORM);
|
|
||||||
const [detailInitial, setDetailInitial] = useState<DetailInitialState | null>(null);
|
|
||||||
const [detailError, setDetailError] = useState<string | null>(null);
|
|
||||||
const [isSavingDetails, setIsSavingDetails] = useState<boolean>(false);
|
|
||||||
|
|
||||||
const [deactivateTarget, setDeactivateTarget] = useState<AdminUser | null>(null);
|
const [deactivateTarget, setDeactivateTarget] = useState<AdminUser | null>(null);
|
||||||
const [isDeactivating, setIsDeactivating] = useState<boolean>(false);
|
const [isDeactivating, setIsDeactivating] = useState<boolean>(false);
|
||||||
|
|
||||||
const loadUsers = useCallback(
|
const [editTarget, setEditTarget] = useState<AdminUser | null>(null);
|
||||||
async (showLoadingState: boolean): Promise<void> => {
|
const [editName, setEditName] = useState<string>("");
|
||||||
try {
|
const [editError, setEditError] = useState<string | null>(null);
|
||||||
if (showLoadingState) {
|
const [isEditing, setIsEditing] = useState<boolean>(false);
|
||||||
setIsLoading(true);
|
|
||||||
} else {
|
|
||||||
setIsRefreshing(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
const response = await fetchAdminUsers(page, USERS_PAGE_SIZE);
|
const loadUsers = useCallback(async (showLoadingState: boolean): Promise<void> => {
|
||||||
const lastValidPage = Math.max(1, response.meta.totalPages);
|
try {
|
||||||
|
if (showLoadingState) {
|
||||||
if (page > lastValidPage) {
|
setIsLoading(true);
|
||||||
setPage(lastValidPage);
|
} else {
|
||||||
return;
|
setIsRefreshing(true);
|
||||||
}
|
|
||||||
|
|
||||||
setUsers(response.data);
|
|
||||||
setMeta(response.meta);
|
|
||||||
setError(null);
|
|
||||||
} catch (err: unknown) {
|
|
||||||
setError(err instanceof Error ? err.message : "Failed to load admin users");
|
|
||||||
} finally {
|
|
||||||
setIsLoading(false);
|
|
||||||
setIsRefreshing(false);
|
|
||||||
}
|
}
|
||||||
},
|
|
||||||
[page]
|
|
||||||
);
|
|
||||||
|
|
||||||
useEffect(() => {
|
const response = await fetchAdminUsers(1, 50);
|
||||||
fetchUserWorkspaces()
|
setUsers(response.data);
|
||||||
.then((workspaces) => {
|
setMeta(response.meta);
|
||||||
const adminRoles: WorkspaceMemberRole[] = [
|
setError(null);
|
||||||
WorkspaceMemberRole.OWNER,
|
} catch (err: unknown) {
|
||||||
WorkspaceMemberRole.ADMIN,
|
setError(err instanceof Error ? err.message : "Failed to load admin users");
|
||||||
];
|
} finally {
|
||||||
|
setIsLoading(false);
|
||||||
setDefaultWorkspaceId(workspaces[0]?.id ?? null);
|
setIsRefreshing(false);
|
||||||
setIsAdmin(workspaces.some((workspace) => adminRoles.includes(workspace.role)));
|
}
|
||||||
})
|
|
||||||
.catch(() => {
|
|
||||||
setDefaultWorkspaceId(null);
|
|
||||||
setIsAdmin(true); // fail open
|
|
||||||
});
|
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (isAdmin !== true) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
void loadUsers(true);
|
void loadUsers(true);
|
||||||
}, [isAdmin, loadUsers, page]);
|
}, [loadUsers]);
|
||||||
|
|
||||||
function resetInviteForm(): void {
|
function resetInviteForm(): void {
|
||||||
setInviteForm(INITIAL_INVITE_FORM);
|
setInviteForm(INITIAL_INVITE_FORM);
|
||||||
setInviteError(null);
|
setInviteError(null);
|
||||||
}
|
}
|
||||||
|
|
||||||
function openUserDetails(user: AdminUser): void {
|
function handleInviteOpenChange(open: boolean): void {
|
||||||
const primaryMembership = getPrimaryMembership(user);
|
if (!open && !isInviting) {
|
||||||
|
resetInviteForm();
|
||||||
const nextDetailForm = {
|
|
||||||
name: user.name,
|
|
||||||
email: user.email,
|
|
||||||
role: primaryMembership?.role ?? WorkspaceMemberRole.MEMBER,
|
|
||||||
workspaceId: primaryMembership?.workspaceId ?? null,
|
|
||||||
workspaceName: primaryMembership?.workspaceName ?? null,
|
|
||||||
};
|
|
||||||
|
|
||||||
setDetailTarget(user);
|
|
||||||
setDetailForm(nextDetailForm);
|
|
||||||
setDetailInitial({
|
|
||||||
name: nextDetailForm.name,
|
|
||||||
email: nextDetailForm.email,
|
|
||||||
role: nextDetailForm.role,
|
|
||||||
workspaceId: nextDetailForm.workspaceId,
|
|
||||||
});
|
|
||||||
setDetailError(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
function resetUserDetails(): void {
|
|
||||||
setDetailTarget(null);
|
|
||||||
setDetailForm(INITIAL_DETAIL_FORM);
|
|
||||||
setDetailInitial(null);
|
|
||||||
setDetailError(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
function handleUserRowKeyDown(event: KeyboardEvent<HTMLDivElement>, user: AdminUser): void {
|
|
||||||
if (event.key === "Enter" || event.key === " ") {
|
|
||||||
event.preventDefault();
|
|
||||||
openUserDetails(user);
|
|
||||||
}
|
}
|
||||||
|
setIsInviteOpen(open);
|
||||||
}
|
}
|
||||||
|
|
||||||
async function handleInviteSubmit(event: SyntheticEvent): Promise<void> {
|
async function handleInviteSubmit(e: SyntheticEvent): Promise<void> {
|
||||||
event.preventDefault();
|
e.preventDefault();
|
||||||
setInviteError(null);
|
setInviteError(null);
|
||||||
|
|
||||||
const email = inviteForm.email.trim();
|
const email = inviteForm.email.trim();
|
||||||
@@ -238,18 +151,17 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!isValidEmail(email)) {
|
const dto: InviteUserDto = { email };
|
||||||
setInviteError("Please enter a valid email address.");
|
|
||||||
return;
|
const name = inviteForm.name.trim();
|
||||||
|
if (name) {
|
||||||
|
dto.name = name;
|
||||||
}
|
}
|
||||||
|
|
||||||
const dto: InviteUserDto = {
|
const workspaceId = inviteForm.workspaceId.trim();
|
||||||
email,
|
if (workspaceId) {
|
||||||
role: inviteForm.role,
|
dto.workspaceId = workspaceId;
|
||||||
};
|
dto.role = inviteForm.role;
|
||||||
|
|
||||||
if (defaultWorkspaceId) {
|
|
||||||
dto.workspaceId = defaultWorkspaceId;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
@@ -265,86 +177,11 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async function handleDetailSubmit(event: SyntheticEvent): Promise<void> {
|
|
||||||
event.preventDefault();
|
|
||||||
|
|
||||||
if (detailTarget === null || detailInitial === null) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const name = detailForm.name.trim();
|
|
||||||
const email = detailForm.email.trim();
|
|
||||||
|
|
||||||
if (!name) {
|
|
||||||
setDetailError("Name is required.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!email) {
|
|
||||||
setDetailError("Email is required.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!isValidEmail(email)) {
|
|
||||||
setDetailError("Please enter a valid email address.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const didUpdateUser = name !== detailInitial.name || email !== detailInitial.email;
|
|
||||||
const didUpdateRole =
|
|
||||||
detailForm.workspaceId !== null &&
|
|
||||||
detailForm.workspaceId === detailInitial.workspaceId &&
|
|
||||||
detailForm.role !== detailInitial.role;
|
|
||||||
|
|
||||||
if (!didUpdateUser && !didUpdateRole) {
|
|
||||||
resetUserDetails();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsSavingDetails(true);
|
|
||||||
setDetailError(null);
|
|
||||||
|
|
||||||
if (didUpdateUser) {
|
|
||||||
const dto: UpdateUserDto = {};
|
|
||||||
|
|
||||||
if (name !== detailInitial.name) {
|
|
||||||
dto.name = name;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (email !== detailInitial.email) {
|
|
||||||
dto.email = email;
|
|
||||||
}
|
|
||||||
|
|
||||||
await updateUser(detailTarget.id, dto);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (didUpdateRole && detailForm.workspaceId !== null) {
|
|
||||||
await updateWorkspaceMemberRole(detailForm.workspaceId, detailTarget.id, {
|
|
||||||
role: detailForm.role,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
resetUserDetails();
|
|
||||||
await loadUsers(false);
|
|
||||||
} catch (err: unknown) {
|
|
||||||
setDetailError(err instanceof Error ? err.message : "Failed to update user");
|
|
||||||
} finally {
|
|
||||||
setIsSavingDetails(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function confirmDeactivate(): Promise<void> {
|
async function confirmDeactivate(): Promise<void> {
|
||||||
if (!deactivateTarget) {
|
if (!deactivateTarget) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (authUser?.id === deactivateTarget.id) {
|
|
||||||
setDeactivateTarget(null);
|
|
||||||
setError("You cannot deactivate your own account.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
setIsDeactivating(true);
|
setIsDeactivating(true);
|
||||||
await deactivateUser(deactivateTarget.id);
|
await deactivateUser(deactivateTarget.id);
|
||||||
@@ -358,18 +195,21 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (isAdmin === null) {
|
async function handleEditSubmit(): Promise<void> {
|
||||||
return (
|
if (editTarget === null) return;
|
||||||
<Card className="max-w-2xl mx-auto mt-8">
|
setIsEditing(true);
|
||||||
<CardContent className="py-12 text-center text-muted-foreground">
|
setEditError(null);
|
||||||
Checking permissions...
|
try {
|
||||||
</CardContent>
|
const dto: UpdateUserDto = {};
|
||||||
</Card>
|
if (editName.trim()) dto.name = editName.trim();
|
||||||
);
|
await updateUser(editTarget.id, dto);
|
||||||
}
|
setEditTarget(null);
|
||||||
|
await loadUsers(false);
|
||||||
if (!isAdmin) {
|
} catch (err: unknown) {
|
||||||
return <SettingsAccessDenied message="You need Admin or Owner role to manage users." />;
|
setEditError(err instanceof Error ? err.message : "Failed to update user");
|
||||||
|
} finally {
|
||||||
|
setIsEditing(false);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
@@ -394,15 +234,7 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
{isRefreshing ? "Refreshing..." : "Refresh"}
|
{isRefreshing ? "Refreshing..." : "Refresh"}
|
||||||
</Button>
|
</Button>
|
||||||
|
|
||||||
<Dialog
|
<Dialog open={isInviteOpen} onOpenChange={handleInviteOpenChange}>
|
||||||
open={isInviteOpen}
|
|
||||||
onOpenChange={(open) => {
|
|
||||||
if (!open && !isInviting) {
|
|
||||||
resetInviteForm();
|
|
||||||
}
|
|
||||||
setIsInviteOpen(open);
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<DialogTrigger asChild>
|
<DialogTrigger asChild>
|
||||||
<Button>
|
<Button>
|
||||||
<UserPlus className="h-4 w-4 mr-2" />
|
<UserPlus className="h-4 w-4 mr-2" />
|
||||||
@@ -413,13 +245,13 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
<DialogHeader>
|
<DialogHeader>
|
||||||
<DialogTitle>Invite User</DialogTitle>
|
<DialogTitle>Invite User</DialogTitle>
|
||||||
<DialogDescription>
|
<DialogDescription>
|
||||||
Invite a new user and assign their role for your default workspace.
|
Create an invited account and optionally assign workspace access.
|
||||||
</DialogDescription>
|
</DialogDescription>
|
||||||
</DialogHeader>
|
</DialogHeader>
|
||||||
|
|
||||||
<form
|
<form
|
||||||
onSubmit={(event) => {
|
onSubmit={(e) => {
|
||||||
void handleInviteSubmit(event);
|
void handleInviteSubmit(e);
|
||||||
}}
|
}}
|
||||||
className="space-y-4"
|
className="space-y-4"
|
||||||
>
|
>
|
||||||
@@ -429,8 +261,8 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
id="invite-email"
|
id="invite-email"
|
||||||
type="email"
|
type="email"
|
||||||
value={inviteForm.email}
|
value={inviteForm.email}
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
onChange={(e: ChangeEvent<HTMLInputElement>) => {
|
||||||
setInviteForm((prev) => ({ ...prev, email: event.target.value }));
|
setInviteForm((prev) => ({ ...prev, email: e.target.value }));
|
||||||
}}
|
}}
|
||||||
placeholder="user@example.com"
|
placeholder="user@example.com"
|
||||||
maxLength={255}
|
maxLength={255}
|
||||||
@@ -438,6 +270,33 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label htmlFor="invite-name">Name (optional)</Label>
|
||||||
|
<Input
|
||||||
|
id="invite-name"
|
||||||
|
type="text"
|
||||||
|
value={inviteForm.name}
|
||||||
|
onChange={(e: ChangeEvent<HTMLInputElement>) => {
|
||||||
|
setInviteForm((prev) => ({ ...prev, name: e.target.value }));
|
||||||
|
}}
|
||||||
|
placeholder="Jane Doe"
|
||||||
|
maxLength={255}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label htmlFor="invite-workspace-id">Workspace ID (optional)</Label>
|
||||||
|
<Input
|
||||||
|
id="invite-workspace-id"
|
||||||
|
type="text"
|
||||||
|
value={inviteForm.workspaceId}
|
||||||
|
onChange={(e: ChangeEvent<HTMLInputElement>) => {
|
||||||
|
setInviteForm((prev) => ({ ...prev, workspaceId: e.target.value }));
|
||||||
|
}}
|
||||||
|
placeholder="UUID workspace id"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
<div className="space-y-2">
|
<div className="space-y-2">
|
||||||
<Label htmlFor="invite-role">Role</Label>
|
<Label htmlFor="invite-role">Role</Label>
|
||||||
<Select
|
<Select
|
||||||
@@ -457,13 +316,9 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
))}
|
))}
|
||||||
</SelectContent>
|
</SelectContent>
|
||||||
</Select>
|
</Select>
|
||||||
{defaultWorkspaceId ? (
|
<p className="text-xs text-muted-foreground">
|
||||||
<p className="text-xs text-muted-foreground">Role will be applied on invite.</p>
|
Role is only applied when workspace ID is provided.
|
||||||
) : (
|
</p>
|
||||||
<p className="text-xs text-muted-foreground">
|
|
||||||
No default workspace found. User will be invited without workspace assignment.
|
|
||||||
</p>
|
|
||||||
)}
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{inviteError ? (
|
{inviteError ? (
|
||||||
@@ -477,10 +332,7 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
type="button"
|
type="button"
|
||||||
variant="outline"
|
variant="outline"
|
||||||
onClick={() => {
|
onClick={() => {
|
||||||
if (!isInviting) {
|
handleInviteOpenChange(false);
|
||||||
setIsInviteOpen(false);
|
|
||||||
resetInviteForm();
|
|
||||||
}
|
|
||||||
}}
|
}}
|
||||||
disabled={isInviting}
|
disabled={isInviting}
|
||||||
>
|
>
|
||||||
@@ -502,13 +354,7 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
</Link>
|
</Link>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{isLoading ? (
|
{error ? (
|
||||||
<Card>
|
|
||||||
<CardContent className="py-12 text-center text-muted-foreground">
|
|
||||||
Loading users...
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
) : error ? (
|
|
||||||
<Card>
|
<Card>
|
||||||
<CardContent className="py-4">
|
<CardContent className="py-4">
|
||||||
<p className="text-sm text-destructive" role="alert">
|
<p className="text-sm text-destructive" role="alert">
|
||||||
@@ -516,6 +362,14 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
</p>
|
</p>
|
||||||
</CardContent>
|
</CardContent>
|
||||||
</Card>
|
</Card>
|
||||||
|
) : null}
|
||||||
|
|
||||||
|
{isLoading ? (
|
||||||
|
<Card>
|
||||||
|
<CardContent className="py-12 text-center text-muted-foreground">
|
||||||
|
Loading users...
|
||||||
|
</CardContent>
|
||||||
|
</Card>
|
||||||
) : users.length === 0 ? (
|
) : users.length === 0 ? (
|
||||||
<Card>
|
<Card>
|
||||||
<CardHeader>
|
<CardHeader>
|
||||||
@@ -527,52 +381,47 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
<Card>
|
<Card>
|
||||||
<CardHeader>
|
<CardHeader>
|
||||||
<CardTitle>User Directory</CardTitle>
|
<CardTitle>User Directory</CardTitle>
|
||||||
<CardDescription>Click a user to view details or edit profile fields.</CardDescription>
|
<CardDescription>Name, email, role, and account status.</CardDescription>
|
||||||
</CardHeader>
|
</CardHeader>
|
||||||
<CardContent className="space-y-3">
|
<CardContent className="space-y-3">
|
||||||
{users.map((user) => {
|
{users.map((user) => {
|
||||||
const primaryMembership = getPrimaryMembership(user);
|
const primaryRole = getPrimaryRole(user);
|
||||||
const isActive = user.deactivatedAt === null;
|
const isActive = user.deactivatedAt === null;
|
||||||
const isCurrentUser = authUser?.id === user.id;
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div
|
<div
|
||||||
key={user.id}
|
key={user.id}
|
||||||
className="rounded-md border p-4 flex flex-col gap-3 md:flex-row md:items-center md:justify-between cursor-pointer hover:bg-muted/30"
|
className="rounded-md border p-4 flex flex-col gap-3 md:flex-row md:items-center md:justify-between"
|
||||||
role="button"
|
|
||||||
tabIndex={0}
|
|
||||||
onClick={() => {
|
|
||||||
openUserDetails(user);
|
|
||||||
}}
|
|
||||||
onKeyDown={(event) => {
|
|
||||||
handleUserRowKeyDown(event, user);
|
|
||||||
}}
|
|
||||||
>
|
>
|
||||||
<div className="space-y-1 min-w-0">
|
<div className="space-y-1 min-w-0">
|
||||||
<p className="font-semibold truncate">
|
<p className="font-semibold truncate">{user.name || "Unnamed User"}</p>
|
||||||
{user.name || "Unnamed User"}
|
|
||||||
{isCurrentUser ? (
|
|
||||||
<span className="ml-2 text-xs font-normal text-muted-foreground">
|
|
||||||
(You)
|
|
||||||
</span>
|
|
||||||
) : null}
|
|
||||||
</p>
|
|
||||||
<p className="text-sm text-muted-foreground truncate">{user.email}</p>
|
<p className="text-sm text-muted-foreground truncate">{user.email}</p>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div className="flex items-center gap-2 flex-wrap md:justify-end">
|
<div className="flex items-center gap-2 flex-wrap md:justify-end">
|
||||||
<Badge variant="outline">
|
<Badge variant="outline">
|
||||||
{primaryMembership ? toRoleLabel(primaryMembership.role) : "No role"}
|
{primaryRole ? toRoleLabel(primaryRole) : "No role"}
|
||||||
</Badge>
|
</Badge>
|
||||||
<Badge variant={isActive ? "secondary" : "destructive"}>
|
<Badge variant={isActive ? "secondary" : "destructive"}>
|
||||||
{isActive ? "Active" : "Inactive"}
|
{isActive ? "Active" : "Inactive"}
|
||||||
</Badge>
|
</Badge>
|
||||||
{isActive && !isCurrentUser ? (
|
<Button
|
||||||
|
variant="outline"
|
||||||
|
size="sm"
|
||||||
|
onClick={() => {
|
||||||
|
setEditTarget(user);
|
||||||
|
setEditName(user.name);
|
||||||
|
setEditError(null);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<Pencil className="h-4 w-4 mr-2" />
|
||||||
|
Edit Role
|
||||||
|
</Button>
|
||||||
|
{isActive ? (
|
||||||
<Button
|
<Button
|
||||||
variant="destructive"
|
variant="destructive"
|
||||||
size="sm"
|
size="sm"
|
||||||
onClick={(event) => {
|
onClick={() => {
|
||||||
event.stopPropagation();
|
|
||||||
setDeactivateTarget(user);
|
setDeactivateTarget(user);
|
||||||
}}
|
}}
|
||||||
>
|
>
|
||||||
@@ -584,151 +433,10 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
})}
|
})}
|
||||||
|
|
||||||
{meta && meta.totalPages > 1 ? (
|
|
||||||
<div className="flex items-center justify-between pt-3 mt-1 border-t">
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
Page {page} of {meta.totalPages}
|
|
||||||
</p>
|
|
||||||
<div className="flex gap-2">
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
size="sm"
|
|
||||||
disabled={page === 1}
|
|
||||||
onClick={() => {
|
|
||||||
setPage((previousPage) => Math.max(1, previousPage - 1));
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
Previous
|
|
||||||
</Button>
|
|
||||||
<Button
|
|
||||||
variant="outline"
|
|
||||||
size="sm"
|
|
||||||
disabled={page >= meta.totalPages}
|
|
||||||
onClick={() => {
|
|
||||||
setPage((previousPage) => Math.min(meta.totalPages, previousPage + 1));
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
Next
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
) : null}
|
|
||||||
</CardContent>
|
</CardContent>
|
||||||
</Card>
|
</Card>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
<Dialog
|
|
||||||
open={detailTarget !== null}
|
|
||||||
onOpenChange={(open) => {
|
|
||||||
if (!open && !isSavingDetails) {
|
|
||||||
resetUserDetails();
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<DialogContent>
|
|
||||||
<DialogHeader>
|
|
||||||
<DialogTitle>User Details</DialogTitle>
|
|
||||||
<DialogDescription>
|
|
||||||
Edit profile details for {detailTarget?.email ?? "selected user"}.
|
|
||||||
</DialogDescription>
|
|
||||||
</DialogHeader>
|
|
||||||
|
|
||||||
<form
|
|
||||||
onSubmit={(event) => {
|
|
||||||
void handleDetailSubmit(event);
|
|
||||||
}}
|
|
||||||
className="space-y-4"
|
|
||||||
>
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="detail-name">Name</Label>
|
|
||||||
<Input
|
|
||||||
id="detail-name"
|
|
||||||
value={detailForm.name}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setDetailForm((prev) => ({ ...prev, name: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="Full name"
|
|
||||||
maxLength={255}
|
|
||||||
disabled={isSavingDetails}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="detail-email">Email</Label>
|
|
||||||
<Input
|
|
||||||
id="detail-email"
|
|
||||||
type="email"
|
|
||||||
value={detailForm.email}
|
|
||||||
onChange={(event: ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setDetailForm((prev) => ({ ...prev, email: event.target.value }));
|
|
||||||
}}
|
|
||||||
placeholder="user@example.com"
|
|
||||||
maxLength={255}
|
|
||||||
disabled={isSavingDetails}
|
|
||||||
required
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="detail-role">Role</Label>
|
|
||||||
<Select
|
|
||||||
value={detailForm.role}
|
|
||||||
disabled={detailForm.workspaceId === null || isSavingDetails}
|
|
||||||
onValueChange={(value) => {
|
|
||||||
setDetailForm((prev) => ({ ...prev, role: value as WorkspaceMemberRole }));
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<SelectTrigger id="detail-role">
|
|
||||||
<SelectValue placeholder="Select role" />
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent>
|
|
||||||
{Object.values(WorkspaceMemberRole).map((role) => (
|
|
||||||
<SelectItem key={role} value={role}>
|
|
||||||
{toRoleLabel(role)}
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
{detailForm.workspaceName ? (
|
|
||||||
<p className="text-xs text-muted-foreground">
|
|
||||||
Role updates apply to: {detailForm.workspaceName}
|
|
||||||
</p>
|
|
||||||
) : (
|
|
||||||
<p className="text-xs text-muted-foreground">
|
|
||||||
This user has no workspace membership. Role cannot be updated.
|
|
||||||
</p>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{detailError !== null ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{detailError}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<DialogFooter>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="outline"
|
|
||||||
onClick={() => {
|
|
||||||
if (!isSavingDetails) {
|
|
||||||
resetUserDetails();
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
disabled={isSavingDetails}
|
|
||||||
>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
<Button type="submit" disabled={isSavingDetails}>
|
|
||||||
{isSavingDetails ? "Saving..." : "Save Changes"}
|
|
||||||
</Button>
|
|
||||||
</DialogFooter>
|
|
||||||
</form>
|
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
|
||||||
|
|
||||||
<AlertDialog
|
<AlertDialog
|
||||||
open={deactivateTarget !== null}
|
open={deactivateTarget !== null}
|
||||||
onOpenChange={(open) => {
|
onOpenChange={(open) => {
|
||||||
@@ -760,4 +468,55 @@ export default function UsersSettingsPage(): ReactElement {
|
|||||||
</AlertDialog>
|
</AlertDialog>
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
|
|
||||||
|
<Dialog
|
||||||
|
open={editTarget !== null}
|
||||||
|
onOpenChange={(open) => {
|
||||||
|
if (!open && !isEditing) {
|
||||||
|
setEditTarget(null);
|
||||||
|
setEditError(null);
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<DialogContent>
|
||||||
|
<DialogHeader>
|
||||||
|
<DialogTitle>Edit User Role</DialogTitle>
|
||||||
|
<DialogDescription>Change role for {editTarget?.email ?? "user"}.</DialogDescription>
|
||||||
|
</DialogHeader>
|
||||||
|
<div className="space-y-4 py-2">
|
||||||
|
{editError !== null ? <p className="text-sm text-destructive">{editError}</p> : null}
|
||||||
|
<div className="space-y-2">
|
||||||
|
<Label htmlFor="edit-name">Display Name</Label>
|
||||||
|
<Input
|
||||||
|
id="edit-name"
|
||||||
|
value={editName}
|
||||||
|
onChange={(e: ChangeEvent<HTMLInputElement>) => {
|
||||||
|
setEditName(e.target.value);
|
||||||
|
}}
|
||||||
|
placeholder="Full name"
|
||||||
|
disabled={isEditing}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<DialogFooter>
|
||||||
|
<Button
|
||||||
|
variant="outline"
|
||||||
|
onClick={() => {
|
||||||
|
setEditTarget(null);
|
||||||
|
}}
|
||||||
|
disabled={isEditing}
|
||||||
|
>
|
||||||
|
Cancel
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
onClick={() => {
|
||||||
|
void handleEditSubmit();
|
||||||
|
}}
|
||||||
|
disabled={isEditing}
|
||||||
|
>
|
||||||
|
{isEditing ? "Saving..." : "Save"}
|
||||||
|
</Button>
|
||||||
|
</DialogFooter>
|
||||||
|
</DialogContent>
|
||||||
|
</Dialog>;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,12 +1,12 @@
|
|||||||
import type { UserWorkspace, WorkspaceMemberEntry } from "@/lib/api/workspaces";
|
import type { UserWorkspace } from "@/lib/api/workspaces";
|
||||||
import type { ReactElement, ReactNode } from "react";
|
import type { ReactElement, ReactNode } from "react";
|
||||||
|
|
||||||
import { WorkspaceMemberRole } from "@mosaic/shared";
|
import { WorkspaceMemberRole } from "@mosaic/shared";
|
||||||
import { render, screen, waitFor } from "@testing-library/react";
|
import { render, screen, waitFor } from "@testing-library/react";
|
||||||
import userEvent from "@testing-library/user-event";
|
import userEvent from "@testing-library/user-event";
|
||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||||
|
import { createWorkspace, fetchUserWorkspaces } from "@/lib/api/workspaces";
|
||||||
|
|
||||||
import { createWorkspace, fetchUserWorkspaces, fetchWorkspaceMembers } from "@/lib/api/workspaces";
|
|
||||||
import WorkspacesPage from "./page";
|
import WorkspacesPage from "./page";
|
||||||
|
|
||||||
vi.mock("next/link", () => ({
|
vi.mock("next/link", () => ({
|
||||||
@@ -21,23 +21,33 @@ vi.mock("next/link", () => ({
|
|||||||
},
|
},
|
||||||
}));
|
}));
|
||||||
|
|
||||||
|
vi.mock("@/components/workspace/WorkspaceCard", () => ({
|
||||||
|
WorkspaceCard: function WorkspaceCardMock({
|
||||||
|
workspace,
|
||||||
|
userRole,
|
||||||
|
memberCount,
|
||||||
|
}: {
|
||||||
|
workspace: { name: string };
|
||||||
|
userRole: WorkspaceMemberRole;
|
||||||
|
memberCount: number;
|
||||||
|
}): ReactElement {
|
||||||
|
return (
|
||||||
|
<div data-testid="workspace-card">
|
||||||
|
{workspace.name} | {userRole} | {String(memberCount)}
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
},
|
||||||
|
}));
|
||||||
|
|
||||||
vi.mock("@/lib/api/workspaces", () => ({
|
vi.mock("@/lib/api/workspaces", () => ({
|
||||||
fetchUserWorkspaces: vi.fn(),
|
fetchUserWorkspaces: vi.fn(),
|
||||||
createWorkspace: vi.fn(),
|
createWorkspace: vi.fn(),
|
||||||
fetchWorkspaceMembers: vi.fn(),
|
|
||||||
addWorkspaceMember: vi.fn(),
|
|
||||||
removeWorkspaceMember: vi.fn(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock("@/lib/api/admin", () => ({
|
|
||||||
fetchAdminUsers: vi.fn(),
|
|
||||||
}));
|
}));
|
||||||
|
|
||||||
const fetchUserWorkspacesMock = vi.mocked(fetchUserWorkspaces);
|
const fetchUserWorkspacesMock = vi.mocked(fetchUserWorkspaces);
|
||||||
const createWorkspaceMock = vi.mocked(createWorkspace);
|
const createWorkspaceMock = vi.mocked(createWorkspace);
|
||||||
const fetchWorkspaceMembersMock = vi.mocked(fetchWorkspaceMembers);
|
|
||||||
|
|
||||||
const workspaceA: UserWorkspace = {
|
const baseWorkspace: UserWorkspace = {
|
||||||
id: "workspace-1",
|
id: "workspace-1",
|
||||||
name: "Personal Workspace",
|
name: "Personal Workspace",
|
||||||
ownerId: "owner-1",
|
ownerId: "owner-1",
|
||||||
@@ -45,93 +55,45 @@ const workspaceA: UserWorkspace = {
|
|||||||
createdAt: "2026-01-01T00:00:00.000Z",
|
createdAt: "2026-01-01T00:00:00.000Z",
|
||||||
};
|
};
|
||||||
|
|
||||||
const workspaceB: UserWorkspace = {
|
|
||||||
id: "workspace-2",
|
|
||||||
name: "Client Workspace",
|
|
||||||
ownerId: "owner-2",
|
|
||||||
role: WorkspaceMemberRole.ADMIN,
|
|
||||||
createdAt: "2026-01-02T00:00:00.000Z",
|
|
||||||
};
|
|
||||||
|
|
||||||
const membersA: WorkspaceMemberEntry[] = [
|
|
||||||
{
|
|
||||||
workspaceId: "workspace-1",
|
|
||||||
userId: "user-a",
|
|
||||||
role: WorkspaceMemberRole.OWNER,
|
|
||||||
joinedAt: "2026-01-03T00:00:00.000Z",
|
|
||||||
user: {
|
|
||||||
id: "user-a",
|
|
||||||
email: "alice@example.com",
|
|
||||||
name: "Alice",
|
|
||||||
image: null,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
const membersB: WorkspaceMemberEntry[] = [
|
|
||||||
{
|
|
||||||
workspaceId: "workspace-2",
|
|
||||||
userId: "user-b",
|
|
||||||
role: WorkspaceMemberRole.MEMBER,
|
|
||||||
joinedAt: "2026-01-04T00:00:00.000Z",
|
|
||||||
user: {
|
|
||||||
id: "user-b",
|
|
||||||
email: "bob@example.com",
|
|
||||||
name: "Bob",
|
|
||||||
image: null,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
];
|
|
||||||
|
|
||||||
describe("WorkspacesPage", () => {
|
describe("WorkspacesPage", () => {
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
vi.clearAllMocks();
|
vi.clearAllMocks();
|
||||||
});
|
});
|
||||||
|
|
||||||
it("loads workspaces and fetches members for the first workspace", async () => {
|
it("loads and renders user workspaces from the API", async () => {
|
||||||
fetchUserWorkspacesMock.mockResolvedValue([workspaceA, workspaceB]);
|
fetchUserWorkspacesMock.mockResolvedValue([baseWorkspace]);
|
||||||
fetchWorkspaceMembersMock.mockResolvedValue(membersA);
|
|
||||||
|
|
||||||
render(<WorkspacesPage />);
|
render(<WorkspacesPage />);
|
||||||
|
|
||||||
expect(await screen.findByText("Your Workspaces (2)")).toBeInTheDocument();
|
expect(screen.getByText("Loading workspaces...")).toBeInTheDocument();
|
||||||
expect(await screen.findByText("Personal Workspace Members")).toBeInTheDocument();
|
|
||||||
|
|
||||||
await waitFor(() => {
|
expect(await screen.findByText("Your Workspaces (1)")).toBeInTheDocument();
|
||||||
expect(fetchWorkspaceMembersMock).toHaveBeenCalledWith("workspace-1");
|
expect(screen.getByTestId("workspace-card")).toHaveTextContent("Personal Workspace");
|
||||||
});
|
expect(fetchUserWorkspacesMock).toHaveBeenCalledTimes(1);
|
||||||
|
|
||||||
expect(screen.getByText("alice@example.com")).toBeInTheDocument();
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it("switches selected workspace and reloads member list", async () => {
|
it("shows fetch errors in the UI", async () => {
|
||||||
fetchUserWorkspacesMock.mockResolvedValue([workspaceA, workspaceB]);
|
fetchUserWorkspacesMock.mockRejectedValue(new Error("Unable to load workspaces"));
|
||||||
fetchWorkspaceMembersMock.mockResolvedValueOnce(membersA).mockResolvedValueOnce(membersB);
|
|
||||||
|
|
||||||
const user = userEvent.setup();
|
|
||||||
render(<WorkspacesPage />);
|
render(<WorkspacesPage />);
|
||||||
|
|
||||||
expect(await screen.findByText("Personal Workspace Members")).toBeInTheDocument();
|
expect(await screen.findByText("Unable to load workspaces")).toBeInTheDocument();
|
||||||
|
|
||||||
await user.click(screen.getByRole("button", { name: /client workspace/i }));
|
|
||||||
|
|
||||||
await waitFor(() => {
|
|
||||||
expect(fetchWorkspaceMembersMock).toHaveBeenLastCalledWith("workspace-2");
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(await screen.findByText("Client Workspace Members")).toBeInTheDocument();
|
|
||||||
expect(screen.getByText("bob@example.com")).toBeInTheDocument();
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it("creates a workspace and refreshes the list", async () => {
|
it("creates a workspace and refreshes the list", async () => {
|
||||||
fetchUserWorkspacesMock
|
fetchUserWorkspacesMock.mockResolvedValueOnce([baseWorkspace]).mockResolvedValueOnce([
|
||||||
.mockResolvedValueOnce([workspaceA])
|
baseWorkspace,
|
||||||
.mockResolvedValueOnce([workspaceA, workspaceB]);
|
{
|
||||||
fetchWorkspaceMembersMock.mockResolvedValue(membersA);
|
...baseWorkspace,
|
||||||
|
id: "workspace-2",
|
||||||
|
name: "New Workspace",
|
||||||
|
role: WorkspaceMemberRole.MEMBER,
|
||||||
|
},
|
||||||
|
]);
|
||||||
createWorkspaceMock.mockResolvedValue({
|
createWorkspaceMock.mockResolvedValue({
|
||||||
id: "workspace-2",
|
id: "workspace-2",
|
||||||
name: "Client Workspace",
|
name: "New Workspace",
|
||||||
ownerId: "owner-2",
|
ownerId: "owner-1",
|
||||||
settings: {},
|
settings: {},
|
||||||
createdAt: "2026-01-02T00:00:00.000Z",
|
createdAt: "2026-01-02T00:00:00.000Z",
|
||||||
updatedAt: "2026-01-02T00:00:00.000Z",
|
updatedAt: "2026-01-02T00:00:00.000Z",
|
||||||
@@ -143,17 +105,31 @@ describe("WorkspacesPage", () => {
|
|||||||
|
|
||||||
expect(await screen.findByText("Your Workspaces (1)")).toBeInTheDocument();
|
expect(await screen.findByText("Your Workspaces (1)")).toBeInTheDocument();
|
||||||
|
|
||||||
await user.type(screen.getByPlaceholderText("Enter workspace name..."), "Client Workspace");
|
await user.type(screen.getByPlaceholderText("Enter workspace name..."), "New Workspace");
|
||||||
await user.click(screen.getByRole("button", { name: "Create Workspace" }));
|
await user.click(screen.getByRole("button", { name: "Create Workspace" }));
|
||||||
|
|
||||||
await waitFor(() => {
|
await waitFor(() => {
|
||||||
expect(createWorkspaceMock).toHaveBeenCalledWith({ name: "Client Workspace" });
|
expect(createWorkspaceMock).toHaveBeenCalledWith({ name: "New Workspace" });
|
||||||
});
|
});
|
||||||
|
|
||||||
await waitFor(() => {
|
await waitFor(() => {
|
||||||
expect(fetchUserWorkspacesMock).toHaveBeenCalledTimes(2);
|
expect(fetchUserWorkspacesMock).toHaveBeenCalledTimes(2);
|
||||||
});
|
});
|
||||||
|
|
||||||
expect(await screen.findByText("Your Workspaces (2)")).toBeInTheDocument();
|
expect(await screen.findByText("Your Workspaces (2)")).toBeInTheDocument();
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it("shows create errors in the UI", async () => {
|
||||||
|
fetchUserWorkspacesMock.mockResolvedValue([baseWorkspace]);
|
||||||
|
createWorkspaceMock.mockRejectedValue(new Error("Workspace creation failed"));
|
||||||
|
|
||||||
|
const user = userEvent.setup();
|
||||||
|
render(<WorkspacesPage />);
|
||||||
|
|
||||||
|
expect(await screen.findByText("Your Workspaces (1)")).toBeInTheDocument();
|
||||||
|
|
||||||
|
await user.type(screen.getByPlaceholderText("Enter workspace name..."), "Bad Workspace");
|
||||||
|
await user.click(screen.getByRole("button", { name: "Create Workspace" }));
|
||||||
|
|
||||||
|
expect(await screen.findByText("Workspace creation failed")).toBeInTheDocument();
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -2,51 +2,10 @@
|
|||||||
|
|
||||||
import type { ReactElement, SyntheticEvent } from "react";
|
import type { ReactElement, SyntheticEvent } from "react";
|
||||||
|
|
||||||
import { useCallback, useEffect, useMemo, useState } from "react";
|
import { useCallback, useEffect, useState } from "react";
|
||||||
|
import { WorkspaceCard } from "@/components/workspace/WorkspaceCard";
|
||||||
|
import { createWorkspace, fetchUserWorkspaces, type UserWorkspace } from "@/lib/api/workspaces";
|
||||||
import Link from "next/link";
|
import Link from "next/link";
|
||||||
import { UserPlus, UserX } from "lucide-react";
|
|
||||||
import { WorkspaceMemberRole } from "@mosaic/shared";
|
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
|
||||||
import {
|
|
||||||
Dialog,
|
|
||||||
DialogContent,
|
|
||||||
DialogDescription,
|
|
||||||
DialogFooter,
|
|
||||||
DialogHeader,
|
|
||||||
DialogTitle,
|
|
||||||
DialogTrigger,
|
|
||||||
} from "@/components/ui/dialog";
|
|
||||||
import {
|
|
||||||
AlertDialog,
|
|
||||||
AlertDialogAction,
|
|
||||||
AlertDialogCancel,
|
|
||||||
AlertDialogContent,
|
|
||||||
AlertDialogDescription,
|
|
||||||
AlertDialogFooter,
|
|
||||||
AlertDialogHeader,
|
|
||||||
AlertDialogTitle,
|
|
||||||
} from "@/components/ui/alert-dialog";
|
|
||||||
import { Input } from "@/components/ui/input";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import {
|
|
||||||
Select,
|
|
||||||
SelectContent,
|
|
||||||
SelectItem,
|
|
||||||
SelectTrigger,
|
|
||||||
SelectValue,
|
|
||||||
} from "@/components/ui/select";
|
|
||||||
import {
|
|
||||||
addWorkspaceMember,
|
|
||||||
createWorkspace,
|
|
||||||
fetchUserWorkspaces,
|
|
||||||
fetchWorkspaceMembers,
|
|
||||||
removeWorkspaceMember,
|
|
||||||
type UserWorkspace,
|
|
||||||
type WorkspaceMemberEntry,
|
|
||||||
} from "@/lib/api/workspaces";
|
|
||||||
import { fetchAdminUsers, type AdminUser } from "@/lib/api/admin";
|
|
||||||
|
|
||||||
function getErrorMessage(error: unknown, fallback: string): string {
|
function getErrorMessage(error: unknown, fallback: string): string {
|
||||||
if (error instanceof Error) {
|
if (error instanceof Error) {
|
||||||
@@ -56,53 +15,18 @@ function getErrorMessage(error: unknown, fallback: string): string {
|
|||||||
return fallback;
|
return fallback;
|
||||||
}
|
}
|
||||||
|
|
||||||
function toRoleLabel(role: WorkspaceMemberRole): string {
|
/**
|
||||||
return `${role.charAt(0)}${role.slice(1).toLowerCase()}`;
|
* Workspaces Page
|
||||||
}
|
* Fetches and creates workspaces through the real API.
|
||||||
|
*/
|
||||||
interface RemoveMemberTarget {
|
|
||||||
userId: string;
|
|
||||||
email: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
const ROLE_BADGE_CLASS: Record<WorkspaceMemberRole, string> = {
|
|
||||||
[WorkspaceMemberRole.OWNER]: "border-purple-200 bg-purple-50 text-purple-700",
|
|
||||||
[WorkspaceMemberRole.ADMIN]: "border-blue-200 bg-blue-50 text-blue-700",
|
|
||||||
[WorkspaceMemberRole.MEMBER]: "border-green-200 bg-green-50 text-green-700",
|
|
||||||
[WorkspaceMemberRole.GUEST]: "border-gray-200 bg-gray-50 text-gray-700",
|
|
||||||
};
|
|
||||||
|
|
||||||
export default function WorkspacesPage(): ReactElement {
|
export default function WorkspacesPage(): ReactElement {
|
||||||
const [workspaces, setWorkspaces] = useState<UserWorkspace[]>([]);
|
const [workspaces, setWorkspaces] = useState<UserWorkspace[]>([]);
|
||||||
const [selectedWorkspaceId, setSelectedWorkspaceId] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const [isLoading, setIsLoading] = useState(true);
|
const [isLoading, setIsLoading] = useState(true);
|
||||||
const [loadError, setLoadError] = useState<string | null>(null);
|
const [loadError, setLoadError] = useState<string | null>(null);
|
||||||
|
|
||||||
const [isCreating, setIsCreating] = useState(false);
|
const [isCreating, setIsCreating] = useState(false);
|
||||||
const [newWorkspaceName, setNewWorkspaceName] = useState("");
|
const [newWorkspaceName, setNewWorkspaceName] = useState("");
|
||||||
const [createError, setCreateError] = useState<string | null>(null);
|
const [createError, setCreateError] = useState<string | null>(null);
|
||||||
|
|
||||||
const [members, setMembers] = useState<WorkspaceMemberEntry[]>([]);
|
|
||||||
const [isMembersLoading, setIsMembersLoading] = useState(false);
|
|
||||||
const [membersError, setMembersError] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const [isAddMemberOpen, setIsAddMemberOpen] = useState(false);
|
|
||||||
const [isAddingMember, setIsAddingMember] = useState(false);
|
|
||||||
const [addMemberError, setAddMemberError] = useState<string | null>(null);
|
|
||||||
const [memberUserId, setMemberUserId] = useState<string>("");
|
|
||||||
const [memberRole, setMemberRole] = useState<WorkspaceMemberRole>(WorkspaceMemberRole.MEMBER);
|
|
||||||
const [availableUsers, setAvailableUsers] = useState<AdminUser[]>([]);
|
|
||||||
const [isLoadingUsers, setIsLoadingUsers] = useState(false);
|
|
||||||
|
|
||||||
const [removeTarget, setRemoveTarget] = useState<RemoveMemberTarget | null>(null);
|
|
||||||
const [isRemovingMember, setIsRemovingMember] = useState(false);
|
|
||||||
|
|
||||||
const selectedWorkspace = useMemo(
|
|
||||||
() => workspaces.find((workspace) => workspace.id === selectedWorkspaceId) ?? null,
|
|
||||||
[selectedWorkspaceId, workspaces]
|
|
||||||
);
|
|
||||||
|
|
||||||
const loadWorkspaces = useCallback(async (): Promise<void> => {
|
const loadWorkspaces = useCallback(async (): Promise<void> => {
|
||||||
setIsLoading(true);
|
setIsLoading(true);
|
||||||
|
|
||||||
@@ -110,57 +34,31 @@ export default function WorkspacesPage(): ReactElement {
|
|||||||
const data = await fetchUserWorkspaces();
|
const data = await fetchUserWorkspaces();
|
||||||
setWorkspaces(data);
|
setWorkspaces(data);
|
||||||
setLoadError(null);
|
setLoadError(null);
|
||||||
setSelectedWorkspaceId((current) => {
|
|
||||||
if (current && data.some((workspace) => workspace.id === current)) {
|
|
||||||
return current;
|
|
||||||
}
|
|
||||||
|
|
||||||
return data[0]?.id ?? null;
|
|
||||||
});
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
setLoadError(getErrorMessage(error, "Failed to load workspaces"));
|
setLoadError(getErrorMessage(error, "Failed to load workspaces"));
|
||||||
setSelectedWorkspaceId(null);
|
|
||||||
} finally {
|
} finally {
|
||||||
setIsLoading(false);
|
setIsLoading(false);
|
||||||
}
|
}
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
const loadMembers = useCallback(async (workspaceId: string): Promise<void> => {
|
|
||||||
setIsMembersLoading(true);
|
|
||||||
setMembersError(null);
|
|
||||||
|
|
||||||
try {
|
|
||||||
const data = await fetchWorkspaceMembers(workspaceId);
|
|
||||||
setMembers(data);
|
|
||||||
} catch (error) {
|
|
||||||
setMembersError(getErrorMessage(error, "Failed to load workspace members"));
|
|
||||||
setMembers([]);
|
|
||||||
} finally {
|
|
||||||
setIsMembersLoading(false);
|
|
||||||
}
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
void loadWorkspaces();
|
void loadWorkspaces();
|
||||||
}, [loadWorkspaces]);
|
}, [loadWorkspaces]);
|
||||||
|
|
||||||
useEffect(() => {
|
const workspacesWithRoles = workspaces.map((workspace) => ({
|
||||||
if (!selectedWorkspaceId) {
|
...workspace,
|
||||||
setMembers([]);
|
settings: {},
|
||||||
setMembersError(null);
|
createdAt: new Date(workspace.createdAt),
|
||||||
return;
|
updatedAt: new Date(workspace.createdAt),
|
||||||
}
|
userRole: workspace.role,
|
||||||
|
memberCount: 1,
|
||||||
|
}));
|
||||||
|
|
||||||
void loadMembers(selectedWorkspaceId);
|
const handleCreateWorkspace = async (e: SyntheticEvent<HTMLFormElement>): Promise<void> => {
|
||||||
}, [loadMembers, selectedWorkspaceId]);
|
e.preventDefault();
|
||||||
|
|
||||||
const handleCreateWorkspace = async (event: SyntheticEvent<HTMLFormElement>): Promise<void> => {
|
|
||||||
event.preventDefault();
|
|
||||||
|
|
||||||
const workspaceName = newWorkspaceName.trim();
|
const workspaceName = newWorkspaceName.trim();
|
||||||
if (!workspaceName) {
|
if (!workspaceName) return;
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsCreating(true);
|
setIsCreating(true);
|
||||||
setCreateError(null);
|
setCreateError(null);
|
||||||
@@ -176,394 +74,91 @@ export default function WorkspacesPage(): ReactElement {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
const eligibleUsers = useMemo(() => {
|
|
||||||
const memberIds = new Set(members.map((member) => member.userId));
|
|
||||||
return availableUsers.filter((user) => !memberIds.has(user.id));
|
|
||||||
}, [availableUsers, members]);
|
|
||||||
|
|
||||||
const loadAvailableUsers = useCallback(async (): Promise<void> => {
|
|
||||||
setIsLoadingUsers(true);
|
|
||||||
|
|
||||||
try {
|
|
||||||
const response = await fetchAdminUsers(1, 200);
|
|
||||||
const activeUsers = response.data.filter((user) => user.deactivatedAt === null);
|
|
||||||
setAvailableUsers(activeUsers);
|
|
||||||
|
|
||||||
if (memberUserId && activeUsers.some((user) => user.id === memberUserId)) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const memberIds = new Set(members.map((member) => member.userId));
|
|
||||||
const firstEligible = activeUsers.find((user) => !memberIds.has(user.id));
|
|
||||||
setMemberUserId(firstEligible?.id ?? "");
|
|
||||||
} catch (error) {
|
|
||||||
setAddMemberError(getErrorMessage(error, "Failed to load users for member assignment"));
|
|
||||||
setAvailableUsers([]);
|
|
||||||
setMemberUserId("");
|
|
||||||
} finally {
|
|
||||||
setIsLoadingUsers(false);
|
|
||||||
}
|
|
||||||
}, [memberUserId, members]);
|
|
||||||
|
|
||||||
const openAddMemberDialog = async (): Promise<void> => {
|
|
||||||
setAddMemberError(null);
|
|
||||||
setMemberRole(WorkspaceMemberRole.MEMBER);
|
|
||||||
setIsAddMemberOpen(true);
|
|
||||||
await loadAvailableUsers();
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleAddMember = async (event: SyntheticEvent<HTMLFormElement>): Promise<void> => {
|
|
||||||
event.preventDefault();
|
|
||||||
|
|
||||||
if (!selectedWorkspaceId) {
|
|
||||||
setAddMemberError("Select a workspace before adding members.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!memberUserId) {
|
|
||||||
setAddMemberError("Select a user to add.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsAddingMember(true);
|
|
||||||
setAddMemberError(null);
|
|
||||||
|
|
||||||
try {
|
|
||||||
await addWorkspaceMember(selectedWorkspaceId, {
|
|
||||||
userId: memberUserId,
|
|
||||||
role: memberRole,
|
|
||||||
});
|
|
||||||
setIsAddMemberOpen(false);
|
|
||||||
await loadMembers(selectedWorkspaceId);
|
|
||||||
} catch (error) {
|
|
||||||
setAddMemberError(getErrorMessage(error, "Failed to add member"));
|
|
||||||
} finally {
|
|
||||||
setIsAddingMember(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleRemoveMember = async (): Promise<void> => {
|
|
||||||
if (!selectedWorkspaceId || !removeTarget) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsRemovingMember(true);
|
|
||||||
|
|
||||||
try {
|
|
||||||
await removeWorkspaceMember(selectedWorkspaceId, removeTarget.userId);
|
|
||||||
setRemoveTarget(null);
|
|
||||||
await loadMembers(selectedWorkspaceId);
|
|
||||||
} catch (error) {
|
|
||||||
setMembersError(getErrorMessage(error, "Failed to remove member"));
|
|
||||||
} finally {
|
|
||||||
setIsRemovingMember(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<main className="max-w-6xl mx-auto p-6 space-y-6">
|
<main className="container mx-auto px-4 py-8 max-w-5xl">
|
||||||
<div className="flex items-start justify-between gap-4">
|
<div className="mb-8">
|
||||||
<div>
|
<div className="flex items-center justify-between mb-2">
|
||||||
<h1 className="text-3xl font-bold">Workspaces</h1>
|
<h1 className="text-3xl font-bold text-gray-900">Workspaces</h1>
|
||||||
<p className="text-muted-foreground mt-1">Manage workspaces and workspace members</p>
|
<Link href="/settings" className="text-sm text-blue-600 hover:text-blue-700">
|
||||||
|
← Back to Settings
|
||||||
|
</Link>
|
||||||
</div>
|
</div>
|
||||||
<Link href="/settings" className="text-sm text-blue-600 hover:text-blue-700">
|
<p className="text-gray-600">Manage your workspaces and collaborate with your team</p>
|
||||||
← Back to Settings
|
|
||||||
</Link>
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<Card>
|
{/* Create New Workspace */}
|
||||||
<CardHeader>
|
<div className="bg-white rounded-lg shadow-sm border border-gray-200 p-6 mb-6">
|
||||||
<CardTitle>Create New Workspace</CardTitle>
|
<h2 className="text-lg font-semibold text-gray-900 mb-4">Create New Workspace</h2>
|
||||||
<CardDescription>Create a workspace for a new team or project.</CardDescription>
|
<form onSubmit={handleCreateWorkspace} className="flex gap-3">
|
||||||
</CardHeader>
|
<input
|
||||||
<CardContent>
|
type="text"
|
||||||
<form onSubmit={handleCreateWorkspace} className="flex gap-3">
|
value={newWorkspaceName}
|
||||||
<Input
|
onChange={(e) => {
|
||||||
type="text"
|
setNewWorkspaceName(e.target.value);
|
||||||
value={newWorkspaceName}
|
}}
|
||||||
onChange={(event) => {
|
placeholder="Enter workspace name..."
|
||||||
setNewWorkspaceName(event.target.value);
|
disabled={isCreating}
|
||||||
}}
|
className="flex-1 px-4 py-2 border border-gray-300 rounded-lg focus:ring-2 focus:ring-blue-500 focus:border-transparent disabled:bg-gray-100"
|
||||||
placeholder="Enter workspace name..."
|
/>
|
||||||
disabled={isCreating}
|
<button
|
||||||
/>
|
type="submit"
|
||||||
<Button type="submit" disabled={isCreating || !newWorkspaceName.trim()}>
|
disabled={isCreating || !newWorkspaceName.trim()}
|
||||||
{isCreating ? "Creating..." : "Create Workspace"}
|
className="px-6 py-2 bg-blue-600 text-white rounded-lg hover:bg-blue-700 disabled:opacity-50 disabled:cursor-not-allowed font-medium"
|
||||||
</Button>
|
>
|
||||||
</form>
|
{isCreating ? "Creating..." : "Create Workspace"}
|
||||||
{createError !== null ? (
|
</button>
|
||||||
<p className="mt-3 text-sm text-destructive" role="alert">
|
</form>
|
||||||
{createError}
|
{createError !== null && (
|
||||||
</p>
|
<div className="mt-3 rounded-md border border-red-200 bg-red-50 px-3 py-2 text-sm text-red-700">
|
||||||
) : null}
|
{createError}
|
||||||
</CardContent>
|
</div>
|
||||||
</Card>
|
)}
|
||||||
|
|
||||||
{loadError !== null ? (
|
|
||||||
<Card>
|
|
||||||
<CardContent className="py-4">
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{loadError}
|
|
||||||
</p>
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<div className="grid grid-cols-1 lg:grid-cols-5 gap-6">
|
|
||||||
<Card className="lg:col-span-2">
|
|
||||||
<CardHeader>
|
|
||||||
<CardTitle>Your Workspaces ({isLoading ? "..." : workspaces.length})</CardTitle>
|
|
||||||
<CardDescription>Click a workspace to manage its members.</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent className="space-y-3">
|
|
||||||
{isLoading ? (
|
|
||||||
<p className="text-sm text-muted-foreground">Loading workspaces...</p>
|
|
||||||
) : workspaces.length === 0 ? (
|
|
||||||
<p className="text-sm text-muted-foreground">
|
|
||||||
No workspaces yet. Create one to begin.
|
|
||||||
</p>
|
|
||||||
) : (
|
|
||||||
workspaces.map((workspace) => {
|
|
||||||
const isSelected = selectedWorkspaceId === workspace.id;
|
|
||||||
|
|
||||||
return (
|
|
||||||
<button
|
|
||||||
key={workspace.id}
|
|
||||||
type="button"
|
|
||||||
onClick={() => {
|
|
||||||
setSelectedWorkspaceId(workspace.id);
|
|
||||||
}}
|
|
||||||
className={`w-full rounded-lg border p-4 text-left transition-colors ${
|
|
||||||
isSelected ? "border-primary bg-muted/40" : "border-border hover:bg-muted/20"
|
|
||||||
}`}
|
|
||||||
>
|
|
||||||
<div className="flex items-start justify-between gap-3">
|
|
||||||
<div className="min-w-0">
|
|
||||||
<p className="font-semibold truncate">{workspace.name}</p>
|
|
||||||
<p className="text-xs text-muted-foreground mt-1">
|
|
||||||
Created {new Date(workspace.createdAt).toLocaleDateString()}
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
<Badge variant="outline">{toRoleLabel(workspace.role)}</Badge>
|
|
||||||
</div>
|
|
||||||
</button>
|
|
||||||
);
|
|
||||||
})
|
|
||||||
)}
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
|
|
||||||
<Card className="lg:col-span-3">
|
|
||||||
<CardHeader>
|
|
||||||
<div className="flex items-start justify-between gap-3">
|
|
||||||
<div>
|
|
||||||
<CardTitle>
|
|
||||||
{selectedWorkspace ? `${selectedWorkspace.name} Members` : "Workspace Members"}
|
|
||||||
</CardTitle>
|
|
||||||
<CardDescription>
|
|
||||||
{selectedWorkspace
|
|
||||||
? "Manage member roles and access for this workspace."
|
|
||||||
: "Select a workspace to view its members."}
|
|
||||||
</CardDescription>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Dialog
|
|
||||||
open={isAddMemberOpen}
|
|
||||||
onOpenChange={(open) => {
|
|
||||||
if (!open && !isAddingMember) {
|
|
||||||
setIsAddMemberOpen(false);
|
|
||||||
setAddMemberError(null);
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<DialogTrigger asChild>
|
|
||||||
<Button
|
|
||||||
onClick={() => {
|
|
||||||
void openAddMemberDialog();
|
|
||||||
}}
|
|
||||||
disabled={!selectedWorkspace}
|
|
||||||
>
|
|
||||||
<UserPlus className="h-4 w-4 mr-2" />
|
|
||||||
Add Member
|
|
||||||
</Button>
|
|
||||||
</DialogTrigger>
|
|
||||||
<DialogContent>
|
|
||||||
<DialogHeader>
|
|
||||||
<DialogTitle>Add Workspace Member</DialogTitle>
|
|
||||||
<DialogDescription>
|
|
||||||
Add an existing user to {selectedWorkspace?.name ?? "this workspace"}.
|
|
||||||
</DialogDescription>
|
|
||||||
</DialogHeader>
|
|
||||||
|
|
||||||
<form
|
|
||||||
onSubmit={(event) => {
|
|
||||||
void handleAddMember(event);
|
|
||||||
}}
|
|
||||||
className="space-y-4"
|
|
||||||
>
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="member-user">User</Label>
|
|
||||||
<Select
|
|
||||||
value={memberUserId}
|
|
||||||
onValueChange={(value) => {
|
|
||||||
setMemberUserId(value);
|
|
||||||
}}
|
|
||||||
disabled={isLoadingUsers || eligibleUsers.length === 0 || isAddingMember}
|
|
||||||
>
|
|
||||||
<SelectTrigger id="member-user">
|
|
||||||
<SelectValue
|
|
||||||
placeholder={
|
|
||||||
isLoadingUsers
|
|
||||||
? "Loading users..."
|
|
||||||
: eligibleUsers.length === 0
|
|
||||||
? "No eligible users"
|
|
||||||
: "Select a user"
|
|
||||||
}
|
|
||||||
/>
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent>
|
|
||||||
{eligibleUsers.map((user) => (
|
|
||||||
<SelectItem key={user.id} value={user.id}>
|
|
||||||
{user.name} ({user.email})
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="space-y-2">
|
|
||||||
<Label htmlFor="member-role">Role</Label>
|
|
||||||
<Select
|
|
||||||
value={memberRole}
|
|
||||||
onValueChange={(value) => {
|
|
||||||
setMemberRole(value as WorkspaceMemberRole);
|
|
||||||
}}
|
|
||||||
disabled={isAddingMember}
|
|
||||||
>
|
|
||||||
<SelectTrigger id="member-role">
|
|
||||||
<SelectValue placeholder="Select role" />
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent>
|
|
||||||
{Object.values(WorkspaceMemberRole)
|
|
||||||
.filter((role) => role !== WorkspaceMemberRole.OWNER)
|
|
||||||
.map((role) => (
|
|
||||||
<SelectItem key={role} value={role}>
|
|
||||||
{toRoleLabel(role)}
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{addMemberError !== null ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{addMemberError}
|
|
||||||
</p>
|
|
||||||
) : null}
|
|
||||||
|
|
||||||
<DialogFooter>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="outline"
|
|
||||||
onClick={() => {
|
|
||||||
if (!isAddingMember) {
|
|
||||||
setIsAddMemberOpen(false);
|
|
||||||
setAddMemberError(null);
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
disabled={isAddingMember}
|
|
||||||
>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
<Button type="submit" disabled={isAddingMember || !memberUserId}>
|
|
||||||
{isAddingMember ? "Adding..." : "Add Member"}
|
|
||||||
</Button>
|
|
||||||
</DialogFooter>
|
|
||||||
</form>
|
|
||||||
</DialogContent>
|
|
||||||
</Dialog>
|
|
||||||
</div>
|
|
||||||
</CardHeader>
|
|
||||||
|
|
||||||
<CardContent className="space-y-3">
|
|
||||||
{selectedWorkspace === null ? (
|
|
||||||
<p className="text-sm text-muted-foreground">Select a workspace to view members.</p>
|
|
||||||
) : membersError !== null ? (
|
|
||||||
<p className="text-sm text-destructive" role="alert">
|
|
||||||
{membersError}
|
|
||||||
</p>
|
|
||||||
) : isMembersLoading ? (
|
|
||||||
<p className="text-sm text-muted-foreground">Loading members...</p>
|
|
||||||
) : members.length === 0 ? (
|
|
||||||
<p className="text-sm text-muted-foreground">No members found for this workspace.</p>
|
|
||||||
) : (
|
|
||||||
members.map((member) => (
|
|
||||||
<div
|
|
||||||
key={member.userId}
|
|
||||||
className="rounded-md border p-4 flex flex-col gap-3 md:flex-row md:items-center md:justify-between"
|
|
||||||
>
|
|
||||||
<div className="space-y-1 min-w-0">
|
|
||||||
<p className="font-semibold truncate">{member.user.name ?? "Unnamed User"}</p>
|
|
||||||
<p className="text-sm text-muted-foreground truncate">{member.user.email}</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="flex items-center gap-2 md:justify-end">
|
|
||||||
<Badge variant="outline" className={ROLE_BADGE_CLASS[member.role]}>
|
|
||||||
{toRoleLabel(member.role)}
|
|
||||||
</Badge>
|
|
||||||
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="destructive"
|
|
||||||
size="sm"
|
|
||||||
onClick={() => {
|
|
||||||
setRemoveTarget({
|
|
||||||
userId: member.userId,
|
|
||||||
email: member.user.email,
|
|
||||||
});
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<UserX className="h-4 w-4 mr-2" />
|
|
||||||
Remove
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
))
|
|
||||||
)}
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<AlertDialog
|
{/* Workspace List */}
|
||||||
open={removeTarget !== null}
|
<div className="space-y-4">
|
||||||
onOpenChange={(open) => {
|
<h2 className="text-xl font-semibold text-gray-900">
|
||||||
if (!open && !isRemovingMember) {
|
Your Workspaces ({isLoading ? "..." : workspacesWithRoles.length})
|
||||||
setRemoveTarget(null);
|
</h2>
|
||||||
}
|
{loadError !== null ? (
|
||||||
}}
|
<div className="rounded-md border border-red-200 bg-red-50 px-4 py-3 text-red-700">
|
||||||
>
|
{loadError}
|
||||||
<AlertDialogContent>
|
</div>
|
||||||
<AlertDialogHeader>
|
) : isLoading ? (
|
||||||
<AlertDialogTitle>Remove Workspace Member</AlertDialogTitle>
|
<div className="bg-white rounded-lg shadow-sm border border-gray-200 p-12 text-center text-gray-600">
|
||||||
<AlertDialogDescription>
|
Loading workspaces...
|
||||||
Remove {removeTarget?.email} from {selectedWorkspace?.name}? They will lose access to
|
</div>
|
||||||
this workspace.
|
) : workspacesWithRoles.length === 0 ? (
|
||||||
</AlertDialogDescription>
|
<div className="bg-white rounded-lg shadow-sm border border-gray-200 p-12 text-center">
|
||||||
</AlertDialogHeader>
|
<svg
|
||||||
<AlertDialogFooter>
|
className="mx-auto h-12 w-12 text-gray-400 mb-4"
|
||||||
<AlertDialogCancel disabled={isRemovingMember}>Cancel</AlertDialogCancel>
|
fill="none"
|
||||||
<AlertDialogAction
|
stroke="currentColor"
|
||||||
disabled={isRemovingMember}
|
viewBox="0 0 24 24"
|
||||||
onClick={() => {
|
|
||||||
void handleRemoveMember();
|
|
||||||
}}
|
|
||||||
>
|
>
|
||||||
{isRemovingMember ? "Removing..." : "Remove"}
|
<path
|
||||||
</AlertDialogAction>
|
strokeLinecap="round"
|
||||||
</AlertDialogFooter>
|
strokeLinejoin="round"
|
||||||
</AlertDialogContent>
|
strokeWidth={2}
|
||||||
</AlertDialog>
|
d="M19 21V5a2 2 0 00-2-2H7a2 2 0 00-2 2v16m14 0h2m-2 0h-5m-9 0H3m2 0h5M9 7h1m-1 4h1m4-4h1m-1 4h1m-5 10v-5a1 1 0 011-1h2a1 1 0 011 1v5m-4 0h4"
|
||||||
|
/>
|
||||||
|
</svg>
|
||||||
|
<h3 className="text-lg font-medium text-gray-900 mb-2">No workspaces yet</h3>
|
||||||
|
<p className="text-gray-600">Create your first workspace to get started</p>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="grid grid-cols-1 md:grid-cols-2 gap-4">
|
||||||
|
{workspacesWithRoles.map((workspace) => (
|
||||||
|
<WorkspaceCard
|
||||||
|
key={workspace.id}
|
||||||
|
workspace={workspace}
|
||||||
|
userRole={workspace.userRole}
|
||||||
|
memberCount={workspace.memberCount}
|
||||||
|
/>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
</main>
|
</main>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,9 +0,0 @@
|
|||||||
import type { ReactNode } from "react";
|
|
||||||
|
|
||||||
export default function OnboardingLayout({ children }: { children: ReactNode }): React.JSX.Element {
|
|
||||||
return (
|
|
||||||
<main className="flex min-h-screen items-center justify-center bg-gradient-to-b from-slate-50 to-white p-4 sm:p-6">
|
|
||||||
<div className="w-full max-w-3xl">{children}</div>
|
|
||||||
</main>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -1,36 +0,0 @@
|
|||||||
import { redirect } from "next/navigation";
|
|
||||||
import { OnboardingWizard } from "@/components/onboarding/OnboardingWizard";
|
|
||||||
import { API_BASE_URL } from "@/lib/config";
|
|
||||||
|
|
||||||
export const dynamic = "force-dynamic";
|
|
||||||
|
|
||||||
interface OnboardingStatusResponse {
|
|
||||||
completed: boolean;
|
|
||||||
}
|
|
||||||
|
|
||||||
async function getOnboardingStatus(): Promise<OnboardingStatusResponse> {
|
|
||||||
try {
|
|
||||||
const response = await fetch(`${API_BASE_URL}/api/onboarding/status`, {
|
|
||||||
method: "GET",
|
|
||||||
cache: "no-store",
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!response.ok) {
|
|
||||||
return { completed: false };
|
|
||||||
}
|
|
||||||
|
|
||||||
return (await response.json()) as OnboardingStatusResponse;
|
|
||||||
} catch {
|
|
||||||
return { completed: false };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export default async function OnboardingPage(): Promise<React.JSX.Element> {
|
|
||||||
const status = await getOnboardingStatus();
|
|
||||||
|
|
||||||
if (status.completed) {
|
|
||||||
redirect("/");
|
|
||||||
}
|
|
||||||
|
|
||||||
return <OnboardingWizard />;
|
|
||||||
}
|
|
||||||
@@ -1,106 +0,0 @@
|
|||||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
||||||
import { render, screen, waitFor } from "@testing-library/react";
|
|
||||||
import userEvent from "@testing-library/user-event";
|
|
||||||
import { OnboardingWizard } from "./OnboardingWizard";
|
|
||||||
|
|
||||||
const mockPush = vi.fn();
|
|
||||||
const mockGetStatus = vi.fn();
|
|
||||||
const mockCreateBreakglass = vi.fn();
|
|
||||||
const mockConfigureOidc = vi.fn();
|
|
||||||
const mockTestProvider = vi.fn();
|
|
||||||
const mockAddProvider = vi.fn();
|
|
||||||
const mockCompleteOnboarding = vi.fn();
|
|
||||||
|
|
||||||
vi.mock("next/navigation", () => ({
|
|
||||||
useRouter: (): { push: typeof mockPush } => ({
|
|
||||||
push: mockPush,
|
|
||||||
}),
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock("@/lib/api/onboarding", () => ({
|
|
||||||
fetchOnboardingStatus: (): ReturnType<typeof mockGetStatus> => mockGetStatus(),
|
|
||||||
createBreakglassAdmin: (...args: unknown[]): ReturnType<typeof mockCreateBreakglass> =>
|
|
||||||
mockCreateBreakglass(...args),
|
|
||||||
configureOidcProvider: (...args: unknown[]): ReturnType<typeof mockConfigureOidc> =>
|
|
||||||
mockConfigureOidc(...args),
|
|
||||||
testOnboardingProvider: (...args: unknown[]): ReturnType<typeof mockTestProvider> =>
|
|
||||||
mockTestProvider(...args),
|
|
||||||
addOnboardingProvider: (...args: unknown[]): ReturnType<typeof mockAddProvider> =>
|
|
||||||
mockAddProvider(...args),
|
|
||||||
completeOnboarding: (): ReturnType<typeof mockCompleteOnboarding> => mockCompleteOnboarding(),
|
|
||||||
}));
|
|
||||||
|
|
||||||
describe("OnboardingWizard", () => {
|
|
||||||
beforeEach(() => {
|
|
||||||
mockPush.mockReset();
|
|
||||||
mockGetStatus.mockReset();
|
|
||||||
mockCreateBreakglass.mockReset();
|
|
||||||
mockConfigureOidc.mockReset();
|
|
||||||
mockTestProvider.mockReset();
|
|
||||||
mockAddProvider.mockReset();
|
|
||||||
mockCompleteOnboarding.mockReset();
|
|
||||||
|
|
||||||
mockGetStatus.mockResolvedValue({ completed: false });
|
|
||||||
mockCreateBreakglass.mockResolvedValue({ id: "bg-1", username: "admin" });
|
|
||||||
mockConfigureOidc.mockResolvedValue(undefined);
|
|
||||||
mockTestProvider.mockResolvedValue({ success: true });
|
|
||||||
mockAddProvider.mockResolvedValue({ id: "provider-1" });
|
|
||||||
mockCompleteOnboarding.mockResolvedValue(undefined);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("renders the first step with admin setup fields", async () => {
|
|
||||||
render(<OnboardingWizard />);
|
|
||||||
|
|
||||||
expect(
|
|
||||||
await screen.findByText("Welcome to Mosaic Stack. Let's get you set up.")
|
|
||||||
).toBeInTheDocument();
|
|
||||||
expect(screen.getByLabelText("Username")).toBeInTheDocument();
|
|
||||||
expect(screen.getByLabelText("Password")).toBeInTheDocument();
|
|
||||||
expect(screen.getByLabelText("Confirm Password")).toBeInTheDocument();
|
|
||||||
expect(screen.getByText("1. Admin")).toBeInTheDocument();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("validates admin form fields before submit", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
render(<OnboardingWizard />);
|
|
||||||
|
|
||||||
await screen.findByText("Welcome to Mosaic Stack. Let's get you set up.");
|
|
||||||
await user.click(screen.getByRole("button", { name: "Create Admin" }));
|
|
||||||
|
|
||||||
expect(screen.getByText("Username must be at least 3 characters.")).toBeInTheDocument();
|
|
||||||
expect(mockCreateBreakglass).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("supports happy path with OIDC skipped", async () => {
|
|
||||||
const user = userEvent.setup();
|
|
||||||
render(<OnboardingWizard />);
|
|
||||||
|
|
||||||
await screen.findByText("Welcome to Mosaic Stack. Let's get you set up.");
|
|
||||||
|
|
||||||
await user.type(screen.getByLabelText("Username"), "admin");
|
|
||||||
await user.type(screen.getByLabelText("Password"), "verysecurepassword");
|
|
||||||
await user.type(screen.getByLabelText("Confirm Password"), "verysecurepassword");
|
|
||||||
await user.click(screen.getByRole("button", { name: "Create Admin" }));
|
|
||||||
|
|
||||||
await screen.findByText("Configure OIDC Provider (Optional)");
|
|
||||||
await user.click(screen.getByRole("button", { name: "Skip" }));
|
|
||||||
|
|
||||||
await screen.findByText("Add Your First LLM Provider");
|
|
||||||
await user.type(screen.getByLabelText("Display Name"), "My OpenAI");
|
|
||||||
await user.type(screen.getByLabelText("API Key"), "sk-test-key");
|
|
||||||
|
|
||||||
await user.click(screen.getByRole("button", { name: "Test Connection" }));
|
|
||||||
await screen.findByText("Connection successful.");
|
|
||||||
|
|
||||||
const addProviderButton = screen.getByRole("button", { name: "Add Provider" });
|
|
||||||
expect(addProviderButton).toBeEnabled();
|
|
||||||
await user.click(addProviderButton);
|
|
||||||
|
|
||||||
await screen.findByText("You're all set");
|
|
||||||
await user.click(screen.getByRole("button", { name: "Launch Mosaic Stack" }));
|
|
||||||
|
|
||||||
await waitFor(() => {
|
|
||||||
expect(mockPush).toHaveBeenCalledWith("/");
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,791 +0,0 @@
|
|||||||
"use client";
|
|
||||||
|
|
||||||
import { useEffect, useMemo, useState } from "react";
|
|
||||||
import { Check, Loader2 } from "lucide-react";
|
|
||||||
import { useRouter } from "next/navigation";
|
|
||||||
import { Button } from "@/components/ui/button";
|
|
||||||
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
|
|
||||||
import { Input } from "@/components/ui/input";
|
|
||||||
import { Label } from "@/components/ui/label";
|
|
||||||
import {
|
|
||||||
Select,
|
|
||||||
SelectContent,
|
|
||||||
SelectItem,
|
|
||||||
SelectTrigger,
|
|
||||||
SelectValue,
|
|
||||||
} from "@/components/ui/select";
|
|
||||||
import {
|
|
||||||
addOnboardingProvider,
|
|
||||||
completeOnboarding,
|
|
||||||
configureOidcProvider,
|
|
||||||
createBreakglassAdmin,
|
|
||||||
fetchOnboardingStatus,
|
|
||||||
testOnboardingProvider,
|
|
||||||
} from "@/lib/api/onboarding";
|
|
||||||
|
|
||||||
type WizardStep = 1 | 2 | 3 | 4;
|
|
||||||
type ProviderType = "openai" | "anthropic" | "zai" | "ollama" | "custom";
|
|
||||||
|
|
||||||
interface StepDefinition {
|
|
||||||
id: WizardStep;
|
|
||||||
label: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
interface ProviderOption {
|
|
||||||
value: ProviderType;
|
|
||||||
label: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
const STEPS: StepDefinition[] = [
|
|
||||||
{ id: 1, label: "1. Admin" },
|
|
||||||
{ id: 2, label: "2. Auth" },
|
|
||||||
{ id: 3, label: "3. Provider" },
|
|
||||||
{ id: 4, label: "4. Launch" },
|
|
||||||
];
|
|
||||||
|
|
||||||
const PROVIDER_OPTIONS: ProviderOption[] = [
|
|
||||||
{ value: "openai", label: "OpenAI" },
|
|
||||||
{ value: "anthropic", label: "Anthropic" },
|
|
||||||
{ value: "zai", label: "Z.ai" },
|
|
||||||
{ value: "ollama", label: "Ollama" },
|
|
||||||
{ value: "custom", label: "Custom" },
|
|
||||||
];
|
|
||||||
|
|
||||||
const CLOUD_PROVIDER_TYPES = new Set<ProviderType>(["openai", "anthropic", "zai"]);
|
|
||||||
const BASE_URL_PROVIDER_TYPES = new Set<ProviderType>(["ollama", "custom"]);
|
|
||||||
|
|
||||||
function getErrorMessage(error: unknown, fallback: string): string {
|
|
||||||
if (error instanceof Error && error.message.trim().length > 0) {
|
|
||||||
return error.message;
|
|
||||||
}
|
|
||||||
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
function isValidHttpUrl(value: string): boolean {
|
|
||||||
try {
|
|
||||||
const parsed = new URL(value);
|
|
||||||
return parsed.protocol === "http:" || parsed.protocol === "https:";
|
|
||||||
} catch {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function mapProviderTypeToApi(type: ProviderType): string {
|
|
||||||
switch (type) {
|
|
||||||
case "anthropic":
|
|
||||||
return "claude";
|
|
||||||
case "zai":
|
|
||||||
return "openai";
|
|
||||||
case "custom":
|
|
||||||
return "openai";
|
|
||||||
default:
|
|
||||||
return type;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function getProviderDefaultBaseUrl(type: ProviderType): string | undefined {
|
|
||||||
switch (type) {
|
|
||||||
case "ollama":
|
|
||||||
return "http://localhost:11434";
|
|
||||||
case "anthropic":
|
|
||||||
return "https://api.anthropic.com/v1";
|
|
||||||
case "zai":
|
|
||||||
return "https://api.z.ai/v1";
|
|
||||||
default:
|
|
||||||
return undefined;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function buildProviderName(displayName: string, type: ProviderType): string {
|
|
||||||
const slug = displayName
|
|
||||||
.trim()
|
|
||||||
.toLowerCase()
|
|
||||||
.replace(/[^a-z0-9]+/g, "-")
|
|
||||||
.replace(/^-+/, "")
|
|
||||||
.replace(/-+$/, "");
|
|
||||||
|
|
||||||
if (slug.length > 0) {
|
|
||||||
return slug;
|
|
||||||
}
|
|
||||||
|
|
||||||
return `${type}-provider`;
|
|
||||||
}
|
|
||||||
|
|
||||||
function constantTimeEquals(left: string, right: string): boolean {
|
|
||||||
if (left.length !== right.length) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
let mismatch = 0;
|
|
||||||
for (let index = 0; index < left.length; index += 1) {
|
|
||||||
mismatch |= left.charCodeAt(index) ^ right.charCodeAt(index);
|
|
||||||
}
|
|
||||||
|
|
||||||
return mismatch === 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
export function OnboardingWizard(): React.JSX.Element {
|
|
||||||
const router = useRouter();
|
|
||||||
|
|
||||||
const [currentStep, setCurrentStep] = useState<WizardStep>(1);
|
|
||||||
const [isCheckingStatus, setIsCheckingStatus] = useState(true);
|
|
||||||
|
|
||||||
const [username, setUsername] = useState("");
|
|
||||||
const [password, setPassword] = useState("");
|
|
||||||
const [confirmPassword, setConfirmPassword] = useState("");
|
|
||||||
const [isCreatingAdmin, setIsCreatingAdmin] = useState(false);
|
|
||||||
const [configuredUsername, setConfiguredUsername] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const [issuerUrl, setIssuerUrl] = useState("");
|
|
||||||
const [clientId, setClientId] = useState("");
|
|
||||||
const [clientSecret, setClientSecret] = useState("");
|
|
||||||
const [isConfiguringOidc, setIsConfiguringOidc] = useState(false);
|
|
||||||
const [oidcConfigured, setOidcConfigured] = useState(false);
|
|
||||||
|
|
||||||
const [providerType, setProviderType] = useState<ProviderType>("openai");
|
|
||||||
const [displayName, setDisplayName] = useState("");
|
|
||||||
const [providerApiKey, setProviderApiKey] = useState("");
|
|
||||||
const [providerBaseUrl, setProviderBaseUrl] = useState("");
|
|
||||||
const [isTestingProvider, setIsTestingProvider] = useState(false);
|
|
||||||
const [isAddingProvider, setIsAddingProvider] = useState(false);
|
|
||||||
const [providerConfigured, setProviderConfigured] = useState<{
|
|
||||||
displayName: string;
|
|
||||||
type: ProviderType;
|
|
||||||
} | null>(null);
|
|
||||||
const [providerTestMessage, setProviderTestMessage] = useState<string | null>(null);
|
|
||||||
const [providerTestSucceeded, setProviderTestSucceeded] = useState(false);
|
|
||||||
const [testedProviderSignature, setTestedProviderSignature] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const [isCompleting, setIsCompleting] = useState(false);
|
|
||||||
const [errorMessage, setErrorMessage] = useState<string | null>(null);
|
|
||||||
|
|
||||||
const requiresApiKey = CLOUD_PROVIDER_TYPES.has(providerType);
|
|
||||||
const requiresBaseUrl = BASE_URL_PROVIDER_TYPES.has(providerType);
|
|
||||||
const apiProviderType = mapProviderTypeToApi(providerType);
|
|
||||||
const resolvedProviderBaseUrl =
|
|
||||||
requiresBaseUrl && providerBaseUrl.trim().length > 0
|
|
||||||
? providerBaseUrl.trim()
|
|
||||||
: getProviderDefaultBaseUrl(providerType);
|
|
||||||
|
|
||||||
const providerTestPayload = useMemo(() => {
|
|
||||||
const payload: { type: string; baseUrl?: string; apiKey?: string } = {
|
|
||||||
type: apiProviderType,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (resolvedProviderBaseUrl !== undefined && resolvedProviderBaseUrl.length > 0) {
|
|
||||||
payload.baseUrl = resolvedProviderBaseUrl;
|
|
||||||
}
|
|
||||||
|
|
||||||
const trimmedApiKey = providerApiKey.trim();
|
|
||||||
if (requiresApiKey && trimmedApiKey.length > 0) {
|
|
||||||
payload.apiKey = trimmedApiKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
return payload;
|
|
||||||
}, [apiProviderType, providerApiKey, requiresApiKey, resolvedProviderBaseUrl]);
|
|
||||||
|
|
||||||
const providerPayloadSignature = useMemo(
|
|
||||||
() => JSON.stringify(providerTestPayload),
|
|
||||||
[providerTestPayload]
|
|
||||||
);
|
|
||||||
|
|
||||||
const canAddProvider =
|
|
||||||
providerTestSucceeded &&
|
|
||||||
testedProviderSignature === providerPayloadSignature &&
|
|
||||||
!isTestingProvider &&
|
|
||||||
!isAddingProvider;
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
let cancelled = false;
|
|
||||||
|
|
||||||
async function loadStatus(): Promise<void> {
|
|
||||||
try {
|
|
||||||
const status = await fetchOnboardingStatus();
|
|
||||||
if (!cancelled && status.completed) {
|
|
||||||
router.push("/");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
// Status check failure should not block setup UI.
|
|
||||||
} finally {
|
|
||||||
if (!cancelled) {
|
|
||||||
setIsCheckingStatus(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void loadStatus();
|
|
||||||
|
|
||||||
return (): void => {
|
|
||||||
cancelled = true;
|
|
||||||
};
|
|
||||||
}, [router]);
|
|
||||||
|
|
||||||
const resetProviderVerification = (): void => {
|
|
||||||
setProviderTestSucceeded(false);
|
|
||||||
setTestedProviderSignature(null);
|
|
||||||
setProviderTestMessage(null);
|
|
||||||
};
|
|
||||||
|
|
||||||
const validateAdminStep = (): boolean => {
|
|
||||||
if (username.trim().length < 3) {
|
|
||||||
setErrorMessage("Username must be at least 3 characters.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (password.length < 8) {
|
|
||||||
setErrorMessage("Password must be at least 8 characters.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!constantTimeEquals(password, confirmPassword)) {
|
|
||||||
setErrorMessage("Passwords do not match.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
};
|
|
||||||
|
|
||||||
const validateOidcStep = (): boolean => {
|
|
||||||
if (issuerUrl.trim().length === 0 || !isValidHttpUrl(issuerUrl.trim())) {
|
|
||||||
setErrorMessage("Issuer URL must be a valid URL.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (clientId.trim().length === 0) {
|
|
||||||
setErrorMessage("Client ID is required.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (clientSecret.trim().length === 0) {
|
|
||||||
setErrorMessage("Client secret is required.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
};
|
|
||||||
|
|
||||||
const validateProviderStep = (): boolean => {
|
|
||||||
if (displayName.trim().length === 0) {
|
|
||||||
setErrorMessage("Display name is required.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (requiresApiKey && providerApiKey.trim().length === 0) {
|
|
||||||
setErrorMessage("API key is required for this provider.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (requiresBaseUrl && providerBaseUrl.trim().length === 0) {
|
|
||||||
setErrorMessage("Base URL is required for this provider.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (requiresBaseUrl && !isValidHttpUrl(providerBaseUrl.trim())) {
|
|
||||||
setErrorMessage("Base URL must be a valid URL.");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleCreateAdmin = async (event: React.SyntheticEvent<HTMLFormElement>): Promise<void> => {
|
|
||||||
event.preventDefault();
|
|
||||||
|
|
||||||
setErrorMessage(null);
|
|
||||||
if (!validateAdminStep()) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsCreatingAdmin(true);
|
|
||||||
try {
|
|
||||||
const result = await createBreakglassAdmin({
|
|
||||||
username: username.trim(),
|
|
||||||
password,
|
|
||||||
});
|
|
||||||
setConfiguredUsername(result.username);
|
|
||||||
setCurrentStep(2);
|
|
||||||
} catch (error) {
|
|
||||||
setErrorMessage(getErrorMessage(error, "Failed to create admin account."));
|
|
||||||
} finally {
|
|
||||||
setIsCreatingAdmin(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleConfigureOidc = async (
|
|
||||||
event: React.SyntheticEvent<HTMLFormElement>
|
|
||||||
): Promise<void> => {
|
|
||||||
event.preventDefault();
|
|
||||||
|
|
||||||
setErrorMessage(null);
|
|
||||||
if (!validateOidcStep()) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsConfiguringOidc(true);
|
|
||||||
try {
|
|
||||||
await configureOidcProvider({
|
|
||||||
issuerUrl: issuerUrl.trim(),
|
|
||||||
clientId: clientId.trim(),
|
|
||||||
clientSecret: clientSecret.trim(),
|
|
||||||
});
|
|
||||||
setOidcConfigured(true);
|
|
||||||
setCurrentStep(3);
|
|
||||||
} catch (error) {
|
|
||||||
setErrorMessage(getErrorMessage(error, "Failed to configure OIDC provider."));
|
|
||||||
} finally {
|
|
||||||
setIsConfiguringOidc(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleSkipOidc = (): void => {
|
|
||||||
setErrorMessage(null);
|
|
||||||
setOidcConfigured(false);
|
|
||||||
setCurrentStep(3);
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleTestProvider = async (): Promise<void> => {
|
|
||||||
setErrorMessage(null);
|
|
||||||
setProviderTestMessage(null);
|
|
||||||
if (!validateProviderStep()) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsTestingProvider(true);
|
|
||||||
try {
|
|
||||||
const response = await testOnboardingProvider(providerTestPayload);
|
|
||||||
if (!response.success) {
|
|
||||||
setProviderTestSucceeded(false);
|
|
||||||
setTestedProviderSignature(null);
|
|
||||||
setErrorMessage(response.error ?? "Connection test failed.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setProviderTestSucceeded(true);
|
|
||||||
setTestedProviderSignature(providerPayloadSignature);
|
|
||||||
setProviderTestMessage("Connection successful.");
|
|
||||||
} catch (error) {
|
|
||||||
setProviderTestSucceeded(false);
|
|
||||||
setTestedProviderSignature(null);
|
|
||||||
setErrorMessage(getErrorMessage(error, "Connection test failed."));
|
|
||||||
} finally {
|
|
||||||
setIsTestingProvider(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleAddProvider = async (): Promise<void> => {
|
|
||||||
setErrorMessage(null);
|
|
||||||
if (!validateProviderStep()) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!canAddProvider) {
|
|
||||||
setErrorMessage("Test connection successfully before adding the provider.");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setIsAddingProvider(true);
|
|
||||||
try {
|
|
||||||
const trimmedDisplayName = displayName.trim();
|
|
||||||
const payload: {
|
|
||||||
name: string;
|
|
||||||
displayName: string;
|
|
||||||
type: string;
|
|
||||||
baseUrl?: string;
|
|
||||||
apiKey?: string;
|
|
||||||
} = {
|
|
||||||
name: buildProviderName(trimmedDisplayName, providerType),
|
|
||||||
displayName: trimmedDisplayName,
|
|
||||||
type: apiProviderType,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (resolvedProviderBaseUrl !== undefined && resolvedProviderBaseUrl.length > 0) {
|
|
||||||
payload.baseUrl = resolvedProviderBaseUrl;
|
|
||||||
}
|
|
||||||
|
|
||||||
const trimmedApiKey = providerApiKey.trim();
|
|
||||||
if (requiresApiKey && trimmedApiKey.length > 0) {
|
|
||||||
payload.apiKey = trimmedApiKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
await addOnboardingProvider(payload);
|
|
||||||
|
|
||||||
setProviderConfigured({ displayName: trimmedDisplayName, type: providerType });
|
|
||||||
setCurrentStep(4);
|
|
||||||
} catch (error) {
|
|
||||||
setErrorMessage(getErrorMessage(error, "Failed to add provider."));
|
|
||||||
} finally {
|
|
||||||
setIsAddingProvider(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleCompleteOnboarding = async (): Promise<void> => {
|
|
||||||
setErrorMessage(null);
|
|
||||||
setIsCompleting(true);
|
|
||||||
try {
|
|
||||||
await completeOnboarding();
|
|
||||||
router.push("/");
|
|
||||||
} catch (error) {
|
|
||||||
setErrorMessage(getErrorMessage(error, "Failed to complete onboarding."));
|
|
||||||
} finally {
|
|
||||||
setIsCompleting(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const providerLabel =
|
|
||||||
PROVIDER_OPTIONS.find((option) => option.value === providerConfigured?.type)?.label ??
|
|
||||||
providerConfigured?.type ??
|
|
||||||
"Unknown";
|
|
||||||
|
|
||||||
return (
|
|
||||||
<Card className="mx-auto w-full max-w-2xl shadow-sm">
|
|
||||||
<CardHeader>
|
|
||||||
<CardTitle>First-boot onboarding</CardTitle>
|
|
||||||
<CardDescription>Set up your admin access, auth, and first provider.</CardDescription>
|
|
||||||
</CardHeader>
|
|
||||||
<CardContent className="space-y-6">
|
|
||||||
<div className="grid grid-cols-2 gap-2 sm:grid-cols-4">
|
|
||||||
{STEPS.map((step) => {
|
|
||||||
const isActive = currentStep === step.id;
|
|
||||||
const isComplete = currentStep > step.id;
|
|
||||||
const badgeClass = isComplete
|
|
||||||
? "bg-emerald-100 text-emerald-700 border-emerald-200"
|
|
||||||
: isActive
|
|
||||||
? "bg-blue-100 text-blue-700 border-blue-200"
|
|
||||||
: "bg-gray-100 text-gray-500 border-gray-200";
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div
|
|
||||||
key={step.id}
|
|
||||||
className={`rounded-md border px-3 py-2 text-sm ${badgeClass}`}
|
|
||||||
aria-current={isActive ? "step" : undefined}
|
|
||||||
>
|
|
||||||
<div className="flex items-center gap-2 font-medium">
|
|
||||||
<span className="inline-flex h-5 w-5 items-center justify-center rounded-full border border-current text-xs">
|
|
||||||
{isComplete ? <Check className="h-3.5 w-3.5" aria-hidden="true" /> : step.id}
|
|
||||||
</span>
|
|
||||||
<span>{step.label}</span>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
})}
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{isCheckingStatus ? (
|
|
||||||
<div className="flex items-center gap-2 text-sm text-gray-600">
|
|
||||||
<Loader2 className="h-4 w-4 animate-spin" aria-hidden="true" />
|
|
||||||
<span>Checking onboarding status...</span>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<>
|
|
||||||
{currentStep === 1 && (
|
|
||||||
<form onSubmit={handleCreateAdmin} className="space-y-4" noValidate>
|
|
||||||
<div className="space-y-1">
|
|
||||||
<h2 className="text-xl font-semibold">
|
|
||||||
Welcome to Mosaic Stack. Let's get you set up.
|
|
||||||
</h2>
|
|
||||||
<p className="text-sm text-gray-600">
|
|
||||||
Create a breakglass admin account for emergency access.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-username">Username</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-username"
|
|
||||||
value={username}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setUsername(event.target.value);
|
|
||||||
}}
|
|
||||||
disabled={isCreatingAdmin}
|
|
||||||
autoComplete="username"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-password">Password</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-password"
|
|
||||||
type="password"
|
|
||||||
value={password}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setPassword(event.target.value);
|
|
||||||
}}
|
|
||||||
disabled={isCreatingAdmin}
|
|
||||||
autoComplete="new-password"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-confirm-password">Confirm Password</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-confirm-password"
|
|
||||||
type="password"
|
|
||||||
value={confirmPassword}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setConfirmPassword(event.target.value);
|
|
||||||
}}
|
|
||||||
disabled={isCreatingAdmin}
|
|
||||||
autoComplete="new-password"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Button type="submit" disabled={isCreatingAdmin}>
|
|
||||||
{isCreatingAdmin && (
|
|
||||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" aria-hidden="true" />
|
|
||||||
)}
|
|
||||||
<span>Create Admin</span>
|
|
||||||
</Button>
|
|
||||||
</form>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{currentStep === 2 && (
|
|
||||||
<form onSubmit={handleConfigureOidc} className="space-y-4" noValidate>
|
|
||||||
<div className="space-y-1">
|
|
||||||
<h2 className="text-xl font-semibold">Configure OIDC Provider (Optional)</h2>
|
|
||||||
<p className="text-sm text-gray-600">
|
|
||||||
You can skip this for now and continue with breakglass-only authentication.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-issuer-url">OIDC Issuer URL</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-issuer-url"
|
|
||||||
value={issuerUrl}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setIssuerUrl(event.target.value);
|
|
||||||
}}
|
|
||||||
disabled={isConfiguringOidc}
|
|
||||||
placeholder="https://auth.example.com/application/o/mosaic/"
|
|
||||||
autoComplete="url"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-client-id">Client ID</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-client-id"
|
|
||||||
value={clientId}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setClientId(event.target.value);
|
|
||||||
}}
|
|
||||||
disabled={isConfiguringOidc}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-client-secret">Client Secret</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-client-secret"
|
|
||||||
type="password"
|
|
||||||
value={clientSecret}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setClientSecret(event.target.value);
|
|
||||||
}}
|
|
||||||
disabled={isConfiguringOidc}
|
|
||||||
autoComplete="off"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="flex flex-wrap gap-2">
|
|
||||||
<Button type="submit" disabled={isConfiguringOidc}>
|
|
||||||
{isConfiguringOidc && (
|
|
||||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" aria-hidden="true" />
|
|
||||||
)}
|
|
||||||
<span>Configure OIDC</span>
|
|
||||||
</Button>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="outline"
|
|
||||||
onClick={handleSkipOidc}
|
|
||||||
disabled={isConfiguringOidc}
|
|
||||||
>
|
|
||||||
Skip
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{currentStep === 3 && (
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div className="space-y-1">
|
|
||||||
<h2 className="text-xl font-semibold">Add Your First LLM Provider</h2>
|
|
||||||
<p className="text-sm text-gray-600">
|
|
||||||
Test the connection before adding your provider.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-provider-type">Provider Type</Label>
|
|
||||||
<Select
|
|
||||||
value={providerType}
|
|
||||||
onValueChange={(value) => {
|
|
||||||
const nextType = value as ProviderType;
|
|
||||||
setProviderType(nextType);
|
|
||||||
setProviderApiKey("");
|
|
||||||
setProviderBaseUrl(
|
|
||||||
BASE_URL_PROVIDER_TYPES.has(nextType)
|
|
||||||
? (getProviderDefaultBaseUrl(nextType) ?? "")
|
|
||||||
: ""
|
|
||||||
);
|
|
||||||
resetProviderVerification();
|
|
||||||
setErrorMessage(null);
|
|
||||||
}}
|
|
||||||
disabled={isTestingProvider || isAddingProvider}
|
|
||||||
>
|
|
||||||
<SelectTrigger id="onboarding-provider-type">
|
|
||||||
<SelectValue placeholder="Select provider type" />
|
|
||||||
</SelectTrigger>
|
|
||||||
<SelectContent>
|
|
||||||
{PROVIDER_OPTIONS.map((option) => (
|
|
||||||
<SelectItem key={option.value} value={option.value}>
|
|
||||||
{option.label}
|
|
||||||
</SelectItem>
|
|
||||||
))}
|
|
||||||
</SelectContent>
|
|
||||||
</Select>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-provider-display-name">Display Name</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-provider-display-name"
|
|
||||||
value={displayName}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setDisplayName(event.target.value);
|
|
||||||
resetProviderVerification();
|
|
||||||
setErrorMessage(null);
|
|
||||||
}}
|
|
||||||
disabled={isTestingProvider || isAddingProvider}
|
|
||||||
placeholder="My OpenAI Provider"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{requiresApiKey && (
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-provider-api-key">API Key</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-provider-api-key"
|
|
||||||
type="password"
|
|
||||||
value={providerApiKey}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setProviderApiKey(event.target.value);
|
|
||||||
resetProviderVerification();
|
|
||||||
setErrorMessage(null);
|
|
||||||
}}
|
|
||||||
disabled={isTestingProvider || isAddingProvider}
|
|
||||||
autoComplete="off"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{requiresBaseUrl && (
|
|
||||||
<div className="grid gap-2">
|
|
||||||
<Label htmlFor="onboarding-provider-base-url">Base URL</Label>
|
|
||||||
<Input
|
|
||||||
id="onboarding-provider-base-url"
|
|
||||||
value={providerBaseUrl}
|
|
||||||
onChange={(event: React.ChangeEvent<HTMLInputElement>) => {
|
|
||||||
setProviderBaseUrl(event.target.value);
|
|
||||||
resetProviderVerification();
|
|
||||||
setErrorMessage(null);
|
|
||||||
}}
|
|
||||||
disabled={isTestingProvider || isAddingProvider}
|
|
||||||
placeholder="http://localhost:11434"
|
|
||||||
autoComplete="url"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{providerTestMessage && (
|
|
||||||
<p className="text-sm text-emerald-700" role="status">
|
|
||||||
{providerTestMessage}
|
|
||||||
</p>
|
|
||||||
)}
|
|
||||||
|
|
||||||
<div className="flex flex-wrap gap-2">
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
variant="outline"
|
|
||||||
onClick={() => {
|
|
||||||
void handleTestProvider();
|
|
||||||
}}
|
|
||||||
disabled={isTestingProvider || isAddingProvider}
|
|
||||||
>
|
|
||||||
{isTestingProvider && (
|
|
||||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" aria-hidden="true" />
|
|
||||||
)}
|
|
||||||
<span>Test Connection</span>
|
|
||||||
</Button>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
onClick={() => {
|
|
||||||
void handleAddProvider();
|
|
||||||
}}
|
|
||||||
disabled={!canAddProvider}
|
|
||||||
>
|
|
||||||
{isAddingProvider && (
|
|
||||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" aria-hidden="true" />
|
|
||||||
)}
|
|
||||||
<span>Add Provider</span>
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{currentStep === 4 && (
|
|
||||||
<div className="space-y-4">
|
|
||||||
<div className="space-y-1">
|
|
||||||
<h2 className="text-xl font-semibold">You're all set</h2>
|
|
||||||
<p className="text-sm text-gray-600">
|
|
||||||
Review the setup summary and launch Mosaic Stack.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="rounded-md border bg-gray-50 p-4">
|
|
||||||
<ul className="space-y-2 text-sm">
|
|
||||||
<li>
|
|
||||||
<span className="font-medium">Admin:</span>{" "}
|
|
||||||
{configuredUsername ? `${configuredUsername} configured` : "Not configured"}
|
|
||||||
</li>
|
|
||||||
<li>
|
|
||||||
<span className="font-medium">OIDC:</span>{" "}
|
|
||||||
{oidcConfigured ? "Configured" : "Skipped for now"}
|
|
||||||
</li>
|
|
||||||
<li>
|
|
||||||
<span className="font-medium">LLM Provider:</span>{" "}
|
|
||||||
{providerConfigured
|
|
||||||
? `${providerConfigured.displayName} (${providerLabel})`
|
|
||||||
: "Not configured"}
|
|
||||||
</li>
|
|
||||||
</ul>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
onClick={() => void handleCompleteOnboarding()}
|
|
||||||
disabled={isCompleting}
|
|
||||||
>
|
|
||||||
{isCompleting && (
|
|
||||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" aria-hidden="true" />
|
|
||||||
)}
|
|
||||||
<span>Launch Mosaic Stack</span>
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{errorMessage && (
|
|
||||||
<p className="text-sm text-red-600" role="alert">
|
|
||||||
{errorMessage}
|
|
||||||
</p>
|
|
||||||
)}
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
export default OnboardingWizard;
|
|
||||||
@@ -1,51 +0,0 @@
|
|||||||
"use client";
|
|
||||||
|
|
||||||
import Link from "next/link";
|
|
||||||
import { usePathname } from "next/navigation";
|
|
||||||
import { Card, CardContent } from "@/components/ui/card";
|
|
||||||
|
|
||||||
interface FleetSettingsLink {
|
|
||||||
href: string;
|
|
||||||
label: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
const FLEET_SETTINGS_LINKS: FleetSettingsLink[] = [
|
|
||||||
{ href: "/settings/providers", label: "Providers" },
|
|
||||||
{ href: "/settings/agent-config", label: "Agent Config" },
|
|
||||||
{ href: "/settings/auth", label: "Authentication" },
|
|
||||||
];
|
|
||||||
|
|
||||||
export function FleetSettingsNav(): React.JSX.Element {
|
|
||||||
const pathname = usePathname();
|
|
||||||
|
|
||||||
return (
|
|
||||||
<Card>
|
|
||||||
<CardContent className="px-4 py-3 flex flex-wrap items-center gap-2">
|
|
||||||
<Link
|
|
||||||
href="/settings"
|
|
||||||
className="inline-flex h-9 items-center rounded-md px-3 text-sm font-medium text-muted-foreground hover:text-foreground hover:bg-muted transition-colors"
|
|
||||||
>
|
|
||||||
All Settings
|
|
||||||
</Link>
|
|
||||||
|
|
||||||
{FLEET_SETTINGS_LINKS.map((link) => {
|
|
||||||
const isActive = pathname === link.href;
|
|
||||||
|
|
||||||
return (
|
|
||||||
<Link
|
|
||||||
key={link.href}
|
|
||||||
href={link.href}
|
|
||||||
className={`inline-flex h-9 items-center rounded-md px-3 text-sm font-medium transition-colors ${
|
|
||||||
isActive
|
|
||||||
? "bg-primary text-primary-foreground"
|
|
||||||
: "text-muted-foreground hover:text-foreground hover:bg-muted"
|
|
||||||
}`}
|
|
||||||
>
|
|
||||||
{link.label}
|
|
||||||
</Link>
|
|
||||||
);
|
|
||||||
})}
|
|
||||||
</CardContent>
|
|
||||||
</Card>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -1,16 +0,0 @@
|
|||||||
import type { ReactElement } from "react";
|
|
||||||
|
|
||||||
interface SettingsAccessDeniedProps {
|
|
||||||
message: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export function SettingsAccessDenied({ message }: SettingsAccessDeniedProps): ReactElement {
|
|
||||||
return (
|
|
||||||
<div className="p-8 max-w-2xl">
|
|
||||||
<div className="rounded-lg border border-red-200 bg-red-50 p-6 text-center">
|
|
||||||
<p className="text-lg font-semibold text-red-700">Access Denied</p>
|
|
||||||
<p className="mt-2 text-sm text-red-600">{message}</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -1,60 +0,0 @@
|
|||||||
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
||||||
import * as client from "./client";
|
|
||||||
import { fetchAdminUsers, inviteUser, updateUser, deactivateUser } from "./admin";
|
|
||||||
|
|
||||||
vi.mock("./client");
|
|
||||||
|
|
||||||
beforeEach((): void => {
|
|
||||||
vi.clearAllMocks();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("fetchAdminUsers", (): void => {
|
|
||||||
it("calls admin/users endpoint without params when none provided", async (): Promise<void> => {
|
|
||||||
vi.mocked(client.apiGet).mockResolvedValueOnce({ data: [], meta: {} } as never);
|
|
||||||
await fetchAdminUsers();
|
|
||||||
expect(client.apiGet).toHaveBeenCalledWith("/api/admin/users");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("appends page and limit params when provided", async (): Promise<void> => {
|
|
||||||
vi.mocked(client.apiGet).mockResolvedValueOnce({ data: [], meta: {} } as never);
|
|
||||||
await fetchAdminUsers(2, 50);
|
|
||||||
expect(client.apiGet).toHaveBeenCalledWith("/api/admin/users?page=2&limit=50");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("throws on API error", async (): Promise<void> => {
|
|
||||||
vi.mocked(client.apiGet).mockRejectedValueOnce(new Error("Network error"));
|
|
||||||
await expect(fetchAdminUsers()).rejects.toThrow("Network error");
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("inviteUser", (): void => {
|
|
||||||
it("posts to invite endpoint", async (): Promise<void> => {
|
|
||||||
vi.mocked(client.apiPost).mockResolvedValueOnce({ id: "inv-1" } as never);
|
|
||||||
await inviteUser({
|
|
||||||
email: "a@b.com",
|
|
||||||
name: "Alice",
|
|
||||||
workspaceId: "ws-1",
|
|
||||||
role: "MEMBER" as never,
|
|
||||||
});
|
|
||||||
expect(client.apiPost).toHaveBeenCalledWith(
|
|
||||||
"/api/admin/users/invite",
|
|
||||||
expect.objectContaining({ email: "a@b.com" })
|
|
||||||
);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("updateUser", (): void => {
|
|
||||||
it("patches correct endpoint with dto", async (): Promise<void> => {
|
|
||||||
vi.mocked(client.apiPatch).mockResolvedValueOnce({ id: "u1", name: "Bob" } as never);
|
|
||||||
await updateUser("u1", { name: "Bob" });
|
|
||||||
expect(client.apiPatch).toHaveBeenCalledWith("/api/admin/users/u1", { name: "Bob" });
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("deactivateUser", (): void => {
|
|
||||||
it("deletes correct endpoint", async (): Promise<void> => {
|
|
||||||
vi.mocked(client.apiDelete).mockResolvedValueOnce({} as never);
|
|
||||||
await deactivateUser("u1");
|
|
||||||
expect(client.apiDelete).toHaveBeenCalledWith("/api/admin/users/u1");
|
|
||||||
});
|
|
||||||
});
|
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user