module MCollective module DDL # The base class for all kinds of DDL files. DDL files when # run gets parsed and builds up a hash of the basic primitive # types, ideally restricted so it can be converted to JSON though # today there are some Ruby Symbols in them which might be fixed # laster on. # # The Hash being built should be stored in @entities, the format # is generally not prescribed but there's a definite feel to how # DDL files look so study the agent and discovery ones to see how # the structure applies to very different use cases. # # For every plugin type you should have a single word name - that # corresponds to the directory in the libdir where these plugins # live. If you need anything above and beyond 'metadata' in your # plugin DDL then add a PlugintypeDDL class here and add your # specific behaviors to those. class Base attr_reader :meta, :entities, :pluginname, :plugintype, :usage, :requirements ACTIVATION_DEFAULT = true def initialize(plugin, plugintype=:agent, loadddl=true) @entities = {} @meta = {} @usage = "" @config = Config.instance @pluginname = plugin @plugintype = plugintype.to_sym @requirements = {} loadddlfile if loadddl end # Generates help using the template based on the data # created with metadata and input. # # If no template name is provided one will be chosen based # on the plugin type. If the provided template path is # not absolute then the template will be loaded either from # the config dir and if that does not exist, default to # /etc/mcollective def help(template=nil) template ||= template_for_plugintype template = Util.templatepath(template) unless Util.absolute_path?(template) template = File.read(template) meta = @meta entities = @entities unless template == "metadata-help.erb" metadata_template = Util.templatepath("metadata-help.erb") metadata_template = File.read(metadata_template) metastring = ERB.new(metadata_template, 0, "%") metastring = metastring.result(binding) end erb = ERB.new(template, 0, "%") erb.result(binding) end # rubocop:disable Lint/DuplicateMethods, Style/TrivialAccessors def usage(usage_text) @usage = usage_text end # rubocop:enable Lint/DuplicateMethods, Style/TrivialAccessors def template_for_plugintype case @plugintype when :agent "rpc-help.erb" else if File.exist?(Util.templatepath("#{@plugintype}-help.erb")) "#{@plugintype}-help.erb" else # Default help template gets loaded if plugintype-help does not exist. "metadata-help.erb" end end end def loadddlfile raise("%s/%s is disabled, cannot load DDL file" % [@plugintype, @pluginname]) if @config.mode == :client && !client_activated? if ddlfile = findddlfile instance_eval(File.read(ddlfile), ddlfile, 1) else raise("Can't find DDL for #{@plugintype} plugin '#{@pluginname}'") end end def findddlfile(ddlname=nil, ddltype=nil) ddlname ||= @pluginname ddltype ||= @plugintype @config.libdir.each do |libdir| ddlfile = File.join([libdir, "mcollective", ddltype.to_s, "#{ddlname}.ddl"]) if File.exist?(ddlfile) Log.debug("Found #{ddlname} ddl at #{ddlfile}") return ddlfile end end false end def client_activated? ACTIVATION_DEFAULT end def validate_requirements requirement = @requirements[:mcollective] if requirement && (Util.versioncmp(Util.mcollective_version, requirement) < 0) raise DDLValidationError, "%s plugin '%s' requires MCollective version %s or newer" % [@plugintype.to_s.capitalize, @pluginname, requirement] end true end # validate strings, lists and booleans, we'll add more types of validators when # all the use cases are clear # # only does validation for arguments actually given, since some might # be optional. We validate the presense of the argument earlier so # this is a safe assumption, just to skip them. # # :string can have maxlength and regex. A maxlength of 0 will bypasss checks # :list has a array of valid values def validate_input_argument(input, key, argument) Validator.load_validators case input[key][:type] when :string Validator.validate(argument, :string) Validator.length(argument, input[key][:maxlength].to_i) Validator.validate(argument, input[key][:validation]) when :list Validator.validate(argument, input[key][:list]) else Validator.validate(argument, input[key][:type]) end true rescue => e raise DDLValidationError, "Cannot validate input %s: %s" % [key, e.to_s], e.backtrace end # Registers an input argument for a given action # # See the documentation for action for how to use this def input(argument, properties) raise "Cannot figure out what entity input #{argument} belongs to" unless @current_entity entity = @current_entity [:prompt, :description, :type].each do |arg| raise "Input needs a :#{arg} property" unless properties.include?(arg) end @entities[entity][:input][argument] = {:prompt => properties[:prompt], :description => properties[:description], :type => properties[:type], :default => properties[:default], :optional => properties[:optional]} case properties[:type] when :string raise "Input type :string needs a :validation argument" unless properties.include?(:validation) raise "Input type :string needs a :maxlength argument" unless properties.include?(:maxlength) @entities[entity][:input][argument][:validation] = properties[:validation] @entities[entity][:input][argument][:maxlength] = properties[:maxlength] when :list raise "Input type :list needs a :list argument" unless properties.include?(:list) @entities[entity][:input][argument][:list] = properties[:list] end end # Registers an output argument for a given action # # See the documentation for action for how to use this def output(argument, properties) raise "Cannot figure out what action input #{argument} belongs to" unless @current_entity raise "Output #{argument} needs a description argument" unless properties.include?(:description) raise "Output #{argument} needs a display_as argument" unless properties.include?(:display_as) action = @current_entity @entities[action][:output][argument] = {:description => properties[:description], :display_as => properties[:display_as], :default => properties[:default]} @entities[action][:output][argument][:type] = properties[:type] if properties[:type] end def requires(requirement) raise "Requirement should be a hash in the form :item => 'requirement'" unless requirement.is_a?(Hash) valid_requirements = [:mcollective] requirement.each_key do |key| raise "Requirement %s is not a valid requirement, only %s is supported" % [key, valid_requirements.join(", ")] unless valid_requirements.include?(key) @requirements[key] = requirement[key] end validate_requirements end # Registers meta data for the introspection hash def metadata(meta) [:name, :description, :author, :license, :version, :url, :timeout].each do |arg| raise "Metadata needs a :#{arg} property" unless meta.include?(arg) end @meta = meta end end end end