openmausbot 0.1.55

package/dist-server/local-computer.js

@@ -0,0 +1,198 @@
+// server/local-computer.ts
+import { lstatSync, readFileSync, realpathSync, statSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, isAbsolute, join } from "node:path";
+var REQUIRED_LINUX_TOOLS = ["click", "get_window_state", "list_apps", "type_text"];
+var DRIVER_FILE_IDENTITY_KEYS = [
+  "dev",
+  "ino",
+  "uid",
+  "gid",
+  "mode",
+  "size",
+  "mtimeNs",
+  "ctimeNs"
+];
+function exactKeys(value, keys) {
+  const expected = new Set(keys);
+  return Object.keys(value).length === expected.size && Object.keys(value).every((key) => expected.has(key));
+}
+function legacyPlatform(platform) {
+  if (platform === "darwin" || platform === "win32") return platform;
+  return null;
+}
+function validDriverFileIdentity(value) {
+  return Boolean(value) && typeof value === "object" && !Array.isArray(value) && exactKeys(value, DRIVER_FILE_IDENTITY_KEYS) && DRIVER_FILE_IDENTITY_KEYS.every(
+    (key) => typeof value[key] === "string" && /^\d+$/.test(value[key])
+  );
+}
+function currentDriverFileIdentity(file) {
+  const stat = statSync(file, { bigint: true });
+  return {
+    dev: String(stat.dev),
+    ino: String(stat.ino),
+    uid: String(stat.uid),
+    gid: String(stat.gid),
+    mode: String(stat.mode),
+    size: String(stat.size),
+    mtimeNs: String(stat.mtimeNs),
+    ctimeNs: String(stat.ctimeNs)
+  };
+}
+function sameDriverFileIdentity(expected, actual) {
+  return validDriverFileIdentity(expected) && DRIVER_FILE_IDENTITY_KEYS.every((key) => expected[key] === actual[key]);
+}
+function decodeLegacyDescriptor(value, platform) {
+  const supportedPlatform = legacyPlatform(platform);
+  if (!supportedPlatform || !value || value.mode === "unavailable" || typeof value.mcpCommand !== "string") {
+    return null;
+  }
+  if (value.mcpArgs !== void 0 && !Array.isArray(value.mcpArgs)) return null;
+  if (value.mcpEnv !== void 0 && (!value.mcpEnv || typeof value.mcpEnv !== "object" || Array.isArray(value.mcpEnv))) {
+    return null;
+  }
+  const args = value.mcpArgs ?? ["mcp"];
+  if (!args.every((arg) => typeof arg === "string")) return null;
+  const env = value.mcpEnv ?? {};
+  if (!Object.values(env).every((entry) => typeof entry === "string")) return null;
+  return {
+    command: value.mcpCommand,
+    args,
+    env,
+    platform: supportedPlatform,
+    scope: "local-computer"
+  };
+}
+function decodeLinuxDescriptor(value) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) return null;
+  const x11 = value.mode === "linux-x11-supervised" && value.session === "x11";
+  const wayland = value.mode === "linux-wayland-gnome-supervised" && value.session === "wayland" && value.compositor === "gnome-mutter";
+  const descriptorKeys = [
+    "schemaVersion",
+    "mode",
+    "platform",
+    "session",
+    "enabled",
+    "status",
+    "ownerPid",
+    "generation",
+    "driver",
+    "daemon",
+    "mcp",
+    "toolNames",
+    "doctorWarnings",
+    ...wayland ? ["compositor"] : []
+  ];
+  if (!x11 && !wayland || !exactKeys(value, descriptorKeys) || value.schemaVersion !== 1 || value.platform !== "linux" || value.enabled !== true || value.status !== "ready" || !Number.isInteger(value.ownerPid) || value.ownerPid <= 0 || typeof value.generation !== "string" || !/^[0-9a-f-]{32,64}$/i.test(value.generation)) {
+    return null;
+  }
+  const driver = value.driver;
+  const daemon = value.daemon;
+  const mcp = value.mcp;
+  if (!driver || !daemon || !mcp || Array.isArray(driver) || Array.isArray(daemon) || Array.isArray(mcp) || !exactKeys(driver, ["path", "version", "source", "manifestSchema", "fileIdentity"]) || !exactKeys(daemon, [
+    "socketPath",
+    "pid",
+    "contractVersion",
+    "toolsListSchemaVersion",
+    "capabilityVersion",
+    "mcpProtocolVersion"
+  ]) || !exactKeys(mcp, ["command", "args", "env"])) {
+    return null;
+  }
+  if (typeof driver.path !== "string" || !isAbsolute(driver.path) || driver.version !== "0.19.3" || !["bundled", "environment", "user-local", "path"].includes(String(driver.source)) || driver.manifestSchema !== "1" || !validDriverFileIdentity(driver.fileIdentity) || typeof daemon.socketPath !== "string" || !isAbsolute(daemon.socketPath) || !Number.isInteger(daemon.pid) || daemon.pid <= 0 || daemon.contractVersion !== "0.6.0" || daemon.toolsListSchemaVersion !== "1" || daemon.capabilityVersion !== "1" || daemon.mcpProtocolVersion !== "2025-06-18" || mcp.command !== driver.path || !Array.isArray(mcp.args) || mcp.args.length !== 4 || mcp.args[0] !== "mcp" || mcp.args[1] !== "--embedded" || mcp.args[2] !== "--socket" || mcp.args[3] !== daemon.socketPath || !mcp.env || typeof mcp.env !== "object" || Array.isArray(mcp.env) || !exactKeys(mcp.env, [
+    "CUA_DRIVER_EMBEDDED",
+    "CUA_DRIVER_HOST_BUNDLE_ID",
+    "CUA_DRIVER_RS_UPDATE_CHECK",
+    "CUA_DRIVER_RS_TELEMETRY_ENABLED",
+    ...wayland ? ["CUA_DRIVER_RS_ENABLE_WAYLAND"] : []
+  ]) || mcp.env.CUA_DRIVER_EMBEDDED !== "1" || mcp.env.CUA_DRIVER_HOST_BUNDLE_ID !== "com.openmausbot.app" || mcp.env.CUA_DRIVER_RS_UPDATE_CHECK !== "false" || mcp.env.CUA_DRIVER_RS_TELEMETRY_ENABLED !== "false" || wayland && mcp.env.CUA_DRIVER_RS_ENABLE_WAYLAND !== "1") {
+    return null;
+  }
+  if (!Array.isArray(value.toolNames) || value.toolNames.some((name) => typeof name !== "string") || REQUIRED_LINUX_TOOLS.some((name) => !value.toolNames.includes(name)) || !Array.isArray(value.doctorWarnings)) {
+    return null;
+  }
+  for (const warning of value.doctorWarnings) {
+    if (!warning || typeof warning !== "object" || Array.isArray(warning) || ![3, 4].includes(Object.keys(warning).length) || !Object.keys(warning).every((key) => ["label", "status", "message", "detail"].includes(key)) || typeof warning.label !== "string" || warning.status !== "warn" || typeof warning.message !== "string" || warning.detail !== void 0 && typeof warning.detail !== "string") {
+      return null;
+    }
+  }
+  return {
+    command: driver.path,
+    args: [...mcp.args],
+    env: { ...mcp.env },
+    platform: "linux",
+    generation: value.generation,
+    scope: "local-computer"
+  };
+}
+function processAlive(pid) {
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch {
+    return false;
+  }
+}
+function ownedPrivate(stat, uid) {
+  return (stat.uid === uid || stat.uid === 0) && (stat.mode & 63) === 0;
+}
+function validateLinuxDescriptorRuntime(descriptorFile, raw, {
+  uid = process.getuid?.() ?? -1,
+  isProcessAlive = processAlive
+} = {}) {
+  try {
+    const descriptorStat = lstatSync(descriptorFile);
+    const descriptorDirectoryStat = lstatSync(dirname(descriptorFile));
+    if (!descriptorStat.isFile() || descriptorStat.isSymbolicLink() || !ownedPrivate(descriptorStat, uid) || !descriptorDirectoryStat.isDirectory() || descriptorDirectoryStat.isSymbolicLink() || !ownedPrivate(descriptorDirectoryStat, uid)) {
+      return false;
+    }
+    const driver = raw.driver;
+    const daemon = raw.daemon;
+    const binaryPath = driver.path;
+    const socketPath = daemon.socketPath;
+    const binaryStat = statSync(binaryPath);
+    const currentFileIdentity = currentDriverFileIdentity(binaryPath);
+    const socketStat = lstatSync(socketPath);
+    const socketDirectoryStat = lstatSync(dirname(socketPath));
+    if (realpathSync(binaryPath) !== binaryPath || !sameDriverFileIdentity(driver.fileIdentity, currentFileIdentity) || !binaryStat.isFile() || binaryStat.uid !== uid && binaryStat.uid !== 0 || (binaryStat.mode & 73) === 0 || (binaryStat.mode & 18) !== 0 || !socketStat.isSocket() || socketStat.isSymbolicLink() || !ownedPrivate(socketStat, uid) || !socketDirectoryStat.isDirectory() || socketDirectoryStat.isSymbolicLink() || !ownedPrivate(socketDirectoryStat, uid) || !isProcessAlive(raw.ownerPid) || !isProcessAlive(daemon.pid)) {
+      return false;
+    }
+    return true;
+  } catch {
+    return false;
+  }
+}
+function readCuaConnection({
+  platform = process.platform,
+  userData = process.env.OMB_USER_DATA,
+  home = homedir(),
+  validateLinuxRuntime = validateLinuxDescriptorRuntime
+} = {}) {
+  const candidates = userData ? [join(userData, "cua-connection.json")] : [];
+  if (platform === "darwin") {
+    for (const directory of ["OpenMausBot", "openmausbot", "OpenGrokBot", "opengrokbot"]) {
+      candidates.push(join(home, "Library", "Application Support", directory, "cua-connection.json"));
+    }
+  }
+  for (const file of [...new Set(candidates)]) {
+    try {
+      const raw = JSON.parse(readFileSync(file, "utf8"));
+      if (platform === "linux") {
+        const decoded = decodeLinuxDescriptor(raw);
+        if (decoded && validateLinuxRuntime(file, raw)) return decoded;
+      } else {
+        const decoded = decodeLegacyDescriptor(raw, platform);
+        if (decoded) return decoded;
+      }
+    } catch {
+    }
+  }
+  return null;
+}
+export {
+  DRIVER_FILE_IDENTITY_KEYS,
+  REQUIRED_LINUX_TOOLS,
+  decodeLinuxDescriptor,
+  readCuaConnection,
+  validateLinuxDescriptorRuntime
+};