@ai-setting/roy-agent-core 1.4.13 → 1.4.14

package/dist/shared/@ai-setting/roy-agent-core-8wzz66qe.js

@@ -0,0 +1,620 @@
+import {
+  createLogger,
+  init_logger
+} from "./roy-agent-core-yn761yve.js";
+import {
+  __require
+} from "./roy-agent-core-fs0mn2jk.js";
+
+// packages/core/src/env/task/storage/sqlite-task-store.ts
+init_logger();
+import { join } from "path";
+var logger = createLogger("task:store");
+function getDefaultDataDir() {
+  const home = process.env.HOME || process.env.USERPROFILE || "/tmp";
+  return join(home, ".local", "share");
+}
+function getDefaultTaskDbPath() {
+  return join(getDefaultDataDir(), "roy-agent", "tasks.db");
+}
+
+class SQLiteTaskStore {
+  db = null;
+  dbPath;
+  initialized = false;
+  tasksCache = new Map;
+  operationsCache = new Map;
+  constructor(dbPath) {
+    this.dbPath = dbPath || getDefaultTaskDbPath();
+  }
+  async initialize() {
+    if (this.initialized)
+      return;
+    const { Database } = __require("bun:sqlite");
+    if (this.dbPath !== ":memory:") {
+      const fs = __require("fs");
+      const path = __require("path");
+      const dir = path.dirname(this.dbPath);
+      if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+      }
+    }
+    this.db = new Database(this.dbPath);
+    this.db.run(`
+      CREATE TABLE IF NOT EXISTS task (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        title TEXT NOT NULL,
+        description TEXT DEFAULT '',
+        status TEXT DEFAULT 'todo',
+        priority TEXT DEFAULT 'medium',
+        progress INTEGER DEFAULT 0,
+        current_status TEXT DEFAULT '',
+        goals_and_expected_deliverables TEXT DEFAULT '',
+        parent_task_id INTEGER,
+        created_at INTEGER NOT NULL,
+        updated_at INTEGER NOT NULL,
+        due_date TEXT,
+        tags TEXT
+      )
+    `);
+    this.db.run(`
+      CREATE TABLE IF NOT EXISTS task_operation (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        task_id INTEGER NOT NULL,
+        session_id TEXT NOT NULL,
+        action_type TEXT NOT NULL,
+        action_title TEXT NOT NULL,
+        action_description TEXT DEFAULT '',
+        timestamp INTEGER NOT NULL,
+        md_path TEXT,
+        FOREIGN KEY (task_id) REFERENCES task(id) ON DELETE CASCADE
+      )
+    `);
+    this.db.run("CREATE INDEX IF NOT EXISTS idx_task_operation_task_id ON task_operation(task_id)");
+    this.db.run("CREATE INDEX IF NOT EXISTS idx_task_operation_session_id ON task_operation(session_id)");
+    this.db.run(`
+      CREATE TABLE IF NOT EXISTS task_tag (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT NOT NULL UNIQUE,
+        created_at INTEGER NOT NULL,
+        task_count INTEGER DEFAULT 0
+      )
+    `);
+    this.db.run(`
+      CREATE TABLE IF NOT EXISTS task_tag_relation (
+        task_id INTEGER NOT NULL,
+        tag_id INTEGER NOT NULL,
+        created_at INTEGER NOT NULL,
+        PRIMARY KEY (task_id, tag_id),
+        FOREIGN KEY (task_id) REFERENCES task(id) ON DELETE CASCADE,
+        FOREIGN KEY (tag_id) REFERENCES task_tag(id) ON DELETE CASCADE
+      )
+    `);
+    this.db.run("CREATE INDEX IF NOT EXISTS idx_task_tag_relation_tag_id ON task_tag_relation(tag_id)");
+    this.initialized = true;
+    logger.info(`TaskStore initialized at ${this.dbPath}`);
+  }
+  async createTask(options) {
+    await this.initialize();
+    const now = Date.now();
+    const stmt = this.db.prepare(`
+      INSERT INTO task (title, description, priority, parent_task_id, 
+        goals_and_expected_deliverables, due_date, tags, status, 
+        current_status, created_at, updated_at)
+      VALUES (?, ?, ?, ?, ?, ?, ?, 'todo', '', ?, ?)
+    `);
+    const result = stmt.run(options.title, options.description || "", options.priority || "medium", options.parent_task_id || null, options.goals_and_expected_deliverables || "", options.due_date || null, options.tags ? JSON.stringify(options.tags) : null, now, now);
+    const task = await this.getTask(result.lastInsertRowid);
+    if (task) {
+      this.tasksCache.set(task.id, task);
+      return task;
+    }
+    throw new Error("Failed to create task");
+  }
+  async getTask(id) {
+    await this.initialize();
+    if (this.tasksCache.has(id)) {
+      return this.tasksCache.get(id);
+    }
+    const stmt = this.db.prepare("SELECT * FROM task WHERE id = ?");
+    const row = stmt.get(id);
+    if (!row)
+      return;
+    const task = this.rowToTask(row);
+    this.tasksCache.set(id, task);
+    return task;
+  }
+  async updateTask(id, options) {
+    await this.initialize();
+    const task = await this.getTask(id);
+    if (!task)
+      return;
+    const updates = [];
+    const values = [];
+    if (options.title !== undefined) {
+      updates.push("title = ?");
+      values.push(options.title);
+    }
+    if (options.description !== undefined) {
+      updates.push("description = ?");
+      values.push(options.description);
+    }
+    if (options.status !== undefined) {
+      updates.push("status = ?");
+      values.push(options.status);
+    }
+    if (options.priority !== undefined) {
+      updates.push("priority = ?");
+      values.push(options.priority);
+    }
+    if (options.progress !== undefined) {
+      updates.push("progress = ?");
+      values.push(options.progress);
+    }
+    if (options.current_status !== undefined) {
+      updates.push("current_status = ?");
+      values.push(options.current_status);
+    }
+    if (options.goals_and_expected_deliverables !== undefined) {
+      updates.push("goals_and_expected_deliverables = ?");
+      values.push(options.goals_and_expected_deliverables);
+    }
+    if (options.due_date !== undefined) {
+      updates.push("due_date = ?");
+      values.push(options.due_date);
+    }
+    if (options.tags !== undefined) {
+      updates.push("tags = ?");
+      values.push(JSON.stringify(options.tags));
+    }
+    if (updates.length === 0)
+      return task;
+    updates.push("updated_at = ?");
+    values.push(Date.now());
+    values.push(id);
+    const stmt = this.db.prepare(`UPDATE task SET ${updates.join(", ")} WHERE id = ?`);
+    stmt.run(...values);
+    this.tasksCache.delete(id);
+    return this.getTask(id);
+  }
+  async deleteTask(id) {
+    await this.initialize();
+    const stmt = this.db.prepare("DELETE FROM task WHERE id = ?");
+    const result = stmt.run(id);
+    if (result.changes > 0) {
+      this.tasksCache.delete(id);
+      this.operationsCache.delete(id);
+    }
+    return result.changes > 0;
+  }
+  async listTasks(options) {
+    await this.initialize();
+    let sql = "SELECT * FROM task WHERE 1=1";
+    const params = [];
+    if (options?.status) {
+      sql += " AND status = ?";
+      params.push(options.status);
+    }
+    if (options?.priority) {
+      sql += " AND priority = ?";
+      params.push(options.priority);
+    }
+    if (options?.parent_task_id !== undefined) {
+      sql += " AND parent_task_id = ?";
+      params.push(options.parent_task_id);
+    }
+    sql += " ORDER BY updated_at DESC";
+    if (options?.limit) {
+      sql += " LIMIT ?";
+      params.push(options.limit);
+    }
+    if (options?.offset) {
+      sql += " OFFSET ?";
+      params.push(options.offset);
+    }
+    const stmt = this.db.prepare(sql);
+    const rows = stmt.all(...params);
+    return rows.map((row) => {
+      const task = this.rowToTask(row);
+      this.tasksCache.set(task.id, task);
+      return task;
+    });
+  }
+  async createOperation(options) {
+    await this.initialize();
+    const now = Date.now();
+    const stmt = this.db.prepare(`
+      INSERT INTO task_operation (task_id, session_id, action_type, action_title, 
+        action_description, timestamp, md_path)
+      VALUES (?, ?, ?, ?, ?, ?, ?)
+    `);
+    const result = stmt.run(options.taskId, options.sessionId, options.actionType, options.actionTitle, options.actionDescription || "", now, options.mdPath || null);
+    this.operationsCache.delete(options.taskId);
+    const operation = await this.getOperation(result.lastInsertRowid);
+    if (!operation) {
+      throw new Error("Failed to create operation");
+    }
+    return operation;
+  }
+  async getOperation(id) {
+    await this.initialize();
+    const stmt = this.db.prepare("SELECT * FROM task_operation WHERE id = ?");
+    const row = stmt.get(id);
+    if (!row)
+      return;
+    return this.rowToOperation(row);
+  }
+  async updateOperation(id, updates) {
+    await this.initialize();
+    const op = await this.getOperation(id);
+    if (!op)
+      return;
+    const setClauses = [];
+    const values = [];
+    if (updates.actionTitle !== undefined) {
+      setClauses.push("action_title = ?");
+      values.push(updates.actionTitle);
+    }
+    if (updates.actionDescription !== undefined) {
+      setClauses.push("action_description = ?");
+      values.push(updates.actionDescription);
+    }
+    if (setClauses.length === 0)
+      return op;
+    const stmt = this.db.prepare(`UPDATE task_operation SET ${setClauses.join(", ")} WHERE id = ?`);
+    stmt.run(...values, id);
+    this.operationsCache.delete(op.taskId);
+    return this.getOperation(id);
+  }
+  async deleteOperation(id) {
+    await this.initialize();
+    const op = await this.getOperation(id);
+    if (!op)
+      return false;
+    const stmt = this.db.prepare("DELETE FROM task_operation WHERE id = ?");
+    const result = stmt.run(id);
+    if (result.changes > 0) {
+      this.operationsCache.delete(op.taskId);
+    }
+    return result.changes > 0;
+  }
+  async listOperations(options) {
+    await this.initialize();
+    const cacheKey = options.taskId;
+    if (!this.operationsCache.has(cacheKey)) {
+      const stmt = this.db.prepare("SELECT * FROM task_operation WHERE task_id = ? ORDER BY timestamp DESC");
+      const rows = stmt.all(options.taskId);
+      this.operationsCache.set(cacheKey, rows.map((row) => this.rowToOperation(row)));
+    }
+    let ops = this.operationsCache.get(cacheKey);
+    if (options.actionType) {
+      ops = ops.filter((op) => op.actionType === options.actionType);
+    }
+    const offset = options.offset || 0;
+    const limit = options.limit || 20;
+    return ops.slice(offset, offset + limit);
+  }
+  rowToTask(row) {
+    return {
+      id: row.id,
+      title: row.title,
+      description: row.description,
+      status: row.status,
+      priority: row.priority,
+      progress: row.progress,
+      current_status: row.current_status,
+      goals_and_expected_deliverables: row.goals_and_expected_deliverables,
+      parent_task_id: row.parent_task_id || undefined,
+      createdAt: new Date(row.created_at).toISOString(),
+      updatedAt: new Date(row.updated_at).toISOString(),
+      due_date: row.due_date || undefined,
+      tags: row.tags ? JSON.parse(row.tags) : undefined
+    };
+  }
+  rowToOperation(row) {
+    return {
+      id: row.id,
+      taskId: row.task_id,
+      sessionId: row.session_id,
+      actionType: row.action_type,
+      actionTitle: row.action_title,
+      actionDescription: row.action_description,
+      timestamp: new Date(row.timestamp).toISOString(),
+      mdPath: row.md_path || undefined
+    };
+  }
+  async createTag(name) {
+    await this.initialize();
+    const now = Date.now();
+    const existing = await this.getTagByName(name);
+    if (existing) {
+      return existing;
+    }
+    const stmt = this.db.prepare(`
+      INSERT INTO task_tag (name, created_at, task_count)
+      VALUES (?, ?, 0)
+    `);
+    const result = stmt.run(name, now);
+    return {
+      id: result.lastInsertRowid,
+      name,
+      createdAt: new Date(now).toISOString(),
+      taskCount: 0
+    };
+  }
+  async getTag(id) {
+    await this.initialize();
+    const stmt = this.db.prepare("SELECT * FROM task_tag WHERE id = ?");
+    const row = stmt.get(id);
+    if (!row)
+      return;
+    return this.rowToTag(row);
+  }
+  async getTagByName(name) {
+    await this.initialize();
+    const stmt = this.db.prepare("SELECT * FROM task_tag WHERE name = ?");
+    const row = stmt.get(name);
+    if (!row)
+      return;
+    return this.rowToTag(row);
+  }
+  async listTags(options) {
+    await this.initialize();
+    let sql = "SELECT * FROM task_tag";
+    const params = [];
+    const orderBy = options?.orderBy || "count";
+    if (orderBy === "count") {
+      sql += " ORDER BY task_count DESC";
+    } else if (orderBy === "name") {
+      sql += " ORDER BY name ASC";
+    } else {
+      sql += " ORDER BY created_at DESC";
+    }
+    if (options?.limit) {
+      sql += " LIMIT ?";
+      params.push(options.limit);
+    }
+    if (options?.offset) {
+      sql += " OFFSET ?";
+      params.push(options.offset);
+    }
+    const stmt = this.db.prepare(sql);
+    const rows = stmt.all(...params);
+    return rows.map((row) => this.rowToTag(row));
+  }
+  async searchTags(options) {
+    await this.initialize();
+    const sql = `
+      SELECT * FROM task_tag 
+      WHERE name LIKE ? 
+      ORDER BY task_count DESC 
+      LIMIT ?
+    `;
+    const stmt = this.db.prepare(sql);
+    const rows = stmt.all(`%${options.query}%`, options.limit || 20);
+    return rows.map((row) => this.rowToTag(row));
+  }
+  async deleteTag(id) {
+    await this.initialize();
+    const stmt = this.db.prepare("DELETE FROM task_tag WHERE id = ?");
+    const result = stmt.run(id);
+    return result.changes > 0;
+  }
+  async addTagsToTask(taskId, tagIds) {
+    await this.initialize();
+    const now = Date.now();
+    const stmt = this.db.prepare(`
+      INSERT OR IGNORE INTO task_tag_relation (task_id, tag_id, created_at)
+      VALUES (?, ?, ?)
+    `);
+    for (const tagId of tagIds) {
+      stmt.run(taskId, tagId, now);
+      this.db.run(`
+        UPDATE task_tag SET task_count = task_count + 1 WHERE id = ?
+      `, tagId);
+    }
+    await this.syncTaskTagsJson(taskId);
+  }
+  async removeTagsFromTask(taskId, tagIds) {
+    await this.initialize();
+    const stmt = this.db.prepare("DELETE FROM task_tag_relation WHERE task_id = ? AND tag_id = ?");
+    for (const tagId of tagIds) {
+      stmt.run(taskId, tagId);
+      this.db.run(`
+        UPDATE task_tag SET task_count = MAX(0, task_count - 1) WHERE id = ?
+      `, tagId);
+    }
+    await this.syncTaskTagsJson(taskId);
+  }
+  async getTagsForTask(taskId) {
+    await this.initialize();
+    const sql = `
+      SELECT t.* FROM task_tag t
+      INNER JOIN task_tag_relation r ON t.id = r.tag_id
+      WHERE r.task_id = ?
+      ORDER BY t.name ASC
+    `;
+    const stmt = this.db.prepare(sql);
+    const rows = stmt.all(taskId);
+    return rows.map((row) => this.rowToTag(row));
+  }
+  async getTaskIdsByTags(tagNames) {
+    await this.initialize();
+    if (tagNames.length === 0)
+      return [];
+    const placeholders = tagNames.map(() => "?").join(", ");
+    const sql = `
+      SELECT DISTINCT r.task_id FROM task_tag_relation r
+      INNER JOIN task_tag t ON r.tag_id = t.id
+      WHERE t.name IN (${placeholders})
+    `;
+    const stmt = this.db.prepare(sql);
+    const rows = stmt.all(...tagNames);
+    return rows.map((row) => row.task_id);
+  }
+  async updateTaskTags(taskId, tagNames) {
+    await this.initialize();
+    const currentTags = await this.getTagsForTask(taskId);
+    if (currentTags.length > 0) {
+      await this.removeTagsFromTask(taskId, currentTags.map((t) => t.id));
+    }
+    const tagIds = [];
+    for (const name of tagNames) {
+      let tag = await this.getTagByName(name);
+      if (!tag) {
+        tag = await this.createTag(name);
+      }
+      tagIds.push(tag.id);
+    }
+    if (tagIds.length > 0) {
+      await this.addTagsToTask(taskId, tagIds);
+    }
+  }
+  async searchTasksByKeywords(keywords, options = {}) {
+    await this.initialize();
+    if (!keywords || keywords.length === 0) {
+      return [];
+    }
+    const { limit = 20, excludeTaskId } = options;
+    const tasks = new Map;
+    let taskSql = "SELECT * FROM task WHERE 1=1";
+    const taskParams = [];
+    if (excludeTaskId) {
+      taskSql += " AND id != ?";
+      taskParams.push(excludeTaskId);
+    }
+    taskSql += " ORDER BY updated_at DESC LIMIT ?";
+    taskParams.push(limit * 2);
+    const taskRows = this.db.prepare(taskSql).all(...taskParams);
+    for (const row of taskRows) {
+      const task = this.rowToTask(row);
+      let matchCount = 0;
+      if (task.title) {
+        const lowerTitle = task.title.toLowerCase();
+        for (const keyword of keywords) {
+          if (lowerTitle.includes(keyword.toLowerCase())) {
+            matchCount++;
+          }
+        }
+      }
+      if (task.tags && Array.isArray(task.tags)) {
+        for (const keyword of keywords) {
+          const lowerKeyword = keyword.toLowerCase();
+          if (task.tags.some((tag) => tag.toLowerCase().includes(lowerKeyword))) {
+            matchCount += 2;
+          }
+        }
+      }
+      if (task.description) {
+        const lowerDesc = task.description.toLowerCase();
+        for (const keyword of keywords) {
+          if (lowerDesc.includes(keyword.toLowerCase())) {
+            matchCount++;
+          }
+        }
+      }
+      if (task.goals_and_expected_deliverables) {
+        const lowerGoals = task.goals_and_expected_deliverables.toLowerCase();
+        for (const keyword of keywords) {
+          if (lowerGoals.includes(keyword.toLowerCase())) {
+            matchCount++;
+          }
+        }
+      }
+      if (matchCount > 0) {
+        tasks.set(task.id, { task, matchCount });
+      }
+    }
+    const allOps = this.db.prepare("SELECT DISTINCT task_id FROM task_operation WHERE 1=1").all();
+    for (const opRow of allOps) {
+      const opSql = `
+        SELECT * FROM task_operation 
+        WHERE task_id = ? 
+          AND (action_title LIKE ? OR action_description LIKE ?)
+        LIMIT 10
+      `;
+      for (const keyword of keywords) {
+        const pattern = `%${keyword}%`;
+        const ops = this.db.prepare(opSql).all(opRow.task_id, pattern, pattern);
+        if (ops.length > 0) {
+          const taskRow = this.db.prepare("SELECT * FROM task WHERE id = ?").get(opRow.task_id);
+          if (taskRow) {
+            const task = this.rowToTask(taskRow);
+            if (excludeTaskId && task.id === excludeTaskId)
+              continue;
+            const existing = tasks.get(task.id);
+            if (existing) {
+              existing.matchCount += 2;
+            } else {
+              tasks.set(task.id, { task, matchCount: 2 });
+            }
+          }
+        }
+      }
+    }
+    const results = Array.from(tasks.values()).sort((a, b) => b.matchCount - a.matchCount).slice(0, limit).map((item) => item.task);
+    return results;
+  }
+  async findSimilarTasksByKeywords(keywords, options = {}) {
+    const mergedOptions = { ...options, limit: options.limit ?? 5 };
+    return this.searchTasksByKeywords(keywords, mergedOptions);
+  }
+  async findSimilarTasks(taskId, limit = 5) {
+    await this.initialize();
+    const taskTags = await this.getTagsForTask(taskId);
+    if (taskTags.length === 0) {
+      return [];
+    }
+    const tagIds = taskTags.map((t) => t.id);
+    const placeholders = tagIds.map(() => "?").join(", ");
+    const sql = `
+      SELECT t.*, COUNT(r2.tag_id) as match_count
+      FROM task t
+      INNER JOIN task_tag_relation r ON t.id = r.task_id
+      INNER JOIN task_tag_relation r2 ON t.id = r2.task_id
+      WHERE r.tag_id IN (${placeholders})
+        AND t.id != ?
+      GROUP BY t.id
+      ORDER BY match_count DESC, t.updated_at DESC
+      LIMIT ?
+    `;
+    const stmt = this.db.prepare(sql);
+    const rows = stmt.all(...tagIds, taskId, limit);
+    return rows.map((row) => this.rowToTask(row));
+  }
+  async searchTasks(query, limit = 20) {
+    await this.initialize();
+    const sql = `
+      SELECT * FROM task 
+      WHERE title LIKE ? OR description LIKE ?
+      ORDER BY updated_at DESC
+      LIMIT ?
+    `;
+    const stmt = this.db.prepare(sql);
+    const rows = stmt.all(`%${query}%`, `%${query}%`, limit);
+    return rows.map((row) => this.rowToTask(row));
+  }
+  rowToTag(row) {
+    return {
+      id: row.id,
+      name: row.name,
+      createdAt: new Date(row.created_at).toISOString(),
+      taskCount: row.task_count
+    };
+  }
+  async syncTaskTagsJson(taskId) {
+    const tags = await this.getTagsForTask(taskId);
+    const tagNames = tags.map((t) => t.name);
+    const stmt = this.db.prepare("UPDATE task SET tags = ? WHERE id = ?");
+    stmt.run(tagNames.length > 0 ? JSON.stringify(tagNames) : null, taskId);
+    this.tasksCache.delete(taskId);
+  }
+  async close() {
+    if (this.db) {
+      this.db.close();
+      this.db = null;
+      this.initialized = false;
+    }
+  }
+}
+export { getDefaultDataDir, getDefaultTaskDbPath, SQLiteTaskStore };