root
165 lines
9.4 KiB
TypeScript
165 lines
9.4 KiB
TypeScript
import { randomUUID } from "node:crypto";
|
|
import pg from "pg";
|
|
|
|
import type { ExecutionReport, Milestone, TaskGroup, TaskItem, TaskStatus, Timeline } from "./contracts.ts";
|
|
|
|
const { Pool } = pg;
|
|
|
|
const schema = [
|
|
`CREATE TABLE IF NOT EXISTS tasktree_schema_migrations (migration_id TEXT PRIMARY KEY, applied_at TIMESTAMPTZ NOT NULL DEFAULT now())`,
|
|
`CREATE TABLE IF NOT EXISTS tasktree_groups (
|
|
id TEXT PRIMARY KEY, name TEXT NOT NULL, description TEXT NOT NULL DEFAULT '',
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT now(), updated_at TIMESTAMPTZ NOT NULL DEFAULT now())`,
|
|
`CREATE TABLE IF NOT EXISTS tasktree_tasks (
|
|
id TEXT PRIMARY KEY, group_id TEXT NOT NULL REFERENCES tasktree_groups(id) ON DELETE CASCADE,
|
|
parent_id TEXT REFERENCES tasktree_tasks(id) ON DELETE CASCADE, kind TEXT NOT NULL CHECK (kind IN ('task','subtask')),
|
|
title TEXT NOT NULL, description TEXT NOT NULL DEFAULT '', status TEXT NOT NULL DEFAULT 'pending'
|
|
CHECK (status IN ('pending','in_progress','completed','blocked')),
|
|
start_at TIMESTAMPTZ, due_at TIMESTAMPTZ, sort_order INTEGER NOT NULL DEFAULT 0,
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT now(), updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
|
|
CHECK (due_at IS NULL OR start_at IS NULL OR due_at >= start_at))`,
|
|
`CREATE TABLE IF NOT EXISTS tasktree_milestones (
|
|
id TEXT PRIMARY KEY, group_id TEXT NOT NULL REFERENCES tasktree_groups(id) ON DELETE CASCADE,
|
|
task_id TEXT REFERENCES tasktree_tasks(id) ON DELETE CASCADE, title TEXT NOT NULL,
|
|
occurs_at TIMESTAMPTZ NOT NULL, created_at TIMESTAMPTZ NOT NULL DEFAULT now())`,
|
|
`CREATE TABLE IF NOT EXISTS tasktree_execution_reports (
|
|
id TEXT PRIMARY KEY, task_id TEXT NOT NULL REFERENCES tasktree_tasks(id) ON DELETE CASCADE,
|
|
title TEXT NOT NULL, body TEXT NOT NULL, status TEXT NOT NULL DEFAULT 'succeeded',
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT now())`,
|
|
`CREATE INDEX IF NOT EXISTS idx_tasktree_tasks_group_order ON tasktree_tasks(group_id, sort_order, created_at)`,
|
|
`CREATE INDEX IF NOT EXISTS idx_tasktree_tasks_parent ON tasktree_tasks(parent_id)`,
|
|
`CREATE INDEX IF NOT EXISTS idx_tasktree_reports_task ON tasktree_execution_reports(task_id, created_at DESC)`
|
|
];
|
|
|
|
export class TaskTreeStore {
|
|
private readonly pool: pg.Pool;
|
|
private ready = false;
|
|
|
|
constructor(databaseUrl: string) {
|
|
if (!databaseUrl) throw new Error("TASKTREE_DATABASE_URL or DATABASE_URL is required");
|
|
this.pool = new Pool({ connectionString: databaseUrl, max: 4, connectionTimeoutMillis: 5000, ssl: { rejectUnauthorized: false } });
|
|
}
|
|
|
|
async ensureSchema() {
|
|
if (this.ready) return;
|
|
const client = await this.pool.connect();
|
|
try {
|
|
await client.query("BEGIN");
|
|
for (const statement of schema) await client.query(statement);
|
|
await client.query("INSERT INTO tasktree_schema_migrations (migration_id) VALUES ($1) ON CONFLICT DO NOTHING", ["tasktree-20260716-v1"]);
|
|
await client.query("COMMIT");
|
|
this.ready = true;
|
|
} catch (error) {
|
|
await client.query("ROLLBACK").catch(() => {});
|
|
throw error;
|
|
} finally {
|
|
client.release();
|
|
}
|
|
}
|
|
|
|
async health() {
|
|
await this.ensureSchema();
|
|
const result = await this.pool.query("SELECT current_database() AS database, now() AS now");
|
|
return { storage: "native-postgresql", schemaReady: true, database: result.rows[0]?.database, checkedAt: result.rows[0]?.now };
|
|
}
|
|
|
|
async listGroups(): Promise<TaskGroup[]> {
|
|
await this.ensureSchema();
|
|
const result = await this.pool.query("SELECT * FROM tasktree_groups ORDER BY updated_at DESC, name");
|
|
return result.rows.map(groupRow);
|
|
}
|
|
|
|
async getGroup(id: string): Promise<TaskGroup | null> {
|
|
await this.ensureSchema();
|
|
const result = await this.pool.query("SELECT * FROM tasktree_groups WHERE id=$1", [id]);
|
|
return result.rows[0] ? groupRow(result.rows[0]) : null;
|
|
}
|
|
|
|
async createGroup(name: string, description = ""): Promise<TaskGroup> {
|
|
await this.ensureSchema();
|
|
const result = await this.pool.query("INSERT INTO tasktree_groups (id,name,description) VALUES ($1,$2,$3) RETURNING *", [`tg_${randomUUID()}`, name, description]);
|
|
return groupRow(result.rows[0]);
|
|
}
|
|
|
|
async deleteGroup(id: string): Promise<boolean> {
|
|
await this.ensureSchema();
|
|
return (await this.pool.query("DELETE FROM tasktree_groups WHERE id=$1", [id])).rowCount === 1;
|
|
}
|
|
|
|
async createTask(input: { groupId: string; title: string; description?: string; startAt?: string; dueAt?: string; parentId?: string }): Promise<TaskItem> {
|
|
await this.ensureSchema();
|
|
let kind = "task";
|
|
if (input.parentId) {
|
|
const parent = await this.pool.query("SELECT group_id,parent_id FROM tasktree_tasks WHERE id=$1", [input.parentId]);
|
|
if (!parent.rows[0]) throw domainError("parent_not_found", "parent task was not found");
|
|
if (parent.rows[0].group_id !== input.groupId) throw domainError("parent_group_mismatch", "parent task belongs to another taskgroup");
|
|
if (parent.rows[0].parent_id) throw domainError("maximum_depth_exceeded", "TaskTree supports task and subtask only");
|
|
kind = "subtask";
|
|
}
|
|
const result = await this.pool.query(
|
|
`INSERT INTO tasktree_tasks (id,group_id,parent_id,kind,title,description,start_at,due_at,sort_order)
|
|
VALUES ($1,$2,$3,$4,$5,$6,$7,$8,(SELECT COALESCE(MAX(sort_order),-1)+1 FROM tasktree_tasks WHERE group_id=$2)) RETURNING *`,
|
|
[`tt_${randomUUID()}`, input.groupId, input.parentId ?? null, kind, input.title, input.description ?? "", input.startAt ?? null, input.dueAt ?? null]
|
|
);
|
|
return taskRow(result.rows[0]);
|
|
}
|
|
|
|
async updateTask(id: string, patch: { title?: string; description?: string; status?: TaskStatus; startAt?: string | null; dueAt?: string | null }): Promise<TaskItem | null> {
|
|
await this.ensureSchema();
|
|
const current = await this.pool.query("SELECT * FROM tasktree_tasks WHERE id=$1", [id]);
|
|
if (!current.rows[0]) return null;
|
|
const row = taskRow(current.rows[0]);
|
|
const result = await this.pool.query(
|
|
`UPDATE tasktree_tasks SET title=$2,description=$3,status=$4,start_at=$5,due_at=$6,updated_at=now() WHERE id=$1 RETURNING *`,
|
|
[id, patch.title ?? row.title, patch.description ?? row.description, patch.status ?? row.status,
|
|
patch.startAt === undefined ? row.startAt : patch.startAt, patch.dueAt === undefined ? row.dueAt : patch.dueAt]
|
|
);
|
|
return taskRow(result.rows[0]);
|
|
}
|
|
|
|
async deleteTask(id: string): Promise<boolean> {
|
|
await this.ensureSchema();
|
|
return (await this.pool.query("DELETE FROM tasktree_tasks WHERE id=$1", [id])).rowCount === 1;
|
|
}
|
|
|
|
async createMilestone(input: { groupId: string; title: string; occursAt: string; taskId?: string }): Promise<Milestone> {
|
|
await this.ensureSchema();
|
|
const result = await this.pool.query(
|
|
"INSERT INTO tasktree_milestones (id,group_id,task_id,title,occurs_at) VALUES ($1,$2,$3,$4,$5) RETURNING *",
|
|
[`tm_${randomUUID()}`, input.groupId, input.taskId ?? null, input.title, input.occursAt]
|
|
);
|
|
return milestoneRow(result.rows[0]);
|
|
}
|
|
|
|
async createReport(input: { taskId: string; title: string; body: string; status?: string }): Promise<ExecutionReport> {
|
|
await this.ensureSchema();
|
|
const result = await this.pool.query(
|
|
"INSERT INTO tasktree_execution_reports (id,task_id,title,body,status) VALUES ($1,$2,$3,$4,$5) RETURNING *",
|
|
[`tr_${randomUUID()}`, input.taskId, input.title, input.body, input.status ?? "succeeded"]
|
|
);
|
|
return reportRow(result.rows[0]);
|
|
}
|
|
|
|
async timeline(groupId: string): Promise<Timeline | null> {
|
|
await this.ensureSchema();
|
|
const group = await this.getGroup(groupId);
|
|
if (!group) return null;
|
|
const [tasks, milestones, reports] = await Promise.all([
|
|
this.pool.query("SELECT * FROM tasktree_tasks WHERE group_id=$1 ORDER BY sort_order,created_at", [groupId]),
|
|
this.pool.query("SELECT * FROM tasktree_milestones WHERE group_id=$1 ORDER BY occurs_at", [groupId]),
|
|
this.pool.query("SELECT r.* FROM tasktree_execution_reports r JOIN tasktree_tasks t ON t.id=r.task_id WHERE t.group_id=$1 ORDER BY r.created_at DESC", [groupId])
|
|
]);
|
|
return { group, tasks: tasks.rows.map(taskRow), milestones: milestones.rows.map(milestoneRow), reports: reports.rows.map(reportRow) };
|
|
}
|
|
|
|
async close() { await this.pool.end(); }
|
|
}
|
|
|
|
function groupRow(row: any): TaskGroup { return { id: row.id, name: row.name, description: row.description, createdAt: iso(row.created_at), updatedAt: iso(row.updated_at) }; }
|
|
function taskRow(row: any): TaskItem { return { id: row.id, groupId: row.group_id, parentId: row.parent_id, kind: row.kind, title: row.title, description: row.description, status: row.status, startAt: nullableIso(row.start_at), dueAt: nullableIso(row.due_at), sortOrder: row.sort_order, createdAt: iso(row.created_at), updatedAt: iso(row.updated_at) }; }
|
|
function milestoneRow(row: any): Milestone { return { id: row.id, groupId: row.group_id, taskId: row.task_id, title: row.title, occursAt: iso(row.occurs_at), createdAt: iso(row.created_at) }; }
|
|
function reportRow(row: any): ExecutionReport { return { id: row.id, taskId: row.task_id, title: row.title, body: row.body, status: row.status, createdAt: iso(row.created_at) }; }
|
|
function iso(value: unknown): string { return new Date(value as any).toISOString(); }
|
|
function nullableIso(value: unknown): string | null { return value == null ? null : iso(value); }
|
|
function domainError(code: string, message: string) { return Object.assign(new Error(message), { code }); }
|