IntelliJ-AceJump/src/main/kotlin/org/acejump/action/AceEditorAction.kt

package org.acejump.action

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

/**
 * 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() {
  final override fun isEnabledForCaret(editor: Editor, caret: Caret, dataContext: DataContext?): Boolean {
    return SessionManager[editor] != null || originalHandler.isEnabled(editor, caret, dataContext)
  }
  
  final override fun doExecute(editor: Editor, caret: Caret?, dataContext: DataContext?) {
    val session = SessionManager[editor]
    
    if (session != null) {
      run(session)
    }
    else if (originalHandler.isEnabled(editor, caret, dataContext)) {
      originalHandler.execute(editor, caret, dataContext)
    }
  }
  
  protected abstract fun run(session: Session)
  
  // Actions
  
  class Reset(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
    override fun run(session: Session) = session.end()
  }
  
  class ClearSearch(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
    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 SearchCodeIndents(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
    override fun run(session: Session) = session.startRegexSearch(Pattern.CODE_INDENTS, StandardBoundaries.WHOLE_FILE)
  }
  
  class SearchLineStarts(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
    override fun run(session: Session) = session.startRegexSearch(Pattern.START_OF_LINE, StandardBoundaries.WHOLE_FILE)
  }
  
  class SearchLineEnds(originalHandler: EditorActionHandler) : AceEditorAction(originalHandler) {
    override fun run(session: Session) = session.startRegexSearch(Pattern.END_OF_LINE, StandardBoundaries.WHOLE_FILE)
  }
}