stim 1.20.2 → 1.22.0

package/dist/{agent-device-usage-output-Dzgv6lgR.mjs → agent-device-usage-output-DUFaTMdd.mjs}

@@ -1,67 +1,23 @@
-import { N as formatBytes, k as process_identity_exports } from "./config-BYDxlMY3.mjs";
-import { t as getExecutor } from "./exec-DqKo9GKJ.mjs";
-import { o as formatLongDuration } from "./command-output-BTy8hZ0F.mjs";
-import { c as listWorkspaceDirs } from "./workspaces-C36s2jb9.mjs";
-import { a as envDir, l as processMatches, s as parseAgentDeviceRecord } from "./activity-BSV745Kz.mjs";
-import { existsSync, mkdirSync, readFileSync, readdirSync, realpathSync, renameSync, rmSync, statSync, writeFileSync } from "node:fs";
-import { dirname, isAbsolute, join, relative, resolve, sep } from "node:path";
-import { agentDeviceUsageFile, readAgentDeviceUsageCache, readSwiftpmCacheUsage, swiftpmCacheUsageFile } from "@stim-cli/core/state";
-import { configDir } from "@stim-cli/core";
-import { randomUUID } from "node:crypto";
+import { g as process_identity_exports } from "./exec-DEoAhcze.mjs";
+import { P as formatBytes } from "./config-DlrsduwG.mjs";
+import { o as formatLongDuration } from "./command-output-gwpm28fu.mjs";
+import { F as writeUsageCache, N as canonical, P as measureDu, c as listWorkspaceDirs } from "./workspaces-CpANmBlq.mjs";
+import { c as parseAgentDeviceRecord, o as envDir, u as processMatches } from "./activity-CMLmzp7o.mjs";
+import { existsSync, readFileSync, readdirSync, statSync } from "node:fs";
 import { homedir } from "node:os";
+import { isAbsolute, join, relative, sep } from "node:path";
+import { configDir } from "@stim-cli/core";
+import { agentDeviceUsageFile, readAgentDeviceUsageCache, readSwiftpmCacheUsage, swiftpmCacheUsageFile } from "@stim-cli/core/state";
 import chalk from "chalk";
-//#region src/devices/report-only-usage.ts
-function canonical(path) {
-	try {
-		return realpathSync.native(path);
-	} catch {
-		return resolve(path);
-	}
-}
-function parseDu(output) {
-	const sizes = /* @__PURE__ */ new Map();
-	for (const line of output.split("\n")) {
-		const match = /^(\d+)\t(.+?)\r?$/.exec(line);
-		if (match) sizes.set(resolve(match[2]), Number(match[1]) * 1024);
-	}
-	return sizes;
-}
-async function measureDu(args) {
-	let output = "";
-	let complete = true;
-	try {
-		output = await getExecutor().runFileAsync("du", args, { timeoutMs: 2e4 });
-	} catch (error) {
-		complete = false;
-		const failure = error;
-		if (failure.status === 1 && !failure.code && typeof failure.stdout === "string") output = failure.stdout;
-	}
-	return {
-		sizes: parseDu(output),
-		complete
-	};
-}
-function writeUsageCache(file, value) {
-	const temporary = `${file}.${randomUUID()}.tmp`;
-	try {
-		mkdirSync(dirname(file), { recursive: true });
-		writeFileSync(temporary, JSON.stringify(value));
-		renameSync(temporary, file);
-	} catch {
-		try {
-			rmSync(temporary, { force: true });
-		} catch {}
-	}
-}
-//#endregion
 //#region src/devices/swiftpm-cache-usage.ts
