teleton 0.1.0

package/dist/chunk-UR2LQEKR.js

@@ -0,0 +1,319 @@
+// src/memory/agent/tasks.ts
+import { randomUUID } from "crypto";
+var TaskStore = class {
+  constructor(db) {
+    this.db = db;
+  }
+  /**
+   * Create a new task
+   */
+  createTask(task) {
+    const id = randomUUID();
+    const now = Math.floor(Date.now() / 1e3);
+    this.db.prepare(
+      `
+      INSERT INTO tasks (id, description, status, priority, created_by, created_at, scheduled_for, payload, reason, scheduled_message_id)
+      VALUES (?, ?, 'pending', ?, ?, ?, ?, ?, ?, ?)
+    `
+    ).run(
+      id,
+      task.description,
+      task.priority ?? 0,
+      task.createdBy ?? null,
+      now,
+      task.scheduledFor ? Math.floor(task.scheduledFor.getTime() / 1e3) : null,
+      task.payload ?? null,
+      task.reason ?? null,
+      task.scheduledMessageId ?? null
+    );
+    if (task.dependsOn && task.dependsOn.length > 0) {
+      for (const parentId of task.dependsOn) {
+        this.addDependency(id, parentId);
+      }
+    }
+    return {
+      id,
+      description: task.description,
+      status: "pending",
+      priority: task.priority ?? 0,
+      createdBy: task.createdBy,
+      createdAt: new Date(now * 1e3),
+      scheduledFor: task.scheduledFor,
+      payload: task.payload,
+      reason: task.reason,
+      scheduledMessageId: task.scheduledMessageId
+    };
+  }
+  /**
+   * Update a task
+   */
+  updateTask(taskId, updates) {
+    const task = this.getTask(taskId);
+    if (!task) return void 0;
+    const now = Math.floor(Date.now() / 1e3);
+    const updateFields = [];
+    const updateValues = [];
+    if (updates.description !== void 0) {
+      updateFields.push("description = ?");
+      updateValues.push(updates.description);
+    }
+    if (updates.status !== void 0) {
+      updateFields.push("status = ?");
+      updateValues.push(updates.status);
+      if (updates.status === "in_progress" && !task.startedAt) {
+        updateFields.push("started_at = ?");
+        updateValues.push(now);
+      }
+      if ((updates.status === "done" || updates.status === "failed" || updates.status === "cancelled") && !task.completedAt) {
+        updateFields.push("completed_at = ?");
+        updateValues.push(now);
+      }
+    }
+    if (updates.priority !== void 0) {
+      updateFields.push("priority = ?");
+      updateValues.push(updates.priority);
+    }
+    if (updates.result !== void 0) {
+      updateFields.push("result = ?");
+      updateValues.push(updates.result);
+    }
+    if (updates.error !== void 0) {
+      updateFields.push("error = ?");
+      updateValues.push(updates.error);
+    }
+    if (updateFields.length === 0) return task;
+    updateValues.push(taskId);
+    this.db.prepare(
+      `
+      UPDATE tasks
+      SET ${updateFields.join(", ")}
+      WHERE id = ?
+    `
+    ).run(...updateValues);
+    return this.getTask(taskId);
+  }
+  /**
+   * Get a task by ID
+   */
+  getTask(id) {
+    const row = this.db.prepare(`SELECT * FROM tasks WHERE id = ?`).get(id);
+    if (!row) return void 0;
+    return {
+      id: row.id,
+      description: row.description,
+      status: row.status,
+      priority: row.priority,
+      createdBy: row.created_by,
+      createdAt: new Date(row.created_at * 1e3),
+      startedAt: row.started_at ? new Date(row.started_at * 1e3) : void 0,
+      completedAt: row.completed_at ? new Date(row.completed_at * 1e3) : void 0,
+      result: row.result,
+      error: row.error,
+      scheduledFor: row.scheduled_for ? new Date(row.scheduled_for * 1e3) : void 0,
+      payload: row.payload,
+      reason: row.reason,
+      scheduledMessageId: row.scheduled_message_id
+    };
+  }
+  /**
+   * List tasks with optional filters
+   */
+  listTasks(filter) {
+    let sql = `SELECT * FROM tasks WHERE 1=1`;
+    const params = [];
+    if (filter?.status) {
+      sql += ` AND status = ?`;
+      params.push(filter.status);
+    }
+    if (filter?.createdBy) {
+      sql += ` AND created_by = ?`;
+      params.push(filter.createdBy);
+    }
+    sql += ` ORDER BY priority DESC, created_at ASC`;
+    const rows = this.db.prepare(sql).all(...params);
+    return rows.map((row) => ({
+      id: row.id,
+      description: row.description,
+      status: row.status,
+      priority: row.priority,
+      createdBy: row.created_by,
+      createdAt: new Date(row.created_at * 1e3),
+      startedAt: row.started_at ? new Date(row.started_at * 1e3) : void 0,
+      completedAt: row.completed_at ? new Date(row.completed_at * 1e3) : void 0,
+      result: row.result,
+      error: row.error,
+      scheduledFor: row.scheduled_for ? new Date(row.scheduled_for * 1e3) : void 0,
+      payload: row.payload,
+      reason: row.reason,
+      scheduledMessageId: row.scheduled_message_id
+    }));
+  }
+  /**
+   * Get active (pending or in_progress) tasks
+   */
+  getActiveTasks() {
+    const rows = this.db.prepare(
+      `
+      SELECT * FROM tasks
+      WHERE status IN ('pending', 'in_progress')
+      ORDER BY priority DESC, created_at ASC
+    `
+    ).all();
+    return rows.map((row) => ({
+      id: row.id,
+      description: row.description,
+      status: row.status,
+      priority: row.priority,
+      createdBy: row.created_by,
+      createdAt: new Date(row.created_at * 1e3),
+      startedAt: row.started_at ? new Date(row.started_at * 1e3) : void 0,
+      completedAt: row.completed_at ? new Date(row.completed_at * 1e3) : void 0,
+      result: row.result,
+      error: row.error,
+      scheduledFor: row.scheduled_for ? new Date(row.scheduled_for * 1e3) : void 0,
+      payload: row.payload,
+      reason: row.reason,
+      scheduledMessageId: row.scheduled_message_id
+    }));
+  }
+  /**
+   * Delete a task
+   */
+  deleteTask(taskId) {
+    const result = this.db.prepare(`DELETE FROM tasks WHERE id = ?`).run(taskId);
+    return result.changes > 0;
+  }
+  /**
+   * Mark task as done
+   */
+  completeTask(taskId, result) {
+    return this.updateTask(taskId, { status: "done", result });
+  }
+  /**
+   * Mark task as failed
+   */
+  failTask(taskId, error) {
+    return this.updateTask(taskId, { status: "failed", error });
+  }
+  /**
+   * Start a task
+   */
+  startTask(taskId) {
+    return this.updateTask(taskId, { status: "in_progress" });
+  }
+  /**
+   * Cancel a task
+   */
+  cancelTask(taskId) {
+    return this.updateTask(taskId, { status: "cancelled" });
+  }
+  /**
+   * Check if adding a dependency would create a cycle
+   * Uses BFS to traverse dependency graph
+   */
+  wouldCreateCycle(taskId, newParentId) {
+    if (taskId === newParentId) {
+      return true;
+    }
+    const visited = /* @__PURE__ */ new Set();
+    const queue = [newParentId];
+    while (queue.length > 0) {
+      const current = queue.shift();
+      if (current === taskId) {
+        return true;
+      }
+      if (visited.has(current)) {
+        continue;
+      }
+      visited.add(current);
+      const deps = this.getDependencies(current);
+      queue.push(...deps);
+    }
+    return false;
+  }
+  /**
+   * Add a dependency (taskId depends on parentTaskId)
+   * Throws error if would create circular dependency
+   */
+  addDependency(taskId, parentTaskId) {
+    if (this.wouldCreateCycle(taskId, parentTaskId)) {
+      throw new Error(
+        `Cannot add dependency: would create circular dependency (${taskId} \u2192 ${parentTaskId})`
+      );
+    }
+    this.db.prepare(`INSERT OR IGNORE INTO task_dependencies (task_id, depends_on_task_id) VALUES (?, ?)`).run(taskId, parentTaskId);
+  }
+  /**
+   * Get all tasks that this task depends on
+   */
+  getDependencies(taskId) {
+    const rows = this.db.prepare(`SELECT depends_on_task_id FROM task_dependencies WHERE task_id = ?`).all(taskId);
+    return rows.map((r) => r.depends_on_task_id);
+  }
+  /**
+   * Get all tasks that depend on this task
+   */
+  getDependents(taskId) {
+    const rows = this.db.prepare(`SELECT task_id FROM task_dependencies WHERE depends_on_task_id = ?`).all(taskId);
+    return rows.map((r) => r.task_id);
+  }
+  /**
+   * Check if a task can execute (all dependencies are done)
+   * Optimized: Single query with JOIN instead of N+1 queries
+   */
+  canExecute(taskId) {
+    const result = this.db.prepare(
+      `
+        SELECT COUNT(*) as pending_count
+        FROM task_dependencies td
+        LEFT JOIN tasks t ON td.depends_on_task_id = t.id
+        WHERE td.task_id = ?
+          AND (t.id IS NULL OR t.status != 'done')
+      `
+    ).get(taskId);
+    return result.pending_count === 0;
+  }
+  /**
+   * Get all parent task results for a dependent task
+   * Optimized: Single query with JOIN instead of N+1 queries
+   */
+  getParentResults(taskId) {
+    const rows = this.db.prepare(
+      `
+        SELECT t.id, t.description, t.result
+        FROM task_dependencies td
+        JOIN tasks t ON td.depends_on_task_id = t.id
+        WHERE td.task_id = ?
+          AND t.status = 'done'
+          AND t.result IS NOT NULL
+      `
+    ).all(taskId);
+    return rows.map((row) => {
+      let parsedResult;
+      try {
+        parsedResult = JSON.parse(row.result);
+      } catch (e) {
+        parsedResult = row.result;
+      }
+      return {
+        taskId: row.id,
+        description: row.description,
+        result: parsedResult
+      };
+    });
+  }
+};
+var instances = /* @__PURE__ */ new WeakMap();
+function getTaskStore(db) {
+  let store = instances.get(db);
+  if (!store) {
+    store = new TaskStore(db);
+    instances.set(db, store);
+  }
+  return store;
+}
+
+export {
+  TaskStore,
+  getTaskStore
+};