openmausbot 0.1.86 → 0.1.88

package/dist-server/drivers/phone-proxy.js

@@ -5,6 +5,7 @@ import { createInterface } from "node:readline";
 import { delimiter, join } from "node:path";
 import { homedir } from "node:os";
 var MAX_ADB_OUTPUT = 16 * 1024 * 1024;
+var PHONE_BUSY_TEXT = "Another thread is using the phone. This call was not performed. Pause phone work until that thread finishes, then read the screen again before acting.";
 var PACKAGE = /^[A-Za-z0-9_]+(?:\.[A-Za-z0-9_]+)+$/;
 var COMPONENT = /^([A-Za-z0-9_]+(?:\.[A-Za-z0-9_]+)+)\/\S+$/;
 var SAFE_TEXT = /^[A-Za-z0-9 _.,@-]+$/;
@@ -217,6 +218,32 @@ var TOOLS = [
   { name: "press", description: "Press an Android navigation or keyboard key.", inputSchema: { type: "object", properties: { key: { type: "string", enum: Object.keys(KEYCODES) }, serial: { type: "string" } }, required: ["key"] } }
 ];
 var textResult = (text, isError = false) => ({ content: [{ type: "text", text }], ...isError ? { isError: true } : {} });
+function createPhoneClaim(env = process.env, fetchImpl = fetch) {
+  const url = env.OMB_HARNESS_URL?.trim();
+  const token = env.OMB_PHONE_TOKEN?.trim();
+  if (!url || !token) {
+    return async () => ({ ok: true });
+  }
+  return async () => {
+    try {
+      const response = await fetchImpl(new URL("/api/internal/phone/claim", url), {
+        method: "POST",
+        redirect: "error",
+        headers: { authorization: `Bearer ${token}`, "content-type": "application/json" },
+        body: "{}",
+        signal: AbortSignal.timeout(1e4)
+      });
+      if (response.ok) return { ok: true };
+      if (response.status === 409) return { ok: false, message: PHONE_BUSY_TEXT };
+      if (response.status === 401 || response.status === 403) {
+        return { ok: false, message: "This turn no longer has phone access. Start a new turn to use the phone." };
+      }
+      return { ok: false, message: "OpenMausBot could not reserve the phone for this call. This call was not performed; try again, or start a new turn if it keeps failing." };
+    } catch {
+      return { ok: false, message: "OpenMausBot could not be reached to reserve the phone. This call was not performed; try again." };
+    }
+  };
+}
 async function callTool(name, args) {
   if (name === "status") {
     const adb = resolveAdbPath();
@@ -301,7 +328,7 @@ var send = (message) => process.stdout.write(`${JSON.stringify(message)}
 `);
 var ok = (id, result) => send({ jsonrpc: "2.0", id, result });
 var rpcError = (id, code, message) => send({ jsonrpc: "2.0", id, error: { code, message } });
-async function handle(message) {
+async function handle(message, ensureClaim) {
   const id = message.id;
   const method = message.method;
   const params = message.params ?? {};
@@ -312,6 +339,8 @@ async function handle(message) {
   if (method === "tools/call") {
     const name = String(params.name ?? "");
     if (!TOOLS.some((tool) => tool.name === name)) return rpcError(id, -32602, `Unknown tool: ${name}`);
+    const claim = await ensureClaim();
+    if (!claim.ok) return ok(id, textResult(claim.message, true));
     try {
       return ok(id, await callTool(name, params.arguments ?? {}));
     } catch (error) {
@@ -321,16 +350,18 @@ async function handle(message) {
   if (id !== void 0) rpcError(id, -32601, `Method not found: ${String(method)}`);
 }
 if (process.argv[1] && existsSync(process.argv[1]) && /phone-proxy\.(?:ts|js)$/.test(process.argv[1])) {
+  const ensureClaim = createPhoneClaim();
   const lines = createInterface({ input: process.stdin, terminal: false });
   lines.on("line", (line) => {
     try {
       const message = JSON.parse(line);
-      void handle(message).catch((error) => rpcError(message.id, -32603, error instanceof Error ? error.message : String(error)));
+      void handle(message, ensureClaim).catch((error) => rpcError(message.id, -32603, error instanceof Error ? error.message : String(error)));
     } catch {
     }
   });
 }
 export {
+  createPhoneClaim,
   findPackages,
   parseDevices,
   parseLaunchablePackages,