@sudobility/testomniac_runner 0.0.130 → 0.0.132

package/bun.lock

@@ -8,8 +8,8 @@
         "@noble/curves": "^1.0.0",
         "@noble/hashes": "^1.0.0",
         "@sudobility/signic_sdk": "^0.1.7",
-        "@sudobility/testomniac_runner_service": "^0.1.128",
-        "@sudobility/testomniac_types": "^0.0.67",
+        "@sudobility/testomniac_runner_service": "^0.1.131",
+        "@sudobility/testomniac_types": "^0.0.69",
         "hono": "^4.7.0",
         "jose": "^6.1.2",
         "openai": "^6.7.0",
@@ -172,9 +172,9 @@
 
     "@sudobility/signic_sdk": ["@sudobility/signic_sdk@0.1.7", "", {}, "sha512-5XSgHSVsmyrMQ/ui1nDywwzt9dbRCsaeJ5tX6mKw2ZXbTZ82OsMr+dqDyV9XV3pfy7IHRIZq73af5KBamx72Fw=="],
 
-    "@sudobility/testomniac_runner_service": ["@sudobility/testomniac_runner_service@0.1.128", "", { "peerDependencies": { "@sudobility/testomniac_types": "^0.0.67", "openai": ">=6.0.0", "react": ">=18.0.0" }, "optionalPeers": ["openai", "react"] }, "sha512-YWFjcMmZROSHSNUZLt9czUJeWoYmIj69JCXUP9CD9sw5/UTzFFLwIdkqtAE2lG4RDo/Yb/5heZPmffZ1EfPPqw=="],
+    "@sudobility/testomniac_runner_service": ["@sudobility/testomniac_runner_service@0.1.131", "", { "peerDependencies": { "@sudobility/testomniac_types": "^0.0.69", "openai": ">=6.0.0", "react": ">=18.0.0" }, "optionalPeers": ["openai", "react"] }, "sha512-1ZJk1EjCJlFVXJXhlU3MLNPL53VwUJ6yOq4O9qvkpEn4VoCBnZlmVNOjicVXZE0UEP2u32GUGvBNEUHJGE5rFg=="],
 
-    "@sudobility/testomniac_types": ["@sudobility/testomniac_types@0.0.67", "", { "peerDependencies": { "@sudobility/types": "^1.9.62" } }, "sha512-kNBDk7AsFx1rqUWLq4nx0nf1alV9G7U5xbxj8qDvZKa+cCnUSiZFRtDB88ENWQzeIPVJAU8MaVo5yUKVL46eYg=="],
+    "@sudobility/testomniac_types": ["@sudobility/testomniac_types@0.0.69", "", { "peerDependencies": { "@sudobility/types": "^1.9.62" } }, "sha512-VkWcWdy4zwddCZspzc0qhHIqyfuuzyMUh6Dvjk614y8+A8GQ8N6D9AiNmFtqrDOoMEnfo9VNU8AvdJGOjjc/1A=="],
 
     "@sudobility/types": ["@sudobility/types@1.9.61", "", {}, "sha512-SODGpstB/iKfK3H/4BvJx/FBcc1h3gutUjGotyxN19VnOfWyzaDoEmW7eyoxOAYhZyXMXagSiii+NIEZvuxKog=="],
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sudobility/testomniac_runner",
-  "version": "0.0.130",
+  "version": "0.0.132",
   "publishConfig": {
     "access": "public"
   },
@@ -24,8 +24,8 @@
     "@noble/curves": "^1.0.0",
     "@noble/hashes": "^1.0.0",
     "@sudobility/signic_sdk": "^0.1.7",
-    "@sudobility/testomniac_runner_service": "^0.1.128",
-    "@sudobility/testomniac_types": "^0.0.67",
+    "@sudobility/testomniac_runner_service": "^0.1.131",
+    "@sudobility/testomniac_types": "^0.0.69",
     "hono": "^4.7.0",
     "jose": "^6.1.2",
     "openai": "^6.7.0",

package/src/adapters/PuppeteerAdapter.ts

@@ -241,9 +241,10 @@ export class PuppeteerAdapter implements BrowserAdapter {
     type?: string;
     quality?: number;
   }): Promise<Uint8Array> {
+    const type = (options?.type as "jpeg" | "png") || "jpeg";
     const buffer = await this.page.screenshot({
-      type: (options?.type as "jpeg" | "png") || "jpeg",
-      quality: options?.quality || 72,
+      type,
+      ...(type === "jpeg" ? { quality: options?.quality || 72 } : {}),
     });
     return new Uint8Array(buffer);
   }

package/src/index.ts

