knife-node-context-exec 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d99cf4455cd6c9ec2db19c61067a19e6c2bfca8c
+  data.tar.gz: b91ea5849c60d64f125375feaa9a80fdb4403483
+SHA512:
+  metadata.gz: f2b39683e071789698cd3d64c8ceb8f8e9f9a01fd5da01d0fbaac5c67b23d8bf0f6335404a77e2acd53b2aad4c74ba8d0e827dc159c81d32fedf3c97181896d7
+  data.tar.gz: 8b1edd59577af9ed084edf6f86a0e75514355bfc4c5ec21f9f7b20c8230ea3e59ae2086442bf23625ff4159c36636ea92e145b07ec1a2a4eae86e6df86cc2621

data/lib/chef/knife/context_exec.rb

@@ -0,0 +1,52 @@
+# The knife
+class ContextExec < Chef::Knife
+  banner 'knife context exec'
+
+  deps do
+    require 'chef/search/query'
+    require 'knife-node-context-exec/runner'
+  end
+
+  option :environment,
+         short: '-E',
+         long: '--environment',
+         description: 'The environment to search.'
+  option :node_query,
+         short: '-Q',
+         long: '--query',
+         description: 'The node query.'
+  option :directory,
+         short: '-D',
+         long: '--directory',
+         description: 'A directory for working files.'
+  option :template_filename,
+         short: '-T',
+         long: '--template',
+         description: 'The filename of a template.'
+  option :script_filename,
+         short: '-S',
+         long: '--script',
+         description: 'The environment to search.'
+  option :command,
+         short: '-C',
+         long: '--command',
+         description: 'The command to run.'
+  option :filter_regex,
+         short: '-R',
+         long: '--regex',
+         description: 'A regex used to filter output.'
+  option :parallel,
+         short: '-P',
+         long: '--parallel',
+         description: 'Run in parallel?'
+
+  def run
+    environment = config[:environment].to_s
+    nodes = Chef::Search::Query.new.search(:node, config[:node_query].to_s).first.select do |node|
+      node.environment == environment
+    end
+    KnifeNodeContextExec.run(nodes, config[:directory].to_s, config[:template_filename].to_s,
+                             config[:script_filename].to_s, config[:command].to_s, config[:filter_regex].to_s,
+                             config[:parallel].to_s == 'true', false)
+  end
+end

data/lib/knife-node-context-exec/runner.rb

@@ -0,0 +1,65 @@
+require 'open3'
+require 'erb'
+
+# Runs commands over a set of nodes
+module KnifeNodeContextExec
+  # Runs a command in the context of each node
+  class NodeRunner
+    attr_reader :node, :working_directory, :template_filename, :script_filename, :command, :output
+
+    def initialize(node, working_directory, template_filename, script_filename, command)
+      @node = node
+      @working_directory = working_directory
+      @template_filename = template_filename
+      @script_filename = script_filename
+      @command = command
+    end
+
+    def run
+      @thread = Thread.new do
+        script_directory = "#{working_directory}/#{node[:name]}"
+        FileUtils.makedirs(script_directory)
+        template = File.read(template_filename)
+        script = ERB.new(template).result(binding)
+        full_script_filename =
+          File.join(script_directory, script_filename).gsub(File::SEPARATOR, File::ALT_SEPARATOR || File::SEPARATOR)
+        File.open(full_script_filename, 'w') { |file| file.puts(script) }
+        cooked_command = command.gsub('%script%', full_script_filename)
+        @output = []
+        Open3.pipeline_r(cooked_command) do |output, _|
+          output.each_line do |line|
+            line.rstrip!
+            @output << line
+            yield(line)
+          end
+        end
+      end
+      self
+    end
+
+    def wait
+      @thread.join
+      self
+    end
+  end
+
+  def self.run(nodes, working_directory, template_filename, script_filename, command, filter_regex, parallel, clean)
+    FileUtils.makedirs(working_directory)
+    temporary_directory = Dir.mktmpdir('x', working_directory)
+    nodes = nodes.map do |node|
+      node = KnifeNodeContextExec::NodeRunner.new(node, temporary_directory,
+                                                  template_filename, script_filename, command).run do |line|
+        puts line if line =~ /#{filter_regex}/
+      end
+      node.wait unless parallel
+      node
+    end
+    nodes.each(&:wait)
+    nodes.each do |node|
+      puts
+      puts "===== FULL OUTPUT - #{node.node[:name]} ======================"
+      node.output.each { |line| puts line }
+    end
+    FileUtils.remove_dir(working_directory) if clean
+  end
+end

metadata

@@ -0,0 +1,46 @@
+--- !ruby/object:Gem::Specification
+name: knife-node-context-exec
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Andrew Heald
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-11-07 00:00:00.000000000 Z
+dependencies: []
+description: Runs a search against Chef server and executes a command for each returned
+  node. See the project home page for more information.
+email: andrew@heald.uk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/chef/knife/context_exec.rb
+- lib/knife-node-context-exec/runner.rb
+homepage: https://github.com/sappho/knife-node-context-exec
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.8
+signing_key: 
+specification_version: 4
+summary: Runs a search against Chef server and executes a command for each returned
+  node
+test_files: []