测试脚本_2

Bin

2025-12-17 262fecaa75b2909ad244f12c3b079ed3ff4ae329

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import type Keymaster from "keymaster";
import { useEffect, useRef } from "react";
import { Hotkey } from "../core/Hotkey";
 
type Keyname = keyof typeof Hotkey.keymap;
 
const hotkeys = Hotkey();
 
const attachHotkey = (key: Keyname, handler: Keymaster.KeyHandler, scope?: string) => {
  if (Hotkey.keymap[key]) {
    hotkeys.overwriteNamed(key as string, handler, scope);
  } else {
    hotkeys.overwriteKey(key as string, handler, scope);
  }
};
 
const removeHotkey = (key: Keyname, scope?: string) => {
  if (Hotkey.keymap[key]) {
    hotkeys.removeNamed(key as string, scope);
  } else {
    hotkeys.removeKey(key as string, scope);
  }
};
 
export const useHotkey = (hotkey?: Keyname, handler?: Keymaster.KeyHandler, scope?: string) => {
  const lastHotkey = useRef<Keyname | null>(null);
  const lastScope = useRef<string | null>(null);
  const handlerFunction = useRef<Keymaster.KeyHandler | undefined>(handler);
 
  // we wanna cache handler function so the prop change does not re-attac a hotkey
  // refs are perfect fit for this purpose as they're mutable and cached during the hook lifecycle
  const handlerWrapper = useRef<Keymaster.KeyHandler>((e, h) => {
    handlerFunction.current?.(e, h);
  });
 
  useEffect(() => {
    const hotkeyChanged = hotkey !== lastHotkey.current;
    const scopeChanged = scope !== lastScope.current;
 
    // hotkey itself only references a cached version of a function
    // so it's never re-attached even if handler changes
    // handler update might happen if it's wrapped with useCallback
    // and will trigger infinite loop if we use it as a dependency for
    // current effect
    (() => {
      if (!hotkeyChanged && !scopeChanged) return;
 
      if (hotkey) {
        attachHotkey(hotkey, handlerWrapper.current, scope);
        lastHotkey.current = hotkey;
      } else if (lastHotkey.current && !hotkey) {
        removeHotkey(lastHotkey.current, lastScope.current);
        lastHotkey.current = null;
      }
    })();
  }, [hotkey, scope]);
 
  // by changing the ref we can safely update the handler
  // as refs are mutable and doesn't trigger react-updates
  // but the hotkey will still have access to a proper function
  useEffect(() => {
    handlerFunction.current = handler;
  }, [handler]);
};