class TmuxClusterSSH

# File lib/tssh.rb, line 2
def initialize options, hosts, logger
  @options = options
  @hosts = hosts
  @logger = logger
  raise "No hosts specified" if @hosts.count == 0
end

# File lib/tssh.rb, line 17
def calc_num_panes(num)
  # XXX should read tmux implementation
  if num > @options[:panes_per_window]
    return @options[:panes_per_window] + calc_num_panes(num - @options[:panes_per_window])
  end

  sqrt_num = Math.sqrt(num)
  panes = sqrt_num.ceil * sqrt_num.floor
  panes >= num ? panes : sqrt_num.ceil ** 2
end

# File lib/tssh.rb, line 39
def run
  num_panes = calc_num_panes(@hosts.count)
  session_name = "tssh-#{$$}"
  (0..num_panes-1).each do |i|
    host = @hosts[i]
    if host.nil?
      command = "'echo;echo empty'"
    else
      command = "'echo #{host};exec #{ssh_command(host)}'"
    end
    STDERR.puts command if @options[:debug] > 0
    if i == 0
      # create new session
      run_command "tmux new -d -s #{session_name} #{command}"
      run_command "tmux setw -t #{session_name} synchronize-panes on > /dev/null"
      run_command "tmux setw -t #{session_name} remain-on-exit on > /dev/null"
    elsif i  % @options[:panes_per_window] == 0
      # create new window on existing session
      run_command "tmux neww -t #{session_name} #{command}"
      run_command "tmux setw -t #{session_name} synchronize-panes on > /dev/null"
      run_command "tmux setw -t #{session_name} remain-on-exit on > /dev/null"
    else
      # add pane to active window
      run_command "tmux splitw -t #{session_name} #{command}"
      run_command "tmux selectl -t #{session_name} tiled > /dev/null"
    end
  end
  # XXX
  run_command 'tmux select-pane -t 0'

  # attach to the session
  exec "tmux attach -t #{session_name}"
end

# File lib/tssh.rb, line 28
def run_command cmd
  @logger.info cmd
  res = system cmd
  case res
  when nil
    raise "command `#{cmd}` failed"
  when false
    raise "command `#{cmd}` exited with status #{$?}"
  end
end

# File lib/tssh.rb, line 9
def ssh_command host
  if @options[:login].nil?
    "#{@options[:ssh]} #{@options[:ssh_args]} #{host}"
  else
    "#{@options[:ssh]} #{@options[:ssh_args]} #{@options[:login]}@#{host}"
  end
end