schasse.tmux-jump/scripts/easymotion.rb

#!/usr/bin/env ruby
require 'timeout'
require 'tempfile'

# SPECIAL STRINGS
GRAY = "\e[0m\e[32m"
# RED = "\e[38;5;124m"
RED = "\e[1m\e[31m"
CLEAR_SEQ = "\e[2J"
HOME_SEQ = "\e[H"
RESET_COLORS = "\e[0m"
ENTER_ALTERNATE_SCREEN = "\e[?1049h"
RESTORE_NORMAL_SCREEN = "\e[?1049l"

# CONFIG
KEYS = 'jfhgkdlsa'.each_char.to_a

# METHODS
def recover_screen_after
  if ALTERNATE_ON == '1'
    recover_alternate_screen_after do
      yield
    end
  else
    recover_normal_screen_after do
      yield
    end
  end
end

def recover_normal_screen_after
  File.open(PANE_TTY_FILE, 'a') do |tty|
    tty << ENTER_ALTERNATE_SCREEN + HOME_SEQ
  end

  returns = yield

  File.open(PANE_TTY_FILE, 'a') do |tty|
    tty << RESTORE_NORMAL_SCREEN
  end
  returns
end

def recover_alternate_screen_after
  saved_screen =
    `tmux capture-pane -ep -t #{PANE_NR}`[0..-2] # with colors...
      .gsub("\n", "\n\r")
  File.open(PANE_TTY_FILE, 'a') do |tty|
    tty << CLEAR_SEQ + HOME_SEQ
  end

  returns = yield

  File.open(PANE_TTY_FILE, 'a') do |tty|
    tty << RESET_COLORS + CLEAR_SEQ
    tty << saved_screen
    tty << "\e[#{CURSOR_Y.to_i + 1};#{CURSOR_X.to_i + 1}H"
    tty << RESET_COLORS
  end
  returns
end

def prompt_char
  tmp_file = Tempfile.new 'tmux-easymotion'
  Kernel.spawn(
    'tmux', 'command-prompt', '-1', '-p', 'char:',
    "run-shell \"printf '%1' >> #{tmp_file.path}\"")
  Timeout.timeout(60*60) do
    last_activity = `tmux display-message -p '\#{session_activity}'`
    loop do
      sleep 0.05
      break if last_activity != `tmux display-message -p '\#{session_activity}'`
    end
  end
  read_char_from_file tmp_file
rescue Timeout::Error
  exit
end

def read_char_from_file(tmp_file)
  char = nil
  Timeout.timeout(0.05) do
    loop do # busy waiting with files :/
      break if char = tmp_file.getc
    end
    File.delete tmp_file
  end
  char
rescue Timeout::Error
  nil
end

def positions_of(jump_to_char, screen_chars)
  positions = []

  positions << 0 if screen_chars[0] =~ /\w/ && screen_chars[0].downcase == jump_to_char
  screen_chars.each_char.with_index do |char, i|
    if (char =~ /\w/).nil? && screen_chars[i+1] && screen_chars[i+1].downcase == jump_to_char
      positions << i+1
    end
  end
  positions
end

def draw_keys_onto_tty(screen_chars, positions, keys, key_len)
  File.open(PANE_TTY_FILE, 'a') do |tty|
    cursor = 0
    positions.each_with_index do |pos, i|
      tty << "#{GRAY}#{screen_chars[cursor..pos-1].gsub("\n", "\n\r")}"
      tty << "#{RED}#{keys[i]}"
      cursor = pos + key_len
    end
    tty << "#{GRAY}#{screen_chars[cursor..-1].gsub("\n", "\n\r")}"
    tty << HOME_SEQ
  end
end

def keys_for(position_count, keys = KEYS)
  if position_count > keys.size
    keys_for(position_count, keys.product(KEYS).map(&:join))
  else
    keys
  end
end

def prompt_position_index(positions, screen_chars)
  return nil if positions.size == 0
  return 0 if positions.size == 1
  keys = keys_for positions.size
  key_len = keys.first.size
  draw_keys_onto_tty screen_chars, positions, keys, key_len
  key_index = KEYS.index(prompt_char)
  if !key_index.nil? && key_len > 1
    magnitude = KEYS.size ** (key_len - 1)
    range_beginning = key_index * magnitude # p.e. 2 * 22^1
    range_ending = range_beginning + magnitude - 1
    remaining_positions = positions[range_beginning..range_ending]
    return nil if remaining_positions.nil?
    lower_index = prompt_position_index(remaining_positions, screen_chars)
    return nil if lower_index.nil?
    range_beginning + lower_index
  else
    key_index
  end
end

def main
  jump_to_char = read_char_from_file File.new(TMP_FILE)
  `tmux send-keys -X -t #{PANE_NR} cancel` if PANE_MODE == '1'
  screen_chars =
    `tmux capture-pane -p -t #{PANE_NR}`[0..-2].gsub("︎", '') # without colors
  positions = positions_of jump_to_char, screen_chars
  position_index = recover_screen_after do
    prompt_position_index positions, screen_chars
  end
  exit 0 if position_index.nil?
  jump_to = positions[position_index]
  `tmux copy-mode -t #{PANE_NR}`
   # begin: tmux weirdness when 1st line is empty
  `tmux send-keys -X -t #{PANE_NR} start-of-line`
  `tmux send-keys -X -t #{PANE_NR} top-line`
  `tmux send-keys -X -t #{PANE_NR} -N 200 cursor-right`
   # end
  `tmux send-keys -X -t #{PANE_NR} start-of-line`
  `tmux send-keys -X -t #{PANE_NR} top-line`
  `tmux send-keys -X -t #{PANE_NR} -N #{jump_to} cursor-right`
end

if $PROGRAM_NAME == __FILE__
  PANE_NR = `tmux display-message -p "\#{pane_id}"`.strip
  format = '#{pane_id};#{pane_tty};#{pane_in_mode};\#{cursor_y};\#{cursor_x};\#{alternate_on}'
  tmux_data = `tmux lsp -a -F "#{format}" | grep #{PANE_NR}`.split(';')
  PANE_TTY_FILE = tmux_data[1]
  PANE_MODE = tmux_data[2]
  CURSOR_Y = tmux_data[3]
  CURSOR_X = tmux_data[4]
  ALTERNATE_ON = tmux_data[5]
  TMP_FILE = ARGV[0]
  main
end