Repeat content operation with a numeric prefix

2 files changed, 103 insertions, 95 deletions

diff --git a/src/content/controllers/KeymapController.ts b/src/content/controllers/KeymapController.ts
index 6157a71..cf59ae5 100644
--- a/src/content/controllers/KeymapController.ts
+++ b/src/content/controllers/KeymapController.ts
@@ -32,71 +32,75 @@ export default class KeymapController {
 
   // eslint-disable-next-line complexity, max-lines-per-function
   press(key: Key): boolean {
-    let op = this.keymapUseCase.nextOp(key);
-    if (op === null) {
+    let ops = this.keymapUseCase.nextOps(key);
+    if (ops.length === 0) {
       return false;
     }
 
-    // do not await due to return a boolean immediately
-    switch (op.type) {
-    case operations.ADDON_ENABLE:
-      this.addonEnabledUseCase.enable();
-      break;
-    case operations.ADDON_DISABLE:
-      this.addonEnabledUseCase.disable();
-      break;
-    case operations.ADDON_TOGGLE_ENABLED:
-      this.addonEnabledUseCase.toggle();
-      break;
-    case operations.FIND_NEXT:
-      this.findSlaveUseCase.findNext();
-      break;
-    case operations.FIND_PREV:
-      this.findSlaveUseCase.findPrev();
-      break;
-    case operations.SCROLL_VERTICALLY:
-      this.scrollUseCase.scrollVertically(op.count);
-      break;
-    case operations.SCROLL_HORIZONALLY:
-      this.scrollUseCase.scrollHorizonally(op.count);
-      break;
-    case operations.SCROLL_PAGES:
-      this.scrollUseCase.scrollPages(op.count);
-      break;
-    case operations.SCROLL_TOP:
-      this.scrollUseCase.scrollToTop();
-      break;
-    case operations.SCROLL_BOTTOM:
-      this.scrollUseCase.scrollToBottom();
-      break;
-    case operations.SCROLL_HOME:
-      this.scrollUseCase.scrollToHome();
-      break;
-    case operations.SCROLL_END:
-      this.scrollUseCase.scrollToEnd();
-      break;
-    case operations.FOLLOW_START:
-      this.followMasterClient.startFollow(op.newTab, op.background);
-      break;
-    case operations.MARK_SET_PREFIX:
-      this.markKeyUseCase.enableSetMode();
-      break;
-    case operations.MARK_JUMP_PREFIX:
-      this.markKeyUseCase.enableJumpMode();
-      break;
-    case operations.FOCUS_INPUT:
-      this.focusUseCase.focusFirstInput();
-      break;
-    case operations.URLS_YANK:
-      this.clipbaordUseCase.yankCurrentURL();
-      break;
-    case operations.URLS_PASTE:
-      this.clipbaordUseCase.openOrSearch(
-        op.newTab ? op.newTab : false,
-      );
-      break;
-    default:
-      this.backgroundClient.execBackgroundOp(op);
+    // Do not await asynchronous methods to return a boolean immidiately. The
+    // caller requires the synchronous response from the callback to identify
+    // to continue of abandon the event propagations.
+    for (let op of ops) {
+      switch (op.type) {
+      case operations.ADDON_ENABLE:
+        this.addonEnabledUseCase.enable();
+        break;
+      case operations.ADDON_DISABLE:
+        this.addonEnabledUseCase.disable();
+        break;
+      case operations.ADDON_TOGGLE_ENABLED:
+        this.addonEnabledUseCase.toggle();
+        break;
+      case operations.FIND_NEXT:
+        this.findSlaveUseCase.findNext();
+        break;
+      case operations.FIND_PREV:
+        this.findSlaveUseCase.findPrev();
+        break;
+      case operations.SCROLL_VERTICALLY:
+        this.scrollUseCase.scrollVertically(op.count);
+        break;
+      case operations.SCROLL_HORIZONALLY:
+        this.scrollUseCase.scrollHorizonally(op.count);
+        break;
+      case operations.SCROLL_PAGES:
+        this.scrollUseCase.scrollPages(op.count);
+        break;
+      case operations.SCROLL_TOP:
+        this.scrollUseCase.scrollToTop();
+        break;
+      case operations.SCROLL_BOTTOM:
+        this.scrollUseCase.scrollToBottom();
+        break;
+      case operations.SCROLL_HOME:
+        this.scrollUseCase.scrollToHome();
+        break;
+      case operations.SCROLL_END:
+        this.scrollUseCase.scrollToEnd();
+        break;
+      case operations.FOLLOW_START:
+        this.followMasterClient.startFollow(op.newTab, op.background);
+        break;
+      case operations.MARK_SET_PREFIX:
+        this.markKeyUseCase.enableSetMode();
+        break;
+      case operations.MARK_JUMP_PREFIX:
+        this.markKeyUseCase.enableJumpMode();
+        break;
+      case operations.FOCUS_INPUT:
+        this.focusUseCase.focusFirstInput();
+        break;
+      case operations.URLS_YANK:
+        this.clipbaordUseCase.yankCurrentURL();
+        break;
+      case operations.URLS_PASTE:
+        this.clipbaordUseCase.openOrSearch(
+          op.newTab ? op.newTab : false,
+        );
+        break;
+      default:
+        this.backgroundClient.execBackgroundOp(op);
+      }
     }
     return true;
   }
diff --git a/src/content/usecases/KeymapUseCase.ts b/src/content/usecases/KeymapUseCase.ts
index 1707f6f..f7de334 100644
--- a/src/content/usecases/KeymapUseCase.ts
+++ b/src/content/usecases/KeymapUseCase.ts
@@ -8,13 +8,16 @@ import Key from '../../shared/settings/Key';
 import KeySequence from '../domains/KeySequence';
 import AddressRepository from '../repositories/AddressRepository';
 
-type KeymapEntityMap = Map<KeySequence, operations.Operation>;
-
 const reservedKeymaps = Keymaps.fromJSON({
   '<Esc>': { type: operations.CANCEL },
   '<C-[>': { type: operations.CANCEL },
 });
 
+const enableAddonOps = [
+  operations.ADDON_ENABLE,
+  operations.ADDON_TOGGLE_ENABLED,
+];
+
 @injectable()
 export default class KeymapUseCase {
   constructor(
@@ -32,53 +35,54 @@ export default class KeymapUseCase {
   ) {
   }
 
-  nextOp(key: Key): operations.Operation | null {
+  // eslint-disable-next-line max-statements
+  nextOps(key: Key): operations.Operation[] {
     let sequence = this.repository.enqueueKey(key);
-    if (sequence.length() === 1 && this.blacklistKey(key)) {
+    let baseSequence = sequence.trimNumericPrefix();
+    if (baseSequence.length() === 1 && this.blacklistKey(key)) {
       // ignore if the input starts with black list keys
       this.repository.clear();
-      return null;
+      return [];
     }
 
     let keymaps = this.keymapEntityMap();
-    let matched = Array.from(keymaps.keys()).filter(
-      (mapping: KeySequence) => {
-        return mapping.startsWith(sequence);
-      });
-    if (!this.addonEnabledRepository.get()) {
-      // available keymaps are only ADDON_ENABLE and ADDON_TOGGLE_ENABLED if
-      // the addon disabled
-      matched = matched.filter((keymap) => {
-        let type = (keymaps.get(keymap) as operations.Operation).type;
-        return type === operations.ADDON_ENABLE ||
-          type === operations.ADDON_TOGGLE_ENABLED;
-      });
-    }
-    if (matched.length === 0) {
-      // No operations to match with inputs
+    let matched = keymaps.filter(([seq]) => seq.startsWith(sequence));
+    let baseMatched = keymaps.filter(([seq]) => seq.startsWith(baseSequence));
+
+    if (matched.length === 1 &&
+        sequence.length() === matched[0][0].length()) {
+      // keys are matched with an operation
       this.repository.clear();
-      return null;
-    } else if (matched.length > 1 ||
-      matched.length === 1 && sequence.length() < matched[0].length()) {
-      // More than one operations are matched
-      return null;
+      return [matched[0][1]];
+    } else if (
+      baseMatched.length === 1 &&
+        baseSequence.length() === baseMatched[0][0].length()) {
+      // keys are matched with an operation with a numeric prefix
+      this.repository.clear();
+      return Array(sequence.repeatCount()).fill(baseMatched[0][1]);
+    } else if (matched.length >= 1 || baseMatched.length >= 1) {
+      // keys are matched with an operation's prefix
+      return [];
     }
-    // Exactly one operation is matched
-    let operation = keymaps.get(matched[0]) as operations.Operation;
-    this.repository.clear();
-    return operation;
-  }
 
-  clear(): void {
+    // matched with no operations
     this.repository.clear();
+    return [];
   }
 
-  private keymapEntityMap(): KeymapEntityMap {
+  private keymapEntityMap(): [KeySequence, operations.Operation][] {
     let keymaps = this.settingRepository.get().keymaps.combine(reservedKeymaps);
     let entries = keymaps.entries().map(
       ([keys, op]) => [KeySequence.fromMapKeys(keys), op]
     ) as [KeySequence, operations.Operation][];
-    return new Map<KeySequence, operations.Operation>(entries);
+    if (!this.addonEnabledRepository.get()) {
+      // available keymaps are only ADDON_ENABLE and ADDON_TOGGLE_ENABLED if
+      // the addon disabled
+      entries = entries.filter(
+        ([_seq, { type }]) => enableAddonOps.includes(type)
+      );
+    }
+    return entries;
   }
 
   private blacklistKey(key: Key): boolean {