Update MotionQuoteAction.kt to support iq and aq motions

JetBrains · Jan 8, 2025 · f0256ac · f0256ac
1 parent 1b030a6
commit f0256ac
Showing 1 changed file with 55 additions and 0 deletions.
diff --git a/vim-engine/src/main/kotlin/com/maddyhome/idea/vim/action/motion/object/MotionQuoteAction.kt b/vim-engine/src/main/kotlin/com/maddyhome/idea/vim/action/motion/object/MotionQuoteAction.kt
@@ -111,6 +111,61 @@ class MotionOuterBlockDoubleQuoteAction : TextObjectActionHandler() {
   }
 }
 
+@CommandOrMotion(keys = ["iq"], modes = [Mode.VISUAL, Mode.OP_PENDING])
+class MotionInnerAnyQuoteProximityAction : TextObjectActionHandler() {
+
+  override val flags: EnumSet<CommandFlags> = enumSetOf(CommandFlags.FLAG_TEXT_BLOCK)
+
+  override val visualType: TextObjectVisualType = TextObjectVisualType.CHARACTER_WISE
+
+  override fun getRange(
+    editor: VimEditor,
+    caret: ImmutableVimCaret,
+    context: ExecutionContext,
+    count: Int,
+    rawCount: Int,
+  ): TextRange? {
+    return listOf('`', '"', '\'')
+      .mapNotNull { quote ->
+        injector.searchHelper.findBlockQuoteInLineRange(editor, caret, quote, false)?.let { range -> range to quote }
+      }
+      .minByOrNull { it.first.distanceTo(caret.offset) }?.first
+  }
+}
+
+@CommandOrMotion(keys = ["aq"], modes = [Mode.VISUAL, Mode.OP_PENDING])
+class MotionOuterAnyQuoteProximityAction : TextObjectActionHandler() {
+
+  override val flags: EnumSet<CommandFlags> = enumSetOf(CommandFlags.FLAG_TEXT_BLOCK)
+
+  override val visualType: TextObjectVisualType = TextObjectVisualType.CHARACTER_WISE
+
+  override fun getRange(
+    editor: VimEditor,
+    caret: ImmutableVimCaret,
+    context: ExecutionContext,
+    count: Int,
+    rawCount: Int,
+  ): TextRange? {
+    return listOf('`', '"', '\'')
+      .mapNotNull { quote ->
+        injector.searchHelper.findBlockQuoteInLineRange(editor, caret, quote, true)?.let { range -> range to quote }
+      }
+      .minByOrNull { it.first.distanceTo(caret.offset) }?.first
+  }
+}
+
+// Extension function to calculate distance
+private fun TextRange.distanceTo(caretOffset: Int): Int {
+  return if (caretOffset < startOffset) {
+    startOffset - caretOffset
+  } else if (caretOffset > endOffset) {
+    caretOffset - endOffset
+  } else {
+    0 // Caret is inside the range
+  }
+}
+
 @CommandOrMotion(keys = ["a'"], modes = [Mode.VISUAL, Mode.OP_PENDING])
 class MotionOuterBlockSingleQuoteAction : TextObjectActionHandler() {