Merge feature/82-personality-module (#82) into develop

Implements Personality Module:
- Personality model and Prisma migration
- CRUD API with controller and service
- Comprehensive test suite
- Integration with workspace
This commit is contained in:
Jason Woltje
2026-01-29 17:59:28 -06:00
43 changed files with 4782 additions and 2 deletions

View File

@@ -0,0 +1,268 @@
# Feature #18: Advanced Filtering and Search - Implementation Summary
## Overview
Implemented comprehensive filtering and search capabilities for Mosaic Stack, including backend query enhancements and a frontend FilterBar component.
## Backend Implementation
### 1. Shared Filter DTOs (`apps/api/src/common/dto/`)
**Files Created:**
- `base-filter.dto.ts` - Base DTO with pagination, sorting, and search
- `base-filter.dto.spec.ts` - Comprehensive validation tests (16 tests)
**Features:**
- Pagination support (page, limit with validation)
- Full-text search with trimming and max length validation
- Multi-field sorting (`sortBy` comma-separated, `sortOrder`)
- Date range filtering (`dateFrom`, `dateTo`)
- Enum `SortOrder` (ASC/DESC)
### 2. Query Builder Utility (`apps/api/src/common/utils/`)
**Files Created:**
- `query-builder.ts` - Reusable Prisma query building utilities
- `query-builder.spec.ts` - Complete test coverage (23 tests)
**Methods:**
- `buildSearchFilter()` - Full-text search across multiple fields (case-insensitive)
- `buildSortOrder()` - Single or multi-field sorting with custom order per field
- `buildDateRangeFilter()` - Date range with gte/lte operators
- `buildInFilter()` - Multi-select filters (supports arrays)
- `buildPaginationParams()` - Calculate skip/take for pagination
- `buildPaginationMeta()` - Rich pagination metadata with hasNextPage/hasPrevPage
### 3. Enhanced Query DTOs
**Updated:**
- `apps/api/src/tasks/dto/query-tasks.dto.ts` - Now extends BaseFilterDto
- `apps/api/src/tasks/dto/query-tasks.dto.spec.ts` - Comprehensive tests (13 tests)
**New Features:**
- Multi-select status filter (TaskStatus[])
- Multi-select priority filter (TaskPriority[])
- Multi-select domain filter (domainId[])
- Full-text search on title/description
- Multi-field sorting
- Date range filtering on dueDate
### 4. Service Layer Updates
**Updated:**
- `apps/api/src/tasks/tasks.service.ts` - Uses QueryBuilder for all filtering
**Improvements:**
- Cleaner, more maintainable filter building
- Consistent pagination across endpoints
- Rich pagination metadata
- Support for complex multi-filter queries
## Frontend Implementation
### 1. FilterBar Component (`apps/web/src/components/filters/`)
**Files Created:**
- `FilterBar.tsx` - Main filter component
- `FilterBar.test.tsx` - Component tests (12 tests)
- `index.ts` - Export barrel
**Features:**
- **Search Input**: Debounced full-text search (customizable debounce delay)
- **Status Filter**: Multi-select dropdown with checkboxes
- **Priority Filter**: Multi-select dropdown with checkboxes
- **Date Range Picker**: From/To date inputs
- **Active Filter Count**: Badge showing number of active filters
- **Clear All Filters**: Button to reset all filters
- **Visual Feedback**: Badges on filter buttons showing selection count
**API:**
```typescript
interface FilterValues {
search?: string;
status?: TaskStatus[];
priority?: TaskPriority[];
dateFrom?: string;
dateTo?: string;
sortBy?: string;
sortOrder?: "asc" | "desc";
}
interface FilterBarProps {
onFilterChange: (filters: FilterValues) => void;
initialFilters?: FilterValues;
debounceMs?: number; // Default: 300ms
}
```
**Usage Example:**
```tsx
import { FilterBar } from "@/components/filters";
import { useState } from "react";
function TaskList() {
const [filters, setFilters] = useState({});
// Fetch tasks with filters
const { data } = useQuery({
queryKey: ["tasks", filters],
queryFn: () => fetchTasks(filters)
});
return (
<div>
<FilterBar
onFilterChange={setFilters}
initialFilters={filters}
debounceMs={300}
/>
{/* Task list rendering */}
</div>
);
}
```
## Test Coverage
### Backend Tests: **72 passing**
- Base Filter DTO: 16 tests ✓
- Query Builder: 23 tests ✓
- Query Tasks DTO: 13 tests ✓
- Common Guards: 20 tests ✓ (existing)
### Frontend Tests: **12 passing**
- FilterBar component: 12 tests ✓
**Total Test Coverage: 84 tests passing**
### Test Categories:
- DTO validation (enum, UUID, string, number types)
- Filter building logic (search, sort, pagination, date ranges)
- Multi-select array handling (status, priority, domain)
- Component rendering and interaction
- Debounced input handling
- Filter state management
- Active filter counting
## API Changes
### Query Parameters (Tasks Endpoint: `GET /api/tasks`)
**New/Enhanced:**
```
?search=urgent # Full-text search
?status=IN_PROGRESS,NOT_STARTED # Multi-select status
?priority=HIGH,MEDIUM # Multi-select priority
?domainId=uuid1,uuid2 # Multi-select domain
?sortBy=priority,dueDate # Multi-field sort
?sortOrder=asc # Sort direction
?dueDateFrom=2024-01-01 # Date range start
?dueDateTo=2024-12-31 # Date range end
?page=2&limit=50 # Pagination
```
**Response Metadata:**
```json
{
"data": [...],
"meta": {
"total": 150,
"page": 2,
"limit": 50,
"totalPages": 3,
"hasNextPage": true,
"hasPrevPage": true
}
}
```
## Integration Points
### Backend Integration:
1. Import `BaseFilterDto` in new query DTOs
2. Use `QueryBuilder` utilities in service layer
3. Transform decorator handles array/single value conversion
4. Prisma queries built consistently across all endpoints
### Frontend Integration:
1. Import `FilterBar` component
2. Pass `onFilterChange` handler
3. Component handles all UI state and debouncing
4. Passes clean filter object to parent component
5. Parent fetches data with filter parameters
## Files Created/Modified
### Created (16 files):
**Backend:**
- `apps/api/src/common/dto/base-filter.dto.ts`
- `apps/api/src/common/dto/base-filter.dto.spec.ts`
- `apps/api/src/common/dto/index.ts`
- `apps/api/src/common/utils/query-builder.ts`
- `apps/api/src/common/utils/query-builder.spec.ts`
- `apps/api/src/common/utils/index.ts`
- `apps/api/src/tasks/dto/query-tasks.dto.spec.ts`
**Frontend:**
- `apps/web/src/components/filters/FilterBar.tsx`
- `apps/web/src/components/filters/FilterBar.test.tsx`
- `apps/web/src/components/filters/index.ts`
### Modified (2 files):
- `apps/api/src/tasks/dto/query-tasks.dto.ts` - Extended BaseFilterDto, added multi-select
- `apps/api/src/tasks/tasks.service.ts` - Uses QueryBuilder utilities
## Technical Decisions
1. **UUID Validation**: Changed from `@IsUUID("4")` to `@IsUUID(undefined)` for broader compatibility
2. **Transform Decorators**: Used to normalize single values to arrays for multi-select filters
3. **Plain HTML + Tailwind**: FilterBar uses native elements instead of complex UI library dependencies
4. **Debouncing**: Implemented in component for better UX on search input
5. **Prisma Query Building**: Centralized in QueryBuilder for consistency and reusability
6. **Test-First Approach**: All features implemented with tests written first (TDD)
## Next Steps / Recommendations
1. **Apply to Other Entities**: Use same pattern for Projects, Events, Knowledge entries
2. **Add More Sort Fields**: Extend sortBy validation to whitelist allowed fields
3. **Cursor Pagination**: Consider adding cursor-based pagination for large datasets
4. **Filter Presets**: Allow saving/loading filter combinations
5. **Advanced Search**: Add support for search operators (AND, OR, NOT)
6. **Performance**: Add database indexes on commonly filtered fields
## Performance Considerations
- Debounced search prevents excessive API calls
- Pagination limits result set size
- Prisma query optimization with proper indexes recommended
- QueryBuilder creates optimized Prisma queries
- Multi-select uses `IN` operator for efficient DB queries
## Accessibility
- Proper ARIA labels on filter buttons
- Keyboard navigation support in dropdowns
- Clear visual feedback for active filters
- Screen reader friendly filter counts
## Commit Message
```
feat(#18): implement advanced filtering and search
Backend:
- Add BaseFilterDto with pagination, search, and sort support
- Create QueryBuilder utility for Prisma query construction
- Enhance QueryTasksDto with multi-select filters
- Update TasksService to use QueryBuilder
- Add comprehensive test coverage (72 passing tests)
Frontend:
- Create FilterBar component with multi-select support
- Implement debounced search input
- Add date range picker
- Support for status, priority, and domain filters
- Add visual feedback with filter counts
- Full test coverage (12 passing tests)
Total: 84 tests passing, 85%+ coverage
```

View File

@@ -0,0 +1,181 @@
# Gantt Chart Implementation Summary
## Issue
**#15: Implement Gantt Chart Component**
- Repository: https://git.mosaicstack.dev/mosaic/stack/issues/15
- Milestone: M3-Features (0.0.3)
- Priority: P0
## Implementation Complete ✅
### Files Created/Modified
#### Component Files
1. **`apps/web/src/components/gantt/GanttChart.tsx`** (299 lines)
- Main Gantt chart component
- Timeline visualization with task bars
- Status-based color coding
- Interactive task selection
- Accessible with ARIA labels
2. **`apps/web/src/components/gantt/types.ts`** (95 lines)
- Type definitions for GanttTask, TimelineRange, etc.
- Helper functions: `toGanttTask()`, `toGanttTasks()`
- Converts base Task to GanttTask with start/end dates
3. **`apps/web/src/components/gantt/index.ts`** (7 lines)
- Module exports
#### Test Files
4. **`apps/web/src/components/gantt/GanttChart.test.tsx`** (401 lines)
- 22 comprehensive component tests
- Tests rendering, interactions, accessibility, edge cases
- Tests PDA-friendly language
5. **`apps/web/src/components/gantt/types.test.ts`** (204 lines)
- 11 tests for helper functions
- Tests type conversions and edge cases
#### Demo Page
6. **`apps/web/src/app/demo/gantt/page.tsx`** (316 lines)
- Interactive demo with sample project data
- Shows 7 sample tasks with various statuses
- Task selection and details display
- Statistics dashboard
### Test Results
```
✓ All 33 tests passing (100%)
- 22 component tests
- 11 helper function tests
Coverage: 96.18% (exceeds 85% requirement)
- GanttChart.tsx: 97.63%
- types.ts: 91.3%
- Overall gantt module: 96.18%
```
### Features Implemented
#### Core Features ✅
- [x] Task name, start date, end date display
- [x] Visual timeline bars
- [x] Status-based color coding
- Completed: Green
- In Progress: Blue
- Paused: Yellow
- Not Started: Gray
- Archived: Light Gray
- [x] Interactive task selection (onClick callback)
- [x] Responsive design with customizable height
#### PDA-Friendly Design ✅
- [x] "Target passed" instead of "OVERDUE"
- [x] "Approaching target" for near-deadline tasks
- [x] Non-judgmental, supportive language throughout
#### Accessibility ✅
- [x] Proper ARIA labels for all interactive elements
- [x] Keyboard navigation support (Tab + Enter)
- [x] Screen reader friendly
- [x] Semantic HTML structure
#### Technical Requirements ✅
- [x] Next.js 16 + React 19
- [x] TypeScript with strict typing (NO `any` types)
- [x] TailwindCSS + Shadcn/ui patterns
- [x] Follows `~/.claude/agent-guides/frontend.md`
- [x] Follows `~/.claude/agent-guides/typescript.md`
#### Testing (TDD) ✅
- [x] Tests written FIRST before implementation
- [x] 96.18% coverage (exceeds 85% requirement)
- [x] All edge cases covered
- [x] Accessibility tests included
### Dependencies (Stretch Goals)
- [ ] Visual dependency lines (planned for future)
- [ ] Drag-to-resize task dates (planned for future)
*Note: Dependencies infrastructure is in place (tasks can have `dependencies` array), but visual rendering is not yet implemented. The `showDependencies` prop is accepted and ready for future implementation.*
### Integration
The component integrates seamlessly with existing Task model from Prisma:
```typescript
// Convert existing tasks to Gantt format
import { toGanttTasks } from '@/components/gantt';
const tasks: Task[] = await fetchTasks();
const ganttTasks = toGanttTasks(tasks);
// Use the component
<GanttChart
tasks={ganttTasks}
onTaskClick={(task) => console.log(task)}
height={500}
/>
```
### Demo
View the interactive demo at: **`/demo/gantt`**
The demo includes:
- 7 sample tasks representing a typical project
- Various task statuses (completed, in-progress, paused, not started)
- Statistics dashboard
- Task selection and detail view
- Toggle for dependencies (UI placeholder)
### Git Commit
```
Branch: feature/15-gantt-chart
Commit: 9ff7718
Message: feat(#15): implement Gantt chart component
```
### Next Steps
1. **Merge to develop** - Ready for code review
2. **Integration testing** - Test with real task data from API
3. **Future enhancements:**
- Implement visual dependency lines
- Add drag-to-resize functionality
- Add task grouping by project
- Add zoom controls for timeline
- Add export to image/PDF
## Blockers/Decisions
### No Blockers ✅
All requirements met without blockers.
### Decisions Made:
1. **Start/End Dates**: Used `metadata.startDate` for flexibility. If not present, falls back to `createdAt`. This avoids schema changes while supporting proper Gantt visualization.
2. **Dependencies**: Stored in `metadata.dependencies` as string array. Visual rendering deferred to future enhancement.
3. **Timeline Range**: Auto-calculated from task dates with 5% padding for better visualization.
4. **Color Scheme**: Based on existing TaskItem component patterns for consistency.
5. **Accessibility**: Full ARIA support with keyboard navigation as first-class feature.
## Summary
**Complete** - Gantt chart component fully implemented with:
- 96.18% test coverage (33 tests passing)
- PDA-friendly language
- Full accessibility support
- Interactive demo page
- Production-ready code
- Strict TypeScript (no `any` types)
- Follows all coding standards
Ready for code review and merge to develop.

163
M3-021-ollama-completion.md Normal file
View File

@@ -0,0 +1,163 @@
# Issue #21: Ollama Integration - Completion Report
**Issue:** https://git.mosaicstack.dev/mosaic/stack/issues/21
**Milestone:** M3-Features (0.0.3)
**Priority:** P1
**Status:** ✅ COMPLETED
## Summary
Successfully implemented Ollama integration for Mosaic Stack, providing local/remote LLM capabilities for AI features including intent classification, summaries, and natural language queries.
## Files Created
### Core Module
- `apps/api/src/ollama/dto/index.ts` - TypeScript DTOs and interfaces (1012 bytes)
- `apps/api/src/ollama/ollama.service.ts` - Service implementation (8855 bytes)
- `apps/api/src/ollama/ollama.controller.ts` - REST API controller (2038 bytes)
- `apps/api/src/ollama/ollama.module.ts` - NestJS module configuration (973 bytes)
### Integration
- `apps/api/src/app.module.ts` - Added OllamaModule to main app imports
## Features Implemented
### Configuration
- ✅ Environment variable based configuration
- `OLLAMA_MODE` - local|remote (default: local)
- `OLLAMA_ENDPOINT` - API endpoint (default: http://localhost:11434)
- `OLLAMA_MODEL` - default model (default: llama3.2)
- `OLLAMA_TIMEOUT` - request timeout in ms (default: 30000)
### Service Methods
-`generate(prompt, options?, model?)` - Text generation from prompts
-`chat(messages, options?, model?)` - Chat conversation completion
-`embed(text, model?)` - Generate text embeddings for vector search
-`listModels()` - List available Ollama models
-`healthCheck()` - Verify Ollama connectivity and status
### API Endpoints
-`POST /ollama/generate` - Text generation
-`POST /ollama/chat` - Chat completion
-`POST /ollama/embed` - Embedding generation
-`GET /ollama/models` - Model listing
-`GET /ollama/health` - Health check
### Error Handling
- ✅ Connection failure handling
- ✅ Request timeout handling with AbortController
- ✅ HTTP error status propagation
- ✅ Typed error responses with HttpException
## Testing
### Test Coverage
**Tests Written (TDD Approach):**
-`ollama.service.spec.ts` - 18 test cases
-`ollama.controller.spec.ts` - 9 test cases
**Total:** 27 tests passing
###Test Categories
- Service configuration and initialization
- Text generation with various options
- Chat completion with message history
- Embedding generation
- Model listing
- Health check (healthy and unhealthy states)
- Error handling (network errors, timeouts, API errors)
- Custom model support
- Options mapping (temperature, max_tokens, stop sequences)
**Coverage:** Comprehensive coverage of all service methods and error paths
## Code Quality
### TypeScript Standards
- ✅ NO `any` types - all functions explicitly typed
- ✅ Explicit return types on all exported functions
- ✅ Proper error type narrowing (`unknown``Error`)
- ✅ Interface definitions for all DTOs
- ✅ Strict null checking compliance
- ✅ Follows `~/.claude/agent-guides/typescript.md`
- ✅ Follows `~/.claude/agent-guides/backend.md`
### NestJS Patterns
- ✅ Proper dependency injection with `@Inject()`
- ✅ Configuration factory pattern
- ✅ Injectable service with `@Injectable()`
- ✅ Controller decorators (`@Controller`, `@Post`, `@Get`, `@Body`)
- ✅ Module exports for service reusability
## Integration Points
### Current
- Health check endpoint available for monitoring
- Service exported from module for use by other modules
- Configuration via environment variables
### Future Ready
- Prepared for Brain query integration
- Service can be injected into other modules
- Embeddings ready for vector search implementation
- Model selection support for different use cases
## Environment Configuration
The `.env.example` file already contains Ollama configuration (no changes needed):
```bash
OLLAMA_MODE=local
OLLAMA_ENDPOINT=http://localhost:11434
OLLAMA_MODEL=llama3.2
OLLAMA_TIMEOUT=30000
```
## Commit
Branch: `feature/21-ollama-integration`
Commit message format:
```
feat(#21): implement Ollama integration
- Created OllamaModule with injectable service
- Support for local and remote Ollama instances
- Configuration via environment variables
- Service methods: generate(), chat(), embed(), listModels()
- Health check endpoint for connectivity verification
- Error handling for connection failures and timeouts
- Request timeout configuration
- Comprehensive unit tests with 100% coverage (27 tests passing)
- Follows TypeScript strict typing guidelines (no any types)
API Endpoints:
- POST /ollama/generate - Text generation
- POST /ollama/chat - Chat completion
- POST /ollama/embed - Embeddings
- GET /ollama/models - List models
- GET /ollama/health - Health check
Refs #21
```
## Next Steps
1. **Testing:** Run integration tests with actual Ollama instance
2. **Documentation:** Add API documentation (Swagger/OpenAPI)
3. **Integration:** Use OllamaService in Brain module for NL queries
4. **Features:** Implement intent classification using chat endpoint
5. **Features:** Add semantic search using embeddings
## Technical Notes
- Uses native `fetch` API (Node.js 18+)
- Implements proper timeout handling with AbortController
- Supports both local (http://localhost:11434) and remote Ollama instances
- Compatible with all Ollama API v1 endpoints
- Designed for easy extension (streaming support can be added later)
---
**Completed By:** Claude (Subagent)
**Date:** 2026-01-29
**Time Spent:** ~30 minutes
**Status:** ✅ Ready for merge

View File

@@ -0,0 +1,31 @@
-- CreateEnum
CREATE TYPE "FormalityLevel" AS ENUM ('VERY_CASUAL', 'CASUAL', 'NEUTRAL', 'FORMAL', 'VERY_FORMAL');
-- CreateTable
CREATE TABLE "personalities" (
"id" UUID NOT NULL,
"workspace_id" UUID NOT NULL,
"name" TEXT NOT NULL,
"description" TEXT,
"tone" TEXT NOT NULL,
"formality_level" "FormalityLevel" NOT NULL DEFAULT 'NEUTRAL',
"system_prompt_template" TEXT NOT NULL,
"is_default" BOOLEAN NOT NULL DEFAULT false,
"is_active" BOOLEAN NOT NULL DEFAULT true,
"created_at" TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
"updated_at" TIMESTAMPTZ NOT NULL,
CONSTRAINT "personalities_pkey" PRIMARY KEY ("id")
);
-- CreateIndex
CREATE INDEX "personalities_workspace_id_idx" ON "personalities"("workspace_id");
-- CreateIndex
CREATE INDEX "personalities_workspace_id_is_default_idx" ON "personalities"("workspace_id", "is_default");
-- CreateIndex
CREATE UNIQUE INDEX "personalities_workspace_id_name_key" ON "personalities"("workspace_id", "name");
-- AddForeignKey
ALTER TABLE "personalities" ADD CONSTRAINT "personalities_workspace_id_fkey" FOREIGN KEY ("workspace_id") REFERENCES "workspaces"("id") ON DELETE CASCADE ON UPDATE CASCADE;

View File

@@ -0,0 +1,41 @@
/*
Warnings:
- You are about to drop the `personalities` table. If the table is not empty, all the data it contains will be lost.
- Added the required column `display_text` to the `knowledge_links` table without a default value. This is not possible if the table is not empty.
- Added the required column `position_end` to the `knowledge_links` table without a default value. This is not possible if the table is not empty.
- Added the required column `position_start` to the `knowledge_links` table without a default value. This is not possible if the table is not empty.
*/
-- DropForeignKey
ALTER TABLE "personalities" DROP CONSTRAINT "personalities_workspace_id_fkey";
-- DropIndex
DROP INDEX "knowledge_links_source_id_target_id_key";
-- AlterTable: Add new columns with temporary defaults for existing records
ALTER TABLE "knowledge_links"
ADD COLUMN "display_text" TEXT DEFAULT '',
ADD COLUMN "position_end" INTEGER DEFAULT 0,
ADD COLUMN "position_start" INTEGER DEFAULT 0,
ADD COLUMN "resolved" BOOLEAN NOT NULL DEFAULT false,
ALTER COLUMN "target_id" DROP NOT NULL;
-- Update existing records: set display_text to link_text and resolved to true if target exists
UPDATE "knowledge_links" SET "display_text" = "link_text" WHERE "display_text" = '';
UPDATE "knowledge_links" SET "resolved" = true WHERE "target_id" IS NOT NULL;
-- Remove defaults for new records
ALTER TABLE "knowledge_links"
ALTER COLUMN "display_text" DROP DEFAULT,
ALTER COLUMN "position_end" DROP DEFAULT,
ALTER COLUMN "position_start" DROP DEFAULT;
-- DropTable
DROP TABLE "personalities";
-- DropEnum
DROP TYPE "FormalityLevel";
-- CreateIndex
CREATE INDEX "knowledge_links_source_id_resolved_idx" ON "knowledge_links"("source_id", "resolved");

View File

@@ -0,0 +1,170 @@
import { describe, expect, it } from "vitest";
import { validate } from "class-validator";
import { plainToClass } from "class-transformer";
import { BaseFilterDto, BasePaginationDto, SortOrder } from "./base-filter.dto";
describe("BasePaginationDto", () => {
it("should accept valid pagination parameters", async () => {
const dto = plainToClass(BasePaginationDto, {
page: 1,
limit: 20,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.page).toBe(1);
expect(dto.limit).toBe(20);
});
it("should use default values when not provided", async () => {
const dto = plainToClass(BasePaginationDto, {});
const errors = await validate(dto);
expect(errors.length).toBe(0);
});
it("should reject page less than 1", async () => {
const dto = plainToClass(BasePaginationDto, {
page: 0,
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors[0].property).toBe("page");
});
it("should reject limit less than 1", async () => {
const dto = plainToClass(BasePaginationDto, {
limit: 0,
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors[0].property).toBe("limit");
});
it("should reject limit greater than 100", async () => {
const dto = plainToClass(BasePaginationDto, {
limit: 101,
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors[0].property).toBe("limit");
});
it("should transform string numbers to integers", async () => {
const dto = plainToClass(BasePaginationDto, {
page: "2" as any,
limit: "30" as any,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.page).toBe(2);
expect(dto.limit).toBe(30);
});
});
describe("BaseFilterDto", () => {
it("should accept valid search parameter", async () => {
const dto = plainToClass(BaseFilterDto, {
search: "test query",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.search).toBe("test query");
});
it("should accept valid sortBy parameter", async () => {
const dto = plainToClass(BaseFilterDto, {
sortBy: "createdAt",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.sortBy).toBe("createdAt");
});
it("should accept valid sortOrder parameter", async () => {
const dto = plainToClass(BaseFilterDto, {
sortOrder: SortOrder.DESC,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.sortOrder).toBe(SortOrder.DESC);
});
it("should reject invalid sortOrder", async () => {
const dto = plainToClass(BaseFilterDto, {
sortOrder: "invalid" as any,
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors.some(e => e.property === "sortOrder")).toBe(true);
});
it("should accept comma-separated sortBy fields", async () => {
const dto = plainToClass(BaseFilterDto, {
sortBy: "priority,createdAt",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.sortBy).toBe("priority,createdAt");
});
it("should accept date range filters", async () => {
const dto = plainToClass(BaseFilterDto, {
dateFrom: "2024-01-01T00:00:00Z",
dateTo: "2024-12-31T23:59:59Z",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
});
it("should reject invalid date format for dateFrom", async () => {
const dto = plainToClass(BaseFilterDto, {
dateFrom: "not-a-date",
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors.some(e => e.property === "dateFrom")).toBe(true);
});
it("should reject invalid date format for dateTo", async () => {
const dto = plainToClass(BaseFilterDto, {
dateTo: "not-a-date",
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors.some(e => e.property === "dateTo")).toBe(true);
});
it("should trim whitespace from search query", async () => {
const dto = plainToClass(BaseFilterDto, {
search: " test query ",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.search).toBe("test query");
});
it("should reject search queries longer than 500 characters", async () => {
const longString = "a".repeat(501);
const dto = plainToClass(BaseFilterDto, {
search: longString,
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors.some(e => e.property === "search")).toBe(true);
});
});

View File

@@ -0,0 +1,82 @@
import {
IsOptional,
IsInt,
Min,
Max,
IsString,
IsEnum,
IsDateString,
MaxLength,
} from "class-validator";
import { Type, Transform } from "class-transformer";
/**
* Enum for sort order
*/
export enum SortOrder {
ASC = "asc",
DESC = "desc",
}
/**
* Base DTO for pagination
*/
export class BasePaginationDto {
@IsOptional()
@Type(() => Number)
@IsInt({ message: "page must be an integer" })
@Min(1, { message: "page must be at least 1" })
page?: number = 1;
@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 = 50;
}
/**
* Base DTO for filtering and sorting
* Provides common filtering capabilities across all entities
*/
export class BaseFilterDto extends BasePaginationDto {
/**
* Full-text search query
* Searches across title, description, and other text fields
*/
@IsOptional()
@IsString({ message: "search must be a string" })
@MaxLength(500, { message: "search must not exceed 500 characters" })
@Transform(({ value }) => (typeof value === "string" ? value.trim() : value))
search?: string;
/**
* Field(s) to sort by
* Can be comma-separated for multi-field sorting (e.g., "priority,createdAt")
*/
@IsOptional()
@IsString({ message: "sortBy must be a string" })
sortBy?: string;
/**
* Sort order (ascending or descending)
*/
@IsOptional()
@IsEnum(SortOrder, { message: "sortOrder must be either 'asc' or 'desc'" })
sortOrder?: SortOrder = SortOrder.DESC;
/**
* Filter by date range - start date
*/
@IsOptional()
@IsDateString({}, { message: "dateFrom must be a valid ISO 8601 date string" })
dateFrom?: Date;
/**
* Filter by date range - end date
*/
@IsOptional()
@IsDateString({}, { message: "dateTo must be a valid ISO 8601 date string" })
dateTo?: Date;
}

View File

@@ -0,0 +1 @@
export * from "./base-filter.dto";

View File

@@ -0,0 +1 @@
export * from "./query-builder";

View File

@@ -0,0 +1,183 @@
import { describe, expect, it } from "vitest";
import { QueryBuilder } from "./query-builder";
import { SortOrder } from "../dto";
describe("QueryBuilder", () => {
describe("buildSearchFilter", () => {
it("should return empty object when search is undefined", () => {
const result = QueryBuilder.buildSearchFilter(undefined, ["title", "description"]);
expect(result).toEqual({});
});
it("should return empty object when search is empty string", () => {
const result = QueryBuilder.buildSearchFilter("", ["title", "description"]);
expect(result).toEqual({});
});
it("should build OR filter for multiple fields", () => {
const result = QueryBuilder.buildSearchFilter("test", ["title", "description"]);
expect(result).toEqual({
OR: [
{ title: { contains: "test", mode: "insensitive" } },
{ description: { contains: "test", mode: "insensitive" } },
],
});
});
it("should handle single field", () => {
const result = QueryBuilder.buildSearchFilter("test", ["title"]);
expect(result).toEqual({
OR: [
{ title: { contains: "test", mode: "insensitive" } },
],
});
});
it("should trim search query", () => {
const result = QueryBuilder.buildSearchFilter(" test ", ["title"]);
expect(result).toEqual({
OR: [
{ title: { contains: "test", mode: "insensitive" } },
],
});
});
});
describe("buildSortOrder", () => {
it("should return default sort when sortBy is undefined", () => {
const result = QueryBuilder.buildSortOrder(undefined, undefined, { createdAt: "desc" });
expect(result).toEqual({ createdAt: "desc" });
});
it("should build single field sort", () => {
const result = QueryBuilder.buildSortOrder("title", SortOrder.ASC);
expect(result).toEqual({ title: "asc" });
});
it("should build multi-field sort", () => {
const result = QueryBuilder.buildSortOrder("priority,dueDate", SortOrder.DESC);
expect(result).toEqual([
{ priority: "desc" },
{ dueDate: "desc" },
]);
});
it("should handle mixed sorting with custom order per field", () => {
const result = QueryBuilder.buildSortOrder("priority:asc,dueDate:desc");
expect(result).toEqual([
{ priority: "asc" },
{ dueDate: "desc" },
]);
});
it("should use default order when not specified per field", () => {
const result = QueryBuilder.buildSortOrder("priority,dueDate", SortOrder.ASC);
expect(result).toEqual([
{ priority: "asc" },
{ dueDate: "asc" },
]);
});
});
describe("buildDateRangeFilter", () => {
it("should return empty object when both dates are undefined", () => {
const result = QueryBuilder.buildDateRangeFilter("createdAt", undefined, undefined);
expect(result).toEqual({});
});
it("should build gte filter when only from date is provided", () => {
const date = new Date("2024-01-01");
const result = QueryBuilder.buildDateRangeFilter("createdAt", date, undefined);
expect(result).toEqual({
createdAt: { gte: date },
});
});
it("should build lte filter when only to date is provided", () => {
const date = new Date("2024-12-31");
const result = QueryBuilder.buildDateRangeFilter("createdAt", undefined, date);
expect(result).toEqual({
createdAt: { lte: date },
});
});
it("should build both gte and lte filters when both dates provided", () => {
const fromDate = new Date("2024-01-01");
const toDate = new Date("2024-12-31");
const result = QueryBuilder.buildDateRangeFilter("createdAt", fromDate, toDate);
expect(result).toEqual({
createdAt: {
gte: fromDate,
lte: toDate,
},
});
});
});
describe("buildInFilter", () => {
it("should return empty object when values is undefined", () => {
const result = QueryBuilder.buildInFilter("status", undefined);
expect(result).toEqual({});
});
it("should return empty object when values is empty array", () => {
const result = QueryBuilder.buildInFilter("status", []);
expect(result).toEqual({});
});
it("should build in filter for single value", () => {
const result = QueryBuilder.buildInFilter("status", ["ACTIVE"]);
expect(result).toEqual({
status: { in: ["ACTIVE"] },
});
});
it("should build in filter for multiple values", () => {
const result = QueryBuilder.buildInFilter("status", ["ACTIVE", "PENDING"]);
expect(result).toEqual({
status: { in: ["ACTIVE", "PENDING"] },
});
});
it("should handle single value as string", () => {
const result = QueryBuilder.buildInFilter("status", "ACTIVE" as any);
expect(result).toEqual({
status: { in: ["ACTIVE"] },
});
});
});
describe("buildPaginationParams", () => {
it("should use default values when not provided", () => {
const result = QueryBuilder.buildPaginationParams(undefined, undefined);
expect(result).toEqual({
skip: 0,
take: 50,
});
});
it("should calculate skip based on page and limit", () => {
const result = QueryBuilder.buildPaginationParams(2, 20);
expect(result).toEqual({
skip: 20,
take: 20,
});
});
it("should handle page 1", () => {
const result = QueryBuilder.buildPaginationParams(1, 25);
expect(result).toEqual({
skip: 0,
take: 25,
});
});
it("should handle large page numbers", () => {
const result = QueryBuilder.buildPaginationParams(10, 50);
expect(result).toEqual({
skip: 450,
take: 50,
});
});
});
});

View File

@@ -0,0 +1,175 @@
import { SortOrder } from "../dto";
/**
* Utility class for building Prisma query filters
* Provides reusable methods for common query operations
*/
export class QueryBuilder {
/**
* Build a full-text search filter across multiple fields
* @param search - Search query string
* @param fields - Fields to search in
* @returns Prisma where clause with OR conditions
*/
static buildSearchFilter(
search: string | undefined,
fields: string[]
): Record<string, any> {
if (!search || search.trim() === "") {
return {};
}
const trimmedSearch = search.trim();
return {
OR: fields.map((field) => ({
[field]: {
contains: trimmedSearch,
mode: "insensitive" as const,
},
})),
};
}
/**
* Build sort order configuration
* Supports single or multi-field sorting with custom order per field
* @param sortBy - Field(s) to sort by (comma-separated)
* @param sortOrder - Default sort order
* @param defaultSort - Fallback sort order if sortBy is undefined
* @returns Prisma orderBy clause
*/
static buildSortOrder(
sortBy?: string,
sortOrder?: SortOrder,
defaultSort?: Record<string, string>
): Record<string, string> | Record<string, string>[] {
if (!sortBy) {
return defaultSort || { createdAt: "desc" };
}
const fields = sortBy.split(",").map((f) => f.trim());
if (fields.length === 1) {
// Check if field has custom order (e.g., "priority:asc")
const [field, customOrder] = fields[0].split(":");
return {
[field]: customOrder || sortOrder || SortOrder.DESC,
};
}
// Multi-field sorting
return fields.map((field) => {
const [fieldName, customOrder] = field.split(":");
return {
[fieldName]: customOrder || sortOrder || SortOrder.DESC,
};
});
}
/**
* Build date range filter
* @param field - Date field name
* @param from - Start date
* @param to - End date
* @returns Prisma where clause with date range
*/
static buildDateRangeFilter(
field: string,
from?: Date,
to?: Date
): Record<string, any> {
if (!from && !to) {
return {};
}
const filter: Record<string, any> = {};
if (from || to) {
filter[field] = {};
if (from) {
filter[field].gte = from;
}
if (to) {
filter[field].lte = to;
}
}
return filter;
}
/**
* Build IN filter for multi-select fields
* @param field - Field name
* @param values - Array of values or single value
* @returns Prisma where clause with IN condition
*/
static buildInFilter<T>(
field: string,
values?: T | T[]
): Record<string, any> {
if (!values) {
return {};
}
const valueArray = Array.isArray(values) ? values : [values];
if (valueArray.length === 0) {
return {};
}
return {
[field]: { in: valueArray },
};
}
/**
* Build pagination parameters
* @param page - Page number (1-indexed)
* @param limit - Items per page
* @returns Prisma skip and take parameters
*/
static buildPaginationParams(
page?: number,
limit?: number
): { skip: number; take: number } {
const actualPage = page || 1;
const actualLimit = limit || 50;
return {
skip: (actualPage - 1) * actualLimit,
take: actualLimit,
};
}
/**
* Build pagination metadata
* @param total - Total count of items
* @param page - Current page
* @param limit - Items per page
* @returns Pagination metadata object
*/
static buildPaginationMeta(
total: number,
page: number,
limit: number
): {
total: number;
page: number;
limit: number;
totalPages: number;
hasNextPage: boolean;
hasPrevPage: boolean;
} {
const totalPages = Math.ceil(total / limit);
return {
total,
page,
limit,
totalPages,
hasNextPage: page < totalPages,
hasPrevPage: page > 1,
};
}
}

View File

@@ -15,6 +15,7 @@ import { AuthGuard } from "../auth/guards/auth.guard";
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
import { Workspace, Permission, RequirePermission } from "../common/decorators";
import { CurrentUser } from "../auth/decorators/current-user.decorator";
import { LinkSyncService } from "./services/link-sync.service";
/**
* Controller for knowledge entry endpoints
@@ -24,7 +25,10 @@ import { CurrentUser } from "../auth/decorators/current-user.decorator";
@Controller("knowledge/entries")
@UseGuards(AuthGuard, WorkspaceGuard, PermissionGuard)
export class KnowledgeController {
constructor(private readonly knowledgeService: KnowledgeService) {}
constructor(
private readonly knowledgeService: KnowledgeService,
private readonly linkSync: LinkSyncService
) {}
/**
* GET /api/knowledge/entries
@@ -100,4 +104,32 @@ export class KnowledgeController {
await this.knowledgeService.remove(workspaceId, slug, user.id);
return { message: "Entry archived successfully" };
}
/**
* GET /api/knowledge/entries/:slug/backlinks
* Get all backlinks for an entry
* Requires: Any workspace member
*/
@Get(":slug/backlinks")
@RequirePermission(Permission.WORKSPACE_ANY)
async getBacklinks(
@Workspace() workspaceId: string,
@Param("slug") slug: string
) {
// First find the entry to get its ID
const entry = await this.knowledgeService.findOne(workspaceId, slug);
// Get backlinks
const backlinks = await this.linkSync.getBacklinks(entry.id);
return {
entry: {
id: entry.id,
slug: entry.slug,
title: entry.title,
},
backlinks,
count: backlinks.length,
};
}
}

View File

@@ -12,13 +12,17 @@ import type {
PaginatedEntries,
} from "./entities/knowledge-entry.entity";
import { renderMarkdown } from "./utils/markdown";
import { LinkSyncService } from "./services/link-sync.service";
/**
* Service for managing knowledge entries
*/
@Injectable()
export class KnowledgeService {
constructor(private readonly prisma: PrismaService) {}
constructor(
private readonly prisma: PrismaService,
private readonly linkSync: LinkSyncService
) {}
/**
@@ -225,6 +229,9 @@ export class KnowledgeService {
throw new Error("Failed to create entry");
}
// Sync wiki links after entry creation
await this.linkSync.syncLinks(workspaceId, result.id, createDto.content);
return {
id: result.id,
workspaceId: result.workspaceId,
@@ -374,6 +381,11 @@ export class KnowledgeService {
throw new Error("Failed to update entry");
}
// Sync wiki links after entry update (only if content changed)
if (updateDto.content !== undefined) {
await this.linkSync.syncLinks(workspaceId, result.id, result.content);
}
return {
id: result.id,
workspaceId: result.workspaceId,

View File

@@ -0,0 +1,410 @@
import { Test, TestingModule } from "@nestjs/testing";
import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
import { LinkSyncService } from "./link-sync.service";
import { LinkResolutionService } from "./link-resolution.service";
import { PrismaService } from "../../prisma/prisma.service";
import * as wikiLinkParser from "../utils/wiki-link-parser";
// Mock the wiki-link parser
vi.mock("../utils/wiki-link-parser");
const mockParseWikiLinks = vi.mocked(wikiLinkParser.parseWikiLinks);
describe("LinkSyncService", () => {
let service: LinkSyncService;
let prisma: PrismaService;
let linkResolver: LinkResolutionService;
const mockWorkspaceId = "workspace-1";
const mockEntryId = "entry-1";
const mockTargetId = "entry-2";
beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
providers: [
LinkSyncService,
{
provide: PrismaService,
useValue: {
knowledgeLink: {
findMany: vi.fn(),
create: vi.fn(),
update: vi.fn(),
delete: vi.fn(),
deleteMany: vi.fn(),
},
$transaction: vi.fn((fn) => fn(prisma)),
},
},
{
provide: LinkResolutionService,
useValue: {
resolveLink: vi.fn(),
resolveLinks: vi.fn(),
},
},
],
}).compile();
service = module.get<LinkSyncService>(LinkSyncService);
prisma = module.get<PrismaService>(PrismaService);
linkResolver = module.get<LinkResolutionService>(LinkResolutionService);
});
afterEach(() => {
vi.clearAllMocks();
});
describe("syncLinks", () => {
it("should be defined", () => {
expect(service).toBeDefined();
});
it("should parse wiki links from content", async () => {
const content = "This is a [[Test Link]] in content";
mockParseWikiLinks.mockReturnValue([
{
raw: "[[Test Link]]",
target: "Test Link",
displayText: "Test Link",
start: 10,
end: 25,
},
]);
vi.spyOn(linkResolver, "resolveLink").mockResolvedValue(mockTargetId);
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([]);
vi.spyOn(prisma.knowledgeLink, "create").mockResolvedValue({} as any);
await service.syncLinks(mockWorkspaceId, mockEntryId, content);
expect(mockParseWikiLinks).toHaveBeenCalledWith(content);
});
it("should create new links when parsing finds wiki links", async () => {
const content = "This is a [[Test Link]] in content";
mockParseWikiLinks.mockReturnValue([
{
raw: "[[Test Link]]",
target: "Test Link",
displayText: "Test Link",
start: 10,
end: 25,
},
]);
vi.spyOn(linkResolver, "resolveLink").mockResolvedValue(mockTargetId);
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([]);
vi.spyOn(prisma.knowledgeLink, "create").mockResolvedValue({
id: "link-1",
sourceId: mockEntryId,
targetId: mockTargetId,
linkText: "Test Link",
displayText: "Test Link",
positionStart: 10,
positionEnd: 25,
resolved: true,
context: null,
createdAt: new Date(),
});
await service.syncLinks(mockWorkspaceId, mockEntryId, content);
expect(prisma.knowledgeLink.create).toHaveBeenCalledWith({
data: {
sourceId: mockEntryId,
targetId: mockTargetId,
linkText: "Test Link",
displayText: "Test Link",
positionStart: 10,
positionEnd: 25,
resolved: true,
},
});
});
it("should create unresolved links when target cannot be found", async () => {
const content = "This is a [[Nonexistent Link]] in content";
mockParseWikiLinks.mockReturnValue([
{
raw: "[[Nonexistent Link]]",
target: "Nonexistent Link",
displayText: "Nonexistent Link",
start: 10,
end: 32,
},
]);
vi.spyOn(linkResolver, "resolveLink").mockResolvedValue(null);
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([]);
vi.spyOn(prisma.knowledgeLink, "create").mockResolvedValue({
id: "link-1",
sourceId: mockEntryId,
targetId: null,
linkText: "Nonexistent Link",
displayText: "Nonexistent Link",
positionStart: 10,
positionEnd: 32,
resolved: false,
context: null,
createdAt: new Date(),
});
await service.syncLinks(mockWorkspaceId, mockEntryId, content);
expect(prisma.knowledgeLink.create).toHaveBeenCalledWith({
data: {
sourceId: mockEntryId,
targetId: null,
linkText: "Nonexistent Link",
displayText: "Nonexistent Link",
positionStart: 10,
positionEnd: 32,
resolved: false,
},
});
});
it("should handle custom display text in links", async () => {
const content = "This is a [[Target|Custom Display]] in content";
mockParseWikiLinks.mockReturnValue([
{
raw: "[[Target|Custom Display]]",
target: "Target",
displayText: "Custom Display",
start: 10,
end: 35,
},
]);
vi.spyOn(linkResolver, "resolveLink").mockResolvedValue(mockTargetId);
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([]);
vi.spyOn(prisma.knowledgeLink, "create").mockResolvedValue({} as any);
await service.syncLinks(mockWorkspaceId, mockEntryId, content);
expect(prisma.knowledgeLink.create).toHaveBeenCalledWith({
data: {
sourceId: mockEntryId,
targetId: mockTargetId,
linkText: "Target",
displayText: "Custom Display",
positionStart: 10,
positionEnd: 35,
resolved: true,
},
});
});
it("should delete orphaned links not present in updated content", async () => {
const content = "This is a [[New Link]] in content";
mockParseWikiLinks.mockReturnValue([
{
raw: "[[New Link]]",
target: "New Link",
displayText: "New Link",
start: 10,
end: 22,
},
]);
// Mock existing link that should be removed
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([
{
id: "old-link-1",
sourceId: mockEntryId,
targetId: "old-target",
linkText: "Old Link",
displayText: "Old Link",
positionStart: 5,
positionEnd: 17,
resolved: true,
context: null,
createdAt: new Date(),
},
] as any);
vi.spyOn(linkResolver, "resolveLink").mockResolvedValue(mockTargetId);
vi.spyOn(prisma.knowledgeLink, "create").mockResolvedValue({} as any);
vi.spyOn(prisma.knowledgeLink, "deleteMany").mockResolvedValue({ count: 1 });
await service.syncLinks(mockWorkspaceId, mockEntryId, content);
expect(prisma.knowledgeLink.deleteMany).toHaveBeenCalledWith({
where: {
sourceId: mockEntryId,
id: {
in: ["old-link-1"],
},
},
});
});
it("should handle empty content by removing all links", async () => {
const content = "";
mockParseWikiLinks.mockReturnValue([]);
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([
{
id: "link-1",
sourceId: mockEntryId,
targetId: mockTargetId,
linkText: "Link",
displayText: "Link",
positionStart: 10,
positionEnd: 18,
resolved: true,
context: null,
createdAt: new Date(),
},
] as any);
vi.spyOn(prisma.knowledgeLink, "deleteMany").mockResolvedValue({ count: 1 });
await service.syncLinks(mockWorkspaceId, mockEntryId, content);
expect(prisma.knowledgeLink.deleteMany).toHaveBeenCalledWith({
where: {
sourceId: mockEntryId,
id: {
in: ["link-1"],
},
},
});
});
it("should handle multiple links in content", async () => {
const content = "Links: [[Link 1]] and [[Link 2]] and [[Link 3]]";
mockParseWikiLinks.mockReturnValue([
{
raw: "[[Link 1]]",
target: "Link 1",
displayText: "Link 1",
start: 7,
end: 17,
},
{
raw: "[[Link 2]]",
target: "Link 2",
displayText: "Link 2",
start: 22,
end: 32,
},
{
raw: "[[Link 3]]",
target: "Link 3",
displayText: "Link 3",
start: 37,
end: 47,
},
]);
vi.spyOn(linkResolver, "resolveLink").mockResolvedValue(mockTargetId);
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([]);
vi.spyOn(prisma.knowledgeLink, "create").mockResolvedValue({} as any);
await service.syncLinks(mockWorkspaceId, mockEntryId, content);
expect(prisma.knowledgeLink.create).toHaveBeenCalledTimes(3);
});
});
describe("getBacklinks", () => {
it("should return all backlinks for an entry", async () => {
const mockBacklinks = [
{
id: "link-1",
sourceId: "source-1",
targetId: mockEntryId,
linkText: "Link Text",
displayText: "Link Text",
positionStart: 10,
positionEnd: 25,
resolved: true,
context: null,
createdAt: new Date(),
source: {
id: "source-1",
title: "Source Entry",
slug: "source-entry",
},
},
];
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue(mockBacklinks as any);
const result = await service.getBacklinks(mockEntryId);
expect(prisma.knowledgeLink.findMany).toHaveBeenCalledWith({
where: {
targetId: mockEntryId,
resolved: true,
},
include: {
source: {
select: {
id: true,
title: true,
slug: true,
},
},
},
orderBy: {
createdAt: "desc",
},
});
expect(result).toEqual(mockBacklinks);
});
it("should return empty array when no backlinks exist", async () => {
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue([]);
const result = await service.getBacklinks(mockEntryId);
expect(result).toEqual([]);
});
});
describe("getUnresolvedLinks", () => {
it("should return all unresolved links for a workspace", async () => {
const mockUnresolvedLinks = [
{
id: "link-1",
sourceId: mockEntryId,
targetId: null,
linkText: "Unresolved Link",
displayText: "Unresolved Link",
positionStart: 10,
positionEnd: 29,
resolved: false,
context: null,
createdAt: new Date(),
},
];
vi.spyOn(prisma.knowledgeLink, "findMany").mockResolvedValue(mockUnresolvedLinks as any);
const result = await service.getUnresolvedLinks(mockWorkspaceId);
expect(prisma.knowledgeLink.findMany).toHaveBeenCalledWith({
where: {
source: {
workspaceId: mockWorkspaceId,
},
resolved: false,
},
include: {
source: {
select: {
id: true,
title: true,
slug: true,
},
},
},
});
expect(result).toEqual(mockUnresolvedLinks);
});
});
});

View File

@@ -0,0 +1,201 @@
import { Injectable } from "@nestjs/common";
import { PrismaService } from "../../prisma/prisma.service";
import { LinkResolutionService } from "./link-resolution.service";
import { parseWikiLinks, WikiLink } from "../utils/wiki-link-parser";
/**
* Represents a backlink to a knowledge entry
*/
export interface Backlink {
id: string;
sourceId: string;
targetId: string;
linkText: string;
displayText: string;
positionStart: number;
positionEnd: number;
resolved: boolean;
context: string | null;
createdAt: Date;
source: {
id: string;
title: string;
slug: string;
};
}
/**
* Represents an unresolved wiki link
*/
export interface UnresolvedLink {
id: string;
sourceId: string;
targetId: string | null;
linkText: string;
displayText: string;
positionStart: number;
positionEnd: number;
resolved: boolean;
context: string | null;
createdAt: Date;
source: {
id: string;
title: string;
slug: string;
};
}
/**
* Service for synchronizing wiki-style links in knowledge entries
*
* Responsibilities:
* - Parse content for wiki links
* - Resolve links to knowledge entries
* - Store/update link records
* - Handle orphaned links
*/
@Injectable()
export class LinkSyncService {
constructor(
private readonly prisma: PrismaService,
private readonly linkResolver: LinkResolutionService
) {}
/**
* Sync links for a knowledge entry
* Parses content, resolves links, and updates the database
*
* @param workspaceId - The workspace scope
* @param entryId - The entry being updated
* @param content - The markdown content to parse
*/
async syncLinks(
workspaceId: string,
entryId: string,
content: string
): Promise<void> {
// Parse wiki links from content
const parsedLinks = parseWikiLinks(content);
// Get existing links for this entry
const existingLinks = await this.prisma.knowledgeLink.findMany({
where: {
sourceId: entryId,
},
});
// Resolve all parsed links
const linkCreations: Array<{
sourceId: string;
targetId: string | null;
linkText: string;
displayText: string;
positionStart: number;
positionEnd: number;
resolved: boolean;
}> = [];
for (const link of parsedLinks) {
const targetId = await this.linkResolver.resolveLink(
workspaceId,
link.target
);
linkCreations.push({
sourceId: entryId,
targetId: targetId,
linkText: link.target,
displayText: link.displayText,
positionStart: link.start,
positionEnd: link.end,
resolved: targetId !== null,
});
}
// Determine which existing links to keep/delete
// We'll use a simple strategy: delete all existing and recreate
// (In production, you might want to diff and only update changed links)
const existingLinkIds = existingLinks.map((link) => link.id);
// Delete all existing links and create new ones in a transaction
await this.prisma.$transaction(async (tx) => {
// Delete all existing links
if (existingLinkIds.length > 0) {
await tx.knowledgeLink.deleteMany({
where: {
sourceId: entryId,
id: {
in: existingLinkIds,
},
},
});
}
// Create new links
for (const linkData of linkCreations) {
await tx.knowledgeLink.create({
data: linkData,
});
}
});
}
/**
* Get all backlinks for an entry
* Returns entries that link TO this entry
*
* @param entryId - The target entry
* @returns Array of backlinks with source entry information
*/
async getBacklinks(entryId: string): Promise<Backlink[]> {
const backlinks = await this.prisma.knowledgeLink.findMany({
where: {
targetId: entryId,
resolved: true,
},
include: {
source: {
select: {
id: true,
title: true,
slug: true,
},
},
},
orderBy: {
createdAt: "desc",
},
});
return backlinks as Backlink[];
}
/**
* Get all unresolved links for a workspace
* Useful for finding broken links or pages that need to be created
*
* @param workspaceId - The workspace scope
* @returns Array of unresolved links
*/
async getUnresolvedLinks(workspaceId: string): Promise<UnresolvedLink[]> {
const unresolvedLinks = await this.prisma.knowledgeLink.findMany({
where: {
source: {
workspaceId,
},
resolved: false,
},
include: {
source: {
select: {
id: true,
title: true,
slug: true,
},
},
},
});
return unresolvedLinks as UnresolvedLink[];
}
}

View File

@@ -0,0 +1,43 @@
import { IsString, IsIn, IsOptional, IsBoolean, MinLength, MaxLength } from "class-validator";
export const FORMALITY_LEVELS = [
"VERY_CASUAL",
"CASUAL",
"NEUTRAL",
"FORMAL",
"VERY_FORMAL",
] as const;
export type FormalityLevel = (typeof FORMALITY_LEVELS)[number];
export class CreatePersonalityDto {
@IsString()
@MinLength(1)
@MaxLength(100)
name!: string;
@IsOptional()
@IsString()
@MaxLength(500)
description?: string;
@IsString()
@MinLength(1)
@MaxLength(50)
tone!: string;
@IsIn(FORMALITY_LEVELS)
formalityLevel!: FormalityLevel;
@IsString()
@MinLength(10)
systemPromptTemplate!: string;
@IsOptional()
@IsBoolean()
isDefault?: boolean;
@IsOptional()
@IsBoolean()
isActive?: boolean;
}

View File

@@ -0,0 +1,2 @@
export * from "./create-personality.dto";
export * from "./update-personality.dto";

View File

@@ -0,0 +1,4 @@
import { PartialType } from "@nestjs/mapped-types";
import { CreatePersonalityDto } from "./create-personality.dto";
export class UpdatePersonalityDto extends PartialType(CreatePersonalityDto) {}

View File

@@ -0,0 +1,15 @@
import { Personality as PrismaPersonality, FormalityLevel } from "@prisma/client";
export class Personality implements PrismaPersonality {
id!: string;
workspaceId!: string;
name!: string;
description!: string | null;
tone!: string;
formalityLevel!: FormalityLevel;
systemPromptTemplate!: string;
isDefault!: boolean;
isActive!: boolean;
createdAt!: Date;
updatedAt!: Date;
}

View File

@@ -0,0 +1,157 @@
import { describe, it, expect, beforeEach, vi } from "vitest";
import { Test, TestingModule } from "@nestjs/testing";
import { PersonalitiesController } from "./personalities.controller";
import { PersonalitiesService } from "./personalities.service";
import { AuthGuard } from "../auth/guards/auth.guard";
import { CreatePersonalityDto, UpdatePersonalityDto } from "./dto";
describe("PersonalitiesController", () => {
let controller: PersonalitiesController;
let service: PersonalitiesService;
const mockWorkspaceId = "workspace-123";
const mockUserId = "user-123";
const mockPersonalityId = "personality-123";
const mockRequest = {
user: { id: mockUserId },
workspaceId: mockWorkspaceId,
};
const mockPersonality = {
id: mockPersonalityId,
workspaceId: mockWorkspaceId,
name: "Professional",
description: "Professional communication style",
tone: "professional",
formalityLevel: "FORMAL" as const,
systemPromptTemplate: "You are a professional assistant.",
isDefault: true,
isActive: true,
createdAt: new Date(),
updatedAt: new Date(),
};
const mockPersonalitiesService = {
findAll: vi.fn(),
findOne: vi.fn(),
findDefault: vi.fn(),
create: vi.fn(),
update: vi.fn(),
remove: vi.fn(),
};
const mockAuthGuard = {
canActivate: vi.fn().mockReturnValue(true),
};
beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
controllers: [PersonalitiesController],
providers: [
{
provide: PersonalitiesService,
useValue: mockPersonalitiesService,
},
],
})
.overrideGuard(AuthGuard)
.useValue(mockAuthGuard)
.compile();
controller = module.get<PersonalitiesController>(PersonalitiesController);
service = module.get<PersonalitiesService>(PersonalitiesService);
// Reset mocks
vi.clearAllMocks();
});
describe("findAll", () => {
it("should return all personalities", async () => {
const mockPersonalities = [mockPersonality];
mockPersonalitiesService.findAll.mockResolvedValue(mockPersonalities);
const result = await controller.findAll(mockRequest as any);
expect(result).toEqual(mockPersonalities);
expect(service.findAll).toHaveBeenCalledWith(mockWorkspaceId, true);
});
it("should filter by active status", async () => {
mockPersonalitiesService.findAll.mockResolvedValue([mockPersonality]);
await controller.findAll(mockRequest as any, false);
expect(service.findAll).toHaveBeenCalledWith(mockWorkspaceId, false);
});
});
describe("findOne", () => {
it("should return a personality by id", async () => {
mockPersonalitiesService.findOne.mockResolvedValue(mockPersonality);
const result = await controller.findOne(mockRequest as any, mockPersonalityId);
expect(result).toEqual(mockPersonality);
expect(service.findOne).toHaveBeenCalledWith(mockWorkspaceId, mockPersonalityId);
});
});
describe("findDefault", () => {
it("should return the default personality", async () => {
mockPersonalitiesService.findDefault.mockResolvedValue(mockPersonality);
const result = await controller.findDefault(mockRequest as any);
expect(result).toEqual(mockPersonality);
expect(service.findDefault).toHaveBeenCalledWith(mockWorkspaceId);
});
});
describe("create", () => {
const createDto: CreatePersonalityDto = {
name: "Casual",
description: "Casual communication style",
tone: "casual",
formalityLevel: "CASUAL",
systemPromptTemplate: "You are a casual assistant.",
};
it("should create a new personality", async () => {
const newPersonality = { ...mockPersonality, ...createDto, id: "new-id" };
mockPersonalitiesService.create.mockResolvedValue(newPersonality);
const result = await controller.create(mockRequest as any, createDto);
expect(result).toEqual(newPersonality);
expect(service.create).toHaveBeenCalledWith(mockWorkspaceId, createDto);
});
});
describe("update", () => {
const updateDto: UpdatePersonalityDto = {
description: "Updated description",
};
it("should update a personality", async () => {
const updatedPersonality = { ...mockPersonality, ...updateDto };
mockPersonalitiesService.update.mockResolvedValue(updatedPersonality);
const result = await controller.update(mockRequest as any, mockPersonalityId, updateDto);
expect(result).toEqual(updatedPersonality);
expect(service.update).toHaveBeenCalledWith(mockWorkspaceId, mockPersonalityId, updateDto);
});
});
describe("remove", () => {
it("should delete a personality", async () => {
mockPersonalitiesService.remove.mockResolvedValue(mockPersonality);
const result = await controller.remove(mockRequest as any, mockPersonalityId);
expect(result).toEqual(mockPersonality);
expect(service.remove).toHaveBeenCalledWith(mockWorkspaceId, mockPersonalityId);
});
});
});

View File

@@ -0,0 +1,77 @@
import {
Controller,
Get,
Post,
Put,
Delete,
Body,
Param,
Query,
UseGuards,
Req,
} from "@nestjs/common";
import { AuthGuard } from "../auth/guards/auth.guard";
import { PersonalitiesService } from "./personalities.service";
import { CreatePersonalityDto, UpdatePersonalityDto } from "./dto";
import { Personality } from "./entities/personality.entity";
@Controller("personalities")
@UseGuards(AuthGuard)
export class PersonalitiesController {
constructor(private readonly personalitiesService: PersonalitiesService) {}
/**
* Get all personalities for the current workspace
*/
@Get()
async findAll(
@Req() req: any,
@Query("isActive") isActive: boolean = true,
): Promise<Personality[]> {
return this.personalitiesService.findAll(req.workspaceId, isActive);
}
/**
* Get the default personality for the current workspace
*/
@Get("default")
async findDefault(@Req() req: any): Promise<Personality> {
return this.personalitiesService.findDefault(req.workspaceId);
}
/**
* Get a specific personality by ID
*/
@Get(":id")
async findOne(@Req() req: any, @Param("id") id: string): Promise<Personality> {
return this.personalitiesService.findOne(req.workspaceId, id);
}
/**
* Create a new personality
*/
@Post()
async create(@Req() req: any, @Body() dto: CreatePersonalityDto): Promise<Personality> {
return this.personalitiesService.create(req.workspaceId, dto);
}
/**
* Update an existing personality
*/
@Put(":id")
async update(
@Req() req: any,
@Param("id") id: string,
@Body() dto: UpdatePersonalityDto,
): Promise<Personality> {
return this.personalitiesService.update(req.workspaceId, id, dto);
}
/**
* Delete a personality
*/
@Delete(":id")
async remove(@Req() req: any, @Param("id") id: string): Promise<Personality> {
return this.personalitiesService.remove(req.workspaceId, id);
}
}

View File

@@ -0,0 +1,13 @@
import { Module } from "@nestjs/common";
import { PrismaModule } from "../prisma/prisma.module";
import { AuthModule } from "../auth/auth.module";
import { PersonalitiesService } from "./personalities.service";
import { PersonalitiesController } from "./personalities.controller";
@Module({
imports: [PrismaModule, AuthModule],
controllers: [PersonalitiesController],
providers: [PersonalitiesService],
exports: [PersonalitiesService],
})
export class PersonalitiesModule {}

View File

@@ -0,0 +1,255 @@
import { describe, it, expect, beforeEach, vi } from "vitest";
import { Test, TestingModule } from "@nestjs/testing";
import { PersonalitiesService } from "./personalities.service";
import { PrismaService } from "../prisma/prisma.service";
import { CreatePersonalityDto, UpdatePersonalityDto } from "./dto";
import { NotFoundException, ConflictException } from "@nestjs/common";
describe("PersonalitiesService", () => {
let service: PersonalitiesService;
let prisma: PrismaService;
const mockWorkspaceId = "workspace-123";
const mockUserId = "user-123";
const mockPersonalityId = "personality-123";
const mockPersonality = {
id: mockPersonalityId,
workspaceId: mockWorkspaceId,
name: "Professional",
description: "Professional communication style",
tone: "professional",
formalityLevel: "FORMAL" as const,
systemPromptTemplate: "You are a professional assistant.",
isDefault: true,
isActive: true,
createdAt: new Date(),
updatedAt: new Date(),
};
const mockPrismaService = {
personality: {
findMany: vi.fn(),
findUnique: vi.fn(),
findFirst: vi.fn(),
create: vi.fn(),
update: vi.fn(),
delete: vi.fn(),
count: vi.fn(),
},
$transaction: vi.fn((callback) => callback(mockPrismaService)),
};
beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
providers: [
PersonalitiesService,
{
provide: PrismaService,
useValue: mockPrismaService,
},
],
}).compile();
service = module.get<PersonalitiesService>(PersonalitiesService);
prisma = module.get<PrismaService>(PrismaService);
// Reset mocks
vi.clearAllMocks();
});
describe("findAll", () => {
it("should return all personalities for a workspace", async () => {
const mockPersonalities = [mockPersonality];
mockPrismaService.personality.findMany.mockResolvedValue(mockPersonalities);
const result = await service.findAll(mockWorkspaceId);
expect(result).toEqual(mockPersonalities);
expect(prisma.personality.findMany).toHaveBeenCalledWith({
where: { workspaceId: mockWorkspaceId, isActive: true },
orderBy: [{ isDefault: "desc" }, { name: "asc" }],
});
});
it("should filter by active status", async () => {
mockPrismaService.personality.findMany.mockResolvedValue([mockPersonality]);
await service.findAll(mockWorkspaceId, false);
expect(prisma.personality.findMany).toHaveBeenCalledWith({
where: { workspaceId: mockWorkspaceId, isActive: false },
orderBy: [{ isDefault: "desc" }, { name: "asc" }],
});
});
});
describe("findOne", () => {
it("should return a personality by id", async () => {
mockPrismaService.personality.findUnique.mockResolvedValue(mockPersonality);
const result = await service.findOne(mockWorkspaceId, mockPersonalityId);
expect(result).toEqual(mockPersonality);
expect(prisma.personality.findUnique).toHaveBeenCalledWith({
where: {
id: mockPersonalityId,
workspaceId: mockWorkspaceId,
},
});
});
it("should throw NotFoundException when personality not found", async () => {
mockPrismaService.personality.findUnique.mockResolvedValue(null);
await expect(service.findOne(mockWorkspaceId, mockPersonalityId)).rejects.toThrow(
NotFoundException,
);
});
});
describe("findDefault", () => {
it("should return the default personality", async () => {
mockPrismaService.personality.findFirst.mockResolvedValue(mockPersonality);
const result = await service.findDefault(mockWorkspaceId);
expect(result).toEqual(mockPersonality);
expect(prisma.personality.findFirst).toHaveBeenCalledWith({
where: { workspaceId: mockWorkspaceId, isDefault: true, isActive: true },
});
});
it("should throw NotFoundException when no default personality exists", async () => {
mockPrismaService.personality.findFirst.mockResolvedValue(null);
await expect(service.findDefault(mockWorkspaceId)).rejects.toThrow(NotFoundException);
});
});
describe("create", () => {
const createDto: CreatePersonalityDto = {
name: "Casual",
description: "Casual communication style",
tone: "casual",
formalityLevel: "CASUAL",
systemPromptTemplate: "You are a casual assistant.",
};
it("should create a new personality", async () => {
mockPrismaService.personality.findFirst.mockResolvedValue(null);
mockPrismaService.personality.create.mockResolvedValue({
...mockPersonality,
...createDto,
id: "new-personality-id",
});
const result = await service.create(mockWorkspaceId, createDto);
expect(result).toMatchObject(createDto);
expect(prisma.personality.create).toHaveBeenCalledWith({
data: {
workspaceId: mockWorkspaceId,
...createDto,
},
});
});
it("should throw ConflictException when name already exists", async () => {
mockPrismaService.personality.findFirst.mockResolvedValue(mockPersonality);
await expect(service.create(mockWorkspaceId, createDto)).rejects.toThrow(ConflictException);
});
it("should unset other defaults when creating a new default personality", async () => {
const createDefaultDto = { ...createDto, isDefault: true };
// First call to findFirst checks for name conflict (should be null)
// Second call to findFirst finds the existing default personality
mockPrismaService.personality.findFirst
.mockResolvedValueOnce(null) // No name conflict
.mockResolvedValueOnce(mockPersonality); // Existing default
mockPrismaService.personality.update.mockResolvedValue({
...mockPersonality,
isDefault: false,
});
mockPrismaService.personality.create.mockResolvedValue({
...mockPersonality,
...createDefaultDto,
});
await service.create(mockWorkspaceId, createDefaultDto);
expect(prisma.personality.update).toHaveBeenCalledWith({
where: { id: mockPersonalityId },
data: { isDefault: false },
});
});
});
describe("update", () => {
const updateDto: UpdatePersonalityDto = {
description: "Updated description",
tone: "updated",
};
it("should update a personality", async () => {
mockPrismaService.personality.findUnique.mockResolvedValue(mockPersonality);
mockPrismaService.personality.findFirst.mockResolvedValue(null);
mockPrismaService.personality.update.mockResolvedValue({
...mockPersonality,
...updateDto,
});
const result = await service.update(mockWorkspaceId, mockPersonalityId, updateDto);
expect(result).toMatchObject(updateDto);
expect(prisma.personality.update).toHaveBeenCalledWith({
where: { id: mockPersonalityId },
data: updateDto,
});
});
it("should throw NotFoundException when personality not found", async () => {
mockPrismaService.personality.findUnique.mockResolvedValue(null);
await expect(
service.update(mockWorkspaceId, mockPersonalityId, updateDto),
).rejects.toThrow(NotFoundException);
});
it("should throw ConflictException when updating to existing name", async () => {
const updateNameDto = { name: "Existing Name" };
mockPrismaService.personality.findUnique.mockResolvedValue(mockPersonality);
mockPrismaService.personality.findFirst.mockResolvedValue({
...mockPersonality,
id: "different-id",
});
await expect(
service.update(mockWorkspaceId, mockPersonalityId, updateNameDto),
).rejects.toThrow(ConflictException);
});
});
describe("remove", () => {
it("should delete a personality", async () => {
mockPrismaService.personality.findUnique.mockResolvedValue(mockPersonality);
mockPrismaService.personality.delete.mockResolvedValue(mockPersonality);
const result = await service.remove(mockWorkspaceId, mockPersonalityId);
expect(result).toEqual(mockPersonality);
expect(prisma.personality.delete).toHaveBeenCalledWith({
where: { id: mockPersonalityId },
});
});
it("should throw NotFoundException when personality not found", async () => {
mockPrismaService.personality.findUnique.mockResolvedValue(null);
await expect(service.remove(mockWorkspaceId, mockPersonalityId)).rejects.toThrow(
NotFoundException,
);
});
});
});

View File

@@ -0,0 +1,156 @@
import {
Injectable,
NotFoundException,
ConflictException,
Logger,
} from "@nestjs/common";
import { PrismaService } from "../prisma/prisma.service";
import { CreatePersonalityDto, UpdatePersonalityDto } from "./dto";
import { Personality } from "./entities/personality.entity";
@Injectable()
export class PersonalitiesService {
private readonly logger = new Logger(PersonalitiesService.name);
constructor(private readonly prisma: PrismaService) {}
/**
* Find all personalities for a workspace
*/
async findAll(workspaceId: string, isActive: boolean = true): Promise<Personality[]> {
return this.prisma.personality.findMany({
where: { workspaceId, isActive },
orderBy: [{ isDefault: "desc" }, { name: "asc" }],
});
}
/**
* Find a specific personality by ID
*/
async findOne(workspaceId: string, id: string): Promise<Personality> {
const personality = await this.prisma.personality.findUnique({
where: { id, workspaceId },
});
if (!personality) {
throw new NotFoundException(`Personality with ID ${id} not found`);
}
return personality;
}
/**
* Find the default personality for a workspace
*/
async findDefault(workspaceId: string): Promise<Personality> {
const personality = await this.prisma.personality.findFirst({
where: { workspaceId, isDefault: true, isActive: true },
});
if (!personality) {
throw new NotFoundException(`No default personality found for workspace ${workspaceId}`);
}
return personality;
}
/**
* Create a new personality
*/
async create(workspaceId: string, dto: CreatePersonalityDto): Promise<Personality> {
// Check for duplicate name
const existing = await this.prisma.personality.findFirst({
where: { workspaceId, name: dto.name },
});
if (existing) {
throw new ConflictException(`Personality with name "${dto.name}" already exists`);
}
// If creating a default personality, unset other defaults
if (dto.isDefault) {
await this.unsetOtherDefaults(workspaceId);
}
const personality = await this.prisma.personality.create({
data: {
workspaceId,
...dto,
},
});
this.logger.log(`Created personality ${personality.id} for workspace ${workspaceId}`);
return personality;
}
/**
* Update an existing personality
*/
async update(
workspaceId: string,
id: string,
dto: UpdatePersonalityDto,
): Promise<Personality> {
// Check existence
await this.findOne(workspaceId, id);
// Check for duplicate name if updating name
if (dto.name) {
const existing = await this.prisma.personality.findFirst({
where: { workspaceId, name: dto.name, id: { not: id } },
});
if (existing) {
throw new ConflictException(`Personality with name "${dto.name}" already exists`);
}
}
// If setting as default, unset other defaults
if (dto.isDefault === true) {
await this.unsetOtherDefaults(workspaceId, id);
}
const personality = await this.prisma.personality.update({
where: { id },
data: dto,
});
this.logger.log(`Updated personality ${id} for workspace ${workspaceId}`);
return personality;
}
/**
* Delete a personality
*/
async remove(workspaceId: string, id: string): Promise<Personality> {
// Check existence
await this.findOne(workspaceId, id);
const personality = await this.prisma.personality.delete({
where: { id },
});
this.logger.log(`Deleted personality ${id} from workspace ${workspaceId}`);
return personality;
}
/**
* Unset the default flag on all other personalities in the workspace
*/
private async unsetOtherDefaults(workspaceId: string, excludeId?: string): Promise<void> {
const currentDefault = await this.prisma.personality.findFirst({
where: {
workspaceId,
isDefault: true,
...(excludeId && { id: { not: excludeId } }),
},
});
if (currentDefault) {
await this.prisma.personality.update({
where: { id: currentDefault.id },
data: { isDefault: false },
});
}
}
}

View File

@@ -0,0 +1,168 @@
import { describe, expect, it } from "vitest";
import { validate } from "class-validator";
import { plainToClass } from "class-transformer";
import { QueryTasksDto } from "./query-tasks.dto";
import { TaskStatus, TaskPriority } from "@prisma/client";
import { SortOrder } from "../../common/dto";
describe("QueryTasksDto", () => {
const validWorkspaceId = "123e4567-e89b-12d3-a456-426614174000";
it("should accept valid workspaceId", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
});
it("should reject invalid workspaceId", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: "not-a-uuid",
});
const errors = await validate(dto);
expect(errors.length).toBeGreaterThan(0);
expect(errors.some(e => e.property === "workspaceId")).toBe(true);
});
it("should accept valid status filter", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
status: TaskStatus.IN_PROGRESS,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(Array.isArray(dto.status)).toBe(true);
expect(dto.status).toEqual([TaskStatus.IN_PROGRESS]);
});
it("should accept multiple status filters", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
status: [TaskStatus.IN_PROGRESS, TaskStatus.NOT_STARTED],
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(Array.isArray(dto.status)).toBe(true);
expect(dto.status).toHaveLength(2);
});
it("should accept valid priority filter", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
priority: TaskPriority.HIGH,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(Array.isArray(dto.priority)).toBe(true);
expect(dto.priority).toEqual([TaskPriority.HIGH]);
});
it("should accept multiple priority filters", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
priority: [TaskPriority.HIGH, TaskPriority.LOW],
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(Array.isArray(dto.priority)).toBe(true);
expect(dto.priority).toHaveLength(2);
});
it("should accept search parameter", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
search: "test task",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.search).toBe("test task");
});
it("should accept sortBy parameter", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
sortBy: "priority,dueDate",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.sortBy).toBe("priority,dueDate");
});
it("should accept sortOrder parameter", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
sortOrder: SortOrder.ASC,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(dto.sortOrder).toBe(SortOrder.ASC);
});
it("should accept domainId filter", async () => {
const domainId = "123e4567-e89b-12d3-a456-426614174001";
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
domainId,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(Array.isArray(dto.domainId)).toBe(true);
expect(dto.domainId).toEqual([domainId]);
});
it("should accept multiple domainId filters", async () => {
const domainIds = [
"123e4567-e89b-12d3-a456-426614174001",
"123e4567-e89b-12d3-a456-426614174002",
];
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
domainId: domainIds,
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
expect(Array.isArray(dto.domainId)).toBe(true);
expect(dto.domainId).toHaveLength(2);
});
it("should accept date range filters", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
dueDateFrom: "2024-01-01T00:00:00Z",
dueDateTo: "2024-12-31T23:59:59Z",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
});
it("should accept all filters combined", async () => {
const dto = plainToClass(QueryTasksDto, {
workspaceId: validWorkspaceId,
status: [TaskStatus.IN_PROGRESS, TaskStatus.NOT_STARTED],
priority: [TaskPriority.HIGH, TaskPriority.MEDIUM],
search: "urgent task",
sortBy: "priority,dueDate",
sortOrder: SortOrder.ASC,
page: 2,
limit: 25,
dueDateFrom: "2024-01-01T00:00:00Z",
dueDateTo: "2024-12-31T23:59:59Z",
});
const errors = await validate(dto);
expect(errors.length).toBe(0);
});
});

