dyoung522/devtools-base

require "config"
require "optparse"

module DevTools
  class OptParse
    attr_reader :parser, :version

    def initialize(opts)
      program_path  = DevTools::PROGRAM
      program_files = []

      %W(~/.#{program_path} ./.#{program_path} ./#{program_path}.conf ./#{program_path}.rc).each do |path|
        program_files.push File.expand_path(path)
      end

      Config.setup do |config|
        config.const_name = 'Options'
        config.use_env    = true
      end

      Config.load_and_set_settings opts[:testing] ? String.new : program_files

      default_options(Options, opts[:defaults]) if opts[:defaults]

      Options.verbose = Options.verbose ? 1 : 0 unless Options.verbose.is_a?(Numeric)
      Options.debug   = true if Options.verbose >= 5

      @parser  ||= common_options opts[:testing]
      @version ||= sprintf "%s v%s (%s v%s)\n",
                           opts[:name] || DevTools::PROGRAM, opts[:version] || DevTools::VERSION,
                           "DevTools", DevTools::VERSION
    end

    def common_options(unit_testing = false)
      OptionParser.new do |opts|
        opts.on_tail("-h", "--help", "Show this message") do
          unless unit_testing
            puts version + "\n"
            puts opts
            exit
          end
        end

        # Config file
        opts.on_tail("-c", "--config FILE", "Specify an alternate configuration file") do |file|
          Options.add_source! file
          Options.reload!
        end

        # dry-run switch
        opts.on_tail("--dry-run", "Don't actually run commands") do
          Options.dryrun = true
        end

        # Verbose switch
        opts.on_tail("-q", "--quiet", "Run quietly") do
          Options.verbose = 0
        end

        opts.on_tail("-v", "--verbose", "Run verbosely (may be specified more than once)") do
          Options.verbose += 1
        end

        opts.on_tail("--version", "Show version") do
          unless unit_testing
            puts version
            exit
          end
        end
      end
    end

    def default_options(options, opts)
      opts.each do |key, value|
        options[key] = value unless options.keys.include?(key)
      end
    end
  end # class Option
end # module DevTools