class CustomOptionParser

WPScan - WordPress Security Scanner
Copyright (C) 2012-2013

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

Attributes

symbols_used[R]

Public Class Methods

new(banner = nil, width = 32, indent = ' ' * 4) click to toggle source

# File lib/common/custom_option_parser.rb, line 23
def initialize(banner = nil, width = 32, indent = ' ' * 4)
  @results         = {}
  @symbols_used    = []
  super(banner, width, indent)
end

Protected Class Methods

option_to_symbol(option) click to toggle source

param Array option

# File lib/common/custom_option_parser.rb, line 73
def self.option_to_symbol(option)
  option_name = nil

  option.each do |option_attr|
    if option_attr =~ %r^--/
      option_name = option_attr
      break
    end
  end

  if option_name
    option_name = option_name.gsub(%r^--/, '').gsub(%r-/, '_').gsub(%r .*$/, '')
    :"#{option_name}"
  else
    raise "Could not find the option name for #{option}"
  end
end

Public Instance Methods

add(options) click to toggle source

param Array(Array) or Array options

# File lib/common/custom_option_parser.rb, line 31
def add(options)
  if options.is_a?(Array)
    if options[0].is_a?(Array)
      options.each do |option|
        add_option(option)
      end
    else
      add_option(options)
    end
  else
    raise "Options must be at least an Array, or an Array(Array). #{options.class} supplied"
  end
end

add_option(option) click to toggle source

param Array option

# File lib/common/custom_option_parser.rb, line 46
def add_option(option)
  if option.is_a?(Array)
    option_symbol = CustomOptionParser::option_to_symbol(option)

    unless @symbols_used.include?(option_symbol)
      @symbols_used << option_symbol

      self.on(*option) do |arg|
        @results[option_symbol] = arg
      end
    else
      raise "The option #{option_symbol} is already used !"
    end
  else
    raise "The option must be an array, #{option.class} supplied : '#{option}'"
  end
end

results(argv = default_argv) click to toggle source

return Hash

# File lib/common/custom_option_parser.rb, line 65
def results(argv = default_argv)
  self.parse!(argv) if @results.empty?

  @results
end