anywhere 0.0.1 → 0.0.2

data/README.md

@@ -34,6 +34,11 @@ Or install it yourself as:
     => [<run_time=0.06875, cmd=<uptime>, stdout=<1 lines, 72 chars>, stderr=<empty>, exit_status=0>,
      <run_time=0.067885, cmd=<uptime>, stdout=<1 lines, 72 chars>, stderr=<empty>, exit_status=0>]
 
+## Running the specs
+
+For the ssh integration rest you need to have some host (could be your local vagrant) you can access as root. This host must be
+set in the `SSH_TEST_HOST` instance variable.
+
 ## Contributing
 
 1. Fork it

data/Rakefile

@@ -1 +1,7 @@
 require "bundler/gem_tasks"
+
+task :version_commit do
+  require File.expand_path("../lib/anywhere/version.rb", __FILE__)
+  cmd = "git add lib/anywhere/version.rb && git commit -m 'version #{Anywhere::VERSION}'"
+  system cmd
+end

data/anywhere.gemspec

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

data/lib/anywhere/base.rb

@@ -13,12 +13,38 @@ module Anywhere
       whoami == "root"
     end
 
-    def execute(cmd, stdin = nil)
-      do_execute(cmd, stdin) do |stream, data|
+    def capture(path, compressed = false)
+      io = StringIO.new
+      cmd = "cat #{path}"
+      cmd << " | gzip " if compressed
+      execute(cmd, nil, io)
+      io.rewind
+      out = io.read
+      if compressed
+        # not sure why this is not working with the uncompress method
+        out, s = Open3.capture2("cat - | gunzip", stdin_data: out)
+        out
+      else
+        out
+      end
+    end
+
+    def uncompress(string)
+      require "zlib"
+      require "base64"
+      zstream = Zlib::Inflate.new
+      buf = zstream.inflate(string)
+      zstream.finish
+      zstream.close
+      buf
+    end
+
+    def execute(cmd, stdin = nil, stdout = nil)
+      do_execute(cmd, stdin, stdout) do |stream, data|
         data.split("\n").each do |line|
           if stream == :stderr
             logger.error line
-          else
+          elsif !stdout
             logger.debug line
           end
         end
@@ -77,7 +103,7 @@ module Anywhere
         end
       end
       logger.info "writing #{content.length} bytes to #{path} (md5: #{md5})"
-      tmp_path = "/opt/urknall/tmp/files.#{md5}"
+      tmp_path = "/tmp/anywhere/files.#{md5}"
       execute("#{sudo_cmd} mkdir -p #{File.dirname(tmp_path)}")
       execute("#{sudo_cmd} chown #{whoami} #{File.dirname(tmp_path)}")
       logger.debug "writing to #{tmp_path}"
@@ -94,7 +120,7 @@ module Anywhere
       logger.debug "moving #{tmp_path} to #{path}"
       if file_exists?(path)
         logger.debug "diff #{path} #{tmp_path}"
-        execute("diff #{tmp_path} #{path}").stdout.split("\n").each do |line|
+        execute("diff #{tmp_path} #{path}; /bin/true").stdout.split("\n").each do |line|
           logger.debug line
         end
       end

data/lib/anywhere/local.rb

@@ -5,7 +5,7 @@ require "anywhere/base"
 
 module Anywhere
   class Local < Base
-    def do_execute(cmd, stdin_data = nil)
+    def do_execute(cmd, stdin_data = nil, stdout_io = nil)
       require "open3"
       result = Result.new(cmd)
       result.started!
@@ -19,10 +19,14 @@ module Anywhere
             stream.each do |line|
               if stream == stdout
                 result.add_stdout(line.strip)
-                logger.info(line) if logger
+                if stdout_io
+                  stdout_io << line
+                else
+                  logger.debug(line)
+                end
               elsif stream == stderr
                 result.add_stderr(line.strip)
-                logger.error(line) if logger
+                logger.error(line)
               end
             end
           end

data/lib/anywhere/logger.rb

@@ -1,9 +1,27 @@
-require "colorize"
+require "rainbow"
 require "time"
+require "logger"
 
 module Anywhere
   class Logger
