nissh 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3101450e4df71a8c85bfe1291a2afc908548fbb2
+  data.tar.gz: 09667cd03ed47386c30ddfe64880c285de8a7fec
+SHA512:
+  metadata.gz: 9aaf6594647384a86aa6e6640d3d84d6c0b29642d764eb8b0fcf09756065251469eac7b0b08b36d4e0eccf0ad7718feb8340e233d2fa9b3895adea16c0b0ea4c
+  data.tar.gz: 3441ccf451bf16810a3f46d29ca991d01ee8d154b480087b21d9bfe1d0aa9a0f7c44e96ee8dad2b489f4e85ec79fa2747eb783abcba1a04ccba18b6e5d023dcd

data/lib/nissh.rb

@@ -0,0 +1 @@
+require 'nissh/session'

data/lib/nissh/response.rb

@@ -0,0 +1,24 @@
+module Nissh
+  class Response
+
+    attr_accessor :stdout, :stderr, :exit_code, :exit_signal, :command
+
+    def initialize
+      reset_output!
+    end
+
+    def success?
+      exit_code == 0
+    end
+
+    def output
+      "#{stdout}\n#{stderr}"
+    end
+
+    def reset_output!
+      @stdout = ""
+      @stderr = ""
+    end
+
+  end
+end

data/lib/nissh/session.rb

@@ -0,0 +1,114 @@
+require 'net/ssh'
+require 'nissh/response'
+
+module Nissh
+  class Session
+
+    class CommandExecutionFailed < StandardError; end
+
+    class << self
+      attr_accessor :logger
+    end
+
+    attr_reader :session
+    attr_accessor :sudo_password
+
+    def initialize(*args)
+      if args.first.is_a?(Net::SSH::Connection::Session)
+        @session = args.first
+      else
+        @session = Net::SSH.start(*args)
+      end
+    end
+
+    def execute!(commands, options = {})
+      unless commands.is_a?(Array)
+        commands = [commands]
+      end
+
+      if options[:sudo]
+        commands = commands.map do |command|
+          "sudo --stdin #{command}"
+        end
+      end
+
+      command = commands.join(' && ')
+      log :info, "\e[44;37m=> #{command}\e[0m"
+
+      response = Nissh::Response.new
+      response.command = command
+      channel = @session.open_channel do |channel|
+        channel.exec(command) do |_, success|
+          raise CommandExecutionFailed, "Command \"#{command}\" was unable to execute" unless success
+          channel.on_data do |_,data|
+            response.stdout += data
+            log :debug, data.gsub(/[\r]/, ''), :tab => 4
+          end
+
+          channel.on_extended_data do |_,_,data|
+            response.stderr += data.gsub(/\r/, '')
+            log :warn, data, :tab => 4
+            if data =~ /^\[sudo\] password for/
+              password = options[:sudo].is_a?(String) ? options[:sudo] : self.sudo_password
+              channel.send_data "#{password}\n"
+            end
+          end
+
+          channel.on_request("exit-status") do |_,data|
+            response.exit_code = data.read_long
+            log :info, "\e[43;37m=> Exit status: #{response.exit_code}\e[0m"
+          end
+
+          channel.on_request("exit-signal") do |_, data|
+            response.exit_signal = data.read_long
+          end
+        end
+      end
+      channel.wait
+      response
+    end
+
+    def execute_with_timeout!(command, timeout = 30)
+      Timeout.timeout(timeout) do
+        execute!(command)
+      end
+    rescue Timeout::Error => e
+      response = Nissh::Response.new
+      response.exit_code -255
+      response.stderr = "Command did not finish executing within the allowed #{timeout} seconds."
+      response.command = command
+      response
+    end
+
+    def execute_with_success!(command, success_code = 0)
+      result = execute!(command)
+      if result.success?
+        result
+      else
+        false
+      end
+    end
+
+    def execute_with_exception!(command, success_code = 0)
+      result = execute!(command)
+      if result.exit_code == success_code
+        result
+      else
+        raise CommandExecutionFailed, result.output
+      end
+    end
+
+    private
+
+    def log(type, text, options = {})
+      if self.class.logger
+        prefix = "\e[45;37m[#{@session.transport.host}]\e[0m"
+        tabs = " " * (options[:tab] || 0)
+        text.split(/\n/).each do |line|
+          self.class.logger.send(type,  prefix + tabs + line)
+        end
+      end
+    end
+
+  end
+end

data/lib/nissh/version.rb

@@ -0,0 +1,3 @@
+module Nissh
+  VERSION = '1.0.0'
+end

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+name: nissh
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Adam Cooke
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-08-31 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2'
+description: A wrapper for net/ssh to make running commands more fun
+email:
+- me@adamcooke.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/nissh.rb
+- lib/nissh/response.rb
+- lib/nissh/session.rb
+- lib/nissh/version.rb
+homepage: https://github.com/adamcooke/nissh
+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.4.5
+signing_key: 
+specification_version: 4
+summary: A wrapper for net/ssh to make running commands more fun
+test_files: []
+has_rdoc: