i2cssh 1.1.0 → 1.2.0

data/README.markdown

@@ -9,13 +9,14 @@ to all sessions.
     $ gem install i2cssh
 
 ## Usage
-
     Usage: i2cssh [options]
         -d, --debug                      Start RIPL after creating terminals
-        -f, --file FILE                  Cluster file
+        -A, --forward-agent              Enable SSH agent forwarding
+        -f, --file FILE                  Cluster file (one hostname per line)
         -F, --fullscreen                 Fullscreen
-        -u, --username USERNAME          SSH username
+        -l, --login LOGIN                SSH login name
         -c, --cluster CLUSTERNAME        Name of the cluster specified in ~/.i2csshrc
+        -m, --machines a,b,c             Comma-separated list of hosts
         -C, --columns COLUMNS            Amount of columns (rows will be calculated)
         -R, --rows ROWS                  Amount of rows (columns will be calculated)
 

data/VERSION

@@ -1 +1 @@
-1.1.0
+1.2.0

data/bin/i2cssh

@@ -1,111 +1,65 @@
 #!/usr/bin/ruby
 require 'rubygems'
-require 'appscript'
 require 'optparse'
+require 'i2cssh'
 
-options = {}
+i2_options, ssh_options, servers = {}, [], []
 
 optparse = OptionParser.new do |opts|
-  opts.banner = "Usage: #{File.basename(__FILE__)} [options]"
-  options[:debug] = false
-  options[:fullscreen] = false
-  opts.on('-d', '--debug', "Start RIPL after creating terminals") { options[:debug] = true }
-  opts.on('-A', '--forward-agent', "Enable SSH agent forwarding") { options[:forward] = true }
-  opts.on('-f', '--file FILE', "Cluster file (one hostname per line)") { |f| options[:file] = f }
-  opts.on('-F', '--fullscreen', "Fullscreen") { options[:fullscreen] = true }
-  opts.on('-l', '--login LOGIN', "SSH login name") { |u| options[:login] = u }
-  opts.on('-c', '--cluster CLUSTERNAME', "Name of the cluster specified in ~/.i2csshrc") { |c| options[:cluster] = c }
-  opts.on('-m', '--machines a,b,c', Array, "Comma-separated list of hosts") { |h| options[:hosts] = h }
-  opts.on('-C', '--columns COLUMNS', "Amount of columns (rows will be calculated)") { |c| options[:columns] = c }
-  opts.on('-R', '--rows ROWS', "Amount of rows (columns will be calculated)") { |r| options[:rows] = r }
-#  opts.on('-s', '--session SESSIONNAME', "Name of an iTerm2 session") { |s| options[:session_name] = s }
+    opts.banner = "Usage: #{File.basename(__FILE__)} [options]"
+
+    # SSH options
+    opts.on '-A', '--forward-agent',
+        'Enable SSH agent forwarding' do
+        ssh_options << '-A'
+    end
+    opts.on '-l', '--login LOGIN',
+        'SSH login name' do |u|
+        ssh_options << "-l #{u}"
+    end
+
+    # iTerm2 options
+    opts.on '-F', '--fullscreen',
+        'Make the window fullscreen' do
+        i2_options[:fullscreen] = true
+    end
+    opts.on '-C', '--columns COLUMNS', Integer,
+        'Number of columns (rows will be calculated)' do |c|
+        i2_options[:columns] = c
+    end
+    opts.on '-R', '--rows ROWS', Integer,
+        'Number of rows (columns will be calculated)' do |r|
+        if i2_options[:columns]
+            puts "ERROR: -C and -R can't be used at the same time"
+            puts optparse.help
+            exit
+        else
+            i2_options[:rows] = r
+        end
+    end
+
+    # Hosts
+    opts.on '-f', '--file FILE',
+        'Cluster file (one hostname per line)' do |f|
+        servers += File.read(f).split "\n"
+    end
+    opts.on '-c', '--cluster CLUSTERNAME',
+        'Name of the cluster specified in ~/.i2csshrc' do |c|
+        require 'yaml'
+        config_hash = YAML.load File.read File.expand_path '~/.i2csshrc'
+        servers += config_hash["clusters"][c]
+    end
+    opts.on '-m', '--machines a,b,c', Array,
+        'Comma-separated list of hosts' do |h|
+        servers += h
+    end
 end
-
 optparse.parse!
 
-unless options[:file] || options[:cluster] || options[:hosts] then
-  puts optparse.help
-  exit
-end
-
-if [ options[:file], options[:cluster], options[:hosts] ].compact.length > 1 then
-  puts "ERROR: neither -f, -c or -h can be combined"
-  puts optparse.help
-  exit
-end
-
-if options[:columns]  && options[:rows] then
-  puts "ERROR: -C and -R can't be used at the same time"
-  puts optparse.help
-  exit
-end
-
-ssh_prefix = "ssh " +
-    (options[:forward] ? "-A " : "") +
-    (options[:login] ? "-l #{options[:login]} " : "")
-
-rows = options[:rows].to_i if options[:rows]
-columns = options[:columns].to_i if options[:columns]
-
-
-if options[:file] then
-  servers = File.read(options[:file]).split("\n")
-elsif options[:cluster] then
-  require 'yaml'
-  config_hash = YAML.load(File.read(File.expand_path("~/.i2csshrc")))
-  servers = config_hash["clusters"][options[:cluster]]
-elsif options[:hosts] then
-  servers = options[:hosts]
-end
-
-count = servers.size
-if rows then
-  columns = (count / rows.to_f).ceil
-elsif columns then
-  rows = (count / columns.to_f).ceil
-else
-  rows = Math.sqrt(count).ceil
-  columns = (count / rows.to_f).ceil
+if servers.empty?
+    puts "ERROR: no servers given"
+    puts optparse.help
+    exit
 end
 
