mirror of
				https://github.com/chylex/IntelliJ-AceJump.git
				synced 2025-11-04 00:40:12 +01:00 
			
		
		
		
	Compare commits
	
		
			21 Commits
		
	
	
		
			35003b0bab
			...
			experiment
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						
						
							
						
						9a435feccc
	
				 | 
					
					
						|||
| 
						
						
							
						
						c1feb891e4
	
				 | 
					
					
						|||
| 
						
						
							
						
						0b6ba62cda
	
				 | 
					
					
						|||
| 
						
						
							
						
						dfd5b122a0
	
				 | 
					
					
						|||
| 
						
						
							
						
						f2400d134d
	
				 | 
					
					
						|||
| 
						
						
							
						
						98997b4d86
	
				 | 
					
					
						|||
| 
						
						
							
						
						1e172e9c49
	
				 | 
					
					
						|||
| 01b37c878e | |||
| 1630c706a9 | |||
| 041a130c5f | |||
| 7561bfde36 | |||
| 724e469f21 | |||
| bce9a5f636 | |||
| bfe0aa536e | |||
| 2ca21b8423 | |||
| 430bcf6883 | |||
| 092650af81 | |||
| 2009123114 | |||
| de19c0e0b8 | |||
| 5021a07fb9 | |||
| 1180547b06 | 
@@ -1,53 +1,19 @@
 | 
			
		||||
import org.jetbrains.changelog.closure
 | 
			
		||||
import org.jetbrains.intellij.tasks.PatchPluginXmlTask
 | 
			
		||||
import org.jetbrains.intellij.tasks.PublishTask
 | 
			
		||||
import org.jetbrains.intellij.tasks.RunIdeTask
 | 
			
		||||
import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 | 
			
		||||
 | 
			
		||||
