ssp 0.4 → 0.5

data/lib/ssp/application/domain.rb

@@ -1,8 +1,10 @@
 require 'linode'
 require 'ssp/config'
 
-class OpenStruct
-  undef :type # fixing object.type deprecation issue
+if OpenStruct.new.respond_to?(:type)
+  class OpenStruct
+    undef :type # fixing object.type deprecation issue
+  end
 end
 
 class SSP::App::Domain < Thor

data/lib/ssp/application/remote.rb

@@ -0,0 +1,160 @@
+require 'chef/config'
+require 'chef/search/query'
+require 'chef/node'
+
+class Chef::Knife::Ssh
+  def print_data(host, data)
+    if data =~ /\n/
+      data.split(/\n/).each { |d| print_data(host, d) }
+    else
+      unless config[:dont_print_hosts]
+        padding = @longest - host.length
+        print h.color(host, :cyan)
+        print " " * padding
+      end
+      puts data
+    end
+  end
+end
+
+class SSP::App::Remote < Thor
+  namespace :remote
+
+  default_task :command
+
+  class_option :chef_config, :aliases => "-c",
+               :desc => "Location of the knife configuration file",
+               :default => File.join(ENV['HOME'], '.chef', 'knife.rb')
+
+  class_option :ssh_user, :aliases => "-u",
+               :desc => <<-EOH.squeeze(" ").strip
+                 The ssh username. Defaults to the project name if only
+                 project nodes are provided or to the current local user
+                 otherwise,
+               EOH
+  class_option :ssh_password, :aliases => "-p",
+               :desc => "The ssh password"
+  class_option :cwd, :aliases => "-d",
+               :desc => <<-EOH.squeeze(" ").strip
+                 The working directory where the command should be run.
+                 If the user is a project user (e.g. dtm) then cwd defaults
+                 to an intelligent choice: with no value it will be ~/current,
+                 but it can be given as a partial SHA1 hash (at least 6 chars
+                 long) in which case it will be used as a release directory.
+                 Provide $HOME to run in the user's home dir.
+               EOH
+
+
+  desc "command NODE_or_QUERY COMMAND", "Runs a command on multiple nodes"
+  def command(nodes, *args)
+    chef_config
+
+    nodes = nodes.split(" ").map {|n| n.index('.') or n.index(':') ? n : "#{n}.sspti.me"}
+    user = ssh_user(nodes)
+    cwd = working_dir(user)
+    command = "cd #{cwd}\n#{args.join(" ")}"
+
+    say "Running #{shell.set_color(args.join(" ").inspect, :green)} in #{shell.set_color(cwd, :yellow)} as #{shell.set_color(user, :blue)} on #{nodes.map {|n| shell.set_color(n, :cyan)}.join(", ").gsub(/,([^,]*)$/, " and\\1")}"
+    ssh_run nodes.join(" "), command, user
+  end
+
+  desc "command NODE_or_QUERY TASK...", "Runs rake task(s) on multiple nodes"
+  def rake(nodes, *args)
+    args.unshift("rake")
+    command(nodes, *args)
+  end
+
+  desc "log LOGFILE", "Tails a logfile"
+  method_option :node, :aliases => "-n",
+                :desc => "The node where the logfile is located",
+                :default => "dtm-db-2"
+  method_option :host, :aliases => "-h",
+                :desc => "Host or project name (e.g. sage or dtm)"
+  method_option :date, :aliases => "-D",
+                :desc => "The date of the logfile you're interested in. Defaults to today, turns off tailing if this is in the past.",
+                :default => Date.today.to_s
+  method_option :history, :aliases => "-N",
+                :desc => "How many lines to show from the past",
+                :default => 10
+  def log(logfile)
+    @working_dir = options[:cwd] || options[:node] == "dtm-db-2" ? "/srv/rsyslog" : "/var/log"
+    @working_dir += '/' + options[:host] if options[:host]
+    @working_dir += '/' + options[:date].tr('-', '/')
+
+    @dont_print_hosts = true
+    tail = "-f " if Date.parse(options[:date]) == Date.today
+    command(options[:node], "tail -n#{options[:history]} #{tail}#{logfile}")
+  rescue Interrupt
+    # we don't care about interrupting a tail
+  end
+
+
+private
+
+  def ssh_run(host, command, user, pass = options[:ssh_password], manual = !host.index(':'))
+    ssh = ::Chef::Knife::Ssh.new
+    ssh.name_args = [ host, command ]
+    ssh.config[:ssh_user] = user
+    ssh.config[:password] = pass
+    ssh.config[:manual] = manual
+    if @dont_print_hosts
+      ssh.config[:dont_print_hosts] = true
+    end
+
+    begin
+      ssh.run
+    rescue Net::SSH::AuthenticationFailed
+      unless pass
+        puts "Failed to authenticate #{user} - trying password auth"
+        ssh = ::Chef::Knife::Ssh.new
+        ssh.name_args = [ host, command ]
+        ssh.config[:ssh_user] = user
+        ssh.config[:manual] = manual
+        ssh.config[:password] = ssh.get_password
+        if @dont_print_hosts
+          ssh.config[:dont_print_hosts] = true
+        end
+        ssh.run
+      end
+    end
+  end
+
+  def working_dir(user)
+    if @working_dir
+      @working_dir
+    elsif %w{dtm}.include?(user)
+      if options[:cwd] =~ /^[a-f0-9]{6,}$/
+        "~/releases/#{options[:cwd]}*"
+      elsif options[:cwd].nil?
+        "~/current"
+      end
+    else
+      options[:cwd] || "~"
+    end
+  end
+
+  def user_from_nodes(nodes)
+    if nodes.all? {|n| n.index('-')}
+      project_names = nodes.map do |node|
+        if node.index(':').nil? or node.split(':').first == "hostname"
+          node.split(':').last.split('-').first
+        end
+      end
+      if project_names.uniq.size == 1
+        project_names.first
+      end
+    end
+  end
+
+  def ssh_user(nodes)
+    options[:ssh_user] || user_from_nodes(nodes) || ENV['USER'].downcase
+  end
+
+  def chef_config
+    unless defined?(@_chef_config_loaded)
+      ::Chef::Config.from_file(options[:chef_config])
+      @_chef_config_loaded = true
+    end
+    ::Chef::Config
+  end
+end

