maoda-commander-tt 0.0.23 → 0.0.24

package/dist/main.js

@@ -4096,6 +4096,213 @@ function compareCanvasMatches(a, b) {
   return a.nodeIndex - b.nodeIndex;
 }
 
+// src/commands/tool/tool-draw-command.ts
+var ToolDrawCommand = class extends BaseCommand {
+  _meta() {
+    return {
+      name: "draw",
+      description: "\u6309\u6982\u7387\u5206\u5E03\u62BD\u7B7E"
+    };
+  }
+  _configureArguments(cmd) {
+    cmd.argument("<count>", "\u8981\u62BD\u53D6\u7684\u4E2A\u6570");
+    cmd.argument("<distribution...>", "\u6982\u7387\u5206\u5E03\uFF0C\u652F\u6301\u7B80\u5199\u6216 JSON \u6570\u7EC4");
+    cmd.addHelpText(
+      "after",
+      [
+        "",
+        "\u8F93\u5165\u6848\u4F8B:",
+        '  tt tool draw 3 "A A:50,B B:30,C C:20"',
+        `  tt tool draw 3 '[{"name":"\u4E2D\u56FD \u6210\u90FD","weight":50},{"name":"Los Angeles","weight":50}]'`
+      ].join("\n")
+    );
+  }
+  async _execute(ctx) {
+    const countResult = parseCount(ctx.args[0]);
+    if (!countResult.ok) {
+      return this._makeError(1, countResult.msg);
+    }
+    const distributionText = stringifyDistributionArg(ctx.args[1]);
+    if (!distributionText) {
+      return this._makeError(1, "\u6982\u7387\u5206\u5E03\u4E0D\u80FD\u4E3A\u7A7A");
+    }
+    const distributionResult = parseDistribution(distributionText);
+    if (!distributionResult.ok) {
+      return this._makeError(1, distributionResult.msg);
+    }
+    this._outputJson({
+      items: drawWeightedItems(countResult.value, distributionResult.value)
+    });
+    return this._makeOk();
+  }
+  _outputJson(data) {
+    this._output(JSON.stringify(data, null, 2));
+  }
+};
+function parseCount(value) {
+  const count = Number(String(value ?? "").trim());
+  if (!Number.isInteger(count) || count <= 0) {
+    return { ok: false, msg: "\u62BD\u53D6\u4E2A\u6570\u5FC5\u987B\u662F\u6B63\u6574\u6570" };
+  }
+  return { ok: true, value: count };
+}
+function stringifyDistributionArg(value) {
+  if (Array.isArray(value)) {
+    return value.map((item) => String(item)).join(" ").trim();
+  }
+  return String(value ?? "").trim();
+}
+function parseDistribution(value) {
+  if (value.trimStart().startsWith("[")) {
+    return parseJsonDistribution(value);
+  }
+  return parseShorthandDistribution(value);
+}
+function parseJsonDistribution(value) {
+  let parsed;
+  try {
+    parsed = JSON.parse(value);
+  } catch (err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    return { ok: false, msg: `JSON \u6982\u7387\u5206\u5E03\u89E3\u6790\u5931\u8D25: ${msg}` };
+  }
+  if (!Array.isArray(parsed)) {
+    return { ok: false, msg: "JSON \u6982\u7387\u5206\u5E03\u5FC5\u987B\u662F\u6570\u7EC4" };
+  }
+  const entries = [];
+  for (let index = 0; index < parsed.length; index += 1) {
+    const entryResult = parseJsonDistributionEntry(parsed[index], index);
+    if (!entryResult.ok) {
+      return entryResult;
+    }
+    entries.push(entryResult.value);
+  }
+  return validateDistribution(entries);
+}
+function parseJsonDistributionEntry(value, index) {
+  const record = asRecord2(value);
+  if (!record) {
+    return { ok: false, msg: `\u7B2C ${index + 1} \u9879\u5FC5\u987B\u662F\u5BF9\u8C61` };
+  }
+  const nameResult = parseEntryName(record.name, `\u7B2C ${index + 1} \u9879`);
+  if (!nameResult.ok) {
+    return nameResult;
+  }
+  const weightResult = parseWeight(
+    getJsonWeightValue(record),
+    `\u7B2C ${index + 1} \u9879`
+  );
+  if (!weightResult.ok) {
+    return weightResult;
+  }
+  return {
+    ok: true,
+    value: {
+      name: nameResult.value,
+      weight: weightResult.value
+    }
+  };
+}
+function getJsonWeightValue(record) {
+  if ("weight" in record) {
+    return record.weight;
+  }
+  if ("probability" in record) {
+    return record.probability;
+  }
+  if ("prob" in record) {
+    return record.prob;
+  }
+  return record.p;
+}
+function parseShorthandDistribution(value) {
+  const parts = value.split(/[,，]/);
+  const entries = [];
+  for (let index = 0; index < parts.length; index += 1) {
+    const part = parts[index].trim();
+    if (!part) {
+      return { ok: false, msg: `\u7B2C ${index + 1} \u9879\u4E3A\u7A7A` };
+    }
+    const separatorIndex = findLastWeightSeparator(part);
+    if (separatorIndex < 0) {
+      return {
+        ok: false,
+        msg: `\u7B2C ${index + 1} \u9879\u7F3A\u5C11\u6982\u7387\u5206\u9694\u7B26 ":"`
+      };
+    }
+    const nameResult = parseEntryName(
+      part.slice(0, separatorIndex),
+      `\u7B2C ${index + 1} \u9879`
+    );
+    if (!nameResult.ok) {
+      return nameResult;
+    }
+    const weightResult = parseWeight(
+      part.slice(separatorIndex + 1),
+      `\u7B2C ${index + 1} \u9879`
+    );
+    if (!weightResult.ok) {
+      return weightResult;
+    }
+    entries.push({
+      name: nameResult.value,
+      weight: weightResult.value
+    });
+  }
+  return validateDistribution(entries);
+}
+function findLastWeightSeparator(value) {
+  return Math.max(value.lastIndexOf(":"), value.lastIndexOf("\uFF1A"));
+}
+function parseEntryName(value, label) {
+  if (typeof value !== "string") {
+    return { ok: false, msg: `${label}\u7684\u540D\u79F0\u5FC5\u987B\u662F\u5B57\u7B26\u4E32` };
+  }
+  const name = value.trim();
+  if (!name) {
+    return { ok: false, msg: `${label}\u7684\u540D\u79F0\u4E0D\u80FD\u4E3A\u7A7A` };
+  }
+  return { ok: true, value: name };
+}
+function parseWeight(value, label) {
+  const rawValue = typeof value === "string" ? value.trim().replace(/[％%]$/, "") : value;
+  const weight = Number(rawValue);
+  if (!Number.isFinite(weight) || weight <= 0) {
+    return { ok: false, msg: `${label}\u7684\u6982\u7387\u5FC5\u987B\u662F\u5927\u4E8E 0 \u7684\u6570\u5B57` };
+  }
+  return { ok: true, value: weight };
+}
+function validateDistribution(entries) {
+  if (entries.length === 0) {
+    return { ok: false, msg: "\u6982\u7387\u5206\u5E03\u81F3\u5C11\u9700\u8981 1 \u9879" };
+  }
+  return { ok: true, value: entries };
+}
+function drawWeightedItems(count, entries) {
+  const totalWeight = entries.reduce((sum, entry) => sum + entry.weight, 0);
+  const items = [];
+  for (let index = 0; index < count; index += 1) {
+    items.push(drawWeightedItem(entries, totalWeight));
+  }
+  return items;
+}
+function drawWeightedItem(entries, totalWeight) {
+  let cursor = Math.random() * totalWeight;
+  for (const entry of entries) {
+    cursor -= entry.weight;
+    if (cursor < 0) {
+      return entry.name;
+    }
+  }
+  return entries[entries.length - 1].name;
+}
+function asRecord2(value) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) {
+    return void 0;
+  }
+  return value;
+}
+
 // src/commands/tool/tool-command.ts
 var ToolCommand = class extends BaseSubcommandHost {
   _meta() {
@@ -4106,6 +4313,7 @@ var ToolCommand = class extends BaseSubcommandHost {
   }
   _registerSubcommands() {
     this._addSubcommand(new ToolCanvasCommand());
+    this._addSubcommand(new ToolDrawCommand());
   }
 };