View File

@@ -0,0 +1,80 @@
"use client";
import { useState, useEffect } from "react";
import type { Domain } from "@mosaic/shared";
import { DomainList } from "@/components/domains/DomainList";
import { fetchDomains, createDomain, updateDomain, deleteDomain } from "@/lib/api/domains";
export default function DomainsPage(): JSX.Element {
const [domains, setDomains] = useState<Domain[]>([]);
const [isLoading, setIsLoading] = useState<boolean>(true);
const [error, setError] = useState<string | null>(null);
useEffect(() => {
loadDomains();
}, []);
async function loadDomains(): Promise<void> {
try {
setIsLoading(true);
const response = await fetchDomains();
setDomains(response.data);
setError(null);
} catch (err) {
setError(err instanceof Error ? err.message : "Failed to load domains");
} finally {
setIsLoading(false);
}
}
function handleEdit(domain: Domain): void {
// TODO: Open edit modal/form
console.log("Edit domain:", domain);
}
async function handleDelete(domain: Domain): Promise<void> {
if (!confirm(`Are you sure you want to delete "${domain.name}"?`)) {
return;
}
try {
await deleteDomain(domain.id);
await loadDomains();
} catch (err) {
setError(err instanceof Error ? err.message : "Failed to delete domain");
}
}
return (
<div className="max-w-6xl mx-auto p-6">
<div className="mb-6">
<h1 className="text-3xl font-bold mb-2">Domains</h1>
<p className="text-gray-600">
Organize your tasks and projects by life areas
</p>
</div>
{error && (
<div className="mb-6 p-4 bg-red-50 border border-red-200 rounded text-red-700">
{error}
</div>
)}
<div className="mb-6">
<button
className="px-4 py-2 bg-gray-900 text-white rounded hover:bg-gray-800"
onClick={() => console.log("TODO: Open create modal")}
>
Create Domain
</button>
</div>
<DomainList
domains={domains}
isLoading={isLoading}
onEdit={handleEdit}
onDelete={handleDelete}
/>
</div>
);
}

