choria-colt 0.5.1 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e5a3d94cfa899f41ced9fc0405ed5f4fb41924267d4b56f6d31ba28420c9b0e7
-  data.tar.gz: df5bef6793badb31de0ee0185d0f8d3ca44288fedcf64b34084dbb60d1a6de30
+  metadata.gz: 4e76e667917bf1958b48d45148712fa5e0d6f2c15280bbbda0d4899b0537e9f8
+  data.tar.gz: 17637d3bf6738bef6c28a282aabb1c4ac7680e4b2e8a6f0fe873a27b59079cc1
 SHA512:
-  metadata.gz: bba0206e6f84679cd6c7ed5fd1831dc8e2beba052b096b265ac272baadeaf20a69585d6165a37ca1f7db2ca48a6fe532eeb77171c75f656bb0aa9f2c50796ccb
-  data.tar.gz: 02d6f79e80a664445793596ea9d96f106a14523ba9824893e7c1d43cc2c3484f63e27a1a20173d30e608e48f4d59872db9878385da2b197021d56374c502defa
+  metadata.gz: bacccbdfd46669936f1271d05e4fdceae1864e07ae7f14b260f6de3ec2ed1e90d0c3270f57ea3faaa424b267eab59e1460abcdf4262feec421c6ea510bc811d5
+  data.tar.gz: ad8aaf32f6f71f349d04955cb795c0439c4ecd30103f290da82b3ce7037c6dce330b7e89123a416c120b61bc194c754debfb97412e6e200af8d53264aaba169c

data/.rubocop_todo.yml

@@ -1,16 +1,11 @@
 # This configuration was generated by
 # `rubocop --auto-gen-config`
-# on 2022-04-28 09:34:37 UTC using RuboCop version 1.26.0.
+# on 2022-05-03 07:08:32 UTC using RuboCop version 1.26.0.
 # The point is for the user to remove these configuration records
 # one by one as the offenses are removed from the code base.
 # Note that changes in the inspected code, or installation of new
 # versions of RuboCop, may require this file to be generated again.
 
-# Offense count: 1
-# Configuration parameters: IgnoredMethods, CountRepeatedAttributes.
-Metrics/AbcSize:
-  Max: 22
-
 # Offense count: 1
 # Configuration parameters: IgnoredMethods.
 Metrics/CyclomaticComplexity:

data/CHANGELOG.md

@@ -1,5 +1,20 @@
 # Changelog
 
