@sendmux/cli 1.4.0 → 1.5.0

package/dist/profiles.js

@@ -1,6 +1,10 @@
-import { chmod, mkdir, readFile, rename, writeFile, } from "node:fs/promises";
+import { chmod, mkdir, open, readFile, rename, stat, unlink, writeFile, } from "node:fs/promises";
+import { createHash } from "node:crypto";
 import { join } from "node:path";
 const CONFIG_FILE = "config.json";
+const CONFIG_LOCK_RETRY_MS = 25;
+const CONFIG_LOCK_STALE_MS = 5_000;
+const CONFIG_LOCK_TIMEOUT_MS = 10_000;
 export async function readCliConfig(configDir) {
     const path = configPath(configDir);
     try {
@@ -18,8 +22,71 @@ export async function readCliConfig(configDir) {
         throw error;
     }
 }
-export async function writeCliConfig(configDir, config) {
+export async function updateCliConfig(configDir, update) {
     await mkdir(configDir, { mode: 0o700, recursive: true });
+    const releaseLock = await acquireConfigWriteLock(configDir);
+    try {
+        const config = await readCliConfig(configDir);
+        const result = update(config);
+        await writeCliConfigFile(configDir, config);
+        return result;
+    }
+    finally {
+        await releaseLock();
+    }
+}
+export async function reserveAgentRegistrationIntent(configDir, profileName, candidate) {
+    await mkdir(configDir, { mode: 0o700, recursive: true });
+    const path = registrationIntentPath(configDir, profileName);
+    const releaseLock = await acquireFileLock(`${path}.lock`, "Timed out waiting for another Sendmux process to reserve the agent registration.");
+    try {
+        while (true) {
+            try {
+                const existing = JSON.parse(await readFile(path, "utf8"));
+                if (isRegisteringAgentProfile(existing)) {
+                    return existing;
+                }
+                await unlink(path);
+            }
+            catch (error) {
+                if (error instanceof SyntaxError) {
+                    await unlink(path).catch((unlinkError) => {
+                        if (!isNodeError(unlinkError) || unlinkError.code !== "ENOENT")
+                            throw unlinkError;
+                    });
+                }
+                else if (!isNodeError(error) || error.code !== "ENOENT") {
+                    throw error;
+                }
+            }
+            try {
+                await writeFile(path, `${JSON.stringify(candidate, null, 2)}\n`, { flag: "wx", mode: 0o600 });
+                await chmod(path, 0o600);
+                return candidate;
+            }
+            catch (error) {
+                if (!isNodeError(error) || error.code !== "EEXIST")
+                    throw error;
+            }
+        }
+    }
+    finally {
+        await releaseLock();
+    }
+}
+export async function clearAgentRegistrationIntent(configDir, profileName) {
+    try {
+        await unlink(registrationIntentPath(configDir, profileName));
+    }
+    catch (error) {
+        if (!isNodeError(error) || error.code !== "ENOENT")
+            throw error;
+    }
+}
+export function configPath(configDir) {
+    return join(configDir, CONFIG_FILE);
+}
+async function writeCliConfigFile(configDir, config) {
     const path = configPath(configDir);
     const tempPath = `${path}.${process.pid}.tmp`;
     await writeFile(tempPath, `${JSON.stringify(config, null, 2)}\n`, { mode: 0o600 });
@@ -27,8 +94,84 @@ export async function writeCliConfig(configDir, config) {
     await rename(tempPath, path);
     await chmod(path, 0o600);
 }
-export function configPath(configDir) {
-    return join(configDir, CONFIG_FILE);
+async function acquireConfigWriteLock(configDir) {
+    return acquireFileLock(`${configPath(configDir)}.lock`, "Timed out waiting for another Sendmux process to finish updating the profile config.");
+}
+async function acquireFileLock(lockPath, timeoutMessage) {
+    const deadline = Date.now() + CONFIG_LOCK_TIMEOUT_MS;
+    while (true) {
+        try {
+            const handle = await open(lockPath, "wx", 0o600);
+            return async () => {
+                try {
+                    await handle.close();
+                }
+                finally {
+                    try {
+                        await unlink(lockPath);
+                    }
+                    catch (error) {
+                        if (!isNodeError(error) || error.code !== "ENOENT")
+                            throw error;
+                    }
+                }
+            };
+        }
+        catch (error) {
+            if (!isNodeError(error) || error.code !== "EEXIST")
+                throw error;
+        }
+        try {
+            const lock = await stat(lockPath);
+            if (Date.now() - lock.mtimeMs >= CONFIG_LOCK_STALE_MS) {
+                try {
+                    await unlink(lockPath);
+                    continue;
+                }
+                catch (error) {
+                    if (isNodeError(error) && error.code === "ENOENT")
+                        continue;
+                    if (!isNodeError(error) || (error.code !== "EACCES" && error.code !== "EPERM"))
+                        throw error;
+                }
+            }
+        }
+        catch (error) {
+            if (isNodeError(error) && error.code === "ENOENT")
+                continue;
+            throw error;
+        }
+        if (Date.now() >= deadline) {
+            throw new Error(timeoutMessage);
+        }
+        await new Promise((resolve) => setTimeout(resolve, CONFIG_LOCK_RETRY_MS));
+    }
+}
+function registrationIntentPath(configDir, profileName) {
+    const profileHash = createHash("sha256").update(profileName, "utf8").digest("hex");
+    return join(configDir, `agent-registration-${profileHash}.json`);
+}
+function isRegisteringAgentProfile(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const profile = value;
+    return (profile.type === "agent" &&
+        profile.state === "registering" &&
+        typeof profile.appApiBaseUrl === "string" &&
+        typeof profile.authBaseUrl === "string" &&
+        typeof profile.idempotencyKey === "string" &&
+        typeof profile.sendingApiBaseUrl === "string" &&
+        (profile.clientName === undefined || typeof profile.clientName === "string") &&
+        (profile.mailboxLocalPart === undefined || typeof profile.mailboxLocalPart === "string"));
+}
+export function isAgentProfile(profile) {
+    return profile.type === "agent";
+}
+export function isApiKeyProfile(profile) {
+    return !isAgentProfile(profile);
+}
+export function isActiveAgentProfile(profile) {
+    return isAgentProfile(profile) && profile.state === "active";
 }
 function isNodeError(error) {
     return Boolean(error && typeof error === "object" && "code" in error);