@integrity-labs/agt-cli 0.28.431 → 0.28.433

package/dist/mcp/direct-chat-channel.js

@@ -18867,7 +18867,7 @@ var require_filters = __commonJS({
       return r.copySafeness(str, res);
     }
     _exports.indent = indent;
-    function join12(arr, del, attr) {
+    function join13(arr, del, attr) {
       del = del || "";
       if (attr) {
         arr = lib.map(arr, function(v) {
@@ -18876,7 +18876,7 @@ var require_filters = __commonJS({
       }
       return arr.join(del);
     }
-    _exports.join = join12;
+    _exports.join = join13;
     function last(arr) {
       return arr[arr.length - 1];
     }
@@ -30399,14 +30399,14 @@ function readLockHolder(path) {
 }
 
 // src/direct-chat-channel.ts
-import { homedir as homedir3 } from "os";
-import { join as join11 } from "path";
+import { homedir as homedir4 } from "os";
+import { join as join12 } from "path";
 import { randomUUID } from "crypto";
 import {
   watch,
   mkdirSync as mkdirSync5,
   writeFileSync as writeFileSync6,
-  readFileSync as readFileSync10,
+  readFileSync as readFileSync11,
   readdirSync as readdirSync4,
   existsSync as existsSync6,
   renameSync as renameSync5,
@@ -34162,6 +34162,28 @@ function buildUsageLimitReplyText(limitedUntil) {
   return `Your agent has hit its Claude Code usage limit until ${formatUtcClock(limitedUntil)}. It'll pick back up once the limit resets \u2014 please try again after then.`;
 }
 
+// ../core/dist/account-enforcement/marker.js
+var ACCOUNT_ENFORCEMENT_MARKER_FILENAME = "account-enforcement.json";
+var ACCOUNT_ENFORCEMENT_MARKER_VERSION = 1;
+function parseAccountEnforcementMarker(raw) {
+  let parsed;
+  try {
+    parsed = JSON.parse(raw);
+  } catch {
+    return null;
+  }
+  if (typeof parsed !== "object" || parsed === null || parsed.version !== ACCOUNT_ENFORCEMENT_MARKER_VERSION) {
+    return null;
+  }
+  const level = parsed.level;
+  if (level !== "warn" && level !== "mute")
+    return null;
+  return { version: ACCOUNT_ENFORCEMENT_MARKER_VERSION, level };
+}
+function buildAccountIssueReplyText() {
+  return "There is an issue with your account, please contact support@augmented.team.";
+}
+
 // ../core/dist/kanban/state-machine.js
 var KANBAN_STATUSES = [
   "backlog",
@@ -35461,6 +35483,26 @@ function readUsageLimitUntil(opts) {
   return resolveUsageLimitUntil(parseUsageLimitMarker(raw), opts.now ?? /* @__PURE__ */ new Date());
 }
 
+// src/account-enforcement-notice.ts
+import { readFileSync as readFileSync10 } from "fs";
+import { homedir as homedir3 } from "os";
+import { join as join11 } from "path";
+function accountEnforcementMarkerPath(codeName) {
+  if (!codeName) return null;
+  return join11(homedir3(), ".augmented", codeName, ACCOUNT_ENFORCEMENT_MARKER_FILENAME);
+}
+function readAccountEnforcementLevel(opts) {
+  const path = opts.filePath ?? accountEnforcementMarkerPath(opts.codeName);
+  if (!path) return null;
+  let raw;
+  try {
+    raw = readFileSync10(path, "utf-8");
+  } catch {
+    return null;
+  }
+  return parseAccountEnforcementMarker(raw)?.level ?? null;
+}
+
 // ../core/dist/channels/governance/sender-policy-decline.js
 function decideDeclineReply(input) {
   const last = input.cache.get(input.key);
@@ -35482,24 +35524,25 @@ var DIRECT_CHAT_MAINTENANCE_COOLDOWN_MS = (() => {
   return Number.isFinite(n) && n >= 0 ? n * 1e3 : 1800 * 1e3;
 })();
 var DIRECT_CHAT_USAGE_LIMIT_CACHE = /* @__PURE__ */ new Map();
+var DIRECT_CHAT_ACCOUNT_MUTE_CACHE = /* @__PURE__ */ new Map();
 var AGT_HOST = process.env.AGT_HOST;
 var AGT_API_KEY = process.env.AGT_API_KEY;
 var AGT_AGENT_ID = process.env.AGT_AGENT_ID;
-var DIRECT_CHAT_AGENT_DIR = AGT_AGENT_ID ? join11(homedir3(), ".augmented", AGT_AGENT_ID) : null;
+var DIRECT_CHAT_AGENT_DIR = AGT_AGENT_ID ? join12(homedir4(), ".augmented", AGT_AGENT_ID) : null;
 var INBOUND_ATTACHMENTS_DIR = resolveInboundAttachmentsDir({
   codeName: process.env.AGT_AGENT_CODE_NAME,
   turnInitiatorFile: process.env.AGT_TURN_INITIATOR_FILE,
   agentId: AGT_AGENT_ID,
-  homeDir: homedir3()
+  homeDir: homedir4()
 });
 var AGT_AGENT_CODE_NAME = process.env.AGT_AGENT_CODE_NAME;
 var directChatStderrLogStream = null;
 if (AGT_AGENT_CODE_NAME) {
   try {
-    const logDir = join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME);
+    const logDir = join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME);
     mkdirSync5(logDir, { recursive: true });
     directChatStderrLogStream = createWriteStream(
-      join11(logDir, "direct-chat-channel-stderr.log"),
+      join12(logDir, "direct-chat-channel-stderr.log"),
       { flags: "a", mode: 384 }
     );
     directChatStderrLogStream.on("error", () => {
@@ -35518,10 +35561,10 @@ if (AGT_AGENT_CODE_NAME) {
   } catch {
   }
 }
-var PROGRESS_HEARTBEAT_PATH = AGT_AGENT_CODE_NAME ? join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME, "channel-progress-heartbeat.json") : null;
-var DIRECT_CHAT_PENDING_INBOUND_DIR = AGT_AGENT_CODE_NAME ? join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME, "direct-chat-pending-inbound") : null;
-var DIRECT_CHAT_DELIVERY_LEDGER_DIR = AGT_AGENT_CODE_NAME ? join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME, ".agt-inbound-delivery-ledger") : null;
-var DIRECT_CHAT_CODE_NAME_AGENT_DIR = AGT_AGENT_CODE_NAME ? join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME) : null;
+var PROGRESS_HEARTBEAT_PATH = AGT_AGENT_CODE_NAME ? join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME, "channel-progress-heartbeat.json") : null;
+var DIRECT_CHAT_PENDING_INBOUND_DIR = AGT_AGENT_CODE_NAME ? join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME, "direct-chat-pending-inbound") : null;
+var DIRECT_CHAT_DELIVERY_LEDGER_DIR = AGT_AGENT_CODE_NAME ? join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME, ".agt-inbound-delivery-ledger") : null;
+var DIRECT_CHAT_CODE_NAME_AGENT_DIR = AGT_AGENT_CODE_NAME ? join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME) : null;
 var DIRECT_CHAT_STALE_MARKER_MS = 24 * 60 * 60 * 1e3;
 function recordDirectChatDelivery(sessionId, messageIds) {
   if (!sessionId) return;
@@ -35532,8 +35575,8 @@ function recordDirectChatDelivery(sessionId, messageIds) {
     delivered_at: (/* @__PURE__ */ new Date()).toISOString()
   });
 }
-var DIRECT_CHAT_RECOVERY_OUTBOX_DIR = AGT_AGENT_CODE_NAME ? join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME, "direct-chat-recovery-outbox") : null;
-var DIRECT_CHAT_RECOVERY_LEDGER_DIR = AGT_AGENT_CODE_NAME ? join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME, ".agt-direct-chat-recovery-ledger") : null;
+var DIRECT_CHAT_RECOVERY_OUTBOX_DIR = AGT_AGENT_CODE_NAME ? join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME, "direct-chat-recovery-outbox") : null;
+var DIRECT_CHAT_RECOVERY_LEDGER_DIR = AGT_AGENT_CODE_NAME ? join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME, ".agt-direct-chat-recovery-ledger") : null;
 var progressReceivedAt = /* @__PURE__ */ new Map();
 var directChatProgressState = { tracked: null };
 var directChatProgressTickRunning = false;
