antigravity-ai-kit 2.1.0 → 3.0.1

package/lib/workflow-persistence.js

@@ -0,0 +1,160 @@
+/**
+ * Antigravity AI Kit — Workflow Persistence
+ *
+ * Extends the workflow engine with checkpoint/resume capability.
+ * Workflow state survives across agent sessions via JSON file snapshots.
+ *
+ * @module lib/workflow-persistence
+ * @author Emre Dursun
+ * @since v3.0.0
+ */
+
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const crypto = require('crypto');
+const workflowEngine = require('./workflow-engine');
+
+const AGENT_DIR = '.agent';
+const ENGINE_DIR = 'engine';
+const CHECKPOINTS_DIR = 'checkpoints';
+
+/**
+ * Resolves the checkpoints directory path.
+ *
+ * @param {string} projectRoot - Root directory of the project
+ * @returns {string} Absolute path to checkpoints directory
+ */
+function resolveCheckpointsDir(projectRoot) {
+  return path.join(projectRoot, AGENT_DIR, ENGINE_DIR, CHECKPOINTS_DIR);
+}
+
+/**
+ * Creates a checkpoint of the current workflow state.
+ *
+ * @param {string} projectRoot - Root directory of the project
+ * @param {string} [label] - Optional human-readable label
+ * @returns {{ checkpointId: string, filePath: string, timestamp: string }}
+ */
+function createCheckpoint(projectRoot, label) {
+  const { state } = workflowEngine.loadWorkflowState(projectRoot);
+  const checkpointsDir = resolveCheckpointsDir(projectRoot);
+
+  if (!fs.existsSync(checkpointsDir)) {
+    fs.mkdirSync(checkpointsDir, { recursive: true });
+  }
+
+  const timestamp = new Date().toISOString();
+  const checkpointId = crypto.randomUUID().slice(0, 8);
+
+  const checkpoint = {
+    id: checkpointId,
+    label: label || `Checkpoint at ${state.currentPhase}`,
+    timestamp,
+    phase: state.currentPhase,
+    state: JSON.parse(JSON.stringify(state)),
+  };
+
+  const filePath = path.join(checkpointsDir, `${checkpointId}.json`);
+  fs.writeFileSync(filePath, JSON.stringify(checkpoint, null, 2) + '\n', 'utf-8');
+
+  return { checkpointId, filePath, timestamp };
+}
+
+/**
+ * Lists all available checkpoints, sorted by timestamp (newest first).
+ *
+ * @param {string} projectRoot - Root directory of the project
+ * @returns {Array<{ id: string, label: string, phase: string, timestamp: string }>}
+ */
+function listCheckpoints(projectRoot) {
+  const checkpointsDir = resolveCheckpointsDir(projectRoot);
+
+  if (!fs.existsSync(checkpointsDir)) {
+    return [];
+  }
+
+  const files = fs.readdirSync(checkpointsDir).filter((f) => f.endsWith('.json'));
+
+  return files
+    .map((file) => {
+      try {
+        const data = JSON.parse(fs.readFileSync(path.join(checkpointsDir, file), 'utf-8'));
+        return { id: data.id, label: data.label, phase: data.phase, timestamp: data.timestamp };
+      } catch {
+        return null;
+      }
+    })
+    .filter(Boolean)
+    .sort((a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime());
+}
+
+/**
+ * Restores workflow state from a checkpoint.
+ *
+ * @param {string} projectRoot - Root directory of the project
+ * @param {string} checkpointId - ID of the checkpoint to restore
+ * @returns {{ success: boolean, restoredPhase: string }}
+ */
+function resumeFromCheckpoint(projectRoot, checkpointId) {
+  const checkpointsDir = resolveCheckpointsDir(projectRoot);
+  const filePath = path.join(checkpointsDir, `${checkpointId}.json`);
+
+  if (!fs.existsSync(filePath)) {
+    throw new Error(`Checkpoint not found: ${checkpointId}`);
+  }
+
+  const checkpoint = JSON.parse(fs.readFileSync(filePath, 'utf-8'));
+  const statePath = path.join(projectRoot, AGENT_DIR, ENGINE_DIR, 'workflow-state.json');
+  const tempPath = `${statePath}.tmp`;
+
+  // Append restore event to history
+  checkpoint.state.history.push({
+    from: 'RESTORED',
+    to: checkpoint.state.currentPhase,
+    trigger: `Restored from checkpoint ${checkpointId}`,
+    timestamp: new Date().toISOString(),
+  });
+
+  fs.writeFileSync(tempPath, JSON.stringify(checkpoint.state, null, 2) + '\n', 'utf-8');
+  fs.renameSync(tempPath, statePath);
+
+  return { success: true, restoredPhase: checkpoint.state.currentPhase };
+}
+
+/**
+ * Returns a rich summary of the current workflow state.
+ *
+ * @param {string} projectRoot - Root directory of the project
+ * @returns {object} Workflow summary with phase durations and checkpoint count
+ */
+function getWorkflowSummary(projectRoot) {
+  const { state } = workflowEngine.loadWorkflowState(projectRoot);
+  const checkpoints = listCheckpoints(projectRoot);
+  const history = state.history || [];
+
+  const phaseDurations = {};
+  for (const [phaseName, phaseData] of Object.entries(state.phases || {})) {
+    if (phaseData.startedAt && phaseData.completedAt) {
+      const duration = new Date(phaseData.completedAt).getTime() - new Date(phaseData.startedAt).getTime();
+      phaseDurations[phaseName] = Math.round(duration / 1000);
+    }
+  }
+
+  return {
+    currentPhase: state.currentPhase,
+    startedAt: state.startedAt,
+    transitionCount: history.length,
+    checkpointCount: checkpoints.length,
+    phaseDurations,
+    lastTransition: history.length > 0 ? history[history.length - 1] : null,
+  };
+}
+
+module.exports = {
+  createCheckpoint,
+  listCheckpoints,
+  resumeFromCheckpoint,
+  getWorkflowSummary,
+};

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "antigravity-ai-kit",
-  "version": "2.1.0",
-  "description": "🚀 Trust-Grade AI development framework — 17 Agents, 27 Skills, 31 Commands, 11 Workflows. Transform your IDE into an autonomous engineering team.",
+  "version": "3.0.1",
+  "description": "🚀 Trust-Grade AI development framework with a 21-module runtime engine — 19 Agents, 31 Skills, 31 Commands, 14 Workflows, 261 Tests. Workflow enforcement, task governance, agent reputation, self-healing, and skill marketplace.",
   "main": "bin/ag-kit.js",
   "bin": {
     "ag-kit": "./bin/ag-kit.js"
@@ -41,8 +41,12 @@
   },
   "files": [
     "bin/",
+    "lib/",
     ".agent/",
     "README.md",
     "LICENSE"
-  ]
+  ],
+  "devDependencies": {
+    "vitest": "^4.1.0"
+  }
 }