@notis_ai/cli 0.2.18 → 0.2.19

package/src/runtime/agent-browser.js

@@ -1,109 +1,7 @@
 import { spawn, spawnSync } from 'node:child_process';
-import { mkdirSync, readFileSync, writeFileSync } from 'node:fs';
+import { mkdirSync, writeFileSync } from 'node:fs';
 import { dirname } from 'node:path';
 
-export const HIDE_HARNESS_STATUS_SCRIPT =
-  "(() => { const el = document.getElementById('harness-status'); if (el) el.style.opacity = '0'; return true; })()";
-
-/**
- * Prepare the mounted harness page for a listing capture: remove the debug
- * banner and the layout the harness adds around the app (top padding for the
- * banner, 100vh min-height) so content height can be measured and framed
- * without artificial whitespace.
- */
-const PREPARE_CAPTURE_SCRIPT =
-  "(() => { const s = document.getElementById('harness-status'); if (s) s.style.display = 'none';" +
-  " const r = document.getElementById('root'); if (r) { r.style.paddingTop = '0'; r.style.minHeight = '0'; }" +
-  " document.body.style.minHeight = '0'; document.documentElement.style.minHeight = '0'; return true; })()";
-
-const FONTS_STATUS_SCRIPT =
-  "(() => (document.fonts ? document.fonts.status : 'loaded'))()";
-
-const CONTENT_HEIGHT_SCRIPT =
-  '(() => { const r = document.getElementById(\'root\');' +
-  ' if (!r) return Math.ceil(document.documentElement.scrollHeight);' +
-  ' return Math.ceil(Math.max(r.scrollHeight, r.getBoundingClientRect().height)); })()';
-
-// CSS viewport bounds for content-aware framing. Content shorter than the
-// minimum gets a little breathing room; content taller than the maximum is
-// cropped at the frame rather than shrunk into unreadably small text.
-const MIN_CSS_VIEWPORT_HEIGHT = 720;
-const MAX_CSS_VIEWPORT_HEIGHT = 1500;
-
-function pngDimensions(path) {
-  try {
-    const buffer = readFileSync(path);
-    if (buffer.length < 24 || buffer.readUInt32BE(12) !== 0x49484452) {
-      return null;
-    }
-    return { width: buffer.readUInt32BE(16), height: buffer.readUInt32BE(20) };
-  } catch {
-    return null;
-  }
-}
-
-/**
- * Pick a CSS viewport that frames `contentHeight` at the output aspect ratio,
- * with a deviceScaleFactor that maps it back to exactly outputWidth x
- * outputHeight physical pixels. Width snaps to a multiple of 8 so the height
- * stays integral for the common 16:10 target.
- */
-function fitCssViewport(contentHeight, outputWidth, outputHeight) {
-  const aspect = outputWidth / outputHeight;
-  const cssH = Math.min(
-    MAX_CSS_VIEWPORT_HEIGHT,
-    Math.max(MIN_CSS_VIEWPORT_HEIGHT, Math.round(contentHeight)),
-  );
-  const cssW = Math.round((cssH * aspect) / 8) * 8;
-  return {
-    width: cssW,
-    height: Math.round(cssW / aspect),
-    scale: outputWidth / cssW,
-  };
-}
-
-async function evalNumber(sessionName, script, fallback = null) {
-  const result = await runAgentBrowser(
-    ['--session', sessionName, '--json', 'eval', script],
-    { timeoutMs: 5000 },
-  );
-  if (result.exitCode !== 0) {
-    return fallback;
-  }
-  try {
-    const payload = parseAgentBrowserJson(result.stdout);
-    const raw = payload?.data?.result ?? payload?.result ?? payload?.data ?? null;
-    const value = typeof raw === 'string' ? JSON.parse(raw) : raw;
-    return Number.isFinite(value) ? value : fallback;
-  } catch {
-    return fallback;
-  }
-}
-
-async function waitForFonts(sessionName, timeoutMs = 3000) {
-  const deadline = Date.now() + timeoutMs;
-  while (Date.now() < deadline) {
-    const result = await runAgentBrowser(
-      ['--session', sessionName, '--json', 'eval', FONTS_STATUS_SCRIPT],
-      { timeoutMs: 5000 },
-    );
-    if (result.exitCode !== 0) {
-      return;
-    }
-    try {
-      const payload = parseAgentBrowserJson(result.stdout);
-      const raw = payload?.data?.result ?? payload?.result ?? payload?.data ?? null;
-      const status = typeof raw === 'string' && raw.startsWith('"') ? JSON.parse(raw) : raw;
-      if (status !== 'loading') {
-        return;
-      }
-    } catch {
-      return;
-    }
-    await delay(150);
-  }
-}
-
 async function setViewport(sessionName, width, height, scale = null) {
   const args = ['--session', sessionName, 'set', 'viewport', String(width), String(height)];
   if (scale != null) {
@@ -135,9 +33,11 @@ function runAgentBrowser(args, { timeoutMs = 30_000 } = {}) {
     let stdout = '';
     let stderr = '';
     let timedOut = false;
+    let hardStop;
     const timer = setTimeout(() => {
       timedOut = true;
       child.kill('SIGTERM');
+      hardStop = setTimeout(() => child.kill('SIGKILL'), 2000);
     }, timeoutMs);
 
     child.stdout.on('data', (chunk) => {
@@ -147,7 +47,7 @@ function runAgentBrowser(args, { timeoutMs = 30_000 } = {}) {
       stderr += chunk.toString();
     });
     child.on('error', (error) => {
-      clearTimeout(timer);
+      clearTimeout(timer); clearTimeout(hardStop);
       resolvePromise({
         exitCode: 1,
         stdout,
@@ -156,7 +56,7 @@ function runAgentBrowser(args, { timeoutMs = 30_000 } = {}) {
       });
     });
     child.on('close', (code) => {
-      clearTimeout(timer);
+      clearTimeout(timer); clearTimeout(hardStop);
       resolvePromise({
         exitCode: timedOut ? 124 : (code ?? 1),
         stdout,
@@ -227,10 +127,19 @@ export async function runHarnessRoute({
   sessionName,
   timeoutMs = 10_000,
   snapshotPath = null,
+  designViewports = DEFAULT_DESIGN_VIEWPORTS,
+  waitForRuntime = false,
+  signal,
 }) {
+  const cancelled = () => ({ mounted: false, renderStarted: false, runtimeCalls: [],
+    errors: [{ phase: 'browser', message: 'Verification cancelled.' }], snapshotPath: null, tool_error: null });
+  if (signal?.aborted) return cancelled();
   const opened = await runAgentBrowser(['--session', sessionName, 'open', url], {
     timeoutMs: Math.min(Math.max(timeoutMs, 5000), 30_000),
   });
+  // Let an in-flight open settle before the owner closes its browser. Closing
+  // concurrently with open can leave a newly launched browser behind.
+  if (signal?.aborted) return cancelled();
   if (opened.exitCode !== 0) {
     return {
       mounted: false,
@@ -243,20 +152,26 @@ export async function runHarnessRoute({
   }
 
   const deadline = Date.now() + timeoutMs;
-  while (Date.now() < deadline) {
+  let readySince = null;
+  while (Date.now() < deadline && !signal?.aborted) {
     const lastRead = await readHarness(sessionName, 5000);
     if (!lastRead.ok) {
+      readySince = null;
       await delay(250);
       continue;
     }
     const harness = lastRead.harness;
-    if (harness?.mounted || (Array.isArray(harness?.errors) && harness.errors.length > 0)) {
-      await delay(250);
-      break;
-    }
+    if (Array.isArray(harness?.errors) && harness.errors.length > 0) break;
+    const pending = (harness?.runtimeCalls || []).some(call => call?.ok == null);
+    if (harness?.mounted && (!waitForRuntime || !pending)) {
+      readySince ??= Date.now();
+      if (Date.now() - readySince >= 250) break;
+    } else readySince = null;
     await delay(250);
   }
 
+  if (signal?.aborted) return cancelled();
+
   const finalRead = await readHarness(sessionName, 5000);
   if (!finalRead.ok) {
     return {
@@ -282,183 +197,179 @@ export async function runHarnessRoute({
     }
   }
 
-  const timedOut = !harness.mounted && Date.now() >= deadline;
+  const runtimeCalls = Array.isArray(harness.runtimeCalls) ? harness.runtimeCalls : [];
+  const timedOut = (!harness.mounted || (waitForRuntime && runtimeCalls.some(call => call?.ok == null))) && Date.now() >= deadline;
+
+  // Design assertions only mean something on a mounted route. Loading
+  // placeholders are excused while a runtime call is still in flight.
+  let design = [];
+  let designToolError = null;
+  if (harness.mounted && Array.isArray(designViewports) && designViewports.length > 0) {
+    await delay(500);
+    const pending = runtimeCalls.some((call) => call && call.ok == null);
+    const collected = await collectDesignFindings(sessionName, {
+      viewports: designViewports,
+      skipLoadingCheck: pending,
+    });
+    design = collected.findings;
+    designToolError = collected.tool_error;
+  }
+
   return {
     mounted: Boolean(harness.mounted),
     renderStarted: Boolean(harness.renderStarted),
     errors: Array.isArray(harness.errors) ? harness.errors : [],
-    runtimeCalls: Array.isArray(harness.runtimeCalls) ? harness.runtimeCalls : [],
+    runtimeCalls,
     snapshotPath: savedSnapshotPath,
     timed_out: timedOut,
     tool_error: null,
+    design,
+    design_tool_error: designToolError,
     raw: harness,
   };
 }
 
-/**
- * Open a harness route, wait for it to mount, then capture a PNG screenshot of
- * the rendered app. Output is always exactly `width` x `height` physical
- * pixels (default 2000x1250, the 16:10 the listing validator expects), but the
- * CSS viewport is fitted to the rendered content height: short pages render
- * larger (no dead whitespace below the app), tall pages get more room before
- * cropping. The deviceScaleFactor maps the fitted CSS viewport back onto the
- * fixed output frame. Used by `notis apps screenshot`.
- */
-export async function captureHarnessScreenshot({
-  url,
-  sessionName,
-  screenshotPath,
-  focusSelector = null,
-  width = 2000,
-  height = 1250,
-  timeoutMs = 15_000,
-}) {
-  const opened = await runAgentBrowser(['--session', sessionName, 'open', url], {
-    timeoutMs: Math.min(Math.max(timeoutMs, 5000), 30_000),
-  });
-  if (opened.exitCode !== 0) {
-    return { ok: false, mounted: false, screenshotPath: null, tool_error: commandError('open', opened) };
-  }
-
-  // Initial fixed frame so mount/layout happen at a deterministic size; the
-  // scale keeps physical output at width x height from the very first paint.
-  const initial = fitCssViewport(900, width, height);
-  let scaledViewport = await setViewport(
-    sessionName,
-    initial.width,
-    initial.height,
-    initial.scale,
-  );
-  if (!scaledViewport) {
-    // Older agent-browser without deviceScaleFactor support: fall back to a
-    // plain fixed viewport at output size.
-    await setViewport(sessionName, width, height);
-  }
+export async function closeAgentBrowserSessionResult(sessionName) {
+  const started = Date.now();
+  // agent-browser waits up to 30s for CDP Browser.close and then up to 5s
+  // for its owned Chrome process. Do not kill the client at the start of the
+  // provider's shutdown grace period and lose its actual close acknowledgement.
+  const result = await runAgentBrowser(['--session', sessionName, '--json', 'close'], { timeoutMs: 40_000 });
+  let acknowledged = false;
+  try {
+    const reply = parseAgentBrowserJson(result.stdout);
+    acknowledged = reply?.success === true && reply?.data?.closed === true;
+  } catch { /* Missing acknowledgement is a failed cleanup, never a pass. */ }
+  return { ok: result.exitCode === 0 && acknowledged, exit_code: result.exitCode,
+    timed_out: Boolean(result.timedOut), duration_ms: Date.now() - started };
+}
 
-  const deadline = Date.now() + timeoutMs;
-  let lastErrors = [];
-  let lastReadError = null;
-  let mounted = false;
-  while (Date.now() < deadline) {
-    const read = await readHarness(sessionName, 5000);
-    if (read.ok) {
-      const harness = read.harness || {};
-      mounted = Boolean(harness.mounted);
-      lastErrors = Array.isArray(harness.errors) ? harness.errors : [];
-      if (mounted || lastErrors.length > 0) {
-        break;
-      }
-    } else {
-      lastReadError = read.toolError;
+// ---------------------------------------------------------------------------
+// Runtime design assertions.
+//
+// The static design lint (app-boundary-validator.js) catches banned class
+// names. This script runs in the mounted harness page and catches what a
+// regex cannot: a bordered box nested in a bordered box across files, tinted
+// panels stacked three deep, text that computes below 12px, horizontal
+// overflow at a phone width, and loading placeholders that outlive the data.
+// Palette colors are deliberately not checked here: at runtime a token and a
+// hardcoded hue resolve to the same RGB, so that stays with the static lint.
+// ---------------------------------------------------------------------------
+export const DESIGN_ASSERTIONS_MARKER = '__notisDesignAssertions';
+
+export const DESIGN_ASSERTIONS_SCRIPT = `(() => {
+  /* ${DESIGN_ASSERTIONS_MARKER} */
+  const root = document.getElementById('root') || document.body;
+  const out = [];
+  const seen = new Set();
+  const push = (kind, el, extra) => {
+    const cls = String(el.className && el.className.baseVal !== undefined ? el.className.baseVal : el.className || '').trim().slice(0, 80);
+    const text = (el.textContent || '').trim().replace(/\\s+/g, ' ').slice(0, 60);
+    const snippet = el.tagName.toLowerCase() + (cls ? '.' + cls.split(/\\s+/).slice(0, 6).join('.') : '') + (text ? ' "' + text + '"' : '');
+    const key = kind + '|' + snippet;
+    if (seen.has(key)) return;
+    seen.add(key);
+    out.push(Object.assign({ kind, snippet }, extra || {}));
+  };
+  const controls = new Set(['INPUT', 'SELECT', 'TEXTAREA', 'BUTTON', 'A', 'SUMMARY']);
+  const isControl = (el) => controls.has(el.tagName) || /^(button|checkbox|switch|radio|combobox|textbox|tab|menuitem)$/.test(el.getAttribute('role') || '');
+  const visible = (el) => { const r = el.getBoundingClientRect(); return r.width > 0 && r.height > 0; };
+  const isBox = (el) => {
+    const cs = getComputedStyle(el);
+    if (!['borderTopWidth', 'borderRightWidth', 'borderBottomWidth', 'borderLeftWidth'].every((k) => parseFloat(cs[k]) > 0)) return false;
+    if (/rgba\\(\\d+, \\d+, \\d+, 0\\)|transparent/.test(cs.borderTopColor)) return false;
+    const r = el.getBoundingClientRect();
+    return r.width > 40 && r.height > 20;
+  };
+  const bg = (el) => getComputedStyle(el).backgroundColor;
+  const transparent = (c) => !c || c === 'transparent' || /rgba\\(\\d+, \\d+, \\d+, 0\\)/.test(c);
+  const isPanel = (el) => {
+    if (transparent(bg(el))) return false;
+    if (parseFloat(getComputedStyle(el).borderRadius) <= 0) return false;
+    const r = el.getBoundingClientRect();
+    return r.width > 160 && r.height > 48;
+  };
+  const all = Array.from(root.querySelectorAll('*'));
+  for (const el of all) {
+    if (!visible(el) || isControl(el)) continue;
+    if (isBox(el)) {
+      let p = el.parentElement; let depth = 0;
+      while (p && p !== root && depth < 6) { if (!isControl(p) && isBox(p)) { push('nested_border_box', el); break; } p = p.parentElement; depth += 1; }
     }
-    await delay(250);
-  }
-
-  if (lastErrors.length > 0) {
-    return { ok: false, mounted, screenshotPath: null, errors: lastErrors, tool_error: null };
-  }
-  if (!mounted) {
-    return {
-      ok: false,
-      mounted: false,
-      screenshotPath: null,
-      errors: [],
-      timed_out: true,
-      tool_error: lastReadError || {
-        phase: 'harness',
-        message: 'Timed out waiting for window.__harness.mounted.',
-      },
-    };
-  }
-
-  // Strip harness chrome (debug banner, banner padding, 100vh min-height) so
-  // the frame contains only the app, then wait for webfonts before measuring.
-  await runAgentBrowser(
-    ['--session', sessionName, 'eval', PREPARE_CAPTURE_SCRIPT],
-    { timeoutMs: 5000 },
-  );
-  await waitForFonts(sessionName);
-
-  // Fit the CSS viewport to the rendered content. Content height depends on
-  // viewport width, so refine once after the first resize reflows the page.
-  let framing = null;
-  if (scaledViewport) {
-    let contentHeight = await evalNumber(sessionName, CONTENT_HEIGHT_SCRIPT);
-    for (let pass = 0; contentHeight != null && pass < 2; pass += 1) {
-      const fitted = fitCssViewport(contentHeight, width, height);
-      if (framing && fitted.width === framing.width) {
-        break;
-      }
-      scaledViewport = await setViewport(sessionName, fitted.width, fitted.height, fitted.scale);
-      if (!scaledViewport) {
-        framing = null;
-        await setViewport(sessionName, width, height);
-        break;
-      }
-      framing = { ...fitted, content_height: contentHeight };
-      await delay(150);
-      const remeasured = await evalNumber(sessionName, CONTENT_HEIGHT_SCRIPT);
-      if (remeasured == null || Math.abs(remeasured - contentHeight) <= 32) {
-        framing.content_height = remeasured ?? contentHeight;
-        break;
-      }
-      contentHeight = remeasured;
+    if (isPanel(el)) {
+      let p = el.parentElement; let panels = 0;
+      while (p && p !== root) { if (isPanel(p)) panels += 1; p = p.parentElement; }
+      if (panels >= 2) push('nested_tinted_panel', el, { depth: panels + 1 });
     }
   }
+  const walker = document.createTreeWalker(root, NodeFilter.SHOW_TEXT);
+  let node;
+  while ((node = walker.nextNode())) {
+    const text = (node.textContent || '').trim();
+    if (!text) continue;
+    const el = node.parentElement;
+    if (!el || !visible(el)) continue;
+    const size = parseFloat(getComputedStyle(el).fontSize);
+    if (size > 0 && size < 12) push('text_below_12px', el, { font_size: size });
+    if (/^Loading(\\.{3}|\\u2026)?(\\s|$)/.test(text)) push('loading_placeholder_after_mount', el);
+  }
+  if (root.querySelector('[data-notis-content-skeleton]')) push('loading_placeholder_after_mount', root.querySelector('[data-notis-content-skeleton]'));
+  const doc = document.documentElement;
+  if (doc.scrollWidth > doc.clientWidth + 1) out.push({ kind: 'horizontal_overflow', snippet: 'document', scroll_width: doc.scrollWidth, client_width: doc.clientWidth });
+  return JSON.stringify(out.slice(0, 40));
+})()`;
 
-  // Let the route settle (async data, resize transitions) before the capture.
-  await delay(500);
+export const DEFAULT_DESIGN_VIEWPORTS = [
+  { name: 'desktop', width: 1280, height: 900 },
+  { name: 'mobile', width: 390, height: 844 },
+];
 
-  mkdirSync(dirname(screenshotPath), { recursive: true });
-  const screenshotArgs = ['--session', sessionName, 'screenshot'];
-  if (focusSelector) {
-    screenshotArgs.push(focusSelector);
+/**
+ * Evaluate the design assertions in the current page. Returns an array of
+ * findings, or `{ tool_error }` when agent-browser could not run the script.
+ */
+export async function evalDesignAssertions(sessionName, { viewport = null, timeoutMs = 8000 } = {}) {
+  const result = await runAgentBrowser(
+    ['--session', sessionName, '--json', 'eval', DESIGN_ASSERTIONS_SCRIPT],
+    { timeoutMs },
+  );
+  if (result.exitCode !== 0) {
+    return { tool_error: commandError('design_eval', result) };
   }
-  screenshotArgs.push(screenshotPath, '--screenshot-format', 'png');
-  const shot = await runAgentBrowser(screenshotArgs, { timeoutMs: 15_000 });
-  if (shot.exitCode !== 0) {
-    return { ok: false, mounted: true, screenshotPath: null, errors: lastErrors, timed_out: false, tool_error: commandError('screenshot', shot) };
+  try {
+    const payload = parseAgentBrowserJson(result.stdout);
+    const raw = payload?.data?.result ?? payload?.result ?? payload?.data ?? null;
+    const value = typeof raw === 'string' ? JSON.parse(raw) : raw;
+    if (!Array.isArray(value)) throw new Error('Design evaluation did not return a findings array.');
+    return value.map((finding) => ({ ...finding, viewport: viewport || finding.viewport || null }));
+  } catch (error) {
+    return { tool_error: { phase: 'design_eval_parse', message: error.message } };
   }
+}
 
-  // The listing validator requires exact output dimensions. If the fitted
-  // capture came out wrong (e.g. the browser ignored the scale factor),
-  // recapture once at a plain fixed viewport.
-  const dimensions = pngDimensions(screenshotPath);
-  if (!focusSelector && framing && (!dimensions || dimensions.width !== width || dimensions.height !== height)) {
-    await setViewport(sessionName, width, height);
-    await delay(300);
-    const retry = await runAgentBrowser(
-      ['--session', sessionName, 'screenshot', screenshotPath, '--screenshot-format', 'png'],
-      { timeoutMs: 15_000 },
-    );
-    if (retry.exitCode !== 0) {
-      return { ok: false, mounted: true, screenshotPath: null, errors: lastErrors, timed_out: false, tool_error: commandError('screenshot', retry) };
+/**
+ * Run every requested viewport. Missing browser checks are incomplete proof,
+ * not a passing design result, even when static source lint passed.
+ */
+export async function collectDesignFindings(sessionName, { viewports = DEFAULT_DESIGN_VIEWPORTS, settleMs = 350, skipLoadingCheck = false } = {}) {
+  const findings = [];
+  let toolError = null;
+  for (const viewport of viewports) {
+    const applied = await setViewport(sessionName, viewport.width, viewport.height);
+    if (!applied) {
+      toolError ||= { phase: 'design_viewport', message: `Could not set ${viewport.name} viewport.` };
+      continue;
+    }
+    await delay(settleMs);
+    const result = await evalDesignAssertions(sessionName, { viewport: viewport.name });
+    if (Array.isArray(result)) {
+      for (const finding of result) {
+        if (skipLoadingCheck && finding.kind === 'loading_placeholder_after_mount') continue;
+        findings.push(finding);
+      }
+    } else if (!toolError) {
+      toolError = result.tool_error;
     }
-    framing = null;
-  }
-
-  if (focusSelector) {
-    framing = {
-      ...(framing || {}),
-      focus_selector: focusSelector,
-      source_dimensions: dimensions,
-    };
   }
-
-  return {
-    ok: true,
-    mounted: true,
-    screenshotPath,
-    errors: lastErrors,
-    timed_out: false,
-    tool_error: null,
-    framing,
-  };
-}
-
-export async function closeAgentBrowserSession(sessionName) {
-  const result = await runAgentBrowser(['--session', sessionName, 'close'], {
-    timeoutMs: 5000,
-  });
-  return result.exitCode === 0;
+  return { findings, tool_error: toolError };
 }