-async function getSwiftpmCacheUsage({ maxAgeMs = 6e5 } = {}) {
-	const dir = canonical(process.platform === "darwin" ? join(homedir(), "Library", "Caches", "org.swift.swiftpm") : join(process.env.XDG_CACHE_HOME || join(homedir(), ".cache"), "org.swift.swiftpm"));
+async function getSwiftpmCacheUsage({ maxAgeMs = 6e5, operation } = {}) {
+	const env = operation?.env ?? process.env;
+	const dir = canonical(process.platform === "darwin" ? join(env.HOME || env.USERPROFILE || homedir(), "Library", "Caches", "org.swift.swiftpm") : join(env.XDG_CACHE_HOME || join(env.HOME || env.USERPROFILE || homedir(), ".cache"), "org.swift.swiftpm"));
 	const cached = readSwiftpmCacheUsage();
 	const age = cached ? Date.now() - Date.parse(cached.measuredAt) : Infinity;
 	if (cached && cached.dir === dir && age >= 0 && age < maxAgeMs) return cached;
 	const present = existsSync(dir);
-	const result = present ? await measureDu(["-sk", dir]) : null;
+	const result = present ? await measureDu(["-sk", dir], operation) : null;
 	const bytes = present ? result?.complete ? result.sizes.get(dir) ?? null : null : 0;
 	const usage = {
 		version: 1,
@@ -71,7 +27,7 @@ async function getSwiftpmCacheUsage({ maxAgeMs = 6e5 } = {}) {
 		bytes,
 		complete: bytes !== null
 	};
-	writeUsageCache(swiftpmCacheUsageFile(), usage);
+	writeUsageCache(swiftpmCacheUsageFile(), usage, operation?.signal);
 	return usage;
 }
 //#endregion
@@ -234,18 +190,18 @@ function readRunnerLock(dir) {
 	} catch {}
 	return "unreadable";
 }
-function discover() {
-	const runnerRoot = canonical(join(homedir(), ".agent-device", "apple-runner"));
+function discover(env, cwd) {
+	const runnerRoot = canonical(join(env?.HOME || env?.USERPROFILE || homedir(), ".agent-device", "apple-runner"));
 	const hostedDir = canonical(join(configDir(), "server", "agent-device"));
 	const workspaceDirs = listWorkspaceDirs().map(({ dir, projectRoot }) => ({
 		dir: canonical(workspaceAgentDeviceDir(dir)),
 		projectRoot,
 		bytes: null
 	}));
-	const override = envDir("AGENT_DEVICE_STATE_DIR");
+	const override = envDir("AGENT_DEVICE_STATE_DIR", env, cwd);
 	const candidate = override ? canonical(override) : null;
 	const overlaps = candidate && [...workspaceDirs.map(({ dir }) => dir), hostedDir].some((dir) => contains(dir, candidate) || contains(candidate, dir));
-	const stateDir = candidate && !overlaps ? candidate : canonical(join(homedir(), ".agent-device"));
+	const stateDir = candidate && !overlaps ? candidate : canonical(join(env?.HOME || env?.USERPROFILE || homedir(), ".agent-device"));
 	const workspaces = workspaceDirs.filter(({ dir }) => existsSync(dir));
 	const roots = {
 		stateDir,
@@ -253,7 +209,7 @@ function discover() {
 		hostedDir,
 		workspaceDirs: workspaces.map(({ dir }) => dir).toSorted()
 	};
-	const leaseDir = envDir("AGENT_DEVICE_IOS_RUNNER_LEASE_DIR") ?? join(runnerRoot, "leases");
+	const leaseDir = envDir("AGENT_DEVICE_IOS_RUNNER_LEASE_DIR", env, cwd) ?? join(runnerRoot, "leases");
 	let leases = [];
 	let unreadable = false;
 	try {
@@ -332,15 +288,15 @@ function discover() {
 		absent
 	};
 }
-async function getAgentDeviceUsage({ maxAgeMs = MAX_AGE_MS } = {}) {
-	const tree = discover();
+async function getAgentDeviceUsage({ maxAgeMs = MAX_AGE_MS, operation } = {}) {
+	const tree = discover(operation?.env, operation?.cwd);
 	const cached = readAgentDeviceUsageCache();
 	const age = cached ? Date.now() - Date.parse(cached.usage.measuredAt) : Infinity;
 	if (cached && age >= 0 && age < maxAgeMs && JSON.stringify(cached.roots) === JSON.stringify(tree.roots)) return cached.usage;
 	const sizes = /* @__PURE__ */ new Map();
 	let complete = true;
 	const measure = async (args) => {
-		const result = await measureDu(args);
+		const result = await measureDu(args, operation);
 		complete &&= result.complete;
 		for (const [dir, bytes] of result.sizes) sizes.set(dir, bytes);
 	};
@@ -362,7 +318,7 @@ async function getAgentDeviceUsage({ maxAgeMs = MAX_AGE_MS } = {}) {
 	writeUsageCache(agentDeviceUsageFile(), {
 		...usage,
 		roots: tree.roots
-	});
+	}, operation?.signal);
 	return usage;
 }
 //#endregion