Contents

module Puppet::Parser::Functions
  newfunction(:add_progress) do |args|
    loaded = false
    begin
      require 'puppet/transaction'
      loaded = true
    rescue LoadError
      ::Puppet.warning 'Unable to load puppet/transaction for progress bar support, this version may not be supported'
    end

    if loaded
      # Monkey patch the transaction to put our wrapper around the report object
      class Puppet::Transaction
        attr_accessor :in_main_catalog

        def is_interesting?(resource)
          return false if resource.class.name == :component  # ignore defined types
          ![:schedule, :class, :stage, :filebucket, :anchor, :concat_fragment].include?(resource.to_s.split('[')[0].downcase.to_sym)
        end

        def tracked_resources
          @tracked_resources ||= catalog.vertices.select { |resource| is_interesting?(resource) }.map(&:to_s)
        end

        def evaluate_with_trigger(*args, &block)
          if catalog.version
            self.in_main_catalog = true
            tracked_resources.each { |r| ::Puppet.info "MONITOR_RESOURCE #{r}" }
          end
          evaluate_without_trigger(*args, &block)
          self.in_main_catalog = false if catalog.version
        end

        if method_defined?(:evaluate) && method_defined?(:report)
          alias_method :evaluate_without_trigger, :evaluate
          alias_method :evaluate, :evaluate_with_trigger
        else
          ::Puppet.warning 'Unable to patch Puppet transactions for progress bar support, this version may not be supported'
        end
      end
    end
  end
end

Version data entries

25 entries across 25 versions & 1 rubygems

Version	Path
kafo-6.5.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.4.1	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.4.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.3.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.2.1	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.2.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.1.2	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.1.1	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.1.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-6.0.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-5.1.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-5.0.1	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-5.0.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-4.1.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-4.0.1	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-4.0.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-3.0.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-2.1.0	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-2.0.2	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb
kafo-2.0.1	modules/kafo_configure/lib/puppet/parser/functions/add_progress.rb