View File

@@ -0,0 +1,263 @@
"use client";
import { useState, useEffect } from "react";
import type { Personality } from "@mosaic/shared";
import { PersonalityPreview } from "@/components/personalities/PersonalityPreview";
import { PersonalityForm, PersonalityFormData } from "@/components/personalities/PersonalityForm";
import {
fetchPersonalities,
createPersonality,
updatePersonality,
deletePersonality,
} from "@/lib/api/personalities";
import { Button } from "@/components/ui/button";
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
import { Badge } from "@/components/ui/badge";
import { Plus, Pencil, Trash2, Eye } from "lucide-react";
import {
AlertDialog,
AlertDialogAction,
AlertDialogCancel,
AlertDialogContent,
AlertDialogDescription,
AlertDialogFooter,
AlertDialogHeader,
AlertDialogTitle,
} from "@/components/ui/alert-dialog";
export default function PersonalitiesPage(): JSX.Element {
const [personalities, setPersonalities] = useState<Personality[]>([]);
const [selectedPersonality, setSelectedPersonality] = useState<Personality | null>(null);
const [isLoading, setIsLoading] = useState<boolean>(true);
const [error, setError] = useState<string | null>(null);
const [mode, setMode] = useState<"list" | "create" | "edit" | "preview">("list");
const [deleteTarget, setDeleteTarget] = useState<Personality | null>(null);
useEffect(() => {
loadPersonalities();
}, []);
async function loadPersonalities(): Promise<void> {
try {
setIsLoading(true);
const response = await fetchPersonalities();
setPersonalities(response.data);
setError(null);
} catch (err) {
setError(err instanceof Error ? err.message : "Failed to load personalities");
} finally {
setIsLoading(false);
}
}
async function handleCreate(data: PersonalityFormData): Promise<void> {
try {
await createPersonality(data);
await loadPersonalities();
setMode("list");
setError(null);
} catch (err) {
setError(err instanceof Error ? err.message : "Failed to create personality");
throw err;
}
}
async function handleUpdate(data: PersonalityFormData): Promise<void> {
if (!selectedPersonality) return;
try {
await updatePersonality(selectedPersonality.id, data);
await loadPersonalities();
setMode("list");
setSelectedPersonality(null);
setError(null);
} catch (err) {
setError(err instanceof Error ? err.message : "Failed to update personality");
throw err;
}
}
async function confirmDelete(): Promise<void> {
if (!deleteTarget) return;
try {
await deletePersonality(deleteTarget.id);
await loadPersonalities();
setDeleteTarget(null);
setError(null);
} catch (err) {
setError(err instanceof Error ? err.message : "Failed to delete personality");
}
}
if (mode === "create") {
return (
<div className="max-w-4xl mx-auto p-6">
<PersonalityForm
onSubmit={handleCreate}
onCancel={() => setMode("list")}
/>
</div>
);
}
if (mode === "edit" && selectedPersonality) {
return (
<div className="max-w-4xl mx-auto p-6">
<PersonalityForm
personality={selectedPersonality}
onSubmit={handleUpdate}
onCancel={() => {
setMode("list");
setSelectedPersonality(null);
}}
/>
</div>
);
}
if (mode === "preview" && selectedPersonality) {
return (
<div className="max-w-4xl mx-auto p-6">
<div className="mb-4">
<Button
variant="outline"
onClick={() => {
setMode("list");
setSelectedPersonality(null);
}}
>
Back to List
</Button>
</div>
<PersonalityPreview personality={selectedPersonality} />
</div>
);
}
return (
<div className="max-w-6xl mx-auto p-6">
{/* Header */}
<div className="mb-6">
<div className="flex items-center justify-between">
<div>
<h1 className="text-3xl font-bold">AI Personalities</h1>
<p className="text-muted-foreground mt-1">
Customize how the AI assistant communicates and responds
</p>
</div>
<Button onClick={() => setMode("create")}>
<Plus className="mr-2 h-4 w-4" />
New Personality
</Button>
</div>
</div>
{/* Error Display */}
{error && (
<div className="mb-4 p-4 bg-destructive/10 text-destructive rounded-md">
{error}
</div>
)}
{/* Loading State */}
{isLoading ? (
<div className="text-center py-12">
<p className="text-muted-foreground">Loading personalities...</p>
</div>
) : personalities.length === 0 ? (
<Card>
<CardContent className="flex flex-col items-center justify-center py-12">
<p className="text-muted-foreground mb-4">No personalities found</p>
<Button onClick={() => setMode("create")}>
<Plus className="mr-2 h-4 w-4" />
Create First Personality
</Button>
</CardContent>
</Card>
) : (
<div className="grid gap-4">
{personalities.map((personality) => (
<Card key={personality.id}>
<CardHeader>
<div className="flex items-start justify-between">
<div>
<CardTitle className="flex items-center gap-2">
{personality.name}
{personality.isDefault && (
<Badge variant="secondary">Default</Badge>
)}
{!personality.isActive && (
<Badge variant="outline">Inactive</Badge>
)}
</CardTitle>
<CardDescription>{personality.description}</CardDescription>
</div>
<div className="flex gap-2">
<Button
variant="ghost"
size="sm"
onClick={() => {
setSelectedPersonality(personality);
setMode("preview");
}}
>
<Eye className="h-4 w-4" />
</Button>
<Button
variant="ghost"
size="sm"
onClick={() => {
setSelectedPersonality(personality);
setMode("edit");
}}
>
<Pencil className="h-4 w-4" />
</Button>
<Button
variant="ghost"
size="sm"
onClick={() => setDeleteTarget(personality)}
>
<Trash2 className="h-4 w-4 text-destructive" />
</Button>
</div>
</div>
</CardHeader>
<CardContent>
<div className="flex gap-4 text-sm">
<div>
<span className="text-muted-foreground">Tone:</span>
<Badge variant="outline" className="ml-2">
{personality.tone}
</Badge>
</div>
<div>
<span className="text-muted-foreground">Formality:</span>
<Badge variant="outline" className="ml-2">
{personality.formalityLevel.replace(/_/g, " ")}
</Badge>
</div>
</div>
</CardContent>
</Card>
))}
</div>
)}
{/* Delete Confirmation Dialog */}
<AlertDialog open={!!deleteTarget} onOpenChange={(open) => !open && setDeleteTarget(null)}>
<AlertDialogContent>
<AlertDialogHeader>
<AlertDialogTitle>Delete Personality</AlertDialogTitle>
<AlertDialogDescription>
Are you sure you want to delete "{deleteTarget?.name}"? This action cannot be undone.
</AlertDialogDescription>
</AlertDialogHeader>
<AlertDialogFooter>
<AlertDialogCancel>Cancel</AlertDialogCancel>
<AlertDialogAction onClick={confirmDelete}>Delete</AlertDialogAction>
</AlertDialogFooter>
</AlertDialogContent>
</AlertDialog>
</div>
);
}

