hotdog 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: be590bcf10423710cc9b86db34d0aaec800d9eaf
-  data.tar.gz: fb1eaa5dd9f40e4ff4889f378e1093a647e0cb90
+  metadata.gz: 380b4a02d1dd654994f3f1ea4d4a69eef6712967
+  data.tar.gz: 4bbb5bb1974ec6cfa48e72780fb06e999935e65a
 SHA512:
-  metadata.gz: 75e2e9648fb2229292f79d70c4609ea7340637891b2a5c5db3cc637cf9e9a420f27dd2c940996cfd4ae2c22b638c2fbedf628a526e55efad474068e7ad34b5ff
-  data.tar.gz: 055408cd88e6f4bc745dd6ed56008107bad40a3bc7db6e8bab7698c0b2278a3b6ebb52b19546871ae05c1f2169fc8f6b72ccd5357d89cd37c6383e463b760322
+  metadata.gz: bd8ac3e16d6b817d81d8e10911e343657643acaf03ffebd9a7227e3d47a38c5d5dc8af433ffc6775cef7ae2a6867bd2f541ff29acbb8c341a8c7d74ca61ceed3
+  data.tar.gz: 8c011e9f95681fcc351365a3d8309ef509dc1329b6895754cbd5b8c12734e01e9b47b993c983d67e25e5bb481d6e55fc6a02d2546c35a5a44ebf54b785582de1

data/hotdog.gemspec

@@ -24,4 +24,5 @@ Gem::Specification.new do |spec|
   spec.add_dependency "dogapi", ">= 1.13.0"
   spec.add_dependency "parslet", "~> 1.6.2"
   spec.add_dependency "sqlite3", "~> 1.3.10"
+  spec.add_dependency "parallel", "~> 1.4.1"
 end

data/lib/hotdog/commands/pssh.rb

@@ -0,0 +1,105 @@
+#!/usr/bin/env ruby
+
+require "json"
+require "parslet"
+require "hotdog/commands/search"
+require "shellwords"
+require "parallel"
+
+module Hotdog
+  module Commands
+    class Pssh < Search
+      def run(args=[])
+        ssh_option = {
+          options: [],
+          user: nil,
+          port: nil,
+          identity_file: nil,
+          max_parallelism: nil,
+        }
+
+        optparse.on("-o SSH_OPTION", "Passes this string to ssh command through shell. This option may be given multiple times") do |option|
+          ssh_option[:options] += [option]
+        end
+        optparse.on("-i SSH_IDENTITY_FILE", "SSH identity file path") do |path|
+          ssh_option[:identity_file] = path
+        end
+        optparse.on("-p PORT", "Port of the remote host", Integer) do |port|
+          ssh_option[:port] = port
+        end
+        optparse.on("-u SSH_USER", "SSH login user name") do |user|
+          ssh_option[:user] = user
+        end
+        optparse.on("-P PARALLELISM", "Max parallelism", Integer) do |n|
+          ssh_option[:max_parallelism] = n
+        end
+
+        search_args = []
+        optparse.order!(args) {|search_arg| search_args.push(search_arg) }
+        expression = search_args.join(" ").strip
+        if expression.empty? || args.empty?
+          exit(1)
+        end
+
+        begin
+          node = parse(expression)
+        rescue Parslet::ParseFailed => error
+          STDERR.puts("syntax error: " + error.cause.ascii_tree)
+          exit(1)
+        end
+
+        result = evaluate(node, self).sort
+        result, fields = get_hosts(result)
+
+        if result.empty?
+          STDERR.puts("no match found: #{search_args.join(" ")}")
+          exit(1)
+        end
+
+        addresses = result.map {|host| [host.first, host.last] }
+
+        # build ssh command
+        cmdline = ["ssh"]
+        ssh_option[:options].each do |option|
+          cmdline << "-o" << option
+        end
+        if path = ssh_option[:identity_file]
+          cmdline << "-i" << Shellwords.escape(path)
+        end
+        if port = ssh_option[:port]
+          cmdline << "-p" << port.to_s
+        end
+        if ssh_option[:forward_agent]
+          cmdline << "-A"
+        end
+
+        cmdline << "-o" << "BatchMode=yes"
+
+        user = ssh_option[:user]
+
+        threads = ssh_option[:max_parallelism] || addresses.size
+        stats = Parallel.map(addresses, in_threads: threads) do |address,name|
+          c = cmdline.dup
+          if user
+            c << (user + "@" + address)
+          else
+            c << address
+          end
+
+          c.concat(args)
+
+          IO.popen([*c, in: :close, err: [:child, :out]]) do |io|
+            io.each_line {|line|
+              STDOUT.write "#{name}: #{line}"
+            }
+          end
+          $?.success?  # $? is thread-local variable
+        end
+
+        unless stats.all? {|success| success }
+          exit(1)
+        end
+      end
+    end
+  end
+end

data/lib/hotdog/version.rb

@@ -1,3 +1,3 @@
 module Hotdog
-  VERSION = "0.1.7"
+  VERSION = "0.1.8"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hotdog
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - Yamashita Yuu
@@ -80,6 +80,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.3.10
+- !ruby/object:Gem::Dependency
+  name: parallel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.4.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.4.1
 description: Yet another command-line tool for Datadog
 email:
 - peek824545201@gmail.com
@@ -101,6 +115,7 @@ files:
 - lib/hotdog/commands/help.rb
 - lib/hotdog/commands/hosts.rb
 - lib/hotdog/commands/ls.rb
+- lib/hotdog/commands/pssh.rb
 - lib/hotdog/commands/search.rb
 - lib/hotdog/commands/ssh.rb
 - lib/hotdog/commands/tags.rb