@nurix/nustack 0.23.8 → 0.23.10

package/dist/commands/census.js

@@ -0,0 +1,48 @@
+import { attributeTranscript } from "../lib/census/attribute.js";
+import { CensusAccumulator, renderCensusText } from "../lib/census/report.js";
+import { discoverSessions, readLines } from "../lib/census/transcripts.js";
+import { printJsonSuccess } from "../lib/json_output.js";
+class CensusArgumentError extends Error {
+    code = "CENSUS_ARGUMENT_INVALID";
+    constructor(message) {
+        super(message);
+        this.name = "CensusArgumentError";
+    }
+}
+function parseSince(value) {
+    if (value === undefined)
+        return undefined;
+    const instant = /^\d{4}-\d{2}-\d{2}$/.test(value) ? Date.parse(`${value}T00:00:00Z`) : Number.NaN;
+    if (Number.isNaN(instant))
+        throw new CensusArgumentError(`--since takes a date as YYYY-MM-DD, not "${value}".`);
+    return instant;
+}
+function parseMaxSessions(value) {
+    if (value === undefined)
+        return undefined;
+    if (!/^\d+$/.test(value) || Number.parseInt(value, 10) < 1) {
+        throw new CensusArgumentError(`--max-sessions takes a positive integer, not "${value}".`);
+    }
+    return Number.parseInt(value, 10);
+}
+export async function runCensus(options) {
+    const sessions = await discoverSessions({
+        root: options.root,
+        sinceMs: parseSince(options.since),
+        maxSessions: parseMaxSessions(options.maxSessions),
+    });
+    const census = new CensusAccumulator();
+    for (const session of sessions) {
+        census.addSession();
+        census.addTranscript(await attributeTranscript(readLines(session.mainPath), "main"), "main");
+        for (const sidechainPath of session.sidechainPaths) {
+            census.addTranscript(await attributeTranscript(readLines(sidechainPath), "sidechain"), "sidechain");
+        }
+    }
+    const report = census.finish();
+    if (options.json) {
+        printJsonSuccess("census", report);
+        return;
+    }
+    console.log(renderCensusText(report));
+}

package/dist/commands/gate.js

@@ -0,0 +1,75 @@
+import os from "node:os";
+import path from "node:path";
+import { hasManagedHook, installManagedHook, removeManagedHook } from "../lib/agent/claudeHooks.js";
+import { readDesktopSettings } from "../lib/desktop_settings.js";
+import { loadOutlineLibrary } from "../lib/gate/outline.js";
+import { resolveThreshold } from "../lib/gate/settings.js";
+import { printJsonSuccess } from "../lib/json_output.js";
+import { readStoreVersion } from "../lib/local_store/database.js";
+export const GATE_HOOK_TIMEOUT_SECONDS = 10;
+export const GATE_HOOKS = [
+    { matcher: "Read", command: "nustack gate read" },
+    { matcher: "Bash", command: "nustack gate bash" },
+];
+const GATE_EVENT = "PreToolUse";
+function claudeSettingsPath() {
+    return path.join(os.homedir(), ".claude", "settings.json");
+}
+export async function runGateInstall() {
+    let isChanged = false;
+    for (const hook of GATE_HOOKS) {
+        const { changed } = await installManagedHook(claudeSettingsPath(), GATE_EVENT, hook.matcher, hook.command, {
+            timeout: GATE_HOOK_TIMEOUT_SECONDS,
+        });
+        isChanged ||= changed;
+    }
+    console.log(isChanged
+        ? "Installed the large-read gate's Read and Bash hooks."
+        : "Installed the large-read gate's Read and Bash hooks (already installed — no change).");
+}
+export async function runGateRemove() {
+    let isRemoved = false;
+    for (const hook of GATE_HOOKS) {
+        const { removed } = await removeManagedHook(claudeSettingsPath(), GATE_EVENT, hook.matcher);
+        isRemoved ||= removed;
+    }
+    console.log(isRemoved
+        ? "Removed the large-read gate's Read and Bash hooks."
+        : "Removed the large-read gate's Read and Bash hooks (already absent — no change).");
+}
+export async function readGateStatus() {
+    const settingsPath = claudeSettingsPath();
+    const installed = {
+        read: await hasManagedHook(settingsPath, GATE_EVENT, "Read"),
+        bash: await hasManagedHook(settingsPath, GATE_EVENT, "Bash"),
+    };
+    const { localReader } = readDesktopSettings().routing;
+    const isInstalled = installed.read && installed.bash;
+    const isUninstalled = !installed.read && !installed.bash;
+    const { thresholdTokens, thresholdSource } = resolveThreshold();
+    const library = await loadOutlineLibrary();
+    return {
+        installed,
+        localReader,
+        isInSync: localReader ? isInstalled : isUninstalled,
+        thresholdTokens,
+        thresholdSource,
+        outline: library !== null && typeof library.outlineSource === "function" ? "tree-sitter" : "fallback",
+        storeVersion: await readStoreVersion(),
+    };
+}
+export async function runGateStatus(options) {
+    const status = await readGateStatus();
+    if (options.json) {
+        printJsonSuccess("gate status", status);
+        return;
+    }
+    const onOff = (isOn) => (isOn ? "installed" : "not installed");
+    console.log([
+        `Read hook: ${onOff(status.installed.read)} · Bash hook: ${onOff(status.installed.bash)}`,
+        `Local Reader switch: ${status.localReader ? "on" : "off"} (${status.isInSync ? "in sync with the hooks" : "out of sync with the hooks — run `nustack gate install` or `nustack gate remove`"})`,
+        `Threshold: ${status.thresholdTokens} tokens (from ${status.thresholdSource})`,
+        `Outline: ${status.outline}`,
+        `Local store version: ${status.storeVersion ?? "no store"}`,
+    ].join("\n"));
+}

