@astrasyncai/verification-gateway 5.9.0 → 5.11.0

package/dist/cursor/extension.mjs

@@ -1,3 +1,7 @@
+// src/cursor/extension.ts
+import { homedir as homedir3 } from "os";
+import { join as join3 } from "path";
+
 // src/adapter-interface/interface.ts
 var ADAPTER_INTERFACE_VERSION = 1;
 
@@ -3965,7 +3969,7 @@ function getTrustLevel(score) {
 }
 
 // src/version.ts
-var SDK_VERSION = "5.9.0";
+var SDK_VERSION = "5.11.0";
 
 // src/http.ts
 var SDK_USER_AGENT = `astrasync-sdk/${SDK_VERSION}`;
@@ -4284,6 +4288,24 @@ async function callVerifyAccessAPI(config, request) {
         }
       };
     }
+    if (response.status === 400 && data?.error === "INVALID_AGENT_ID") {
+      const message = typeof data?.message === "string" ? data.message : "Agent identifier must be a canonical ASTRA-* id.";
+      return {
+        success: true,
+        attemptId: typeof data?.attemptId === "string" ? data.attemptId : void 0,
+        access: {
+          allowed: false,
+          reason: "invalid_agent_id",
+          failures: [
+            {
+              dimension: "identity.invalid_agent_id",
+              message,
+              guidance: typeof data?.guidance === "string" ? data.guidance : "Present your canonical ASTRA-* id (discover it via GET /api/agents/request-registration/{requestId} after owner approval)."
+            }
+          ]
+        }
+      };
+    }
     if (!response.ok) {
       if (response.status === 401) {
         return {
@@ -5114,11 +5136,445 @@ var AstraSyncGateway = class {
   }
 };
 
+// src/claude-code/guard-config.ts
+import { homedir as homedir2 } from "os";
+
+// src/fleet/enrolment-store.ts
+import { chmodSync, existsSync as existsSync3, mkdirSync as mkdirSync3, readFileSync as readFileSync3, writeFileSync as writeFileSync2 } from "fs";
+import { homedir } from "os";
+import { dirname as dirname2, join as join2 } from "path";
+function machineScopeDir(platform = process.platform, env = process.env) {
+  if (platform === "darwin") return "/Library/Application Support/AstraSync";
+  if (platform === "win32") return join2(env.ProgramData || "C:\\ProgramData", "AstraSync");
+  return "/etc/astrasync";
+}
+function fleetPaths(options = {}) {
+  const home = options.home ?? homedir();
+  const machineDir = options.machineDir ?? options.env?.ASTRASYNC_FLEET_MACHINE_DIR ?? machineScopeDir(options.platform, options.env);
+  const userDir = join2(home, ".astrasync", "fleet");
+  return {
+    machineDevice: join2(machineDir, "fleet.json"),
+    machinePolicy: join2(machineDir, "policy.jws"),
+    userDir,
+    userDevice: join2(userDir, "device.json"),
+    userPolicy: join2(userDir, "policy.jws"),
+    userKeyChain: join2(userDir, "keychain.json"),
+    state: join2(userDir, "state.json")
+  };
+}
+function isEnrolment(value) {
+  const v = value;
+  return !!v && typeof v.deviceId === "string" && typeof v.credential === "string" && typeof v.apiBaseUrl === "string" && !!v.org && typeof v.org.ownerId === "string" && !!v.jwks && Array.isArray(v.jwks.keys);
+}
+function loadEnrolment(paths) {
+  const candidates = [
+    [paths.machineDevice, "machine"],
+    [paths.userDevice, "user"]
+  ];
+  for (const [path, scope] of candidates) {
+    if (!existsSync3(path)) continue;
+    try {
+      const parsed = JSON.parse(readFileSync3(path, "utf-8"));
+      if (isEnrolment(parsed)) return { enrolment: parsed, scope };
+    } catch {
+    }
+    return { corrupt: true, path };
+  }
+  return null;
+}
+function writeFileMode(path, content, mode) {
+  mkdirSync3(dirname2(path), { recursive: true, mode: 448 });
+  writeFileSync2(path, content, { encoding: "utf-8", mode });
+  try {
+    chmodSync(path, mode);
+  } catch {
+  }
+}
+function readCachedPolicies(paths) {
+  const out = [];
+  for (const p of [paths.userPolicy, paths.machinePolicy]) {
+    try {
+      if (existsSync3(p)) out.push(readFileSync3(p, "utf-8").trim());
+    } catch {
+    }
+  }
+  return out.filter(Boolean);
+}
+function writeCachedPolicy(paths, jws) {
+  writeFileMode(paths.userPolicy, jws, 384);
+}
+function readKeyChain(paths) {
+  try {
+    const parsed = JSON.parse(readFileSync3(paths.userKeyChain, "utf-8"));
+    return Array.isArray(parsed) ? parsed.filter((j) => typeof j === "string") : [];
+  } catch {
+    return [];
+  }
+}
+function writeKeyChain(paths, chain) {
+  writeFileMode(paths.userKeyChain, JSON.stringify(chain), 384);
+}
+function readState(paths) {
+  try {
+    const parsed = JSON.parse(readFileSync3(paths.state, "utf-8"));
+    const c = parsed.counts;
+    return {
+      ...parsed,
+      counts: { allow: c?.allow ?? 0, deny: c?.deny ?? 0, ask: c?.ask ?? 0 }
+    };
+  } catch {
+    return { counts: { allow: 0, deny: 0, ask: 0 } };
+  }
+}
+function writeState(paths, state) {
+  try {
+    writeFileMode(paths.state, JSON.stringify(state, null, 2), 384);
+  } catch {
+  }
+}
+
+// src/claude-code/guard-config.ts
+var FLEET_ENROL_INSTRUCTION = "This machine must be enrolled: run `astrasync guard enroll <token>` (token from your organisation admin)";
+function resolveFleet(env, home = homedir2(), paths = fleetPaths({ home, env })) {
+  const loaded = loadEnrolment(paths);
+  if (loaded && "corrupt" in loaded) {
+    return {
+      kind: "fleet-missing",
+      reason: `fleet enrolment at ${loaded.path} is unreadable \u2014 ${FLEET_ENROL_INSTRUCTION}`,
+      paths
+    };
+  }
+  if (loaded) return { kind: "fleet", enrolment: loaded.enrolment, scope: loaded.scope, paths };
+  if (env.ASTRASYNC_FLEET_REQUIRED === "1") {
+    return { kind: "fleet-missing", reason: FLEET_ENROL_INSTRUCTION, paths };
+  }
+  return null;
+}
+
+// src/fleet/client.ts
+var FleetApiError = class extends Error {
+  constructor(message, status, code) {
+    super(message);
+    this.status = status;
+    this.code = code;
+    this.name = "FleetApiError";
+  }
+};
+function base(url) {
+  return url.replace(/\/+$/, "");
+}
+async function call(url, init, timeoutMs) {
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), timeoutMs);
+  try {
+    const res = await sdkFetch(url, { ...init, signal: controller.signal });
+    const text = await res.text();
+    let body = text;
+    try {
+      body = text ? JSON.parse(text) : null;
+    } catch {
+    }
+    return { status: res.status, body, headers: res.headers };
+  } finally {
+    clearTimeout(timer);
+  }
+}
+function apiError(status, body, what) {
+  const b = body ?? {};
+  const code = b.code ?? (typeof b.error === "string" ? b.error : void 0);
+  return new FleetApiError(`${what} failed (${status}${code ? ` ${code}` : ""})`, status, code);
+}
+function deviceHeaders(enrolment) {
+  return { Authorization: `Device ${enrolment.credential}` };
+}
+function policyEtag(sha256, kid) {
+  return `${sha256}.${kid ?? ""}`;
+}
+async function fetchFleetPolicy(enrolment, ifNoneMatch, timeoutMs = 1500) {
+  const headers = deviceHeaders(enrolment);
+  if (ifNoneMatch) headers["If-None-Match"] = `"${ifNoneMatch}"`;
+  const r = await call(`${base(enrolment.apiBaseUrl)}/fleet/policy`, { headers }, timeoutMs);
+  if (r.status === 304) return { status: "unchanged" };
+  if (r.status !== 200) throw apiError(r.status, r.body, "Policy fetch");
+  const data = r.body;
+  if (!data?.policyJws) throw new FleetApiError("Policy response has no policyJws", r.status);
+  return {
+    status: "ok",
+    policyJws: data.policyJws,
+    sha256: data.sha256 ?? "",
+    version: data.version ?? 0,
+    keyChain: Array.isArray(data.keyChain) ? data.keyChain.filter((j) => typeof j === "string") : []
+  };
+}
+async function sendHeartbeat(enrolment, request, timeoutMs = 1500) {
+  const r = await call(
+    `${base(enrolment.apiBaseUrl)}/fleet/heartbeat`,
+    {
+      method: "POST",
+      headers: { ...deviceHeaders(enrolment), "Content-Type": "application/json" },
+      body: JSON.stringify(request)
+    },
+    timeoutMs
+  );
+  if (r.status !== 200) throw apiError(r.status, r.body, "Heartbeat");
+  return r.body;
+}
+
+// src/fleet/policy-verify.ts
+import { createHash as createHash4 } from "crypto";
+import { compactVerify, createLocalJWKSet } from "jose";
+
+// src/fleet/types.ts
+var FLEET_POLICY_TYP = "fleet-policy+jwt";
+var FLEET_POLICY_AUD = "astrasync:fleet-policy";
+var FLEET_POLICY_ISS = "astrasync.ai";
+var FLEET_KEYS_TYP = "fleet-keys+jwt";
+var FLEET_KEYS_AUD = "astrasync:fleet-keys";
+var DEFAULT_HEARTBEAT_INTERVAL_SEC = 900;
+
+// src/fleet/policy-verify.ts
+var FleetPolicyError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "FleetPolicyError";
+  }
+};
+function sha256Hex(text) {
+  return createHash4("sha256").update(text, "utf8").digest("hex");
+}
+function parseSettings(raw) {
+  if (!raw || typeof raw !== "object") throw new FleetPolicyError("settings missing");
+  const s = raw;
+  if (s.posture !== "active" && s.posture !== "passive") {
+    throw new FleetPolicyError("settings.posture must be active or passive");
+  }
+  if (s.failOpen !== void 0 && typeof s.failOpen !== "boolean") {
+    throw new FleetPolicyError("settings.failOpen must be a boolean");
+  }
+  if (s.governMcp !== void 0 && typeof s.governMcp !== "boolean") {
+    throw new FleetPolicyError("settings.governMcp must be a boolean");
+  }
+  if (s.skipTools !== void 0 && !(Array.isArray(s.skipTools) && s.skipTools.every((t) => typeof t === "string"))) {
+    throw new FleetPolicyError("settings.skipTools must be a string array");
+  }
+  const num = (v) => typeof v === "number" && Number.isFinite(v) && v > 0 ? v : void 0;
+  return {
+    posture: s.posture,
+    failOpen: s.failOpen === true,
+    ...typeof s.governMcp === "boolean" && { governMcp: s.governMcp },
+    ...Array.isArray(s.skipTools) && { skipTools: s.skipTools },
+    ...num(s.heartbeatIntervalSec) && { heartbeatIntervalSec: num(s.heartbeatIntervalSec) },
+    ...num(s.staleAfterSec) && { staleAfterSec: num(s.staleAfterSec) }
+  };
+}
+async function verifyFleetPolicy(jws, jwks, options) {
+  let payloadBytes;
+  let header;
+  try {
+    const keySet = createLocalJWKSet(jwks);
+    const result = await compactVerify(jws, keySet, { algorithms: ["ES256"] });
+    payloadBytes = result.payload;
+    header = result.protectedHeader;
+  } catch (err) {
+    throw new FleetPolicyError(
+      `signature verification failed (${err instanceof Error ? err.message : String(err)})`
+    );
+  }
+  if (header.typ !== FLEET_POLICY_TYP) throw new FleetPolicyError(`unexpected typ ${header.typ}`);
+  let claims;
+  try {
+    claims = JSON.parse(new TextDecoder().decode(payloadBytes));
+  } catch {
+    throw new FleetPolicyError("payload is not JSON");
+  }
+  if (claims.aud !== FLEET_POLICY_AUD) throw new FleetPolicyError(`unexpected aud ${claims.aud}`);
+  if (claims.iss !== FLEET_POLICY_ISS) throw new FleetPolicyError(`unexpected iss ${claims.iss}`);
+  if (!options.expectedSubs.includes(claims.sub)) {
+    throw new FleetPolicyError("policy was issued for a different organisation or device");
+  }
+  if (!Number.isInteger(claims.ver) || claims.ver < 1) {
+    throw new FleetPolicyError("ver must be a positive integer");
+  }
+  if (options.minVer !== void 0 && claims.ver < options.minVer) {
+    throw new FleetPolicyError(
+      `policy version ${claims.ver} is older than the accepted version ${options.minVer} (rollback)`
+    );
+  }
+  if (typeof claims.yaml !== "string" || sha256Hex(claims.yaml) !== claims.sha256) {
+    throw new FleetPolicyError("policy hash mismatch");
+  }
+  const settings = parseSettings(claims.settings);
+  const parsed = parseYaml(claims.yaml);
+  if (!parsed.policy) {
+    const first = parsed.validation.errors[0];
+    throw new FleetPolicyError(
+      `signed policy is invalid${first ? ` (${first.path}: ${first.message})` : ""}`
+    );
+  }
+  return { kid: header.kid, claims: { ...claims, settings }, policy: parsed.policy, settings, jws };
+}
+async function resolveFleetJwks(pinned, chain, ownerId) {
+  let trusted = pinned;
+  let lastSeq = 0;
+  for (const link of chain) {
+    try {
+      const { payload, protectedHeader } = await compactVerify(
+        link,
+        createLocalJWKSet(trusted),
+        { algorithms: ["ES256"] }
+      );
+      if (protectedHeader.typ !== FLEET_KEYS_TYP) continue;
+      const c = JSON.parse(new TextDecoder().decode(payload));
+      if (c.aud !== FLEET_KEYS_AUD || c.iss !== FLEET_POLICY_ISS || c.sub !== ownerId) continue;
+      if (!Number.isInteger(c.seq) || c.seq <= lastSeq) continue;
+      const keys = Array.isArray(c.keys) ? c.keys.filter((k) => k && k.kty === "EC" && k.crv === "P-256" && !("d" in k)) : [];
+      if (keys.length === 0) continue;
+      trusted = { keys };
+      lastSeq = c.seq;
+    } catch {
+    }
+  }
+  return trusted;
+}
+async function pickNewestValid(candidates, jwks, options) {
+  let best = null;
+  let lastError = new FleetPolicyError("no cached fleet policy");
+  for (const jws of candidates) {
+    try {
+      const v = await verifyFleetPolicy(jws, jwks, options);
+      if (!best || v.claims.ver > best.claims.ver) best = v;
+    } catch (err) {
+      lastError = err instanceof Error ? err : new FleetPolicyError(String(err));
+    }
+  }
+  if (!best) throw lastError;
+  return best;
+}
+
+// src/fleet/runtime.ts
+var FLEET_SYNC_BUDGET_MS = 1500;
+function expectedSubs(enrolment) {
+  return [enrolment.org.ownerId, enrolment.deviceId];
+}
+function fleetJwks(ctx, chain = readKeyChain(ctx.paths)) {
+  return resolveFleetJwks(ctx.enrolment.jwks, chain, ctx.enrolment.org.ownerId);
+}
+async function acceptFetched(ctx, fetched, verifyOpts) {
+  const jwks = await fleetJwks(ctx, fetched.keyChain);
+  const verified = await verifyFleetPolicy(fetched.policyJws, jwks, verifyOpts);
+  if (fetched.keyChain.length > 0) writeKeyChain(ctx.paths, fetched.keyChain);
+  writeCachedPolicy(ctx.paths, fetched.policyJws);
+  return verified;
+}
+function remember(paths, state, verified) {
+  state.appliedSha256 = verified.claims.sha256;
+  state.maxVer = Math.max(state.maxVer ?? 0, verified.claims.ver);
+  writeState(paths, state);
+}
+async function loadFleetPolicy(ctx, options = {}) {
+  const state = readState(ctx.paths);
+  const verifyOpts = { expectedSubs: expectedSubs(ctx.enrolment), minVer: state.maxVer };
+  const cached = readCachedPolicies(ctx.paths);
+  let cacheError = null;
+  if (cached.length > 0) {
+    try {
+      const verified = await pickNewestValid(cached, await fleetJwks(ctx), verifyOpts);
+      if (verified.claims.sha256 !== state.appliedSha256 || verified.claims.ver !== state.maxVer) {
+        remember(ctx.paths, state, verified);
+      }
+      return verified;
+    } catch (err) {
+      cacheError = err instanceof Error ? err : new FleetPolicyError(String(err));
+    }
+  }
+  try {
+    const fetched = await fetchFleetPolicy(
+      ctx.enrolment,
+      null,
+      options.fetchTimeoutMs ?? FLEET_SYNC_BUDGET_MS
+    );
+    if (fetched.status === "ok") {
+      const verified = await acceptFetched(ctx, fetched, verifyOpts);
+      remember(ctx.paths, state, verified);
+      return verified;
+    }
+  } catch (err) {
+    if (err instanceof FleetApiError && err.code === "DEVICE_REVOKED") {
+      state.revoked = true;
+      writeState(ctx.paths, state);
+    }
+    const detail = err instanceof Error ? err.message : String(err);
+    throw new FleetPolicyError(
+      cacheError ? `${cacheError.message}; refresh failed (${detail})` : `no fleet policy (${detail})`
+    );
+  }
+  throw cacheError ?? new FleetPolicyError("no fleet policy available");
+}
+async function recordDecisionAndHeartbeat(ctx, applied, decision, options) {
+  const now = options.now ?? Date.now;
+  const state = readState(ctx.paths);
+  if (decision) state.counts[decision] += 1;
+  const intervalSec = applied.settings.heartbeatIntervalSec ?? ctx.enrolment.heartbeatIntervalSec ?? DEFAULT_HEARTBEAT_INTERVAL_SEC;
+  const last = state.lastHeartbeatAt ? Date.parse(state.lastHeartbeatAt) : 0;
+  if (now() - last < intervalSec * 1e3) {
+    writeState(ctx.paths, state);
+    return "throttled";
+  }
+  const deadline = now() + (options.budgetMs ?? FLEET_SYNC_BUDGET_MS);
+  const remaining = () => Math.max(1, deadline - now());
+  state.lastHeartbeatAt = new Date(now()).toISOString();
+  let outcome = "sent";
+  try {
+    const hb = await sendHeartbeat(
+      ctx.enrolment,
+      {
+        appliedSha256: applied.claims.sha256,
+        guardVersion: options.guardVersion,
+        adapters: [{ kind: options.adapterKind, managed: true }],
+        posture: applied.settings.posture,
+        decisionCounts: { ...state.counts }
+      },
+      remaining()
+    );
+    state.counts = { allow: 0, deny: 0, ask: 0 };
+    state.revoked = false;
+    delete state.lastError;
+    const policyChanged = !!hb.activeSha256 && hb.activeSha256 !== applied.claims.sha256;
+    const keyChanged = !!hb.activeKid && !!applied.kid && hb.activeKid !== applied.kid;
+    if (policyChanged || keyChanged) {
+      const fetched = await fetchFleetPolicy(
+        ctx.enrolment,
+        policyEtag(applied.claims.sha256, applied.kid),
+        remaining()
+      );
+      if (fetched.status === "ok") {
+        const verified = await acceptFetched(ctx, fetched, {
+          expectedSubs: expectedSubs(ctx.enrolment),
+          minVer: Math.max(state.maxVer ?? 0, applied.claims.ver)
+        });
+        state.appliedSha256 = verified.claims.sha256;
+        state.maxVer = Math.max(state.maxVer ?? 0, verified.claims.ver);
+        outcome = "refreshed";
+      }
+    }
+  } catch (err) {
+    if (err instanceof FleetApiError && err.code === "DEVICE_REVOKED") {
+      state.revoked = true;
+      outcome = "revoked";
+    } else {
+      state.lastError = err instanceof Error ? err.message : String(err);
+      outcome = "failed";
+    }
+  }
+  writeState(ctx.paths, state);
+  return outcome;
+}
+
 // src/cursor/extension.ts
 var adapter = null;
 var gateway = null;
