fix: add repr method for debugging

violentmonkey · Mar 8, 2024 · 845d0b5 · 845d0b5
1 parent 504c70b
commit 845d0b5
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 10 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1,3 +1,12 @@
+import { modifiers } from './constants';
+import {
+  addKeyNode,
+  createKeyNode,
+  getKeyNode,
+  removeKeyNode,
+  reprNodeTree,
+} from './node';
+import { Subject } from './subject';
 import {
   IKeyNode,
   IShortcut,
@@ -6,10 +15,7 @@ import {
   IShortcutOptions,
   IShortcutServiceOptions,
 } from './types';
-import { normalizeSequence, parseCondition, buildKey } from './util';
-import { addKeyNode, createKeyNode, getKeyNode, removeKeyNode } from './node';
-import { modifiers } from './constants';
-import { Subject } from './subject';
+import { buildKey, normalizeSequence, parseCondition } from './util';
 
 export * from './constants';
 export * from './types';
@@ -238,6 +244,10 @@ export class KeyboardService {
     }
     this._timer = window.setTimeout(this._reset, this.options.sequenceTimeout);
   };
+
+  repr() {
+    return reprNodeTree(this._root);
+  }
 }
 
 let service: KeyboardService;

diff --git a/src/node.ts b/src/node.ts
@@ -8,11 +8,11 @@ export function createKeyNode(): IKeyNode {
 }
 
 export function addKeyNode(
-  parent: IKeyNode,
+  root: IKeyNode,
   sequence: string[],
   shortcut: IShortcut,
 ) {
-  let node: IKeyNode = parent;
+  let node: IKeyNode = root;
   for (const key of sequence) {
     let child = node.children.get(key);
     if (!child) {
@@ -24,8 +24,8 @@ export function addKeyNode(
   node.shortcuts.add(shortcut);
 }
 
-export function getKeyNode(parent: IKeyNode, sequence: string[]) {
-  let node: IKeyNode | undefined = parent;
+export function getKeyNode(root: IKeyNode, sequence: string[]) {
+  let node: IKeyNode | undefined = root;
   for (const key of sequence) {
     node = node.children.get(key);
     if (!node) break;
@@ -34,11 +34,11 @@ export function getKeyNode(parent: IKeyNode, sequence: string[]) {
 }
 
 export function removeKeyNode(
-  parent: IKeyNode,
+  root: IKeyNode,
   sequence: string[],
   shortcut?: IShortcut,
 ) {
-  let node: IKeyNode | undefined = parent;
+  let node: IKeyNode | undefined = root;
   const ancestors = [node];
   for (const key of sequence) {
     node = node.children.get(key);
@@ -56,3 +56,21 @@ export function removeKeyNode(
     i -= 1;
   }
 }
+
+export function reprNodeTree(root: IKeyNode) {
+  const result: string[] = [];
+  const reprChildren = (node: IKeyNode, level = 0) => {
+    for (const [key, child] of node.children.entries()) {
+      result.push(
+        [
+          '  '.repeat(level),
+          key,
+          child.shortcuts.size ? ` (${child.shortcuts.size})` : '',
+        ].join(''),
+      );
+      reprChildren(child, level + 1);
+    }
+  };
+  reprChildren(root);
+  return result.join('\n');
+}