@skyramp/mcp 0.0.49 → 0.0.51

package/build/utils/AnalysisStateManager.js

@@ -3,45 +3,68 @@ import * as path from "path";
 import * as os from "os";
 import { logger } from "./logger.js";
 /**
- * Manages persistent state for test analysis workflow
+ * File prefix mapping for each state type
+ */
+const STATE_FILE_PREFIXES = {
+    analysis: "skyramp-analysis", // For test maintenance workflow
+    recommendation: "skyramp-recommendation", // For test recommendation workflow
+};
+/**
+ * Generic State Manager for persisting workflow data
  * Reduces token usage by storing intermediate results in filesystem
+ *
+ * Supports two state types:
+ * - analysis: Test maintenance workflow (discovery, drift, execution, health)
+ * - recommendation: Test recommendation workflow (analyze repo, map tests, recommendations)
  */
-export class AnalysisStateManager {
+export class StateManager {
     stateFile;
     sessionId;
+    stateType;
     /**
      * Create a new state manager
+     * @param stateType Type of state (analysis, recommendation)
      * @param sessionId Unique session identifier (defaults to timestamp)
      * @param stateDir Directory to store state files (defaults to /tmp)
      */
-    constructor(sessionId, stateDir) {
+    constructor(stateType = "analysis", sessionId, stateDir) {
+        this.stateType = stateType;
         this.sessionId = sessionId || Date.now().toString();
         const baseDir = stateDir || os.tmpdir();
-        this.stateFile = path.join(baseDir, `skyramp-analysis-${this.sessionId}.json`);
+        const prefix = STATE_FILE_PREFIXES[stateType];
+        this.stateFile = path.join(baseDir, `${prefix}-${this.sessionId}.json`);
     }
     /**
      * Create state manager from existing state file path
      */
     static fromStatePath(stateFilePath) {
         const basename = path.basename(stateFilePath, ".json");
-        const sessionId = basename.replace("skyramp-analysis-", "");
         const stateDir = path.dirname(stateFilePath);
-        return new AnalysisStateManager(sessionId, stateDir);
+        // Determine state type from filename
+        let stateType = "analysis";
+        let sessionId = basename;
+        for (const [type, prefix] of Object.entries(STATE_FILE_PREFIXES)) {
+            if (basename.startsWith(prefix)) {
+                stateType = type;
+                sessionId = basename.replace(`${prefix}-`, "");
+                break;
+            }
+        }
+        return new StateManager(stateType, sessionId, stateDir);
     }
     /**
-     * Read current state from file
-     * @returns Test analysis results, or empty array if file doesn't exist
+     * Read data from state file (excludes metadata)
      */
-    async readState() {
+    async readData() {
         if (!this.exists()) {
             logger.debug(`State file does not exist: ${this.stateFile}`);
-            return [];
+            return null;
         }
         try {
             const content = await fs.promises.readFile(this.stateFile, "utf-8");
-            const state = JSON.parse(content);
-            logger.debug(`Read ${state.tests.length} tests from state file: ${this.stateFile}`);
-            return state.tests;
+            const { metadata, ...data } = JSON.parse(content);
+            logger.debug(`Read data from state file: ${this.stateFile}`);
+            return data;
         }
         catch (error) {
             logger.error(`Failed to read state file: ${error.message}`);
@@ -65,11 +88,10 @@ export class AnalysisStateManager {
         }
     }
     /**
-     * Write test results to state file
-     * @param tests Test analysis results
-     * @param options Additional metadata options
+     * Write data to state file
+     * Data fields are spread at root level alongside metadata
      */
-    async writeState(tests, options) {
+    async writeData(data, options) {
         try {
             // Read existing metadata if file exists
             let existingMetadata;
@@ -78,9 +100,10 @@ export class AnalysisStateManager {
                 existingMetadata = existing?.metadata;
             }
             const state = {
-                tests,
+                ...data,
                 metadata: {
                     sessionId: this.sessionId,
+                    stateType: this.stateType,
                     repositoryPath: options?.repositoryPath || existingMetadata?.repositoryPath,
                     createdAt: existingMetadata?.createdAt || new Date().toISOString(),
                     updatedAt: new Date().toISOString(),
@@ -88,49 +111,13 @@ export class AnalysisStateManager {
                 },
             };
             await fs.promises.writeFile(this.stateFile, JSON.stringify(state, null, 2), "utf-8");
-            logger.debug(`Wrote ${tests.length} tests to state file: ${this.stateFile}`);
+            logger.debug(`Wrote data to state file: ${this.stateFile}`);
         }
         catch (error) {
             logger.error(`Failed to write state file: ${error.message}`);
             throw new Error(`Failed to write state file: ${error.message}`);
         }
     }
-    // /**
-    //  * Read state with filtering
-    //  * @param filter Function to filter tests
-    //  */
-    // async readStateFiltered(
-    //   filter: (test: TestAnalysisResult) => boolean,
-    // ): Promise<TestAnalysisResult[]> {
-    //   const allTests = await this.readState();
-    //   return allTests.filter(filter);
-    // }
-    // /**
-    //  * Read state with pagination
-    //  * @param page Page number (0-indexed)
-    //  * @param pageSize Number of tests per page
-    //  */
-    // async readStatePaginated(
-    //   page: number,
-    //   pageSize: number = 100,
-    // ): Promise<{
-    //   tests: TestAnalysisResult[];
-    //   hasMore: boolean;
-    //   total: number;
-    //   page: number;
-    //   pageSize: number;
-    // }> {
-    //   const allTests = await this.readState();
-    //   const start = page * pageSize;
-    //   const end = start + pageSize;
-    //   return {
-    //     tests: allTests.slice(start, end),
-    //     hasMore: end < allTests.length,
-    //     total: allTests.length,
-    //     page,
-    //     pageSize,
-    //   };
-    // }
     /**
      * Get the state file path
      */
@@ -143,6 +130,12 @@ export class AnalysisStateManager {
     getSessionId() {
         return this.sessionId;
     }
+    /**
+     * Get the state type
+     */
+    getStateType() {
+        return this.stateType;
+    }
     /**
      * Check if state file exists
      */
@@ -184,12 +177,17 @@ export class AnalysisStateManager {
      * Cleanup old state files
      * @param maxAgeHours Maximum age in hours
      * @param stateDir Directory to clean (defaults to /tmp)
+     * @param stateTypes Which state types to clean (defaults to all)
      * @returns Number of files deleted
      */
-    static async cleanupOldStateFiles(maxAgeHours = 24, stateDir) {
+    static async cleanupOldStateFiles(maxAgeHours = 24, stateDir, stateTypes) {
         const baseDir = stateDir || os.tmpdir();
         const files = await fs.promises.readdir(baseDir);
-        const stateFiles = files.filter((f) => f.startsWith("skyramp-analysis-"));
+        // Get prefixes to clean
+        const prefixesToClean = stateTypes
+            ? stateTypes.map((t) => STATE_FILE_PREFIXES[t])
+            : Object.values(STATE_FILE_PREFIXES);
+        const stateFiles = files.filter((f) => prefixesToClean.some((prefix) => f.startsWith(prefix)));
         let deletedCount = 0;
         const now = Date.now();
         const maxAge = maxAgeHours * 60 * 60 * 1000;
@@ -216,19 +214,32 @@ export class AnalysisStateManager {
     /**
      * List all state files in directory
      * @param stateDir Directory to search (defaults to /tmp)
+     * @param stateTypes Which state types to list (defaults to all)
      */
-    static async listStateFiles(stateDir) {
+    static async listStateFiles(stateDir, stateTypes) {
         const baseDir = stateDir || os.tmpdir();
         const files = await fs.promises.readdir(baseDir);
-        const stateFiles = files.filter((f) => f.startsWith("skyramp-analysis-"));
+        // Get prefixes to list
+        const prefixesToList = stateTypes
+            ? stateTypes.map((t) => STATE_FILE_PREFIXES[t])
+            : Object.values(STATE_FILE_PREFIXES);
+        const stateFiles = files.filter((f) => prefixesToList.some((prefix) => f.startsWith(prefix)));
         const fileInfos = await Promise.all(stateFiles.map(async (file) => {
             const filePath = path.join(baseDir, file);
             const stats = await fs.promises.stat(filePath);
-            const sessionId = file
-                .replace("skyramp-analysis-", "")
-                .replace(".json", "");
+            // Determine state type and session ID
+            let stateType = "analysis";
+            let sessionId = file.replace(".json", "");
+            for (const [type, prefix] of Object.entries(STATE_FILE_PREFIXES)) {
+                if (file.startsWith(prefix)) {
+                    stateType = type;
+                    sessionId = file.replace(`${prefix}-`, "").replace(".json", "");
+                    break;
+                }
+            }
             return {
                 sessionId,
+                stateType,
                 path: filePath,
                 size: stats.size,
                 createdAt: stats.birthtime,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skyramp/mcp",
-  "version": "0.0.49",
+  "version": "0.0.51",
   "main": "build/index.js",
   "type": "module",
   "bin": {
@@ -14,6 +14,8 @@
     "start": "node build/index.js",
     "test:startup": "time node build/index.js --help",
     "pretty": "npx prettier --write .",
+    "lint": "eslint src --ext .ts",
+    "lint:fix": "eslint src --ext .ts --fix",
     "test": "npx jest"
   },
   "files": [
@@ -43,7 +45,7 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.24.3",
-    "@skyramp/skyramp": "^1.3.1",
+    "@skyramp/skyramp": "^1.3.3",
     "@playwright/test": "^1.55.0",
     "dockerode": "^4.0.6",
     "fast-glob": "^3.3.3",
@@ -55,6 +57,9 @@
     "@types/jest": "^29.5.14",
     "@types/mocha": "^10.0.10",
     "@types/node": "^22.15.19",
+    "@typescript-eslint/eslint-plugin": "^8.0.0",
+    "@typescript-eslint/parser": "^8.0.0",
+    "eslint": "^9.0.0",
     "jest": "^29.7.0",
     "ts-jest": "^29.3.4",
     "typescript": "^5.8.3"