+    LEVELS = {
+      DEBUG:    0,
+      INFO:     1,
+      WARN:     2,
+      ERROR:    3,
+      FATAL:    4,
+      UNKNOWN:  5
+    }
+
+    COLORS = {
+      0 => "#6d6d6d", # 109, 109, 109
+      1 => "#227000", # 34, 112
+      2 => "#f79700", # 247, 151
+      3 => "#f00000", # 240, 0
+    }
+
     attr_accessor :prefix
+    attr_writer :log_level
 
     class << self
       def mutex
@@ -11,6 +29,10 @@ module Anywhere
       end
     end
 
+    def log_level
+      @log_level ||= ::Logger::INFO
+    end
+
     def initialize(attributes = {})
       @attributes = attributes
     end
@@ -23,16 +45,15 @@ module Anywhere
       @attributes[:stream] ||= STDOUT
     end
 
-    def info(message)
-      print_with_prefix "INFO ".green + " #{message}"
-    end
-
-    def error(message)
-      print_with_prefix "ERROR".red + " #{message}"
-    end
-
-    def debug(message)
-      print_with_prefix "DEBUG".blue + " #{message}"
+    LEVELS.each do |name, l|
+      define_method(name.downcase) do |message|
+        return if log_level > l
+        prefix = "%05s" % [name.to_s]
+        if color = COLORS[l]
+          prefix = prefix.color(color)
+        end
+        print_with_prefix "%s %s" % [prefix, message]
+      end
     end
 
     def print_with_prefix(message)

data/lib/anywhere/ssh.rb

@@ -14,7 +14,7 @@ module Anywhere
       @attributes = attributes.merge(paranoid: false)
     end
 
-    def do_execute(cmd, stdin = nil)
+    def do_execute(cmd, stdin = nil, stdout = nil)
       result = Result.new(cmd)
       result.started!
       ssh.open_channel do |ch|
@@ -22,7 +22,11 @@ module Anywhere
           raise "could not execute command" unless success
           ch.on_data do |ch, data|
             result.add_stdout data
-            yield(:stdout, data) if block_given?
+            if stdout
+              stdout << data
+            elsif block_given?
+              yield(:stdout, data)
+            end
           end
 
           ch.on_extended_data do |ch, type, data|

data/lib/anywhere/version.rb

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

data/spec/integration/ssh_integration_spec.rb

@@ -16,8 +16,9 @@ describe "SSH integration spec" do
   end
 
   before :all do
-    @user = "vagrant"
-    host = "192.168.34.12"
+    @user = "root"
+    host = ENV["SSH_TEST_HOST"]
+    raise %(ENV["SSH_TEST_HOST"] must be set) if host.nil?
     port = 22
 
     @runner = Anywhere::SSH.new(host, @user, port: port)
@@ -32,27 +33,45 @@ describe "SSH integration spec" do
   end
 
   describe "#home_dir" do
-    it { runner.home_dir.should eq("/home/vagrant") }
+    it { runner.home_dir.should eq("/root") }
   end
 
   describe "#extract_tar" do
     let(:path) { FIXTURES_PATH.join("arch.tgz") }
 
     it "extracts the provided path" do
-      runner.extract_tar(path, "/tmp/urknall/tar")
-      runner.execute("ls /tmp/urknall/tar").stdout.split("\n").should eq(["a.txt", "b.txt"])
-      runner.execute("cat /tmp/urknall/tar/a.txt").stdout.should eq("this is a\n")
+      runner.extract_tar(path, "/tmp/anywhere/tar")
+      runner.execute("ls /tmp/anywhere/tar").stdout.split("\n").should eq(["a.txt", "b.txt"])
+      runner.execute("cat /tmp/anywhere/tar/a.txt").stdout.should eq("this is a\n")
     end
   end
 
   describe "#mkdir_p" do
     before :each do
-      @runner.execute("rm -Rf /tmp/urknall")
+      @runner.execute("rm -Rf /tmp/anywhere")
     end
 
     it "should create a new directory" do
