@stackmemoryai/stackmemory 0.2.9 → 0.3.0

package/dist/core/context/frame-manager.js

@@ -0,0 +1,682 @@
+import { v4 as uuidv4 } from "uuid";
+import { logger } from "../monitoring/logger.js";
+import { trace } from "../trace/index.js";
+import {
+  DatabaseError,
+  FrameError,
+  ErrorCode,
+  createErrorHandler
+} from "../errors/index.js";
+import { sessionManager, FrameQueryMode } from "../session/index.js";
+class FrameManager {
+  db;
+  currentRunId;
+  sessionId;
+  projectId;
+  activeStack = [];
+  // Stack of active frame IDs
+  queryMode = FrameQueryMode.PROJECT_ACTIVE;
+  constructor(db, projectId, runId) {
+    this.db = db;
+    this.projectId = projectId;
+    const session = sessionManager.getCurrentSession();
+    if (session) {
+      this.currentRunId = session.runId;
+      this.sessionId = session.sessionId;
+    } else {
+      this.currentRunId = runId || uuidv4();
+      this.sessionId = this.currentRunId;
+    }
+    this.initializeSchema();
+    this.loadActiveStack();
+  }
+  setQueryMode(mode) {
+    this.queryMode = mode;
+    this.loadActiveStack();
+  }
+  initializeSchema() {
+    const errorHandler = createErrorHandler({
+      operation: "initializeSchema",
+      projectId: this.projectId,
+      runId: this.currentRunId
+    });
+    try {
+      this.db.exec(`
+        CREATE TABLE IF NOT EXISTS frames (
+          frame_id TEXT PRIMARY KEY,
+          run_id TEXT NOT NULL,
+          project_id TEXT NOT NULL,
+          parent_frame_id TEXT REFERENCES frames(frame_id),
+          depth INTEGER NOT NULL DEFAULT 0,
+          type TEXT NOT NULL,
+          name TEXT NOT NULL,
+          state TEXT DEFAULT 'active',
+          inputs TEXT DEFAULT '{}',
+          outputs TEXT DEFAULT '{}',
+          digest_text TEXT,
+          digest_json TEXT DEFAULT '{}',
+          created_at INTEGER DEFAULT (unixepoch()),
+          closed_at INTEGER
+        );
+
+        CREATE TABLE IF NOT EXISTS events (
+          event_id TEXT PRIMARY KEY,
+          run_id TEXT NOT NULL,
+          frame_id TEXT NOT NULL,
+          seq INTEGER NOT NULL,
+          event_type TEXT NOT NULL,
+          payload TEXT NOT NULL,
+          ts INTEGER DEFAULT (unixepoch()),
+          FOREIGN KEY(frame_id) REFERENCES frames(frame_id)
+        );
+
+        CREATE TABLE IF NOT EXISTS anchors (
+          anchor_id TEXT PRIMARY KEY,
+          frame_id TEXT NOT NULL,
+          project_id TEXT NOT NULL,
+          type TEXT NOT NULL,
+          text TEXT NOT NULL,
+          priority INTEGER DEFAULT 0,
+          created_at INTEGER DEFAULT (unixepoch()),
+          metadata TEXT DEFAULT '{}',
+          FOREIGN KEY(frame_id) REFERENCES frames(frame_id)
+        );
+
+        CREATE INDEX IF NOT EXISTS idx_frames_run ON frames(run_id);
+        CREATE INDEX IF NOT EXISTS idx_frames_parent ON frames(parent_frame_id);
+        CREATE INDEX IF NOT EXISTS idx_frames_state ON frames(state);
+        CREATE INDEX IF NOT EXISTS idx_events_frame ON events(frame_id);
+        CREATE INDEX IF NOT EXISTS idx_events_seq ON events(frame_id, seq);
+        CREATE INDEX IF NOT EXISTS idx_anchors_frame ON anchors(frame_id);
+      `);
+    } catch (error) {
+      const dbError = errorHandler(error, {
+        operation: "initializeSchema",
+        schema: "frames"
+      });
+      if (dbError instanceof DatabaseError) {
+        throw new DatabaseError(
+          "Failed to initialize frame database schema",
+          ErrorCode.DB_MIGRATION_FAILED,
+          {
+            projectId: this.projectId,
+            operation: "initializeSchema",
+            originalError: error
+          },
+          error instanceof Error ? error : void 0
+        );
+      }
+      throw dbError;
+    }
+  }
+  loadActiveStack() {
+    const errorHandler = createErrorHandler({
+      operation: "loadActiveStack",
+      runId: this.currentRunId,
+      projectId: this.projectId
+    });
+    try {
+      let query;
+      let params;
+      switch (this.queryMode) {
+        case FrameQueryMode.ALL_ACTIVE:
+          query = `
+            SELECT frame_id, parent_frame_id, depth
+            FROM frames
+            WHERE state = 'active'
+            ORDER BY created_at DESC, depth ASC
+          `;
+          params = [];
+          break;
+        case FrameQueryMode.PROJECT_ACTIVE:
+          query = `
+            SELECT frame_id, parent_frame_id, depth, run_id
+            FROM frames
+            WHERE state = 'active' AND project_id = ?
+            ORDER BY created_at DESC, depth ASC
+          `;
+          params = [this.projectId];
+          break;
+        case FrameQueryMode.HISTORICAL:
+          query = `
+            SELECT frame_id, parent_frame_id, depth
+            FROM frames
+            WHERE project_id = ?
+            ORDER BY created_at DESC, depth ASC
+          `;
+          params = [this.projectId];
+          break;
+        case FrameQueryMode.CURRENT_SESSION:
+        default:
+          query = `
+            SELECT frame_id, parent_frame_id, depth
+            FROM frames
+            WHERE run_id = ? AND state = 'active'
+            ORDER BY depth ASC
+          `;
+          params = [this.currentRunId];
+          break;
+      }
+      const activeFrames = this.db.prepare(query).all(...params);
+      this.activeStack = this.buildStackOrder(activeFrames);
+      logger.info("Loaded active stack", {
+        runId: this.currentRunId,
+        stackDepth: this.activeStack.length,
+        activeFrames: this.activeStack,
+        queryMode: this.queryMode
+      });
+    } catch (error) {
+      const dbError = errorHandler(error, {
+        query: "Frame loading query",
+        runId: this.currentRunId,
+        queryMode: this.queryMode
+      });
+      if (dbError instanceof DatabaseError) {
+        throw new DatabaseError(
+          "Failed to load active frame stack",
+          ErrorCode.DB_QUERY_FAILED,
+          {
+            runId: this.currentRunId,
+            projectId: this.projectId,
+            operation: "loadActiveStack"
+          },
+          error instanceof Error ? error : void 0
+        );
+      }
+      throw dbError;
+    }
+  }
+  buildStackOrder(frames) {
+    const stack = [];
+    const rootFrame = frames.find((f) => !f.parent_frame_id);
+    if (!rootFrame) return [];
+    let currentFrame = rootFrame;
+    stack.push(currentFrame.frame_id);
+    while (currentFrame) {
+      const childFrame = frames.find(
+        (f) => f.parent_frame_id === currentFrame.frame_id
+      );
+      if (!childFrame) break;
+      stack.push(childFrame.frame_id);
+      currentFrame = childFrame;
+    }
+    return stack;
+  }
+  /**
+   * Create a new frame and push to stack
+   */
+  createFrame(options) {
+    return trace.traceSync("function", "FrameManager.createFrame", options, () => this._createFrame(options));
+  }
+  _createFrame(options) {
+    const frameId = uuidv4();
+    const parentFrameId = options.parentFrameId || this.getCurrentFrameId();
+    const depth = parentFrameId ? this.getFrameDepth(parentFrameId) + 1 : 0;
+    const frame = {
+      frame_id: frameId,
+      run_id: this.currentRunId,
+      project_id: this.projectId,
+      parent_frame_id: parentFrameId,
+      depth,
+      type: options.type,
+      name: options.name,
+      state: "active",
+      inputs: options.inputs || {},
+      created_at: Math.floor(Date.now() / 1e3)
+    };
+    try {
+      this.db.prepare(
+        `
+        INSERT INTO frames (
+          frame_id, run_id, project_id, parent_frame_id, depth, type, name, state, inputs, created_at
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+      `
+      ).run(
+        frame.frame_id,
+        frame.run_id,
+        frame.project_id,
+        frame.parent_frame_id,
+        frame.depth,
+        frame.type,
+        frame.name,
+        frame.state,
+        JSON.stringify(frame.inputs),
+        frame.created_at
+      );
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to create frame: ${options.name}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          frameId,
+          frameType: options.type,
+          frameName: options.name,
+          parentFrameId,
+          depth,
+          operation: "createFrame"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+    this.activeStack.push(frameId);
+    logger.info("Created frame", {
+      frameId,
+      type: options.type,
+      name: options.name,
+      depth,
+      parentFrameId,
+      stackDepth: this.activeStack.length
+    });
+    return frameId;
+  }
+  /**
+   * Close the current frame and generate digest
+   */
+  closeFrame(frameId, outputs) {
+    trace.traceSync("function", "FrameManager.closeFrame", { frameId, outputs }, () => this._closeFrame(frameId, outputs));
+  }
+  _closeFrame(frameId, outputs) {
+    const targetFrameId = frameId || this.getCurrentFrameId();
+    if (!targetFrameId) {
+      throw new FrameError(
+        "No active frame to close",
+        ErrorCode.FRAME_INVALID_STATE,
+        {
+          operation: "closeFrame",
+          activeStack: this.activeStack,
+          stackDepth: this.activeStack.length
+        }
+      );
+    }
+    const frame = this.getFrame(targetFrameId);
+    if (!frame) {
+      throw new FrameError(
+        `Frame not found: ${targetFrameId}`,
+        ErrorCode.FRAME_NOT_FOUND,
+        {
+          frameId: targetFrameId,
+          operation: "closeFrame",
+          runId: this.currentRunId
+        }
+      );
+    }
+    if (frame.state === "closed") {
+      logger.warn("Attempted to close already closed frame", {
+        frameId: targetFrameId
+      });
+      return;
+    }
+    const digest = this.generateDigest(targetFrameId);
+    const finalOutputs = { ...outputs, ...digest.structured };
+    try {
+      this.db.prepare(
+        `
+        UPDATE frames
+        SET state = 'closed',
+            outputs = ?,
+            digest_text = ?,
+            digest_json = ?,
+            closed_at = unixepoch()
+        WHERE frame_id = ?
+      `
+      ).run(
+        JSON.stringify(finalOutputs),
+        digest.text,
+        JSON.stringify(digest.structured),
+        targetFrameId
+      );
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to close frame: ${targetFrameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          frameId: targetFrameId,
+          frameName: frame.name,
+          operation: "closeFrame"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+    this.activeStack = this.activeStack.filter((id) => id !== targetFrameId);
+    this.closeChildFrames(targetFrameId);
+    logger.info("Closed frame", {
+      frameId: targetFrameId,
+      name: frame.name,
+      duration: Math.floor(Date.now() / 1e3) - frame.created_at,
+      digestLength: digest.text.length,
+      stackDepth: this.activeStack.length
+    });
+  }
+  closeChildFrames(parentFrameId) {
+    try {
+      const children = this.db.prepare(
+        `
+        SELECT frame_id FROM frames
+        WHERE parent_frame_id = ? AND state = 'active'
+      `
+      ).all(parentFrameId);
+      children.forEach((child) => {
+        try {
+          this.closeFrame(child.frame_id);
+        } catch (error) {
+          logger.error(
+            "Failed to close child frame",
+            error instanceof Error ? error : new Error(String(error)),
+            {
+              parentFrameId,
+              childFrameId: child.frame_id
+            }
+          );
+        }
+      });
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to close child frames for parent: ${parentFrameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          parentFrameId,
+          operation: "closeChildFrames"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+  }
+  /**
+   * Generate digest for a frame
+   */
+  generateDigest(frameId) {
+    const frame = this.getFrame(frameId);
+    const events = this.getFrameEvents(frameId);
+    const anchors = this.getFrameAnchors(frameId);
+    if (!frame) {
+      throw new FrameError(
+        `Cannot generate digest: frame not found ${frameId}`,
+        ErrorCode.FRAME_NOT_FOUND,
+        {
+          frameId,
+          operation: "generateDigest",
+          runId: this.currentRunId
+        }
+      );
+    }
+    const decisions = anchors.filter((a) => a.type === "DECISION");
+    const constraints = anchors.filter((a) => a.type === "CONSTRAINT");
+    const risks = anchors.filter((a) => a.type === "RISK");
+    const toolCalls = events.filter((e) => e.event_type === "tool_call");
+    const artifacts = events.filter((e) => e.event_type === "artifact");
+    const structured = {
+      result: frame.name,
+      decisions: decisions.map((d) => ({ id: d.anchor_id, text: d.text })),
+      constraints: constraints.map((c) => ({ id: c.anchor_id, text: c.text })),
+      risks: risks.map((r) => ({ id: r.anchor_id, text: r.text })),
+      artifacts: artifacts.map((a) => ({
+        kind: a.payload.kind || "unknown",
+        ref: a.payload.ref
+      })),
+      tool_calls_count: toolCalls.length,
+      duration_seconds: frame.closed_at ? frame.closed_at - frame.created_at : 0
+    };
+    const text = this.generateDigestText(frame, structured, events.length);
+    return { text, structured };
+  }
+  generateDigestText(frame, structured, eventCount) {
+    let summary = `Completed: ${frame.name}
+`;
+    if (structured.decisions.length > 0) {
+      summary += `
+Decisions made:
+${structured.decisions.map((d) => `- ${d.text}`).join("\n")}`;
+    }
+    if (structured.constraints.length > 0) {
+      summary += `
+Constraints established:
+${structured.constraints.map((c) => `- ${c.text}`).join("\n")}`;
+    }
+    if (structured.risks.length > 0) {
+      summary += `
+Risks identified:
+${structured.risks.map((r) => `- ${r.text}`).join("\n")}`;
+    }
+    summary += `
+Activity: ${eventCount} events, ${structured.tool_calls_count} tool calls`;
+    if (structured.duration_seconds > 0) {
+      summary += `, ${Math.floor(structured.duration_seconds / 60)}m ${structured.duration_seconds % 60}s duration`;
+    }
+    return summary;
+  }
+  /**
+   * Add event to current frame
+   */
+  addEvent(eventType, payload, frameId) {
+    const targetFrameId = frameId || this.getCurrentFrameId();
+    if (!targetFrameId) {
+      throw new FrameError(
+        "No active frame for event",
+        ErrorCode.FRAME_INVALID_STATE,
+        {
+          operation: "addEvent",
+          eventType,
+          activeStack: this.activeStack
+        }
+      );
+    }
+    const eventId = uuidv4();
+    const seq = this.getNextEventSequence(targetFrameId);
+    try {
+      this.db.prepare(
+        `
+        INSERT INTO events (event_id, run_id, frame_id, seq, event_type, payload)
+        VALUES (?, ?, ?, ?, ?, ?)
+      `
+      ).run(
+        eventId,
+        this.currentRunId,
+        targetFrameId,
+        seq,
+        eventType,
+        JSON.stringify(payload)
+      );
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to add event to frame: ${targetFrameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          eventId,
+          frameId: targetFrameId,
+          eventType,
+          seq,
+          operation: "addEvent"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+    return eventId;
+  }
+  /**
+   * Add anchor to frame
+   */
+  addAnchor(type, text, priority = 0, metadata = {}, frameId) {
+    const targetFrameId = frameId || this.getCurrentFrameId();
+    if (!targetFrameId) {
+      throw new FrameError(
+        "No active frame for anchor",
+        ErrorCode.FRAME_INVALID_STATE,
+        {
+          operation: "addAnchor",
+          anchorType: type,
+          text: text.substring(0, 100),
+          activeStack: this.activeStack
+        }
+      );
+    }
+    const anchorId = uuidv4();
+    try {
+      this.db.prepare(
+        `
+        INSERT INTO anchors (anchor_id, frame_id, project_id, type, text, priority, metadata)
+        VALUES (?, ?, ?, ?, ?, ?, ?)
+      `
+      ).run(
+        anchorId,
+        targetFrameId,
+        this.projectId,
+        type,
+        text,
+        priority,
+        JSON.stringify(metadata)
+      );
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to add anchor to frame: ${targetFrameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          anchorId,
+          frameId: targetFrameId,
+          anchorType: type,
+          operation: "addAnchor"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+    return anchorId;
+  }
+  /**
+   * Get hot stack context for current active frames
+   */
+  getHotStackContext(maxEvents = 20) {
+    return this.activeStack.map((frameId) => {
+      const frame = this.getFrame(frameId);
+      if (!frame) return null;
+      return {
+        frameId,
+        header: {
+          goal: frame.name,
+          constraints: this.extractConstraints(frame.inputs),
+          definitions: frame.inputs.definitions
+        },
+        anchors: this.getFrameAnchors(frameId),
+        recentEvents: this.getFrameEvents(frameId, maxEvents),
+        activeArtifacts: this.getActiveArtifacts(frameId)
+      };
+    }).filter(Boolean);
+  }
+  /**
+   * Get active frame path (root to current)
+   */
+  getActiveFramePath() {
+    return this.activeStack.map((frameId) => this.getFrame(frameId)).filter(Boolean);
+  }
+  // Utility methods
+  getCurrentFrameId() {
+    return this.activeStack[this.activeStack.length - 1];
+  }
+  getStackDepth() {
+    return this.activeStack.length;
+  }
+  getFrameDepth(frameId) {
+    const frame = this.getFrame(frameId);
+    return frame?.depth || 0;
+  }
+  getFrame(frameId) {
+    try {
+      const row = this.db.prepare(
+        `
+        SELECT * FROM frames WHERE frame_id = ?
+      `
+      ).get(frameId);
+      if (!row) return void 0;
+      return {
+        ...row,
+        inputs: JSON.parse(row.inputs || "{}"),
+        outputs: JSON.parse(row.outputs || "{}"),
+        digest_json: JSON.parse(row.digest_json || "{}")
+      };
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to get frame: ${frameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          frameId,
+          operation: "getFrame"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+  }
+  getFrameEvents(frameId, limit) {
+    try {
+      const query = limit ? `SELECT * FROM events WHERE frame_id = ? ORDER BY seq DESC LIMIT ?` : `SELECT * FROM events WHERE frame_id = ? ORDER BY seq ASC`;
+      const params = limit ? [frameId, limit] : [frameId];
+      const rows = this.db.prepare(query).all(...params);
+      return rows.map((row) => ({
+        ...row,
+        payload: JSON.parse(row.payload)
+      }));
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to get frame events: ${frameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          frameId,
+          limit,
+          operation: "getFrameEvents"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+  }
+  getFrameAnchors(frameId) {
+    try {
+      const rows = this.db.prepare(
+        `
+        SELECT * FROM anchors WHERE frame_id = ? ORDER BY priority DESC, created_at ASC
+      `
+      ).all(frameId);
+      return rows.map((row) => ({
+        ...row,
+        metadata: JSON.parse(row.metadata || "{}")
+      }));
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to get frame anchors: ${frameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          frameId,
+          operation: "getFrameAnchors"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+  }
+  getNextEventSequence(frameId) {
+    try {
+      const result = this.db.prepare(
+        `
+        SELECT MAX(seq) as max_seq FROM events WHERE frame_id = ?
+      `
+      ).get(frameId);
+      return (result.max_seq || 0) + 1;
+    } catch (error) {
+      throw new DatabaseError(
+        `Failed to get next event sequence for frame: ${frameId}`,
+        ErrorCode.DB_QUERY_FAILED,
+        {
+          frameId,
+          operation: "getNextEventSequence"
+        },
+        error instanceof Error ? error : void 0
+      );
+    }
+  }
+  extractConstraints(inputs) {
+    return inputs.constraints;
+  }
+  getActiveArtifacts(frameId) {
+    const artifacts = this.getFrameEvents(frameId).filter((e) => e.event_type === "artifact").map((e) => e.payload.ref).filter(Boolean);
+    return artifacts;
+  }
+}
+export {
+  FrameManager
+};
+//# sourceMappingURL=frame-manager.js.map