data/lib/ssp/version.rb

@@ -1,3 +1,3 @@
 module SSP
-  VERSION = "0.4"
+  VERSION = "0.5"
 end

metadata

@@ -1,22 +1,19 @@
 --- !ruby/object:Gem::Specification 
 name: ssp
 version: !ruby/object:Gem::Version 
-  hash: 3
   prerelease: false
   segments: 
   - 0
-  - 4
-  version: "0.4"
+  - 5
+  version: "0.5"
 platform: ruby
 authors: 
-- !binary |
-  TMOhc3psw7MgQsOhY3Np
-
+- "L\xC3\xA1szl\xC3\xB3 B\xC3\xA1csi"
 autorequire: 
 bindir: bin
 cert_chain: []
 
-date: 2010-10-03 00:00:00 +02:00
+date: 2010-10-14 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -27,7 +24,6 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        hash: 17
         segments: 
         - 0
         - 13
@@ -42,7 +38,6 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        hash: 25
         segments: 
         - 0
         - 9
@@ -57,7 +52,6 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        hash: 11
         segments: 
         - 0
         - 2
@@ -73,7 +67,6 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        hash: 21
         segments: 
         - 1
         - 0
@@ -89,7 +82,6 @@ dependencies:
     requirements: 
     - - ~>
       - !ruby/object:Gem::Version 
-        hash: 3
         segments: 
         - 0
         - 6
@@ -116,6 +108,7 @@ files:
 - lib/ssp/application/node.rb
 - lib/ssp/application/office_node.rb
 - lib/ssp/application/pair.rb
+- lib/ssp/application/remote.rb
 - lib/ssp/config.rb
 - lib/ssp/version.rb
 has_rdoc: true
@@ -132,7 +125,6 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 3
       segments: 
       - 0
       version: "0"
@@ -141,7 +133,6 @@ required_rubygems_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 3
       segments: 
       - 0
       version: "0"