maoda-commander-tt 0.0.17 → 0.0.18

package/dist/main.js

@@ -3662,6 +3662,132 @@ function inferVideoExtension(remoteUrl, response) {
   return "mp4";
 }
 
+// src/commands/ai/ai-querytask-command.ts
+import { spawn as spawn3 } from "child_process";
+var BIN = "xx";
+var PPE_ENV = "ppe_new_login_cli";
+var POLL_INTERVAL_MS2 = 5e3;
+var POLL_TIMEOUT_MS2 = 10 * 60 * 1e3;
+var RUNNING_STATUS = "running";
+var AiQueryTaskCommand = class extends AiBaseCommand {
+  _meta() {
+    return {
+      name: "querytask",
+      description: "\u8F6E\u8BE2 xx query-video-direct \u4EFB\u52A1\u7ED3\u679C\uFF0Crunning \u65F6\u81EA\u52A8\u91CD\u8BD5\u76F4\u81F3\u5B8C\u6210\u6216\u8D85\u65F6"
+    };
+  }
+  _configureArguments(cmd) {
+    cmd.argument("<task_id>", "\u8981\u67E5\u8BE2\u7684\u4EFB\u52A1 ID");
+  }
+  async _execute(ctx) {
+    const taskId = String(ctx.args[0] ?? "").trim();
+    if (!taskId) {
+      return this._makeError(1, "\u8BF7\u4F20\u5165 task_id");
+    }
+    const startedAt = Date.now();
+    while (true) {
+      const queryResult = await this._runQuery(taskId);
+      if (!queryResult.ok) {
+        return this._makeError(queryResult.code, queryResult.msg);
+      }
+      const { stdout, stderr } = queryResult.value;
+      const status = extractStatus(stdout);
+      if (status !== RUNNING_STATUS) {
+        if (stderr.trim().length > 0) {
+          process.stderr.write(stderr);
+        }
+        process.stdout.write(stdout.endsWith("\n") ? stdout : `${stdout}
+`);
+        return this._makeOk();
+      }
+      const elapsed = Date.now() - startedAt;
+      if (elapsed + POLL_INTERVAL_MS2 >= POLL_TIMEOUT_MS2) {
+        return this._makeError(
+          1,
+          `\u8F6E\u8BE2\u8D85\u65F6 ${Math.round(POLL_TIMEOUT_MS2 / 1e3)}s\uFF0C\u4EFB\u52A1 ${taskId} \u4ECD\u4E3A ${RUNNING_STATUS} \u72B6\u6001`
+        );
+      }
+      await delay2(POLL_INTERVAL_MS2);
+    }
+  }
+  _runQuery(taskId) {
+    const args = [
+      "query-video-direct",
+      "--task-id",
+      taskId,
+      "--ppe-env",
+      PPE_ENV,
+      "--json"
+    ];
+    return new Promise((resolve8) => {
+      const child = spawn3(BIN, args, { cwd: process.cwd() });
+      let stdout = "";
+      let stderr = "";
+      child.stdout.on("data", (chunk) => {
+        stdout += chunk.toString();
+      });
+      child.stderr.on("data", (chunk) => {
+        stderr += chunk.toString();
+      });
+      child.on("error", (error) => {
+        resolve8(
+          makeError(
+            1,
+            `\u6267\u884C ${BIN} ${args.join(" ")} \u5931\u8D25: ${error.message}`
+          )
+        );
+      });
+      child.on("close", (code, signal) => {
+        if (code === 0) {
+          resolve8(makeOkWith({ stdout, stderr }));
+          return;
+        }
+        const signalText = signal ? `, signal: ${signal}` : "";
+        const stderrText = stderr.trim() ? `
+${stderr.trim()}` : "";
+        resolve8(
+          makeError(
+            code ?? 1,
+            `\u6267\u884C ${BIN} ${args.join(" ")} \u5931\u8D25${signalText}${stderrText}`
+          )
+        );
+      });
+    });
+  }
+};
+function extractStatus(stdout) {
+  const trimmed = stdout.trim();
+  if (!trimmed) {
+    return void 0;
+  }
+  const parsed = tryParseJson(trimmed);
+  if (!parsed || typeof parsed !== "object") {
+    return void 0;
+  }
+  const record = parsed;
+  if (typeof record.status === "string" && record.status.trim()) {
+    return record.status.trim();
+  }
+  const data = record.data;
+  if (data && typeof data === "object") {
+    const dataRecord = data;
+    if (typeof dataRecord.status === "string" && dataRecord.status.trim()) {
+      return dataRecord.status.trim();
+    }
+  }
+  return void 0;
+}
+function tryParseJson(text) {
+  try {
+    return JSON.parse(text);
+  } catch {
+    return void 0;
+  }
+}
+function delay2(ms) {
+  return new Promise((resolve8) => setTimeout(resolve8, ms));
+}
+
 // src/commands/ai/ai-command.ts
 var AiCommand = class extends BaseSubcommandHost {
   _meta() {
@@ -3678,6 +3804,7 @@ var AiCommand = class extends BaseSubcommandHost {
     this._addSubcommand(new AiImageCommand());
     this._addSubcommand(new AiVideoCommand());
     this._addSubcommand(new AiVidCommand());
+    this._addSubcommand(new AiQueryTaskCommand());
   }
 };