Sha256: 2f9a3ce645a9c63c1f2e86d28f1dab4672db4ebb7cc774428f563e639ce35554

Contents?: true

Size: 1.59 KB

Versions: 8

Compression:

Stored size: 1.59 KB

Contents

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

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

        super(name, description, author, license, version, url, timeout)
        @plugin_name = plugin_name
        @actions = actions || []
        @ddl = create_ddl
        @mod_name = "Agent"
        @pclass = "RPC::Agent"
        @content = create_plugin_content
        @plugin = create_plugin_string
        write_plugins
      end

      def create_ddl
        action_text = ""
        @actions.each_with_index do |action, i|
          action_text += "action \"#{action}\", :description => \"%ACTIONDESCRIPTION%\" do\n"
          action_text += action_help if i == 0
          action_text += "end\n"
          action_text += "\n" unless @actions.size == (i + 1)
        end
        # Use inherited method to create metadata part of the ddl
        create_metadata_string + action_text
      end

      def create_plugin_content
        content_text = ""

        # Add actions to agent file
        @actions.each_with_index do |action, i|
          content_text +=  "%6s%s" % [" ", "action \"#{action}\" do\n"]
          content_text +=  "%6s%s" % [" ", "end\n"]
          content_text += "\n" unless @actions.size == (i + 1)
        end
        content_text
      end

      def action_help
        action_snippet = File.read(File.join(File.dirname(__FILE__), "templates", "action_snippet.erb"))
        ERB.new(action_snippet).result
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
choria-mcorpc-support-2.24.2 lib/mcollective/generators/agent_generator.rb
choria-mcorpc-support-2.24.1 lib/mcollective/generators/agent_generator.rb
choria-mcorpc-support-2.24.0 lib/mcollective/generators/agent_generator.rb
choria-mcorpc-support-2.23.3 lib/mcollective/generators/agent_generator.rb
choria-mcorpc-support-2.23.2 lib/mcollective/generators/agent_generator.rb
choria-mcorpc-support-2.23.1 lib/mcollective/generators/agent_generator.rb
choria-mcorpc-support-2.23.0 lib/mcollective/generators/agent_generator.rb
choria-mcorpc-support-2.23.0.pre lib/mcollective/generators/agent_generator.rb