sshkit_addon 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 646bed166f7237a05479be5c16db53ef4b7d4544
+  data.tar.gz: a2247967ece7259e48510e37c3fdc4dfc6b71cc2
+SHA512:
+  metadata.gz: aa77e0c45d9303a9d2a61819371c2bf7c7631f37c505f0af345d7d4ed63657651c1395745327de3664418de56408f99adb73fff04c9bf5048cb46ba24b05885b
+  data.tar.gz: 8175e2d4c4e13689efad3a605f06dd1eb49e0cbb4c31727a9b86b18919da46da5e7c898cabe6ffc52dc5b9de8a26632f37ba66ebbd55b95e160852d00d8a4879

data/lib/addon/cap_base.rb

@@ -0,0 +1,33 @@
+require "sshkit"
+require "sshkit/dsl"
+include SSHKit::DSL
+
+require File.dirname(__FILE__) + "/sshkit_monkey_patch"
+
+class CapBase
+	attr_reader :servers, :servers_by_hash
+	def initialize server_hash
+		@servers = []
+		@servers_by_hash = {}
+
+		server_hash.each do |name, hash|
+			args = [:host,:user,:port,:password].each_with_object({}) do |key, h|
+				if key == :host
+					h[:hostname] = hash[key]
+				else
+					h[key] = hash[key] if hash.has_key? key
+				end
+			end
+
+			# host = SSHKit::Host.new :hostname => hash[:host], :user => hash[:user], :password => hash[:password]
+			host = SSHKit::Host.new args
+			host.key = hash[:key] if hash.has_key? :key #use ssh private key file
+
+			host.properties.options = hash
+
+			@servers.push host
+
+			@servers_by_hash[name] = host
+		end
+	end
+end

data/lib/addon/shell_command_constructor.rb

@@ -0,0 +1,16 @@
+module ShellCommandConstructor
+	def self.construct_command strings_or_list, connector = " && "
+   	list = case strings_or_list
+   	when Array
+   		strings_or_list
+   	when String
+   		strings_or_list.split(/\n/)
+   	end
+   	list.each_with_object([]) do |line, obj|
+       	line.strip!
+       	next if line.empty?
+       	next if line =~ /^#/
+       	obj.push line
+   	end.join connector
+	end
+end

data/lib/addon/sshkit_monkey_patch.rb