@@ -44,6 +44,7 @@ if (import.meta.main) {
       "runner-id": { type: "string" },
       "base-url": { type: "string" },
       "size-class": { type: "string", default: "desktop" },
+      "scan-mode": { type: "string" },
     },
     strict: false,
   });
@@ -54,6 +55,7 @@ if (import.meta.main) {
     const runnerId = Number(args["runner-id"]);
     const baseUrl = String(args["base-url"] ?? "");
     const sizeClass = String(args["size-class"] ?? "desktop");
+    const scanMode = readScanMode(args["scan-mode"]);
 
     logger.info({ scanId, runnerId }, "one-shot mode: executing test run");
     try {
@@ -63,6 +65,7 @@ if (import.meta.main) {
         scanUrl: baseUrl,
         baseUrl,
         sizeClass,
+        scanMode,
         runnerInstanceId: crypto.randomUUID(),
         runnerInstanceName: "mcp-runner",
       });
@@ -104,6 +107,14 @@ if (import.meta.main) {
   }
 }
 
+function readScanMode(
+  value: unknown
+): "full" | "partial" | "minimum" | undefined {
+  return value === "full" || value === "partial" || value === "minimum"
+    ? value
+    : undefined;
+}
+
 export default {
   port,
   fetch: app.fetch,

package/src/orchestrator.ts

@@ -40,14 +40,12 @@ export interface RunOptions {
   loginUrl?: string;
   entityCredentialId?: number;
   quickScan?: boolean;
+  scanMode?: "full" | "partial" | "minimum";
 }
 
 export async function runFullScan(options: RunOptions): Promise<void> {
   const config = loadConfig();
-  const api = getApiClient(
-    config.apiUrl + "/api/v1/scanner",
-    config.scannerApiKey
-  );
+  const api = getApiClient(config.apiUrl, config.scannerApiKey);
   const { scanUrl, baseUrl, userEmail } = options;
   const runnerName = options.runnerName || new URL(scanUrl).hostname;
   const sizeClass = (options.sizeClass as SizeClass) || SizeClass.Desktop;
@@ -65,7 +63,12 @@ export async function runFullScan(options: RunOptions): Promise<void> {
     const page = await chromium.newPage(defaultScreen);
     const adapter = new PuppeteerAdapter(page);
 
-    const eventHandler: ScanEventHandler = {
+    const eventHandler: ScanEventHandler & {
+      onStatusUpdate?: (update: {
+        message: string;
+        testRunId?: number;
+      }) => void;
+    } = {
       onPageFound: p =>
         logger.info(
           { relativePath: p.relativePath, pageId: p.pageId },
@@ -91,6 +94,8 @@ export async function runFullScan(options: RunOptions): Promise<void> {
       onFindingCreated: f =>
         logger.warn({ type: f.type, title: f.title }, "finding created"),
       onStatsUpdated: stats => logger.debug(stats, "stats updated"),
+      onStatusUpdate: update =>
+        logger.info({ testRunId: update.testRunId }, update.message),
       onScreenshotCaptured: () => {},
       onScanComplete: summary => logger.info(summary, "scan complete"),
       onError: err => logger.error(err, "scan error"),
@@ -115,6 +120,7 @@ export async function runFullScan(options: RunOptions): Promise<void> {
         entityCredentialId: options.entityCredentialId,
         credentials: options.credentials,
         quickScan: options.quickScan,
+        scanMode: options.scanMode,
       },
       api,
       expertises,
@@ -138,7 +144,11 @@ export async function runFullScan(options: RunOptions): Promise<void> {
       });
     }
 
-    await page.close();
+    try {
+      await page.close();
+    } catch (err) {
+      logger.debug({ err }, "page already closed during scan cleanup");
+    }
 
     logger.info(
       { scanId, sizeClass, totalDurationMs: elapsed(runStart) },
@@ -159,10 +169,7 @@ export async function runSequenceScan(
   options: SequenceRunOptions
 ): Promise<void> {
   const config = loadConfig();
-  const api = getApiClient(
-    config.apiUrl + "/api/v1/scanner",
-    config.scannerApiKey
-  );
+  const api = getApiClient(config.apiUrl, config.scannerApiKey);
 
   const runStart = Date.now();
   const sizeClass = (options.sizeClass as SizeClass) || SizeClass.Desktop;
@@ -177,7 +184,12 @@ export async function runSequenceScan(
     const adapter = new PuppeteerAdapter(page);
     const expertises = createDefaultExpertises();
 
-    const eventHandler: ScanEventHandler = {
+    const eventHandler: ScanEventHandler & {
+      onStatusUpdate?: (update: {
+        message: string;
+        testRunId?: number;
+      }) => void;
+    } = {
       onPageFound: p => logger.info(p, "page found"),
       onPageStateCreated: s => logger.info(s, "page state created"),
       onTestSurfaceCreated: s => logger.info(s, "surface created"),
@@ -186,6 +198,8 @@ export async function runSequenceScan(
       onTestRunCompleted: r => logger.info(r, "run completed"),
       onFindingCreated: f => logger.warn(f, "finding created"),
       onStatsUpdated: s => logger.debug(s, "stats"),
+      onStatusUpdate: update =>
+        logger.info({ testRunId: update.testRunId }, update.message),
       onScreenshotCaptured: () => {},
       onScanComplete: s => logger.info(s, "complete"),
       onError: e => logger.error(e, "error"),
@@ -205,7 +219,11 @@ export async function runSequenceScan(
       eventHandler
     );
 
-    await page.close();
+    try {
+      await page.close();
+    } catch (err) {
+      logger.debug({ err }, "page already closed during sequence cleanup");
+    }
 
     logger.info(
       {

package/src/runner-manager.ts

@@ -33,10 +33,7 @@ export class RunnerManager {
     this.tickInFlight = true;
     try {
       const config = loadConfig();
-      const api = getApiClient(
-        config.apiUrl + "/api/v1/scanner",
-        config.scannerApiKey
-      );
+      const api = getApiClient(config.apiUrl, config.scannerApiKey);
 
       while (this.activeRuns.size < this.maxConcurrentRunners) {
         const pendingRun = await api.getPendingTestRun();
@@ -57,23 +54,50 @@ export class RunnerManager {
         }
 
         const slotNumber = this.activeRuns.size + 1;
-        const runner = await api.getRunner(pendingRun.runnerId);
+
+        let runner;
+        try {
+          runner = await api.getRunner(pendingRun.runnerId);
+        } catch (err) {
+          logger.error(
+            { err, runnerId: pendingRun.runnerId, runId: pendingRun.id },
+            "failed to fetch runner, will retry next tick"
+          );
+          break;
+        }
         if (!runner) {
           logger.error(
             { runnerId: pendingRun.runnerId, runId: pendingRun.id },
             "runner not found for pending run"
           );
-          await api.completeTestRun(pendingRun.id, { status: "failed" });
+          try {
+            await api.completeTestRun(pendingRun.id, { status: "failed" });
+          } catch (err) {
+            logger.error(
+              { err, runId: pendingRun.id },
+              "failed to mark run as failed"
+            );
+          }
           continue;
         }
 
         const runnerInstanceId = `${this.processInstanceId}:${slotNumber}`;
         const runnerInstanceName = `${runner.title} [slot ${slotNumber}]`;
-        const claimed = await api.claimTestRun(
-          pendingRun.id,
-          runnerInstanceId,
-          runnerInstanceName
-        );
+
+        let claimed;
+        try {
+          claimed = await api.claimTestRun(
+            pendingRun.id,
+            runnerInstanceId,
+            runnerInstanceName
+          );
+        } catch (err) {
+          logger.error(
+            { err, runId: pendingRun.id },
+            "failed to claim run, will retry next tick"
+          );
+          break;
+        }
         if (!claimed) {
           logger.info(
             { runId: pendingRun.id },
@@ -106,8 +130,11 @@ export class RunnerManager {
           runnerInstanceId,
           runnerInstanceName,
           quickScan: pendingRun.quickScan ?? false,
+          scanMode: readScanMode(pendingRun),
         });
       }
+    } catch (err) {
+      logger.error({ err }, "tick failed");
     } finally {
       this.tickInFlight = false;
     }
@@ -122,6 +149,7 @@ export class RunnerManager {
     runnerInstanceId: string;
     runnerInstanceName: string;
     quickScan: boolean;
+    scanMode?: "full" | "partial" | "minimum";
   }): Promise<void> {
     const config = loadConfig();
     const api = getApiClient(
@@ -140,6 +168,7 @@ export class RunnerManager {
         runnerInstanceId: params.runnerInstanceId,
         runnerInstanceName: params.runnerInstanceName,
         quickScan: params.quickScan,
+        scanMode: params.scanMode,
       });
 
       logger.info({ runId: params.runId }, "run completed successfully");
@@ -161,3 +190,12 @@ export class RunnerManager {
     }
   }
 }
+
+function readScanMode(
+  run: unknown
+): "full" | "partial" | "minimum" | undefined {
+  const value = (run as { scanMode?: unknown }).scanMode;
+  return value === "full" || value === "partial" || value === "minimum"
+    ? value
+    : undefined;
+}