snapshot of https://github.com/toeverything/AFFiNE.git (canary, 2026-09-16; blobs > 131072B omitted)

blocksuite/framework/std/src/extension/keymap.ts

@@ -0,0 +1,49 @@
+import type { ExtensionType } from '@blocksuite/store';
+
+import type { EventOptions, UIEventHandler } from '../event/index.js';
+import { KeymapIdentifier } from '../identifier.js';
+import type { BlockStdScope } from '../scope/index.js';
+
+let id = 1;
+
+/**
+ * Create a keymap extension.
+ *
+ * @param keymapFactory
+ * Create keymap of the extension.
+ * It should return an object with `keymap` and `options`.
+ *
+ * `keymap` is a record of keymap.
+ *
+ * @param options
+ * `options` is an optional object that restricts the event to be handled.
+ *
+ * @example
+ * ```ts
+ * import { KeymapExtension } from '@blocksuite/std';
+ *
+ * const MyKeymapExtension = KeymapExtension(std => {
+ *   return {
+ *     keymap: {
+ *       'mod-a': SelectAll
+ *     }
+ *     options: {
+ *       flavour: 'affine:paragraph'
+ *     }
+ *   }
+ * });
+ * ```
+ */
+export function KeymapExtension(
+  keymapFactory: (std: BlockStdScope) => Record<string, UIEventHandler>,
+  options?: EventOptions
+): ExtensionType {
+  return {
+    setup: di => {
+      di.addImpl(KeymapIdentifier(`Keymap-${id++}`), {
+        getter: keymapFactory,
+        options,
+      });
+    },
+  };
+}