+## [v0.6.0](https://github.com/opus-codium/choria-colt/tree/v0.6.0) (2022-05-16)
+
+[Full Changelog](https://github.com/opus-codium/choria-colt/compare/v0.5.1...v0.6.0)
+
+**Merged pull requests:**
+
+- Allow to run a task from any puppet environment [\#28](https://github.com/opus-codium/choria-colt/pull/28) ([neomilium](https://github.com/neomilium))
+- CLI: Use symbol notation to target `:all` nodes [\#27](https://github.com/opus-codium/choria-colt/pull/27) ([neomilium](https://github.com/neomilium))
+- CLI: Allow targeting on task status requests [\#26](https://github.com/opus-codium/choria-colt/pull/26) ([neomilium](https://github.com/neomilium))
+- Fix `task status` command [\#25](https://github.com/opus-codium/choria-colt/pull/25) ([neomilium](https://github.com/neomilium))
+
+## [v0.5.1](https://github.com/opus-codium/choria-colt/tree/v0.5.1) (2022-05-02)
+
+[Full Changelog](https://github.com/opus-codium/choria-colt/compare/v0.5.0...v0.5.1)
+
 ## [v0.5.0](https://github.com/opus-codium/choria-colt/tree/v0.5.0) (2022-05-02)
 
 [Full Changelog](https://github.com/opus-codium/choria-colt/compare/v0.4.0...v0.5.0)

data/README.md

@@ -51,8 +51,8 @@ To install this gem onto your local machine, run `bundle exec rake install`.
 
 
 To release a new version:
-  1. update the version number in `version.rb`
-  1. generate CHANGELOG.md with `bundle exec rake changelog`
+  1. update the version number in `lib/choria/colt/version.rb`
+  1. generate `CHANGELOG.md` with `bundle exec rake changelog`
   1. commit changes
   1. run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 

data/lib/choria/colt/cli.rb

@@ -9,6 +9,15 @@ module Choria
   class Colt
     class CLI < Thor
       class Tasks < Thor
+        def self.define_targets_and_filters_options
+          option :targets,
+                 aliases: ['--target', '-t'],
+                 desc: 'Identifies the targets of the command.'
+          option :targets_with_classes,
+                 aliases: ['--targets-with-class', '-C'],
+                 desc: 'Select the targets which have the specified Puppet classes.'
+        end
+
         class_option :log_level,
                      desc: 'Set log level for CLI',
                      default: 'info'
@@ -19,28 +28,22 @@ module Choria
 
           Parameters take the form parameter=value.
         DESC
-        option :targets,
-               aliases: ['--target', '-t'],
-               desc: 'Identifies the targets of the command.'
-        option :targets_with_classes,
-               aliases: ['--targets-with-class', '-C'],
-               desc: 'Select the targets which have the specified Puppet classes.'
-        def run(*args) # rubocop:disable Metrics/AbcSize
-          input = extract_task_parameters_from_args(args)
-          targets = extract_targets_from_options
-
+        define_targets_and_filters_options
+        option :environment,
+               aliases: ['-E'],
+               desc: 'Puppet environment to grab tasks from',
+               default: 'production'
+        def run(*args)
           raise Thor::Error, 'Task name is required' if args.empty?
-          raise Thor::Error, "Too many arguments: #{args}" unless args.count == 1
 
-          raise Thor::Error, 'Flag --targets or --targets-with-class is required' if options['targets'].nil? && options['targets_with_classes'].nil?
+          input = extract_task_parameters_from_args(args)
+          raise Thor::Error, "Too many arguments: #{args}" unless args.count == 1
 
           task_name = args.shift
+          targets, targets_with_classes = extract_targets_and_filters_from_options
 
-          logger.debug "Targets: #{targets}"
-
-          targets_with_classes = options['targets_with_classes']&.split(',')
-
-          results = colt.run_bolt_task task_name, input: input, targets: targets, targets_with_classes: targets_with_classes do |result|
+          environment = options['environment']
+          results = colt.run_bolt_task task_name, input: input, targets: targets, targets_with_classes: targets_with_classes, environment: environment do |result|
             $stdout.puts formatter.process_result(result)
           end
 
@@ -84,8 +87,11 @@ module Choria
 
           A task ID is required to request Choria services and retrieve results.
         DESC
+        define_targets_and_filters_options
         def status(task_id)
-          results = colt.wait_bolt_task task_id do |result|
+          targets, targets_with_classes = extract_targets_and_filters_from_options
+
+          results = colt.wait_bolt_task(task_id, targets: targets, targets_with_classes: targets_with_classes) do |result|
             $stdout.puts formatter.process_result(result)
           end
 
@@ -130,8 +136,17 @@ module Choria
             end.to_h
           end
 
+          def extract_targets_and_filters_from_options
+            raise Thor::Error, 'Flag --targets or --targets-with-class is required' if options['targets'].nil? && options['targets_with_classes'].nil?
+
+            targets = extract_targets_from_options
+            targets_with_classes = options['targets_with_classes']&.split(',')
+
+            [targets, targets_with_classes]
+          end
+
           def extract_targets_from_options
-            return nil if options['targets'] == 'all'
+            return nil if options['targets'] == ':all'
 
             targets = options['targets']&.split(',')
             targets&.map do |t|

data/lib/choria/colt/version.rb

@@ -2,6 +2,6 @@
 
 module Choria
   class Colt
-    VERSION = '0.5.1'
+    VERSION = '0.6.0'
   end
 end

data/lib/choria/colt.rb

@@ -19,9 +19,9 @@ module Choria
       @orchestrator = Choria::Orchestrator.new logger: @logger
     end
 
-    def run_bolt_task(task_name, input: {}, targets: nil, targets_with_classes: nil, &block)
+    def run_bolt_task(task_name, input: {}, targets: nil, targets_with_classes: nil, environment: 'production', &block)
       logger.debug "Instantiate task '#{task_name}' and validate input"
-      task = Choria::Orchestrator::Task.new(name: task_name, input: input, orchestrator: orchestrator)
+      task = Choria::Orchestrator::Task.new(name: task_name, input: input, environment: environment, orchestrator: orchestrator)
 
       task.on_result(&block) if block_given?
 
@@ -33,11 +33,11 @@ module Choria
       raise
     end
 
-    def wait_bolt_task(task_id, &block)
-      task = Choria::Orchestrator::Task.new(id: task_id, orchestrator: orchestrator)
+    def wait_bolt_task(task_id, targets: nil, targets_with_classes: nil, &block)
+      orchestrator.discover(targets: targets, targets_with_classes: targets_with_classes)
 
+      task = Choria::Orchestrator::Task.new(id: task_id, orchestrator: orchestrator)
       task.on_result(&block) if block_given?
-
       task.wait
       task.results
     rescue Choria::Orchestrator::Error => e

data/lib/choria/orchestrator/task.rb

@@ -65,8 +65,8 @@ module Choria
       end
 
       def rpc_results=(results)
-        pending_results, completed_results = results.partition { |res| res[:data][:exitcode] == -1 }
-        @pending_targets ||= pending_results.map { |res| res[:sender] }
+        completed_results = results.reject { |res| res[:data][:exitcode] == -1 }
+        @pending_targets ||= results.map { |res| res[:sender] }
 
         new_results = completed_results.select { |res| @pending_targets.include? res[:sender] }
         new_results.each do |res|

data/lib/choria/orchestrator.rb

@@ -39,10 +39,16 @@ module Choria
       @tasks_support ||= MCollective::Util::Choria.new.tasks_support
     end
 
-    def run(task, targets: nil, targets_with_classes: nil, verbose: false) # rubocop:disable Metrics/AbcSize
+    def run(task, targets: nil, targets_with_classes: nil, verbose: false)
       rpc_client.progress = verbose
+      discover(targets: targets, targets_with_classes: targets_with_classes)
+      raise DiscoverError, 'No requests sent, no nodes discovered' if rpc_client.discover.size.zero?
+
+      task.run
+    end
 
-      logger.debug "Running task: '#{task.name}' (targets: #{targets.nil? ? 'all' : targets})"
+    def discover(targets: nil, targets_with_classes: nil)
+      logger.debug "Targets: #{targets.nil? ? 'all' : targets})"
       targets&.each { |target| rpc_client.identity_filter target }
 
       unless targets_with_classes.nil?
@@ -51,9 +57,7 @@ module Choria
       end
 
       logger.info 'Discovering targets…'
-      raise DiscoverError, 'No requests sent, no nodes discovered' if rpc_client.discover.size.zero?
-
-      task.run
+      rpc_client.discover
     end
 
     def rpc_client

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: choria-colt
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.6.0
 platform: ruby
 authors:
 - Romuald Conty
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-05-02 00:00:00.000000000 Z
+date: 2022-05-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport