yaak/apps/yaak-proxy/lib/hotkeys.ts

import type { ActionInvocation, ActionMetadata } from '@yaakapp-internal/proxy-lib';
import { rpc } from './rpc';

type ActionBinding = {
  invocation: ActionInvocation;
  meta: ActionMetadata;
  keys: { key: string; ctrl: boolean; shift: boolean; alt: boolean; meta: boolean };
};

/** Parse a hotkey string like "Ctrl+Shift+P" into its parts. */
function parseHotkey(hotkey: string): ActionBinding['keys'] {
  const parts = hotkey.split('+').map((p) => p.trim().toLowerCase());
  return {
    ctrl: parts.includes('ctrl') || parts.includes('control'),
    shift: parts.includes('shift'),
    alt: parts.includes('alt'),
    meta: parts.includes('meta') || parts.includes('cmd') || parts.includes('command'),
    key:
      parts.filter(
        (p) => !['ctrl', 'control', 'shift', 'alt', 'meta', 'cmd', 'command'].includes(p),
      )[0] ?? '',
  };
}

function matchesEvent(binding: ActionBinding['keys'], e: KeyboardEvent): boolean {
  return (
    e.ctrlKey === binding.ctrl &&
    e.shiftKey === binding.shift &&
    e.altKey === binding.alt &&
    e.metaKey === binding.meta &&
    e.key.toLowerCase() === binding.key
  );
}

/** Fetch all actions from Rust and register a global keydown listener. */
export async function initHotkeys(): Promise<() => void> {
  const { actions } = await rpc('list_actions', {});

  const bindings: ActionBinding[] = actions
    .filter(
      (entry): entry is [ActionInvocation, ActionMetadata & { defaultHotkey: string }] =>
        entry[1].defaultHotkey != null,
    )
    .map(([invocation, meta]) => ({
      invocation,
      meta,
      keys: parseHotkey(meta.defaultHotkey),
    }));

  function onKeyDown(e: KeyboardEvent) {
    for (const binding of bindings) {
      if (matchesEvent(binding.keys, e)) {
        e.preventDefault();
        rpc('execute_action', binding.invocation);
        return;
      }
    }
  }

  window.addEventListener('keydown', onKeyDown);
  return () => window.removeEventListener('keydown', onKeyDown);
}