voidraft/frontend/src/views/editor/extensions/contextMenu/index.ts

import { EditorView } from '@codemirror/view';
import { Extension } from '@codemirror/state';
import { copyCommand, cutCommand, pasteCommand } from '../codeblock/copyPaste';
import { KeyBindingKey } from '@/../bindings/voidraft/internal/models/models';
import { useKeybindingStore } from '@/stores/keybindingStore';
import { undo, redo } from '@codemirror/commands';
import i18n from '@/i18n';
import { useSystemStore } from '@/stores/systemStore';
import { showContextMenu } from './manager';
import {
  buildRegisteredMenu,
  createMenuContext,
  registerMenuNodes
} from './menuSchema';
import type { MenuSchemaNode } from './menuSchema';


function t(key: string): string {
  return i18n.global.t(key);
}


function formatKeyBinding(keyBinding: string): string {
  const systemStore = useSystemStore();
  const isMac = systemStore.isMacOS;

  return keyBinding
    .replace("Mod", isMac ? "Cmd" : "Ctrl")
    .replace("Alt", isMac ? "Option" : "Alt")
    .replace(/-/g, " + ");
}

const shortcutCache = new Map<KeyBindingKey, string>();


function getShortcutText(keyBindingKey?: KeyBindingKey): string {
  if (keyBindingKey === undefined) {
    return "";
  }

  const cached = shortcutCache.get(keyBindingKey);
  if (cached !== undefined) {
    return cached;
  }

  try {
    const keybindingStore = useKeybindingStore();
    // binding.key 是命令标识符，binding.command 是快捷键组合
    const binding = keybindingStore.keyBindings.find(
      (kb) => kb.key === keyBindingKey && kb.enabled
    );

    if (binding?.command) {
      const formatted = formatKeyBinding(binding.command);
      shortcutCache.set(keyBindingKey, formatted);
      return formatted;
    }
  } catch (error) {
    console.warn("An error occurred while getting the shortcut:", error);
  }

  shortcutCache.set(keyBindingKey, "");
  return "";
}


function builtinMenuNodes(): MenuSchemaNode[] {
  return [
    {
      id: "copy",
      labelKey: "keybindings.commands.blockCopy",
      command: copyCommand,
      keyBindingKey: KeyBindingKey.BlockCopyKeyBindingKey,
      enabled: (context) => context.hasSelection
    },
    {
      id: "cut",
      labelKey: "keybindings.commands.blockCut",
      command: cutCommand,
      keyBindingKey: KeyBindingKey.BlockCutKeyBindingKey,
      visible: (context) => context.isEditable,
      enabled: (context) => context.hasSelection && context.isEditable
    },
    {
      id: "paste",
      labelKey: "keybindings.commands.blockPaste",
      command: pasteCommand,
      keyBindingKey: KeyBindingKey.BlockPasteKeyBindingKey,
      visible: (context) => context.isEditable
    },
    {
      id: "undo",
      labelKey: "keybindings.commands.historyUndo",
      command: undo,
      keyBindingKey: KeyBindingKey.HistoryUndoKeyBindingKey,
      visible: (context) => context.isEditable
    },
    {
      id: "redo",
      labelKey: "keybindings.commands.historyRedo",
      command: redo,
      keyBindingKey: KeyBindingKey.HistoryRedoKeyBindingKey,
      visible: (context) => context.isEditable
    }
  ];
}

let builtinMenuRegistered = false;

function ensureBuiltinMenuRegistered(): void {
  if (builtinMenuRegistered) return;
  registerMenuNodes(builtinMenuNodes());
  builtinMenuRegistered = true;
}


export function createEditorContextMenu(): Extension {
  ensureBuiltinMenuRegistered();

  return EditorView.domEventHandlers({
    contextmenu: (event, view) => {
      event.preventDefault();

      const context = createMenuContext(view, event as MouseEvent);
      const menuItems = buildRegisteredMenu(context, {
        translate: t,
        formatShortcut: getShortcutText
      });

      if (menuItems.length === 0) {
        return false;
      }

      showContextMenu(view, event.clientX, event.clientY, menuItems);
      return true;
    }
  });
}

export default createEditorContextMenu;