blower 4.7 → 6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cd4ef69511d05d0f1ee58cd814ddc58a85102c467fe695a0520cf02345dbad69
-  data.tar.gz: 82aeb0b7c6ead150c17d78f50513c300b5aa326cd7ccf4c1932eb84017e712d2
+  metadata.gz: b43c163f31a76c68c8327d1c696bda453f2c6fefcfc673fdcf338f8c840e3a6c
+  data.tar.gz: e33fde4f43223527937dd57d11da5609bb2f124a66dc21dfa71d2f79a119a765
 SHA512:
-  metadata.gz: e932789921c9553fbefd425a38db2d0d3db063681a660065ad9e642b0335d364e09a392fd34535fb64ba57d05d87da40f8a8f9420d7749f53a8ff5a8851efc79
-  data.tar.gz: 40bf21b8b30968869d6a3f425155d66a4cb562fc1f1241e40f14c5987ac1b670713e10a4bbc148373e2b3012c376ada391f6157da79df24e947790bf3aa767f1
+  metadata.gz: 11ee37db482dad0ea13f9cdf42cb29007d6018a995f404bc06fb27f971ba3c6c2073f0a44165d0790c75f8cd931d37cd197ffda7d59e2f43368d9ebad14957a1
+  data.tar.gz: '0860054ed01d01203c7a616ae03fbc0653e0e3519e3516786530a88163207aef206f3da6e4be1ddcf78642314ddbde62d4cd427c1aecbfa64e0b73dc9f7465a9'

data/bin/blow

@@ -2,6 +2,9 @@
 require 'blower'
 require 'pathname'
 require 'optparse'
+require 'dotenv'
+
+Dotenv.load
 
 path = []
 
@@ -30,6 +33,7 @@ if File.directory?(File.join(Dir.pwd, "lib"))
 end
 
 context.run "Blowfile", optional: true
+
 begin
   until ARGV.empty?
     context.run ARGV.shift

data/lib/blower.rb

@@ -8,5 +8,7 @@ require 'blower/logger'
 require 'blower/errors'
 require 'blower/context'
 require 'blower/host'
+require 'blower/local'
+require 'blower/target'
 require 'blower/plugin'
 require 'blower/version'

data/lib/blower/context.rb

@@ -1,3 +1,4 @@
+require 'colorize'
 require 'find'
 require 'erb'
 require 'json'
@@ -32,6 +33,9 @@ module Blower
     # The target hosts.
     attr_accessor :hosts
 
+    # The failed hosts.
+    attr_accessor :failures
+
     # Username override. If not-nil, this user is used for all remote accesses.
     attr_accessor :user
 
@@ -45,6 +49,7 @@ module Blower
       @path = path
       @data = {}
       @hosts = []
+      @failures = []
     end
 
     # Return a context variable.
@@ -90,7 +95,7 @@ module Blower
     def on (*hosts, quiet: false)
       let :@hosts => hosts.flatten do
         log.info "on #{@hosts.map(&:name).join(", ")}", quiet: quiet do
-          yield
+          yield *hosts
         end
       end
     end
@@ -115,12 +120,20 @@ module Blower
     # @raise Whatever the task itself raises.
     # @return The result of evaluating the task file.
     def run (task, optional: false, quiet: false, once: nil)
+      @run_cache ||= {}
       once once, quiet: quiet do
         log.info "run #{task}", quiet: quiet do
           file = find_task(task)
-          code = File.read(file)
-          let :@file => file do
-            instance_eval(code, file)
+          if @run_cache.has_key? file
+            log.info "*cached*"
+            @run_cache[file]
+          else
+            @run_cache[file] = begin
+              code = File.read(file)
+              let :@file => file do
+                instance_eval(code, file)
+              end
+            end
           end
         end
       end
@@ -129,6 +142,12 @@ module Blower
       raise e
     end
 
+    def sh? (command, as: user, on: hosts, quiet: false, once: nil)
+      sh command, as: as, on: on, quiet: quiet, once: once
+    rescue
+      nil
+    end
+
     # Execute a shell command on each host
     # @macro onable
     # @macro asable
@@ -136,7 +155,7 @@ module Blower
     # @macro onceable
     def sh (command, as: user, on: hosts, quiet: false, once: nil)
       self.once once, quiet: quiet do
-        log.info "sh #{command}", quiet: quiet do
+        log.info "sh: #{command}", quiet: quiet do
           hash_map(hosts) do |host|
             host.sh command, as: as, quiet: quiet
           end
@@ -282,7 +301,7 @@ module Blower
 
       def to_s
         map do |host, data|
-          "#{host.name.blue} (#{host.address.green})\n" + data.strip.to_s.gsub(/^/, "  ")
+          "#{host.name.blue}\n" + data.strip.to_s.gsub(/^/, "  ")
         end.join("\n")
       end
 
@@ -302,40 +321,80 @@ module Blower
     end
 
     def hash_map (hosts = self.hosts)