View File

@@ -0,0 +1,136 @@
import { describe, it, expect, vi } from "vitest";
import { render, screen } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import { DomainFilter } from "./DomainFilter";
import type { Domain } from "@mosaic/shared";
describe("DomainFilter", () => {
const mockDomains: Domain[] = [
{
id: "domain-1",
workspaceId: "workspace-1",
name: "Work",
slug: "work",
description: "Work-related tasks",
color: "#3B82F6",
icon: "💼",
sortOrder: 0,
metadata: {},
createdAt: new Date("2026-01-28"),
updatedAt: new Date("2026-01-28"),
},
{
id: "domain-2",
workspaceId: "workspace-1",
name: "Personal",
slug: "personal",
description: null,
color: "#10B981",
icon: "🏠",
sortOrder: 1,
metadata: {},
createdAt: new Date("2026-01-28"),
updatedAt: new Date("2026-01-28"),
},
];
it("should render All button", () => {
const onFilterChange = vi.fn();
render(
<DomainFilter
domains={mockDomains}
selectedDomain={null}
onFilterChange={onFilterChange}
/>
);
expect(screen.getByRole("button", { name: /all/i })).toBeInTheDocument();
});
it("should render domain filter buttons", () => {
const onFilterChange = vi.fn();
render(
<DomainFilter
domains={mockDomains}
selectedDomain={null}
onFilterChange={onFilterChange}
/>
);
expect(screen.getByRole("button", { name: /filter by work/i })).toBeInTheDocument();
expect(screen.getByRole("button", { name: /filter by personal/i })).toBeInTheDocument();
});
it("should highlight All when no domain selected", () => {
const onFilterChange = vi.fn();
render(
<DomainFilter
domains={mockDomains}
selectedDomain={null}
onFilterChange={onFilterChange}
/>
);
const allButton = screen.getByRole("button", { name: /all/i });
expect(allButton.getAttribute("aria-pressed")).toBe("true");
});
it("should highlight selected domain", () => {
const onFilterChange = vi.fn();
render(
<DomainFilter
domains={mockDomains}
selectedDomain="domain-1"
onFilterChange={onFilterChange}
/>
);
const workButton = screen.getByRole("button", { name: /filter by work/i });
expect(workButton.getAttribute("aria-pressed")).toBe("true");
});
it("should call onFilterChange when All clicked", async () => {
const user = userEvent.setup();
const onFilterChange = vi.fn();
render(
<DomainFilter
domains={mockDomains}
selectedDomain="domain-1"
onFilterChange={onFilterChange}
/>
);
const allButton = screen.getByRole("button", { name: /all/i });
await user.click(allButton);
expect(onFilterChange).toHaveBeenCalledWith(null);
});
it("should call onFilterChange when domain clicked", async () => {
const user = userEvent.setup();
const onFilterChange = vi.fn();
render(
<DomainFilter
domains={mockDomains}
selectedDomain={null}
onFilterChange={onFilterChange}
/>
);
const workButton = screen.getByRole("button", { name: /filter by work/i });
await user.click(workButton);
expect(onFilterChange).toHaveBeenCalledWith("domain-1");
});
it("should display domain icons", () => {
const onFilterChange = vi.fn();
render(
<DomainFilter
domains={mockDomains}
selectedDomain={null}
onFilterChange={onFilterChange}
/>
);
expect(screen.getByText("💼")).toBeInTheDocument();
expect(screen.getByText("🏠")).toBeInTheDocument();
});
});

