awt 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ba28acfb7c2b085b5932197f7a2145c82bb74085
-  data.tar.gz: fb4d005cf06fda90dc483f721fa6d8de189c7126
+  metadata.gz: 8e3158b8b1980b0db7939acfe0af10e5c4f6940b
+  data.tar.gz: c17259140354d4debd4ede725e46c8f4281bb2ff
 SHA512:
-  metadata.gz: 82cfc32ee11022ee12b2699091c821ffedc00e5bee4e976ad70a50f8b6d858699d66258e1fffa137107be491664a4f114913200c0201907fb2aa8d7279d341fe
-  data.tar.gz: b6b16cfaff66874e4769e4c790408f33f2a3504bf08813ed423ed5f01cfc8f0531c55dba80694fc1a0432465ab7f13d8c23e3b598751bdb521c50584b7c57deb
+  metadata.gz: 7abeaab52b73ce75ff0884a8d83139f86e862b98f2ed593cfa25fe17dcf98b3a1f963b4914440b8d6d19348a4cc3981d933a4be0d8f09fe82730dc0130dc60c4
+  data.tar.gz: 100d3833184ce553f7378ab6a63ae4bf1bbdee267fe78a47176738f996afb3633ddee07a1708391dede024816d319a2582ae4e8f367934232265a4730949c66a

data/README.md

@@ -1,5 +1,6 @@
 # Awt
 
