Sha256: 0a28122f07373d7589902951ef3279dcbbb01d81842f8d2c256eea25436e7b9a

Contents?: true

Size: 1.61 KB

Versions: 5

Compression:

Stored size: 1.61 KB

Contents

module MCollective
  module Generators
    class DataGenerator < Base
      attr_accessor :ddl, :content

      def initialize(plugin_name, outputs=[], name=nil, description=nil, author=nil,
                     license=nil, version=nil, url=nil, timeout=nil)

        super(name, description, author, license, version, url, timeout)
        @mod_name = "Data"
        @pclass = "Base"
        @plugin_name = plugin_name
        @outputs = outputs
        @ddl = create_ddl
        @content = create_plugin_content
        @plugin = create_plugin_string
        write_plugins
      end

      def create_ddl
        query_text = "dataquery :description => \"Query information\" do\n"
        query_text += ERB.new(File.read(File.join(File.dirname(__FILE__), "templates", "data_input_snippet.erb"))).result

        @outputs.each_with_index do |output, i|
          query_text += "%2s%s" % [" ", "output :#{output},\n"]
          query_text += "%9s%s" % [" ", ":description => \"%DESCRIPTION%\",\n"]
          query_text += "%9s%s" % [" ", ":display_as => \"%DESCRIPTION%\"\n"]
          query_text += "\n" unless @outputs.size == (i + 1)
        end

        query_text += "end"

        # Use inherited method to create metadata part of the ddl
        create_metadata_string + query_text
      end

      def create_plugin_content
        content_text = "%6s%s" % [" ", "query do |what|\n"]

        @outputs.each do |output|
          content_text += "%8s%s" % [" ", "result[:#{output}] = nil\n"]
        end
        content_text += "%6s%s" % [" ", "end\n"]

        # Add actions to agent file
        content_text
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
choria-mcorpc-support-2.23.3 lib/mcollective/generators/data_generator.rb
choria-mcorpc-support-2.23.2 lib/mcollective/generators/data_generator.rb
choria-mcorpc-support-2.23.1 lib/mcollective/generators/data_generator.rb
choria-mcorpc-support-2.23.0 lib/mcollective/generators/data_generator.rb
choria-mcorpc-support-2.23.0.pre lib/mcollective/generators/data_generator.rb