plugins {
 | 
			
		||||
  idea apply true
 | 
			
		||||
  kotlin("jvm") version "1.3.72"
 | 
			
		||||
  kotlin("jvm") version "1.5.0"
 | 
			
		||||
  id("org.jetbrains.intellij") version "0.7.2"
 | 
			
		||||
  id("org.jetbrains.changelog") version "1.1.2"
 | 
			
		||||
  id("com.github.ben-manes.versions") version "0.38.0"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
tasks {
 | 
			
		||||
  withType<KotlinCompile> {
 | 
			
		||||
    kotlinOptions.jvmTarget = JavaVersion.VERSION_1_8.toString()
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  named<Zip>("buildPlugin") {
 | 
			
		||||
    dependsOn("test")
 | 
			
		||||
    archiveFileName.set("AceJump.zip")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  withType<RunIdeTask> {
 | 
			
		||||
    dependsOn("test")
 | 
			
		||||
    findProperty("luginDev")?.let { args = listOf(projectDir.absolutePath) }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  withType<PublishTask> {
 | 
			
		||||
    val intellijPublishToken: String? by project
 | 
			
		||||
    token(intellijPublishToken)
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  withType<PatchPluginXmlTask> {
 | 
			
		||||
    sinceBuild("201.6668.0")
 | 
			
		||||
    changeNotes({ changelog.getLatest().toHTML() })
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
changelog {
 | 
			
		||||
  path = "${project.projectDir}/CHANGES.md"
 | 
			
		||||
  header = closure { "${project.version}" }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
dependencies {
 | 
			
		||||
  // gradle-intellij-plugin doesn't attach sources properly for Kotlin :(
 | 
			
		||||
  compileOnly(kotlin("stdlib-jdk8"))
 | 
			
		||||
  // https://github.com/promeG/TinyPinyin
 | 
			
		||||
  implementation("com.github.promeg:tinypinyin:2.0.3")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
repositories {
 | 
			
		||||
@@ -56,11 +22,11 @@ repositories {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
intellij {
 | 
			
		||||
  version = "2020.2"
 | 
			
		||||
  version = "2021.1"
 | 
			
		||||
  pluginName = "AceJump"
 | 
			
		||||
  updateSinceUntilBuild = false
 | 
			
		||||
  setPlugins("java")
 | 
			
		||||
  setPlugins("java", "IdeaVIM:0.66")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
group = "org.acejump"
 | 
			
		||||
version = "3.7"
 | 
			
		||||
version = "chylex-8"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										
											BIN
										
									
								
								gradle/wrapper/gradle-wrapper.jar
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										
											BIN
										
									
								
								gradle/wrapper/gradle-wrapper.jar
									
									
									
									
										vendored
									
									
								
							
										
											Binary file not shown.
										
									
								
							
							
								
								
									
										2
									
								
								gradle/wrapper/gradle-wrapper.properties
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								gradle/wrapper/gradle-wrapper.properties
									
									
									
									
										vendored
									
									
								
							@@ -1,5 +1,5 @@
 | 
			
		||||
distributionBase=GRADLE_USER_HOME
 | 
			
		||||
distributionPath=wrapper/dists
 | 
			
		||||
distributionUrl=https\://services.gradle.org/distributions/gradle-6.8-bin.zip
 | 
			
		||||
distributionUrl=https\://services.gradle.org/distributions/gradle-7.1-bin.zip
 | 
			
		||||
zipStoreBase=GRADLE_USER_HOME
 | 
			
		||||
zipStorePath=wrapper/dists
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										2
									
								
								gradlew
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								gradlew
									
									
									
									
										vendored
									
									
								
							@@ -72,7 +72,7 @@ case "`uname`" in
 | 
			
		||||
  Darwin* )
 | 
			
		||||
    darwin=true
 | 
			
		||||
    ;;
 | 
			
		||||
  MINGW* )
 | 
			
		||||
  MSYS* | MINGW* )
 | 
			
		||||
    msys=true
 | 
			
		||||
    ;;
 | 
			
		||||
  NONSTOP* )
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,8 @@
 | 
			
		||||
package org.acejump
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.openapi.editor.actions.EditorActionUtil
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.IntArrayList
 | 
			
		||||
 | 
			
		||||
annotation class ExternalUsage
 | 
			
		||||
 | 
			
		||||
@@ -36,7 +38,7 @@ fun CharSequence.countMatchingCharacters(selfOffset: Int, otherText: String): In
 | 
			
		||||
 * Determines which characters form a "word" for the purposes of functions below.
 | 
			
		||||
 */
 | 
			
		||||
val Char.isWordPart
 | 
			
		||||
  get() = this.isJavaIdentifierPart()
 | 
			
		||||
  get() = this in 'a'..'z' || this.isJavaIdentifierPart()
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Finds index of the first character in a word.
 | 
			
		||||
@@ -56,8 +58,35 @@ inline fun CharSequence.wordStart(pos: Int, isPartOfWord: (Char) -> Boolean = Ch
 | 
			
		||||
 */
 | 
			
		||||
inline fun CharSequence.wordEnd(pos: Int, isPartOfWord: (Char) -> Boolean = Char::isWordPart): Int {
 | 
			
		||||
  var end = pos
 | 
			
		||||
  val limit = length - 1
 | 
			
		||||
  
 | 
			
		||||
  while (end < length - 1 && isPartOfWord(this[end + 1])) {
 | 
			
		||||
  while (end < limit && isPartOfWord(this[end + 1])) {
 | 
			
		||||
    ++end
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  return end
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Finds index of the previous "camelHumps" hump in a word.
 | 
			
		||||
 */
 | 
			
		||||
inline fun CharSequence.humpStart(pos: Int, isPartOfWord: (Char) -> Boolean = Char::isWordPart): Int {
 | 
			
		||||
  var start = pos
 | 
			
		||||
  
 | 
			
		||||
  while (start > 0 && isPartOfWord(this[start - 1]) && !EditorActionUtil.isHumpBound(this, start, true)) {
 | 
			
		||||
    --start
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  return start
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Finds index of the next "camelHumps" hump in a word.
 | 
			
		||||
 */
 | 
			
		||||
inline fun CharSequence.humpEnd(pos: Int, isPartOfWord: (Char) -> Boolean = Char::isWordPart): Int {
 | 
			
		||||
  var end = pos
 | 
			
		||||
  
 | 
			
		||||
  while (end < length - 1 && isPartOfWord(this[end + 1]) && !EditorActionUtil.isHumpBound(this, end + 1, false)) {
 | 
			
		||||
    ++end
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
@@ -69,14 +98,25 @@ inline fun CharSequence.wordEnd(pos: Int, isPartOfWord: (Char) -> Boolean = Char
 | 
			
		||||
 */
 | 
			
		||||
inline fun CharSequence.wordEndPlus(pos: Int, isPartOfWord: (Char) -> Boolean = Char::isWordPart): Int {
 | 
			
		||||
  var end = this.wordEnd(pos, isPartOfWord)
 | 
			
		||||
  val limit = length - 1
 | 
			
		||||
  
 | 
			
		||||
  while (end < length - 1 && !isPartOfWord(this[end + 1])) {
 | 
			
		||||
  while (end < limit && !isPartOfWord(this[end + 1])) {
 | 
			
		||||
    ++end
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  if (end < length - 1 && isPartOfWord(this[end + 1])) {
 | 
			
		||||
  if (end < limit && isPartOfWord(this[end + 1])) {
 | 
			
		||||
    ++end
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  return end
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fun MutableMap<Editor, IntArrayList>.clone(): MutableMap<Editor, IntArrayList> {
 | 
			
		||||
  val clone = HashMap<Editor, IntArrayList>(size)
 | 
			
		||||
  
 | 
			
		||||
  for ((editor, offsets) in this) {
 | 
			
		||||
    clone[editor] = offsets.clone()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  return clone
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,71 +0,0 @@
 | 
			
		||||
package org.acejump.action
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.actionSystem.AnActionEvent
 | 
			
		||||
import com.intellij.openapi.actionSystem.CommonDataKeys.EDITOR
 | 
			
		||||
import com.intellij.openapi.project.DumbAwareAction
 | 
			
		||||
import org.acejump.boundaries.Boundaries
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries
 | 
			
		||||
import org.acejump.input.JumpMode
 | 
			
		||||
import org.acejump.search.Pattern
 | 
			
		||||
import org.acejump.session.Session
 | 
			
		||||
import org.acejump.session.SessionManager
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Base class for keyboard-activated actions that create or update an AceJump [Session].
 | 
			
		||||
 */
 | 
			
		||||
sealed class AceAction : DumbAwareAction() {
 | 
			
		||||
  final override fun update(action: AnActionEvent) {
 | 
			
		||||
    action.presentation.isEnabled = action.getData(EDITOR) != null
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  final override fun actionPerformed(e: AnActionEvent) {
 | 
			
		||||
    invoke(SessionManager.start(e.getData(EDITOR) ?: return))
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  abstract operator fun invoke(session: Session)
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Generic action type that toggles a specific [JumpMode].
 | 
			
		||||
   */
 | 
			
		||||
  abstract class BaseToggleJumpModeAction(private val mode: JumpMode) : AceAction() {
 | 
			
		||||
    final override fun invoke(session: Session) = session.toggleJumpMode(mode)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Generic action type that starts a regex search.
 | 
			
		||||
   */
 | 
			
		||||
  abstract class BaseRegexSearchAction(private val pattern: Pattern, private val boundaries: Boundaries) : AceAction() {
 | 
			
		||||
    override fun invoke(session: Session) = session.startRegexSearch(pattern, boundaries)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Initiates an AceJump session in the first [JumpMode], or cycles to the next [JumpMode] as defined in configuration.
 | 
			
		||||
   */
 | 
			
		||||
  object ActivateOrCycleMode : AceAction() {
 | 
			
		||||
    override fun invoke(session: Session) = session.cycleNextJumpMode()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Initiates an AceJump session in the last [JumpMode], or cycles to the previous [JumpMode] as defined in configuration.
 | 
			
		||||
   */
 | 
			
		||||
  object ActivateOrReverseCycleMode : AceAction() {
 | 
			
		||||
    override fun invoke(session: Session) = session.cyclePreviousJumpMode()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  // @formatter:off
 | 
			
		||||
  
 | 
			
		||||
  object ToggleJumpMode        : BaseToggleJumpModeAction(JumpMode.JUMP)
 | 
			
		||||
  object ToggleJumpEndMode     : BaseToggleJumpModeAction(JumpMode.JUMP_END)
 | 
			
		||||
  object ToggleTargetMode      : BaseToggleJumpModeAction(JumpMode.TARGET)
 | 
			
		||||
  object ToggleDeclarationMode : BaseToggleJumpModeAction(JumpMode.DEFINE)
 | 
			
		||||
  
 | 
			
		||||
  object StartAllWordsMode          : BaseRegexSearchAction(Pattern.ALL_WORDS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
  object StartAllWordsBackwardsMode : BaseRegexSearchAction(Pattern.ALL_WORDS, StandardBoundaries.BEFORE_CARET)
 | 
			
		||||
  object StartAllWordsForwardMode   : BaseRegexSearchAction(Pattern.ALL_WORDS, StandardBoundaries.AFTER_CARET)
 | 
			
		||||
  object StartAllLineStartsMode     : BaseRegexSearchAction(Pattern.LINE_STARTS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
  object StartAllLineEndsMode       : BaseRegexSearchAction(Pattern.LINE_ENDS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
  object StartAllLineIndentsMode    : BaseRegexSearchAction(Pattern.LINE_INDENTS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
  object StartAllLineMarksMode      : BaseRegexSearchAction(Pattern.LINE_ALL_MARKS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
  
 | 
			
		||||
  // @formatter:on
 | 
			
		||||
}
 | 
			
		||||
@@ -4,7 +4,6 @@ import com.intellij.openapi.actionSystem.DataContext
 | 
			
		||||
import com.intellij.openapi.editor.Caret
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.openapi.editor.actionSystem.EditorActionHandler
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries
 | 
			
		||||
import org.acejump.search.Pattern
 | 
			
		||||
import org.acejump.session.Session
 | 
			
		||||
import org.acejump.session.SessionManager
 | 
			
		||||
@@ -12,7 +11,7 @@ import org.acejump.session.SessionManager
 | 
			
		||||
/**
 | 
			
		||||
 * Base class for keyboard-activated overrides of existing editor actions, that have a different meaning during an AceJump [Session].
 | 
			
		||||
 */
 | 
			
		||||
sealed class AceEditorAction(private val originalHandler: EditorActionHandler) : EditorActionHandler() {
 | 
			
		||||
abstract class AceEditorAction(private val originalHandler: EditorActionHandler) : EditorActionHandler() {
 | 
			
		||||
  final override fun isEnabledForCaret(editor: Editor, caret: Caret, dataContext: DataContext?): Boolean {
 | 
			
		||||
    return SessionManager[editor] != null || originalHandler.isEnabled(editor, caret, dataContext)
 | 
			
		||||
  }
 | 
			
		||||
@@ -40,23 +39,19 @@ sealed class AceEditorAction(private val originalHandler: EditorActionHandler) :
 | 
			
		||||
    override fun run(session: Session) = session.restart()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class SelectBackward(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
 | 
			
		||||
    override fun run(session: Session) = session.visitPreviousTag()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class SelectForward(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
 | 
			
		||||
    override fun run(session: Session) = session.visitNextTag()
 | 
			
		||||
  class TagImmediately(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
 | 
			
		||||
    override fun run(session: Session) = session.tagImmediately()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class SearchLineStarts(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
 | 
			
		||||
    override fun run(session: Session) = session.startRegexSearch(Pattern.LINE_STARTS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
    override fun run(session: Session) = session.startRegexSearch(Pattern.LINE_STARTS)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class SearchLineEnds(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
 | 
			
		||||
    override fun run(session: Session) = session.startRegexSearch(Pattern.LINE_ENDS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
    override fun run(session: Session) = session.startRegexSearch(Pattern.LINE_ENDS)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class SearchLineIndents(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
 | 
			
		||||
    override fun run(session: Session) = session.startRegexSearch(Pattern.LINE_INDENTS, StandardBoundaries.WHOLE_FILE)
 | 
			
		||||
    override fun run(session: Session) = session.startRegexSearch(Pattern.LINE_INDENTS)
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										81
									
								
								src/main/kotlin/org/acejump/action/AceKeyboardAction.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										81
									
								
								src/main/kotlin/org/acejump/action/AceKeyboardAction.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,81 @@
 | 
			
		||||
package org.acejump.action
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.actionSystem.AnActionEvent
 | 
			
		||||
import com.intellij.openapi.actionSystem.CommonDataKeys.EDITOR
 | 
			
		||||
import com.intellij.openapi.fileEditor.TextEditor
 | 
			
		||||
import com.intellij.openapi.fileEditor.ex.FileEditorManagerEx
 | 
			
		||||
import com.intellij.openapi.project.DumbAwareAction
 | 
			
		||||
import com.intellij.util.IncorrectOperationException
 | 
			
		||||
import org.acejump.boundaries.Boundaries
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries.*
 | 
			
		||||
import org.acejump.search.Pattern
 | 
			
		||||
import org.acejump.session.Session
 | 
			
		||||
import org.acejump.session.SessionManager
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Base class for keyboard-activated actions that create or update an AceJump [Session].
 | 
			
		||||
 */
 | 
			
		||||
abstract class AceKeyboardAction : DumbAwareAction() {
 | 
			
		||||
  final override fun update(action: AnActionEvent) {
 | 
			
		||||
    action.presentation.isEnabled = action.getData(EDITOR) != null
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  final override fun actionPerformed(e: AnActionEvent) {
 | 
			
		||||
    val editor = e.getData(EDITOR) ?: return
 | 
			
		||||
    val project = e.project
 | 
			
		||||
    
 | 
			
		||||
    if (project != null) {
 | 
			
		||||
      try {
 | 
			
		||||
        val openEditors = FileEditorManagerEx.getInstanceEx(project)
 | 
			
		||||
          .splitters
 | 
			
		||||
          .selectedEditors
 | 
			
		||||
          .mapNotNull { (it as? TextEditor)?.editor }
 | 
			
		||||
          .sortedBy { if (it === editor) 0 else 1 }
 | 
			
		||||
        invoke(SessionManager.start(editor, openEditors))
 | 
			
		||||
      } catch (e: IncorrectOperationException) {
 | 
			
		||||
        invoke(SessionManager.start(editor))
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    else {
 | 
			
		||||
      invoke(SessionManager.start(editor))
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  abstract operator fun invoke(session: Session)
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Generic action type that starts a regex search.
 | 
			
		||||
   */
 | 
			
		||||
  abstract class BaseRegexSearchAction(private val pattern: Pattern, private val boundaries: Boundaries) : AceKeyboardAction() {
 | 
			
		||||
    override fun invoke(session: Session) {
 | 
			
		||||
      session.defaultBoundary = boundaries
 | 
			
		||||
      session.startRegexSearch(pattern)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Starts or ends an AceJump session in quick jump mode.
 | 
			
		||||
   */
 | 
			
		||||
  object ActivateAceJump : AceKeyboardAction() {
 | 
			
		||||
    override fun invoke(session: Session) = session.startJumpMode()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Starts or cycles main AceJump modes.
 | 
			
		||||
   */
 | 
			
		||||
  object ActivateAceJumpSpecial : AceKeyboardAction() {
 | 
			
		||||
    override fun invoke(session: Session) = session.startOrCycleSpecialModes()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  // @formatter:off
 | 
			
		||||
  
 | 
			
		||||
  object StartAllWordsMode          : BaseRegexSearchAction(Pattern.ALL_WORDS, VISIBLE_ON_SCREEN)
 | 
			
		||||
  object StartAllWordsBackwardsMode : BaseRegexSearchAction(Pattern.ALL_WORDS, BEFORE_CARET.intersection(VISIBLE_ON_SCREEN))
 | 
			
		||||
  object StartAllWordsForwardMode   : BaseRegexSearchAction(Pattern.ALL_WORDS, AFTER_CARET.intersection(VISIBLE_ON_SCREEN))
 | 
			
		||||
  object StartAllLineStartsMode     : BaseRegexSearchAction(Pattern.LINE_STARTS, VISIBLE_ON_SCREEN)
 | 
			
		||||
  object StartAllLineEndsMode       : BaseRegexSearchAction(Pattern.LINE_ENDS, VISIBLE_ON_SCREEN)
 | 
			
		||||
  object StartAllLineIndentsMode    : BaseRegexSearchAction(Pattern.LINE_INDENTS, VISIBLE_ON_SCREEN)
 | 
			
		||||
  object StartAllLineMarksMode      : BaseRegexSearchAction(Pattern.LINE_ALL_MARKS, VISIBLE_ON_SCREEN)
 | 
			
		||||
  
 | 
			
		||||
  // @formatter:on
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										484
									
								
								src/main/kotlin/org/acejump/action/AceTagAction.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										484
									
								
								src/main/kotlin/org/acejump/action/AceTagAction.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,484 @@
 | 
			
		||||
package org.acejump.action
 | 
			
		||||
 | 
			
		||||
import com.intellij.find.actions.ShowUsagesAction
 | 
			
		||||
import com.intellij.openapi.actionSystem.ActionManager
 | 
			
		||||
import com.intellij.openapi.actionSystem.IdeActions
 | 
			
		||||
import com.intellij.openapi.application.ApplicationManager
 | 
			
		||||
import com.intellij.openapi.command.CommandProcessor
 | 
			
		||||
import com.intellij.openapi.command.UndoConfirmationPolicy
 | 
			
		||||
import com.intellij.openapi.command.WriteCommandAction
 | 
			
		||||
import com.intellij.openapi.editor.CaretState
 | 
			
		||||
import com.intellij.openapi.editor.Document
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.openapi.editor.LogicalPosition
 | 
			
		||||
import com.intellij.openapi.editor.actionSystem.DocCommandGroupId
 | 
			
		||||
import com.intellij.openapi.editor.actions.EditorActionUtil
 | 
			
		||||
import com.intellij.openapi.fileEditor.TextEditor
 | 
			
		||||
import com.intellij.openapi.fileEditor.ex.FileEditorManagerEx
 | 
			
		||||
import com.intellij.openapi.fileEditor.ex.IdeDocumentHistory
 | 
			
		||||
import com.intellij.openapi.project.Project
 | 
			
		||||
import com.intellij.openapi.ui.playback.commands.ActionCommand
 | 
			
		||||
import com.intellij.openapi.util.TextRange
 | 
			
		||||
import com.intellij.psi.PsiDocumentManager
 | 
			
		||||
import com.intellij.psi.codeStyle.CodeStyleManager
 | 
			
		||||
import org.acejump.*
 | 
			
		||||
import org.acejump.search.SearchProcessor
 | 
			
		||||
import kotlin.math.max
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Base class for actions available after typing a tag.
 | 
			
		||||
 */
 | 
			
		||||
sealed class AceTagAction {
 | 
			
		||||
  abstract operator fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean)
 | 
			
		||||
  
 | 
			
		||||
  abstract class BaseJumpAction : AceTagAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
      val caretModel = editor.caretModel
 | 
			
		||||
      val oldCarets = if (shiftMode) caretModel.caretsAndSelections else emptyList()
 | 
			
		||||
      
 | 
			
		||||
      recordCaretPosition(editor)
 | 
			
		||||
      
 | 
			
		||||
      if (isFinal) {
 | 
			
		||||
        ensureEditorFocused(editor)
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      moveCaretTo(editor, getCaretOffset(editor, searchProcessor, offset))
 | 
			
		||||
      
 | 
			
		||||
      if (shiftMode) {
 | 
			
		||||
        caretModel.caretsAndSelections = oldCarets + caretModel.caretsAndSelections
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    abstract fun getCaretOffset(editor: Editor, searchProcessor: SearchProcessor, offset: Int): Int
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  abstract class BaseWordAction : BaseJumpAction() {
 | 
			
		||||
    final override fun getCaretOffset(editor: Editor, searchProcessor: SearchProcessor, offset: Int): Int {
 | 
			
		||||
      val matchingChars = countMatchingCharacters(editor, searchProcessor, offset)
 | 
			
		||||
      val targetOffset = offset + matchingChars
 | 
			
		||||
      val isInsideWord = matchingChars > 0 && editor.immutableText.let { it[targetOffset - 1].isWordPart && it[targetOffset].isWordPart }
 | 
			
		||||
      
 | 
			
		||||
      return getCaretOffset(editor, offset, targetOffset, isInsideWord)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    abstract fun getCaretOffset(editor: Editor, queryStartOffset: Int, queryEndOffset: Int, isInsideWord: Boolean): Int
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  abstract class BaseSelectAction : AceTagAction() {
 | 
			
		||||
    final override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
      if (shiftMode) {
 | 
			
		||||
        val caretModel = editor.caretModel
 | 
			
		||||
        val oldCarets = caretModel.caretsAndSelections
 | 
			
		||||
        val oldOffsetPosition = caretModel.logicalPosition
 | 
			
		||||
        
 | 
			
		||||
        invoke(editor, searchProcessor, offset)
 | 
			
		||||
        
 | 
			
		||||
        if (caretModel.caretsAndSelections.any { isSelectionOverlapping(oldOffsetPosition, it) }) {
 | 
			
		||||
          oldCarets.removeAll { isSelectionOverlapping(oldOffsetPosition, it) }
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        caretModel.caretsAndSelections = oldCarets + caretModel.caretsAndSelections
 | 
			
		||||
      }
 | 
			
		||||
      else {
 | 
			
		||||
        invoke(editor, searchProcessor, offset)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun isSelectionOverlapping(offset: LogicalPosition, oldCaret: CaretState): Boolean {
 | 
			
		||||
      return oldCaret.caretPosition == offset || oldCaret.selectionStart == offset || oldCaret.selectionEnd == offset
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    protected abstract operator fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  abstract class BasePerCaretWriteAction(private val selector: AceTagAction) : AceTagAction() {
 | 
			
		||||
    final override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
      val oldCarets = editor.caretModel.caretsAndSelections
 | 
			
		||||
      selector(editor, searchProcessor, offset, shiftMode = false, isFinal = isFinal)
 | 
			
		||||
      val range = editor.selectionModel.let { TextRange(it.selectionStart, it.selectionEnd) }
 | 
			
		||||
      
 | 
			
		||||
      editor.caretModel.caretsAndSelections = oldCarets
 | 
			
		||||
      invoke(editor, range, shiftMode)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    protected abstract operator fun invoke(editor: Editor, range: TextRange, shiftMode: Boolean)
 | 
			
		||||
    
 | 
			
		||||
    protected fun insertAtCarets(editor: Editor, text: String) {
 | 
			
		||||
      val document = editor.document
 | 
			
		||||
      
 | 
			
		||||
      editor.caretModel.runForEachCaret {
 | 
			
		||||
        if (it.hasSelection()) {
 | 
			
		||||
          document.replaceString(it.selectionStart, it.selectionEnd, text)
 | 
			
		||||
          fixIndents(editor, it.selectionStart, it.selectionEnd)
 | 
			
		||||
        }
 | 
			
		||||
        else {
 | 
			
		||||
          document.insertString(it.offset, text)
 | 
			
		||||
          fixIndents(editor, it.offset, it.offset + text.length)
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun fixIndents(editor: Editor, startOffset: Int, endOffset: Int) {
 | 
			
		||||
      val project = editor.project ?: return
 | 
			
		||||
      val document = editor.document
 | 
			
		||||
      val documentManager = PsiDocumentManager.getInstance(project)
 | 
			
		||||
      
 | 
			
		||||
      documentManager.commitAllDocuments()
 | 
			
		||||
      
 | 
			
		||||
      val file = documentManager.getPsiFile(document) ?: return
 | 
			
		||||
      val text = document.charsSequence
 | 
			
		||||
      
 | 
			
		||||
      if (startOffset > 0 && endOffset > startOffset + 1 && text[endOffset - 1] == '\n' && text[startOffset - 1] == '\n') {
 | 
			
		||||
        CodeStyleManager.getInstance(project).adjustLineIndent(file, TextRange(startOffset, endOffset - 1))
 | 
			
		||||
      }
 | 
			
		||||
      else {
 | 
			
		||||
        CodeStyleManager.getInstance(project).adjustLineIndent(file, TextRange(startOffset, endOffset))
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private companion object {
 | 
			
		||||
    fun countMatchingCharacters(editor: Editor, searchProcessor: SearchProcessor, offset: Int): Int {
 | 
			
		||||
      return editor.immutableText.countMatchingCharacters(offset, searchProcessor.query.rawText)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fun recordCaretPosition(editor: Editor) = with(editor) {
 | 
			
		||||
      project?.let { addCurrentPositionToHistory(it, document) }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fun moveCaretTo(editor: Editor, offset: Int) = with(editor) {
 | 
			
		||||
      selectionModel.removeSelection(true)
 | 
			
		||||
      caretModel.removeSecondaryCarets()
 | 
			
		||||
      caretModel.moveToOffset(offset)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fun selectRange(editor: Editor, fromOffset: Int, toOffset: Int, cursorOffset: Int = toOffset) = with(editor) {
 | 
			
		||||
      selectionModel.removeSelection(true)
 | 
			
		||||
      selectionModel.setSelection(fromOffset, toOffset)
 | 
			
		||||
      caretModel.moveToOffset(cursorOffset)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fun performAction(actionName: String) {
 | 
			
		||||
      val actionManager = ActionManager.getInstance()
 | 
			
		||||
      val action = actionManager.getAction(actionName)
 | 
			
		||||
      if (action != null) {
 | 
			
		||||
        actionManager.tryToExecute(action, ActionCommand.getInputEvent(null), null, null, true)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fun ensureEditorFocused(editor: Editor) {
 | 
			
		||||
      val project = editor.project ?: return
 | 
			
		||||
      val fem = FileEditorManagerEx.getInstanceEx(project)
 | 
			
		||||
      
 | 
			
		||||
      val window = fem.windows.firstOrNull { (it.selectedEditor?.selectedWithProvider?.fileEditor as? TextEditor)?.editor === editor }
 | 
			
		||||
      if (window != null && window !== fem.currentWindow) {
 | 
			
		||||
        fem.currentWindow = window
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun addCurrentPositionToHistory(project: Project, document: Document) {
 | 
			
		||||
      CommandProcessor.getInstance().executeCommand(project, {
 | 
			
		||||
        with(IdeDocumentHistory.getInstance(project)) {
 | 
			
		||||
          setCurrentCommandHasMoves()
 | 
			
		||||
          includeCurrentCommandAsNavigation()
 | 
			
		||||
          includeCurrentPlaceAsChangePlace()
 | 
			
		||||
        }
 | 
			
		||||
      }, "AceJumpHistoryAppender", DocCommandGroupId.noneGroupId(document), UndoConfirmationPolicy.DO_NOT_REQUEST_CONFIRMATION, document)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  // Actions
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the first character of the search query.
 | 
			
		||||
   * On shift action, adds the new caret to existing carets.
 | 
			
		||||
   */
 | 
			
		||||
  object JumpToSearchStart : BaseJumpAction() {
 | 
			
		||||
    override fun getCaretOffset(editor: Editor, searchProcessor: SearchProcessor, offset: Int): Int {
 | 
			
		||||
      return offset
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the last character of the search query.
 | 
			
		||||
   * On shift action, adds the new caret to existing carets.
 | 
			
		||||
   */
 | 
			
		||||
  object JumpToSearchEnd : BaseJumpAction() {
 | 
			
		||||
    override fun getCaretOffset(editor: Editor, searchProcessor: SearchProcessor, offset: Int): Int {
 | 
			
		||||
      return offset + max(0, countMatchingCharacters(editor, searchProcessor, offset) - 1)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret just past the last character of the search query.
 | 
			
		||||
   * On shift action, adds the new caret to existing carets.
 | 
			
		||||
   */
 | 
			
		||||
  object JumpPastSearchEnd : BaseJumpAction() {
 | 
			
		||||
    override fun getCaretOffset(editor: Editor, searchProcessor: SearchProcessor, offset: Int): Int {
 | 
			
		||||
      return offset + countMatchingCharacters(editor, searchProcessor, offset)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the start of a word. Word detection uses [Character.isJavaIdentifierPart] to count some special
 | 
			
		||||
   * characters, such as underscores, as part of a word. If there is no word at the last character of the search query, then the caret is
 | 
			
		||||
   * placed at the first character of the search query.
 | 
			
		||||
   *
 | 
			
		||||
   * On shift action, adds the new caret to existing carets.
 | 
			
		||||
   */
 | 
			
		||||
  object JumpToWordStart : BaseWordAction() {
 | 
			
		||||
    override fun getCaretOffset(editor: Editor, queryStartOffset: Int, queryEndOffset: Int, isInsideWord: Boolean): Int {
 | 
			
		||||
      return if (isInsideWord)
 | 
			
		||||
        editor.immutableText.wordStart(queryEndOffset)
 | 
			
		||||
      else
 | 
			
		||||
        queryStartOffset
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the end of a word. Word detection uses [Character.isJavaIdentifierPart] to count some special
 | 
			
		||||
   * characters, such as underscores, as part of a word. If there is no word at the last character of the search query, then the caret is
 | 
			
		||||
   * placed after the last character of the search query.
 | 
			
		||||
   *
 | 
			
		||||
   * On shift action, adds the new caret to existing carets.
 | 
			
		||||
   */
 | 
			
		||||
  object JumpToWordEnd : BaseWordAction() {
 | 
			
		||||
    override fun getCaretOffset(editor: Editor, queryStartOffset: Int, queryEndOffset: Int, isInsideWord: Boolean): Int {
 | 
			
		||||
      return if (isInsideWord)
 | 
			
		||||
        editor.immutableText.wordEnd(queryEndOffset) + 1
 | 
			
		||||
      else
 | 
			
		||||
        queryEndOffset
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the end of the line.
 | 
			
		||||
   * On shift action, adds the new caret to existing carets.
 | 
			
		||||
   */
 | 
			
		||||
  object JumpToLineEnd : BaseWordAction() {
 | 
			
		||||
    override fun getCaretOffset(editor: Editor, queryStartOffset: Int, queryEndOffset: Int, isInsideWord: Boolean): Int {
 | 
			
		||||
      val document = editor.document
 | 
			
		||||
      val line = document.getLineNumber(queryEndOffset)
 | 
			
		||||
      return document.getLineEndOffset(line)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, selects all characters covered by the search query.
 | 
			
		||||
   * On shift action, adds the new selection to existing selections.
 | 
			
		||||
   */
 | 
			
		||||
  object SelectQuery : BaseSelectAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int) {
 | 
			
		||||
      recordCaretPosition(editor)
 | 
			
		||||
      
 | 
			
		||||
      val startOffset = JumpToSearchStart.getCaretOffset(editor, searchProcessor, offset)
 | 
			
		||||
      val endOffset = JumpPastSearchEnd.getCaretOffset(editor, searchProcessor, offset)
 | 
			
		||||
      
 | 
			
		||||
      selectRange(editor, startOffset, endOffset)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the end of a word, and also selects the entire word. Word detection uses
 | 
			
		||||
   * [Character.isJavaIdentifierPart] to count some special characters, such as underscores, as part of a word. If there is no word at the
 | 
			
		||||
   * last character of the search query, then the caret is placed after the last character of the search query, and all text between the
 | 
			
		||||
   * start and end of the search query is selected.
 | 
			
		||||
   *
 | 
			
		||||
   * On shift action, adds the new selection to existing selections.
 | 
			
		||||
   */
 | 
			
		||||
  object SelectWord : BaseSelectAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int) {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      val queryEndOffset = JumpToSearchEnd.getCaretOffset(editor, searchProcessor, offset)
 | 
			
		||||
      
 | 
			
		||||
      if (chars[queryEndOffset].isWordPart) {
 | 
			
		||||
        recordCaretPosition(editor)
 | 
			
		||||
        
 | 
			
		||||
        val startOffset = JumpToWordStart.getCaretOffset(editor, offset, queryEndOffset, isInsideWord = true)
 | 
			
		||||
        val endOffset = JumpToWordEnd.getCaretOffset(editor, offset, queryEndOffset, isInsideWord = true)
 | 
			
		||||
        
 | 
			
		||||
        selectRange(editor, startOffset, endOffset)
 | 
			
		||||
      }
 | 
			
		||||
      else {
 | 
			
		||||
        SelectQuery(editor, searchProcessor, offset, shiftMode = false, isFinal = true)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the end of a camel hump inside a word, and also selects the hump. If there is no word at the
 | 
			
		||||
   * last character of the search query, then the search query is selected. See [SelectWord] and [SelectQuery] for details.
 | 
			
		||||
   *
 | 
			
		||||
   * On shift action, adds the new selection to existing selections.
 | 
			
		||||
   */
 | 
			
		||||
  object SelectHump : BaseSelectAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int) {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      val queryEndOffset = JumpToSearchEnd.getCaretOffset(editor, searchProcessor, offset)
 | 
			
		||||
      
 | 
			
		||||
      if (chars[queryEndOffset].isWordPart) {
 | 
			
		||||
        recordCaretPosition(editor)
 | 
			
		||||
        
 | 
			
		||||
        val startOffset = chars.humpStart(queryEndOffset)
 | 
			
		||||
        val endOffset = chars.humpEnd(queryEndOffset) + 1
 | 
			
		||||
        
 | 
			
		||||
        selectRange(editor, startOffset, endOffset)
 | 
			
		||||
      }
 | 
			
		||||
      else {
 | 
			
		||||
        SelectQuery(editor, searchProcessor, offset, shiftMode = false, isFinal = true)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, selects the line at the tag, excluding the indent.
 | 
			
		||||
   * On shift action, adds the new selection to existing selections.
 | 
			
		||||
   */
 | 
			
		||||
  object SelectLine : BaseSelectAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int) {
 | 
			
		||||
      JumpToSearchEnd(editor, searchProcessor, offset, shiftMode = false, isFinal = true)
 | 
			
		||||
      
 | 
			
		||||
      val document = editor.document
 | 
			
		||||
      val line = editor.caretModel.logicalPosition.line
 | 
			
		||||
      val lineStart = EditorActionUtil.findFirstNonSpaceOffsetOnTheLine(document, line)
 | 
			
		||||
      val lineEnd = document.getLineEndOffset(line)
 | 
			
		||||
      
 | 
			
		||||
      selectRange(editor, lineStart, lineEnd)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, places the caret at the last character of the search query, and then performs Extend Selection a set amount of
 | 
			
		||||
   * times.
 | 
			
		||||
   *
 | 
			
		||||
   * On shift action, adds the new selection to existing selections.
 | 
			
		||||
   */
 | 
			
		||||
  class SelectExtended(private val extendCount: Int) : BaseSelectAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int) {
 | 
			
		||||
      JumpToSearchEnd(editor, searchProcessor, offset, shiftMode = false, isFinal = true)
 | 
			
		||||
      
 | 
			
		||||
      repeat(extendCount) {
 | 
			
		||||
        performAction(IdeActions.ACTION_EDITOR_SELECT_WORD_AT_CARET)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, selects the range between the caret and a position decided by the provided [BaseJumpAction].
 | 
			
		||||
   * On shift action, adds the new selection to existing selections.
 | 
			
		||||
   */
 | 
			
		||||
  class SelectToCaret(private val jumper: BaseJumpAction) : BaseSelectAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int) {
 | 
			
		||||
      val caretModel = editor.caretModel
 | 
			
		||||
      val oldOffset = caretModel.offset
 | 
			
		||||
      val oldSelection = editor.selectionModel.takeIf { it.hasSelection(false) }?.let { it.selectionStart..it.selectionEnd }
 | 
			
		||||
  
 | 
			
		||||
      jumper(editor, searchProcessor, offset, shiftMode = false, isFinal = true)
 | 
			
		||||
      
 | 
			
		||||
      val newOffset = caretModel.offset
 | 
			
		||||
      
 | 
			
		||||
      if (oldSelection == null) {
 | 
			
		||||
        selectRange(editor, oldOffset, newOffset)
 | 
			
		||||
      }
 | 
			
		||||
      else {
 | 
			
		||||
        selectRange(editor, minOf(oldOffset, newOffset, oldSelection.first), maxOf(oldOffset, newOffset, oldSelection.last), newOffset)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, selects the range between [firstOffset] and a position decided by the provided [BaseJumpAction].
 | 
			
		||||
   * On shift action, adds the new selection to existing selections.
 | 
			
		||||
   */
 | 
			
		||||
  class SelectBetweenPoints(private val firstOffset: Int, private val secondOffsetJumper: BaseJumpAction) : BaseSelectAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int) {
 | 
			
		||||
      secondOffsetJumper(editor, searchProcessor, offset, shiftMode = false, isFinal = true)
 | 
			
		||||
      selectRange(editor, firstOffset, editor.caretModel.offset)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Selects text based on the provided [selector] action and clones it at every existing caret, selecting the cloned text. If a caret
 | 
			
		||||
   * has a selection, the selected text will be replaced.
 | 
			
		||||
   */
 | 
			
		||||
  class CloneToCaret(selector: AceTagAction) : BasePerCaretWriteAction(selector) {
 | 
			
		||||
    override fun invoke(editor: Editor, range: TextRange, shiftMode: Boolean) {
 | 
			
		||||
      WriteCommandAction.writeCommandAction(editor.project).withName("AceJump Clone").run<Throwable> {
 | 
			
		||||
        insertAtCarets(editor, editor.document.getText(range))
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Selects text based on the provided [selector] action and clones it to every existing caret, selecting the cloned text and deleting
 | 
			
		||||
   * the original. If a caret has a selection, the selected text will be replaced.
 | 
			
		||||
   */
 | 
			
		||||
  open class MoveToCaret(selector: AceTagAction) : BasePerCaretWriteAction(selector) {
 | 
			
		||||
    override fun invoke(editor: Editor, range: TextRange, shiftMode: Boolean) {
 | 
			
		||||
      val difference = if (shiftMode) editor.caretModel.caretsAndSelections.sumBy {
 | 
			
		||||
        val start = it.selectionStart?.let(editor::logicalPositionToOffset)
 | 
			
		||||
        val end = it.selectionEnd?.let(editor::logicalPositionToOffset)
 | 
			
		||||
        if (start == null || end == null || end > range.endOffset) 0 else range.length - (end - start)
 | 
			
		||||
      } else 0
 | 
			
		||||
      
 | 
			
		||||
      WriteCommandAction.writeCommandAction(editor.project).withName("AceJump Move").run<Throwable> {
 | 
			
		||||
        val document = editor.document
 | 
			
		||||
        val text = document.getText(range)
 | 
			
		||||
        document.deleteString(range.startOffset, range.endOffset)
 | 
			
		||||
        insertAtCarets(editor, text)
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      if (shiftMode) {
 | 
			
		||||
        editor.selectionModel.removeSelection(true)
 | 
			
		||||
        editor.caretModel.moveToOffset(range.startOffset + difference)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, performs the Go To Declaration action, available via `Navigate | Declaration or Usages`.
 | 
			
		||||
   * On shift action, performs the Go To Type Declaration action, available via `Navigate | Type Declaration`.
 | 
			
		||||
   * Always places the caret at the start of the word.
 | 
			
		||||
   */
 | 
			
		||||
  object GoToDeclaration : AceTagAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
      JumpToWordStart(editor, searchProcessor, offset, shiftMode = false, isFinal = isFinal)
 | 
			
		||||
      ApplicationManager.getApplication().invokeLater { performAction(if (shiftMode) IdeActions.ACTION_GOTO_TYPE_DECLARATION else IdeActions.ACTION_GOTO_DECLARATION) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, performs the Show Usages action, available via the context menu.
 | 
			
		||||
   * On shift action, performs the Find Usages action, available via the context menu.
 | 
			
		||||
   * Always places the caret at the start of the word.
 | 
			
		||||
   */
 | 
			
		||||
  object ShowUsages : AceTagAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
      JumpToWordStart(editor, searchProcessor, offset, shiftMode = false, isFinal = isFinal)
 | 
			
		||||
      ApplicationManager.getApplication().invokeLater { performAction(if (shiftMode) IdeActions.ACTION_FIND_USAGES else ShowUsagesAction.ID) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Performs the Show Context Actions action, available via the context menu or Alt+Enter.
 | 
			
		||||
   * Always places the caret at the start of the word.
 | 
			
		||||
   */
 | 
			
		||||
  object ShowIntentions : AceTagAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
      JumpToWordStart(editor, searchProcessor, offset, shiftMode = false, isFinal = isFinal)
 | 
			
		||||
      ApplicationManager.getApplication().invokeLater { performAction(IdeActions.ACTION_SHOW_INTENTION_ACTIONS) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default action, performs the Refactor This action, available via the main menu.
 | 
			
		||||
   * On shift action, performs the Rename... refactoring, available via the main menu.
 | 
			
		||||
   * Always places the caret at the start of the word.
 | 
			
		||||
   */
 | 
			
		||||
  object Refactor : AceTagAction() {
 | 
			
		||||
    override fun invoke(editor: Editor, searchProcessor: SearchProcessor, offset: Int, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
      JumpToWordStart(editor, searchProcessor, offset, shiftMode = false, isFinal = isFinal)
 | 
			
		||||
      ApplicationManager.getApplication().invokeLater { performAction(if (shiftMode) IdeActions.ACTION_RENAME else "Refactorings.QuickListPopupAction") }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										148
									
								
								src/main/kotlin/org/acejump/action/AceVimAction.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										148
									
								
								src/main/kotlin/org/acejump/action/AceVimAction.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,148 @@
 | 
			
		||||
package org.acejump.action
 | 
			
		||||
 | 
			
		||||
import org.acejump.boundaries.Boundaries
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries.AFTER_CARET
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries.BEFORE_CARET
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries.VISIBLE_ON_SCREEN
 | 
			
		||||
import org.acejump.modes.ActionMode
 | 
			
		||||
import org.acejump.modes.VimJumpMode
 | 
			
		||||
import org.acejump.search.Pattern
 | 
			
		||||
import org.acejump.session.Session
 | 
			
		||||
 | 
			
		||||
sealed class AceVimAction : AceKeyboardAction() {
 | 
			
		||||
  protected abstract val boundary: Boundaries
 | 
			
		||||
  
 | 
			
		||||
  final override fun invoke(session: Session) {
 | 
			
		||||
    session.defaultBoundary = boundary
 | 
			
		||||
    start(session)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  protected open fun start(session: Session) {
 | 
			
		||||
    session.startJumpMode(::VimJumpMode)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class JumpToChar : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class JumpToCharAfterCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(AFTER_CARET)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class JumpToCharBeforeCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(BEFORE_CARET)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class LWordsAfterCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(AFTER_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_LWORD)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class UWordsAfterCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(AFTER_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_UWORD)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class LWordsBeforeCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(BEFORE_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_LWORD)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class UWordsBeforeCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(BEFORE_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_UWORD)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class LWordEndsAfterCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(AFTER_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_LWORD_END)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class UWordEndsAfterCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(AFTER_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_UWORD_END)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class LWordEndsBeforeCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(BEFORE_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_LWORD_END)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class UWordEndsBeforeCaret : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN.intersection(BEFORE_CARET)
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      super.start(session)
 | 
			
		||||
      session.startRegexSearch(Pattern.VIM_UWORD_END)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class GoToDeclaration : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      session.startJumpMode { ActionMode(AceTagAction.GoToDeclaration, shiftMode = false) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class GoToTypeDeclaration : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      session.startJumpMode { ActionMode(AceTagAction.GoToDeclaration, shiftMode = true) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class ShowIntentions : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      session.startJumpMode { ActionMode(AceTagAction.ShowIntentions, shiftMode = false) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class ShowUsages : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      session.startJumpMode { ActionMode(AceTagAction.ShowUsages, shiftMode = false) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class FindUsages : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      session.startJumpMode { ActionMode(AceTagAction.ShowUsages, shiftMode = true) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class Refactor : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      session.startJumpMode { ActionMode(AceTagAction.Refactor, shiftMode = false) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  class Rename : AceVimAction() {
 | 
			
		||||
    override val boundary = VISIBLE_ON_SCREEN
 | 
			
		||||
    override fun start(session: Session) {
 | 
			
		||||
      session.startJumpMode { ActionMode(AceTagAction.Refactor, shiftMode = true) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -1,108 +0,0 @@
 | 
			
		||||
package org.acejump.action
 | 
			
		||||
 | 
			
		||||
import com.intellij.codeInsight.navigation.actions.GotoDeclarationAction
 | 
			
		||||
import com.intellij.codeInsight.navigation.actions.GotoTypeDeclarationAction
 | 
			
		||||
import com.intellij.openapi.actionSystem.ActionManager
 | 
			
		||||
import com.intellij.openapi.actionSystem.AnAction
 | 
			
		||||
import com.intellij.openapi.command.CommandProcessor
 | 
			
		||||
import com.intellij.openapi.command.UndoConfirmationPolicy
 | 
			
		||||
import com.intellij.openapi.editor.Document
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.openapi.editor.actionSystem.DocCommandGroupId
 | 
			
		||||
import com.intellij.openapi.fileEditor.ex.IdeDocumentHistory
 | 
			
		||||
import com.intellij.openapi.project.Project
 | 
			
		||||
import com.intellij.openapi.ui.playback.commands.ActionCommand
 | 
			
		||||
import org.acejump.*
 | 
			
		||||
import org.acejump.input.JumpMode
 | 
			
		||||
import org.acejump.input.JumpMode.*
 | 
			
		||||
import org.acejump.search.SearchProcessor
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Performs [JumpMode] navigation and actions.
 | 
			
		||||
 */
 | 
			
		||||
internal class TagJumper(private val editor: Editor, private val mode: JumpMode, private val searchProcessor: SearchProcessor?) {
 | 
			
		||||
  /**
 | 
			
		||||
   * Moves caret to a specific offset in the editor according to the positioning and selection rules of the current [JumpMode].
 | 
			
		||||
   */
 | 
			
		||||
  fun visit(offset: Int) {
 | 
			
		||||
    if (mode === JUMP_END || mode === TARGET) {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      val matchingChars = searchProcessor?.let { chars.countMatchingCharacters(offset, it.query.rawText) } ?: 0
 | 
			
		||||
      val targetOffset = offset + matchingChars
 | 
			
		||||
      val isInsideWord = matchingChars > 0 && chars[targetOffset - 1].isWordPart && chars[targetOffset].isWordPart
 | 
			
		||||
      val finalTargetOffset = if (isInsideWord) chars.wordEnd(targetOffset) + 1 else targetOffset
 | 
			
		||||
      
 | 
			
		||||
      if (mode === JUMP_END) {
 | 
			
		||||
        moveCaretTo(editor, finalTargetOffset)
 | 
			
		||||
      }
 | 
			
		||||
      else if (mode === TARGET) {
 | 
			
		||||
        if (isInsideWord) {
 | 
			
		||||
          selectRange(editor, chars.wordStart(targetOffset), finalTargetOffset)
 | 
			
		||||
        }
 | 
			
		||||
        else {
 | 
			
		||||
          selectRange(editor, offset, finalTargetOffset)
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    else {
 | 
			
		||||
      moveCaretTo(editor, offset)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Updates caret and selection by [visit]ing a specific offset in the editor, and applying session-finalizing [JumpMode] actions such as
 | 
			
		||||
   * using the Go To Declaration action, or selecting text between caret and target offset/word if Shift was held during the jump.
 | 
			
		||||
   */
 | 
			
		||||
  fun jump(offset: Int, shiftMode: Boolean) {
 | 
			
		||||
    val oldOffset = editor.caretModel.offset
 | 
			
		||||
    
 | 
			
		||||
    visit(offset)
 | 
			
		||||
    
 | 
			
		||||
    if (mode === DEFINE) {
 | 
			
		||||
      performAction(if (shiftMode) GotoTypeDeclarationAction() else GotoDeclarationAction())
 | 
			
		||||
      return
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    if (shiftMode) {
 | 
			
		||||
      val newOffset = editor.caretModel.offset
 | 
			
		||||
      
 | 
			
		||||
      if (mode === TARGET) {
 | 
			
		||||
        selectRange(editor, oldOffset, when {
 | 
			
		||||
          newOffset < oldOffset -> editor.selectionModel.selectionStart
 | 
			
		||||
          else                  -> editor.selectionModel.selectionEnd
 | 
			
		||||
        })
 | 
			
		||||
      }
 | 
			
		||||
      else {
 | 
			
		||||
        selectRange(editor, oldOffset, newOffset)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private companion object {
 | 
			
		||||
    private fun moveCaretTo(editor: Editor, offset: Int) = with(editor) {
 | 
			
		||||
      project?.let { addCurrentPositionToHistory(it, document) }
 | 
			
		||||
      selectionModel.removeSelection(true)
 | 
			
		||||
      caretModel.moveToOffset(offset)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun selectRange(editor: Editor, fromOffset: Int, toOffset: Int) = with(editor) {
 | 
			
		||||
      selectionModel.removeSelection(true)
 | 
			
		||||
      selectionModel.setSelection(fromOffset, toOffset)
 | 
			
		||||
      caretModel.moveToOffset(toOffset)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun addCurrentPositionToHistory(project: Project, document: Document) {
 | 
			
		||||
      CommandProcessor.getInstance().executeCommand(project, {
 | 
			
		||||
        with(IdeDocumentHistory.getInstance(project)) {
 | 
			
		||||
          setCurrentCommandHasMoves()
 | 
			
		||||
          includeCurrentCommandAsNavigation()
 | 
			
		||||
          includeCurrentPlaceAsChangePlace()
 | 
			
		||||
        }
 | 
			
		||||
      }, "AceJumpHistoryAppender", DocCommandGroupId.noneGroupId(document), UndoConfirmationPolicy.DO_NOT_REQUEST_CONFIRMATION, document)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun performAction(action: AnAction) {
 | 
			
		||||
      ActionManager.getInstance().tryToExecute(action, ActionCommand.getInputEvent(null), null, null, true)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -1,73 +0,0 @@
 | 
			
		||||
package org.acejump.action
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.openapi.editor.ScrollType
 | 
			
		||||
import com.intellij.openapi.editor.SelectionModel
 | 
			
		||||
import org.acejump.search.SearchProcessor
 | 
			
		||||
import kotlin.math.abs
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Enables navigation between currently active tags.
 | 
			
		||||
 */
 | 
			
		||||
internal class TagVisitor(private val editor: Editor, private val searchProcessor: SearchProcessor, private val tagJumper: TagJumper) {
 | 
			
		||||
  /**
 | 
			
		||||
   * Places caret at the closest tag following the caret position, according to the rules of the current jump mode (see [TagJumper.visit]).
 | 
			
		||||
   * If the caret is at or past the last tag, it moves to the first tag instead.
 | 
			
		||||
   * If there is only one tag, it immediately performs the jump action as described in [TagJumper.jump].
 | 
			
		||||
   */
 | 
			
		||||
  fun visitNext(): Boolean {
 | 
			
		||||
    return visit(SelectionModel::getSelectionEnd) { if (it < 0) -it - 1 else it + 1 }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Places caret at the closest tag preceding the caret position, according to the rules of the current jump mode (see [TagJumper.visit]).
 | 
			
		||||
   * If the caret is at or before the first tag, it moves to the last tag instead.
 | 
			
		||||
   * If there is only one tag, it immediately performs the jump action as described in [TagJumper.jump].
 | 
			
		||||
   */
 | 
			
		||||
  fun visitPrevious(): Boolean {
 | 
			
		||||
    return visit(SelectionModel::getSelectionStart) { if (it < 0) -it - 2 else it - 1 }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Scrolls to the closest result to the caret.
 | 
			
		||||
   */
 | 
			
		||||
  fun scrollToClosest() {
 | 
			
		||||
    val caret = editor.caretModel.offset
 | 
			
		||||
    val results = searchProcessor.results.takeUnless { it.isEmpty } ?: return
 | 
			
		||||
    val index = results.binarySearch(caret).let { if (it < 0) -it - 1 else it }
 | 
			
		||||
    
 | 
			
		||||
    val targetOffset = listOfNotNull(
 | 
			
		||||
      results.getOrNull(index - 1),
 | 
			
		||||
      results.getOrNull(index)
 | 
			
		||||
    ).minBy {
 | 
			
		||||
      abs(it - caret)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    if (targetOffset != null) {
 | 
			
		||||
      editor.scrollingModel.scrollTo(editor.offsetToLogicalPosition(targetOffset), ScrollType.RELATIVE)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private inline fun visit(caretPosition: SelectionModel.() -> Int, indexModifier: (Int) -> Int): Boolean {
 | 
			
		||||
    val results = searchProcessor.results.takeUnless { it.isEmpty } ?: return false
 | 
			
		||||
    val nextIndex = indexModifier(results.binarySearch(caretPosition(editor.selectionModel)))
 | 
			
		||||
    
 | 
			
		||||
    val targetOffset = results.getInt(when {
 | 
			
		||||
      nextIndex < 0                 -> results.lastIndex
 | 
			
		||||
      nextIndex > results.lastIndex -> 0
 | 
			
		||||
      else                          -> nextIndex
 | 
			
		||||
    })
 | 
			
		||||
    
 | 
			
		||||
    val onlyResult = results.size == 1
 | 
			
		||||
    
 | 
			
		||||
    if (onlyResult) {
 | 
			
		||||
      tagJumper.jump(targetOffset, shiftMode = false)
 | 
			
		||||
    }
 | 
			
		||||
    else {
 | 
			
		||||
      tagJumper.visit(targetOffset)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    editor.scrollingModel.scrollToCaret(ScrollType.RELATIVE)
 | 
			
		||||
    return onlyResult
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -3,16 +3,6 @@ package org.acejump.boundaries
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
 | 
			
		||||
enum class StandardBoundaries : Boundaries {
 | 
			
		||||
  WHOLE_FILE {
 | 
			
		||||
    override fun getOffsetRange(editor: Editor, cache: EditorOffsetCache): IntRange {
 | 
			
		||||
      return 0 until editor.document.textLength
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    override fun isOffsetInside(editor: Editor, offset: Int, cache: EditorOffsetCache): Boolean {
 | 
			
		||||
      return offset in (0 until editor.document.textLength)
 | 
			
		||||
    }
 | 
			
		||||
  },
 | 
			
		||||
  
 | 
			
		||||
  VISIBLE_ON_SCREEN {
 | 
			
		||||
    override fun getOffsetRange(editor: Editor, cache: EditorOffsetCache): IntRange {
 | 
			
		||||
      val (topLeft, bottomRight) = cache.visibleArea(editor)
 | 
			
		||||
@@ -45,21 +35,21 @@ enum class StandardBoundaries : Boundaries {
 | 
			
		||||
  
 | 
			
		||||
  BEFORE_CARET {
 | 
			
		||||
    override fun getOffsetRange(editor: Editor, cache: EditorOffsetCache): IntRange {
 | 
			
		||||
      return 0..(editor.caretModel.offset)
 | 
			
		||||
      return 0 until editor.caretModel.offset
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    override fun isOffsetInside(editor: Editor, offset: Int, cache: EditorOffsetCache): Boolean {
 | 
			
		||||
      return offset <= editor.caretModel.offset
 | 
			
		||||
      return offset < editor.caretModel.offset
 | 
			
		||||
    }
 | 
			
		||||
  },
 | 
			
		||||
  
 | 
			
		||||
  AFTER_CARET {
 | 
			
		||||
    override fun getOffsetRange(editor: Editor, cache: EditorOffsetCache): IntRange {
 | 
			
		||||
      return editor.caretModel.offset until editor.document.textLength
 | 
			
		||||
      return (editor.caretModel.offset + 1) until editor.document.textLength
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    override fun isOffsetInside(editor: Editor, offset: Int, cache: EditorOffsetCache): Boolean {
 | 
			
		||||
      return offset >= editor.caretModel.offset
 | 
			
		||||
      return offset > editor.caretModel.offset
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -19,16 +19,14 @@ class AceConfig : PersistentStateComponent<AceSettings> {
 | 
			
		||||
      get() = ServiceManager.getService(AceConfig::class.java).aceSettings
 | 
			
		||||
    
 | 
			
		||||
    val layout get() = settings.layout
 | 
			
		||||
    val cycleModes get() = settings.let { arrayOf(it.cycleMode1, it.cycleMode2, it.cycleMode3, it.cycleMode4) }
 | 
			
		||||
    val minQueryLength get() = settings.minQueryLength
 | 
			
		||||
    val jumpModeColor get() = settings.jumpModeColor
 | 
			
		||||
    val jumpEndModeColor get() = settings.jumpEndModeColor
 | 
			
		||||
    val targetModeColor get() = settings.targetModeColor
 | 
			
		||||
    val definitionModeColor get() = settings.definitionModeColor
 | 
			
		||||
    val advancedModeColor get() = settings.advancedModeColor
 | 
			
		||||
    val betweenPointsModeColor get() = settings.betweenPointsModeColor
 | 
			
		||||
    val textHighlightColor get() = settings.textHighlightColor
 | 
			
		||||
    val tagForegroundColor get() = settings.tagForegroundColor
 | 
			
		||||
    val tagBackgroundColor get() = settings.tagBackgroundColor
 | 
			
		||||
    val searchWholeFile get() = settings.searchWholeFile
 | 
			
		||||
    val acceptedTagColor get() = settings.acceptedTagColor
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun getState(): AceSettings {
 | 
			
		||||
 
 | 
			
		||||
@@ -14,36 +14,26 @@ class AceConfigurable : Configurable {
 | 
			
		||||
  override fun isModified() =
 | 
			
		||||
    panel.allowedChars != settings.allowedChars ||
 | 
			
		||||
      panel.keyboardLayout != settings.layout ||
 | 
			
		||||
      panel.cycleMode1 != settings.cycleMode1 ||
 | 
			
		||||
      panel.cycleMode2 != settings.cycleMode2 ||
 | 
			
		||||
      panel.cycleMode3 != settings.cycleMode3 ||
 | 
			
		||||
      panel.cycleMode4 != settings.cycleMode4 ||
 | 
			
		||||
      panel.minQueryLengthInt != settings.minQueryLength ||
 | 
			
		||||
      panel.jumpModeColor != settings.jumpModeColor ||
 | 
			
		||||
      panel.jumpEndModeColor != settings.jumpEndModeColor ||
 | 
			
		||||
      panel.targetModeColor != settings.targetModeColor ||
 | 
			
		||||
      panel.definitionModeColor != settings.definitionModeColor ||
 | 
			
		||||
      panel.advancedModeColor != settings.advancedModeColor ||
 | 
			
		||||
      panel.betweenPointsModeColor != settings.betweenPointsModeColor ||
 | 
			
		||||
      panel.textHighlightColor != settings.textHighlightColor ||
 | 
			
		||||
      panel.tagForegroundColor != settings.tagForegroundColor ||
 | 
			
		||||
      panel.tagBackgroundColor != settings.tagBackgroundColor ||
 | 
			
		||||
      panel.searchWholeFile != settings.searchWholeFile
 | 
			
		||||
      panel.acceptedTagColor != settings.acceptedTagColor
 | 
			
		||||
  
 | 
			
		||||
  override fun apply() {
 | 
			
		||||
    settings.allowedChars = panel.allowedChars
 | 
			
		||||
    settings.layout = panel.keyboardLayout
 | 
			
		||||
    settings.cycleMode1 = panel.cycleMode1
 | 
			
		||||
    settings.cycleMode2 = panel.cycleMode2
 | 
			
		||||
    settings.cycleMode3 = panel.cycleMode3
 | 
			
		||||
    settings.cycleMode4 = panel.cycleMode4
 | 
			
		||||
    settings.minQueryLength = panel.minQueryLengthInt ?: settings.minQueryLength
 | 
			
		||||
    panel.jumpModeColor?.let { settings.jumpModeColor = it }
 | 
			
		||||
    panel.jumpEndModeColor?.let { settings.jumpEndModeColor = it }
 | 
			
		||||
    panel.targetModeColor?.let { settings.targetModeColor = it }
 | 
			
		||||
    panel.definitionModeColor?.let { settings.definitionModeColor = it }
 | 
			
		||||
    panel.advancedModeColor?.let { settings.advancedModeColor = it }
 | 
			
		||||
    panel.betweenPointsModeColor?.let { settings.betweenPointsModeColor = it }
 | 
			
		||||
    panel.textHighlightColor?.let { settings.textHighlightColor = it }
 | 
			
		||||
    panel.tagForegroundColor?.let { settings.tagForegroundColor = it }
 | 
			
		||||
    panel.tagBackgroundColor?.let { settings.tagBackgroundColor = it }
 | 
			
		||||
    settings.searchWholeFile = panel.searchWholeFile
 | 
			
		||||
    panel.acceptedTagColor?.let { settings.acceptedTagColor = it }
 | 
			
		||||
    KeyLayoutCache.reset(settings)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
 
 | 
			
		||||
@@ -1,7 +1,6 @@
 | 
			
		||||
package org.acejump.config
 | 
			
		||||
 | 
			
		||||
import com.intellij.util.xmlb.annotations.OptionTag
 | 
			
		||||
import org.acejump.input.JumpMode
 | 
			
		||||
import org.acejump.input.KeyLayout
 | 
			
		||||
import org.acejump.input.KeyLayout.QWERTY
 | 
			
		||||
import java.awt.Color
 | 
			
		||||
@@ -9,23 +8,16 @@ import java.awt.Color
 | 
			
		||||
data class AceSettings(
 | 
			
		||||
  var layout: KeyLayout = QWERTY,
 | 
			
		||||
  var allowedChars: String = layout.allChars,
 | 
			
		||||
  var cycleMode1: JumpMode = JumpMode.JUMP,
 | 
			
		||||
  var cycleMode2: JumpMode = JumpMode.DEFINE,
 | 
			
		||||
  var cycleMode3: JumpMode = JumpMode.TARGET,
 | 
			
		||||
  var cycleMode4: JumpMode = JumpMode.JUMP_END,
 | 
			
		||||
  var minQueryLength: Int = 1,
 | 
			
		||||
  
 | 
			
		||||
  @OptionTag("jumpModeRGB", converter = ColorConverter::class)
 | 
			
		||||
  var jumpModeColor: Color = Color(0xFFFFFF),
 | 
			
		||||
  
 | 
			
		||||
  @OptionTag("jumpEndModeRGB", converter = ColorConverter::class)
 | 
			
		||||
  var jumpEndModeColor: Color = Color(0x33E78A),
 | 
			
		||||
  @OptionTag("advancedModeRGB", converter = ColorConverter::class)
 | 
			
		||||
  var advancedModeColor: Color = Color(0xFFB700),
 | 
			
		||||
  
 | 
			
		||||
  @OptionTag("targetModeRGB", converter = ColorConverter::class)
 | 
			
		||||
  var targetModeColor: Color = Color(0xFFB700),
 | 
			
		||||
  
 | 
			
		||||
  @OptionTag("definitionModeRGB", converter = ColorConverter::class)
 | 
			
		||||
  var definitionModeColor: Color = Color(0x6FC5FF),
 | 
			
		||||
  @OptionTag("betweenPointsModeRGB", converter = ColorConverter::class)
 | 
			
		||||
  var betweenPointsModeColor: Color = Color(0x6FC5FF),
 | 
			
		||||
  
 | 
			
		||||
  @OptionTag("textHighlightRGB", converter = ColorConverter::class)
 | 
			
		||||
  var textHighlightColor: Color = Color(0x394B58),
 | 
			
		||||
@@ -36,5 +28,6 @@ data class AceSettings(
 | 
			
		||||
  @OptionTag("tagBackgroundRGB", converter = ColorConverter::class)
 | 
			
		||||
  var tagBackgroundColor: Color = Color(0x008299),
 | 
			
		||||
  
 | 
			
		||||
  var searchWholeFile: Boolean = true
 | 
			
		||||
  @OptionTag("acceptedTagRGB", converter = ColorConverter::class)
 | 
			
		||||
  var acceptedTagColor: Color = Color(0x394B58)
 | 
			
		||||
)
 | 
			
		||||
 
 | 
			
		||||
@@ -2,14 +2,12 @@ package org.acejump.config
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.ui.ComboBox
 | 
			
		||||
import com.intellij.ui.ColorPanel
 | 
			
		||||
import com.intellij.ui.components.JBCheckBox
 | 
			
		||||
import com.intellij.ui.components.JBTextArea
 | 
			
		||||
import com.intellij.ui.components.JBTextField
 | 
			
		||||
import com.intellij.ui.layout.Cell
 | 
			
		||||
import com.intellij.ui.layout.GrowPolicy.MEDIUM_TEXT
 | 
			
		||||
import com.intellij.ui.layout.GrowPolicy.SHORT_TEXT
 | 
			
		||||
import com.intellij.ui.layout.panel
 | 
			
		||||
import org.acejump.input.JumpMode
 | 
			
		||||
import org.acejump.input.KeyLayout
 | 
			
		||||
import java.awt.Color
 | 
			
		||||
import java.awt.Font
 | 
			
		||||
@@ -27,28 +25,19 @@ internal class AceSettingsPanel {
 | 
			
		||||
  private val tagCharsField = JBTextField()
 | 
			
		||||
  private val keyboardLayoutCombo = ComboBox<KeyLayout>()
 | 
			
		||||
  private val keyboardLayoutArea = JBTextArea().apply { isEditable = false }
 | 
			
		||||
  private val cycleModeCombo1 = ComboBox<JumpMode>()
 | 
			
		||||
  private val cycleModeCombo2 = ComboBox<JumpMode>()
 | 
			
		||||
  private val cycleModeCombo3 = ComboBox<JumpMode>()
 | 
			
		||||
  private val cycleModeCombo4 = ComboBox<JumpMode>()
 | 
			
		||||
  private val minQueryLengthField = JBTextField()
 | 
			
		||||
  private val jumpModeColorWheel = ColorPanel()
 | 
			
		||||
  private val jumpEndModeColorWheel = ColorPanel()
 | 
			
		||||
  private val targetModeColorWheel = ColorPanel()
 | 
			
		||||
  private val definitionModeColorWheel = ColorPanel()
 | 
			
		||||
  private val advancedModeColorWheel = ColorPanel()
 | 
			
		||||
  private val betweenPointsModeColorWheel = ColorPanel()
 | 
			
		||||
  private val textHighlightColorWheel = ColorPanel()
 | 
			
		||||
  private val tagForegroundColorWheel = ColorPanel()
 | 
			
		||||
  private val tagBackgroundColorWheel = ColorPanel()
 | 
			
		||||
  private val searchWholeFileCheckBox = JBCheckBox()
 | 
			
		||||
  private val acceptedTagColorWheel = ColorPanel()
 | 
			
		||||
  
 | 
			
		||||
  init {
 | 
			
		||||
    tagCharsField.apply { font = Font("monospaced", font.style, font.size) }
 | 
			
		||||
    keyboardLayoutArea.apply { font = Font("monospaced", font.style, font.size) }
 | 
			
		||||
    keyboardLayoutCombo.setupEnumItems { keyChars = it.rows.joinToString("\n") }
 | 
			
		||||
    cycleModeCombo1.setupEnumItems { cycleMode1 = it }
 | 
			
		||||
    cycleModeCombo2.setupEnumItems { cycleMode2 = it }
 | 
			
		||||
    cycleModeCombo3.setupEnumItems { cycleMode3 = it }
 | 
			
		||||
    cycleModeCombo4.setupEnumItems { cycleMode4 = it }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  internal val rootPanel: JPanel = panel {
 | 
			
		||||
@@ -61,30 +50,18 @@ internal class AceSettingsPanel {
 | 
			
		||||
      row("Keyboard design:") { short(keyboardLayoutArea) }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    titledRow("Modes") {
 | 
			
		||||
      row("Cycle order:") {
 | 
			
		||||
        cell(isVerticalFlow = false, isFullWidth = false) {
 | 
			
		||||
          cycleModeCombo1()
 | 
			
		||||
          cycleModeCombo2()
 | 
			
		||||
          cycleModeCombo3()
 | 
			
		||||
          cycleModeCombo4()
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    titledRow("Behavior") {
 | 
			
		||||
      row("Minimum typed characters (1-10):") { short(minQueryLengthField) }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    titledRow("Colors") {
 | 
			
		||||
      row("Jump mode caret background:") { short(jumpModeColorWheel) }
 | 
			
		||||
      row("Jump to End mode caret background:") { short(jumpEndModeColorWheel) }
 | 
			
		||||
      row("Target mode caret background:") { short(targetModeColorWheel) }
 | 
			
		||||
      row("Definition mode caret background:") { short(definitionModeColorWheel) }
 | 
			
		||||
      row("Advanced mode caret background:") { short(advancedModeColorWheel) }
 | 
			
		||||
      row("Between Points mode caret background:") { short(betweenPointsModeColorWheel) }
 | 
			
		||||
      row("Searched text background:") { short(textHighlightColorWheel) }
 | 
			
		||||
      row("Tag foreground:") { short(tagForegroundColorWheel) }
 | 
			
		||||
      row("Tag background:") { short(tagBackgroundColorWheel) }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    titledRow("Behavior") {
 | 
			
		||||
      row { short(searchWholeFileCheckBox.apply { text = "Search whole file" }) }
 | 
			
		||||
      row("Minimum typed characters (1-10):") { short(minQueryLengthField) }
 | 
			
		||||
      row("Accepted tag position background:") { short(acceptedTagColorWheel) }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
@@ -92,19 +69,14 @@ internal class AceSettingsPanel {
 | 
			
		||||
  internal var allowedChars by tagCharsField
 | 
			
		||||
  internal var keyboardLayout by keyboardLayoutCombo
 | 
			
		||||
  internal var keyChars by keyboardLayoutArea
 | 
			
		||||
  internal var cycleMode1 by cycleModeCombo1
 | 
			
		||||
  internal var cycleMode2 by cycleModeCombo2
 | 
			
		||||
  internal var cycleMode3 by cycleModeCombo3
 | 
			
		||||
  internal var cycleMode4 by cycleModeCombo4
 | 
			
		||||
  internal var minQueryLength by minQueryLengthField
 | 
			
		||||
  internal var jumpModeColor by jumpModeColorWheel
 | 
			
		||||
  internal var jumpEndModeColor by jumpEndModeColorWheel
 | 
			
		||||
  internal var targetModeColor by targetModeColorWheel
 | 
			
		||||
  internal var definitionModeColor by definitionModeColorWheel
 | 
			
		||||
  internal var advancedModeColor by advancedModeColorWheel
 | 
			
		||||
  internal var betweenPointsModeColor by betweenPointsModeColorWheel
 | 
			
		||||
  internal var textHighlightColor by textHighlightColorWheel
 | 
			
		||||
  internal var tagForegroundColor by tagForegroundColorWheel
 | 
			
		||||
  internal var tagBackgroundColor by tagBackgroundColorWheel
 | 
			
		||||
  internal var searchWholeFile by searchWholeFileCheckBox
 | 
			
		||||
  internal var acceptedTagColor by acceptedTagColorWheel
 | 
			
		||||
  
 | 
			
		||||
  internal var minQueryLengthInt
 | 
			
		||||
    get() = minQueryLength.toIntOrNull()?.coerceIn(1, 10)
 | 
			
		||||
@@ -113,19 +85,14 @@ internal class AceSettingsPanel {
 | 
			
		||||
  fun reset(settings: AceSettings) {
 | 
			
		||||
    allowedChars = settings.allowedChars
 | 
			
		||||
    keyboardLayout = settings.layout
 | 
			
		||||
    cycleMode1 = settings.cycleMode1
 | 
			
		||||
    cycleMode2 = settings.cycleMode2
 | 
			
		||||
    cycleMode3 = settings.cycleMode3
 | 
			
		||||
    cycleMode4 = settings.cycleMode4
 | 
			
		||||
    minQueryLength = settings.minQueryLength.toString()
 | 
			
		||||
    jumpModeColor = settings.jumpModeColor
 | 
			
		||||
    jumpEndModeColor = settings.jumpEndModeColor
 | 
			
		||||
    targetModeColor = settings.targetModeColor
 | 
			
		||||
    definitionModeColor = settings.definitionModeColor
 | 
			
		||||
    advancedModeColor = settings.advancedModeColor
 | 
			
		||||
    betweenPointsModeColor = settings.betweenPointsModeColor
 | 
			
		||||
    textHighlightColor = settings.textHighlightColor
 | 
			
		||||
    tagForegroundColor = settings.tagForegroundColor
 | 
			
		||||
    tagBackgroundColor = settings.tagBackgroundColor
 | 
			
		||||
    searchWholeFile = settings.searchWholeFile
 | 
			
		||||
    acceptedTagColor = settings.acceptedTagColor
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  // Removal pending support for https://youtrack.jetbrains.com/issue/KT-8575
 | 
			
		||||
 
 | 
			
		||||
@@ -1,68 +0,0 @@
 | 
			
		||||
package org.acejump.input
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.colors.impl.AbstractColorsScheme
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import java.awt.Color
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Describes modes that determine the behavior of a "jump" to a tag. Most modes have two variations:
 | 
			
		||||
 * - **Default jump** happens when jumping without holding the Shift key
 | 
			
		||||
 * - **Shift jump** happens when jumping while holding the Shift key
 | 
			
		||||
 */
 | 
			
		||||
enum class JumpMode {
 | 
			
		||||
  /**
 | 
			
		||||
   * Default value at the start of a session. If the session does not get assigned a proper [JumpMode] by the time the user requests a jump,
 | 
			
		||||
   * the results of the jump are undefined.
 | 
			
		||||
   */
 | 
			
		||||
  DISABLED,
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default jump, places the caret at the first character of the search query.
 | 
			
		||||
   * On shift jump, does the above but also selects all text between the original and new caret positions.
 | 
			
		||||
   */
 | 
			
		||||
  JUMP,
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default jump, places the caret at the end of a word. Word detection uses [Character.isJavaIdentifierPart] to count some special
 | 
			
		||||
   * characters, such as underscores, as part of a word. If there is no word at the first character of the search query, then the caret is
 | 
			
		||||
   * placed after the last character of the search query.
 | 
			
		||||
   *
 | 
			
		||||
   * On shift jump, does the above but also selects all text between the original and new caret positions.
 | 
			
		||||
   */
 | 
			
		||||
  JUMP_END,
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default jump, places the caret at the end of a word, and also selects the entire word. Word detection uses
 | 
			
		||||
   * [Character.isJavaIdentifierPart] to count some special characters, such as underscores, as part of a word. If there is no word at the
 | 
			
		||||
   * first character of the search query, then the caret is placed after the last character of the search query, and all text between the
 | 
			
		||||
   * start and end of the search query is selected.
 | 
			
		||||
   *
 | 
			
		||||
   * On shift jump, does the above but also selects all text between the original caret position and the new selection, merging the
 | 
			
		||||
   * selections into one.
 | 
			
		||||
   */
 | 
			
		||||
  TARGET,
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * On default jump, performs the Go To Declaration action, available via `Navigate | Declaration or Usages`.
 | 
			
		||||
   * On shift jump, performs the Go To Type Declaration action, available via `Navigate | Type Declaration`.
 | 
			
		||||
   * Always places the caret at the first character of the search query.
 | 
			
		||||
   */
 | 
			
		||||
  DEFINE;
 | 
			
		||||
  
 | 
			
		||||
  val caretColor: Color
 | 
			
		||||
    get() = when (this) {
 | 
			
		||||
      JUMP     -> AceConfig.jumpModeColor
 | 
			
		||||
      JUMP_END -> AceConfig.jumpEndModeColor
 | 
			
		||||
      DEFINE   -> AceConfig.definitionModeColor
 | 
			
		||||
      TARGET   -> AceConfig.targetModeColor
 | 
			
		||||
      DISABLED -> AbstractColorsScheme.INHERITED_COLOR_MARKER
 | 
			
		||||
    }
 | 
			
		||||
  
 | 
			
		||||
  override fun toString() = when (this) {
 | 
			
		||||
    DISABLED -> "(Skip)"
 | 
			
		||||
    JUMP     -> "Jump"
 | 
			
		||||
    JUMP_END -> "Jump to End"
 | 
			
		||||
    TARGET   -> "Target"
 | 
			
		||||
    DEFINE   -> "Definition"
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -1,53 +0,0 @@
 | 
			
		||||
package org.acejump.input
 | 
			
		||||
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Remembers the current [JumpMode] for a session. Allows cycling [JumpMode]s according to the order defined in configuration, or toggling
 | 
			
		||||
 * one specific [JumpMode] on or off.
 | 
			
		||||
 */
 | 
			
		||||
internal class JumpModeTracker {
 | 
			
		||||
  private var currentMode = JumpMode.DISABLED
 | 
			
		||||
  private var currentIndex = 0
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Switches to the next/previous [JumpMode] defined in configuration, skipping any [JumpMode]s that are not assigned. If at least two
 | 
			
		||||
   * [JumpMode]s are assigned in the cycle order, then cycling will wrap around. If only one [JumpMode] is assigned, then cycling will
 | 
			
		||||
   * toggle that one mode.
 | 
			
		||||
   */
 | 
			
		||||
  fun cycle(forward: Boolean): JumpMode {
 | 
			
		||||
    val cycleModes = AceConfig.cycleModes
 | 
			
		||||
    val direction = if (forward) 1 else -1
 | 
			
		||||
    val start = if (currentIndex == 0 && !forward) 0 else currentIndex - 1
 | 
			
		||||
    
 | 
			
		||||
    for (offset in 1 until cycleModes.size) {
 | 
			
		||||
      val index = (start + cycleModes.size + (offset * direction)) % cycleModes.size
 | 
			
		||||
      
 | 
			
		||||
      if (cycleModes[index] != JumpMode.DISABLED) {
 | 
			
		||||
        currentMode = cycleModes[index]
 | 
			
		||||
        currentIndex = index + 1
 | 
			
		||||
        return currentMode
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    currentMode = JumpMode.DISABLED
 | 
			
		||||
    currentIndex = 0
 | 
			
		||||
    return currentMode
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Switches to the specified [JumpMode]. If the current mode already equals the specified one, it resets to [JumpMode.DISABLED].
 | 
			
		||||
   */
 | 
			
		||||
  fun toggle(newMode: JumpMode): JumpMode {
 | 
			
		||||
    if (currentMode == newMode) {
 | 
			
		||||
      currentMode = JumpMode.DISABLED
 | 
			
		||||
      currentIndex = 0
 | 
			
		||||
    }
 | 
			
		||||
    else {
 | 
			
		||||
      currentMode = newMode
 | 
			
		||||
      currentIndex = AceConfig.cycleModes.indexOfFirst { it == newMode } + 1
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return currentMode
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										12
									
								
								src/main/kotlin/org/acejump/modes/ActionMode.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								src/main/kotlin/org/acejump/modes/ActionMode.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
			
		||||
package org.acejump.modes
 | 
			
		||||
 | 
			
		||||
import org.acejump.action.AceTagAction
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import org.acejump.session.SessionState
 | 
			
		||||
 | 
			
		||||
class ActionMode(private val action: AceTagAction, private val shiftMode: Boolean) : JumpMode() {
 | 
			
		||||
  override fun accept(state: SessionState, acceptedTag: Tag): Boolean {
 | 
			
		||||
    state.act(action, acceptedTag, shiftMode, isFinal = true)
 | 
			
		||||
    return true
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										73
									
								
								src/main/kotlin/org/acejump/modes/AdvancedMode.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										73
									
								
								src/main/kotlin/org/acejump/modes/AdvancedMode.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,73 @@
 | 
			
		||||
package org.acejump.modes
 | 
			
		||||
 | 
			
		||||
import org.acejump.action.AceTagAction
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import org.acejump.session.SessionState
 | 
			
		||||
import org.acejump.session.TypeResult
 | 
			
		||||
 | 
			
		||||
class AdvancedMode : SessionMode {
 | 
			
		||||
  companion object {
 | 
			
		||||
    private val JUMP_HINT = arrayOf(
 | 
			
		||||
      "<f>[J]</f>ump / <f>[L]</f> past Query",
 | 
			
		||||
      "<f>[E]</f> Word End / <f>[M]</f> Line End"
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    val JUMP_ALT_HINT = JUMP_HINT.map { it.replace("<f>[J]</f>ump ", "<f>[J]</f> at Tag ") }.toTypedArray()
 | 
			
		||||
    
 | 
			
		||||
    val JUMP_ACTION_MAP = mapOf(
 | 
			
		||||
      'J' to AceTagAction.JumpToSearchStart,
 | 
			
		||||
      'L' to AceTagAction.JumpPastSearchEnd,
 | 
			
		||||
      'E' to AceTagAction.JumpToWordEnd,
 | 
			
		||||
      'M' to AceTagAction.JumpToLineEnd
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    val SELECT_HINT = arrayOf(
 | 
			
		||||
      "Select <f>[W]</f>ord / <f>[H]</f>ump",
 | 
			
		||||
      "Select <f>[Q]</f>uery / <f>[N]</f> Line / <f>[1-9]</f> Expansion"
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    val SELECT_ACTION_MAP = mapOf(
 | 
			
		||||
      'W' to AceTagAction.SelectWord,
 | 
			
		||||
      'H' to AceTagAction.SelectHump,
 | 
			
		||||
      'Q' to AceTagAction.SelectQuery,
 | 
			
		||||
      'N' to AceTagAction.SelectLine,
 | 
			
		||||
      *('1'..'9').mapIndexed { index, char -> char to AceTagAction.SelectExtended(index + 1) }.toTypedArray()
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    private val ALL_HINTS = arrayOf(
 | 
			
		||||
      *JUMP_HINT,
 | 
			
		||||
      *SELECT_HINT,
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    private val ALL_ACTION_MAP = mapOf(
 | 
			
		||||
      *JUMP_ACTION_MAP.map { it.key to it.value }.toTypedArray(),
 | 
			
		||||
      *SELECT_ACTION_MAP.map { it.key to it.value }.toTypedArray(),
 | 
			
		||||
    )
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override val caretColor
 | 
			
		||||
    get() = AceConfig.advancedModeColor
 | 
			
		||||
  
 | 
			
		||||
  override fun type(state: SessionState, charTyped: Char, acceptedTag: Tag?): TypeResult {
 | 
			
		||||
    if (acceptedTag == null) {
 | 
			
		||||
      return state.type(charTyped)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val action = ALL_ACTION_MAP[charTyped.toUpperCase()]
 | 
			
		||||
    if (action != null) {
 | 
			
		||||
      state.act(action, acceptedTag, shiftMode = charTyped.isUpperCase(), isFinal = true)
 | 
			
		||||
      return TypeResult.EndSession
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return TypeResult.Nothing
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun accept(state: SessionState, acceptedTag: Tag): Boolean {
 | 
			
		||||
    return false
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun getHint(acceptedTag: Int?, hasQuery: Boolean): Array<String>? {
 | 
			
		||||
    return ALL_HINTS.takeIf { acceptedTag != null }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										92
									
								
								src/main/kotlin/org/acejump/modes/BetweenPointsMode.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										92
									
								
								src/main/kotlin/org/acejump/modes/BetweenPointsMode.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,92 @@
 | 
			
		||||
package org.acejump.modes
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.CaretState
 | 
			
		||||
import org.acejump.action.AceTagAction
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import org.acejump.session.SessionState
 | 
			
		||||
import org.acejump.session.TypeResult
 | 
			
		||||
 | 
			
		||||
class BetweenPointsMode : SessionMode {
 | 
			
		||||
  private companion object {
 | 
			
		||||
    private val TYPE_TAG_HINT = arrayOf(
 | 
			
		||||
      "<b>Type to Search...</b>"
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    private val ACTION_MODE_HINT = arrayOf(
 | 
			
		||||
      "<f>[C]</f>lone to Caret...",
 | 
			
		||||
      "<f>[M]</f>ove to Caret..."
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    private val ACTION_MODE_MAP = mapOf(
 | 
			
		||||
      'C' to (AceTagAction::CloneToCaret),
 | 
			
		||||
      'M' to (AceTagAction::MoveToCaret)
 | 
			
		||||
    )
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override val caretColor
 | 
			
		||||
    get() = AceConfig.betweenPointsModeColor
 | 
			
		||||
  
 | 
			
		||||
  private var actionMode: ((AceTagAction.BaseSelectAction) -> AceTagAction)? = null
 | 
			
		||||
  private var originalCarets: List<CaretState>? = null
 | 
			
		||||
  private var firstOffset: Int? = null
 | 
			
		||||
  
 | 
			
		||||
  override fun type(state: SessionState, charTyped: Char, acceptedTag: Tag?): TypeResult {
 | 
			
		||||
    val actionMode = actionMode
 | 
			
		||||
    if (actionMode == null) {
 | 
			
		||||
      this.actionMode = ACTION_MODE_MAP[charTyped.toUpperCase()]
 | 
			
		||||
      return TypeResult.Nothing
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    if (acceptedTag == null) {
 | 
			
		||||
      return state.type(charTyped)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    if (firstOffset == null) {
 | 
			
		||||
      val selectAction = AdvancedMode.SELECT_ACTION_MAP[charTyped.toUpperCase()]
 | 
			
		||||
      if (selectAction != null) {
 | 
			
		||||
        state.act(actionMode(selectAction), acceptedTag, shiftMode = charTyped.isUpperCase(), isFinal = true)
 | 
			
		||||
        return TypeResult.EndSession
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val jumpAction = AdvancedMode.JUMP_ACTION_MAP[charTyped.toUpperCase()]
 | 
			
		||||
    if (jumpAction == null) {
 | 
			
		||||
      return TypeResult.Nothing
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val firstOffset = firstOffset
 | 
			
		||||
    if (firstOffset == null) {
 | 
			
		||||
      val caretModel = acceptedTag.editor.caretModel
 | 
			
		||||
      this.originalCarets = caretModel.caretsAndSelections
 | 
			
		||||
      
 | 
			
		||||
      state.act(jumpAction, acceptedTag, shiftMode = false, isFinal = false)
 | 
			
		||||
      this.firstOffset = caretModel.offset
 | 
			
		||||
      return TypeResult.RestartSearch
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    originalCarets?.let { acceptedTag.editor.caretModel.caretsAndSelections = it }
 | 
			
		||||
    
 | 
			
		||||
    state.act(
 | 
			
		||||
      actionMode(AceTagAction.SelectBetweenPoints(firstOffset, jumpAction)),
 | 
			
		||||
      acceptedTag,
 | 
			
		||||
      shiftMode = charTyped.isUpperCase(),
 | 
			
		||||
      isFinal = true
 | 
			
		||||
    )
 | 
			
		||||
    
 | 
			
		||||
    return TypeResult.EndSession
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun accept(state: SessionState, acceptedTag: Tag): Boolean {
 | 
			
		||||
    return false
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun getHint(acceptedTag: Int?, hasQuery: Boolean): Array<String>? {
 | 
			
		||||
    return when {
 | 
			
		||||
      actionMode == null  -> ACTION_MODE_HINT
 | 
			
		||||
      acceptedTag == null -> TYPE_TAG_HINT.takeUnless { hasQuery }
 | 
			
		||||
      firstOffset == null -> AdvancedMode.JUMP_ALT_HINT + AdvancedMode.SELECT_HINT
 | 
			
		||||
      else                -> AdvancedMode.JUMP_ALT_HINT
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										29
									
								
								src/main/kotlin/org/acejump/modes/JumpMode.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								src/main/kotlin/org/acejump/modes/JumpMode.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,29 @@
 | 
			
		||||
package org.acejump.modes
 | 
			
		||||
 | 
			
		||||
import org.acejump.action.AceTagAction
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import org.acejump.session.SessionState
 | 
			
		||||
import org.acejump.session.TypeResult
 | 
			
		||||
 | 
			
		||||
open class JumpMode : SessionMode {
 | 
			
		||||
  override val caretColor
 | 
			
		||||
    get() = AceConfig.jumpModeColor
 | 
			
		||||
  
 | 
			
		||||
  protected var wasUpperCase = false
 | 
			
		||||
    private set
 | 
			
		||||
  
 | 
			
		||||
  override fun type(state: SessionState, charTyped: Char, acceptedTag: Tag?): TypeResult {
 | 
			
		||||
    wasUpperCase = charTyped.isUpperCase()
 | 
			
		||||
    return state.type(charTyped)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun accept(state: SessionState, acceptedTag: Tag): Boolean {
 | 
			
		||||
    state.act(AceTagAction.JumpToSearchStart, acceptedTag, wasUpperCase, isFinal = true)
 | 
			
		||||
    return true
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun getHint(acceptedTag: Int?, hasQuery: Boolean): Array<String>? {
 | 
			
		||||
    return null
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										14
									
								
								src/main/kotlin/org/acejump/modes/SessionMode.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								src/main/kotlin/org/acejump/modes/SessionMode.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
			
		||||
package org.acejump.modes
 | 
			
		||||
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import org.acejump.session.SessionState
 | 
			
		||||
import org.acejump.session.TypeResult
 | 
			
		||||
import java.awt.Color
 | 
			
		||||
 | 
			
		||||
interface SessionMode {
 | 
			
		||||
  val caretColor: Color
 | 
			
		||||
  
 | 
			
		||||
  fun type(state: SessionState, charTyped: Char, acceptedTag: Tag?): TypeResult
 | 
			
		||||
  fun accept(state: SessionState, acceptedTag: Tag): Boolean
 | 
			
		||||
  fun getHint(acceptedTag: Int?, hasQuery: Boolean): Array<String>?
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										17
									
								
								src/main/kotlin/org/acejump/modes/VimJumpMode.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								src/main/kotlin/org/acejump/modes/VimJumpMode.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
			
		||||
package org.acejump.modes
 | 
			
		||||
 | 
			
		||||
import org.acejump.action.AceTagAction
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import org.acejump.session.SessionState
 | 
			
		||||
 | 
			
		||||
class VimJumpMode : JumpMode() {
 | 
			
		||||
  override fun accept(state: SessionState, acceptedTag: Tag): Boolean {
 | 
			
		||||
    val action = if (acceptedTag.editor.selectionModel.hasSelection())
 | 
			
		||||
      AceTagAction.SelectToCaret(AceTagAction.JumpToSearchStart)
 | 
			
		||||
    else
 | 
			
		||||
      AceTagAction.JumpToSearchStart
 | 
			
		||||
    
 | 
			
		||||
    state.act(action, acceptedTag, wasUpperCase, isFinal = true)
 | 
			
		||||
    return true
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -5,5 +5,9 @@ enum class Pattern(val regex: String) {
 | 
			
		||||
  LINE_ENDS("\\n|\\Z"),
 | 
			
		||||
  LINE_INDENTS("[^\\s].*|^\\n"),
 | 
			
		||||
  LINE_ALL_MARKS(LINE_ENDS.regex + "|" + LINE_STARTS.regex + "|" + LINE_INDENTS.regex),
 | 
			
		||||
  ALL_WORDS("(?<=[^a-zA-Z0-9_]|\\A)[a-zA-Z0-9_]");
 | 
			
		||||
  ALL_WORDS("(?<=[^a-zA-Z0-9_]|\\A)[a-zA-Z0-9_]"),
 | 
			
		||||
  VIM_LWORD("(?<=[^a-zA-Z0-9_]|\\A)[a-zA-Z0-9_]"),
 | 
			
		||||
  VIM_UWORD("(?<=\\s|\\A)[^\\s]"),
 | 
			
		||||
  VIM_LWORD_END("[a-zA-Z0-9_](?=[^a-zA-Z0-9_]|\\Z)"),
 | 
			
		||||
  VIM_UWORD_END("[^\\s](?=\\s|\\Z)")
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -3,6 +3,7 @@ package org.acejump.search
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.IntArrayList
 | 
			
		||||
import org.acejump.boundaries.Boundaries
 | 
			
		||||
import org.acejump.clone
 | 
			
		||||
import org.acejump.immutableText
 | 
			
		||||
import org.acejump.isWordPart
 | 
			
		||||
import org.acejump.matchesAt
 | 
			
		||||
@@ -10,27 +11,26 @@ import org.acejump.matchesAt
 | 
			
		||||
/**
 | 
			
		||||
 * Searches editor text for matches of a [SearchQuery], and updates previous results when the user [type]s a character.
 | 
			
		||||
 */
 | 
			
		||||
internal class SearchProcessor private constructor(private val editor: Editor, query: SearchQuery, boundaries: Boundaries) {
 | 
			
		||||
class SearchProcessor private constructor(
 | 
			
		||||
  private val editors: List<Editor>, query: SearchQuery, results: MutableMap<Editor, IntArrayList>
 | 
			
		||||
) {
 | 
			
		||||
  companion object {
 | 
			
		||||
    fun fromChar(editor: Editor, char: Char, boundaries: Boundaries): SearchProcessor {
 | 
			
		||||
      return SearchProcessor(editor, SearchQuery.Literal(char.toString()), boundaries)
 | 
			
		||||
    fun fromChar(editors: List<Editor>, char: Char, boundaries: Boundaries): SearchProcessor {
 | 
			
		||||
      return SearchProcessor(editors, SearchQuery.Literal(char.toString()), boundaries)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fun fromRegex(editor: Editor, pattern: String, boundaries: Boundaries): SearchProcessor {
 | 
			
		||||
      return SearchProcessor(editor, SearchQuery.RegularExpression(pattern), boundaries)
 | 
			
		||||
    fun fromRegex(editors: List<Editor>, pattern: String, boundaries: Boundaries): SearchProcessor {
 | 
			
		||||
      return SearchProcessor(editors, SearchQuery.RegularExpression(pattern), boundaries)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  var query = query
 | 
			
		||||
    private set
 | 
			
		||||
  
 | 
			
		||||
  var results = IntArrayList(0)
 | 
			
		||||
    private set
 | 
			
		||||
  
 | 
			
		||||
  init {
 | 
			
		||||
  private constructor(editors: List<Editor>, query: SearchQuery, boundaries: Boundaries) : this(editors, query, mutableMapOf()) {
 | 
			
		||||
    val regex = query.toRegex()
 | 
			
		||||
    
 | 
			
		||||
    if (regex != null) {
 | 
			
		||||
      for (editor in editors) {
 | 
			
		||||
        val offsets = IntArrayList()
 | 
			
		||||
        
 | 
			
		||||
        val offsetRange = boundaries.getOffsetRange(editor)
 | 
			
		||||
        var result = regex.find(editor.immutableText, offsetRange.first)
 | 
			
		||||
        
 | 
			
		||||
@@ -42,13 +42,22 @@ internal class SearchProcessor private constructor(private val editor: Editor, q
 | 
			
		||||
            break
 | 
			
		||||
          }
 | 
			
		||||
          else if (boundaries.isOffsetInside(editor, index)) {
 | 
			
		||||
          results.add(index)
 | 
			
		||||
            offsets.add(index)
 | 
			
		||||
          }
 | 
			
		||||
          
 | 
			
		||||
          result = result.next()
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        results[editor] = offsets
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  internal var query = query
 | 
			
		||||
    private set
 | 
			
		||||
  
 | 
			
		||||
  internal var results = results
 | 
			
		||||
    private set
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Appends a character to the search query and removes all search results that no longer match the query. If the last typed character
 | 
			
		||||
@@ -57,13 +66,12 @@ internal class SearchProcessor private constructor(private val editor: Editor, q
 | 
			
		||||
   */
 | 
			
		||||
  fun type(char: Char, tagger: Tagger): Boolean {
 | 
			
		||||
    val newQuery = query.rawText + char
 | 
			
		||||
    val chars = editor.immutableText
 | 
			
		||||
    val canMatchTag = tagger.canQueryMatchAnyTag(newQuery)
 | 
			
		||||
    
 | 
			
		||||
    // If the typed character is not compatible with any existing tag or as a continuation of any previous occurrence, reject the query
 | 
			
		||||
    // change and return false to indicate that nothing else should happen.
 | 
			
		||||
    
 | 
			
		||||
    if (newQuery.length > 1 && !canMatchTag && results.none { chars.matchesAt(it, newQuery, ignoreCase = true) }) {
 | 
			
		||||
    if (newQuery.length > 1 && !canMatchTag && !isContinuation(newQuery)) {
 | 
			
		||||
      return false
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
@@ -76,7 +84,10 @@ internal class SearchProcessor private constructor(private val editor: Editor, q
 | 
			
		||||
      query = SearchQuery.Literal(char.toString())
 | 
			
		||||
      tagger.unmark()
 | 
			
		||||
      
 | 
			
		||||
      val iter = results.iterator()
 | 
			
		||||
      for ((editor, offsets) in results) {
 | 
			
		||||
        val chars = editor.immutableText
 | 
			
		||||
        val iter = offsets.iterator()
 | 
			
		||||
        
 | 
			
		||||
        while (iter.hasNext()) {
 | 
			
		||||
          val movedOffset = iter.nextInt() + newQuery.length - 1
 | 
			
		||||
          
 | 
			
		||||
@@ -88,6 +99,7 @@ internal class SearchProcessor private constructor(private val editor: Editor, q
 | 
			
		||||
          }
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    else {
 | 
			
		||||
      removeObsoleteResults(newQuery, tagger)
 | 
			
		||||
      query = SearchQuery.Literal(newQuery)
 | 
			
		||||
@@ -96,6 +108,20 @@ internal class SearchProcessor private constructor(private val editor: Editor, q
 | 
			
		||||
    return true
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Returns true if the new query is a continuation of any remaining search query.
 | 
			
		||||
   */
 | 
			
		||||
  private fun isContinuation(newQuery: String): Boolean {
 | 
			
		||||
    for ((editor, offsets) in results) {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      if (offsets.any { chars.matchesAt(it, newQuery, ignoreCase = true) }) {
 | 
			
		||||
        return true
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return false
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * After updating the query, removes all results that no longer match the search query.
 | 
			
		||||
   */
 | 
			
		||||
@@ -103,21 +129,28 @@ internal class SearchProcessor private constructor(private val editor: Editor, q
 | 
			
		||||
    val lastCharOffset = newQuery.lastIndex
 | 
			
		||||
    val lastChar = newQuery[lastCharOffset]
 | 
			
		||||
    val ignoreCase = newQuery[0].isLowerCase()
 | 
			
		||||
    
 | 
			
		||||
    for ((editor, offsets) in results.entries.toList()) {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      
 | 
			
		||||
      val remaining = IntArrayList()
 | 
			
		||||
    val iter = results.iterator()
 | 
			
		||||
      val iter = offsets.iterator()
 | 
			
		||||
      
 | 
			
		||||
      while (iter.hasNext()) {
 | 
			
		||||
        val offset = iter.nextInt()
 | 
			
		||||
        val endOffset = offset + lastCharOffset
 | 
			
		||||
        val lastTypedCharMatches = endOffset < chars.length && chars[endOffset].equals(lastChar, ignoreCase)
 | 
			
		||||
        
 | 
			
		||||
      if (lastTypedCharMatches || tagger.isQueryCompatibleWithTagAt(newQuery, offset)) {
 | 
			
		||||
        if (lastTypedCharMatches || tagger.isQueryCompatibleWithTagAt(newQuery, Tag(editor, offset))) {
 | 
			
		||||
          remaining.add(offset)
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
    results = remaining
 | 
			
		||||
      results[editor] = remaining
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun clone(): SearchProcessor {
 | 
			
		||||
    return SearchProcessor(editors, query, results.clone())
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -25,7 +25,7 @@ internal sealed class SearchQuery {
 | 
			
		||||
   * Each occurrence must either match the entire query, or match the query up to a point so that the rest of the query matches the
 | 
			
		||||
   * beginning of a tag at the location of the occurrence.
 | 
			
		||||
   */
 | 
			
		||||
  class Literal(override var rawText: String) : SearchQuery() {
 | 
			
		||||
  class Literal(override val rawText: String) : SearchQuery() {
 | 
			
		||||
    init {
 | 
			
		||||
      require(rawText.isNotEmpty())
 | 
			
		||||
    }
 | 
			
		||||
@@ -48,7 +48,7 @@ internal sealed class SearchQuery {
 | 
			
		||||
  /**
 | 
			
		||||
   * Searches for all matches of a regular expression.
 | 
			
		||||
   */
 | 
			
		||||
  class RegularExpression(private var pattern: String) : SearchQuery() {
 | 
			
		||||
  class RegularExpression(private val pattern: String) : SearchQuery() {
 | 
			
		||||
    override val rawText = ""
 | 
			
		||||
    
 | 
			
		||||
    override fun getHighlightLength(text: CharSequence, offset: Int): Int {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,8 +1,8 @@
 | 
			
		||||
package org.acejump.search
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.*
 | 
			
		||||
import it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.IntList
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.IntOpenHashSet
 | 
			
		||||
import org.acejump.boundaries.EditorOffsetCache
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
@@ -10,9 +10,7 @@ import org.acejump.immutableText
 | 
			
		||||
import org.acejump.input.KeyLayoutCache
 | 
			
		||||
import org.acejump.isWordPart
 | 
			
		||||
import org.acejump.wordEndPlus
 | 
			
		||||
import java.util.*
 | 
			
		||||
import kotlin.collections.HashMap
 | 
			
		||||
import kotlin.collections.HashSet
 | 
			
		||||
import java.util.IdentityHashMap
 | 
			
		||||
import kotlin.math.max
 | 
			
		||||
 | 
			
		||||
/*
 | 
			
		||||
@@ -53,50 +51,83 @@ import kotlin.math.max
 | 
			
		||||
 */
 | 
			
		||||
 | 
			
		||||
internal class Solver private constructor(
 | 
			
		||||
  private val editor: Editor,
 | 
			
		||||
  private val editorPriority: List<Editor>,
 | 
			
		||||
  private val queryLength: Int,
 | 
			
		||||
  private val newResults: IntList,
 | 
			
		||||
  private val allResults: IntList
 | 
			
		||||
  private val newResults: Map<Editor, IntList>,
 | 
			
		||||
  private val allResults: Map<Editor, IntList>,
 | 
			
		||||
) {
 | 
			
		||||
  companion object {
 | 
			
		||||
    fun solve(
 | 
			
		||||
      editor: Editor, query: SearchQuery, newResults: IntList, allResults: IntList, tags: List<String>, cache: EditorOffsetCache
 | 
			
		||||
    ): Map<String, Int> {
 | 
			
		||||
      return Solver(editor, max(1, query.rawText.length), newResults, allResults).map(tags, cache)
 | 
			
		||||
      editorPriority: List<Editor>,
 | 
			
		||||
      query: SearchQuery,
 | 
			
		||||
      newResults: Map<Editor, IntList>,
 | 
			
		||||
      allResults: Map<Editor, IntList>,
 | 
			
		||||
      tags: List<String>,
 | 
			
		||||
      caches: Map<Editor, EditorOffsetCache>,
 | 
			
		||||
    ): Map<String, Tag> {
 | 
			
		||||
      return Solver(editorPriority, max(1, query.rawText.length), newResults, allResults).map(tags, caches)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private var newTags = Object2IntOpenHashMap<String>(KeyLayoutCache.allPossibleTags.size)
 | 
			
		||||
  private val newTagIndices = IntOpenHashSet()
 | 
			
		||||
  private var newTags = HashMap<String, Tag>(KeyLayoutCache.allPossibleTags.size)
 | 
			
		||||
  private val newTagIndices = newResults.keys.associateWith { IntOpenHashSet() }
 | 
			
		||||
  
 | 
			
		||||
  private var allWordFragments = HashSet<String>(allResults.size).apply {
 | 
			
		||||
    val iter = allResults.iterator()
 | 
			
		||||
  private var allWordFragments = HashSet<String>(allResults.values.sumBy(IntList::size)).apply {
 | 
			
		||||
    for ((editor, offsets) in allResults) {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      val iter = offsets.iterator()
 | 
			
		||||
      while (iter.hasNext()) {
 | 
			
		||||
      forEachWordFragment(iter.nextInt()) { add(it) }
 | 
			
		||||
        forEachWordFragment(chars, iter.nextInt(), this::add)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun map(availableTags: List<String>, cache: EditorOffsetCache): Map<String, Int> {
 | 
			
		||||
    val eligibleSitesByTag = HashMap<String, IntList>(100)
 | 
			
		||||
  private fun generateEligibleSites(availableTags: List<String>): Map<String, MutableList<Tag>> {
 | 
			
		||||
    val eligibleSitesByTag = HashMap<String, MutableList<Tag>>(100)
 | 
			
		||||
    val tagsByFirstLetter = availableTags.groupBy { it[0] }
 | 
			
		||||
    
 | 
			
		||||
    val iter = newResults.iterator()
 | 
			
		||||
    for ((editor, offsets) in newResults) {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      val iter = offsets.iterator()
 | 
			
		||||
      while (iter.hasNext()) {
 | 
			
		||||
        val site = iter.nextInt()
 | 
			
		||||
        
 | 
			
		||||
        for ((firstLetter, tags) in tagsByFirstLetter.entries) {
 | 
			
		||||
        if (canTagBeginWithChar(site, firstLetter)) {
 | 
			
		||||
          if (canTagBeginWithChar(chars, site, firstLetter)) {
 | 
			
		||||
            for (tag in tags) {
 | 
			
		||||
            eligibleSitesByTag.getOrPut(tag) { IntArrayList(10) }.add(site)
 | 
			
		||||
              eligibleSitesByTag.getOrPut(tag, ::mutableListOf).add(Tag(editor, site))
 | 
			
		||||
            }
 | 
			
		||||
          }
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val matchingSites = HashMap<IntList, IntArray>()
 | 
			
		||||
    val matchingSitesAsArrays = IdentityHashMap<String, IntArray>() // Keys are guaranteed to be from a single collection.
 | 
			
		||||
    return eligibleSitesByTag
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun generateMatchingSites(
 | 
			
		||||
    eligibleSites: Map<String, MutableList<Tag>>,
 | 
			
		||||
    caches: Map<Editor, EditorOffsetCache>,
 | 
			
		||||
  ): Map<String, Iterator<Tag>> {
 | 
			
		||||
    val matchingSites = HashMap<MutableList<Tag>, Iterator<Tag>>()
 | 
			
		||||
    val matchingSitesSorted = IdentityHashMap<String, Iterator<Tag>>() // Keys are guaranteed to be from a single collection.
 | 
			
		||||
    
 | 
			
		||||
    val siteOrder = siteOrder(caches)
 | 
			
		||||
    
 | 
			
		||||
    for ((mark, tags) in eligibleSites.entries) {
 | 
			
		||||
      matchingSitesSorted[mark] = matchingSites.getOrPut(tags) {
 | 
			
		||||
        @Suppress("ConvertLambdaToReference")
 | 
			
		||||
        tags.toMutableList().apply { sortWith(siteOrder) }.iterator()
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return matchingSitesSorted
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun map(availableTags: List<String>, caches: Map<Editor, EditorOffsetCache>): Map<String, Tag> {
 | 
			
		||||
    val eligibleSitesByTag = generateEligibleSites(availableTags)
 | 
			
		||||
    val matchingSitesSorted = generateMatchingSites(eligibleSitesByTag, caches)
 | 
			
		||||
    
 | 
			
		||||
    val siteOrder = siteOrder(cache)
 | 
			
		||||
    val tagOrder = KeyLayoutCache.tagOrder
 | 
			
		||||
      .thenComparingInt { eligibleSitesByTag.getValue(it).size }
 | 
			
		||||
      .thenBy(AceConfig.layout.priority(String::last))
 | 
			
		||||
@@ -105,20 +136,15 @@ internal class Solver private constructor(
 | 
			
		||||
      sortWith(tagOrder)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    for ((key, value) in eligibleSitesByTag.entries) {
 | 
			
		||||
      matchingSitesAsArrays[key] = matchingSites.getOrPut(value) {
 | 
			
		||||
        value.toIntArray().apply { IntArrays.mergeSort(this, siteOrder) }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    var totalAssigned = 0
 | 
			
		||||
    val totalResults = newResults.values.sumBy(IntList::size)
 | 
			
		||||
    
 | 
			
		||||
    for (tag in sortedTags) {
 | 
			
		||||
      if (totalAssigned == newResults.size) {
 | 
			
		||||
      if (totalAssigned == totalResults) {
 | 
			
		||||
        break
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      if (tryToAssignTag(tag, matchingSitesAsArrays.getValue(tag))) {
 | 
			
		||||
      if (tryToAssignTag(tag, matchingSitesSorted.getValue(tag))) {
 | 
			
		||||
        totalAssigned++
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
@@ -126,50 +152,63 @@ internal class Solver private constructor(
 | 
			
		||||
    return newTags
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun tryToAssignTag(tag: String, sites: IntArray): Boolean {
 | 
			
		||||
    if (newTags.containsKey(tag)) {
 | 
			
		||||
  private fun tryToAssignTag(mark: String, tags: Iterator<Tag>): Boolean {
 | 
			
		||||
    if (newTags.containsKey(mark)) {
 | 
			
		||||
      return false
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val index = sites.firstOrNull { it !in newTagIndices } ?: return false
 | 
			
		||||
    while (tags.hasNext()) {
 | 
			
		||||
      val tag = tags.next()
 | 
			
		||||
      val assigned = newTagIndices.getValue(tag.editor)
 | 
			
		||||
      
 | 
			
		||||
    @Suppress("ReplacePutWithAssignment")
 | 
			
		||||
    newTags.put(tag, index)
 | 
			
		||||
    newTagIndices.add(index)
 | 
			
		||||
      if (tag.offset !in assigned) {
 | 
			
		||||
        newTags[mark] = tag
 | 
			
		||||
        assigned.add(tag.offset)
 | 
			
		||||
        return true
 | 
			
		||||
      }
 | 
			
		||||
  
 | 
			
		||||
  private fun siteOrder(cache: EditorOffsetCache) = IntComparator { a, b ->
 | 
			
		||||
    val aIsVisible = StandardBoundaries.VISIBLE_ON_SCREEN.isOffsetInside(editor, a, cache)
 | 
			
		||||
    val bIsVisible = StandardBoundaries.VISIBLE_ON_SCREEN.isOffsetInside(editor, b, cache)
 | 
			
		||||
    
 | 
			
		||||
    if (aIsVisible != bIsVisible) {
 | 
			
		||||
      // Sites in immediate view should come first.
 | 
			
		||||
      return@IntComparator if (aIsVisible) -1 else 1
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val chars = editor.immutableText
 | 
			
		||||
    val aIsNotWordStart = chars[max(0, a - 1)].isWordPart
 | 
			
		||||
    val bIsNotWordStart = chars[max(0, b - 1)].isWordPart
 | 
			
		||||
    return false
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun siteOrder(caches: Map<Editor, EditorOffsetCache>) = Comparator<Tag> { a, b ->
 | 
			
		||||
    val aEditor = a.editor
 | 
			
		||||
    val bEditor = b.editor
 | 
			
		||||
    
 | 
			
		||||
    if (aEditor !== bEditor) {
 | 
			
		||||
      val aEditorIndex = editorPriority.indexOf(aEditor)
 | 
			
		||||
      val bEditorIndex = editorPriority.indexOf(bEditor)
 | 
			
		||||
      // For multiple editors, prioritize them based on the provided order.
 | 
			
		||||
      return@Comparator if (aEditorIndex < bEditorIndex) -1 else 1
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val aIsVisible = StandardBoundaries.VISIBLE_ON_SCREEN.isOffsetInside(aEditor, a.offset, caches.getValue(aEditor))
 | 
			
		||||
    val bIsVisible = StandardBoundaries.VISIBLE_ON_SCREEN.isOffsetInside(bEditor, b.offset, caches.getValue(bEditor))
 | 
			
		||||
    if (aIsVisible != bIsVisible) {
 | 
			
		||||
      // Sites in immediate view should come first.
 | 
			
		||||
      return@Comparator if (aIsVisible) -1 else 1
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val aIsNotWordStart = aEditor.immutableText[max(0, a.offset - 1)].isWordPart
 | 
			
		||||
    val bIsNotWordStart = bEditor.immutableText[max(0, b.offset - 1)].isWordPart
 | 
			
		||||
    if (aIsNotWordStart != bIsNotWordStart) {
 | 
			
		||||
      // Ensure that the first letter of a word is prioritized for tagging.
 | 
			
		||||
      return@IntComparator if (bIsNotWordStart) -1 else 1
 | 
			
		||||
      return@Comparator if (bIsNotWordStart) -1 else 1
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    when {
 | 
			
		||||
      a < b -> -1
 | 
			
		||||
      a > b -> 1
 | 
			
		||||
      a.offset < b.offset -> -1
 | 
			
		||||
      a.offset > b.offset -> 1
 | 
			
		||||
      else                -> 0
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun canTagBeginWithChar(site: Int, char: Char): Boolean {
 | 
			
		||||
  private fun canTagBeginWithChar(chars: CharSequence, site: Int, char: Char): Boolean {
 | 
			
		||||
    if (char.toString() in allWordFragments) {
 | 
			
		||||
      return false
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    forEachWordFragment(site) {
 | 
			
		||||
    forEachWordFragment(chars, site) {
 | 
			
		||||
      if (it + char in allWordFragments) {
 | 
			
		||||
        return false
 | 
			
		||||
      }
 | 
			
		||||
@@ -178,8 +217,7 @@ internal class Solver private constructor(
 | 
			
		||||
    return true
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private inline fun forEachWordFragment(site: Int, callback: (String) -> Unit) {
 | 
			
		||||
    val chars = editor.immutableText
 | 
			
		||||
  private inline fun forEachWordFragment(chars: CharSequence, site: Int, callback: (String) -> Unit) {
 | 
			
		||||
    val left = max(0, site + queryLength - 1)
 | 
			
		||||
    val right = chars.wordEndPlus(site)
 | 
			
		||||
    
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										13
									
								
								src/main/kotlin/org/acejump/search/Tag.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								src/main/kotlin/org/acejump/search/Tag.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,13 @@
 | 
			
		||||
package org.acejump.search
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
 | 
			
		||||
data class Tag(val editor: Editor, val offset: Int) {
 | 
			
		||||
  override fun equals(other: Any?): Boolean {
 | 
			
		||||
    return other is Tag && other.offset == offset && other.editor === editor
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun hashCode(): Int {
 | 
			
		||||
    return (offset * 31) + editor.hashCode()
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -1,5 +1,6 @@
 | 
			
		||||
package org.acejump.search
 | 
			
		||||
 | 
			
		||||
import com.google.common.collect.ArrayListMultimap
 | 
			
		||||
import com.google.common.collect.HashBiMap
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.IntArrayList
 | 
			
		||||
@@ -11,39 +12,41 @@ import org.acejump.immutableText
 | 
			
		||||
import org.acejump.input.KeyLayoutCache.allPossibleTags
 | 
			
		||||
import org.acejump.isWordPart
 | 
			
		||||
import org.acejump.matchesAt
 | 
			
		||||
import org.acejump.view.Tag
 | 
			
		||||
import org.acejump.view.TagMarker
 | 
			
		||||
import java.util.AbstractMap.SimpleImmutableEntry
 | 
			
		||||
import kotlin.collections.component1
 | 
			
		||||
import kotlin.collections.component2
 | 
			
		||||
import kotlin.math.min
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Assigns tags to search occurrences, updates them when the search query changes, and requests a jump if the search query matches a tag.
 | 
			
		||||
 * The ordering of [editors] may be used to prioritize tagging editors earlier in the list in case of conflicts.
 | 
			
		||||
 */
 | 
			
		||||
internal class Tagger(private val editor: Editor) {
 | 
			
		||||
  private var tagMap = HashBiMap.create<String, Int>()
 | 
			
		||||
class Tagger(private val editors: List<Editor>) {
 | 
			
		||||
  private var tagMap = HashBiMap.create<String, Tag>()
 | 
			
		||||
  
 | 
			
		||||
  val hasTags
 | 
			
		||||
    get() = tagMap.isNotEmpty()
 | 
			
		||||
  
 | 
			
		||||
  @ExternalUsage
 | 
			
		||||
  val tags
 | 
			
		||||
    get() = tagMap.map { SimpleImmutableEntry(it.key, it.value) }.sortedBy { it.value }
 | 
			
		||||
  internal val tags
 | 
			
		||||
    get() = tagMap.map { SimpleImmutableEntry(it.key, it.value) }.sortedBy { it.value.offset }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Removes all markers, allowing them to be regenerated from scratch.
 | 
			
		||||
   */
 | 
			
		||||
  fun unmark() {
 | 
			
		||||
  internal fun unmark() {
 | 
			
		||||
    tagMap = HashBiMap.create()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Assigns tags to as many results as possible, keeping previously assigned tags. Returns a [TaggingResult.Jump] if the current search
 | 
			
		||||
   * Assigns tags to as many results as possible, keeping previously assigned tags. Returns a [TaggingResult.Accept] if the current search
 | 
			
		||||
   * query matches any existing tag and we should jump to it and end the session, or [TaggingResult.Mark] to continue the session with
 | 
			
		||||
   * updated tag markers.
 | 
			
		||||
   *
 | 
			
		||||
   * Note that the [results] collection will be mutated.
 | 
			
		||||
   */
 | 
			
		||||
  fun markOrJump(query: SearchQuery, results: IntList): TaggingResult {
 | 
			
		||||
  internal fun update(query: SearchQuery, results: Map<Editor, IntList>): TaggingResult {
 | 
			
		||||
    val isRegex = query is SearchQuery.RegularExpression
 | 
			
		||||
    val queryText = if (isRegex) " ${query.rawText}" else query.rawText[0] + query.rawText.drop(1).toLowerCase()
 | 
			
		||||
    
 | 
			
		||||
@@ -52,12 +55,14 @@ internal class Tagger(private val editor: Editor) {
 | 
			
		||||
    if (!isRegex) {
 | 
			
		||||
      for (entry in tagMap.entries) {
 | 
			
		||||
        if (entry solves queryText) {
 | 
			
		||||
          return TaggingResult.Jump(entry.value)
 | 
			
		||||
          return TaggingResult.Accept(entry.value)
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      if (queryText.length == 1) {
 | 
			
		||||
        removeResultsWithOverlappingTags(results)
 | 
			
		||||
        for ((editor, offsets) in results) {
 | 
			
		||||
          removeResultsWithOverlappingTags(editor, offsets)
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
@@ -65,16 +70,26 @@ internal class Tagger(private val editor: Editor) {
 | 
			
		||||
      tagMap = assignTagsAndMerge(results, availableTags, query, queryText)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return TaggingResult.Mark(createTagMarkers(results, query.rawText.ifEmpty { null }))
 | 
			
		||||
    val resultTags = results.flatMap { (editor, offsets) -> offsets.map { Tag(editor, it) } }
 | 
			
		||||
    return TaggingResult.Mark(createTagMarkers(resultTags, query.rawText.ifEmpty { null }))
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun clone(): Tagger {
 | 
			
		||||
    return Tagger(editors).also { it.tagMap.putAll(tagMap) }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Assigns as many unassigned tags as possible, and merges them with the existing compatible tags.
 | 
			
		||||
   */
 | 
			
		||||
  private fun assignTagsAndMerge(results: IntList, availableTags: List<String>, query: SearchQuery, queryText: String): HashBiMap<String, Int> {
 | 
			
		||||
    val cache = EditorOffsetCache.new()
 | 
			
		||||
  private fun assignTagsAndMerge(
 | 
			
		||||
    results: Map<Editor, IntList>, availableTags: List<String>, query: SearchQuery, queryText: String,
 | 
			
		||||
  ): HashBiMap<String, Tag> {
 | 
			
		||||
    val caches = results.keys.associateWith { EditorOffsetCache.new() }
 | 
			
		||||
    
 | 
			
		||||
    results.sort { a, b ->
 | 
			
		||||
    for ((editor, offsets) in results) {
 | 
			
		||||
      val cache = caches.getValue(editor)
 | 
			
		||||
      
 | 
			
		||||
      offsets.sort { a, b ->
 | 
			
		||||
        val aIsVisible = StandardBoundaries.VISIBLE_ON_SCREEN.isOffsetInside(editor, a, cache)
 | 
			
		||||
        val bIsVisible = StandardBoundaries.VISIBLE_ON_SCREEN.isOffsetInside(editor, b, cache)
 | 
			
		||||
        
 | 
			
		||||
@@ -84,52 +99,61 @@ internal class Tagger(private val editor: Editor) {
 | 
			
		||||
          else                      -> 0
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val allAssignedTags = mutableMapOf<String, Int>()
 | 
			
		||||
    val oldCompatibleTags = tagMap.filter { isTagCompatibleWithQuery(it.key, it.value, queryText) || it.value in results }
 | 
			
		||||
    val vacantResults: IntList
 | 
			
		||||
    val allAssignedTags = mutableMapOf<String, Tag>()
 | 
			
		||||
    val oldCompatibleTags = tagMap.filter { (mark, tag) ->
 | 
			
		||||
      isTagCompatibleWithQuery(mark, tag, queryText) || results[tag.editor]?.contains(tag.offset) == true
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val vacantResults: Map<Editor, IntList>
 | 
			
		||||
    
 | 
			
		||||
    if (oldCompatibleTags.isEmpty()) {
 | 
			
		||||
      vacantResults = results
 | 
			
		||||
    }
 | 
			
		||||
    else {
 | 
			
		||||
      vacantResults = IntArrayList()
 | 
			
		||||
      val vacant = mutableMapOf<Editor, IntList>()
 | 
			
		||||
      
 | 
			
		||||
      for ((editor, offsets) in results) {
 | 
			
		||||
        val list = IntArrayList()
 | 
			
		||||
        val iter = offsets.iterator()
 | 
			
		||||
        
 | 
			
		||||
      val iter = results.iterator()
 | 
			
		||||
        while (iter.hasNext()) {
 | 
			
		||||
        val offset = iter.nextInt()
 | 
			
		||||
          val tag = Tag(editor, iter.nextInt())
 | 
			
		||||
          if (tag !in oldCompatibleTags.values) {
 | 
			
		||||
            list.add(tag.offset)
 | 
			
		||||
          }
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        if (offset !in oldCompatibleTags.values) {
 | 
			
		||||
          vacantResults.add(offset)
 | 
			
		||||
        }
 | 
			
		||||
        vacant[editor] = list
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      vacantResults = vacant
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    allAssignedTags.putAll(oldCompatibleTags)
 | 
			
		||||
    allAssignedTags.putAll(Solver.solve(editor, query, vacantResults, results, availableTags, cache))
 | 
			
		||||
    allAssignedTags.putAll(Solver.solve(editors, query, vacantResults, results, availableTags, caches))
 | 
			
		||||
    
 | 
			
		||||
    val assignedMarkers = allAssignedTags.keys.groupBy { it[0] }
 | 
			
		||||
    
 | 
			
		||||
    return allAssignedTags.mapKeysTo(HashBiMap.create(allAssignedTags.size)) { (tag, _) ->
 | 
			
		||||
      // Avoid matching query - will trigger a jump.
 | 
			
		||||
      // TODO: lift this constraint.
 | 
			
		||||
      val queryEndsWith = queryText.endsWith(tag[0]) || queryText.endsWith(tag)
 | 
			
		||||
      
 | 
			
		||||
      if (!queryEndsWith && canShortenTag(tag, allAssignedTags))
 | 
			
		||||
      if (canShortenTag(tag, assignedMarkers, queryText))
 | 
			
		||||
        tag[0].toString()
 | 
			
		||||
      else
 | 
			
		||||
        tag
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private infix fun Map.Entry<String, Int>.solves(query: String): Boolean {
 | 
			
		||||
  private infix fun Map.Entry<String, Tag>.solves(query: String): Boolean {
 | 
			
		||||
    return query.endsWith(key, true) && isTagCompatibleWithQuery(key, value, query)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun isTagCompatibleWithQuery(tag: String, offset: Int, query: String): Boolean {
 | 
			
		||||
    return editor.immutableText.matchesAt(offset, getPlaintextPortion(query, tag), ignoreCase = true)
 | 
			
		||||
  private fun isTagCompatibleWithQuery(marker: String, tag: Tag, query: String): Boolean {
 | 
			
		||||
    return tag.editor.immutableText.matchesAt(tag.offset, getPlaintextPortion(query, marker), ignoreCase = true)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun isQueryCompatibleWithTagAt(query: String, offset: Int): Boolean {
 | 
			
		||||
    return tagMap.inverse()[offset].let { it != null && isTagCompatibleWithQuery(it, offset, query) }
 | 
			
		||||
  fun isQueryCompatibleWithTagAt(query: String, tag: Tag): Boolean {
 | 
			
		||||
    return tagMap.inverse()[tag].let { it != null && isTagCompatibleWithQuery(it, tag, query) }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun canQueryMatchAnyTag(query: String): Boolean {
 | 
			
		||||
@@ -139,8 +163,8 @@ internal class Tagger(private val editor: Editor) {
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun removeResultsWithOverlappingTags(results: IntList) {
 | 
			
		||||
    val iter = results.iterator()
 | 
			
		||||
  private fun removeResultsWithOverlappingTags(editor: Editor, offsets: IntList) {
 | 
			
		||||
    val iter = offsets.iterator()
 | 
			
		||||
    val chars = editor.immutableText
 | 
			
		||||
    
 | 
			
		||||
    while (iter.hasNext()) {
 | 
			
		||||
@@ -150,9 +174,18 @@ internal class Tagger(private val editor: Editor) {
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun createTagMarkers(results: IntList, literalQueryText: String?): List<Tag> {
 | 
			
		||||
  private fun createTagMarkers(tags: Collection<Tag>, literalQueryText: String?): MutableMap<Editor, Collection<TagMarker>> {
 | 
			
		||||
    val tagMapInv = tagMap.inverse()
 | 
			
		||||
    return results.mapNotNull { index -> tagMapInv[index]?.let { tag -> Tag.create(editor, tag, index, literalQueryText) } }
 | 
			
		||||
    val markers = ArrayListMultimap.create<Editor, TagMarker>(editors.size, min(tags.size, 50))
 | 
			
		||||
    
 | 
			
		||||
    for (tag in tags) {
 | 
			
		||||
      val mark = tagMapInv[tag] ?: continue
 | 
			
		||||
      val editor = tag.editor
 | 
			
		||||
      val marker = TagMarker.create(editor, mark, tag.offset, literalQueryText)
 | 
			
		||||
      markers.put(editor, marker)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return markers.asMap()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private companion object {
 | 
			
		||||
@@ -173,26 +206,28 @@ internal class Tagger(private val editor: Editor) {
 | 
			
		||||
      return this.isWordPart xor other.isWordPart || this.isWhitespace() xor other.isWhitespace()
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun getPlaintextPortion(query: String, tag: String) = when {
 | 
			
		||||
      query.endsWith(tag, true)         -> query.dropLast(tag.length)
 | 
			
		||||
      query.endsWith(tag.first(), true) -> query.dropLast(1)
 | 
			
		||||
    private fun getPlaintextPortion(query: String, marker: String) = when {
 | 
			
		||||
      query.endsWith(marker, true)         -> query.dropLast(marker.length)
 | 
			
		||||
      query.endsWith(marker.first(), true) -> query.dropLast(1)
 | 
			
		||||
      else                                 -> query
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun getTagPortion(query: String, tag: String) = when {
 | 
			
		||||
      query.endsWith(tag, true)         -> query.takeLast(tag.length)
 | 
			
		||||
      query.endsWith(tag.first(), true) -> query.takeLast(1)
 | 
			
		||||
    private fun getTagPortion(query: String, marker: String) = when {
 | 
			
		||||
      query.endsWith(marker, true)         -> query.takeLast(marker.length)
 | 
			
		||||
      query.endsWith(marker.first(), true) -> query.takeLast(1)
 | 
			
		||||
      else                                 -> ""
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun canShortenTag(tag: String, tagMap: Map<String, Int>): Boolean {
 | 
			
		||||
      for (other in tagMap.keys) {
 | 
			
		||||
        if (tag != other && tag[0] == other[0]) {
 | 
			
		||||
    private fun canShortenTag(marker: String, markers: Map<Char, List<String>>, queryText: String): Boolean {
 | 
			
		||||
      // Avoid matching query - will trigger a jump.
 | 
			
		||||
      // TODO: lift this constraint.
 | 
			
		||||
      val queryEndsWith = queryText.endsWith(marker[0]) || queryText.endsWith(marker)
 | 
			
		||||
      if (queryEndsWith) {
 | 
			
		||||
        return false
 | 
			
		||||
      }
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      return true
 | 
			
		||||
      val startingWithSameLetter = markers[marker[0]]
 | 
			
		||||
      return startingWithSameLetter == null || startingWithSameLetter.singleOrNull() == marker
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,8 +1,9 @@
 | 
			
		||||
package org.acejump.search
 | 
			
		||||
 | 
			
		||||
import org.acejump.view.Tag
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import org.acejump.view.TagMarker
 | 
			
		||||
 | 
			
		||||
internal sealed class TaggingResult {
 | 
			
		||||
  class Jump(val offset: Int) : TaggingResult()
 | 
			
		||||
  class Mark(val tags: List<Tag>) : TaggingResult()
 | 
			
		||||
  class Accept(val tag: Tag) : TaggingResult()
 | 
			
		||||
  class Mark(val markers: MutableMap<Editor, Collection<TagMarker>>) : TaggingResult()
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -26,6 +26,22 @@ internal data class EditorSettings(private val isBlockCursor: Boolean, private v
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun startEditing(editor: Editor) {
 | 
			
		||||
    editor.document.setReadOnly(isReadOnly)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun stopEditing(editor: Editor) {
 | 
			
		||||
    editor.document.setReadOnly(true)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun onTagAccepted(editor: Editor) = editor.let {
 | 
			
		||||
    it.settings.isBlockCursor = isBlockCursor
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun onTagUnaccepted(editor: Editor) = editor.let {
 | 
			
		||||
    it.settings.isBlockCursor = true
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun restore(editor: Editor) {
 | 
			
		||||
    val settings = editor.settings
 | 
			
		||||
    val document = editor.document
 | 
			
		||||
 
 | 
			
		||||
@@ -1,179 +1,219 @@
 | 
			
		||||
package org.acejump.session
 | 
			
		||||
 | 
			
		||||
import com.intellij.codeInsight.hint.HintManager
 | 
			
		||||
import com.intellij.codeInsight.hint.HintManagerImpl
 | 
			
		||||
import com.intellij.codeInsight.hint.HintUtil
 | 
			
		||||
import com.intellij.openapi.actionSystem.DataContext
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.openapi.editor.ScrollType
 | 
			
		||||
import com.intellij.openapi.editor.actionSystem.TypedActionHandler
 | 
			
		||||
import com.intellij.openapi.editor.colors.EditorColors
 | 
			
		||||
import com.intellij.openapi.editor.colors.impl.AbstractColorsScheme
 | 
			
		||||
import com.intellij.ui.LightweightHint
 | 
			
		||||
import org.acejump.ExternalUsage
 | 
			
		||||
import org.acejump.action.TagJumper
 | 
			
		||||
import org.acejump.action.TagVisitor
 | 
			
		||||
import org.acejump.boundaries.Boundaries
 | 
			
		||||
import org.acejump.boundaries.EditorOffsetCache
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries
 | 
			
		||||
import org.acejump.clone
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import org.acejump.immutableText
 | 
			
		||||
import org.acejump.input.EditorKeyListener
 | 
			
		||||
import org.acejump.input.JumpMode
 | 
			
		||||
import org.acejump.input.JumpModeTracker
 | 
			
		||||
import org.acejump.input.KeyLayoutCache
 | 
			
		||||
import org.acejump.search.Pattern
 | 
			
		||||
import org.acejump.search.SearchProcessor
 | 
			
		||||
import org.acejump.search.Tagger
 | 
			
		||||
import org.acejump.search.TaggingResult
 | 
			
		||||
import org.acejump.modes.AdvancedMode
 | 
			
		||||
import org.acejump.modes.BetweenPointsMode
 | 
			
		||||
import org.acejump.modes.JumpMode
 | 
			
		||||
import org.acejump.modes.SessionMode
 | 
			
		||||
import org.acejump.search.*
 | 
			
		||||
import org.acejump.view.TagCanvas
 | 
			
		||||
import org.acejump.view.TextHighlighter
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Manages an AceJump session for a single [Editor].
 | 
			
		||||
 */
 | 
			
		||||
class Session(private val editor: Editor) {
 | 
			
		||||
  private companion object {
 | 
			
		||||
    private val defaultBoundaries
 | 
			
		||||
      get() = if (AceConfig.searchWholeFile) StandardBoundaries.WHOLE_FILE else StandardBoundaries.VISIBLE_ON_SCREEN
 | 
			
		||||
  }
 | 
			
		||||
class Session(private val mainEditor: Editor, private val jumpEditors: List<Editor>) {
 | 
			
		||||
  private val editorSettings = EditorSettings.setup(mainEditor)
 | 
			
		||||
  private lateinit var mode: SessionMode
 | 
			
		||||
  
 | 
			
		||||
  private val originalSettings = EditorSettings.setup(editor)
 | 
			
		||||
  private var state: SessionStateImpl? = null
 | 
			
		||||
  private var tagger = Tagger(jumpEditors)
 | 
			
		||||
  
 | 
			
		||||
  private val jumpModeTracker = JumpModeTracker()
 | 
			
		||||
  private var jumpMode = JumpMode.DISABLED
 | 
			
		||||
  private var acceptedTag: Tag? = null
 | 
			
		||||
    set(value) {
 | 
			
		||||
      field = value
 | 
			
		||||
      
 | 
			
		||||
      if (value === JumpMode.DISABLED) {
 | 
			
		||||
        end()
 | 
			
		||||
      }
 | 
			
		||||
      else {
 | 
			
		||||
        searchProcessor?.let { textHighlighter.render(it.results, it.query, jumpMode) }
 | 
			
		||||
        editor.colorsScheme.setColor(EditorColors.CARET_COLOR, value.caretColor)
 | 
			
		||||
        editor.contentComponent.repaint()
 | 
			
		||||
      if (value != null) {
 | 
			
		||||
        tagCanvases.values.forEach(TagCanvas::removeMarkers)
 | 
			
		||||
        editorSettings.onTagAccepted(mainEditor)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  
 | 
			
		||||
  private var searchProcessor: SearchProcessor? = null
 | 
			
		||||
  private var tagger = Tagger(editor)
 | 
			
		||||
  
 | 
			
		||||
  private val tagJumper
 | 
			
		||||
    get() = TagJumper(editor, jumpMode, searchProcessor)
 | 
			
		||||
  
 | 
			
		||||
  private val tagVisitor
 | 
			
		||||
    get() = searchProcessor?.let { TagVisitor(editor, it, tagJumper) }
 | 
			
		||||
  
 | 
			
		||||
  private val textHighlighter = TextHighlighter(editor)
 | 
			
		||||
  private val tagCanvas = TagCanvas(editor)
 | 
			
		||||
  private val textHighlighter = TextHighlighter()
 | 
			
		||||
  private val tagCanvases = jumpEditors.associateWith(::TagCanvas)
 | 
			
		||||
  
 | 
			
		||||
  @ExternalUsage
 | 
			
		||||
  val tags
 | 
			
		||||
    get() = tagger.tags
 | 
			
		||||
  
 | 
			
		||||
  var defaultBoundary: Boundaries = StandardBoundaries.VISIBLE_ON_SCREEN
 | 
			
		||||
  
 | 
			
		||||
  init {
 | 
			
		||||
    KeyLayoutCache.ensureInitialized(AceConfig.settings)
 | 
			
		||||
    
 | 
			
		||||
    EditorKeyListener.attach(editor, object : TypedActionHandler {
 | 
			
		||||
    EditorKeyListener.attach(mainEditor, object : TypedActionHandler {
 | 
			
		||||
      override fun execute(editor: Editor, charTyped: Char, context: DataContext) {
 | 
			
		||||
        var processor = searchProcessor
 | 
			
		||||
        val state = state ?: return
 | 
			
		||||
        val hadTags = tagger.hasTags
 | 
			
		||||
        
 | 
			
		||||
        if (processor == null) {
 | 
			
		||||
          processor = SearchProcessor.fromChar(editor, charTyped, defaultBoundaries).also { searchProcessor = it }
 | 
			
		||||
        }
 | 
			
		||||
        else if (!processor.type(charTyped, tagger)) {
 | 
			
		||||
          return
 | 
			
		||||
        editorSettings.startEditing(editor)
 | 
			
		||||
        val result = mode.type(state, charTyped, acceptedTag)
 | 
			
		||||
        editorSettings.stopEditing(editor)
 | 
			
		||||
        
 | 
			
		||||
        when (result) {
 | 
			
		||||
          TypeResult.Nothing          -> updateHint()
 | 
			
		||||
          is TypeResult.UpdateResults -> updateSearch(result.processor, markImmediately = hadTags)
 | 
			
		||||
          is TypeResult.ChangeMode    -> setMode(result.mode)
 | 
			
		||||
          
 | 
			
		||||
          TypeResult.RestartSearch    -> restart().also {
 | 
			
		||||
            this@Session.state = SessionStateImpl(jumpEditors, tagger, defaultBoundary)
 | 
			
		||||
            updateHint()
 | 
			
		||||
          }
 | 
			
		||||
          
 | 
			
		||||
        updateSearch(processor, markImmediately = hadTags, shiftMode = charTyped.isUpperCase())
 | 
			
		||||
          TypeResult.EndSession       -> end()
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    })
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Updates text highlights and tag markers according to the current search state. Dispatches jumps if the search query matches a tag.
 | 
			
		||||
   * If all tags are outside view, scrolls to the closest one.
 | 
			
		||||
   */
 | 
			
		||||
  private fun updateSearch(processor: SearchProcessor, markImmediately: Boolean, shiftMode: Boolean = false) {
 | 
			
		||||
  private fun updateSearch(processor: SearchProcessor, markImmediately: Boolean) {
 | 
			
		||||
    val query = processor.query
 | 
			
		||||
    val results = processor.results
 | 
			
		||||
    
 | 
			
		||||
    textHighlighter.render(results, query, jumpMode)
 | 
			
		||||
    
 | 
			
		||||
    if (!markImmediately && query.rawText.let { it.length < AceConfig.minQueryLength && it.all(Char::isLetterOrDigit) }) {
 | 
			
		||||
      textHighlighter.renderOccurrences(results, query)
 | 
			
		||||
      return
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    when (val result = tagger.markOrJump(query, results.clone())) {
 | 
			
		||||
      is TaggingResult.Jump -> {
 | 
			
		||||
        tagJumper.jump(result.offset, shiftMode)
 | 
			
		||||
        tagCanvas.removeMarkers()
 | 
			
		||||
    when (val result = tagger.update(query, results.clone())) {
 | 
			
		||||
      is TaggingResult.Accept -> {
 | 
			
		||||
        acceptedTag = result.tag
 | 
			
		||||
        textHighlighter.renderFinal(result.tag, processor.query)
 | 
			
		||||
        
 | 
			
		||||
        if (state?.let { mode.accept(it, result.tag) } == true) {
 | 
			
		||||
          end()
 | 
			
		||||
          return
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      is TaggingResult.Mark   -> {
 | 
			
		||||
        val tags = result.tags
 | 
			
		||||
        tagCanvas.setMarkers(tags)
 | 
			
		||||
        for ((editor, canvas) in tagCanvases) {
 | 
			
		||||
          canvas.setMarkers(result.markers[editor].orEmpty())
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        val cache = EditorOffsetCache.new()
 | 
			
		||||
        val boundaries = StandardBoundaries.VISIBLE_ON_SCREEN
 | 
			
		||||
        textHighlighter.renderOccurrences(results, query)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
        if (tags.none { boundaries.isOffsetInside(editor, it.offsetL, cache) || boundaries.isOffsetInside(editor, it.offsetR, cache) }) {
 | 
			
		||||
          tagVisitor?.scrollToClosest()
 | 
			
		||||
    updateHint()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun setMode(mode: SessionMode) {
 | 
			
		||||
    this.mode = mode
 | 
			
		||||
    mainEditor.colorsScheme.setColor(EditorColors.CARET_COLOR, mode.caretColor)
 | 
			
		||||
    updateHint()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private fun updateHint() {
 | 
			
		||||
    val acceptedTag = acceptedTag
 | 
			
		||||
    val editor = mainEditor
 | 
			
		||||
    val offset = when {
 | 
			
		||||
      acceptedTag == null           -> null
 | 
			
		||||
      acceptedTag.editor === editor -> acceptedTag.offset
 | 
			
		||||
      else                          -> mainEditor.caretModel.offset
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val hintArray = mode.getHint(offset, state?.currentProcessor.let { it != null && it.query.rawText.isNotEmpty() }) ?: return
 | 
			
		||||
    val hintText = hintArray
 | 
			
		||||
      .joinToString("\n")
 | 
			
		||||
      .replace("<f>", "<span style=\"font-family:'${editor.colorsScheme.editorFontName}';font-weight:bold\">")
 | 
			
		||||
      .replace("</f>", "</span>")
 | 
			
		||||
    
 | 
			
		||||
    val hint = LightweightHint(HintUtil.createInformationLabel(hintText))
 | 
			
		||||
    val pos = offset?.let(editor::offsetToLogicalPosition) ?: editor.caretModel.logicalPosition
 | 
			
		||||
    val point = HintManagerImpl.getHintPosition(hint, editor, pos, HintManager.ABOVE)
 | 
			
		||||
    val info = HintManagerImpl.createHintHint(editor, point, hint, HintManager.ABOVE).setShowImmediately(true)
 | 
			
		||||
    val flags = HintManager.UPDATE_BY_SCROLLING or HintManager.HIDE_BY_ESCAPE
 | 
			
		||||
    HintManagerImpl.getInstanceImpl().showEditorHint(hint, editor, point, flags, 0, true, info)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun startJumpMode() {
 | 
			
		||||
    startJumpMode(::JumpMode)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun startJumpMode(mode: () -> JumpMode) {
 | 
			
		||||
    if (this::mode.isInitialized && mode is JumpMode) {
 | 
			
		||||
      end()
 | 
			
		||||
      return
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    if (this::mode.isInitialized) {
 | 
			
		||||
      restart()
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    setMode(mode())
 | 
			
		||||
    state = SessionStateImpl(jumpEditors, tagger, defaultBoundary)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun startOrCycleSpecialModes() {
 | 
			
		||||
    if (!this::mode.isInitialized) {
 | 
			
		||||
      setMode(AdvancedMode())
 | 
			
		||||
      state = SessionStateImpl(jumpEditors, tagger, defaultBoundary)
 | 
			
		||||
      return
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    restart()
 | 
			
		||||
    setMode(when (mode) {
 | 
			
		||||
      is AdvancedMode -> BetweenPointsMode()
 | 
			
		||||
      else            -> AdvancedMode()
 | 
			
		||||
    })
 | 
			
		||||
    
 | 
			
		||||
    state = SessionStateImpl(jumpEditors, tagger, defaultBoundary)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Starts a regular expression search. If a search was already active, it will be reset alongside its tags and highlights.
 | 
			
		||||
   */
 | 
			
		||||
  fun startRegexSearch(pattern: String, boundaries: Boundaries) {
 | 
			
		||||
    tagger = Tagger(editor)
 | 
			
		||||
    tagCanvas.setMarkers(emptyList())
 | 
			
		||||
  fun startRegexSearch(pattern: Pattern) {
 | 
			
		||||
    if (!this::mode.isInitialized) {
 | 
			
		||||
      setMode(JumpMode())
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    tagger = Tagger(jumpEditors)
 | 
			
		||||
    tagCanvases.values.forEach { it.setMarkers(emptyList()) }
 | 
			
		||||
    
 | 
			
		||||
    val processor = SearchProcessor.fromRegex(jumpEditors, pattern.regex, defaultBoundary).also {
 | 
			
		||||
      state = SessionStateImpl(jumpEditors, tagger, defaultBoundary, it)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    val processor = SearchProcessor.fromRegex(editor, pattern, boundaries.intersection(defaultBoundaries)).also { searchProcessor = it }
 | 
			
		||||
    updateSearch(processor, markImmediately = true)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Starts a regular expression search. If a search was already active, it will be reset alongside its tags and highlights.
 | 
			
		||||
   */
 | 
			
		||||
  fun startRegexSearch(pattern: Pattern, boundaries: Boundaries) {
 | 
			
		||||
    startRegexSearch(pattern.regex, boundaries)
 | 
			
		||||
  }
 | 
			
		||||
  fun tagImmediately() {
 | 
			
		||||
    val state = state ?: return
 | 
			
		||||
    val processor = state.currentProcessor
 | 
			
		||||
    
 | 
			
		||||
  /**
 | 
			
		||||
   * See [JumpModeTracker.cycle].
 | 
			
		||||
   */
 | 
			
		||||
  fun cycleNextJumpMode() {
 | 
			
		||||
    jumpMode = jumpModeTracker.cycle(forward = true)
 | 
			
		||||
    if (processor != null) {
 | 
			
		||||
      updateSearch(processor, markImmediately = true)
 | 
			
		||||
    }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * See [JumpModeTracker.cycle].
 | 
			
		||||
   */
 | 
			
		||||
  fun cyclePreviousJumpMode() {
 | 
			
		||||
    jumpMode = jumpModeTracker.cycle(forward = false)
 | 
			
		||||
    else if (mode is AdvancedMode) {
 | 
			
		||||
      val offset = mainEditor.caretModel.offset
 | 
			
		||||
      val result = mainEditor.immutableText.getOrNull(offset)?.let(state::type)
 | 
			
		||||
      if (result is TypeResult.UpdateResults) {
 | 
			
		||||
        val tag = Tag(mainEditor, offset).also { acceptedTag = it }
 | 
			
		||||
        textHighlighter.renderFinal(tag, result.processor.query)
 | 
			
		||||
        updateHint()
 | 
			
		||||
      }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * See [JumpModeTracker.toggle]
 | 
			
		||||
   */
 | 
			
		||||
  fun toggleJumpMode(newMode: JumpMode) {
 | 
			
		||||
    jumpMode = jumpModeTracker.toggle(newMode)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * See [TagVisitor.visitPrevious]. If there are no tags, nothing happens.
 | 
			
		||||
   */
 | 
			
		||||
  fun visitPreviousTag() {
 | 
			
		||||
    if (tagVisitor?.visitPrevious() == true) {
 | 
			
		||||
      end()
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * See [TagVisitor.visitNext]. If there are no tags, nothing happens.
 | 
			
		||||
   */
 | 
			
		||||
  fun visitNextTag() {
 | 
			
		||||
    if (tagVisitor?.visitNext() == true) {
 | 
			
		||||
      end()
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
@@ -181,32 +221,41 @@ class Session(private val editor: Editor) {
 | 
			
		||||
   * Ends this session.
 | 
			
		||||
   */
 | 
			
		||||
  fun end() {
 | 
			
		||||
    SessionManager.end(editor)
 | 
			
		||||
    SessionManager.end(mainEditor)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Clears any currently active search, tags, and highlights. Does not reset [JumpMode].
 | 
			
		||||
   * Clears any currently active search, tags, and highlights.
 | 
			
		||||
   */
 | 
			
		||||
  fun restart() {
 | 
			
		||||
    tagger = Tagger(editor)
 | 
			
		||||
    searchProcessor = null
 | 
			
		||||
    tagCanvas.removeMarkers()
 | 
			
		||||
    state = null
 | 
			
		||||
    tagger = Tagger(jumpEditors)
 | 
			
		||||
    acceptedTag = null
 | 
			
		||||
    tagCanvases.values.forEach(TagCanvas::removeMarkers)
 | 
			
		||||
    textHighlighter.reset()
 | 
			
		||||
    
 | 
			
		||||
    HintManagerImpl.getInstanceImpl().hideAllHints()
 | 
			
		||||
    editorSettings.onTagUnaccepted(mainEditor)
 | 
			
		||||
    mainEditor.colorsScheme.setColor(EditorColors.CARET_COLOR, mode.caretColor)
 | 
			
		||||
    jumpEditors.forEach { it.contentComponent.repaint() }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Should only be used from [SessionManager] to dispose a successfully ended session.
 | 
			
		||||
   */
 | 
			
		||||
  internal fun dispose() {
 | 
			
		||||
    tagger = Tagger(editor)
 | 
			
		||||
    EditorKeyListener.detach(editor)
 | 
			
		||||
    tagCanvas.unbind()
 | 
			
		||||
    tagger = Tagger(jumpEditors)
 | 
			
		||||
    tagCanvases.values.forEach(TagCanvas::unbind)
 | 
			
		||||
    textHighlighter.reset()
 | 
			
		||||
    EditorKeyListener.detach(mainEditor)
 | 
			
		||||
    
 | 
			
		||||
    if (!editor.isDisposed) {
 | 
			
		||||
      originalSettings.restore(editor)
 | 
			
		||||
      editor.colorsScheme.setColor(EditorColors.CARET_COLOR, JumpMode.DISABLED.caretColor)
 | 
			
		||||
      editor.scrollingModel.scrollToCaret(ScrollType.MAKE_VISIBLE)
 | 
			
		||||
    if (!mainEditor.isDisposed) {
 | 
			
		||||
      HintManagerImpl.getInstanceImpl().hideAllHints()
 | 
			
		||||
      editorSettings.restore(mainEditor)
 | 
			
		||||
      mainEditor.colorsScheme.setColor(EditorColors.CARET_COLOR, AbstractColorsScheme.INHERITED_COLOR_MARKER)
 | 
			
		||||
      
 | 
			
		||||
      val focusedEditor = acceptedTag?.editor ?: mainEditor
 | 
			
		||||
      focusedEditor.scrollingModel.scrollToCaret(ScrollType.MAKE_VISIBLE)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -16,7 +16,16 @@ object SessionManager {
 | 
			
		||||
   * Starts a new [Session], or returns an existing [Session] if the specified [Editor] already has one.
 | 
			
		||||
   */
 | 
			
		||||
  fun start(editor: Editor): Session {
 | 
			
		||||
    return sessions.getOrPut(editor) { cleanup(); Session(editor) }
 | 
			
		||||
    return start(editor, listOf(editor))
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Starts a new multi-editor [Session], or returns an existing [Session] if the specified main [Editor] already has one.
 | 
			
		||||
   * The [mainEditor] is used for typing the search query and tag.
 | 
			
		||||
   * The [jumpEditors] are all editors that will be searched and tagged.
 | 
			
		||||
   */
 | 
			
		||||
  fun start(mainEditor: Editor, jumpEditors: List<Editor>): Session {
 | 
			
		||||
    return sessions.getOrPut(mainEditor) { cleanup(); Session(mainEditor, jumpEditors) }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										9
									
								
								src/main/kotlin/org/acejump/session/SessionState.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								src/main/kotlin/org/acejump/session/SessionState.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,9 @@
 | 
			
		||||
package org.acejump.session
 | 
			
		||||
 | 
			
		||||
import org.acejump.action.AceTagAction
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
 | 
			
		||||
interface SessionState {
 | 
			
		||||
  fun type(char: Char): TypeResult
 | 
			
		||||
  fun act(action: AceTagAction, tag: Tag, shiftMode: Boolean, isFinal: Boolean)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										36
									
								
								src/main/kotlin/org/acejump/session/SessionStateImpl.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										36
									
								
								src/main/kotlin/org/acejump/session/SessionStateImpl.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,36 @@
 | 
			
		||||
package org.acejump.session
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import org.acejump.action.AceTagAction
 | 
			
		||||
import org.acejump.boundaries.Boundaries
 | 
			
		||||
import org.acejump.search.SearchProcessor
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import org.acejump.search.Tagger
 | 
			
		||||
 | 
			
		||||
internal class SessionStateImpl(
 | 
			
		||||
  private val jumpEditors: List<Editor>,
 | 
			
		||||
  private val tagger: Tagger,
 | 
			
		||||
  private val defaultBoundary: Boundaries,
 | 
			
		||||
  processor: SearchProcessor? = null
 | 
			
		||||
) : SessionState {
 | 
			
		||||
  internal var currentProcessor: SearchProcessor? = processor
 | 
			
		||||
  
 | 
			
		||||
  override fun type(char: Char): TypeResult {
 | 
			
		||||
    val processor = currentProcessor
 | 
			
		||||
    
 | 
			
		||||
    if (processor == null) {
 | 
			
		||||
      val newProcessor = SearchProcessor.fromChar(jumpEditors, char, defaultBoundary)
 | 
			
		||||
      return TypeResult.UpdateResults(newProcessor.also { currentProcessor = it })
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    if (processor.type(char, tagger)) {
 | 
			
		||||
      return TypeResult.UpdateResults(processor)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return TypeResult.Nothing
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  override fun act(action: AceTagAction, tag: Tag, shiftMode: Boolean, isFinal: Boolean) {
 | 
			
		||||
    currentProcessor?.let { action(tag.editor, it, tag.offset, shiftMode, isFinal) }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										12
									
								
								src/main/kotlin/org/acejump/session/TypeResult.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								src/main/kotlin/org/acejump/session/TypeResult.kt
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
			
		||||
package org.acejump.session
 | 
			
		||||
 | 
			
		||||
import org.acejump.modes.SessionMode
 | 
			
		||||
import org.acejump.search.SearchProcessor
 | 
			
		||||
 | 
			
		||||
sealed class TypeResult {
 | 
			
		||||
  object Nothing : TypeResult()
 | 
			
		||||
  class UpdateResults(val processor: SearchProcessor) : TypeResult()
 | 
			
		||||
  class ChangeMode(val mode: SessionMode) : TypeResult()
 | 
			
		||||
  object RestartSearch : TypeResult()
 | 
			
		||||
  object EndSession : TypeResult()
 | 
			
		||||
}
 | 
			
		||||
@@ -17,7 +17,7 @@ import javax.swing.SwingUtilities
 | 
			
		||||
 * Holds all active tag markers and renders them on top of the editor.
 | 
			
		||||
 */
 | 
			
		||||
internal class TagCanvas(private val editor: Editor) : JComponent(), CaretListener {
 | 
			
		||||
  private var markers: List<Tag>? = null
 | 
			
		||||
  private var markers: Collection<TagMarker>? = null
 | 
			
		||||
  
 | 
			
		||||
  init {
 | 
			
		||||
    val contentComponent = editor.contentComponent
 | 
			
		||||
@@ -45,7 +45,7 @@ internal class TagCanvas(private val editor: Editor) : JComponent(), CaretListen
 | 
			
		||||
    repaint()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun setMarkers(markers: List<Tag>) {
 | 
			
		||||
  fun setMarkers(markers: Collection<TagMarker>) {
 | 
			
		||||
    this.markers = markers
 | 
			
		||||
    repaint()
 | 
			
		||||
  }
 | 
			
		||||
@@ -64,14 +64,15 @@ internal class TagCanvas(private val editor: Editor) : JComponent(), CaretListen
 | 
			
		||||
    super.paintChildren(g)
 | 
			
		||||
    
 | 
			
		||||
    val markers = markers ?: return
 | 
			
		||||
    
 | 
			
		||||
    (g as Graphics2D).setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON)
 | 
			
		||||
    
 | 
			
		||||
    val font = TagFont(editor)
 | 
			
		||||
    
 | 
			
		||||
    val cache = EditorOffsetCache.new()
 | 
			
		||||
    val viewRange = StandardBoundaries.VISIBLE_ON_SCREEN.getOffsetRange(editor, cache)
 | 
			
		||||
    val occupied = mutableListOf<Rectangle>()
 | 
			
		||||
    
 | 
			
		||||
    (g as Graphics2D).setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON)
 | 
			
		||||
    
 | 
			
		||||
    // If there is a tag at the caret location, prioritize its rendering over all other tags. This is helpful for seeing which tag is
 | 
			
		||||
    // currently selected while navigating highly clustered tags, although it does end up rearranging nearby tags which can be confusing.
 | 
			
		||||
    
 | 
			
		||||
 
 | 
			
		||||
@@ -2,16 +2,22 @@ package org.acejump.view
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.openapi.editor.colors.EditorFontType
 | 
			
		||||
import com.intellij.ui.ColorUtil
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import java.awt.Font
 | 
			
		||||
import java.awt.FontMetrics
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Stores font metrics for aligning and rendering [Tag]s.
 | 
			
		||||
 * Stores font metrics for aligning and rendering [TagMarker]s.
 | 
			
		||||
 */
 | 
			
		||||
internal class TagFont(editor: Editor) {
 | 
			
		||||
  val tagFont: Font = editor.colorsScheme.getFont(EditorFontType.BOLD)
 | 
			
		||||
  val tagCharWidth = editor.component.getFontMetrics(tagFont).charWidth('W')
 | 
			
		||||
  
 | 
			
		||||
  val foregroundColor = AceConfig.tagForegroundColor
 | 
			
		||||
  var backgroundColor = AceConfig.tagBackgroundColor
 | 
			
		||||
  val isForegroundDark = ColorUtil.isDark(foregroundColor)
 | 
			
		||||
  
 | 
			
		||||
  val editorFontMetrics: FontMetrics = editor.component.getFontMetrics(editor.colorsScheme.getFont(EditorFontType.PLAIN))
 | 
			
		||||
  val lineHeight = editor.lineHeight
 | 
			
		||||
  val baselineDistance = editor.ascent
 | 
			
		||||
 
 | 
			
		||||
@@ -1,11 +1,10 @@
 | 
			
		||||
package org.acejump.view
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.editor.Editor
 | 
			
		||||
import com.intellij.ui.ColorUtil
 | 
			
		||||
import com.intellij.openapi.util.SystemInfo
 | 
			
		||||
import com.intellij.ui.scale.JBUIScale
 | 
			
		||||
import org.acejump.boundaries.EditorOffsetCache
 | 
			
		||||
import org.acejump.boundaries.StandardBoundaries
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import org.acejump.countMatchingCharacters
 | 
			
		||||
import org.acejump.immutableText
 | 
			
		||||
import java.awt.Color
 | 
			
		||||
@@ -17,7 +16,7 @@ import kotlin.math.max
 | 
			
		||||
/**
 | 
			
		||||
 * Describes a 1 or 2 character shortcut that points to a specific character in the editor.
 | 
			
		||||
 */
 | 
			
		||||
internal class Tag(
 | 
			
		||||
internal class TagMarker(
 | 
			
		||||
  private val tag: String,
 | 
			
		||||
  val offsetL: Int,
 | 
			
		||||
  val offsetR: Int,
 | 
			
		||||
@@ -29,11 +28,18 @@ internal class Tag(
 | 
			
		||||
  companion object {
 | 
			
		||||
    private const val ARC = 1
 | 
			
		||||
    
 | 
			
		||||
    /**
 | 
			
		||||
     * TODO This might be due to DPI settings.
 | 
			
		||||
     */
 | 
			
		||||
    private val HIGHLIGHT_OFFSET = if (SystemInfo.isMac) -0.5 else 0.0
 | 
			
		||||
    
 | 
			
		||||
    private val SHADOW_COLOR = Color(0F, 0F, 0F, 0.35F)
 | 
			
		||||
    
 | 
			
		||||
    /**
 | 
			
		||||
     * Creates a new tag, precomputing some information about the nearby characters to reduce rendering overhead. If the last typed
 | 
			
		||||
     * character ([literalQueryText]) matches the first [tag] character, only the second [tag] character is displayed.
 | 
			
		||||
     */
 | 
			
		||||
    fun create(editor: Editor, tag: String, offset: Int, literalQueryText: String?): Tag {
 | 
			
		||||
    fun create(editor: Editor, tag: String, offset: Int, literalQueryText: String?): TagMarker {
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      val matching = literalQueryText?.let { chars.countMatchingCharacters(offset, it) } ?: 0
 | 
			
		||||
      val hasSpaceRight = offset + 1 >= chars.length || chars[offset + 1].isWhitespace()
 | 
			
		||||
@@ -43,7 +49,7 @@ internal class Tag(
 | 
			
		||||
      else
 | 
			
		||||
        tag.toUpperCase()
 | 
			
		||||
      
 | 
			
		||||
      return Tag(displayedTag, offset, offset + max(0, matching - 1), tag.length - displayedTag.length, hasSpaceRight)
 | 
			
		||||
      return TagMarker(displayedTag, offset, offset + max(0, matching - 1), tag.length - displayedTag.length, hasSpaceRight)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    /**
 | 
			
		||||
@@ -51,7 +57,9 @@ internal class Tag(
 | 
			
		||||
     */
 | 
			
		||||
    private fun drawHighlight(g: Graphics2D, rect: Rectangle, color: Color) {
 | 
			
		||||
      g.color = color
 | 
			
		||||
      g.translate(0.0, HIGHLIGHT_OFFSET)
 | 
			
		||||
      g.fillRoundRect(rect.x, rect.y, rect.width, rect.height + 1, ARC, ARC)
 | 
			
		||||
      g.translate(0.0, -HIGHLIGHT_OFFSET)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    /**
 | 
			
		||||
@@ -63,12 +71,12 @@ internal class Tag(
 | 
			
		||||
      
 | 
			
		||||
      g.font = font.tagFont
 | 
			
		||||
      
 | 
			
		||||
      if (!ColorUtil.isDark(AceConfig.tagForegroundColor)) {
 | 
			
		||||
        g.color = Color(0F, 0F, 0F, 0.35F)
 | 
			
		||||
      if (!font.isForegroundDark) {
 | 
			
		||||
        g.color = SHADOW_COLOR
 | 
			
		||||
        g.drawString(text, x + 1, y + 1)
 | 
			
		||||
      }
 | 
			
		||||
      
 | 
			
		||||
      g.color = AceConfig.tagForegroundColor
 | 
			
		||||
      g.color = font.foregroundColor
 | 
			
		||||
      g.drawString(text, x, y)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
@@ -88,7 +96,7 @@ internal class Tag(
 | 
			
		||||
  fun paint(g: Graphics2D, editor: Editor, cache: EditorOffsetCache, font: TagFont, occupied: MutableList<Rectangle>): Rectangle? {
 | 
			
		||||
    val rect = alignTag(editor, cache, font, occupied) ?: return null
 | 
			
		||||
    
 | 
			
		||||
    drawHighlight(g, rect, AceConfig.tagBackgroundColor)
 | 
			
		||||
    drawHighlight(g, rect, font.backgroundColor)
 | 
			
		||||
    drawForeground(g, font, rect.location, tag)
 | 
			
		||||
    
 | 
			
		||||
    occupied.add(JBUIScale.scale(2).let { Rectangle(rect.x - it, rect.y, rect.width + (2 * it), rect.height) })
 | 
			
		||||
@@ -6,56 +6,59 @@ import com.intellij.openapi.editor.markup.CustomHighlighterRenderer
 | 
			
		||||
import com.intellij.openapi.editor.markup.HighlighterLayer
 | 
			
		||||
import com.intellij.openapi.editor.markup.HighlighterTargetArea
 | 
			
		||||
import com.intellij.openapi.editor.markup.RangeHighlighter
 | 
			
		||||
import com.intellij.util.ui.JBUI
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.IntArrayList
 | 
			
		||||
import it.unimi.dsi.fastutil.ints.IntList
 | 
			
		||||
import org.acejump.boundaries.EditorOffsetCache
 | 
			
		||||
import org.acejump.config.AceConfig
 | 
			
		||||
import org.acejump.immutableText
 | 
			
		||||
import org.acejump.input.JumpMode
 | 
			
		||||
import org.acejump.isWordPart
 | 
			
		||||
import org.acejump.search.SearchQuery
 | 
			
		||||
import org.acejump.wordEnd
 | 
			
		||||
import org.acejump.wordStart
 | 
			
		||||
import org.acejump.search.Tag
 | 
			
		||||
import java.awt.Color
 | 
			
		||||
import java.awt.Graphics
 | 
			
		||||
import kotlin.math.max
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Renders highlights for search occurrences.
 | 
			
		||||
 */
 | 
			
		||||
internal class TextHighlighter(private val editor: Editor) {
 | 
			
		||||
  private companion object {
 | 
			
		||||
    private const val LAYER = HighlighterLayer.LAST + 1
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private var previousHighlights: Array<RangeHighlighter>? = null
 | 
			
		||||
internal class TextHighlighter {
 | 
			
		||||
  private var previousHighlights = mutableMapOf<Editor, Array<RangeHighlighter>>()
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Removes all current highlights and re-creates them from scratch. Must be called whenever any of the method parameters change.
 | 
			
		||||
   */
 | 
			
		||||
  fun render(offsets: IntList, query: SearchQuery, jumpMode: JumpMode) {
 | 
			
		||||
    val markup = editor.markupModel
 | 
			
		||||
    val chars = editor.immutableText
 | 
			
		||||
    
 | 
			
		||||
    val renderer = when {
 | 
			
		||||
      query is SearchQuery.RegularExpression -> RegexRenderer
 | 
			
		||||
      jumpMode === JumpMode.TARGET           -> SearchedWordWithOutlineRenderer
 | 
			
		||||
  fun renderOccurrences(results: Map<Editor, IntList>, query: SearchQuery) {
 | 
			
		||||
    render(results, when (query) {
 | 
			
		||||
      is SearchQuery.RegularExpression -> RegexRenderer
 | 
			
		||||
      else                             -> SearchedWordRenderer
 | 
			
		||||
    }, query::getHighlightLength)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
    val modifications = (previousHighlights?.size ?: 0) + offsets.size
 | 
			
		||||
    val enableBulkEditing = modifications > 1000
 | 
			
		||||
  /**
 | 
			
		||||
   * Removes all current highlights and re-adds a single highlight at the position of the accepted tag with a different color.
 | 
			
		||||
   */
 | 
			
		||||
  fun renderFinal(tag: Tag, query: SearchQuery) {
 | 
			
		||||
    render(mutableMapOf(tag.editor to IntArrayList(intArrayOf(tag.offset))), AcceptedTagRenderer, query::getHighlightLength)
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private inline fun render(results: Map<Editor, IntList>, renderer: CustomHighlighterRenderer, getHighlightLength: (CharSequence, Int) -> Int) {
 | 
			
		||||
    for ((editor, offsets) in results) {
 | 
			
		||||
      val highlights = previousHighlights[editor]
 | 
			
		||||
      
 | 
			
		||||
      val markup = editor.markupModel
 | 
			
		||||
      val document = editor.document
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      
 | 
			
		||||
      val modifications = (highlights?.size ?: 0) + offsets.size
 | 
			
		||||
      val enableBulkEditing = modifications > 1000
 | 
			
		||||
      
 | 
			
		||||
      try {
 | 
			
		||||
        if (enableBulkEditing) {
 | 
			
		||||
          document.isInBulkUpdate = true
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
      previousHighlights?.forEach(markup::removeHighlighter)
 | 
			
		||||
      previousHighlights = Array(offsets.size) { index ->
 | 
			
		||||
        highlights?.forEach(markup::removeHighlighter)
 | 
			
		||||
        previousHighlights[editor] = Array(offsets.size) { index ->
 | 
			
		||||
          val start = offsets.getInt(index)
 | 
			
		||||
        val end = start + query.getHighlightLength(chars, start)
 | 
			
		||||
          val end = start + getHighlightLength(chars, start)
 | 
			
		||||
          
 | 
			
		||||
          markup.addRangeHighlighter(start, end, LAYER, null, HighlighterTargetArea.EXACT_RANGE).apply {
 | 
			
		||||
            customRenderer = renderer
 | 
			
		||||
@@ -68,9 +71,16 @@ internal class TextHighlighter(private val editor: Editor) {
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    for (editor in previousHighlights.keys.toList()) {
 | 
			
		||||
      if (!results.containsKey(editor)) {
 | 
			
		||||
        previousHighlights.remove(editor)?.forEach(editor.markupModel::removeHighlighter)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun reset() {
 | 
			
		||||
    editor.markupModel.removeAllHighlighters()
 | 
			
		||||
    previousHighlights = null
 | 
			
		||||
    previousHighlights.keys.forEach { it.markupModel.removeAllHighlighters() }
 | 
			
		||||
    previousHighlights.clear()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
@@ -78,43 +88,7 @@ internal class TextHighlighter(private val editor: Editor) {
 | 
			
		||||
   */
 | 
			
		||||
  private object SearchedWordRenderer : CustomHighlighterRenderer {
 | 
			
		||||
    override fun paint(editor: Editor, highlighter: RangeHighlighter, g: Graphics) {
 | 
			
		||||
      drawFilled(g, editor, highlighter.startOffset, highlighter.endOffset)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun drawFilled(g: Graphics, editor: Editor, startOffset: Int, endOffset: Int) {
 | 
			
		||||
      val start = EditorOffsetCache.Uncached.offsetToXY(editor, startOffset)
 | 
			
		||||
      val end = EditorOffsetCache.Uncached.offsetToXY(editor, endOffset)
 | 
			
		||||
      
 | 
			
		||||
      g.color = AceConfig.textHighlightColor
 | 
			
		||||
      g.fillRect(start.x, start.y + 1, end.x - start.x, editor.lineHeight - 1)
 | 
			
		||||
  
 | 
			
		||||
      g.color = AceConfig.tagBackgroundColor
 | 
			
		||||
      g.drawRect(start.x, start.y, end.x - start.x, editor.lineHeight)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  /**
 | 
			
		||||
   * Renders a filled highlight in the background of a searched text occurrence, as well as an outline indicating the range of characters
 | 
			
		||||
   * that will be selected by [JumpMode.TARGET].
 | 
			
		||||
   */
 | 
			
		||||
  private object SearchedWordWithOutlineRenderer : CustomHighlighterRenderer {
 | 
			
		||||
    override fun paint(editor: Editor, highlighter: RangeHighlighter, g: Graphics) {
 | 
			
		||||
      SearchedWordRenderer.paint(editor, highlighter, g)
 | 
			
		||||
      
 | 
			
		||||
      val chars = editor.immutableText
 | 
			
		||||
      val startOffset = highlighter.startOffset
 | 
			
		||||
      
 | 
			
		||||
      if (chars.getOrNull(startOffset)?.isWordPart == true) {
 | 
			
		||||
        drawOutline(g, editor, chars.wordStart(startOffset), chars.wordEnd(startOffset) + 1)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun drawOutline(g: Graphics, editor: Editor, startOffset: Int, endOffset: Int) {
 | 
			
		||||
      val start = EditorOffsetCache.Uncached.offsetToXY(editor, startOffset)
 | 
			
		||||
      val end = EditorOffsetCache.Uncached.offsetToXY(editor, endOffset)
 | 
			
		||||
      
 | 
			
		||||
      g.color = AceConfig.targetModeColor
 | 
			
		||||
      g.drawRect(max(0, start.x - JBUI.scale(1)), start.y, end.x - start.x + JBUI.scale(2), editor.lineHeight)
 | 
			
		||||
      drawFilled(g, editor, highlighter.startOffset, highlighter.endOffset, AceConfig.textHighlightColor)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
@@ -123,16 +97,40 @@ internal class TextHighlighter(private val editor: Editor) {
 | 
			
		||||
   */
 | 
			
		||||
  private object RegexRenderer : CustomHighlighterRenderer {
 | 
			
		||||
    override fun paint(editor: Editor, highlighter: RangeHighlighter, g: Graphics) {
 | 
			
		||||
      drawSingle(g, editor, highlighter.startOffset)
 | 
			
		||||
      drawSingle(g, editor, highlighter.startOffset, AceConfig.textHighlightColor)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
    private fun drawSingle(g: Graphics, editor: Editor, offset: Int) {
 | 
			
		||||
  /**
 | 
			
		||||
   * Renders a filled highlight in the background of the accepted tag position and search query.
 | 
			
		||||
   */
 | 
			
		||||
  private object AcceptedTagRenderer : CustomHighlighterRenderer {
 | 
			
		||||
    override fun paint(editor: Editor, highlighter: RangeHighlighter, g: Graphics) {
 | 
			
		||||
      drawFilled(g, editor, highlighter.startOffset, highlighter.endOffset, AceConfig.acceptedTagColor)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  private companion object {
 | 
			
		||||
    private const val LAYER = HighlighterLayer.LAST + 1
 | 
			
		||||
    
 | 
			
		||||
    private fun drawFilled(g: Graphics, editor: Editor, startOffset: Int, endOffset: Int, color: Color) {
 | 
			
		||||
      val start = EditorOffsetCache.Uncached.offsetToXY(editor, startOffset)
 | 
			
		||||
      val end = EditorOffsetCache.Uncached.offsetToXY(editor, endOffset)
 | 
			
		||||
      
 | 
			
		||||
      g.color = color
 | 
			
		||||
      g.fillRect(start.x, start.y + 1, end.x - start.x, editor.lineHeight - 1)
 | 
			
		||||
      
 | 
			
		||||
      g.color = AceConfig.tagBackgroundColor
 | 
			
		||||
      g.drawRect(start.x, start.y, end.x - start.x, editor.lineHeight)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    private fun drawSingle(g: Graphics, editor: Editor, offset: Int, color: Color) {
 | 
			
		||||
      val pos = EditorOffsetCache.Uncached.offsetToXY(editor, offset)
 | 
			
		||||
      val char = editor.immutableText.getOrNull(offset)?.takeUnless { it == '\n' || it == '\t' } ?: ' '
 | 
			
		||||
      val font = editor.colorsScheme.getFont(EditorFontType.PLAIN)
 | 
			
		||||
      val lastCharWidth = editor.component.getFontMetrics(font).charWidth(char)
 | 
			
		||||
      
 | 
			
		||||
      g.color = AceConfig.textHighlightColor
 | 
			
		||||
      g.color = color
 | 
			
		||||
      g.fillRect(pos.x, pos.y + 1, lastCharWidth, editor.lineHeight - 1)
 | 
			
		||||
      
 | 
			
		||||
      g.color = AceConfig.tagBackgroundColor
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
<idea-plugin url="https://github.com/acejump/AceJump">
 | 
			
		||||
<idea-plugin>
 | 
			
		||||
  <name>AceJump</name>
 | 
			
		||||
  <id>AceJump</id>
 | 
			
		||||
  <id>AceJump-chylex</id>
 | 
			
		||||
 | 
			
		||||
  <description><![CDATA[
 | 
			
		||||
    AceJump allows you to quickly navigate the caret to any position visible in the editor.
 | 
			
		||||
@@ -23,10 +23,8 @@
 | 
			
		||||
                         implementationClass="org.acejump.action.AceEditorAction$Reset"/>
 | 
			
		||||
    <editorActionHandler action="EditorBackSpace" order="first"
 | 
			
		||||
                         implementationClass="org.acejump.action.AceEditorAction$ClearSearch"/>
 | 
			
		||||
    <editorActionHandler action="EditorStartNewLine" order="first"
 | 
			
		||||
                         implementationClass="org.acejump.action.AceEditorAction$SelectBackward"/>
 | 
			
		||||
    <editorActionHandler action="EditorEnter" order="first"
 | 
			
		||||
                         implementationClass="org.acejump.action.AceEditorAction$SelectForward"/>
 | 
			
		||||
                         implementationClass="org.acejump.action.AceEditorAction$TagImmediately"/>
 | 
			
		||||
    <editorActionHandler action="EditorUp" order="first"
 | 
			
		||||
                         implementationClass="org.acejump.action.AceEditorAction$SearchLineStarts"/>
 | 
			
		||||
    <editorActionHandler action="EditorLeft" order="first"
 | 
			
		||||
@@ -41,55 +39,97 @@
 | 
			
		||||
 | 
			
		||||
  <actions>
 | 
			
		||||
    <action id="AceAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$ActivateOrCycleMode"
 | 
			
		||||
            text="Activate / Cycle AceJump Mode">
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$ActivateAceJump"
 | 
			
		||||
            text="Activate AceJump">
 | 
			
		||||
      <keyboard-shortcut keymap="Mac OS X" first-keystroke="ctrl SEMICOLON"/>
 | 
			
		||||
      <keyboard-shortcut keymap="Mac OS X 10.5+" first-keystroke="ctrl SEMICOLON"/>
 | 
			
		||||
      <keyboard-shortcut keymap="$default" first-keystroke="ctrl SEMICOLON"/>
 | 
			
		||||
    </action>
 | 
			
		||||
    <action id="AceReverseAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$ActivateOrReverseCycleMode"
 | 
			
		||||
            text="Activate / Reverse Cycle AceJump Mode"/>
 | 
			
		||||
    <action id="AceWordStartAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$ToggleJumpMode"
 | 
			
		||||
            text="Start AceJump in Jump Mode"/>
 | 
			
		||||
    <action id="AceWordEndAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$ToggleJumpEndMode"
 | 
			
		||||
            text="Start AceJump in Jump End Mode"/>
 | 
			
		||||
    <action id="AceTargetAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$ToggleTargetMode"
 | 
			
		||||
            text="Start AceJump in Target Mode">
 | 
			
		||||
    <action id="AceSpecialAction"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$ActivateAceJumpSpecial"
 | 
			
		||||
            text="Activate / Cycle AceJump Special Modes">
 | 
			
		||||
      <keyboard-shortcut keymap="Mac OS X" first-keystroke="ctrl alt SEMICOLON"/>
 | 
			
		||||
      <keyboard-shortcut keymap="Mac OS X 10.5+" first-keystroke="ctrl alt SEMICOLON"/>
 | 
			
		||||
      <keyboard-shortcut keymap="$default" first-keystroke="ctrl alt SEMICOLON"/>
 | 
			
		||||
    </action>
 | 
			
		||||
    <action id="AceDeclarationAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$ToggleDeclarationMode"
 | 
			
		||||
            text="Start AceJump in Declaration Mode"/>
 | 
			
		||||
    <action id="AceLineAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$StartAllLineMarksMode"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$StartAllLineMarksMode"
 | 
			
		||||
            text="Start AceJump in All Line Marks Mode">
 | 
			
		||||
      <keyboard-shortcut keymap="Mac OS X" first-keystroke="ctrl shift SEMICOLON"/>
 | 
			
		||||
      <keyboard-shortcut keymap="Mac OS X 10.5+" first-keystroke="ctrl shift SEMICOLON"/>
 | 
			
		||||
      <keyboard-shortcut keymap="$default" first-keystroke="ctrl shift SEMICOLON"/>
 | 
			
		||||
    </action>
 | 
			
		||||
    <action id="AceLineStartsAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$StartAllLineStartsMode"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$StartAllLineStartsMode"
 | 
			
		||||
            text="Start AceJump in All Line Starts Mode"/>
 | 
			
		||||
    <action id="AceLineEndsAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$StartAllLineEndsMode"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$StartAllLineEndsMode"
 | 
			
		||||
            text="Start AceJump in All Line Ends Mode"/>
 | 
			
		||||
    <action id="AceLineIndentsAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$StartAllLineIndentsMode"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$StartAllLineIndentsMode"
 | 
			
		||||
            text="Start AceJump in All Line Indents Mode"/>
 | 
			
		||||
    <action id="AceWordAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$StartAllWordsMode"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$StartAllWordsMode"
 | 
			
		||||
            text="Start AceJump in All Words Mode"/>
 | 
			
		||||
    <action id="AceWordForwardAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$StartAllWordsForwardMode"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$StartAllWordsForwardMode"
 | 
			
		||||
            text="Start AceJump in All Words After Caret Mode"/>
 | 
			
		||||
    <action id="AceWordBackwardsAction"
 | 
			
		||||
            class="org.acejump.action.AceAction$StartAllWordsBackwardsMode"
 | 
			
		||||
            class="org.acejump.action.AceKeyboardAction$StartAllWordsBackwardsMode"
 | 
			
		||||
            text="Start AceJump in All Words Before Caret Mode"/>
 | 
			
		||||
    <action id="AceVimAction_JumpToChar"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$JumpToChar"
 | 
			
		||||
            text="AceJump Vim - JumpToChar"/>
 | 
			
		||||
    <action id="AceVimAction_JumpToCharBeforeCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$JumpToCharBeforeCaret"
 | 
			
		||||
            text="AceJump Vim - JumpToCharBeforeCaret"/>
 | 
			
		||||
    <action id="AceVimAction_JumpToCharAfterCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$JumpToCharAfterCaret"
 | 
			
		||||
            text="AceJump Vim - JumpToCharAfterCaret"/>
 | 
			
		||||
    <action id="AceVimAction_LWordsAfterCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$LWordsAfterCaret"
 | 
			
		||||
            text="AceJump Vim - LWordsAfterCaret"/>
 | 
			
		||||
    <action id="AceVimAction_UWordsAfterCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$UWordsAfterCaret"
 | 
			
		||||
            text="AceJump Vim - UWordsAfterCaret"/>
 | 
			
		||||
    <action id="AceVimAction_LWordsBeforeCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$LWordsBeforeCaret"
 | 
			
		||||
            text="AceJump Vim - LWordsBeforeCaret"/>
 | 
			
		||||
    <action id="AceVimAction_UWordsBeforeCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$UWordsBeforeCaret"
 | 
			
		||||
            text="AceJump Vim - UWordsBeforeCaret"/>
 | 
			
		||||
    <action id="AceVimAction_LWordEndsAfterCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$LWordEndsAfterCaret"
 | 
			
		||||
            text="AceJump Vim - LWordEndsAfterCaret"/>
 | 
			
		||||
    <action id="AceVimAction_UWordEndsAfterCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$UWordEndsAfterCaret"
 | 
			
		||||
            text="AceJump Vim - UWordEndsAfterCaret"/>
 | 
			
		||||
    <action id="AceVimAction_LWordEndsBeforeCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$LWordEndsBeforeCaret"
 | 
			
		||||
            text="AceJump Vim - LWordEndsBeforeCaret"/>
 | 
			
		||||
    <action id="AceVimAction_UWordEndsBeforeCaret"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$UWordEndsBeforeCaret"
 | 
			
		||||
            text="AceJump Vim - UWordEndsBeforeCaret"/>
 | 
			
		||||
    <action id="AceVimAction_GoToDeclaration"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$GoToDeclaration"
 | 
			
		||||
            text="AceJump Vim - GoToDeclaration"/>
 | 
			
		||||
    <action id="AceVimAction_GoToTypeDeclaration"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$GoToTypeDeclaration"
 | 
			
		||||
            text="AceJump Vim - GoToTypeDeclaration"/>
 | 
			
		||||
    <action id="AceVimAction_ShowIntentions"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$ShowIntentions"
 | 
			
		||||
            text="AceJump Vim - ShowIntentions"/>
 | 
			
		||||
    <action id="AceVimAction_ShowUsages"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$ShowUsages"
 | 
			
		||||
            text="AceJump Vim - ShowUsages"/>
 | 
			
		||||
    <action id="AceVimAction_FindUsages"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$FindUsages"
 | 
			
		||||
            text="AceJump Vim - FindUsages"/>
 | 
			
		||||
    <action id="AceVimAction_Refactor"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$Refactor"
 | 
			
		||||
            text="AceJump Vim - Refactor"/>
 | 
			
		||||
    <action id="AceVimAction_Rename"
 | 
			
		||||
            class="org.acejump.action.AceVimAction$Rename"
 | 
			
		||||
            text="AceJump Vim - Rename"/>
 | 
			
		||||
  </actions>
 | 
			
		||||
</idea-plugin>
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,5 @@
 | 
			
		||||
import com.intellij.openapi.actionSystem.IdeActions.ACTION_EDITOR_ENTER
 | 
			
		||||
import com.intellij.openapi.actionSystem.IdeActions.ACTION_EDITOR_START_NEW_LINE
 | 
			
		||||
import org.acejump.action.AceAction
 | 
			
		||||
 | 
			
		||||
import org.acejump.action.AceKeyboardAction
 | 
			
		||||
import org.acejump.test.util.BaseTest
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
@@ -28,30 +27,6 @@ class AceTest : BaseTest() {
 | 
			
		||||
  fun `test a query containing a { character`() =
 | 
			
		||||
    assertEquals("abcd{dabc cdab".search("cd{"), setOf(2))
 | 
			
		||||
 | 
			
		||||
  fun `test that jumping to first occurrence succeeds`() {
 | 
			
		||||
    "<caret>testing 1234".search("1")
 | 
			
		||||
 | 
			
		||||
    takeAction(ACTION_EDITOR_ENTER)
 | 
			
		||||
 | 
			
		||||
    myFixture.checkResult("testing <caret>1234")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fun `test that jumping to second occurrence succeeds`() {
 | 
			
		||||
    "<caret>testing 1234".search("ti")
 | 
			
		||||
 | 
			
		||||
    takeAction(ACTION_EDITOR_ENTER)
 | 
			
		||||
 | 
			
		||||
    myFixture.checkResult("tes<caret>ting 1234")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fun `test that jumping to previous occurrence succeeds`() {
 | 
			
		||||
    "te<caret>sting 1234".search("t")
 | 
			
		||||
 | 
			
		||||
    takeAction(ACTION_EDITOR_START_NEW_LINE)
 | 
			
		||||
 | 
			
		||||
    myFixture.checkResult("<caret>testing 1234")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fun `test tag selection`() {
 | 
			
		||||
    "<caret>testing 1234".search("g")
 | 
			
		||||
 | 
			
		||||
@@ -60,18 +35,10 @@ class AceTest : BaseTest() {
 | 
			
		||||
    myFixture.checkResult("testin<caret>g 1234")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fun `test shift selection`() {
 | 
			
		||||
    "<caret>testing 1234".search("4")
 | 
			
		||||
 | 
			
		||||
    typeAndWaitForResults(session.tags[0].key.toUpperCase())
 | 
			
		||||
 | 
			
		||||
    myFixture.checkResult("<selection>testing 123<caret></selection>4")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fun `test words before caret action`() {
 | 
			
		||||
    makeEditor("test words <caret> before caret is two")
 | 
			
		||||
 | 
			
		||||
    takeAction(AceAction.StartAllWordsBackwardsMode)
 | 
			
		||||
    takeAction(AceKeyboardAction.StartAllWordsBackwardsMode)
 | 
			
		||||
 | 
			
		||||
    assertEquals(2, session.tags.size)
 | 
			
		||||
  }
 | 
			
		||||
@@ -79,7 +46,7 @@ class AceTest : BaseTest() {
 | 
			
		||||
  fun `test words after caret action`() {
 | 
			
		||||
    makeEditor("test words <caret> after caret is four")
 | 
			
		||||
 | 
			
		||||
    takeAction(AceAction.StartAllWordsForwardMode)
 | 
			
		||||
    takeAction(AceKeyboardAction.StartAllWordsForwardMode)
 | 
			
		||||
 | 
			
		||||
    assertEquals(4, session.tags.size)
 | 
			
		||||
  }
 | 
			
		||||
@@ -87,7 +54,7 @@ class AceTest : BaseTest() {
 | 
			
		||||
  fun `test word mode`() {
 | 
			
		||||
    makeEditor("test word action")
 | 
			
		||||
 | 
			
		||||
    takeAction(AceAction.StartAllWordsMode)
 | 
			
		||||
    takeAction(AceKeyboardAction.StartAllWordsMode)
 | 
			
		||||
 | 
			
		||||
    assertEquals(3, session.tags.size)
 | 
			
		||||
 | 
			
		||||
@@ -96,20 +63,11 @@ class AceTest : BaseTest() {
 | 
			
		||||
    myFixture.checkResult("test <caret>word action")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fun `test target mode`() {
 | 
			
		||||
    "<caret>test target action".search("target")
 | 
			
		||||
 | 
			
		||||
    takeAction(AceAction.ToggleTargetMode)
 | 
			
		||||
    typeAndWaitForResults(session.tags[0].key)
 | 
			
		||||
 | 
			
		||||
    myFixture.checkResult("test <selection>target<caret></selection> action")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fun `test line mode`() {
 | 
			
		||||
    makeEditor("    test\n    three\n    lines\n")
 | 
			
		||||
 | 
			
		||||
    takeAction(AceAction.StartAllLineMarksMode)
 | 
			
		||||
    takeAction(AceKeyboardAction.StartAllLineMarksMode)
 | 
			
		||||
 | 
			
		||||
    assertEquals(8, session.tags.size) // last empty line does not count
 | 
			
		||||
    assertEquals(9, session.tags.size)
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,4 +1,4 @@
 | 
			
		||||
import org.acejump.action.AceAction
 | 
			
		||||
import org.acejump.action.AceKeyboardAction
 | 
			
		||||
import org.acejump.test.util.BaseTest
 | 
			
		||||
import org.junit.Ignore
 | 
			
		||||
import java.io.File
 | 
			
		||||
@@ -15,7 +15,7 @@ class LatencyTest : BaseTest() {
 | 
			
		||||
      
 | 
			
		||||
      for (query in chars) {
 | 
			
		||||
        makeEditor(editorText)
 | 
			
		||||
        myFixture.testAction(AceAction.ActivateOrCycleMode)
 | 
			
		||||
        myFixture.testAction(AceKeyboardAction.ActivateAceJumpSpecial)
 | 
			
		||||
        time += measureTimeMillis { typeAndWaitForResults("$query") }
 | 
			
		||||
        // TODO assert(Tagger.markers.isNotEmpty()) { "Should be tagged: $query" }
 | 
			
		||||
        resetEditor()
 | 
			
		||||
 
 | 
			
		||||
@@ -2,11 +2,12 @@ package org.acejump.test.util
 | 
			
		||||
 | 
			
		||||
import com.intellij.openapi.actionSystem.AnAction
 | 
			
		||||
import com.intellij.openapi.actionSystem.IdeActions
 | 
			
		||||
import com.intellij.openapi.editor.impl.EditorImpl
 | 
			
		||||
import com.intellij.openapi.fileTypes.PlainTextFileType
 | 
			
		||||
import com.intellij.psi.PsiFile
 | 
			
		||||
import com.intellij.testFramework.fixtures.BasePlatformTestCase
 | 
			
		||||
import com.intellij.util.ui.UIUtil
 | 
			
		||||
import org.acejump.action.AceAction
 | 
			
		||||
import org.acejump.action.AceKeyboardAction
 | 
			
		||||
import org.acejump.session.SessionManager
 | 
			
		||||
 | 
			
		||||
abstract class BaseTest : BasePlatformTestCase() {
 | 
			
		||||
@@ -38,7 +39,9 @@ abstract class BaseTest : BasePlatformTestCase() {
 | 
			
		||||
  fun takeAction(action: AnAction) = myFixture.testAction(action)
 | 
			
		||||
  
 | 
			
		||||
  fun makeEditor(contents: String): PsiFile {
 | 
			
		||||
    return myFixture.configureByText(PlainTextFileType.INSTANCE, contents)
 | 
			
		||||
    val file = myFixture.configureByText(PlainTextFileType.INSTANCE, contents)
 | 
			
		||||
    (myFixture.editor as EditorImpl).scrollPane.viewport.setSize(1000, 100)
 | 
			
		||||
    return file
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun resetEditor() {
 | 
			
		||||
@@ -52,10 +55,10 @@ abstract class BaseTest : BasePlatformTestCase() {
 | 
			
		||||
    UIUtil.dispatchAllInvocationEvents()
 | 
			
		||||
  }
 | 
			
		||||
  
 | 
			
		||||
  fun String.executeQuery(query: String) {
 | 
			
		||||
  private fun String.executeQuery(query: String) {
 | 
			
		||||
    myFixture.run {
 | 
			
		||||
      makeEditor(this@executeQuery)
 | 
			
		||||
      testAction(AceAction.ActivateOrCycleMode)
 | 
			
		||||
      testAction(AceKeyboardAction.ActivateAceJump)
 | 
			
		||||
      typeAndWaitForResults(query)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user