-      @runner.mkdir_p("mkdir -p /tmp/urknall/test")
-      @runner.execute("file /tmp/urknall/test").stdout.should include("/tmp/urknall/test: directory")
+      @runner.mkdir_p("mkdir -p /tmp/anywhere/test")
+      @runner.execute("file /tmp/anywhere/test").stdout.should include("/tmp/anywhere/test: directory")
+    end
+  end
+
+  describe "#capture" do
+    subject(:str) do
+      runner.capture("/etc/passwd")
+    end
+
+    it { should be_kind_of(String) }
+    it { should start_with("root") }
+
+    describe "compressed" do
+      subject(:str) do
+        runner.capture("/etc/passwd", compressed = true)
+      end
+
+      it { should be_kind_of(String) }
+      it { should start_with("root") }
     end
   end
 
@@ -63,6 +82,18 @@ describe "SSH integration spec" do
     it { subject.stdout.should include("load average") }
     it { subject.stderr.length.should eq(0) }
 
+    describe "capturing" do
+      subject(:str) do
+        io = StringIO.new
+        runner.execute("cat /etc/passwd", nil, io)
+        io.rewind
+        io.read
+      end
+
+      it { str.should be_kind_of(String) }
+      it { str.length.should_not eq(0) }
+    end
+
     describe "with unknwon command" do
       subject(:error) do
         begin
@@ -105,6 +136,12 @@ describe "SSH integration spec" do
     subject { runner.write_file(@path, "hello world") }
     it { should be_success }
     it { runner.execute("cat #{@path}").stdout.should eq("hello world") }
+
+    it "should overwrite files" do
+      runner.write_file(@path, "hello world")
+      runner.write_file(@path, "hello world2")
+      runner.capture(@path).should eq("hello world2")
+    end
   end
 
   describe "write big files" do

data/spec/lib/anywhere/local_spec.rb

@@ -2,11 +2,12 @@ require "spec_helper"
 require "anywhere/local"
 
 describe "Anywhere::Local" do
-  before do
-    Anywhere.stub(:logger) { double("logger", error: true, info: true) }
+  subject(:local) do
+    local = Anywhere::Local.new
+    local.logger.stub(:stream) { StringIO.new }
+    local
   end
 
-  subject(:local) { Anywhere::Local.new }
   it { should_not be_nil }
 
   describe "whoami" do
@@ -21,6 +22,19 @@ describe "Anywhere::Local" do
     it { subject.exit_status.should eq(0) }
     it { subject.stdout.should be_kind_of(String) }
 
+    describe "#capture" do
+      subject(:str) do
+        io = StringIO.new
+        local.logger.unstub(:stream) { StringIO.new }
+        local.execute("cat anywhere.gemspec", nil, io)
+        io.rewind
+        io.read
+      end
+
+      it { str.should be_kind_of(String) }
+      it { str.length.should_not eq(0) }
+    end
+
     describe "with command raising an error" do
       subject(:error) do
         caught_err = nil

data/spec/lib/anywhere/logger_spec.rb

@@ -6,11 +6,44 @@ describe "Anywhere::Logger" do
   it { should_not be_nil }
   it { subject.prefix.should be_nil }
 
+  it "sets the default log level to INFO" do
+    logger.log_level.should eq(Logger::INFO)
+  end
+
   describe "setting the prefix" do
     it "changes the prefix" do
       subject.prefix = "test"
       subject.prefix.should eq("test")
     end
   end
+
+  describe "log levels" do
+    let(:stringio) { StringIO.new }
+    let(:log_level) { Logger::DEBUG }
+
+    subject(:lines) do
+      logger.log_level = log_level
+      logger.stub(:stream) { stringio }
+      logger.info "log info"
+      logger.debug "log debug"
+      stringio.rewind
+      stringio.read.split("\n")
+    end
+
+    it { should be_kind_of(Array) }
+    it { subject.count.should eq(2) }
+    it { subject.at(0).should include("INFO") }
+    it { subject.at(1).should include("DEBUG") }
+
+    describe "with log level being info" do
+      let(:log_level) { Logger::INFO }
+
+      it { subject.count.should eq(1) }
+      it { subject.at(0).should include("INFO") }
+    end
+
+    describe "" do
+    end
+  end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: anywhere
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-06-10 00:00:00.000000000 Z
+date: 2013-06-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: net-ssh
@@ -28,7 +28,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: colorize
+  name: rainbow
   requirement: !ruby/object:Gem::Requirement
     none: false
     requirements: