auto-feedback 0.1.0 → 1.1.0

package/build/screenshot/capture.d.ts

@@ -12,10 +12,8 @@ export declare function capturePlaywrightPage(page: Page, options?: {
 }): Promise<Buffer>;
 /**
  * Capture a screenshot of a Windows desktop window by PID.
- * Uses node-screenshots to enumerate windows and match by process ID.
+ * Uses tasklist to resolve PID to process name, then node-screenshots
+ * to enumerate windows and match by appName.
  * Returns raw PNG buffer for further optimization.
- *
- * Note: node-screenshots Window objects may not expose PID in all versions.
- * Falls back to matching by window title if PID is unavailable.
  */
 export declare function captureDesktopWindow(pid: number): Promise<Buffer>;

package/build/screenshot/capture.js

@@ -14,34 +14,48 @@ export async function capturePlaywrightPage(page, options) {
 }
 /**
  * Capture a screenshot of a Windows desktop window by PID.
- * Uses node-screenshots to enumerate windows and match by process ID.
+ * Uses tasklist to resolve PID to process name, then node-screenshots
+ * to enumerate windows and match by appName.
  * Returns raw PNG buffer for further optimization.
- *
- * Note: node-screenshots Window objects may not expose PID in all versions.
- * Falls back to matching by window title if PID is unavailable.
  */
 export async function captureDesktopWindow(pid) {
-    // Dynamic import to avoid load-time failures on non-Windows
+    const { execSync } = await import("child_process");
     const { Window } = await import("node-screenshots");
+    // Resolve PID to process name via tasklist
+    let processName;
+    try {
+        const output = execSync(`tasklist /FI "PID eq ${pid}" /FO CSV /NH`, { encoding: "utf-8" }).trim();
+        // tasklist returns "INFO: No tasks are running..." when PID not found
+        if (output.includes("INFO:") || !output.startsWith('"')) {
+            throw new Error(`Could not find process name for PID ${pid}. The process may have exited.`);
+        }
+        // CSV format: "processname.exe","1234","Console","1","12,345 K"
+        const match = output.match(/^"([^"]+)"/);
+        if (!match) {
+            throw new Error(`Could not find process name for PID ${pid}. The process may have exited.`);
+        }
+        // Strip .exe suffix for matching against appName
+        processName = match[1].replace(/\.exe$/i, "");
+    }
+    catch (error) {
+        if (error instanceof Error && error.message.includes("Could not find")) {
+            throw error;
+        }
+        throw new Error(`Could not find process name for PID ${pid}. The process may have exited.`);
+    }
+    console.error(`[captureDesktopWindow] PID ${pid} resolved to process name: ${processName}`);
+    // Find matching window by appName
     const windows = Window.all();
-    // node-screenshots Window type doesn't expose pid in current typings,
-    // but some builds include it at runtime. Use any cast for defensive check.
-    const target = windows.find((w) => {
-        const wAny = w;
-        if (typeof wAny.processId === "number")
-            return wAny.processId === pid;
-        if (typeof wAny.pid === "number")
-            return wAny.pid === pid;
-        if (typeof wAny.pid === "function")
-            return wAny.pid() === pid;
-        return false;
-    });
+    const target = windows.find((w) => w.appName.toLowerCase().includes(processName.toLowerCase()));
     if (!target) {
-        throw new Error(`No window found for PID ${pid}. The process may not have a visible window yet.`);
+        throw new Error(`No window found for process "${processName}" (PID ${pid}). ` +
+            `The process may not have a visible window yet. ` +
+            `Available windows: ${windows.map((w) => w.appName).join(", ")}`);
     }
     if (target.isMinimized) {
-        throw new Error(`Window for PID ${pid} is minimized. Restore it before capturing.`);
+        throw new Error(`Window for "${processName}" (PID ${pid}) is minimized. Restore it before capturing.`);
     }
+    console.error(`[captureDesktopWindow] Capturing window: "${target.appName}" (${target.width}x${target.height})`);
     const image = await target.captureImage();
     const pngData = await image.toPng();
     return Buffer.from(pngData);

package/build/session-manager.d.ts