+[![Gem Version](https://badge.fury.io/rb/awt.png)](http://badge.fury.io/rb/awt)
 [![Build Status](https://travis-ci.org/i2bskn/awt.png)](https://travis-ci.org/i2bskn/awt)
 [![Coverage Status](https://coveralls.io/repos/i2bskn/awt/badge.png)](https://coveralls.io/r/i2bskn/awt)
 [![Code Climate](https://codeclimate.com/github/i2bskn/awt.png)](https://codeclimate.com/github/i2bskn/awt)
@@ -43,6 +44,8 @@ server "hostname", user: "awt", port: 22, key: "/path/to/id_rsa"
 
 task :task_name do
   run "do something"
+  put "/path/to/local/file", "/path/to/remote/file"
+  get "/path/to/remote/file", "/path/to/local/file"
 end
 ```
 

data/awt.gemspec

@@ -19,6 +19,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_dependency "net-ssh"
+  spec.add_dependency "net-scp"
 
   spec.add_development_dependency "bundler", "~> 1.3"
   spec.add_development_dependency "rake"

data/bin/awt

@@ -3,38 +3,6 @@
 path = File.expand_path("../../lib", __FILE__)
 $:.unshift(path) unless $:.include?(path)
 
-require "optparse"
+require "awt/cli"
 
-require "awt"
-require "awt/dsl"
-
-include Awt::DSL
-
-options = {}
-OptionParser.new do |opt|
-  opt.on("-H VAL") {|v| options[:hosts] = v.split(",")}
-  opt.on("-u VAL") {|v| options[:user] = v}
-  opt.on("-p VAL") {|v| options[:port] = v.to_i}
-  opt.on("-i VAL") {|v| options[:key] = v}
-  opt.on("-f VAL") {|v| options[:task_file] = v}
-
-  begin
-    opt.parse!(ARGV)
-  rescue => e
-    puts e
-    exit 1
-  end
-end
-
-task_name = ARGV.first.to_sym
-hosts = options.delete(:hosts) || []
-task_file = options.delete(:task_file) || task_find
-
-hosts.each do |host|
-  server host, options
-end
-
-unless task_file.nil?
-  load task_file
-  $AWT_TASKS[task_name].exec($AWT_TARGETS)
-end
+Awt::CLI.start

data/lib/awt.rb

@@ -1,4 +1,5 @@
 require "net/ssh"
+require "net/scp"
 
 require "awt/printer"
 require "awt/server"

data/lib/awt/cli.rb

@@ -0,0 +1,60 @@
+require "optparse"
+
+require "awt"
+require "awt/dsl"
+
+include Awt::DSL
+
+module Awt
+  class CLI
+    attr_reader :executable
+
+    def initialize
+      parse_opt
+    end
+
+    def execute
+      register_targets(@hosts)
+      raise "Awtfile not found." if @task_file.nil?
+      load @task_file
+      $AWT_TASKS[@task_name].exec($AWT_TARGETS)
+    end
+
+    class << self
+      def start
+        begin
+          cli = self.new
+          cli.execute
+        rescue => e
+          puts e
+        end
+      end
+    end
+
+    private
+    def parse_opt
+      @options = {}
+      @hosts = []
+      @task_file = task_find
+      OptionParser.new do |opt|
+        opt.version = VERSION
+        opt.banner = "Usage: awt [options] TASK_NAME"
+        opt.on("-H HOSTNAME", "target host names.") {|v| @hosts = v.split(",")}
+        opt.on("-u USER", "login user.") {|v| @options[:user] = v}
+        opt.on("-p PORT", "ssh port number.") {|v| @options[:port] = v.to_i}
+        opt.on("-i IDENTITY_FILE", "SSH private key file.") {|v| @options[:key] = v}
+        opt.on("-f Awtfile", "Task file.") {|v| @task_file = v}
+        opt.parse!(ARGV)
+      end
+
+      raise "Task name is not specified." if ARGV.empty?
+      @task_name = ARGV.first.to_sym
+    end
+
+    def register_targets(hosts)
+      hosts.each do |host|
+        server host, @options
+      end
+    end
+  end
+end

data/lib/awt/server.rb

@@ -10,12 +10,33 @@ module Awt
     end
 
     def run(cmd)
-      @printer.host = @host if @printer.host != @host
+      reset_printer_host
       Net::SSH.start(@host, @user, @options) do |ssh|
         @printer.print_run cmd
         out = ssh.exec!(cmd)
         @printer.print_out out
       end
     end
+
+    def put(local, remote)
+      reset_printer_host
+      Net::SSH.start(@host, @user, @options) do |ssh|
+        @printer.print_upload local
+        ssh.scp.upload! local, remote
+      end
+    end
+
+    def get(remote, local)
+      reset_printer_host
+      Net::SSH.start(@host, @user, @options) do |ssh|
+        @printer.print_download remote
+        ssh.scp.download! remote, local
+      end
+    end
+
+    private
+    def reset_printer_host
+      @printer.host = @host if @printer.host != @host
+    end
   end
 end

data/lib/awt/version.rb

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

data/spec/awt/cli_spec.rb

@@ -0,0 +1,111 @@
+require "spec_helper"
+
+describe Awt::CLI do
+  let(:cli) {Awt::CLI.new}
+
+  before do
+    ARGV.clear
+    ["-H", "host1,host2", "-u", "awt", "-p", "30022", "-i", "/path/to/id_rsa", "-f", "Awtfile", "example"].each do |opt|
+      ARGV << opt
+    end
+  end
+
+  describe "#initialize" do
+    it "should call CLI#parse_opt" do
+      Awt::CLI.any_instance.should_receive(:parse_opt)
+      Awt::CLI.new
+    end
+  end
+
+  describe "#execute" do
+    before do
+      Awt::Server.any_instance.stub(:run)
+      Awt::CLI.any_instance.stub(:load)
+      cli.instance_eval do
+        task :example do
+          run "something"
+        end
+      end
+    end
+
+    after do
+      $AWT_TASKS.clear
+      $AWT_TARGETS.clear
+    end
+
+    it "should call #register_targets" do
+      Awt::CLI.any_instance.should_receive(:register_targets).with(["host1", "host2"])
+      cli.execute
+    end
+
+    it "should throw exception if @task_file is nil" do
+      cli.instance_eval {@task_file = nil}
+      expect {
+        cli.execute
+        }.to raise_error
+    end
+
+    it "should load task file" do
+      Awt::CLI.any_instance.should_receive(:load)
+      cli.execute
+    end
+
+    it "should call Awt::Task#exec" do
+      mock = double("Awt::Task mock").as_null_object
+      mock.should_receive(:exec)
+      $AWT_TASKS.store(:example, mock)
+      cli.execute
+    end
+  end
+
+  describe "#parse_opt" do
+    it "should set task_name" do
+      expect(cli.instance_eval{@task_name}).to eq(:example)
+    end
+
+    it "should set hosts" do
+      expect(cli.instance_eval{@hosts}).to eq(["host1", "host2"])
+    end
+
+    it "should set user" do
+      expect(cli.instance_eval{@options[:user]}).to eq("awt")
+    end
+
+    it "should set port" do
+      expect(cli.instance_eval{@options[:port]}).to eq(30022)
+    end
+
+    it "should set port" do
+      expect(cli.instance_eval{@options[:key]}).to eq("/path/to/id_rsa")
+    end
+
+    it "should set task file" do
+      expect(cli.instance_eval{@task_file}).to eq("Awtfile")
+    end
+  end
+
+  describe "#register_targets" do
+    before {$AWT_TARGETS.clear}
+
+    it "should add targets" do
+      cli.send(:register_targets, cli.instance_eval{@hosts})
+      expect($AWT_TARGETS.size).to eq(2)
+    end
+  end
+
+  describe ".#start" do
+    it "should call Awt::CLI#execute" do
+      Awt::CLI.any_instance.should_receive(:execute)
+      Awt::CLI.start
+    end
+
+    it "should print error message if raise error" do
+      Awt::CLI.any_instance.should_receive(:execute).and_raise("error")
+      expect(
+        capture(:stdout){
+          Awt::CLI.start
+        }.chomp
+      ).to eq("error")
+    end
+  end
+end

data/spec/awt/server_spec.rb

@@ -1,10 +1,14 @@
 require "spec_helper"
 
 describe Awt::Server do
+  let(:server) {Awt::Server.new}
+  let(:ssh) {double("Net::SSH mock").as_null_object}
+  let(:scp) {double("Net::SCP mock").as_null_object}
+  let(:local) {"/local/file"}
+  let(:remote) {"/remote/file"}
+
   describe "#initialize" do
     context "with default params" do
-      let(:server) {Awt::Server.new}
-
       it "should set default host" do
         expect(server.host).to eq("localhost")
       end
@@ -50,9 +54,6 @@ describe Awt::Server do
   end
 
   describe "#run" do
-    let(:server) {Awt::Server.new}
-    let(:ssh) {double("Net::SSH mock").as_null_object}
-
     before do
       Net::SSH.stub(:start).and_yield(ssh)
       Awt::Printer.any_instance.stub(:print_run)
@@ -66,13 +67,69 @@ describe Awt::Server do
       Awt::Printer.any_instance.should_receive(:print_out)
     end
 
-    it "should call Net::SSH::Connection::Session#exec" do
+    it "should call #exec! of Net::SSH" do
       ssh.should_receive(:exec!)
     end
 
-    it "should re-set host of Printer" do
+    it "should call #reset_printer_host" do
+      Awt::Server.any_instance.should_receive(:reset_printer_host)
+    end
+  end
+
+  describe "#put" do
+    before do
+      ssh.stub(:scp).and_return(scp)
+      Net::SSH.stub(:start).and_yield(ssh)
+      Awt::Printer.any_instance.stub(:print_upload)
+    end
+
+    after {server.put(local, remote)}
+
+    it "should call Awt::Printer#print_*" do
+      Awt::Printer.any_instance.should_receive(:print_upload)
+    end
+
+    it "should call #upload! of Net::SCP" do
+      scp.should_receive(:upload!)
+    end
+
+    it "should call #reset_printer_host" do
+      Awt::Server.any_instance.should_receive(:reset_printer_host)
+    end
+  end
+
+  describe "#get" do
+    before do
+      ssh.stub(:scp).and_return(scp)
+      Net::SSH.stub(:start).and_yield(ssh)
+      Awt::Printer.any_instance.stub(:print_download)
+    end
+
+    after {server.get(remote, local)}
+
+    it "should call Awt::Printer#print_*" do
+      Awt::Printer.any_instance.should_receive(:print_download)
+    end
+
+    it "should call #download! of Net::SCP" do
+      scp.should_receive(:download!)
+    end
+
+    it "should call #reset_printer_host" do
+      Awt::Server.any_instance.should_receive(:reset_printer_host)
+    end
+  end
+
+  describe "#reset_printer_host" do
+    after {server.send(:reset_printer_host)}
+
+    it "should re-set host of Printer if changed host" do
       Awt::Printer.any_instance.should_receive(:host=)
       server.host = "example.com"
     end
+
+    it "should not re-set host of Printer" do
+      Awt::Printer.any_instance.should_not_receive(:host=)
+    end
   end
 end

data/spec/spec_helper.rb

@@ -13,7 +13,7 @@ SimpleCov.start do
 end
 
 require "awt"
-require "awt/dsl"
+require "awt/cli"
 
 support_files = File.join(File.expand_path("..", __FILE__), "support/*.rb")
 Dir[support_files].each {|f| require f}

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: awt
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - i2bskn
@@ -24,6 +24,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-scp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -84,11 +98,13 @@ files:
 - awt.gemspec
 - bin/awt
 - lib/awt.rb
+- lib/awt/cli.rb
 - lib/awt/dsl.rb
 - lib/awt/printer.rb
 - lib/awt/server.rb
 - lib/awt/task.rb
 - lib/awt/version.rb
+- spec/awt/cli_spec.rb
 - spec/awt/dsl_spec.rb
 - spec/awt/printer_spec.rb
 - spec/awt/server_spec.rb
@@ -115,11 +131,12 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.1.9
+rubygems_version: 2.1.6
 signing_key: 
 specification_version: 4
 summary: Awt is cli tool for system administration.
 test_files:
+- spec/awt/cli_spec.rb
 - spec/awt/dsl_spec.rb
 - spec/awt/printer_spec.rb
 - spec/awt/server_spec.rb