-      HostHash.new.tap do |result|
-        each(hosts) do |host|
-          result[host] = yield(host)
+      hh = HostHash.new.tap do |result|
+        each(hosts) do |host, i|
+          result[host] = yield(host, i)
+        end
+      end
+      if @singular
+        hh.values.first
+      else
+        hh
+      end
+    end
+
+    def singularly (flag = true)
+      was, @singular = @singular, flag
+      yield
+    ensure
+      @singular = was
+    end
+
+    def on_one (host = self.hosts.sample, serial: true)
+      ret = nil
+      each([host], serial: serial) do |host, i|
+        on host do
+          singularly do
+            ret = yield host, i
+          end
         end
       end
+      ret
     end
 
     def on_each (hosts = self.hosts, serial: true)
-      each(hosts, serial: serial) do |host|
+      each(hosts.dup, serial: serial) do |host, i|
         on host do
-          yield host
+          singularly do
+            yield host, i
+          end
         end
       end
     end
 
+    def locally (&block)
+      on Local.new("<local>") do
+        singularly &block
+      end
+    end
+
     def each (hosts = self.hosts, serial: false)
       fail "No hosts" if hosts.empty?
-      if false
-        [hosts].flatten.each do |host|
-          yield host
-        end
-      else
-        threads = [hosts].flatten.map do |host|
-          Thread.new do
-            begin
-              yield host
-            rescue => e
-              host.log.error e.full_message
-              hosts.delete host
-            end
+      q = (@threads || serial) && Queue.new
+      if q && serial
+        q.push nil
+      elsif q
+        @threads.times { q.push nil }
+      end
+      indent = Thread.current[:indent]
+      i = -1
+      threads = [hosts].flatten.map.with_index do |host|
+        Thread.new do
+          Thread.current[:indent] = indent
+          begin
+            q.pop if q
+            yield host, i += 1
+          rescue => e
+            host.log.error e.message
+            hosts.delete host
+            @failures |= [host]
+          ensure
+            q.push nil if q
+            sleep @delay if @delay
           end
         end
-        threads.each(&:join)
       end
+      threads.each(&:join)
       fail "No hosts remaining" if hosts.empty?
     end
 

data/lib/blower/host.rb

@@ -2,7 +2,6 @@ require 'net/ssh'
 require 'net/ssh/gateway'
 require 'net/scp'
 require 'monitor'
-require 'colorize'
 require 'base64'
 require 'timeout'
 
@@ -56,6 +55,7 @@ module Blower
     # Copy files or directories to the host.
     # @api private
     def cp (froms, to, as: nil, quiet: false, delete: false)
+      sleep DELAY if defined?(DELAY)
       as ||= @user
       output = ""
       synchronize do
@@ -111,6 +111,7 @@ module Blower
     # Execute a command on the host and return its output.
     # @api private
     def sh (command, as: nil, quiet: false)
+      sleep DELAY if defined?(DELAY)
       as ||= @user
       output = ""
       synchronize do
@@ -127,7 +128,7 @@ module Blower
               end
               ch.on_extended_data do |_, _, data|
                 log.trace "received #{data.bytesize} bytes stderr", quiet: quiet
-                output << data.colorize(:red)
+                output << data
               end
               ch.on_request("exit-status") do |_, data|
                 result = data.read_long

data/lib/blower/local.rb

@@ -0,0 +1,48 @@
+require 'net/ssh'
+require 'net/ssh/gateway'
+require 'net/scp'
+require 'monitor'
+require 'base64'
+require 'timeout'
+
+module Blower
+
+  class Local
+    include MonitorMixin
+    extend Forwardable
+
+    attr_reader :name, :data
+
+    def_delegators :data, :[], :[]=
+
+    def initialize (name, proxy: nil)
+      @name, @proxy = name, proxy
+      @data = {}
+    end
+
+    # Represent the host as a string.
+    def to_s
+      @name
+    end
+
+    def sh (command, as: nil, quiet: false)
+      command = "#{@proxy} #{command.shellescape}" if @proxy
+      result = IO.popen(command) do |io|
+        io.read
+      end
+      if $?.success?
+        result
+      else
+        raise "Command failed"
+      end
+    end
+
+    # Produce a Logger prefixed with the host name.
+    # @api private
+    def log
+      @log ||= Logger.instance.with_prefix("on #{name}: ")
+    end
+
+  end
+
+end

data/lib/blower/logger.rb

@@ -1,3 +1,4 @@
+require 'colorize'
 require "singleton"
 
 module Blower
@@ -33,12 +34,11 @@ module Blower
       attr_accessor :indent
     end
 
-    self.indent = 0
-
     self.level = :info
 
     def initialize (prefix = "")
       @prefix = prefix
+      thread[:indent] = 0
       super()
     end
 
@@ -49,10 +49,10 @@ module Blower
 
     # Yield with a temporarily incremented indent counter
     def with_indent ()
-      Logger.indent += 1
+      thread[:indent] += 1
       yield
     ensure
-      Logger.indent -= 1
+      thread[:indent] -= 1
     end
 
     # Display a log message. The block, if specified, is executed in an indented region after the log message is shown.