@@ -50,6 +50,13 @@ export declare class SessionManager {
      * Remove a specific page reference
      */
     removePageRef(sessionId: string, identifier: string): void;
+    /**
+     * Re-key all Maps associated with a page identifier.
+     * Used by navigate when goto changes the page URL.
+     * Updates: pageRefs, consoleCollectors, errorCollectors, networkCollectors, processCollectors.
+     * No-op for entries that don't exist under the old key.
+     */
+    rekeyIdentifier(sessionId: string, oldIdentifier: string, newIdentifier: string): void;
     /**
      * Store the latest auto-captured screenshot for a session
      */

package/build/session-manager.js

@@ -87,6 +87,36 @@ export class SessionManager {
     removePageRef(sessionId, identifier) {
         this.pageRefs.delete(`${sessionId}:${identifier}`);
     }
+    /**
+     * Re-key all Maps associated with a page identifier.
+     * Used by navigate when goto changes the page URL.
+     * Updates: pageRefs, consoleCollectors, errorCollectors, networkCollectors, processCollectors.
+     * No-op for entries that don't exist under the old key.
+     */
+    rekeyIdentifier(sessionId, oldIdentifier, newIdentifier) {
+        const oldKey = `${sessionId}:${oldIdentifier}`;
+        const newKey = `${sessionId}:${newIdentifier}`;
+        // Re-key page ref
+        const pageRef = this.pageRefs.get(oldKey);
+        if (pageRef) {
+            this.pageRefs.delete(oldKey);
+            this.pageRefs.set(newKey, pageRef);
+        }
+        // Re-key all collector maps (silently skip if old key not present)
+        const maps = [
+            this.consoleCollectors,
+            this.errorCollectors,
+            this.networkCollectors,
+            this.processCollectors,
+        ];
+        for (const map of maps) {
+            const value = map.get(oldKey);
+            if (value) {
+                map.delete(oldKey);
+                map.set(newKey, value);
+            }
+        }
+    }
     /**
      * Store the latest auto-captured screenshot for a session
      */

package/build/tools/index.js

@@ -2,7 +2,12 @@
  * MCP tool registration
  */
 import { z } from "zod";
+import { readFileSync } from "fs";
+import { resolve, dirname } from "path";
+import { fileURLToPath } from "url";
 import { createToolError, createToolResult } from "../utils/errors.js";
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const pkg = JSON.parse(readFileSync(resolve(__dirname, "../../package.json"), "utf-8"));
 import { registerCheckPortTool } from "./check-port.js";
 import { registerLaunchWebServerTool } from "./launch-web-server.js";
 import { registerLaunchElectronTool } from "./launch-electron.js";
@@ -30,7 +35,7 @@ export function registerTools(server, sessionManager) {
     server.tool("get_version", "Get server version and capabilities. Use to verify the server is running and responsive.", {}, async () => {
         return createToolResult({
             name: "feedback",
-            version: "0.1.0",
+            version: pkg.version,
             status: "ready",
             capabilities: [
                 "process_lifecycle",

package/build/tools/navigate.js

@@ -70,16 +70,14 @@ export function registerNavigateTool(server, sessionManager) {
                     waitUntil: effectiveWaitUntil,
                     timeout: effectiveTimeout,
                 });
-                // Update PageReference URL in SessionManager for web pages
-                // so page discovery continues to work with the new URL
+                // Re-key page ref AND all collector maps atomically
+                // so page discovery and diagnostic lookups work with the new URL
                 if (pageType === "web" && currentIdentifier !== "electron") {
-                    const oldRef = sessionManager.getPageRef(sessionId, currentIdentifier);
-                    if (oldRef) {
-                        sessionManager.removePageRef(sessionId, currentIdentifier);
-                        sessionManager.setPageRef(sessionId, url, {
-                            ...oldRef,
-                            url: url,
-                        });
+                    sessionManager.rekeyIdentifier(sessionId, currentIdentifier, url);
+                    // Update the URL field in the re-keyed page ref
+                    const updatedRef = sessionManager.getPageRef(sessionId, url);
+                    if (updatedRef) {
+                        updatedRef.url = url;
                     }
                 }
             }

package/build/tools/screenshot-web.js

@@ -60,8 +60,7 @@ export function registerScreenshotWebTool(server, sessionManager) {
                     viewport: { width: 1280, height: 720 },
                 });
                 const page = await context.newPage();
-                await page.goto(url, { waitUntil: "load", timeout: 30000 });
-                // Store page reference for reuse
+                // Store page reference before goto (URL is known from parameter)
                 pageRef = {
                     type: "web",
                     page,
@@ -72,7 +71,7 @@ export function registerScreenshotWebTool(server, sessionManager) {
                 sessionManager.setPageRef(sessionId, url, pageRef);
                 // Attach auto-capture on navigation
                 setupAutoCapture(page, sessionId, sessionManager);
-                // Attach diagnostic collectors
+                // Attach diagnostic collectors BEFORE goto so initial load events are captured
                 const consoleCollector = attachConsoleCollector(page);
                 const errorCollector = attachErrorCollector(page);
                 const networkCollector = attachNetworkCollector(page);
@@ -87,6 +86,8 @@ export function registerScreenshotWebTool(server, sessionManager) {
                         await browser.close().catch(() => { });
                     },
                 });
+                // Navigate AFTER collectors are attached to capture all initial load events
+                await page.goto(url, { waitUntil: "load", timeout: 30000 });
                 console.error(`[screenshot_web] Browser ready for ${url}`);
             }
             else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "auto-feedback",
-  "version": "0.1.0",
+  "version": "1.1.0",
   "type": "module",
   "description": "MCP server for GUI testing and feedback - gives Claude Code eyes and hands",
   "bin": {
@@ -12,7 +12,11 @@
   "scripts": {
     "build": "tsc",
     "dev": "tsx src/index.ts",
-    "prepublishOnly": "npm run build"
+    "prepublishOnly": "npm run build",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:unit": "vitest run tests/unit",
+    "test:integration": "vitest run tests/integration"
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.0",
@@ -27,8 +31,11 @@
   "devDependencies": {
     "@types/node": "^20.0.0",
     "@types/wait-on": "^5.3.4",
+    "electron": "^40.2.1",
     "tsx": "^4.0.0",
-    "typescript": "^5.0.0"
+    "typescript": "^5.0.0",
+    "vite": "^7.3.1",
+    "vitest": "^4.0.18"
   },
   "engines": {
     "node": ">=18.0.0"