@@ -0,0 +1,31 @@
+#take out the support of ssh gateway as the upload/download is not really supported
+require File.dirname(__FILE__) + "/sshkit_my_formatter"
+SSHKit.config.use_format :myformatter
+
+module SSHKit; module Backend; class Netssh
+    def put strings, remote, options = {}
+        local = StringIO.new strings
+        upload! local, remote, options
+    end
+
+    def su_run user, cmd
+      execute %Q(su - #{user} -c "#{cmd}")
+    end
+    alias_method :su_execute, :su_run
+
+    def sudo cmd
+        SSHKit::Backend::Netssh.config.pty = true
+        # p host
+        password = host.password
+        execute %Q( echo #{password} | sudo -kS bash -c "#{cmd}" )
+
+        # execute %Q(sudo bash -c "#{cmd}"), interaction_handler: {
+        #     /\(yes\/no\)/i => "yes\n",
+        #     /[sudo] password/i => "#{password}\n",
+        # }
+
+    end
+
+end;end;end
+
+SSHKit.config.output_verbosity = SSHKit::Logger::DEBUG

data/lib/addon/sshkit_my_formatter.rb

@@ -0,0 +1,58 @@
+module SSHKit
+  module Formatter
+    class MyFormatter < Abstract
+      LEVEL_NAMES = %w{ DEBUG INFO WARN ERROR FATAL }.freeze
+      LEVEL_COLORS = [:black, :blue, :yellow, :red, :red].freeze
+
+      def write(obj)
+        if obj.kind_of?(SSHKit::LogMessage)
+          write_message(obj.verbosity, obj.to_s)
+        else
+          raise "write only supports formatting SSHKit::LogMessage, called with #{obj.class}: #{obj.inspect}"
+        end
+      end
+
+      def log_command_start(command)
+        host_prefix = command.host.user ? "as #{colorize(command.host.user, :blue)}@" : 'on '
+        message = "Running #{colorize(command, :yellow, :bold)} #{host_prefix}#{colorize(command.host, :blue)}"
+        write_message(command.verbosity, message, command.uuid)
+        write_message(Logger::DEBUG, "Command: #{colorize(command.to_command, :blue)}", command.uuid)
+      end
+
+      def log_command_data(command, stream_type, stream_data)
+        color = \
+          case stream_type
+          when :stdout then :green
+          when :stderr then :red
+          else raise "Unrecognised stream_type #{stream_type}, expected :stdout or :stderr"
+          end
+        uuid = "#{command.host}: (#{command.uuid})"
+        write_message(Logger::DEBUG, colorize("\t#{stream_data}".chomp, color), uuid)
+      end
+
+      def log_command_exit(command)
+        runtime = sprintf('%5.3f seconds', command.runtime)
+        successful_or_failed = command.failure? ? colorize('failed', :red, :bold) : colorize('successful', :green, :bold)
+        message = "Finished in #{runtime} with exit status #{command.exit_status} (#{successful_or_failed})."
+        write_message(command.verbosity, message, command.uuid)
+      end
+
+      protected
+
+      def format_message(verbosity, message, uuid=nil)
+        message = "[#{colorize(uuid, :green)}] #{message}" unless uuid.nil?
+        level = colorize(LEVEL_NAMES[verbosity].rjust(6), LEVEL_COLORS[verbosity])
+        "#{level} #{message}"
+      end
+
+      private
+
+      def write_message(verbosity, message, uuid=nil)
+        original_output << "#{format_message(verbosity, message, uuid)}\n" if verbosity >= SSHKit.config.output_verbosity
+      end
+
+    end
+
+  end
+
+end

data/lib/sshkit_addon.rb

@@ -0,0 +1,9 @@
+require "sshkit"
+require "sshkit/dsl"
+include SSHKit::DSL
+
+require File.dirname(__FILE__) + "/addon/sshkit_monkey_patch"
+require File.dirname(__FILE__) + "/addon/cap_base"
+require File.dirname(__FILE__) + "/addon/shell_command_constructor"
+
+

data/readme.md

@@ -0,0 +1,29 @@
+# An addon for sshkit
+
+define the sshkit host with hash structure. then define tasks by choosing the host
+```
+cap3 = CapBase.new({
+  :core1 => { :host => "192.168.5.21", :user => "core", :key => %q(mykey.openssh) },
+  :core2 => { :host => "192.168.5.22", :user => "core", :password => 'xxxxx' },
+  :core3 => { :host => "192.168.5.22", :user => "core", :password => 'xxxxx' },
+})
+servers = cap3.servers
+
+desc "run command"
+task :run_command, [:cmd] do |t, args|
+    cmd = args.cmd
+    on servers do |host|
+      execute cmd
+    end
+end
+
+desc "run command on a specific host"
+task :run_command_on, [:host_key, :cmd] do |t, args|
+    cmd = args.cmd
+    on cap3.servers_by_hash[args.host_key] do |host|
+      execute cmd
+    end
+end
+
+```
+

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+name: sshkit_addon
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Zhimin Wen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2010-04-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sshkit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+description: Addon of the sshkit
+email: zhimin.wen@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/addon/cap_base.rb
+- lib/addon/shell_command_constructor.rb
+- lib/addon/sshkit_monkey_patch.rb
+- lib/addon/sshkit_my_formatter.rb
+- lib/sshkit_addon.rb
+- readme.md
+homepage: http://github.com/zhiminwen/sshkit_addon
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.11
+signing_key: 
+specification_version: 4
+summary: sshkit addon
+test_files: []