76 lines
1.3 KiB
Ruby
76 lines
1.3 KiB
Ruby
require 'securerandom'
|
|
|
|
class TerminalSession
|
|
ZSH_BIN = ENV['TEST_ZSH_BIN'] || 'zsh'
|
|
|
|
def initialize(opts = {})
|
|
opts = {
|
|
width: 80,
|
|
height: 24,
|
|
prompt: '',
|
|
term: 'xterm-256color',
|
|
zsh_bin: ZSH_BIN
|
|
}.merge(opts)
|
|
|
|
cmd="PS1=#{opts[:prompt]} TERM=#{opts[:term]} #{ZSH_BIN} -f"
|
|
tmux_command("new-session -d -x #{opts[:width]} -y #{opts[:height]} '#{cmd}'")
|
|
end
|
|
|
|
def run_command(command)
|
|
send_string(command)
|
|
send_keys('enter')
|
|
|
|
self
|
|
end
|
|
|
|
def send_string(str)
|
|
tmux_command("send-keys -t 0 -l '#{str.gsub("'", "\\'")}'")
|
|
|
|
self
|
|
end
|
|
|
|
def send_keys(*keys)
|
|
tmux_command("send-keys -t 0 #{keys.join(' ')}")
|
|
|
|
self
|
|
end
|
|
|
|
def content(esc_seqs: false)
|
|
cmd = 'capture-pane -p -t 0'
|
|
cmd += ' -e' if esc_seqs
|
|
tmux_command(cmd).strip
|
|
end
|
|
|
|
def clear_screen
|
|
send_keys('C-l')
|
|
sleep(0.1) until content == ''
|
|
|
|
self
|
|
end
|
|
|
|
def destroy
|
|
tmux_command('kill-session')
|
|
end
|
|
|
|
def cursor
|
|
tmux_command("display-message -t 0 -p '\#{cursor_x},\#{cursor_y}'").
|
|
strip.
|
|
split(',').
|
|
map(&:to_i)
|
|
end
|
|
|
|
private
|
|
|
|
def tmux_socket_name
|
|
@tmux_socket_name ||= SecureRandom.hex(6)
|
|
end
|
|
|
|
def tmux_command(cmd)
|
|
out = `tmux -u -L #{tmux_socket_name} #{cmd}`
|
|
|
|
raise("tmux error running: '#{cmd}'") unless $?.success?
|
|
|
|
out
|
|
end
|
|
end
|