package/dist/commands/telemetry.js

@@ -6,7 +6,11 @@ import { getBindingByRepositoryRoot, listActiveBindings, setBindingTelemetry, }
 import { upsertFilebeatState } from "../lib/local_store/filebeat_state.js";
 import { ensureFilebeatBinary } from "../lib/filebeat_installer.js";
 import { collectTelemetryStatus } from "../lib/telemetry/health_report.js";
-import { ensureBindingTelemetry, setLocalOutput, switchMachineOff } from "../lib/session_telemetry.js";
+import { ensureBindingTelemetry, reconcileAllBindings, setLocalOutput, switchMachineOff } from "../lib/session_telemetry.js";
+import { readPartState, telemetryReconcileLockPath } from "../lib/telemetry/part_state.js";
+import { TELEMETRY_CHILD_VARIABLE } from "../lib/telemetry/environment.js";
+import { isFlagSet } from "../lib/records/environment.js";
+import { acquireDrainLock, releaseDrainLock } from "../lib/records/drain.js";
 import { resolveRepository } from "../lib/git-sync/core/repository.js";
 function resolveTargetBinding(store, options) {
     if (options.device)
@@ -28,13 +32,37 @@ export async function runTelemetry(action, options) {
     const store = await openLocalStore();
     try {
         const verbose = { localVerbose: options.localVerbose === true };
+        if (action === "reconcile") {
+            const isChild = isFlagSet(TELEMETRY_CHILD_VARIABLE);
+            const lockPath = telemetryReconcileLockPath();
+            if (isChild && !acquireDrainLock(lockPath))
+                return;
+            try {
+                const result = await reconcileAllBindings({ baseUrl: resolveNuStackUrl(options), store });
+                if (options.quiet)
+                    return;
+                if (json) {
+                    printJsonSuccess("telemetry reconcile", { scope: "device", ...result });
+                    return;
+                }
+                p.intro("nustack telemetry reconcile");
+                p.log.success(`Reconciled ${result.reconciled} folder${result.reconciled === 1 ? "" : "s"}; parts on ${result.parts.on}, off ${result.parts.off}, unknown ${result.parts.unknown}${result.conflict ? "; two telemetry destinations conflict, so nothing ships" : ""}.`);
+                p.outro("Telemetry reconciled.");
+            }
+            finally {
+                if (isChild)
+                    releaseDrainLock(lockPath);
+            }
+            return;
+        }
         if (action === "status") {
             const status = await collectTelemetryStatus(store, undefined, verbose);
             if (json) {
                 printJsonSuccess("telemetry status", status);
                 return;
             }
-            const shipping = status.folders.filter((f) => f.enabled).length;
+            const shipping = status.folders.filter((f) => f.enabled && f.shipping).length;
+            const partsOff = status.folders.filter((f) => f.partState === "off").length;
             p.intro("nustack telemetry status");
             p.note([
                 `telemetry_enabled: ${status.telemetryEnabled}`,
@@ -45,6 +73,7 @@ export async function runTelemetry(action, options) {
                 `install state:     ${status.filebeatState}${status.lastErrorCode ? ` (${status.lastErrorCode})` : ""}`,
                 `config version:    ${status.desiredConfigVersion}`,
                 `folders:           ${shipping} of ${status.folders.length} shipping`,
+                `telemetry part:    ${partsOff === 0 ? "on for every folder's project" : `off for ${partsOff} folder${partsOff === 1 ? "" : "s"}`}`,
                 `last cloud report: ${status.lastHealthAtMs ? new Date(status.lastHealthAtMs).toISOString() : "never"}`,
             ].join("\n"), "Telemetry");
             p.outro("Telemetry status.");
@@ -99,9 +128,12 @@ export async function runTelemetry(action, options) {
                 return;
             }
             p.intro(`nustack telemetry ${action}`);
-            p.log.success(enabled
-                ? `${folderLabel(target)} is shipping. Other folders are unchanged.`
-                : `${folderLabel(target)} has stopped shipping; nothing it records while off will ever ship. Other folders are unchanged.`);
+            const isPartOff = readPartState().projects[target.projectId]?.state === "off";
+            p.log.success(enabled && isPartOff
+                ? `${folderLabel(target)} is switched on, but the tenant's telemetry part is off for this project; this folder ships when it is switched back on.`
+                : enabled
+                    ? `${folderLabel(target)} is shipping. Other folders are unchanged.`
+                    : `${folderLabel(target)} has stopped shipping; nothing it records while off will ever ship. Other folders are unchanged.`);
             p.outro(`Run \`nustack telemetry ${enabled ? "disable" : "enable"}\` here to reverse it.`);
             return;
         }