View File

@@ -0,0 +1,52 @@
"use client";
import type { Domain } from "@mosaic/shared";
interface DomainFilterProps {
domains: Domain[];
selectedDomain: string | null;
onFilterChange: (domainId: string | null) => void;
}
export function DomainFilter({
domains,
selectedDomain,
onFilterChange,
}: DomainFilterProps): JSX.Element {
return (
<div className="flex gap-2 flex-wrap">
<button
onClick={() => onFilterChange(null)}
className={`px-3 py-1 rounded-full text-sm ${
selectedDomain === null
? "bg-gray-900 text-white"
: "bg-gray-100 text-gray-700 hover:bg-gray-200"
}`}
aria-label="Show all domains"
aria-pressed={selectedDomain === null}
>
All
</button>
{domains.map((domain) => (
<button
key={domain.id}
onClick={() => onFilterChange(domain.id)}
className={`px-3 py-1 rounded-full text-sm flex items-center gap-1 ${
selectedDomain === domain.id
? "text-white"
: "bg-gray-100 text-gray-700 hover:bg-gray-200"
}`}
style={{
backgroundColor:
selectedDomain === domain.id ? domain.color || "#374151" : undefined,
}}
aria-label={`Filter by ${domain.name}`}
aria-pressed={selectedDomain === domain.id}
>
{domain.icon && <span>{domain.icon}</span>}
<span>{domain.name}</span>
</button>
))}
</div>
);
}

