@aria_asi/cli 0.2.29 → 0.2.31

package/dist/assets/hooks/aria-harness-via-sdk.mjs

@@ -26,7 +26,10 @@ const LOG_FILE = `${HOME}/.claude/aria-harness-history.log`;
 const LAST_URL_CACHE = `${HOME}/.claude/.aria-harness-last-url`;
 const PACKET_CACHE = `${HOME}/.claude/.aria-harness-last-packet.json`;
 const SHARED_PACKET_CACHE = `${HOME}/.aria/.aria-harness-last-packet.json`;
+const MIZAN_RECEIPT_DIR = `${HOME}/.claude/.aria-mizan-receipts`;
 const HEADER_PREFIX = '🔐 Aria Harness';
+const DEFAULT_RUNTIME_URL = process.env.ARIA_RUNTIME_URL || 'http://127.0.0.1:4319';
+const RUNTIME_PACKET_SUFFIX = '/packet';
 
 const args = process.argv.slice(2);
 const MODE = args.includes('--mode') ? args[args.indexOf('--mode') + 1] : 'session';
@@ -133,6 +136,8 @@ function buildUrlList() {
   const list = [];
   const cached = loadCachedUrl();
   if (cached) list.push(cached);
+  if (process.env.ARIA_RUNTIME_URL) list.push(process.env.ARIA_RUNTIME_URL.replace(/\/+$/, ''));
+  list.push(DEFAULT_RUNTIME_URL.replace(/\/+$/, ''));
   if (process.env.ARIA_HIVE_RUNTIME_URL) list.push(process.env.ARIA_HIVE_RUNTIME_URL.replace(/\/+$/, ''));
   if (process.env.ARIA_HARNESS_BASE_URL) list.push(process.env.ARIA_HARNESS_BASE_URL.replace(/\/+$/, ''));
   if (process.env.ARIA_HARNESS_URL) list.push(process.env.ARIA_HARNESS_URL.replace(/\/+$/, ''));
@@ -155,6 +160,21 @@ function buildUrlList() {
   return list.filter((u) => (seen.has(u) ? false : (seen.add(u), true)));
 }
 
+function isMountedRuntimeUrl(baseUrl) {
+  return /:\/\/(?:127\.0\.0\.1|localhost):4319$/i.test(String(baseUrl || '').replace(/\/+$/, ''));
+}
+
+function normalizeHarnessPacketPayload(payload) {
+  let current = payload;
+  for (let depth = 0; depth < 3; depth++) {
+    if (!current || typeof current !== 'object' || Array.isArray(current)) break;
+    if (!('packet' in current) || !current.packet || typeof current.packet !== 'object') break;
+    if (!('timestamp' in current) && !('version' in current)) break;
+    current = current.packet;
+  }
+  return current;
+}
+
 // SDK loader — dynamic-import the bundled HTTPHarnessClient from the shared
 // ~/.aria/sdk bundle first, then client-local bundles. Module-cached after
 // first load so we don't repeatedly read disk.
@@ -204,6 +224,7 @@ async function loadSdkClass() {
 // before forwarding to avoid blowing the POST body size limit; the most recent
 // messages are the most relevant for history-count purposes.
 let HOOK_EVENT_MESSAGES = undefined;
+let HOOK_EVENT = null;
 try {
   // Claude Code hooks receive the event JSON on stdin. We read it synchronously
   // only if data is already available (i.e. piped); we don't block waiting for
@@ -211,6 +232,7 @@ try {
   // a terminal (e.g. manual invocation for testing).
   const stdinBuf = readFileSync('/dev/stdin', { flag: 'r' });
   const hookEvent = JSON.parse(stdinBuf.toString('utf8'));
+  HOOK_EVENT = hookEvent;
   if (Array.isArray(hookEvent?.messages) && hookEvent.messages.length > 0) {
     // Cap to last 50 messages. The server-side handler slices further if needed.
     HOOK_EVENT_MESSAGES = hookEvent.messages.slice(-50);
@@ -219,6 +241,81 @@ try {
   // stdin not available / not JSON / no messages field — HOOK_EVENT_MESSAGES stays undefined.
 }
 
+function sanitizeSessionId(sessionId) {
+  return String(sessionId || 'claude-code').replace(/[^a-zA-Z0-9_-]/g, '_');
+}
+
+function lastUserMessageFromEvent() {
+  const messages = Array.isArray(HOOK_EVENT?.messages) ? HOOK_EVENT.messages : [];
+  for (let index = messages.length - 1; index >= 0; index -= 1) {
+    const message = messages[index];
+    const role = message?.role || message?.message?.role || message?.type;
+    if (role !== 'user') continue;
+    const content = message?.content ?? message?.message?.content;
+    if (typeof content === 'string' && content.trim()) return content.trim();
+    if (Array.isArray(content)) {
+      const text = content
+        .filter((entry) => entry?.type === 'text' && typeof entry?.text === 'string')
+        .map((entry) => entry.text.trim())
+        .filter(Boolean)
+        .join('\n');
+      if (text) return text;
+    }
+  }
+  return isTurn ? 'Claude turn refresh via harness hook' : 'Claude session start via harness hook';
+}
+
+function receiptPathForSession(sessionId) {
+  return `${MIZAN_RECEIPT_DIR}/${sanitizeSessionId(sessionId)}.json`;
+}
+
+function persistMizanReceipt(sessionId, payload) {
+  try {
+    mkdirSync(MIZAN_RECEIPT_DIR, { recursive: true });
+    writeFileSync(receiptPathForSession(sessionId), JSON.stringify(payload, null, 2) + '\n');
+  } catch {}
+}
+
+async function runMizanPre(apiKey, packet, sourceLabel) {
+  if (!isTurn) return null;
+  const sessionId = HOOK_EVENT?.session_id || HOOK_EVENT?.sessionId || 'claude-code';
+  const response = await fetch(`${DEFAULT_RUNTIME_URL.replace(/\/+$/, '')}/mizan/pre`, {
+    method: 'POST',
+    headers: {
+      Authorization: `Bearer ${apiKey}`,
+      'Content-Type': 'application/json',
+    },
+    body: JSON.stringify({
+      sessionId,
+      packet,
+      context: {
+        sessionId,
+        message: lastUserMessageFromEvent().slice(0, 4000),
+        intendedAction: 'Establish canonical pre-turn cognition receipt for this Claude turn before any tool or output work.',
+        rationale: `Claude turn-start harness sync via ${sourceLabel} must mint a canonical Mizan pre receipt for downstream tool and output enforcement.`,
+        platform: 'claude-code',
+        stage: 'hook-turn-pre',
+      },
+    }),
+  });
+  const payload = await response.json().catch(() => ({}));
+  if (!response.ok) {
+    throw new Error(payload?.error || `mizan/pre failed (${response.status})`);
+  }
+  if (payload?.receipt) {
+    persistMizanReceipt(sessionId, {
+      updatedAt: new Date().toISOString(),
+      source: sourceLabel,
+      sessionId,
+      receipt: payload.receipt,
+      result: payload.result || null,
+      contract: payload.contract || null,
+      summary: payload.summary || null,
+    });
+  }
+  return payload;
+}
+
 async function tryViaSdk(baseUrl, apiKey) {
   // Canonical path: HTTPHarnessClient.getHarnessPacket(). The SDK POSTs to
   // /api/harness/codex with the right shape and returns { packet, timestamp,
@@ -228,16 +325,19 @@ async function tryViaSdk(baseUrl, apiKey) {
   // SDK's `body.packet ?? body` passes the body through unchanged).
   const Cls = await loadSdkClass();
   if (Cls) {
+    const packetUrl = isMountedRuntimeUrl(baseUrl)
+      ? `${baseUrl}${RUNTIME_PACKET_SUFFIX}`
+      : `${baseUrl}/api/harness/codex`;
     const client = new Cls({
       baseUrl,
       apiKey,
-      harnessPacketUrl: `${baseUrl}/api/harness/codex`,
+      harnessPacketUrl: packetUrl,
     });
     // Pass a bodyOverride that does NOT include isHamza:false. The server
     // identifies owner tier from the Bearer token (isMasterTokenRequest).
     // Hardcoding isHamza:false in the body would override that server-side
     // signal and produce hamza:false in the packet even for master-token callers.
-    const bodyOverride = {
+      const bodyOverride = {
       message: isTurn ? 'Claude turn refresh — harness via SDK' : 'Claude session start — harness via SDK',
       stage: isTurn ? 'checkpoint' : 'preflight',
       actor: 'claude-code',
@@ -246,15 +346,19 @@ async function tryViaSdk(baseUrl, apiKey) {
       // Bonus #74: forward conversation history so codex handler can report
       // a non-zero conversation_history_count in the packet.
       ...(HOOK_EVENT_MESSAGES ? { messages: HOOK_EVENT_MESSAGES } : {}),
-    };
-    const wrapped = await client.getHarnessPacket(bodyOverride);
-    const json = wrapped.packet;
-    if (json && json.ok === false) throw new Error(`ok=false: ${json.error || 'unknown'}`);
-    return { json, raw: JSON.stringify(json) };
-  }
+      };
+      const wrapped = await client.getHarnessPacket(bodyOverride);
+      const json = normalizeHarnessPacketPayload(wrapped.packet);
+      if (json && json.ok === false) throw new Error(`ok=false: ${json.error || 'unknown'}`);
+      await runMizanPre(apiKey, wrapped.packet, `${baseUrl}${RUNTIME_PACKET_SUFFIX}`);
+      return { json, raw: JSON.stringify(json) };
+    }
 
+  const packetUrl = isMountedRuntimeUrl(baseUrl)
+    ? `${baseUrl}${RUNTIME_PACKET_SUFFIX}`
+    : `${baseUrl}/api/harness/codex`;
   // SDK absent (dev environment) — direct fetch with identical wire shape.
-  const resp = await fetch(`${baseUrl}/api/harness/codex`, {
+  const resp = await fetch(packetUrl, {
     method: 'POST',
     headers: {
       Authorization: `Bearer ${apiKey}`,
@@ -272,8 +376,10 @@ async function tryViaSdk(baseUrl, apiKey) {
     }),
   });
   if (!resp.ok) throw new Error(`HTTP ${resp.status}`);
-  const json = await resp.json();
+  const responseBody = await resp.json();
+  const json = normalizeHarnessPacketPayload(responseBody.packet ?? responseBody);
   if (json && json.ok === false) throw new Error(`ok=false: ${json.error || 'unknown'}`);
+  await runMizanPre(apiKey, responseBody.packet ?? responseBody, packetUrl);
   return { json, raw: JSON.stringify(json) };
 }
 
@@ -284,6 +390,9 @@ function renderPacket(json, source, ageNote = '') {
   const con = json.contractGate || {};
   const mc = json.missingCritical || [];
   const loaded = json.loadedByClass || {};
+  const offlineBundle = json.runtimeOfflineBundle && typeof json.runtimeOfflineBundle === 'object'
+    ? json.runtimeOfflineBundle
+    : null;
 
   let header = `${HEADER_PREFIX} (${MODE}${ageNote}, sdk=harness-http-client) hash=${phash.slice(0, 16)} via=${source}\n`;
   header += `  preStateGate: passed=${pre.passed} score=${(pre.score || 0).toFixed(2)}`;
@@ -292,6 +401,9 @@ function renderPacket(json, source, ageNote = '') {
   if (mc.length) header += `\n  missingCritical: ${JSON.stringify(mc.slice(0, 5))}`;
   const loadedSummary = Object.entries(loaded).sort().map(([k, v]) => `${k}=${v}`).join(' ');
   if (loadedSummary) header += `\n  loadedByClass: ${loadedSummary}`;
+  if (offlineBundle) {
+    header += `\n  offlineBundle: phase=${offlineBundle.phase || 'unknown'} age=${offlineBundle.ageSeconds ?? 'unknown'}s source=${offlineBundle.source || 'runtime-cache'}`;
+  }
 
   let ctx = header;
   if (harness) {
@@ -344,8 +456,9 @@ async function main() {
       const stat = fs.statSync(PACKET_CACHE);
       const ageSec = (Date.now() - stat.mtimeMs) / 1000;
       if (ageSec < PACKET_CACHE_TTL_SEC) {
-        const cached = JSON.parse(fs.readFileSync(PACKET_CACHE, 'utf8'));
-        return renderPacket(cached, '(cache)', ` cached ${Math.round(ageSec)}s`);
+      const cached = JSON.parse(fs.readFileSync(PACKET_CACHE, 'utf8'));
+      await runMizanPre(apiKey, cached, '(cache)');
+      return renderPacket(cached, '(cache)', ` cached ${Math.round(ageSec)}s`);
       }
     } catch {}
   }
@@ -396,6 +509,7 @@ async function main() {
       const ageSec = (Date.now() - stat.mtimeMs) / 1000;
       if (ageSec >= 0) {
         const cached = JSON.parse(fs.readFileSync(PACKET_CACHE, 'utf8'));
+        await runMizanPre(apiKey, cached, '(stale-cache)');
         try {
           appendFileSync(LOG_FILE, `${new Date().toISOString()} stale mode=${MODE} cache_age=${Math.round(ageSec)}s last_err="${lastErr.replace(/"/g, "'")}" sdk=harness-http-client\n`);
         } catch {}