ttrpg-tools/src/components/md-commander/hooks/useCommander.ts

import { createSignal } from "solid-js";
import {
  MdCommanderCommand,
  CommanderEntry,
  CompletionItem,
} from "../types";
import { useDiceRoller } from "./useDiceRoller";

// ==================== 默认命令 ====================

export const defaultCommands: Record<string, MdCommanderCommand> = {
  help: {
    command: "help",
    description: "显示帮助信息或特定命令的帮助",
    options: {
      cmd: {
        option: "cmd",
        description: "要查询的命令名",
        type: "enum",
        values: [], // 运行时填充
      },
    },
    handler: (args) => {
      if (args.cmd) {
        return {
          message: `命令：${args.cmd}\n描述：${defaultCommands[args.cmd]?.description || "无描述"}`,
          type: "info",
        };
      }
      const cmdList = Object.keys(defaultCommands).join(", ");
      return {
        message: `可用命令：${cmdList}`,
        type: "info",
      };
    },
  },
  clear: {
    command: "clear",
    description: "清空命令历史",
    handler: () => ({
      message: "命令历史已清空",
      type: "success",
    }),
  },
  roll: {
    command: "roll",
    description: "掷骰子 - 支持骰池、修饰符和组合",
    options: {
      formula: {
        option: "formula",
        description: "骰子公式，如 3d6+{d4,d8}kh2-5",
        type: "string",
        required: true,
      },
    },
    handler: (args) => {
      const formula = args.formula || "1d6";
      const { rollSimple } = useDiceRoller();
      const result = rollSimple(formula);
      return {
        message: result.text,
        isHtml: result.isHtml,
        type: result.text.startsWith("错误") ? "error" : "success",
      };
    },
  },
};

// ==================== 工具函数 ====================

export function parseInput(input: string): {
  command?: string;
  options: Record<string, string>;
  incompleteOption?: string;
} {
  const result: {
    command?: string;
    options: Record<string, string>;
    incompleteOption?: string;
  } = {
    options: {},
  };

  const trimmed = input.trim();
  if (!trimmed) return result;

  const parts = trimmed.split(/\s+/);
  let i = 0;

  if (parts[0] && !parts[0].startsWith("-")) {
    result.command = parts[0];
    i = 1;
  }

  while (i < parts.length) {
    const part = parts[i];
    if (part.startsWith("--")) {
      const eqIndex = part.indexOf("=");
      if (eqIndex !== -1) {
        const key = part.slice(2, eqIndex);
        const value = part.slice(eqIndex + 1);
        result.options[key] = value;
      } else {
        const key = part.slice(2);
        if (i + 1 < parts.length && !parts[i + 1].startsWith("-")) {
          result.options[key] = parts[i + 1];
          i++;
        } else {
          result.incompleteOption = key;
        }
      }
    }
    i++;
  }

  return result;
}

export function getCompletions(
  input: string,
  commands: Record<string, MdCommanderCommand>,
): CompletionItem[] {
  const trimmed = input.trim();

  if (!trimmed || /^\s*$/.test(trimmed)) {
    return Object.values(commands).map((cmd) => ({
      label: cmd.command,
      type: "command",
      description: cmd.description,
      insertText: cmd.command,
    }));
  }

  const parsed = parseInput(trimmed);

  if (!parsed.command || !commands[parsed.command]) {
    const commandCompletions = Object.values(commands)
      .filter((cmd) => cmd.command.startsWith(parsed.command || ""))
      .map((cmd) => ({
        label: cmd.command,
        type: "command" as "command",
        description: cmd.description,
        insertText: cmd.command,
      }));
    if (commandCompletions.length > 0) {
      return commandCompletions;
    }
  }

  const cmd = commands[parsed.command!];
  if (!cmd || !cmd.options) return [];

  if (parsed.incompleteOption) {
    const option = cmd.options[parsed.incompleteOption];
    if (option?.type === "enum" && option.values) {
      return option.values
        .filter((v) => parsed.incompleteOption && v.startsWith(parsed.incompleteOption))
        .map((v) => ({
          label: v,
          type: "value",
          insertText: v,
        }));
    }
  }

  const usedOptions = Object.keys(parsed.options);
  return Object.values(cmd.options)
    .filter((opt) => !usedOptions.includes(opt.option))
    .map((opt) => ({
      label: `--${opt.option}`,
      type: "option",
      description: opt.description,
      insertText: `--${opt.option}=${opt.type === "boolean" ? "" : ""}`,
    }));
}