View File

@@ -0,0 +1,62 @@
"use client";
import type { Domain } from "@mosaic/shared";
interface DomainItemProps {
domain: Domain;
onEdit?: (domain: Domain) => void;
onDelete?: (domain: Domain) => void;
}
export function DomainItem({
domain,
onEdit,
onDelete,
}: DomainItemProps): JSX.Element {
return (
<div className="border rounded-lg p-4 hover:shadow-md transition-shadow">
<div className="flex items-start justify-between">
<div className="flex-1">
<div className="flex items-center gap-2 mb-2">
{domain.icon && <span className="text-2xl">{domain.icon}</span>}
{domain.color && (
<div
className="w-4 h-4 rounded-full"
style={{ backgroundColor: domain.color }}
/>
)}
<h3 className="font-semibold text-lg">{domain.name}</h3>
</div>
{domain.description && (
<p className="text-sm text-gray-600">{domain.description}</p>
)}
<div className="mt-2">
<span className="text-xs text-gray-500 font-mono">
{domain.slug}
</span>
</div>
</div>
<div className="flex gap-2 ml-4">
{onEdit && (
<button
onClick={() => onEdit(domain)}
className="text-sm px-3 py-1 border rounded hover:bg-gray-50"
aria-label={`Edit ${domain.name}`}
>
Edit
</button>
)}
{onDelete && (
<button
onClick={() => onDelete(domain)}
className="text-sm px-3 py-1 border border-red-300 text-red-600 rounded hover:bg-red-50"
aria-label={`Delete ${domain.name}`}
>
Delete
</button>
)}
</div>
</div>
</div>
);
}

View File

@@ -0,0 +1,93 @@
import { describe, it, expect, vi } from "vitest";
import { render, screen } from "@testing-library/react";
import { DomainList } from "./DomainList";
import type { Domain } from "@mosaic/shared";
describe("DomainList", () => {
const mockDomains: Domain[] = [
{
id: "domain-1",
workspaceId: "workspace-1",
name: "Work",
slug: "work",
description: "Work-related tasks",
color: "#3B82F6",
icon: "💼",
sortOrder: 0,
metadata: {},
createdAt: new Date("2026-01-28"),
updatedAt: new Date("2026-01-28"),
},
{
id: "domain-2",
workspaceId: "workspace-1",
name: "Personal",
slug: "personal",
description: "Personal tasks and projects",
color: "#10B981",
icon: "🏠",
sortOrder: 1,
metadata: {},
createdAt: new Date("2026-01-28"),
updatedAt: new Date("2026-01-28"),
},
];
it("should render empty state when no domains", () => {
render(<DomainList domains={[]} isLoading={false} />);
expect(screen.getByText(/no domains created yet/i)).toBeInTheDocument();
});
it("should render loading state", () => {
render(<DomainList domains={[]} isLoading={true} />);
expect(screen.getByText(/loading domains/i)).toBeInTheDocument();
});
it("should render domains list", () => {
render(<DomainList domains={mockDomains} isLoading={false} />);
expect(screen.getByText("Work")).toBeInTheDocument();
expect(screen.getByText("Personal")).toBeInTheDocument();
});
it("should call onEdit when edit button clicked", () => {
const onEdit = vi.fn();
render(
<DomainList
domains={mockDomains}
isLoading={false}
onEdit={onEdit}
/>
);
const editButtons = screen.getAllByRole("button", { name: /edit/i });
editButtons[0].click();
expect(onEdit).toHaveBeenCalledWith(mockDomains[0]);
});
it("should call onDelete when delete button clicked", () => {
const onDelete = vi.fn();
render(
<DomainList
domains={mockDomains}
isLoading={false}
onDelete={onDelete}
/>
);
const deleteButtons = screen.getAllByRole("button", { name: /delete/i });
deleteButtons[0].click();
expect(onDelete).toHaveBeenCalledWith(mockDomains[0]);
});
it("should handle undefined domains gracefully", () => {
// @ts-expect-error Testing error state
render(<DomainList domains={undefined} isLoading={false} />);
expect(screen.getByText(/no domains created yet/i)).toBeInTheDocument();
});
it("should handle null domains gracefully", () => {
// @ts-expect-error Testing error state
render(<DomainList domains={null} isLoading={false} />);
expect(screen.getByText(/no domains created yet/i)).toBeInTheDocument();
});
});

View File

@@ -0,0 +1,51 @@
"use client";
import type { Domain } from "@mosaic/shared";
import { DomainItem } from "./DomainItem";
interface DomainListProps {
domains: Domain[];
isLoading: boolean;
onEdit?: (domain: Domain) => void;
onDelete?: (domain: Domain) => void;
}
export function DomainList({
domains,
isLoading,
onEdit,
onDelete,
}: DomainListProps): JSX.Element {
if (isLoading) {
return (
<div className="flex justify-center items-center p-8">
<div className="animate-spin rounded-full h-8 w-8 border-b-2 border-gray-900"></div>
<span className="ml-3 text-gray-600">Loading domains...</span>
</div>
);
}
if (!domains || domains.length === 0) {
return (
<div className="text-center p-8 text-gray-500">
<p className="text-lg">No domains created yet</p>
<p className="text-sm mt-2">
Create domains to organize your tasks and projects
</p>
</div>
);
}
return (
<div className="grid gap-4 md:grid-cols-2 lg:grid-cols-3">
{domains.map((domain) => (
<DomainItem
key={domain.id}
domain={domain}
onEdit={onEdit}
onDelete={onDelete}
/>
))}
</div>
);
}

View File

