claw-code-parity/ts-worker/state/TaskStore.ts

/**
 * TaskStore — manages task lists and runtime tasks.
 *
 * Tracks the todo-list style tasks (TaskListRecord) and the
 * background runtime tasks (RuntimeTaskRecord) like agents and
 * shell processes.
 */

import type {
  TaskListRecord,
  TaskListStatus,
  RuntimeTaskRecord,
  RuntimeTaskStatus,
} from './records.js'

export class TaskStore {
  private taskListId = 'default'
  private tasks = new Map<string, TaskListRecord>()
  private runtimeTasks = new Map<string, RuntimeTaskRecord>()

  getTaskListId(): string {
    return this.taskListId
  }

  // ── Task list (todos) ─────────────────────────────────────────────

  upsertTask(task: TaskListRecord): void {
    this.tasks.set(task.id, task)
  }

  getTask(id: string): TaskListRecord | undefined {
    return this.tasks.get(id)
  }

  listTasks(): TaskListRecord[] {
    return Array.from(this.tasks.values())
  }

  removeTask(id: string): boolean {
    return this.tasks.delete(id)
  }

  // ── Runtime tasks (agents, shells) ────────────────────────────────

  upsertRuntimeTask(task: RuntimeTaskRecord): void {
    this.runtimeTasks.set(task.task_id, task)
  }

  getRuntimeTask(taskId: string): RuntimeTaskRecord | undefined {
    return this.runtimeTasks.get(taskId)
  }

  listRuntimeTasks(): RuntimeTaskRecord[] {
    return Array.from(this.runtimeTasks.values())
  }

  listAgents(): RuntimeTaskRecord[] {
    return this.listRuntimeTasks().filter((t) => t.kind === 'agent')
  }

  getAgent(agentId: string): RuntimeTaskRecord | undefined {
    for (const task of this.runtimeTasks.values()) {
      if (task.kind === 'agent' && (task.agent_id === agentId || task.task_id === agentId)) {
        return task
      }
    }
    return undefined
  }

  stopRuntimeTask(taskId: string): boolean {
    const task = this.runtimeTasks.get(taskId)
    if (!task || task.status !== 'running') return false
    task.status = 'stopped'
    task.completed_at = Date.now()
    return true
  }

  reset(): void {
    this.tasks.clear()
    this.runtimeTasks.clear()
  }
}