maoda-commander-tt 0.0.26 → 0.0.27

package/dist/main.js

@@ -2227,9 +2227,9 @@ var ShortcutBaseCommand = class extends BaseCommand {
       stdio: "inherit",
       shell: true
     });
-    return new Promise((resolve9) => {
+    return new Promise((resolve10) => {
       child.on("error", (error) => {
-        resolve9(
+        resolve10(
           this._makeError(
             1,
             `\u6267\u884C ${cmdLine} \u5931\u8D25: ${error.message}`
@@ -2238,11 +2238,11 @@ var ShortcutBaseCommand = class extends BaseCommand {
       });
       child.on("close", (code, signal) => {
         if (code === 0) {
-          resolve9(this._makeOk());
+          resolve10(this._makeOk());
           return;
         }
         const signalText = signal ? `, signal: ${signal}` : "";
-        resolve9(
+        resolve10(
           this._makeError(
             code ?? 1,
             `\u6267\u884C ${cmdLine} \u5931\u8D25${signalText}`
@@ -2337,9 +2337,9 @@ var AiBaseCommand = class extends BaseCommand {
       cwd: process.cwd(),
       stdio: "inherit"
     });
-    return new Promise((resolve9) => {
+    return new Promise((resolve10) => {
       child.on("error", (error) => {
-        resolve9(
+        resolve10(
           this._makeError(
             1,
             `\u6267\u884C ${command} ${args.join(" ")} \u5931\u8D25: ${error.message}`
@@ -2348,11 +2348,11 @@ var AiBaseCommand = class extends BaseCommand {
       });
       child.on("close", (code, signal) => {
         if (code === 0) {
-          resolve9(this._makeOk());
+          resolve10(this._makeOk());
           return;
         }
         const signalText = signal ? `, signal: ${signal}` : "";
-        resolve9(
+        resolve10(
           this._makeError(
             code ?? 1,
             `\u6267\u884C ${command} ${args.join(" ")} \u5931\u8D25${signalText}`
@@ -2747,9 +2747,9 @@ var AiImageCommandV2 = class extends AiBaseCommand {
       cwd: process.cwd(),
       stdio: "ignore"
     });
-    return new Promise((resolve9) => {
+    return new Promise((resolve10) => {
       child.on("error", (error) => {
-        resolve9(
+        resolve10(
           this._makeError(
             1,
             `\u6267\u884C codex ${args.join(" ")} \u5931\u8D25: ${error.message}`
@@ -2758,11 +2758,11 @@ var AiImageCommandV2 = class extends AiBaseCommand {
       });
       child.on("close", (code, signal) => {
         if (code === 0) {
-          resolve9(this._makeOk());
+          resolve10(this._makeOk());
           return;
         }
         const signalText = signal ? `, signal: ${signal}` : "";
-        resolve9(
+        resolve10(
           this._makeError(
             code ?? 1,
             `\u6267\u884C codex ${args.join(" ")} \u5931\u8D25${signalText}`
@@ -3542,7 +3542,7 @@ function extractErrorText(payload) {
   return void 0;
 }
 function delay(ms) {
-  return new Promise((resolve9) => setTimeout(resolve9, ms));
+  return new Promise((resolve10) => setTimeout(resolve10, ms));
 }
 async function appendPollingUrl(pollingUrl, generationId) {
   const locationResult = await resolveConfigLocation();
@@ -3743,7 +3743,7 @@ var AiQueryTaskCommand = class extends AiBaseCommand {
       PPE_ENV,
       "--json"
     ];
-    return new Promise((resolve9) => {
+    return new Promise((resolve10) => {
       const child = spawn4(BIN, args, { cwd: process.cwd() });
       let stdout = "";
       let stderr = "";
@@ -3754,7 +3754,7 @@ var AiQueryTaskCommand = class extends AiBaseCommand {
         stderr += chunk.toString();
       });
       child.on("error", (error) => {
-        resolve9(
+        resolve10(
           makeError(
             1,
             `\u6267\u884C ${BIN} ${args.join(" ")} \u5931\u8D25: ${error.message}`
@@ -3763,13 +3763,13 @@ var AiQueryTaskCommand = class extends AiBaseCommand {
       });
       child.on("close", (code, signal) => {
         if (code === 0) {
-          resolve9(makeOkWith({ stdout, stderr }));
+          resolve10(makeOkWith({ stdout, stderr }));
           return;
         }
         const signalText = signal ? `, signal: ${signal}` : "";
         const stderrText = stderr.trim() ? `
 ${stderr.trim()}` : "";
-        resolve9(
+        resolve10(
           makeError(
             code ?? 1,
             `\u6267\u884C ${BIN} ${args.join(" ")} \u5931\u8D25${signalText}${stderrText}`
@@ -3809,7 +3809,7 @@ function tryParseJson(text) {
   }
 }
 function delay2(ms) {
-  return new Promise((resolve9) => setTimeout(resolve9, ms));
+  return new Promise((resolve10) => setTimeout(resolve10, ms));
 }
 
 // src/commands/ai/ai-command.ts
@@ -4266,6 +4266,248 @@ function asRecord2(value) {
   return value;
 }
 
+// src/commands/tool/tool-random-files-command.ts
+import { readdir as readdir3, stat as stat5 } from "fs/promises";
+import { homedir as homedir9 } from "os";
+import { extname as extname4, join as join13, resolve as resolve9 } from "path";
+var DEFAULT_COUNT = 1;
+var DEFAULT_SUFFIXES = ["png", "jpeg", "jpg"];
+var DEFAULT_DEPTH = 3;
+var ToolRandomFilesCommand = class extends BaseCommand {
+  _meta() {
+    return {
+      name: "random-files",
+      description: "\u4ECE\u6587\u4EF6\u5939\u968F\u673A\u9009\u62E9\u6587\u4EF6",
+      aliases: ["randf"]
+    };
+  }
+  _configureArguments(cmd) {
+    cmd.argument("<folder...>", "\u76EE\u6807\u6587\u4EF6\u5939");
+  }
+  _configureOptions(cmd) {
+    cmd.option("-c, --count <count>", "\u968F\u673A\u6587\u4EF6\u6570\u91CF", String(DEFAULT_COUNT));
+    cmd.option(
+      "-s, --suffix <suffixes>",
+      "\u6587\u4EF6\u540E\u7F00\uFF0C\u591A\u4E2A\u540E\u7F00\u7528\u9017\u53F7\u5206\u9694",
+      DEFAULT_SUFFIXES.join(",")
+    );
+    cmd.option(
+      "-d, --depth <depth>",
+      "\u9012\u5F52\u6587\u4EF6\u5C42\u7EA7\uFF0C\u76EE\u6807\u6587\u4EF6\u5939\u4E3A\u7B2C 1 \u5C42",
+      String(DEFAULT_DEPTH)
+    );
+    cmd.addHelpText(
+      "after",
+      [
+        "",
+        "\u8F93\u5165\u6848\u4F8B:",
+        "  tt tool random-files ~/Pictures -c 3",
+        "  tt tool random-files ./assets --suffix webp,gif --depth 2"
+      ].join("\n")
+    );
+  }
+  async _execute(ctx) {
+    const folderInputResult = parseFolderInput(ctx.args[0]);
+    if (!folderInputResult.ok) {
+      return this._makeError(1, folderInputResult.msg);
+    }
+    const countResult = parsePositiveInteger(
+      ctx.options.count,
+      DEFAULT_COUNT,
+      "\u968F\u673A\u6587\u4EF6\u6570\u91CF"
+    );
+    if (!countResult.ok) {
+      return this._makeError(1, countResult.msg);
+    }
+    const suffixResult = parseSuffixes(ctx.options.suffix);
+    if (!suffixResult.ok) {
+      return this._makeError(1, suffixResult.msg);
+    }
+    const depthResult = parsePositiveInteger(
+      ctx.options.depth,
+      DEFAULT_DEPTH,
+      "\u9012\u5F52\u6587\u4EF6\u5C42\u7EA7"
+    );
+    if (!depthResult.ok) {
+      return this._makeError(1, depthResult.msg);
+    }
+    const folderResult = await resolveFolder(folderInputResult.value);
+    if (!folderResult.ok) {
+      return this._makeError(1, folderResult.msg);
+    }
+    const candidates = await collectMatchingFiles({
+      folder: folderResult.value,
+      suffixes: suffixResult.value,
+      maxDepth: depthResult.value,
+      currentDepth: 1
+    });
+    if (candidates.length === 0) {
+      return this._makeError(
+        1,
+        `\u672A\u627E\u5230\u5339\u914D\u540E\u7F00\u7684\u6587\u4EF6: ${Array.from(suffixResult.value).join(",")}`
+      );
+    }
+    if (candidates.length < countResult.value) {
+      return this._makeError(
+        1,
+        `\u5339\u914D\u6587\u4EF6\u6570\u91CF\u4E0D\u8DB3: \u9700\u8981 ${countResult.value} \u4E2A\uFF0C\u627E\u5230 ${candidates.length} \u4E2A`
+      );
+    }
+    this._outputJson({
+      files: pickRandomFiles(candidates, countResult.value)
+    });
+    return this._makeOk();
+  }
+  _outputJson(data) {
+    this._output(JSON.stringify(data, null, 2));
+  }
+};
+function parseFolderInput(value) {
+  const folder = stringifyFolderArg(value);
+  if (!folder) {
+    return { ok: false, msg: "\u76EE\u6807\u6587\u4EF6\u5939\u4E0D\u80FD\u4E3A\u7A7A" };
+  }
+  return {
+    ok: true,
+    value: {
+      raw: folder,
+      candidates: createFolderCandidates(folder)
+    }
+  };
+}
+function stringifyFolderArg(value) {
+  if (Array.isArray(value)) {
+    return value.map((item) => String(item)).join(" ").trim();
+  }
+  return String(value ?? "").trim();
+}
+function createFolderCandidates(value) {
+  const candidates = [
+    value,
+    trimWrappingQuotes(value),
+    unescapeShellPath(trimWrappingQuotes(value))
+  ];
+  return uniqueStrings(candidates.map((candidate) => expandHomeDir2(candidate)));
+}
+function trimWrappingQuotes(value) {
+  if (value.length < 2) {
+    return value;
+  }
+  const firstChar = value[0];
+  const lastChar = value[value.length - 1];
+  if (firstChar === '"' && lastChar === '"' || firstChar === "'" && lastChar === "'") {
+    return value.slice(1, -1);
+  }
+  return value;
+}
+function unescapeShellPath(value) {
+  return value.replace(
+    /\\([ \t\n\r"'`$&;()<>|*?[\]{}!#~\\])/g,
+    "$1"
+  );
+}
+function uniqueStrings(values) {
+  return [...new Set(values)];
+}
+async function resolveFolder(folderInput) {
+  let firstExistingPath;
+  for (const candidate of folderInput.candidates) {
+    const folderStat = await safeStat2(candidate);
+    if (!folderStat) {
+      continue;
+    }
+    if (folderStat.isDirectory()) {
+      return { ok: true, value: candidate };
+    }
+    firstExistingPath ??= candidate;
+  }
+  if (firstExistingPath) {
+    return { ok: false, msg: `\u76EE\u6807\u8DEF\u5F84\u4E0D\u662F\u6587\u4EF6\u5939: ${firstExistingPath}` };
+  }
+  return { ok: false, msg: `\u76EE\u6807\u6587\u4EF6\u5939\u4E0D\u5B58\u5728: ${folderInput.raw}` };
+}
+function parsePositiveInteger(value, defaultValue, label) {
+  const rawValue = String(value ?? defaultValue).trim();
+  const parsedValue = Number(rawValue);
+  if (!Number.isInteger(parsedValue) || parsedValue <= 0) {
+    return { ok: false, msg: `${label}\u5FC5\u987B\u662F\u6B63\u6574\u6570` };
+  }
+  return { ok: true, value: parsedValue };
+}
+function parseSuffixes(value) {
+  const suffixes = String(value ?? DEFAULT_SUFFIXES.join(",")).split(/[,，\s]+/).map((suffix) => normalizeSuffix(suffix)).filter((suffix) => suffix.length > 0);
+  if (suffixes.length === 0) {
+    return { ok: false, msg: "\u6587\u4EF6\u540E\u7F00\u4E0D\u80FD\u4E3A\u7A7A" };
+  }
+  return { ok: true, value: new Set(suffixes) };
+}
+function normalizeSuffix(value) {
+  return value.trim().replace(/^\.+/, "").toLowerCase();
+}
+async function collectMatchingFiles(options) {
+  const entries = await safeReadDir(options.folder);
+  const files = [];
+  for (const entry of entries) {
+    const entryPath = join13(options.folder, entry.name);
+    if (entry.isFile()) {
+      if (hasAllowedSuffix(entry.name, options.suffixes)) {
+        files.push(entryPath);
+      }
+      continue;
+    }
+    if (entry.isDirectory() && options.currentDepth < options.maxDepth) {
+      files.push(
+        ...await collectMatchingFiles({
+          folder: entryPath,
+          suffixes: options.suffixes,
+          maxDepth: options.maxDepth,
+          currentDepth: options.currentDepth + 1
+        })
+      );
+    }
+  }
+  return files;
+}
+async function safeReadDir(folder) {
+  try {
+    const entries = await readdir3(folder, { withFileTypes: true });
+    return entries.sort((a, b) => a.name.localeCompare(b.name));
+  } catch {
+    return [];
+  }
+}
+function hasAllowedSuffix(fileName, suffixes) {
+  const suffix = normalizeSuffix(extname4(fileName));
+  return suffixes.has(suffix);
+}
+function pickRandomFiles(files, count) {
+  const shuffled = [...files];
+  for (let index = shuffled.length - 1; index > 0; index -= 1) {
+    const randomIndex = Math.floor(Math.random() * (index + 1));
+    [shuffled[index], shuffled[randomIndex]] = [
+      shuffled[randomIndex],
+      shuffled[index]
+    ];
+  }
+  return shuffled.slice(0, count);
+}
+async function safeStat2(path10) {
+  try {
+    return await stat5(path10);
+  } catch {
+    return void 0;
+  }
+}
+function expandHomeDir2(path10) {
+  if (path10 === "~") {
+    return homedir9();
+  }
+  if (path10.startsWith("~/")) {
+    return join13(homedir9(), path10.slice(2));
+  }
+  return resolve9(path10);
+}
+
 // src/commands/tool/tool-command.ts
 var ToolCommand = class extends BaseSubcommandHost {
   _meta() {
@@ -4277,6 +4519,7 @@ var ToolCommand = class extends BaseSubcommandHost {
   _registerSubcommands() {
     this._addSubcommand(new ToolCanvasCommand());
     this._addSubcommand(new ToolDrawCommand());
+    this._addSubcommand(new ToolRandomFilesCommand());
   }
 };