+module MCollective
+ # A simple helper to build cli tools that supports a uniform command line
+ # layout.
+ class Optionparser
+ attr_reader :parser
+
+ # Creates a new instance of the parser, you can supply defaults and include named groups of options.
+ #
+ # Starts a parser that defaults to verbose and that includs the filter options:
+ #
+ # oparser = MCollective::Optionparser.new({:verbose => true}, "filter")
+ #
+ # Stats a parser in non verbose mode that does support discovery
+ #
+ # oparser = MCollective::Optionparser.new()
+ #
+ # Starts a parser in verbose mode that does not show the common options:
+ #
+ # oparser = MCollective::Optionparser.new({:verbose => true}, "filter", "common")
+ def initialize(defaults = {}, include_sections = nil, exclude_sections = nil)
+ @parser = ::OptionParser.new
+
+ @include = [include_sections].flatten
+ @exclude = [exclude_sections].flatten
+
+ @options = Util.default_options
+
+ @options.merge!(defaults)
+ end
+
+ # Parse the options returning the options, you can pass a block that adds additional options
+ # to the Optionparser.
+ #
+ # The sample below starts a parser that also prompts for --arguments in addition to the defaults.
+ # It also sets the description and shows a usage message specific to this app.
+ #
+ # options = oparser.parse{|parser, options|
+ # parser.define_head "Control the mcollective controller daemon"
+ # parser.banner = "Usage: sh-mcollective [options] command"
+ #
+ # parser.on('--arg', '--argument ARGUMENT', 'Argument to pass to agent') do |v|
+ # options[:argument] = v
+ # end
+ # }
+ #
+ # Users can set default options that get parsed in using the MCOLLECTIVE_EXTRA_OPTS environemnt
+ # variable
+ def parse(&block)
+ yield(@parser, @options) if block_given?
+
+ add_required_options
+
+ add_common_options unless @exclude.include?("common")
+
+ @include.each do |i|
+ next if @exclude.include?(i)
+
+ options_name = "add_#{i}_options"
+ send(options_name) if respond_to?(options_name)
+ end
+
+ @parser.environment("MCOLLECTIVE_EXTRA_OPTS")
+
+ @parser.parse!
+
+ @options[:collective] = Config.instance.main_collective unless @options[:collective]
+
+ @options
+ end
+
+ # These options will be added if you pass 'filter' into the include list of the
+ # constructor.
+ def add_filter_options
+ @parser.separator ""
+ @parser.separator "Host Filters"
+
+ @parser.on('-W', '--with FILTER', 'Combined classes and facts filter') do |f|
+ f.split(" ").each do |filter|
+ begin
+ fact_parsed = parse_fact(filter)
+ @options[:filter]["fact"] << fact_parsed
+ rescue
+ @options[:filter]["cf_class"] << filter
+ end
+ end
+ end
+
+ @parser.on('-S', '--select FILTER', 'Compound filter combining facts and classes') do |f|
+ @options[:filter]["compound"] << Matcher.create_compound_callstack(f)
+ end
+
+ @parser.on('-F', '--wf', '--with-fact fact=val', 'Match hosts with a certain fact') do |f|
+ fact_parsed = parse_fact(f)
+
+ @options[:filter]["fact"] << fact_parsed if fact_parsed
+ end
+
+ @parser.on('-C', '--wc', '--with-class CLASS', 'Match hosts with a certain config management class') do |f|
+ @options[:filter]["cf_class"] << f
+ end
+
+ @parser.on('-A', '--wa', '--with-agent AGENT', 'Match hosts with a certain agent') do |a|
+ @options[:filter]["agent"] << a
+ end
+
+ @parser.on('-I', '--wi', '--with-identity IDENT', 'Match hosts with a certain configured identity') do |a|
+ @options[:filter]["identity"] << a
+ end
+ end
+
+ # These options should always be present
+ def add_required_options
+ @parser.on('-c', '--config FILE', 'Load configuratuion from file rather than default') do |f|
+ @options[:config] = f
+ end
+
+ @parser.on('-v', '--verbose', 'Be verbose') do |v|
+ @options[:verbose] = v
+ end
+
+ @parser.on('-h', '--help', 'Display this screen') do
+ puts @parser
+ exit! 1
+ end
+ end
+
+ # These options will be added to most cli tools
+ def add_common_options
+ @parser.separator ""
+ @parser.separator "Common Options"
+
+ @parser.on('-T', '--target COLLECTIVE', 'Target messages to a specific sub collective') do |f|
+ @options[:collective] = f
+ end
+
+ @parser.on('--dt', '--discovery-timeout SECONDS', Integer, 'Timeout for doing discovery') do |t|
+ @options[:disctimeout] = t
+ end
+
+ @parser.on('-t', '--timeout SECONDS', Integer, 'Timeout for calling remote agents') do |t|
+ @options[:timeout] = t
+ end
+
+ @parser.on('-q', '--quiet', 'Do not be verbose') do |v|
+ @options[:verbose] = false
+ end
+
+ @parser.on('--ttl TTL', 'Set the message validity period') do |v|
+ @options[:ttl] = v.to_i
+ end
+
+ @parser.on('--reply-to TARGET', 'Set a custom target for replies') do |v|
+ @options[:reply_to] = v
+ end
+
+ @parser.on('--dm', '--disc-method METHOD', 'Which discovery method to use') do |v|
+ raise "Discovery method is already set by a competing option" if @options[:discovery_method] && @options[:discovery_method] != v
+ @options[:discovery_method] = v
+ end
+
+ @parser.on('--do', '--disc-option OPTION', 'Options to pass to the discovery method') do |a|
+ @options[:discovery_options] << a
+ end
+
+ @parser.on("--nodes FILE", "List of nodes to address") do |v|
+ raise "Cannot mix --disc-method, --disc-option and --nodes" if @options[:discovery_method] || @options[:discovery_options].size > 0
+ raise "Cannot read the discovery file #{v}" unless File.readable?(v)
+
+ @options[:discovery_method] = "flatfile"
+ @options[:discovery_options] = v
+ end
+ end
+
+ private
+ # Parse a fact filter string like foo=bar into the tuple hash thats needed
+ def parse_fact(fact)
+ Util.parse_fact_string(fact)
+ end
+
+ end
+end