@brunobracaioli/aibrain 0.2.0-beta.1 → 0.3.0-beta.13

package/dist/telemetry-report.mjs

@@ -0,0 +1,133 @@
+import { readConfig, telemetryConfigBinding } from './state.mjs';
+import { reportContextTelemetry } from './mcp.mjs';
+// @ts-expect-error This validator is generated into dist from the shared TypeScript contract.
+import { parseContextTelemetryReport } from './context-telemetry.mjs';
+import { resolve } from 'node:path';
+import { fileURLToPath } from 'node:url';
+
+export const MAX_STDIN_BYTES = 8 * 1024;
+const ERROR_CODES = new Set([
+  'telemetry_input_too_large',
+  'telemetry_input_invalid',
+  'telemetry_consent_required',
+  'telemetry_state_invalid',
+  'telemetry_state_mismatch',
+  'telemetry_state_changed',
+  'telemetry_report_failed',
+]);
+
+/** @param {unknown} error */
+function fixedErrorCode(error) {
+  const value = error instanceof Error ? error.message : '';
+  return ERROR_CODES.has(value) ? value : 'telemetry_report_failed';
+}
+
+/** @param {unknown[]} argv */
+function parseStateDir(argv) {
+  if (
+    argv.length !== 4 ||
+    argv[0] !== '--state-dir' ||
+    typeof argv[1] !== 'string' ||
+    argv[1].length === 0 ||
+    !argv[1].startsWith('/') ||
+    Buffer.byteLength(argv[1], 'utf8') > 4096 ||
+    argv[2] !== '--config-digest' ||
+    typeof argv[3] !== 'string' ||
+    !/^[0-9a-f]{64}$/u.test(argv[3])
+  )
+    throw new Error('telemetry_state_invalid');
+  return { stateDir: argv[1], configDigest: argv[3] };
+}
+
+/** @param {unknown} raw @param {Date} [now] */
+export function parseTelemetryInput(raw, now = new Date()) {
+  if (typeof raw !== 'string' || Buffer.byteLength(raw, 'utf8') > MAX_STDIN_BYTES)
+    throw new Error('telemetry_input_too_large');
+  let value;
+  try {
+    value = JSON.parse(raw);
+  } catch {
+    throw new Error('telemetry_input_invalid');
+  }
+  try {
+    return parseContextTelemetryReport(value, now);
+  } catch {
+    throw new Error('telemetry_input_invalid');
+  }
+}
+
+/**
+ * @param {string} stateDir
+ * @param {string} configDigest
+ * @param {Record<string, unknown>} report
+ * @param {(endpoint: string, token: string, report: Record<string, unknown>, deadlineMs: number) => Promise<{accepted: true, duplicate: boolean}>} [transport]
+ */
+export async function executeTelemetryReport(
+  stateDir,
+  configDigest,
+  report,
+  transport = reportContextTelemetry,
+) {
+  const previousStateDir = process.env.AIBRAIN_STATE_DIR;
+  process.env.AIBRAIN_STATE_DIR = stateDir;
+  try {
+    const config = await readConfig();
+    if (
+      config === null ||
+      config.telemetry?.enabled !== true ||
+      typeof config.telemetry.disclosedAt !== 'string'
+    )
+      throw new Error('telemetry_consent_required');
+    let currentDigest;
+    try {
+      currentDigest = telemetryConfigBinding(config);
+    } catch {
+      throw new Error('telemetry_state_mismatch');
+    }
+    if (currentDigest !== configDigest) throw new Error('telemetry_state_changed');
+    return await transport(config.endpoint, config.token, report, 8_000);
+  } catch (error) {
+    if (error instanceof Error && ERROR_CODES.has(error.message)) throw error;
+    throw new Error('telemetry_report_failed');
+  } finally {
+    if (previousStateDir === undefined) delete process.env.AIBRAIN_STATE_DIR;
+    else process.env.AIBRAIN_STATE_DIR = previousStateDir;
+  }
+}
+
+async function readStdin() {
+  const chunks = [];
+  let total = 0;
+  for await (const chunk of process.stdin) {
+    const bytes = Buffer.isBuffer(chunk) ? chunk : Buffer.from(String(chunk));
+    total += bytes.byteLength;
+    if (total > MAX_STDIN_BYTES) throw new Error('telemetry_input_too_large');
+    chunks.push(bytes);
+  }
+  try {
+    return new TextDecoder('utf-8', { fatal: true }).decode(Buffer.concat(chunks));
+  } catch {
+    throw new Error('telemetry_input_invalid');
+  }
+}
+
+/** @param {string[]} [argv] */
+export async function main(argv = process.argv.slice(2)) {
+  try {
+    const { stateDir, configDigest } = parseStateDir(argv);
+    const report = parseTelemetryInput(await readStdin());
+    const result = await executeTelemetryReport(stateDir, configDigest, report);
+    process.stdout.write(
+      `${JSON.stringify({ schema_version: 1, accepted: true, duplicate: result.duplicate })}\n`,
+    );
+    return 0;
+  } catch (error) {
+    process.stdout.write(
+      `${JSON.stringify({ schema_version: 1, accepted: false, error_code: fixedErrorCode(error) })}\n`,
+    );
+    return 1;
+  }
+}
+
+if (process.argv[1] !== undefined && fileURLToPath(import.meta.url) === resolve(process.argv[1]))
+  process.exitCode = await main();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@brunobracaioli/aibrain",
-  "version": "0.2.0-beta.1",
+  "version": "0.3.0-beta.13",
   "private": false,
   "description": "AI Brain assisted context beta client for Claude Code and Codex",
   "license": "UNLICENSED",