-async function activate(vscodeApi) {
-  const config = loadConfiguration();
+async function activate(vscodeApi, options = {}) {
+  const config = await loadConfiguration(options);
   gateway = new AstraSyncGateway(config.gatewayConfig);
   adapter = new CursorAdapter({
     vscode: vscodeApi,
@@ -5141,10 +5597,18 @@ Action: ${context.action} \u2192 ${context.target}`,
     const channel = vscodeApi.window.createOutputChannel("AstraSync Policy");
     channel.appendLine(`Mode: ${gateway.mode}`);
     channel.appendLine(`Posture: ${gateway.posture}`);
-    channel.appendLine(`Policy file: ${config.gatewayConfig.policyFile || "(inline/none)"}`);
+    channel.appendLine(
+      config.fleetManaged ? `Policy: organisation fleet policy (${config.fleetManaged})` : `Policy file: ${config.gatewayConfig.policyFile || "(inline/none)"}`
+    );
     channel.show();
   });
   vscodeApi.commands.registerCommand("astrasync.switchMode", async () => {
+    if (config.fleetManaged) {
+      vscodeApi.window.showInformationMessage(
+        "AstraSync: this machine is governed by your organisation\u2019s fleet policy \u2014 mode is managed centrally."
+      );
+      return;
+    }
     if (!config.gatewayConfig.apiKey) {
       vscodeApi.window.showInformationMessage(
         "AstraSync: Set astrasync.apiKey in settings to enable online/hybrid mode."
@@ -5174,7 +5638,62 @@ async function deactivate() {
     gateway = null;
   }
 }
-function loadConfiguration() {
+function fleetLockdownPolicy() {
+  return {
+    version: "1.0",
+    name: "fleet-enrolment-required",
+    purposes: [
+      "file.read",
+      "file.write",
+      "file.delete",
+      "shell.exec",
+      "network.fetch",
+      "payment.execute",
+      "email.send"
+    ].map((id) => ({ id, allowed: false }))
+  };
+}
+async function loadConfiguration(options) {
+  const env = options.env ?? process.env;
+  const fleet = resolveFleet(env, options.home ?? homedir3(), options.fleetPaths);
+  if (fleet?.kind === "fleet") {
+    const ctx = { enrolment: fleet.enrolment, paths: fleet.paths };
+    try {
+      const verified = await loadFleetPolicy(ctx);
+      void recordDecisionAndHeartbeat(ctx, verified, null, {
+        guardVersion: SDK_VERSION,
+        adapterKind: "cursor"
+      }).catch(() => void 0);
+      return {
+        fleetManaged: `${fleet.enrolment.org.name ?? fleet.enrolment.org.ownerId} v${verified.claims.ver}`,
+        gatewayConfig: {
+          mode: "local",
+          policy: verified.policy,
+          posture: verified.settings.posture,
+          traceEnabled: true,
+          tracePath: join3(fleet.paths.userDir, "traces"),
+          debug: false
+        }
+      };
+    } catch {
+    }
+  }
+  if (fleet) {
+    return {
+      fleetManaged: "unavailable \u2014 enrol this machine with `astrasync guard enroll <token>`",
+      gatewayConfig: {
+        mode: "local",
+        policy: fleetLockdownPolicy(),
+        posture: "active",
+        traceEnabled: true,
+        tracePath: join3(fleet.paths.userDir, "traces"),
+        debug: false
+      }
+    };
+  }
+  return loadLocalConfiguration();
+}
+function loadLocalConfiguration() {
   return {
     gatewayConfig: {
       mode: "local",