@@ -64,9 +64,10 @@ module Blower
     def log (level, message, quiet: false, &block)
       if !quiet && (LEVELS.index(level) >= LEVELS.index(Logger.level))
         synchronize do
+          message = message.to_s.colorize(COLORS[level]) if level
           message = message.to_s.colorize(COLORS[level]) if level
           message.split("\n").each do |line|
-            STDERR.puts "  " * Logger.indent + @prefix + line
+            STDERR.puts "  " * thread[:indent] + @prefix + line
           end
         end
         with_indent(&block) if block
@@ -95,6 +96,10 @@ module Blower
     define_helper :error
     define_helper :fatal
 
+    def thread
+      Thread.current
+    end
+
   end
 
 end

data/lib/blower/target.rb

@@ -0,0 +1,106 @@
+require 'monitor'
+require 'base64'
+require 'timeout'
+require 'open3'
+
+module Blower
+
+  class Target
+    include MonitorMixin
+    extend Forwardable
+
+    attr_reader :name, :data
+
+    def_delegators :data, :[], :[]=
+
+    def initialize (name, ssh: "ssh", scp: "scp")
+      @name, @ssh, @scp = name, ssh, scp
+      @data = {}
+      super()
+    end
+
+    # Represent the host as a string.
+    def to_s
+      @name
+    end
+
+    # Copy files or directories to the host.
+    # @api private
+    def cp (froms, to, as: nil, quiet: false, delete: false)
+      as ||= @user
+      output = ""
+      synchronize do
+        [froms].flatten.each do |from|
+          if from.is_a?(String)
+            to += "/" if to[-1] != "/" && from.is_a?(Array)
+            command = ["rsync", "-e", @ssh, "-r"]
+            if File.exist?(".blowignore")
+              command += ["--exclude-from", ".blowignore"]
+            end
+            command += ["--delete"] if delete
+            command += [*from, ":#{to}"]
+            log.trace command.shelljoin, quiet: quiet
+            IO.popen(command, in: :close, err: %i(child out)) do |io|
+              until io.eof?
+                begin
+                  output << io.read_nonblock(100)
+                rescue IO::WaitReadable
+                  IO.select([io])
+                  retry
+                end
+              end
+              io.close
+              if !$?.success?
+                log.fatal "exit status #{$?.exitstatus}: #{command}", quiet: quiet
+                log.fatal output, quiet: quiet
+                fail "failed to copy files"
+              end
+            end
+          elsif from.respond_to?(:read)
+            cmd = "echo #{Base64.strict_encode64(from.read).shellescape} | base64 -d > #{to.shellescape}"
+            sh cmd, quiet: quiet
+          else
+            fail "Don't know how to copy a #{from.class}: #{from}"
+          end
+        end
+      end
+      true
+    end
+
+    def sh (command, as: nil, quiet: false)
+      marker, output = SecureRandom.hex(32), nil
+      ssh do |i, o, _|
+        i.puts "echo #{marker}"
+        i.puts "sh -c #{command.shellescape} 2>&1"
+        i.puts "STATUS_#{marker}=$?"
+        i.puts "echo #{marker}"
+        i.flush
+        o.readline("#{marker}\n")
+        output = o.readline("#{marker}\n")[0..-(marker.length + 2)]
+        i.puts "echo $STATUS_#{marker}"
+        status = o.readline.to_i
+        if status != 0
+          fail FailedCommand, output
+        end
+        output
+      end
+    end
+
+    # Produce a Logger prefixed with the host name.
+    # @api private
+    def log
+      @log ||= Logger.instance.with_prefix("on #{name}: ")
+    end
+
+    private
+
+    def ssh
+      unless @wait
+        @stdin, @stdout, @stderr, @wait = Open3.popen3(@ssh)
+      end
+      yield @stdin, @stdout, @stderr
+    end
+
+  end
+
+end

data/lib/blower/util.rb

@@ -26,3 +26,9 @@ class Object
   end
 
 end
+
+def try (err = nil, &block)
+  block.()
+rescue
+  err
+end

data/lib/blower/version.rb

@@ -1,3 +1,3 @@
 module Blower
-  VERSION = "4.7"
+  VERSION = "6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: blower
 version: !ruby/object:Gem::Version
-  version: '4.7'
+  version: '6'
 platform: ruby
 authors:
 - Nathan Baum
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2018-10-27 00:00:00.000000000 Z
+date: 2021-06-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: net-ssh
@@ -94,6 +94,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: ed25519
+  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: dotenv
+  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: yard
   requirement: !ruby/object:Gem::Requirement
@@ -120,15 +148,17 @@ files:
 - lib/blower/context.rb
 - lib/blower/errors.rb
 - lib/blower/host.rb
+- lib/blower/local.rb
 - lib/blower/logger.rb
 - lib/blower/plugin.rb
+- lib/blower/target.rb
 - lib/blower/util.rb
 - lib/blower/version.rb
 homepage: http://www.github.org/nbaum/blower
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -143,9 +173,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.7
-signing_key: 
+rubygems_version: 3.2.3
+signing_key:
 specification_version: 4
 summary: Really simple server orchestration
 test_files: []