CedarCopilot/cedar-mail

import { useCallback } from 'react';

const keyStates = new Map<string, boolean>();

let listenersInit = false;

function initKeyListeners() {
  if (typeof window !== 'undefined' && !listenersInit) {
    window.addEventListener('keydown', (e) => {
      // Store the key state
      keyStates.set(e.key, true);

      // Also store specific states for modifier keys
      if (e.altKey) {
        keyStates.set('Alt', true);
        keyStates.set('AltLeft', true);
        keyStates.set('AltRight', true);
      }

      if (e.shiftKey) {
        keyStates.set('Shift', true);
        keyStates.set('ShiftLeft', true);
        keyStates.set('ShiftRight', true);
      }
    });

    window.addEventListener('keyup', (e) => {
      // Clear the key state
      keyStates.set(e.key, false);

      // Also clear specific states for modifier keys
      if (e.key === 'Alt' || e.key === 'AltLeft' || e.key === 'AltRight') {
        keyStates.set('Alt', false);
        keyStates.set('AltLeft', false);
        keyStates.set('AltRight', false);
      }

      if (e.key === 'Shift' || e.key === 'ShiftLeft' || e.key === 'ShiftRight') {
        keyStates.set('Shift', false);
        keyStates.set('ShiftLeft', false);
        keyStates.set('ShiftRight', false);
      }
    });

    window.addEventListener('blur', () => {
      // Clear all key states when window loses focus
      keyStates.forEach((_, key) => {
        keyStates.set(key, false);
      });
    });

    listenersInit = true;
  }
}

if (typeof window !== 'undefined') {
  setTimeout(() => initKeyListeners(), 0);
}

export function useKeyState() {
  return useCallback((key=[redacted] => keyStates.get(key) || false, []);
}