@pmate/cli 0.29.6 → 0.29.8

package/dist/index.d.ts

@@ -12,7 +12,6 @@ declare enum Command {
     Delete = "delete",
     Dev = "dev",
     Env = "env",
-    Pr = "pr",
     Commit = "commit",
     Summary = "summary",
     Linear = "linear",
@@ -124,11 +123,6 @@ type EnvCommandArgs = {
     envVerbose?: boolean;
     envCwd: string;
 };
-type PrCommandArgs = {
-    type: Command.Pr;
-    action: "create" | "merge";
-    prRef?: string;
-};
 type CommitCommandArgs = {
     type: Command.Commit;
     push: boolean;
@@ -743,6 +737,8 @@ type CallCommandArgs = {
     name: string;
     payload?: Record<string, unknown>;
     filePath?: string;
+    binaryFilePath?: string;
+    contentType?: string;
     hubBaseUrl?: string;
     timeoutMs?: number;
     json: boolean;
@@ -1129,7 +1125,7 @@ type VercelDeploysCommandArgs = {
     lastWeek: boolean;
 };
 type VercelCommandArgs = VercelProjectsCommandArgs | VercelDeploysCommandArgs;
-type CommandArgs = NoneCommandArgs | ListCommandArgs | LoginCommandArgs | AddDnsCommandArgs | StsCommandArgs | BucketCommandArgs | DeployCommandArgs | LogCommandArgs | InitCommandArgs | DeleteCommandArgs | DevCommandArgs | EnvCommandArgs | PrCommandArgs | CommitCommandArgs | SummaryCommandArgs | QuizCommandArgs | ServiceCommandArgs | LinearCommandArgs | NotionCommandArgs | OssCommandArgs | QrcodeCommandArgs | RandomCommandArgs | CollectCommandArgs | ChatRoomCommandArgs | FunctionCommandArgs | WorkerCommandArgs | CallCommandArgs | NsCommandArgs | RoleCommandArgs | AccountCommandArgs | OrgCommandArgs | StoreCommandArgs | AppCommandArgs | VercelCommandArgs | HostCommandArgs;
+type CommandArgs = NoneCommandArgs | ListCommandArgs | LoginCommandArgs | AddDnsCommandArgs | StsCommandArgs | BucketCommandArgs | DeployCommandArgs | LogCommandArgs | InitCommandArgs | DeleteCommandArgs | DevCommandArgs | EnvCommandArgs | CommitCommandArgs | SummaryCommandArgs | QuizCommandArgs | ServiceCommandArgs | LinearCommandArgs | NotionCommandArgs | OssCommandArgs | QrcodeCommandArgs | RandomCommandArgs | CollectCommandArgs | ChatRoomCommandArgs | FunctionCommandArgs | WorkerCommandArgs | CallCommandArgs | NsCommandArgs | RoleCommandArgs | AccountCommandArgs | OrgCommandArgs | StoreCommandArgs | AppCommandArgs | VercelCommandArgs | HostCommandArgs;
 
 interface ParsedArgs {
     help: boolean;

package/dist/index.js

@@ -8249,26 +8249,51 @@ var init_src3 = __esm({
   }
 });
 
-// node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/AgentClient.js
+// node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/AgentClient.js
 var init_AgentClient = __esm({
-  "node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/AgentClient.js"() {
+  "node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/AgentClient.js"() {
     "use strict";
     init_src();
     init_src3();
   }
 });
 
-// node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/FunctionService.js
+// node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/FunctionService.js
 var init_FunctionService = __esm({
-  "node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/FunctionService.js"() {
+  "node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/FunctionService.js"() {
     "use strict";
   }
 });
 
-// node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/HubClient.js
+// node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/HubClient.js
 function createCallId(functionId) {
   return uniqHash(`${functionId}-${Date.now()}-${Math.random()}`, "task-v1" /* Task */);
 }
+function defaultTransportFactory(url) {
+  return new WebsocketClientV2(url);
+}
+function streamContentType(value) {
+  if (typeof Blob !== "undefined" && value instanceof Blob) {
+    return value.type || "application/octet-stream";
+  }
+  if (value instanceof Uint8Array || value instanceof ArrayBuffer) {
+    return "application/octet-stream";
+  }
+  if (typeof value === "string")
+    return "text";
+  return "application/json";
+}
+async function streamInputForWire(value) {
+  if (typeof Blob !== "undefined" && value instanceof Blob) {
+    return new Uint8Array(await value.arrayBuffer());
+  }
+  return value;
+}
+function abortError(reason) {
+  if (reason instanceof Error)
+    return reason;
+  return new Error(typeof reason === "string" ? reason : "Function stream aborted");
+}
 function ensureWsProtocol(baseUrl) {
   if (/^wss?:\/\//i.test(baseUrl))
     return baseUrl;
@@ -8299,13 +8324,14 @@ function withTimeout(promise, timeoutMs, message) {
     });
   });
 }
-var DEFAULT_TIMEOUT_MS, HubClient, AsyncCallQueue;
+var DEFAULT_TIMEOUT_MS, DEFAULT_STREAM_TIMEOUT_MS, HubClient, AsyncCallQueue;
 var init_HubClient = __esm({
-  "node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/HubClient.js"() {
+  "node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/HubClient.js"() {
     "use strict";
     init_src();
     init_src3();
     DEFAULT_TIMEOUT_MS = 4e4;
+    DEFAULT_STREAM_TIMEOUT_MS = 12e4;
     HubClient = class {
       options;
       ws;
@@ -8313,7 +8339,10 @@ var init_HubClient = __esm({
       token;
       authenticated = false;
       calls = /* @__PURE__ */ new Map();
+      streamInputs = /* @__PURE__ */ new Map();
       timeoutMs;
+      streamTimeoutMs;
+      minChunkSizeBytes;
       connectWaiter;
       authWaiter;
       reconnectEnabled = false;
@@ -8321,6 +8350,8 @@ var init_HubClient = __esm({
       constructor(options) {
         this.options = options;
         this.timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+        this.streamTimeoutMs = options.streamTimeoutMs ?? DEFAULT_STREAM_TIMEOUT_MS;
+        this.minChunkSizeBytes = options.minChunkSizeBytes ?? 0;
       }
       isConnected() {
         return this.authenticated;
@@ -8331,7 +8362,7 @@ var init_HubClient = __esm({
         this.token = options.token;
         this.connectWaiter = deferred();
         this.authWaiter = deferred();
-        const ws = new WebsocketClientV2(ensureWsProtocol(this.options.baseUrl));
+        const ws = (this.options.transportFactory ?? defaultTransportFactory)(ensureWsProtocol(this.options.baseUrl));
         this.ws = ws;
         const stream = ws.getStream();
         void this.consume(stream);
@@ -8392,6 +8423,39 @@ var init_HubClient = __esm({
           this.cancelPendingCall(callId);
         }
       }
+      stream(request2) {
+        this.ensureAuthenticated();
+        const callId = createCallId(request2.functionId);
+        const timeoutMs = request2.timeoutMs ?? this.streamTimeoutMs;
+        const result = new Promise((resolve, reject) => {
+          const timer = setTimeout(() => {
+            void this.abortFunctionStream(callId, new Error(`Function stream timed out: ${request2.functionId}`));
+          }, timeoutMs);
+          this.calls.set(callId, {
+            resolve,
+            reject,
+            timer,
+            abort: (error) => this.sendFunctionStreamCancel(callId, request2.functionId, error)
+          });
+        });
+        void result.catch(() => {
+        });
+        const onAbort = () => {
+          void this.abortFunctionStream(callId, abortError(request2.signal?.reason));
+        };
+        request2.signal?.addEventListener("abort", onAbort, { once: true });
+        if (request2.signal?.aborted)
+          onAbort();
+        void this.sendFunctionStream(callId, request2).catch((error) => {
+          void this.abortFunctionStream(callId, error instanceof Error ? error : new Error(String(error)));
+        });
+        void result.finally(() => request2.signal?.removeEventListener("abort", onAbort)).catch(() => {
+        });
+        return {
+          finish: () => result,
+          cancel: (reason) => this.abortFunctionStream(callId, new Error(reason ?? "Function stream cancelled"))
+        };
+      }
       close() {
         this.reconnectEnabled = false;
         this.reauthenticating = void 0;
@@ -8484,7 +8548,13 @@ var init_HubClient = __esm({
           return;
         clearTimeout(call.timer);
         this.calls.delete(callId);
-        call.reject(error);
+        const input = this.streamInputs.get(callId);
+        this.streamInputs.delete(callId);
+        void input?.return?.();
+        if (call.abort)
+          void call.abort(error).then(() => call.reject(error), () => call.reject(error));
+        else
+          call.reject(error);
       }
       handleConnectionLoss(error) {
         this.authenticated = false;
@@ -8498,6 +8568,72 @@ var init_HubClient = __esm({
           profileId
         }, 0 /* PIPELINE */));
       }
+      async sendFunctionStream(callId, request2) {
+        this.ensureAuthenticated();
+        const ws = this.ws;
+        const profileId = this.profileId;
+        const auth = { token: this.token, profileId };
+        const send = (op, contentType, input2) => ws.sendMsg(Msg.create(profileId, "@hub", 6 /* CALL */, {
+          kind: "request",
+          callId,
+          functionId: request2.functionId,
+          op,
+          auth,
+          contentType,
+          ...input2 !== void 0 ? { input: input2 } : {}
+        }, 0 /* PIPELINE */));
+        await send(0 /* Start */, "application/json", request2.params);
+        const minChunkSizeBytes = request2.minChunkSizeBytes ?? this.minChunkSizeBytes;
+        const input = minChunkSizeBytes > 0 ? AsyncStream.chunkBlob(request2.stream, { minBytes: minChunkSizeBytes }) : request2.stream;
+        const iterator = input[Symbol.asyncIterator]();
+        this.streamInputs.set(callId, iterator);
+        try {
+          while (this.calls.has(callId)) {
+            const next = await iterator.next();
+            if (next.done)
+              break;
+            await send(1 /* Data */, streamContentType(next.value), await streamInputForWire(next.value));
+          }
+          if (!this.calls.has(callId))
+            return;
+          await send(2 /* End */, "application/json");
+        } finally {
+          this.streamInputs.delete(callId);
+        }
+      }
+      async abortFunctionStream(callId, error) {
+        const call = this.calls.get(callId);
+        if (!call)
+          return;
+        clearTimeout(call.timer);
+        this.calls.delete(callId);
+        const input = this.streamInputs.get(callId);
+        this.streamInputs.delete(callId);
+        await input?.return?.();
+        try {
+          await call.abort?.(error);
+        } finally {
+          call.reject(error);
+        }
+      }
+      async sendFunctionStreamCancel(callId, functionId, error) {
+        if (!this.ws || !this.profileId || !this.token || !this.authenticated)
+          return;
+        const body = {
+          kind: "request",
+          callId,
+          functionId,
+          op: 3 /* Cancel */,
+          input: { reason: error.message },
+          auth: { token: this.token, profileId: this.profileId }
+        };
+        await this.ws.sendMsg(Msg.create(this.profileId, "@hub", 6 /* CALL */, body, 0 /* PIPELINE */));
+      }
+      ensureAuthenticated() {
+        if (!this.ws || !this.profileId || !this.token || !this.authenticated) {
+          throw new Error("HubClient is not logged in");
+        }
+      }
       reauthenticate() {
         if (this.reauthenticating)
           return this.reauthenticating;
@@ -8610,9 +8746,9 @@ var init_HubClient = __esm({
   }
 });
 
-// node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/index.js
+// node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/index.js
 var init_dist2 = __esm({
-  "node_modules/.pnpm/@pmate+hub-sdk@0.3.2/node_modules/@pmate/hub-sdk/dist/index.js"() {
+  "node_modules/.pnpm/@pmate+hub-sdk@0.4.1/node_modules/@pmate/hub-sdk/dist/index.js"() {
     "use strict";
     init_AgentClient();
     init_FunctionService();
@@ -11376,54 +11512,6 @@ var PrCommandHandler = class {
   constructor(cwd = process.cwd()) {
     this.cwd = cwd;
   }
-  async handle(args) {
-    if (args.action === "merge") {
-      await this.handleMerge(args.prRef);
-      return;
-    }
-    await this.ensureGitRepo();
-    await this.ensureGhCli();
-    await this.ensureGhAuth();
-    const context = await this.getRepoContext();
-    let existingPrUrl = await this.findExistingPrUrl(context.currentBranch);
-    if (!this.hasLocalChanges(context)) {
-      await this.handleNoLocalChanges(context, existingPrUrl);
-      return;
-    }
-    const recentLog = await this.runCapture("git", [
-      "log",
-      "--oneline",
-      "-n",
-      "20"
-    ]);
-    console.log("Generating PR metadata using a PMate Function...");
-    const suggestion = await this.getAiSuggestion({
-      status: context.status,
-      diff: truncate(context.diff, 6e3),
-      diffCached: truncate(context.diffCached, 6e3),
-      recentLog: truncate(recentLog, 2e3),
-      baseBranch: context.baseBranch
-    });
-    this.assertNonEmpty("PR title", suggestion.prTitle);
-    this.assertNonEmpty("Commit message", suggestion.commitMessage);
-    const prTitle = normalizePrTitle(suggestion.prTitle);
-    if (prTitle !== suggestion.prTitle.trim()) {
-      ui.info("Normalized PR title", prTitle);
-    }
-    const headBranch = await this.ensureWorkBranch(context, suggestion);
-    if (headBranch !== context.currentBranch) {
-      existingPrUrl = await this.findExistingPrUrl(headBranch);
-    }
-    await this.commitIfNeeded(suggestion.commitMessage);
-    await this.rebaseOntoBase(context.baseBranch);
-    await this.pushBranchIfNeeded(headBranch, context.baseBranch);
-    await this.ensurePr(
-      existingPrUrl,
-      context.baseBranch,
-      headBranch,
-      prTitle
-    );
-  }
   async handleCommit(push = false) {
     await this.ensureGitRepo();
     const context = await this.getRepoContext();
@@ -16742,11 +16830,8 @@ var CallCommandHandler = class {
     const input = args.filePath ? readJson2(args.filePath) : args.payload;
     try {
       await hub.login({ token: session.token, profileId: session.profileId, timeoutMs: args.timeoutMs });
-      const output = await hub.call({
-        functionId: `${args.namespace}:${args.name}`,
-        input,
-        timeoutMs: args.timeoutMs
-      });
+      const functionId = `${args.namespace}:${args.name}`;
+      const output = args.binaryFilePath ? await callWithBinaryFile(hub, functionId, args.binaryFilePath, args.contentType, args.timeoutMs) : await hub.call({ functionId, input, timeoutMs: args.timeoutMs });
       if (args.json || typeof output !== "string") ui.plain(JSON.stringify(output ?? null, null, 2));
       else ui.plain(output);
     } finally {
@@ -16754,6 +16839,35 @@ var CallCommandHandler = class {
     }
   }
 };
+async function callWithBinaryFile(hub, functionId, filePath, contentType, timeoutMs) {
+  const absolutePath = import_node_path20.default.resolve(process.cwd(), filePath);
+  if (!(0, import_node_fs15.existsSync)(absolutePath)) throw new Error(`Binary file not found: ${absolutePath}`);
+  const mimeType = contentType || inferContentType(absolutePath);
+  const result = hub.stream({
+    functionId,
+    stream: streamFile(absolutePath, mimeType),
+    timeoutMs
+  });
+  return result.finish();
+}
+async function* streamFile(filePath, contentType) {
+  for await (const chunk of (0, import_node_fs15.createReadStream)(filePath)) {
+    yield new Blob([chunk], { type: contentType });
+  }
+}
+function inferContentType(filePath) {
+  const contentTypes = {
+    ".aac": "audio/aac",
+    ".flac": "audio/flac",
+    ".m4a": "audio/mp4",
+    ".mp3": "audio/mpeg",
+    ".ogg": "audio/ogg",
+    ".opus": "audio/opus",
+    ".wav": "audio/wav",
+    ".webm": "audio/webm"
+  };
+  return contentTypes[import_node_path20.default.extname(filePath).toLowerCase()] || "application/octet-stream";
+}
 function readJson2(filePath) {
   const absolutePath = import_node_path20.default.resolve(process.cwd(), filePath);
   if (!(0, import_node_fs15.existsSync)(absolutePath)) throw new Error(`JSON file not found: ${absolutePath}`);
@@ -22739,10 +22853,12 @@ init_namespacedRef();
 function parseCallArgs(tokens) {
   const rest = [...tokens];
   const ref = rest.shift();
-  if (!ref) throw new Error("Usage: call <namespace:functionId> [--payload <json>|--file <path>]");
+  if (!ref) throw new Error("Usage: call <namespace:functionId> [--payload <json>|--file <path>|--binary-file <path>]");
   const parsed = parseNamespacedRef(ref, { entityLabel: "function ref" });
   let payload;
   let filePath;
+  let binaryFilePath;
+  let contentType;
   let hubBaseUrl;
   let timeoutMs;
   let json = false;
@@ -22760,12 +22876,17 @@ function parseCallArgs(tokens) {
       }
       payload = decoded;
     } else if (option === "--file") filePath = value();
+    else if (option === "--binary-file") binaryFilePath = value();
+    else if (option === "--content-type") contentType = value();
     else if (option === "--hub-base-url") hubBaseUrl = value();
     else if (option === "--timeout-ms") timeoutMs = Number.parseInt(value(), 10);
     else if (option === "--json") json = true;
     else throw new Error(`Unknown call option: ${option}`);
   }
-  if (payload && filePath) throw new Error("Provide only one of --payload or --file");
+  if ([payload, filePath, binaryFilePath].filter(Boolean).length > 1) {
+    throw new Error("Provide only one of --payload, --file or --binary-file");
+  }
+  if (contentType && !binaryFilePath) throw new Error("--content-type requires --binary-file");
   if (timeoutMs !== void 0 && (!Number.isInteger(timeoutMs) || timeoutMs < 1)) {
     throw new Error("--timeout-ms must be a positive integer");
   }
@@ -22775,6 +22896,8 @@ function parseCallArgs(tokens) {
     name: parsed.objectId,
     payload,
     filePath,
+    binaryFilePath,
+    contentType,
     hubBaseUrl,
     timeoutMs,
     json
@@ -25785,8 +25908,6 @@ var HELP_TEXT = [
   `  ${import_chalk4.default.gray("                  <app-name> [--cwd <dir>]")}`,
   `  ${import_chalk4.default.cyan("env")}              Manage app env vars on the server`,
   `  ${import_chalk4.default.gray("                  <app-name> --list|--set <key> --value <value> [--delete <key>] [--cwd <dir>] [--verbose|-v]")}`,
-  `  ${import_chalk4.default.cyan("pr")}               Create or update GitHub PRs with AI assistance`,
-  `  ${import_chalk4.default.gray("                  merge [<pr>] (defaults: --squash --delete-branch)")}`,
   `  ${import_chalk4.default.cyan("commit")}           Create an AI-generated commit from local changes`,
   `  ${import_chalk4.default.gray("                  [--push] [--cwd <dir>]")}`,
   `  ${import_chalk4.default.cyan("summary")}          Summarize a GitHub PR with AI`,
@@ -26060,9 +26181,6 @@ function parseCommandArgs(args) {
   if (args.rest[0] === "env") {
     return parseEnvArgs(args.rest.slice(1));
   }
-  if (args.rest[0] === "pr") {
-    return parsePrArgs(args.rest.slice(1));
-  }
   if (args.rest[0] === "commit") {
     return parseCommitArgs(args.rest.slice(1));
   }
@@ -26426,29 +26544,6 @@ function parseEnvArgs(tokens) {
     envCwd: cwd
   };
 }
-function parsePrArgs(tokens) {
-  const rest = [...tokens];
-  if (rest[0] === "merge") {
-    rest.shift();
-    let prRef;
-    while (rest.length > 0) {
-      const token = rest.shift();
-      if (token.startsWith("-")) {
-        throw new Error(`Unknown pr merge option: ${token}`);
-      }
-      if (prRef) {
-        throw new Error(`Unexpected extra argument for pr merge: ${token}`);
-      }
-      prRef = token;
-    }
-    return { type: "pr" /* Pr */, action: "merge", prRef };
-  }
-  while (rest.length > 0) {
-    const token = rest.shift();
-    throw new Error(`Unknown pr option: ${token}`);
-  }
-  return { type: "pr" /* Pr */, action: "create" };
-}
 function parseCommitArgs(tokens) {
   const rest = [...tokens];
   let push = false;
@@ -27746,11 +27841,6 @@ async function main() {
         await handler.handle(commandArgs);
         return;
       }
-      case "pr" /* Pr */: {
-        const handler = new PrCommandHandler();
-        await handler.handle(commandArgs);
-        return;
-      }
       case "commit" /* Commit */: {
         const handler = new CommitCommandHandler();
         await handler.handle(commandArgs);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pmate/cli",
-  "version": "0.29.6",
+  "version": "0.29.8",
   "description": "CLI utilities for the pMate project",
   "license": "MIT",
   "bin": {
@@ -31,7 +31,7 @@
     "@alicloud/sts20150401": "^1.2.0",
     "@linear/sdk": "^70.0.0",
     "@pmate/chat-sdk": "^0.2.0",
-    "@pmate/hub-sdk": "^0.3.2",
+    "@pmate/hub-sdk": "^0.4.1",
     "@types/ali-oss": "^6.16.13",
     "ajv": "^8.17.1",
     "ali-oss": "^6.23.0",