@cleocode/core 2026.3.42 → 2026.3.44

package/src/store/export.ts

@@ -7,7 +7,7 @@
  */
 
 import { createHash } from 'node:crypto';
-import type { Task, TaskFile } from '@cleocode/contracts';
+import type { Task } from '@cleocode/contracts';
 
 /** Export format version. */
 const EXPORT_FORMAT_VERSION = '1.0.0';
@@ -137,17 +137,17 @@ export function buildRelationshipGraph(tasks: Task[]): RelationshipGraph {
  */
 export function buildExportPackage(
   tasks: Task[],
-  taskData: TaskFile,
   options: {
     mode: string;
     rootTaskIds: string[];
     includeChildren: boolean;
     cleoVersion?: string;
     filters?: unknown;
+    projectName?: string;
   },
 ): ExportPackage {
   const now = new Date().toISOString();
-  const projectName = taskData.project?.name ?? 'unknown';
+  const projectName = options.projectName ?? 'unknown';
   const maxId = tasks.reduce((max, t) => {
     const num = parseInt(t.id.replace('T', ''), 10);
     return num > max ? num : max;
@@ -188,41 +188,51 @@ export function buildExportPackage(
 /**
  * Export a single task.
  */
-export function exportSingle(taskId: string, taskData: TaskFile): ExportPackage | null {
-  const task = taskData.tasks.find((t) => t.id === taskId);
+export function exportSingle(
+  taskId: string,
+  allTasks: Task[],
+  projectName?: string,
+): ExportPackage | null {
+  const task = allTasks.find((t) => t.id === taskId);
   if (!task) return null;
 
-  return buildExportPackage([task], taskData, {
+  return buildExportPackage([task], {
     mode: 'single',
     rootTaskIds: [taskId],
     includeChildren: false,
+    projectName,
   });
 }
 
 /**
  * Export a subtree (task + all descendants).
  */
-export function exportSubtree(rootId: string, taskData: TaskFile): ExportPackage | null {
-  const root = taskData.tasks.find((t) => t.id === rootId);
+export function exportSubtree(
+  rootId: string,
+  allTasks: Task[],
+  projectName?: string,
+): ExportPackage | null {
+  const root = allTasks.find((t) => t.id === rootId);
   if (!root) return null;
 
-  // Collect all descendants
+  // Collect all descendants via BFS
   const collected = new Map<string, Task>();
   const queue = [rootId];
 
   while (queue.length > 0) {
     const id = queue.shift()!;
-    const task = taskData.tasks.find((t) => t.id === id);
+    const task = allTasks.find((t) => t.id === id);
     if (!task || collected.has(id)) continue;
     collected.set(id, task);
-    const children = taskData.tasks.filter((t) => t.parentId === id);
+    const children = allTasks.filter((t) => t.parentId === id);
     queue.push(...children.map((c) => c.id));
   }
 
   const tasks = [...collected.values()];
-  return buildExportPackage(tasks, taskData, {
+  return buildExportPackage(tasks, {
     mode: 'subtree',
     rootTaskIds: [rootId],
     includeChildren: true,
+    projectName,
   });
 }

package/src/store/tasks-schema.ts

@@ -16,6 +16,7 @@ import {
   primaryKey,
   sqliteTable,
   text,
+  unique,
 } from 'drizzle-orm/sqlite-core';
 import {
   ADR_STATUSES,
@@ -27,6 +28,22 @@ import {
   TASK_STATUSES,
 } from './status-registry.js';
 
+export type {
+  AgentErrorLogRow,
+  AgentErrorType,
+  AgentInstanceRow,
+  AgentInstanceStatus,
+  AgentType,
+  NewAgentErrorLogRow,
+  NewAgentInstanceRow,
+} from '../agents/agent-schema.js';
+// Re-export agent schema tables so drizzle-kit picks them up for migrations.
+export {
+  AGENT_INSTANCE_STATUSES,
+  AGENT_TYPES,
+  agentErrorLog,
+  agentInstances,
+} from '../agents/agent-schema.js';
 export type {
   NewWarpChainInstanceRow,
   NewWarpChainRow,
@@ -95,6 +112,26 @@ export const TOKEN_USAGE_CONFIDENCE = ['real', 'high', 'estimated', 'coarse'] as
 /** Transport types for token telemetry. */
 export const TOKEN_USAGE_TRANSPORTS = ['cli', 'mcp', 'api', 'agent', 'unknown'] as const;
 
+/** Task relation types matching DB CHECK constraint on task_relations.relation_type. */
+export const TASK_RELATION_TYPES = [
+  'related',
+  'blocks',
+  'duplicates',
+  'absorbs',
+  'fixes',
+  'extends',
+  'supersedes',
+] as const;
+
+/** Lifecycle transition types matching DB CHECK constraint on lifecycle_transitions.transition_type. */
+export const LIFECYCLE_TRANSITION_TYPES = ['automatic', 'manual', 'forced'] as const;
+
+/** External task link types matching DB constraint on external_task_links.link_type. */
+export const EXTERNAL_LINK_TYPES = ['created', 'matched', 'manual'] as const;
+
+/** Sync direction types matching DB constraint on external_task_links.sync_direction. */
+export const SYNC_DIRECTIONS = ['inbound', 'outbound', 'bidirectional'] as const;
+
 // === TASKS TABLE ===
 
 export const tasks = sqliteTable(
@@ -160,6 +197,7 @@ export const tasks = sqliteTable(
     index('idx_tasks_phase').on(table.phase),
     index('idx_tasks_type').on(table.type),
     index('idx_tasks_priority').on(table.priority),
+    index('idx_tasks_session_id').on(table.sessionId),
   ],
 );
 
@@ -193,13 +231,16 @@ export const taskRelations = sqliteTable(
       .notNull()
       .references(() => tasks.id, { onDelete: 'cascade' }),
     relationType: text('relation_type', {
-      enum: ['related', 'blocks', 'duplicates', 'absorbs', 'fixes', 'extends', 'supersedes'],
+      enum: TASK_RELATION_TYPES,
     })
       .notNull()
       .default('related'),
     reason: text('reason'),
   },
-  (table) => [primaryKey({ columns: [table.taskId, table.relatedTo] })],
+  (table) => [
+    primaryKey({ columns: [table.taskId, table.relatedTo] }),
+    index('idx_task_relations_related_to').on(table.relatedTo),
+  ],
 );
 
 // === SESSIONS ===
@@ -225,8 +266,12 @@ export const sessions = sqliteTable(
     startedAt: text('started_at').notNull().default(sql`(datetime('now'))`),
     endedAt: text('ended_at'),
     // Session chain columns (T4959)
-    previousSessionId: text('previous_session_id'),
-    nextSessionId: text('next_session_id'),
+    previousSessionId: text('previous_session_id').references((): AnySQLiteColumn => sessions.id, {
+      onDelete: 'set null',
+    }),
+    nextSessionId: text('next_session_id').references((): AnySQLiteColumn => sessions.id, {
+      onDelete: 'set null',
+    }),
     agentIdentifier: text('agent_identifier'),
     handoffConsumedAt: text('handoff_consumed_at'),
     handoffConsumedBy: text('handoff_consumed_by'),
@@ -242,6 +287,7 @@ export const sessions = sqliteTable(
     index('idx_sessions_status').on(table.status),
     index('idx_sessions_previous').on(table.previousSessionId),
     index('idx_sessions_agent_identifier').on(table.agentIdentifier),
+    index('idx_sessions_started_at').on(table.startedAt),
   ],
 );
 
@@ -325,6 +371,7 @@ export const lifecycleStages = sqliteTable(
     index('idx_lifecycle_stages_pipeline_id').on(table.pipelineId),
     index('idx_lifecycle_stages_stage_name').on(table.stageName),
     index('idx_lifecycle_stages_status').on(table.status),
+    index('idx_lifecycle_stages_validated_by').on(table.validatedBy),
   ],
 );
 
@@ -381,7 +428,7 @@ export const lifecycleTransitions = sqliteTable(
     fromStageId: text('from_stage_id').notNull(),
     toStageId: text('to_stage_id').notNull(),
     transitionType: text('transition_type', {
-      enum: ['automatic', 'manual', 'forced'],
+      enum: LIFECYCLE_TRANSITION_TYPES,
     })
       .notNull()
       .default('automatic'),
@@ -524,6 +571,7 @@ export const auditLog = sqliteTable(
     index('idx_audit_log_domain').on(table.domain),
     index('idx_audit_log_request_id').on(table.requestId),
     index('idx_audit_log_project_hash').on(table.projectHash),
+    index('idx_audit_log_actor').on(table.actor),
   ],
 );
 
@@ -567,6 +615,7 @@ export const tokenUsage = sqliteTable(
     index('idx_token_usage_transport').on(table.transport),
     index('idx_token_usage_domain_operation').on(table.domain, table.operation),
     index('idx_token_usage_method').on(table.method),
+    index('idx_token_usage_gateway').on(table.gateway),
   ],
 );
 
@@ -603,7 +652,10 @@ export const architectureDecisions = sqliteTable(
     keywords: text('keywords'),
     topics: text('topics'),
   },
-  (table) => [index('idx_arch_decisions_status').on(table.status)],
+  (table) => [
+    index('idx_arch_decisions_status').on(table.status),
+    index('idx_arch_decisions_amends_id').on(table.amendsId),
+  ],
 );
 
 // === ADR JUNCTION TABLES (ADR-017 §5.3) ===
@@ -673,11 +725,11 @@ export const externalTaskLinks = sqliteTable(
     externalTitle: text('external_title'),
     /** How this link was established. */
     linkType: text('link_type', {
-      enum: ['created', 'matched', 'manual'],
+      enum: EXTERNAL_LINK_TYPES,
     }).notNull(),
     /** Direction of the sync that created this link. */
     syncDirection: text('sync_direction', {
-      enum: ['inbound', 'outbound', 'bidirectional'],
+      enum: SYNC_DIRECTIONS,
     })
       .notNull()
       .default('inbound'),
@@ -692,6 +744,11 @@ export const externalTaskLinks = sqliteTable(
     index('idx_ext_links_task_id').on(table.taskId),
     index('idx_ext_links_provider_external').on(table.providerId, table.externalId),
     index('idx_ext_links_provider_id').on(table.providerId),
+    unique('uq_ext_links_task_provider_external').on(
+      table.taskId,
+      table.providerId,
+      table.externalId,
+    ),
   ],
 );
 

package/src/store/typed-query.ts

@@ -0,0 +1,17 @@
+import type { SQLInputValue, StatementSync } from 'node:sqlite';
+
+/**
+ * Type-safe wrapper for {@link StatementSync.all} — centralizes the
+ * `as unknown as` cast required by node:sqlite's untyped return type.
+ */
+export function typedAll<T>(stmt: StatementSync, ...params: SQLInputValue[]): T[] {
+  return stmt.all(...params) as unknown as T[];
+}
+
+/**
+ * Type-safe wrapper for {@link StatementSync.get} — centralizes the
+ * `as unknown as` cast required by node:sqlite's untyped return type.
+ */
+export function typedGet<T>(stmt: StatementSync, ...params: SQLInputValue[]): T | undefined {
+  return stmt.get(...params) as unknown as T | undefined;
+}