cli_tool 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 17d8d9e6d914296940ccbd28dd3eabb488dc7b46
-  data.tar.gz: 026adfb4294b909858ab0ff98c9d3c89d4a1f53a
+  metadata.gz: 5d6bebdc51c5d3fa303521a576030ef09a2a2f1e
+  data.tar.gz: 44314ab4226493adced0d5300540c55de8194801
 SHA512:
-  metadata.gz: 66bdb094264f87a4928d380b04e5244472dce78b85aef3c047be6400b88645e887612ba8d7abedc1958c8053a7ce246ffbe4724b47476d7234378d7f6f4e2532
-  data.tar.gz: 8250abdc7bbfb6d3d6bf5ee8fed267b0598d1e1dba65b205072695689a15f5e0c7897f813d07e9f53c8317cdefb73ded1db51ae36a66685d113429d94431e2ce
+  metadata.gz: c4f62e534448942a0f7ef307b7c3be7b37807ca658692f89c7df97a11db30de081c6ec5eb82fdc3a6f0408df5e922e92edcc1e2351d636d64124e8bfffa5b57d
+  data.tar.gz: 0f5cc7400c20e4c3732aba67f733873d89843eff4f17901a362617e021518d72dfc22a65dd8d1217b12abdda98eb4c838b35553f26bc48854408397de2f12ec0

data/lib/cli_tool/remote.rb

@@ -0,0 +1,130 @@
+module CliTool
+  module Remote
+    class WaitingForSSH < StandardError; end;
+    class Failure < StandardError; end;
+
+    class Script < String
+      def <<(script)
+        script = script.strip + ';' unless script.match(/;$/)
+        script = Script.new(script)
+        super("#{script}\n")
+      end
+
+      def prepend(script)
+        script = script.strip + ';' unless script.match(/;$/)
+        script = Script.new(script)
+        super("#{script}\n")
+      end
+    end
+
+    def self.included(base)
+      base.__send__(:include, ::CliTool::StdinOut)
+      base.__send__(:include, ::CliTool::OptionParser)
+      base.__send__(:include, ::CliTool::SoftwareDependencies)
+      base.extend(ClassMethods)
+      base.software(:ssh, :nc)
+      base.options host: :required,
+        identity: :required,
+        debug: :none,
+        user: {
+          dependency: :required,
+          default: %x{whoami}.strip
+        },
+        port: {
+          dependency: :required,
+          default: '22'
+        }
+    end
+
+    def script(script = nil, sudo = false, sudouser = nil)
+      @_script ||= Script.new
+      return Script.new(@_script.strip) if script.nil?
+      return @_script = Script.new if script == :reset
+      @_script << (sudo == :sudo ? %{sudo su -l -c "#{script.strip.gsub('"','\"')}" #{sudouser}} : script).strip
+    end
+
+    def script_exec
+      wait4ssh
+
+      command =[ "ssh -t -t" ]
+      command << "-I #{@identity}" if @identity
+      command << "-p #{@port}" if @port
+      command << "#{@user}@#{@host}"
+      command << "<<-SCRIPT\n#{script}\nexit;\nSCRIPT"
+      command  = command.join(' ')
+      script :reset
+
+      puts("Running Remotely:\n#{command}\n", [:blue, :white_bg])
+
+      system(command)
+
+      unless $?.success?
+        raise Failure, "Error running \"#{command}\" on #{@host} exited with code #{$?.to_i}."
+      end
+    end
+
+    def aptget(action = :update, *software)
+      software = software.map(&:to_s).join(' ')
+      script "apt-get #{action} -q -y --force-yes #{software}", :sudo
+    end
+
+    def aptkeyadd(*keys)
+      keys = keys.map(&:to_s).join(' ')
+      script "apt-key adv --keyserver keyserver.ubuntu.com --recv-keys #{keys}", :sudo
+    end
+
+    def restart
+      script :reset
+      script "shutdown -r now &", :sudo
+      script_exec
+
+      # Let the server shutdown
+      sleep 5
+    end
+
+    def adduser(user, system = false)
+      script "adduser --disabled-password --quiet --gecos '' #{user}".squeeze(' '), :sudo
+    end
+
+    def wait4ssh
+      _retry ||= false
+      %x{nc -z #{@host} #{@port}}
+      raise WaitingForSSH unless $?.success?
+      puts("\nSSH is now available!", :green) if _retry
+    rescue WaitingForSSH
+      print "Waiting for ssh..." unless _retry
+      _retry = true
+      print '.'
+      sleep 2
+      retry
+    end
+
+    module ClassMethods
+
+      # Create the options array
+      def software(*soft)
+        @@software ||= []
+
+        # If no software dependencies were passed then return
+        return @@software.uniq unless soft
+
+        # Find missing software
+        missing = []
+        soft.each do |app|
+          %x{which #{app}}
+          missing << app unless $?.success?
+        end
+
+        # Raise if there were any missing software's
+        unless missing.empty?
+          missing = missing.join(', ')
+          raise CliTool::MissingDependencies,
+            %{The required software packages "#{missing}" could not be found in your $PATH.}
+        end
+
+        # Append to the software list
+        @@software = @@software.concat(soft).uniq
+      end
+    end
+  end
+end

data/lib/cli_tool/version.rb

@@ -1,3 +1,3 @@
 module CliTool
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/cli_tool.rb

@@ -2,6 +2,7 @@ require "cli_tool/version"
 require "cli_tool/stdin_out"
 require "cli_tool/option_parser"
 require "cli_tool/software_dependencies"
+require "cli_tool/remote"
 
 module CliTool
   # Your code goes here...

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cli_tool
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Kelly Becker
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-10-24 00:00:00.000000000 Z
+date: 2013-10-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -71,6 +71,7 @@ files:
 - cli_tool.gemspec
 - lib/cli_tool.rb
 - lib/cli_tool/option_parser.rb
+- lib/cli_tool/remote.rb
 - lib/cli_tool/software_dependencies.rb
 - lib/cli_tool/stdin_out.rb
 - lib/cli_tool/version.rb