@@ -0,0 +1,127 @@
import { describe, it, expect, vi } from "vitest";
import { render, screen } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import { DomainSelector } from "./DomainSelector";
import type { Domain } from "@mosaic/shared";
describe("DomainSelector", () => {
const mockDomains: Domain[] = [
{
id: "domain-1",
workspaceId: "workspace-1",
name: "Work",
slug: "work",
description: "Work-related tasks",
color: "#3B82F6",
icon: "💼",
sortOrder: 0,
metadata: {},
createdAt: new Date("2026-01-28"),
updatedAt: new Date("2026-01-28"),
},
{
id: "domain-2",
workspaceId: "workspace-1",
name: "Personal",
slug: "personal",
description: null,
color: "#10B981",
icon: null,
sortOrder: 1,
metadata: {},
createdAt: new Date("2026-01-28"),
updatedAt: new Date("2026-01-28"),
},
];
it("should render with default placeholder", () => {
const onChange = vi.fn();
render(
<DomainSelector domains={mockDomains} value={null} onChange={onChange} />
);
expect(screen.getByText("Select a domain")).toBeInTheDocument();
});
it("should render with custom placeholder", () => {
const onChange = vi.fn();
render(
<DomainSelector
domains={mockDomains}
value={null}
onChange={onChange}
placeholder="Choose domain"
/>
);
expect(screen.getByText("Choose domain")).toBeInTheDocument();
});
it("should render all domains as options", () => {
const onChange = vi.fn();
render(
<DomainSelector domains={mockDomains} value={null} onChange={onChange} />
);
expect(screen.getByText("💼 Work")).toBeInTheDocument();
expect(screen.getByText("Personal")).toBeInTheDocument();
});
it("should call onChange when selection changes", async () => {
const user = userEvent.setup();
const onChange = vi.fn();
render(
<DomainSelector domains={mockDomains} value={null} onChange={onChange} />
);
const select = screen.getByRole("combobox");
await user.selectOptions(select, "domain-1");
expect(onChange).toHaveBeenCalledWith("domain-1");
});
it("should call onChange with null when cleared", async () => {
const user = userEvent.setup();
const onChange = vi.fn();
render(
<DomainSelector
domains={mockDomains}
value="domain-1"
onChange={onChange}
/>
);
const select = screen.getByRole("combobox");
await user.selectOptions(select, "");
expect(onChange).toHaveBeenCalledWith(null);
});
it("should show selected value", () => {
const onChange = vi.fn();
render(
<DomainSelector
domains={mockDomains}
value="domain-1"
onChange={onChange}
/>
);
const select = screen.getByRole("combobox") as HTMLSelectElement;
expect(select.value).toBe("domain-1");
});
it("should apply custom className", () => {
const onChange = vi.fn();
render(
<DomainSelector
domains={mockDomains}
value={null}
onChange={onChange}
className="custom-class"
/>
);
const select = screen.getByRole("combobox");
expect(select.className).toContain("custom-class");
});
});

View File

@@ -0,0 +1,38 @@
"use client";
import type { Domain } from "@mosaic/shared";
interface DomainSelectorProps {
domains: Domain[];
value: string | null;
onChange: (domainId: string | null) => void;
placeholder?: string;
className?: string;
}
export function DomainSelector({
domains,
value,
onChange,
placeholder = "Select a domain",
className = "",
}: DomainSelectorProps): JSX.Element {
return (
<select
value={value ?? ""}
onChange={(e: React.ChangeEvent<HTMLSelectElement>) =>
onChange(e.target.value || null)
}
className={`border rounded px-3 py-2 ${className}`}
aria-label="Domain selector"
>
<option value="">{placeholder}</option>
{domains.map((domain) => (
<option key={domain.id} value={domain.id}>
{domain.icon ? `${domain.icon} ` : ""}
{domain.name}
</option>
))}
</select>
);
}

View File

@@ -0,0 +1,140 @@
import { describe, it, expect, vi, beforeEach } from "vitest";
import { render, screen, waitFor } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import { FilterBar } from "./FilterBar";
import { TaskStatus, TaskPriority } from "@mosaic/shared";
describe("FilterBar", () => {
const mockOnFilterChange = vi.fn();
beforeEach(() => {
vi.clearAllMocks();
});
it("should render search input", () => {
render(<FilterBar onFilterChange={mockOnFilterChange} />);
expect(screen.getByPlaceholderText(/search/i)).toBeInTheDocument();
});
it("should render status filter", () => {
render(<FilterBar onFilterChange={mockOnFilterChange} />);
expect(screen.getByRole("button", { name: /status/i })).toBeInTheDocument();
});
it("should render priority filter", () => {
render(<FilterBar onFilterChange={mockOnFilterChange} />);
expect(screen.getByRole("button", { name: /priority/i })).toBeInTheDocument();
});
it("should render date range picker", () => {
render(<FilterBar onFilterChange={mockOnFilterChange} />);
expect(screen.getByPlaceholderText(/from date/i)).toBeInTheDocument();
expect(screen.getByPlaceholderText(/to date/i)).toBeInTheDocument();
});
it("should render clear filters button when filters applied", () => {
render(
<FilterBar
onFilterChange={mockOnFilterChange}
initialFilters={{ search: "test" }}
/>
);
expect(screen.getByRole("button", { name: /clear filters/i })).toBeInTheDocument();
});
it("should not render clear filters button when no filters applied", () => {
render(<FilterBar onFilterChange={mockOnFilterChange} />);
expect(screen.queryByRole("button", { name: /clear filters/i })).not.toBeInTheDocument();
});
it("should debounce search input", async () => {
const user = userEvent.setup();
render(<FilterBar onFilterChange={mockOnFilterChange} debounceMs={300} />);
const searchInput = screen.getByPlaceholderText(/search/i);
await user.type(searchInput, "test query");
// Should not call immediately
expect(mockOnFilterChange).not.toHaveBeenCalled();
// Should call after debounce delay
await waitFor(
() => {
expect(mockOnFilterChange).toHaveBeenCalledWith(
expect.objectContaining({ search: "test query" })
);
},
{ timeout: 500 }
);
});
it("should clear all filters when clear button clicked", async () => {
const user = userEvent.setup();
render(
<FilterBar
onFilterChange={mockOnFilterChange}
initialFilters={{
search: "test",
status: [TaskStatus.IN_PROGRESS],
priority: [TaskPriority.HIGH],
}}
/>
);
const clearButton = screen.getByRole("button", { name: /clear filters/i });
await user.click(clearButton);
expect(mockOnFilterChange).toHaveBeenCalledWith({});
});
it("should handle status selection", async () => {
const user = userEvent.setup();
render(<FilterBar onFilterChange={mockOnFilterChange} />);
const statusButton = screen.getByRole("button", { name: /status/i });
await user.click(statusButton);
// Note: Actual multi-select implementation would need to open a dropdown
// This is a simplified test
});
it("should handle priority selection", async () => {
const user = userEvent.setup();
render(<FilterBar onFilterChange={mockOnFilterChange} />);
const priorityButton = screen.getByRole("button", { name: /priority/i });
await user.click(priorityButton);
// Note: Actual implementation would need to open a dropdown
});
it("should handle date range selection", async () => {
const user = userEvent.setup();
render(<FilterBar onFilterChange={mockOnFilterChange} />);
const fromDate = screen.getByPlaceholderText(/from date/i);
const toDate = screen.getByPlaceholderText(/to date/i);
await user.type(fromDate, "2024-01-01");
await user.type(toDate, "2024-12-31");
await waitFor(() => {
expect(mockOnFilterChange).toHaveBeenCalled();
});
});
it("should display active filter count", () => {
render(
<FilterBar
onFilterChange={mockOnFilterChange}
initialFilters={{
status: [TaskStatus.IN_PROGRESS, TaskStatus.NOT_STARTED],
priority: [TaskPriority.HIGH],
}}
/>
);
// Should show 3 active filters (2 statuses + 1 priority)
expect(screen.getByText(/3/)).toBeInTheDocument();
});
});

View File

@@ -0,0 +1,207 @@
"use client";
import { useState, useEffect, useCallback } from "react";
import { TaskStatus, TaskPriority } from "@mosaic/shared";
export interface FilterValues {
search?: string;
status?: TaskStatus[];
priority?: TaskPriority[];
dateFrom?: string;
dateTo?: string;
sortBy?: string;
sortOrder?: "asc" | "desc";
}
interface FilterBarProps {
onFilterChange: (filters: FilterValues) => void;
initialFilters?: FilterValues;
debounceMs?: number;
}
export function FilterBar({
onFilterChange,
initialFilters = {},
debounceMs = 300,
}: FilterBarProps) {
const [filters, setFilters] = useState<FilterValues>(initialFilters);
const [searchValue, setSearchValue] = useState(initialFilters.search || "");
const [showStatusDropdown, setShowStatusDropdown] = useState(false);
const [showPriorityDropdown, setShowPriorityDropdown] = useState(false);
// Debounced search
useEffect(() => {
const timer = setTimeout(() => {
if (searchValue !== filters.search) {
const newFilters = { ...filters, search: searchValue || undefined };
setFilters(newFilters);
onFilterChange(newFilters);
}
}, debounceMs);
return () => clearTimeout(timer);
}, [searchValue, debounceMs]);
const handleFilterChange = useCallback(
(key: keyof FilterValues, value: any) => {
const newFilters = { ...filters, [key]: value };
if (!value || (Array.isArray(value) && value.length === 0)) {
delete newFilters[key];
}
setFilters(newFilters);
onFilterChange(newFilters);
},
[filters, onFilterChange]
);
const handleStatusToggle = (status: TaskStatus) => {
const currentStatuses = filters.status || [];
const newStatuses = currentStatuses.includes(status)
? currentStatuses.filter((s) => s !== status)
: [...currentStatuses, status];
handleFilterChange("status", newStatuses.length > 0 ? newStatuses : undefined);
};
const handlePriorityToggle = (priority: TaskPriority) => {
const currentPriorities = filters.priority || [];
const newPriorities = currentPriorities.includes(priority)
? currentPriorities.filter((p) => p !== priority)
: [...currentPriorities, priority];
handleFilterChange("priority", newPriorities.length > 0 ? newPriorities : undefined);
};
const clearAllFilters = () => {
setFilters({});
setSearchValue("");
onFilterChange({});
};
const activeFilterCount =
(filters.status?.length || 0) +
(filters.priority?.length || 0) +
(filters.search ? 1 : 0) +
(filters.dateFrom ? 1 : 0) +
(filters.dateTo ? 1 : 0);
const hasActiveFilters = activeFilterCount > 0;
return (
<div className="flex flex-wrap items-center gap-2 p-4 bg-gray-50 rounded-lg">
{/* Search Input */}
<div className="relative flex-1 min-w-[200px]">
<input
type="text"
placeholder="Search tasks..."
value={searchValue}
onChange={(e) => setSearchValue(e.target.value)}
className="w-full px-3 py-2 border border-gray-300 rounded-md focus:outline-none focus:ring-2 focus:ring-blue-500"
/>
</div>
{/* Status Filter */}
<div className="relative">
<button
onClick={() => setShowStatusDropdown(!showStatusDropdown)}
className="px-3 py-2 border border-gray-300 rounded-md hover:bg-gray-100 flex items-center gap-2"
aria-label="Status filter"
>
Status
{filters.status && filters.status.length > 0 && (
<span className="bg-blue-500 text-white text-xs px-2 py-0.5 rounded-full">
{filters.status.length}
</span>
)}
</button>
{showStatusDropdown && (
<div className="absolute top-full mt-1 bg-white border border-gray-300 rounded-md shadow-lg z-10 min-w-[150px]">
{Object.values(TaskStatus).map((status) => (
<label
key={status}
className="flex items-center px-3 py-2 hover:bg-gray-100 cursor-pointer"
>
<input
type="checkbox"
checked={filters.status?.includes(status) || false}
onChange={() => handleStatusToggle(status)}
className="mr-2"
/>
{status.replace(/_/g, " ")}
</label>
))}
</div>
)}
</div>
{/* Priority Filter */}
<div className="relative">
<button
onClick={() => setShowPriorityDropdown(!showPriorityDropdown)}
className="px-3 py-2 border border-gray-300 rounded-md hover:bg-gray-100 flex items-center gap-2"
aria-label="Priority filter"
>
Priority
{filters.priority && filters.priority.length > 0 && (
<span className="bg-blue-500 text-white text-xs px-2 py-0.5 rounded-full">
{filters.priority.length}
</span>
)}
</button>
{showPriorityDropdown && (
<div className="absolute top-full mt-1 bg-white border border-gray-300 rounded-md shadow-lg z-10 min-w-[150px]">
{Object.values(TaskPriority).map((priority) => (
<label
key={priority}
className="flex items-center px-3 py-2 hover:bg-gray-100 cursor-pointer"
>
<input
type="checkbox"
checked={filters.priority?.includes(priority) || false}
onChange={() => handlePriorityToggle(priority)}
className="mr-2"
/>
{priority}
</label>
))}
</div>
)}
</div>
{/* Date Range */}
<div className="flex items-center gap-2">
<input
type="date"
placeholder="From date"
value={filters.dateFrom || ""}
onChange={(e) => handleFilterChange("dateFrom", e.target.value || undefined)}
className="px-3 py-2 border border-gray-300 rounded-md focus:outline-none focus:ring-2 focus:ring-blue-500"
/>
<span className="text-gray-500">to</span>
<input
type="date"
placeholder="To date"
value={filters.dateTo || ""}
onChange={(e) => handleFilterChange("dateTo", e.target.value || undefined)}
className="px-3 py-2 border border-gray-300 rounded-md focus:outline-none focus:ring-2 focus:ring-blue-500"
/>
</div>
{/* Clear Filters */}
{hasActiveFilters && (
<button
onClick={clearAllFilters}
className="ml-auto px-3 py-2 text-gray-700 hover:bg-gray-100 rounded-md"
aria-label="Clear filters"
>
Clear filters
</button>
)}
{/* Active Filter Count Badge */}
{activeFilterCount > 0 && (
<span className="bg-blue-500 text-white text-sm px-3 py-1 rounded-full">
{activeFilterCount}
</span>
)}
</div>
);
}

View File

@@ -0,0 +1 @@
export * from "./FilterBar";

View File

@@ -0,0 +1,195 @@
"use client";
import { useState } from "react";
import type { Personality, FormalityLevel } from "@mosaic/shared";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Label } from "@/components/ui/label";
import { Textarea } from "@/components/ui/textarea";
import {
Select,
SelectContent,
SelectItem,
SelectTrigger,
SelectValue,
} from "@/components/ui/select";
import { Switch } from "@/components/ui/switch";
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
export interface PersonalityFormData {
name: string;
description?: string;
tone: string;
formalityLevel: FormalityLevel;
systemPromptTemplate: string;
isDefault?: boolean;
isActive?: boolean;
}
interface PersonalityFormProps {
personality?: Personality;
onSubmit: (data: PersonalityFormData) => Promise<void>;
onCancel?: () => void;
}
const FORMALITY_OPTIONS = [
{ value: "VERY_CASUAL", label: "Very Casual" },
{ value: "CASUAL", label: "Casual" },
{ value: "NEUTRAL", label: "Neutral" },
{ value: "FORMAL", label: "Formal" },
{ value: "VERY_FORMAL", label: "Very Formal" },
];
export function PersonalityForm({ personality, onSubmit, onCancel }: PersonalityFormProps): JSX.Element {
const [formData, setFormData] = useState<PersonalityFormData>({
name: personality?.name || "",
description: personality?.description || "",
tone: personality?.tone || "",
formalityLevel: personality?.formalityLevel || "NEUTRAL",
systemPromptTemplate: personality?.systemPromptTemplate || "",
isDefault: personality?.isDefault || false,
isActive: personality?.isActive ?? true,
});
const [isSubmitting, setIsSubmitting] = useState(false);
async function handleSubmit(e: React.FormEvent): Promise<void> {
e.preventDefault();
setIsSubmitting(true);
try {
await onSubmit(formData);
} finally {
setIsSubmitting(false);
}
}
return (
<form onSubmit={handleSubmit}>
<Card>
<CardHeader>
<CardTitle>{personality ? "Edit Personality" : "Create New Personality"}</CardTitle>
<CardDescription>
Customize how the AI assistant communicates and responds
</CardDescription>
</CardHeader>
<CardContent className="space-y-4">
{/* Name */}
<div className="space-y-2">
<Label htmlFor="name">Name *</Label>
<Input
id="name"
value={formData.name}
onChange={(e) => setFormData({ ...formData, name: e.target.value })}
placeholder="e.g., Professional, Casual, Friendly"
required
/>
</div>
{/* Description */}
<div className="space-y-2">
<Label htmlFor="description">Description</Label>
<Textarea
id="description"
value={formData.description}
onChange={(e) => setFormData({ ...formData, description: e.target.value })}
placeholder="Brief description of this personality style"
rows={2}
/>
</div>
{/* Tone */}
<div className="space-y-2">
<Label htmlFor="tone">Tone *</Label>
<Input
id="tone"
value={formData.tone}
onChange={(e) => setFormData({ ...formData, tone: e.target.value })}
placeholder="e.g., professional, friendly, enthusiastic"
required
/>
</div>
{/* Formality Level */}
<div className="space-y-2">
<Label htmlFor="formality">Formality Level *</Label>
<Select
value={formData.formalityLevel}
onValueChange={(value) =>
setFormData({ ...formData, formalityLevel: value as FormalityLevel })
}
>
<SelectTrigger id="formality">
<SelectValue />
</SelectTrigger>
<SelectContent>
{FORMALITY_OPTIONS.map((option) => (
<SelectItem key={option.value} value={option.value}>
{option.label}
</SelectItem>
))}
</SelectContent>
</Select>
</div>
{/* System Prompt Template */}
<div className="space-y-2">
<Label htmlFor="systemPrompt">System Prompt Template *</Label>
<Textarea
id="systemPrompt"
value={formData.systemPromptTemplate}
onChange={(e) =>
setFormData({ ...formData, systemPromptTemplate: e.target.value })
}
placeholder="You are a helpful AI assistant..."
rows={6}
required
/>
<p className="text-xs text-muted-foreground">
This template guides the AI's communication style and behavior
</p>
</div>
{/* Switches */}
<div className="flex items-center justify-between">
<div className="space-y-0.5">
<Label htmlFor="isDefault">Set as Default</Label>
<p className="text-xs text-muted-foreground">
Use this personality by default for new conversations
</p>
</div>
<Switch
id="isDefault"
checked={formData.isDefault}
onCheckedChange={(checked) => setFormData({ ...formData, isDefault: checked })}
/>
</div>
<div className="flex items-center justify-between">
<div className="space-y-0.5">
<Label htmlFor="isActive">Active</Label>
<p className="text-xs text-muted-foreground">
Make this personality available for selection
</p>
</div>
<Switch
id="isActive"
checked={formData.isActive}
onCheckedChange={(checked) => setFormData({ ...formData, isActive: checked })}
/>
</div>
{/* Actions */}
<div className="flex justify-end gap-2 pt-4">
{onCancel && (
<Button type="button" variant="outline" onClick={onCancel}>
Cancel
</Button>
)}
<Button type="submit" disabled={isSubmitting}>
{isSubmitting ? "Saving..." : personality ? "Update" : "Create"}
</Button>
</div>
</CardContent>
</Card>
</form>
);
}

View File

@@ -0,0 +1,121 @@
"use client";
import { useState } from "react";
import type { Personality } from "@mosaic/shared";
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
import { Button } from "@/components/ui/button";
import { Textarea } from "@/components/ui/textarea";
import { Badge } from "@/components/ui/badge";
import { Sparkles } from "lucide-react";
interface PersonalityPreviewProps {
personality: Personality;
}
const SAMPLE_PROMPTS = [
"Explain quantum computing in simple terms",
"What's the best way to organize my tasks?",
"Help me brainstorm ideas for a new project",
];
const FORMALITY_LABELS: Record<string, string> = {
VERY_CASUAL: "Very Casual",
CASUAL: "Casual",
NEUTRAL: "Neutral",
FORMAL: "Formal",
VERY_FORMAL: "Very Formal",
};
export function PersonalityPreview({ personality }: PersonalityPreviewProps): JSX.Element {
const [selectedPrompt, setSelectedPrompt] = useState<string>(SAMPLE_PROMPTS[0]);
return (
<Card>
<CardHeader>
<div className="flex items-center justify-between">
<div>
<CardTitle className="flex items-center gap-2">
<Sparkles className="h-5 w-5" />
{personality.name}
</CardTitle>
<CardDescription>{personality.description}</CardDescription>
</div>
{personality.isDefault && (
<Badge variant="secondary">Default</Badge>
)}
</div>
</CardHeader>
<CardContent className="space-y-4">
{/* Personality Attributes */}
<div className="grid grid-cols-2 gap-4 text-sm">
<div>
<span className="text-muted-foreground">Tone:</span>
<Badge variant="outline" className="ml-2">
{personality.tone}
</Badge>
</div>
<div>
<span className="text-muted-foreground">Formality:</span>
<Badge variant="outline" className="ml-2">
{FORMALITY_LABELS[personality.formalityLevel]}
</Badge>
</div>
</div>
{/* Sample Interaction */}
<div className="space-y-2">
<label className="text-sm font-medium">Preview with Sample Prompt:</label>
<div className="flex flex-wrap gap-2">
{SAMPLE_PROMPTS.map((prompt) => (
<Button
key={prompt}
variant={selectedPrompt === prompt ? "default" : "outline"}
size="sm"
onClick={() => setSelectedPrompt(prompt)}
>
{prompt.substring(0, 30)}...
</Button>
))}
</div>
</div>
{/* System Prompt Template */}
<div className="space-y-2">
<label className="text-sm font-medium">System Prompt Template:</label>
<Textarea
value={personality.systemPromptTemplate}
readOnly
className="min-h-[100px] bg-muted"
/>
</div>
{/* Mock Response Preview */}
<div className="space-y-2">
<label className="text-sm font-medium">Sample Response Style:</label>
<div className="rounded-md border bg-muted/50 p-4 text-sm">
<p className="italic text-muted-foreground">
"{selectedPrompt}"
</p>
<div className="mt-2 text-foreground">
{personality.formalityLevel === "VERY_CASUAL" && (
<p>Hey! So quantum computing is like... imagine if your computer could be in multiple places at once. Pretty wild, right? 🤯</p>
)}
{personality.formalityLevel === "CASUAL" && (
<p>Sure! Think of quantum computing like a super-powered calculator that can try lots of solutions at the same time.</p>
)}
{personality.formalityLevel === "NEUTRAL" && (
<p>Quantum computing uses quantum mechanics principles to process information differently from classical computers, enabling parallel computation.</p>
)}
{personality.formalityLevel === "FORMAL" && (
<p>Quantum computing represents a paradigm shift in computational methodology, leveraging quantum mechanical phenomena to perform calculations.</p>
)}
{personality.formalityLevel === "VERY_FORMAL" && (
<p>Quantum computing constitutes a fundamental departure from classical computational architectures, employing quantum superposition and entanglement principles.</p>
)}
</div>
</div>
</div>
</CardContent>
</Card>
);
}

View File

@@ -0,0 +1,76 @@
"use client";
import { useState, useEffect } from "react";
import type { Personality } from "@mosaic/shared";
import { fetchPersonalities } from "@/lib/api/personalities";
import {
Select,
SelectContent,
SelectItem,
SelectTrigger,
SelectValue,
} from "@/components/ui/select";
import { Label } from "@/components/ui/label";
import { Badge } from "@/components/ui/badge";
interface PersonalitySelectorProps {
value?: string;
onChange?: (personalityId: string) => void;
label?: string;
className?: string;
}
export function PersonalitySelector({
value,
onChange,
label = "Select Personality",
className,
}: PersonalitySelectorProps): JSX.Element {
const [personalities, setPersonalities] = useState<Personality[]>([]);
const [isLoading, setIsLoading] = useState<boolean>(true);
useEffect(() => {
loadPersonalities();
}, []);
async function loadPersonalities(): Promise<void> {
try {
setIsLoading(true);
const response = await fetchPersonalities();
setPersonalities(response.data);
} catch (err) {
console.error("Failed to load personalities:", err);
} finally {
setIsLoading(false);
}
}
return (
<div className={className}>
{label && (
<Label htmlFor="personality-select" className="mb-2">
{label}
</Label>
)}
<Select value={value} onValueChange={onChange} disabled={isLoading}>
<SelectTrigger id="personality-select">
<SelectValue placeholder={isLoading ? "Loading..." : "Choose a personality"} />
</SelectTrigger>
<SelectContent>
{personalities.map((personality) => (
<SelectItem key={personality.id} value={personality.id}>
<div className="flex items-center gap-2">
<span>{personality.name}</span>
{personality.isDefault && (
<Badge variant="secondary" className="ml-2">
Default
</Badge>
)}
</div>
</SelectItem>
))}
</SelectContent>
</Select>
</div>
);
}

View File

@@ -0,0 +1,81 @@
/**
* Personality API Client
* Handles personality-related API requests
*/
import type { Personality, FormalityLevel } from "@mosaic/shared";
import { apiGet, apiPost, apiPatch, apiDelete, type ApiResponse } from "./client";
/**
* Create personality DTO
*/
export interface CreatePersonalityDto {
name: string;
description?: string;
tone: string;
formalityLevel: FormalityLevel;
systemPromptTemplate: string;
isDefault?: boolean;
isActive?: boolean;
}
/**
* Update personality DTO
*/
export interface UpdatePersonalityDto {
name?: string;
description?: string;
tone?: string;
formalityLevel?: FormalityLevel;
systemPromptTemplate?: string;
isDefault?: boolean;
isActive?: boolean;
}
/**
* Fetch all personalities
*/
export async function fetchPersonalities(
isActive: boolean = true
): Promise<ApiResponse<Personality[]>> {
const endpoint = `/api/personalities?isActive=${isActive}`;
return apiGet<ApiResponse<Personality[]>>(endpoint);
}
/**
* Fetch the default personality
*/
export async function fetchDefaultPersonality(): Promise<Personality> {
return apiGet<Personality>("/api/personalities/default");
}
/**
* Fetch a single personality by ID
*/
export async function fetchPersonality(id: string): Promise<Personality> {
return apiGet<Personality>(`/api/personalities/${id}`);
}
/**
* Create a new personality
*/
export async function createPersonality(data: CreatePersonalityDto): Promise<Personality> {
return apiPost<Personality>("/api/personalities", data);
}
/**
* Update a personality
*/
export async function updatePersonality(
id: string,
data: UpdatePersonalityDto
): Promise<Personality> {
return apiPatch<Personality>(`/api/personalities/${id}`, data);
}
/**
* Delete a personality
*/
export async function deletePersonality(id: string): Promise<void> {
return apiDelete<void>(`/api/personalities/${id}`);
}

View File

@@ -0,0 +1,122 @@
import { describe, it, expect, vi } from 'vitest';
import { render, screen } from '@testing-library/react';
import { WebSocketProvider, useWebSocketContext } from './WebSocketProvider';
import * as useWebSocketModule from '../hooks/useWebSocket';
// Mock the useWebSocket hook
vi.mock('../hooks/useWebSocket');
describe('WebSocketProvider', () => {
it('should provide WebSocket context to children', () => {
const mockUseWebSocket = vi.spyOn(useWebSocketModule, 'useWebSocket');
mockUseWebSocket.mockReturnValue({
isConnected: true,
socket: null,
});
function TestComponent(): React.JSX.Element {
const { isConnected } = useWebSocketContext();
return <div>{isConnected ? 'Connected' : 'Disconnected'}</div>;
}
render(
<WebSocketProvider workspaceId="workspace-123" token="auth-token">
<TestComponent />
</WebSocketProvider>
);
expect(screen.getByText('Connected')).toBeInTheDocument();
});
it('should pass callbacks to useWebSocket hook', () => {
const mockUseWebSocket = vi.spyOn(useWebSocketModule, 'useWebSocket');
mockUseWebSocket.mockReturnValue({
isConnected: false,
socket: null,
});
const onTaskCreated = vi.fn();
const onTaskUpdated = vi.fn();
const onTaskDeleted = vi.fn();
render(
<WebSocketProvider
workspaceId="workspace-123"
token="auth-token"
onTaskCreated={onTaskCreated}
onTaskUpdated={onTaskUpdated}
onTaskDeleted={onTaskDeleted}
>
<div>Test</div>
</WebSocketProvider>
);
expect(mockUseWebSocket).toHaveBeenCalledWith(
'workspace-123',
'auth-token',
{
onTaskCreated,
onTaskUpdated,
onTaskDeleted,
onEventCreated: undefined,
onEventUpdated: undefined,
onEventDeleted: undefined,
onProjectUpdated: undefined,
}
);
});
it('should throw error when useWebSocketContext is used outside provider', () => {
function TestComponent(): React.JSX.Element {
useWebSocketContext();
return <div>Test</div>;
}
// Suppress console.error for this test
const originalError = console.error;
console.error = vi.fn();
expect(() => {
render(<TestComponent />);
}).toThrow('useWebSocketContext must be used within WebSocketProvider');
console.error = originalError;
});
it('should update context when connection status changes', () => {
const mockUseWebSocket = vi.spyOn(useWebSocketModule, 'useWebSocket');
// Initially disconnected
mockUseWebSocket.mockReturnValue({
isConnected: false,
socket: null,
});
function TestComponent(): React.JSX.Element {
const { isConnected } = useWebSocketContext();
return <div data-testid="status">{isConnected ? 'Connected' : 'Disconnected'}</div>;
}
const { rerender } = render(
<WebSocketProvider workspaceId="workspace-123" token="auth-token">
<TestComponent />
</WebSocketProvider>
);
expect(screen.getByTestId('status')).toHaveTextContent('Disconnected');
// Update to connected
mockUseWebSocket.mockReturnValue({
isConnected: true,
socket: null,
});
rerender(
<WebSocketProvider workspaceId="workspace-123" token="auth-token">
<TestComponent />
</WebSocketProvider>
);
expect(screen.getByTestId('status')).toHaveTextContent('Connected');
});
});

View File

@@ -0,0 +1,94 @@
import React, { createContext, useContext, ReactNode } from 'react';
import { useWebSocket } from '../hooks/useWebSocket';
import { Socket } from 'socket.io-client';
interface Task {
id: string;
[key: string]: unknown;
}
interface Event {
id: string;
[key: string]: unknown;
}
interface Project {
id: string;
[key: string]: unknown;
}
interface DeletePayload {
id: string;
}
interface WebSocketContextValue {
isConnected: boolean;
socket: Socket | null;
}
interface WebSocketProviderProps {
workspaceId: string;
token: string;
onTaskCreated?: (task: Task) => void;
onTaskUpdated?: (task: Task) => void;
onTaskDeleted?: (payload: DeletePayload) => void;
onEventCreated?: (event: Event) => void;
onEventUpdated?: (event: Event) => void;
onEventDeleted?: (payload: DeletePayload) => void;
onProjectUpdated?: (project: Project) => void;
children: ReactNode;
}
const WebSocketContext = createContext<WebSocketContextValue | undefined>(undefined);
/**
* WebSocket Provider component
* Manages WebSocket connection and provides context to children
*/
export function WebSocketProvider({
workspaceId,
token,
onTaskCreated,
onTaskUpdated,
onTaskDeleted,
onEventCreated,
onEventUpdated,
onEventDeleted,
onProjectUpdated,
children,
}: WebSocketProviderProps): React.JSX.Element {
const { isConnected, socket } = useWebSocket(workspaceId, token, {
onTaskCreated: onTaskCreated ?? undefined,
onTaskUpdated: onTaskUpdated ?? undefined,
onTaskDeleted: onTaskDeleted ?? undefined,
onEventCreated: onEventCreated ?? undefined,
onEventUpdated: onEventUpdated ?? undefined,
onEventDeleted: onEventDeleted ?? undefined,
onProjectUpdated: onProjectUpdated ?? undefined,
});
const value: WebSocketContextValue = {
isConnected,
socket,
};
return (
<WebSocketContext.Provider value={value}>
{children}
</WebSocketContext.Provider>
);
}
/**
* Hook to access WebSocket context
* @throws Error if used outside WebSocketProvider
*/
export function useWebSocketContext(): WebSocketContextValue {
const context = useContext(WebSocketContext);
if (context === undefined) {
throw new Error('useWebSocketContext must be used within WebSocketProvider');
}
return context;
}