@microsoft/agent-governance-antigravity-cli 4.0.0

package/assets/extensions/agt-global-policy/hooks/session-start.mjs

@@ -0,0 +1,18 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { loadHookPolicyState, runHookMain, writeHookOutput } from "../lib/hook-runtime.mjs";
+
+await runHookMain(async () => {
+  const state = await loadHookPolicyState(import.meta.url);
+  await writeHookOutput({
+    hookSpecificOutput: {
+      additionalContext: state.policy.additionalContext.join("\n"),
+    },
+    systemMessage: "AGT Antigravity governance policy is active.",
+  });
+}, async (error) => {
+  await writeHookOutput({
+    systemMessage: `AGT Antigravity governance could not load startup context: ${error.message}`,
+  });
+});

package/assets/extensions/agt-global-policy/lib/hook-runtime.mjs

@@ -0,0 +1,62 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { fileURLToPath } from "node:url";
+import { loadPolicy } from "./policy.mjs";
+
+export async function loadHookInput() {
+  const chunks = [];
+  for await (const chunk of process.stdin) {
+    chunks.push(chunk);
+  }
+
+  const payload = Buffer.concat(chunks).toString("utf8").trim();
+  return payload ? JSON.parse(payload) : {};
+}
+
+export async function loadHookPolicyState(importMetaUrl) {
+  return loadPolicy({
+    extensionRoot: fileURLToPath(new URL("..", importMetaUrl)),
+  });
+}
+
+export function extractAntigravityToolResponse(toolResponse) {
+  return {
+    error: toolResponse?.error,
+    llmContent: toolResponse?.llmContent,
+    returnDisplay: toolResponse?.returnDisplay,
+  };
+}
+
+export async function writeHookOutput(output) {
+  await new Promise((resolve, reject) => {
+    const onError = (error) => {
+      process.stdout.off("error", onError);
+      reject(error);
+    };
+    process.stdout.once("error", onError);
+    process.stdout.write(`${JSON.stringify(output ?? {})}\n`, "utf8", () => {
+      process.stdout.off("error", onError);
+      resolve();
+    });
+  });
+}
+
+export async function writeHookStderr(message) {
+  await new Promise((resolve) => {
+    process.stderr.write(`${String(message ?? "").trim()}\n`, "utf8", resolve);
+  });
+}
+
+export async function runHookMain(handler, onError) {
+  try {
+    await handler();
+  } catch (error) {
+    await onError(error instanceof Error ? error : new Error(String(error)));
+  }
+}
+
+export async function emitSystemBlock(message) {
+  await writeHookStderr(message);
+  process.exitCode = 2;
+}

package/assets/extensions/agt-global-policy/lib/poisoning.mjs

@@ -0,0 +1,61 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+export function flattenText(value) {
+  if (value === undefined || value === null) {
+    return "";
+  }
+  if (typeof value === "string") {
+    return value;
+  }
+  if (typeof value === "number" || typeof value === "boolean") {
+    return String(value);
+  }
+  if (Array.isArray(value)) {
+    return value.map(flattenText).join("\n");
+  }
+  if (typeof value === "object") {
+    return Object.values(value).map(flattenText).join("\n");
+  }
+  return "";
+}
+
+export function summarizeText(text, maxLength = 4000) {
+  const normalized = flattenText(text).replace(/\s+/g, " ").trim();
+  if (normalized.length <= maxLength) {
+    return normalized;
+  }
+  return `${normalized.slice(0, maxLength)}...`;
+}
+
+export function summarizeTextWindows(text, maxLength = 12000) {
+  const normalized = flattenText(text).replace(/\s+/g, " ").trim();
+  if (!normalized) {
+    return [];
+  }
+  if (normalized.length <= maxLength) {
+    return [normalized];
+  }
+
+  const windowSize = Math.max(Math.floor(maxLength / 4), 1000);
+  const midStart = Math.max(Math.floor(normalized.length / 2) - Math.floor(windowSize / 2), 0);
+  const tailStart = Math.max(normalized.length - windowSize, 0);
+
+  return dedupeStrings([
+    `${normalized.slice(0, windowSize)}...`,
+    `...${normalized.slice(midStart, midStart + windowSize)}...`,
+    `...${normalized.slice(tailStart)}`,
+  ]);
+}
+
+export function safeJsonStringify(value, space = 0) {
+  try {
+    return JSON.stringify(value, null, space);
+  } catch {
+    return "[unserializable]";
+  }
+}
+
+function dedupeStrings(values) {
+  return [...new Set(values.filter(Boolean))];
+}