export function getResultClass(
  type?: "success" | "error" | "warning" | "info",
): string {
  switch (type) {
    case "success":
      return "text-green-600";
    case "error":
      return "text-red-600";
    case "warning":
      return "text-yellow-600";
    case "info":
    default:
      return "text-blue-600";
  }
}

// ==================== Commander Hook ====================

export interface UseCommanderReturn {
  inputValue: () => string;
  entries: () => CommanderEntry[];
  showCompletions: () => boolean;
  completions: () => CompletionItem[];
  selectedCompletion: () => number;
  isFocused: () => boolean;
  setInputValue: (v: string) => void;
  setEntries: (updater: (prev: CommanderEntry[]) => CommanderEntry[]) => void;
  setShowCompletions: (v: boolean) => void;
  setSelectedCompletion: (v: number | ((prev: number) => number)) => void;
  setIsFocused: (v: boolean) => void;
  handleCommand: () => void;
  updateCompletions: () => void;
  acceptCompletion: () => void;
  commands: Record<string, MdCommanderCommand>;
}

export function useCommander(
  customCommands?: Record<string, MdCommanderCommand>,
): UseCommanderReturn {
  const [inputValue, setInputValue] = createSignal("");
  const [entries, setEntries] = createSignal<CommanderEntry[]>([]);
  const [showCompletions, setShowCompletions] = createSignal(false);
  const [completions, setCompletions] = createSignal<CompletionItem[]>([]);
  const [selectedCompletion, setSelectedCompletion] = createSignal(0);
  const [isFocused, setIsFocused] = createSignal(false);

  const commands = { ...defaultCommands, ...customCommands };

  // 更新 help 命令的选项值
  if (commands.help?.options?.cmd) {
    commands.help.options.cmd.values = Object.keys(commands).filter(
      (k) => k !== "help",
    );
  }

  const handleCommand = () => {
    const input = inputValue().trim();
    if (!input) return;

    const parsed = parseInput(input);
    const commandName = parsed.command;
    const cmd = commands[commandName!];

    let result: { message: string; type?: "success" | "error" | "warning" | "info" };

    if (!cmd) {
      result = { message: `未知命令：${commandName}`, type: "error" };
    } else if (cmd.handler) {
      try {
        result = cmd.handler(parsed.options);
      } catch (e) {
        result = {
          message: `执行错误：${e instanceof Error ? e.message : String(e)}`,
          type: "error",
        };
      }
    } else {
      result = { message: `命令 ${commandName} 已执行（无处理器）`, type: "info" };
    }

    const newEntry: CommanderEntry = {
      id: Date.now().toString() + Math.random().toString(36).slice(2),
      command: input,
      args: parsed.options,
      result,
      timestamp: new Date(),
    };

    setEntries((prev) => [...prev, newEntry]);
    setInputValue("");
    setShowCompletions(false);

    if (commandName === "clear") {
      setEntries([]);
    }
  };

  const updateCompletions = () => {
    const input = inputValue();
    const comps = getCompletions(input, commands);
    setCompletions(comps);
    setShowCompletions(comps.length > 0 && isFocused());
    setSelectedCompletion(0);
  };

  const acceptCompletion = () => {
    const idx = selectedCompletion();
    const comp = completions()[idx];
    if (!comp) return;

    const input = inputValue();
    const parsed = parseInput(input);

    let newValue: string;
    if (comp.type === "command") {
      newValue = comp.insertText + " ";
    } else if (comp.type === "option") {
      const base = parsed.command || "";
      const existingOptions = Object.entries(parsed.options)
        .map(([k, v]) => `--${k}=${v}`)
        .join(" ");
      newValue = `${base} ${existingOptions}${existingOptions ? " " : ""}${comp.insertText}`;
    } else {
      const optKey = parsed.incompleteOption;
      if (optKey) {
        const base = parsed.command || "";
        const otherOptions = Object.entries(parsed.options)
          .filter(([k]) => k !== optKey)
          .map(([k, v]) => `--${k}=${v}`)
          .join(" ");
        newValue = `${base} --${optKey}=${comp.insertText}${otherOptions ? " " + otherOptions : ""}`;
      } else {
        newValue = input;
      }
    }

    setInputValue(newValue.trim());
    setShowCompletions(false);
  };

  return {
    inputValue,
    entries,
    showCompletions,
    completions,
    selectedCompletion,
    isFocused,
    setInputValue,
    setEntries,
    setShowCompletions,
    setSelectedCompletion,
    setIsFocused,
    handleCommand,
    updateCompletions,
    acceptCompletion,
    commands,
  };
}