awt 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8e3158b8b1980b0db7939acfe0af10e5c4f6940b
-  data.tar.gz: c17259140354d4debd4ede725e46c8f4281bb2ff
+  metadata.gz: c1814106982fdedf5ba321096decea1ea902bec0
+  data.tar.gz: 4ef116fdbc3c99b4fd309fe35d1a265c04a71131
 SHA512:
-  metadata.gz: 7abeaab52b73ce75ff0884a8d83139f86e862b98f2ed593cfa25fe17dcf98b3a1f963b4914440b8d6d19348a4cc3981d933a4be0d8f09fe82730dc0130dc60c4
-  data.tar.gz: 100d3833184ce553f7378ab6a63ae4bf1bbdee267fe78a47176738f996afb3633ddee07a1708391dede024816d319a2582ae4e8f367934232265a4730949c66a
+  metadata.gz: ee9dc653d1a9ac8feafa9ea221a46a533dffb28a1991421a2f9e774817750dcad133e0285b38e9002e663e1488b0947c8c6784751d3e14bf8fe0a50ad3216135
+  data.tar.gz: f2b372d74d5df8c0695d30e12d12a8e6b126550c10629bcf05c73f565810784a6cffcd40755334ebf0171b10a966a47f3315abe05490a4efc413812d226afaad

data/README.md

@@ -49,6 +49,28 @@ task :task_name do
 end
 ```
 
+`run` is returns a result object.  
+Result object with a exit-status, etc.
+
+```ruby
+if run("cat /etc/centos-release").status == 0
+  puts "Hello CentOS"
+end
+
+if run("uname -a").data.chomp == "Linux"
+  puts "Hello Linux"
+end
+```
+
+Using the `with_env` method if you want to use the environment variable.  
+Key is applied upcase it is converted to String.
+
+```ruby
+with_env key: "value" do
+  run "echo $KEY"
+end
+```
+
 #### Execute
 
 Awt task executable as follows:

data/lib/awt.rb

@@ -1,3 +1,4 @@
+require "ostruct"
 require "net/ssh"
 require "net/scp"
 

data/lib/awt/server.rb

@@ -1,6 +1,6 @@
 module Awt
   class Server
-    attr_accessor :host, :user, :options
+    attr_accessor :host, :user, :options, :envs
 
     def initialize(host: "localhost", port: 22, user: nil, key: "~/.ssh/id_rsa")
       @host = host
@@ -9,13 +9,34 @@ module Awt
       @printer = Printer.new(@host)
     end
 
+    def with_env(envs, &block)
+      env = []
+      envs.each {|k,v| env << "export #{k.to_s.upcase}=\"#{v}\""}
+      @envs = env.join(";") unless env.empty?
+      self.instance_eval &block
+      @envs = nil
+    end
+
     def run(cmd)
       reset_printer_host
+      cmd = "#{@envs};#{cmd}" if @envs
+      result = OpenStruct.new
+
       Net::SSH.start(@host, @user, @options) do |ssh|
         @printer.print_run cmd
-        out = ssh.exec!(cmd)
-        @printer.print_out out
+        ssh.open_channel do |channel|
+          channel.exec cmd do |ch, success|
+            channel.on_data {|ch,data| result.data = data}
+            channel.on_extended_data {|ch,type,data| result.extended_data = data}
+            channel.on_request("exit-status") {|ch,data| result.status = data.read_long}
+          end
+        end
+        ssh.loop
       end
+
+      @printer.print_out result.data if result.data
+      @printer.print_ext result.extended_data if result.extended_data
+      result
     end
 
     def put(local, remote)

data/lib/awt/version.rb

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

data/spec/awt/server_spec.rb

@@ -53,22 +53,47 @@ describe Awt::Server do
     end
   end
 
+  describe "#with_env" do
+    it "should call instance methods" do
+      server.should_receive(:run)
+      server.with_env(key: :value) {run "example"}
+    end
+  end
+
   describe "#run" do
+    let(:channel) {double.as_null_object}
+
     before do
+      channel.stub(:exec).with("example").and_yield(double, double)
+      ssh.stub(:open_channel).and_yield(channel)
       Net::SSH.stub(:start).and_yield(ssh)
       Awt::Printer.any_instance.stub(:print_run)
       Awt::Printer.any_instance.stub(:print_out)
+      Awt::Printer.any_instance.stub(:print_ext)
     end
 
     after {server.run("example")}
 
     it "should call Awt::Printer#print_*" do
       Awt::Printer.any_instance.should_receive(:print_run)
+    end
+
+    it "should call #on_data" do
+      channel.should_receive(:on_data).and_yield(double, double)
       Awt::Printer.any_instance.should_receive(:print_out)
     end
 
-    it "should call #exec! of Net::SSH" do
-      ssh.should_receive(:exec!)
+    it "should call #on_extended_data" do
+      channel.should_receive(:on_extended_data).and_yield(double, double, double)
+      Awt::Printer.any_instance.should_receive(:print_ext)
+    end
+
+    it "should call #on_request" do
+      channel.should_receive(:on_request).with("exit-status").and_yield(double, double.as_null_object)
+    end
+
+    it "should call #loop" do
+      ssh.should_receive(:loop)
     end
 
     it "should call #reset_printer_host" do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: awt
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - i2bskn
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-11-18 00:00:00.000000000 Z
+date: 2013-11-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: net-ssh
@@ -131,7 +131,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.1.6
+rubygems_version: 2.1.9
 signing_key: 
 specification_version: 4
 summary: Awt is cli tool for system administration.