@@ -35552,7 +35595,7 @@ var directChatKanbanCardClient = createKanbanCardActiveClient({
 function readProgressHeartbeat() {
   if (!PROGRESS_HEARTBEAT_PATH || !existsSync6(PROGRESS_HEARTBEAT_PATH)) return null;
   try {
-    return parseProgressHeartbeat(readFileSync10(PROGRESS_HEARTBEAT_PATH, "utf-8"));
+    return parseProgressHeartbeat(readFileSync11(PROGRESS_HEARTBEAT_PATH, "utf-8"));
   } catch {
     return null;
   }
@@ -35561,7 +35604,7 @@ function seedProgressHeartbeat() {
   if (!PROGRESS_HEARTBEAT_PATH) return;
   const tmp = `${PROGRESS_HEARTBEAT_PATH}.${process.pid}.tmp`;
   try {
-    mkdirSync5(join11(homedir3(), ".augmented", AGT_AGENT_CODE_NAME), { recursive: true });
+    mkdirSync5(join12(homedir4(), ".augmented", AGT_AGENT_CODE_NAME), { recursive: true });
     writeFileSync6(tmp, serializeProgressHeartbeat(SEED_PROGRESS_STEP, Date.now()), { mode: 384 });
     renameSync5(tmp, PROGRESS_HEARTBEAT_PATH);
   } catch {
@@ -35721,7 +35764,7 @@ var inboundAttachmentDeps = {
   },
   ensureDir: (dir) => mkdirSync5(dir, { recursive: true }),
   writeFile: (path, bytes) => writeFileSync6(path, bytes, { mode: 384 }),
-  joinPath: (...parts) => join11(...parts),
+  joinPath: (...parts) => join12(...parts),
   warn: (msg) => process.stderr.write(`${msg}
 `)
 };
@@ -36051,7 +36094,7 @@ function scheduleDirectChatBusyAck(sessionId, messageId, arrivedWhileBusy) {
     let paneLogFreshAgeMs = null;
     if (DIRECT_CHAT_CODE_NAME_AGENT_DIR) {
       try {
-        const paneMtimeMs = statSync2(join11(DIRECT_CHAT_CODE_NAME_AGENT_DIR, "pane.log")).mtimeMs;
+        const paneMtimeMs = statSync2(join12(DIRECT_CHAT_CODE_NAME_AGENT_DIR, "pane.log")).mtimeMs;
         paneLogFreshAgeMs = Math.max(0, Date.now() - paneMtimeMs);
       } catch {
       }
@@ -36124,8 +36167,9 @@ async function pollForMessages(sinceMs) {
       if (processedIds.has(msg.id)) continue;
       const isNotice = msg.kind === "notice";
       const maintenanceActive = !isNotice && isMaintenanceModeActive();
-      const usageLimitUntil = !isNotice && !maintenanceActive ? readUsageLimitUntil({ codeName: AGT_AGENT_CODE_NAME ?? null }) : null;
-      if (!isNotice && !maintenanceActive && !usageLimitUntil) {
+      const accountMuted = !isNotice && !maintenanceActive ? readAccountEnforcementLevel({ codeName: AGT_AGENT_CODE_NAME ?? null }) === "mute" : false;
+      const usageLimitUntil = !isNotice && !maintenanceActive && !accountMuted ? readUsageLimitUntil({ codeName: AGT_AGENT_CODE_NAME ?? null }) : null;
+      if (!isNotice && !maintenanceActive && !accountMuted && !usageLimitUntil) {
         markProcessed(msg.id);
         claimTracker.track(msg.session_id, msg.id);
         if (!progressReceivedAt.has(msg.session_id)) {
@@ -36159,6 +36203,37 @@ async function pollForMessages(sinceMs) {
           if (throttle.reply) DIRECT_CHAT_MAINTENANCE_CACHE.delete(throttleKey);
           process.stderr.write(
             `direct-chat-channel: maintenance offline reply failed: ${err.message}
+`
+          );
+        }
+        continue;
+      }
+      if (accountMuted) {
+        const throttleKey = `${msg.session_id}|account_mute`;
+        const throttle = decideDeclineReply({
+          cache: DIRECT_CHAT_ACCOUNT_MUTE_CACHE,
+          key: throttleKey,
+          cooldownMs: DIRECT_CHAT_MAINTENANCE_COOLDOWN_MS,
+          now: Date.now()
+        });
+        const route = throttle.reply ? "/host/direct-chat/reply" : "/host/direct-chat/consume";
+        const body = throttle.reply ? { agent_id: AGT_AGENT_ID, session_id: msg.session_id, content: buildAccountIssueReplyText(), message_ids: [msg.id] } : { agent_id: AGT_AGENT_ID, session_id: msg.session_id, message_ids: [msg.id] };
+        try {
+          const res2 = await apiPost(route, body);
+          const data2 = await res2.json().catch(() => ({}));
+          if (res2.ok && data2.error == null) {
+            markProcessed(msg.id);
+          } else {
+            if (throttle.reply) DIRECT_CHAT_ACCOUNT_MUTE_CACHE.delete(throttleKey);
+            process.stderr.write(
+              `direct-chat-channel: account-mute ${throttle.reply ? "reply" : "consume"} failed: ${data2.error ?? `HTTP ${res2.status}`}
+`
+            );
+          }
+        } catch (err) {
+          if (throttle.reply) DIRECT_CHAT_ACCOUNT_MUTE_CACHE.delete(throttleKey);
+          process.stderr.write(
+            `direct-chat-channel: account-mute reply failed: ${err.message}
 `
           );
         }
@@ -36377,7 +36452,7 @@ function sweepAgedDirectChatMarkersNow(thresholdMs) {
   const now = Date.now();
   const res = sweepAgedDirectChatMarkers(DIRECT_CHAT_PENDING_INBOUND_DIR, {
     readdir: (dir) => readdirSync4(dir),
-    readFile: (p2) => readFileSync10(p2, "utf8"),
+    readFile: (p2) => readFileSync11(p2, "utf8"),
     unlink: (p2) => {
       if (existsSync6(p2)) unlinkSync4(p2);
     },
@@ -36413,7 +36488,7 @@ function sanitizeRecoveryText(text) {
 }
 function directChatRecoveryDeps() {
   return {
-    readFile: (p2) => readFileSync10(p2, "utf-8"),
+    readFile: (p2) => readFileSync11(p2, "utf-8"),
     renameFile: (from, to) => renameSync5(from, to),
     unlinkFile: (p2) => {
       if (existsSync6(p2)) unlinkSync4(p2);
@@ -36426,7 +36501,7 @@ function directChatRecoveryDeps() {
       if (!DIRECT_CHAT_RECOVERY_LEDGER_DIR) return;
       if (markerName.includes("/") || markerName.includes("\\") || markerName.includes("..")) return;
       try {
-        const p2 = join11(DIRECT_CHAT_RECOVERY_LEDGER_DIR, markerName);
+        const p2 = join12(DIRECT_CHAT_RECOVERY_LEDGER_DIR, markerName);
         if (existsSync6(p2)) unlinkSync4(p2);
       } catch {
       }
@@ -36462,7 +36537,7 @@ async function processDirectChatRecoveryOutboxFile(filename) {
   if (!enabled) return;
   directChatRecoveryInFlight.add(filename);
   try {
-    const fullPath = join11(DIRECT_CHAT_RECOVERY_OUTBOX_DIR, filename);
+    const fullPath = join12(DIRECT_CHAT_RECOVERY_OUTBOX_DIR, filename);
     await consumeDirectChatRecoveryFile(fullPath, filename, directChatRecoveryDeps());
   } catch (err) {
     process.stderr.write(
@@ -36490,7 +36565,7 @@ if (DIRECT_CHAT_RECOVERY_OUTBOX_DIR) {
       if (!filename) return;
       const name = filename.toString();
       if (!name.endsWith(".json")) return;
-      if (existsSync6(join11(DIRECT_CHAT_RECOVERY_OUTBOX_DIR, name))) {
+      if (existsSync6(join12(DIRECT_CHAT_RECOVERY_OUTBOX_DIR, name))) {
         void processDirectChatRecoveryOutboxFile(name);
       }
     });