-iterm = Appscript.app.by_name('iTerm')
-finder = Appscript.app.by_name('Finder')
-sys_events = Appscript.app.by_name('System Events')
-term = iterm.make(:new => :terminal)
-
-session = term.sessions.after.make(:new => :session)
-session.exec(:command => "/bin/bash -l")
-
-if options[:fullscreen] then
-  window_bounds = finder.desktop.window.bounds
-  window = iterm.windows.get.sort_by{|x| x.id_.get}.last
-  window.bounds.set(window_bounds.get)
-end
-
-(columns - 1).times do
-  sys_events.keystroke("d", :using => :command_down)
-end
-(rows - 1).times do
-  (columns - 1).times do
-    sys_events.key_code(123, :using => [:command_down, :option_down])
-  end
-  (columns).times do |x|
-    sys_events.keystroke("D", :using => :command_down)
-    sys_events.key_code(124, :using => [:command_down, :option_down]) unless (columns - 1) == x
-  end
-end
-
-(rows * columns).times do |i|
-  if servers && servers[i] then
-    term.sessions[i+1].write(:text => "history -d $(($HISTCMD-1)) && #{ssh_prefix}#{servers[i]}") 
-  else
-    term.sessions[i+1].foreground_color.set("red")
-    term.sessions[i+1].write(:text => "history -d $(($HISTCMD-1)) && stty -isig -icanon -echo && echo -e '#{"\n"*100}UNUSED' && cat > /dev/null")
-  end
-end
-if options[:debug] then
-  require 'ripl'
-  Ripl.start :binding => binding
-end
-
-
+I2Cssh.new servers, ssh_options, i2_options

data/i2cssh.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{i2cssh}
-  s.version = "1.1.0"
+  s.version = "1.2.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Wouter de Bie"]

data/lib/i2cssh.rb

@@ -0,0 +1,71 @@
+require 'appscript'
+
+class I2Cssh
+    def initialize servers, ssh_options, i2_options
+        @ssh_prefix  = "ssh " + ssh_options.join(' ')
+        @ssh_options = ssh_options
+        @i2_options  = i2_options
+        @servers     = servers
+
+        raise Exception.new 'No servers given' if servers.empty?
+
+        @sys_events = Appscript.app.by_name('System Events')
+        @iterm = Appscript.app.by_name('iTerm')
+        @term = @iterm.make(:new => :terminal)
+        session = @term.sessions.after.make :new => :session
+        session.exec :command => "/bin/bash -l"
+
+        compute_geometry
+        split_session
+        maximize if i2_options[:fullscreen]
+        start_ssh
+    end
+
+    private
+    def maximize
+      fullscreen_bounds = Appscript.app.by_name('Finder').desktop.window.bounds
+      window = @iterm.windows.get.sort_by{|x| x.id_.get}.last
+      window.bounds.set fullscreen_bounds.get
+    end
+
+    def compute_geometry
+        count = @servers.size
+        if @i2_options[:rows] then
+          @columns = (count / @i2_options[:rows].to_f).ceil
+        elsif @i2_options[:rows] then
+          @rows = (count / @i2_options[:columns].to_f).ceil
+        else
+          @columns = Math.sqrt(count).ceil
+          @rows = (count / @columns.to_f).ceil
+        end
+        puts @rows
+        puts @columns
+    end
+
+    def split_session
+        2.upto @columns do
+          @sys_events.keystroke "d", :using => :command_down
+        end
+        2.upto @rows do
+          1.upto @columns do
+            @sys_events.key_code 123, :using => [:command_down, :option_down]
+          end
+          @columns.times do |x|
+            @sys_events.keystroke "D", :using => :command_down
+            @sys_events.key_code 124, :using => [:command_down, :option_down] unless @columns - 1 == x
+          end
+        end
+    end
+
+    def start_ssh
+        1.upto(@rows*@columns) do |i|
+            server = @servers[i-1]
+            if server then
+                @term.sessions[i].write :text => "unset HISTFILE && #{@ssh_prefix} #{server}"
+            else
+                @term.sessions[i].foreground_color.set "red"
+                @term.sessions[i].write :text => "unset HISTFILE && stty -isig -icanon -echo && echo -e '#{"\n"*100}UNUSED' && cat > /dev/null"
+            end
+        end
+    end
+end

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 1
-  - 1
+  - 2
   - 0
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors: 
 - Wouter de Bie