Method: NicInfo::Main#initialize

Defined in:

lib/nicinfo/nicinfo_main.rb

#initialize(args, config = nil) ⇒ Main

Returns a new instance of Main.

# File 'lib/nicinfo/nicinfo_main.rb', line 83

def initialize args, config = nil

  if config
    @config = config
  else
    @config = NicInfo::Config.new(NicInfo::Config::formulate_app_data_dir())
  end

  @config.options.require_query = true
  @config.options.jcr = JcrMode::NO_VALIDATION

  @opts = OptionParser.new do |opts|

    opts.banner = "Usage: nicinfo [options] QUERY_VALUE"
    opts.version = NicInfo::VERSION

    opts.separator ""
    opts.separator "Query Options:"

    opts.on("-t", "--type TYPE",
            "Specify type of the query value.",
            "  ip4addr      - IPv4 address",
            "  ip6addr      - IPv6 address",
            "  ip4cidr      - IPv4 cidr block",
            "  ip6cidr      - IPv6 cidr block",
            "  asnumber     - autonomous system number",
            "  domain       - domain name",
            "  entityhandle - handle or id of a contact, organization, registrar or other entity",
            "  nameserver   - fully qualified domain name of a nameserver",
            "  result       - result from a previous query",
            "  esbyname     - entity search by name",
            "  dsbyname     - domain search by name",
            "  dsbynsname   - domain search by nameserver name",
            "  dsbynsip     - domain search by nameserver IP address",
            "  nsbyname     - nameserver search by nameserver name",
            "  nsbyip       - nameserver search by IP address",
            "  trace        - trace route",
            "  url          - RDAP URL",
            "  help         - server help") do |type|
      uptype = type.upcase
      raise OptionParser::InvalidArgument, type.to_s unless QueryType.has_value?(uptype)
      @config.options.query_type = uptype
      @config.options.require_query = false if uptype == "HELP"
    end

    opts.on("-r", "--reverse",
            "Creates a reverse DNS name from an IP address. ") do |reverse|
      @config.options.reverse_ip = true
    end

    opts.on("-b", "--base (or bootstrap) URL",
            "The base URL of the RDAP Service.",
            "When set, the internal bootstrap is bypassed.") do |url|
      @config.config[ NicInfo::BOOTSTRAP][ NicInfo::BOOTSTRAP_URL ] = url
    end

    opts.separator ""
    opts.separator "Cache Options:"

    opts.on("--cache-expiry SECONDS",
            "Age in seconds of items in the cache to be considered expired.") do |s|
      @config.config[ NicInfo::CACHE ][ NicInfo::CACHE_EXPIRY ] = s
    end

    opts.on("--cache YES|NO|TRUE|FALSE",
            "Controls if the cache is used or not.") do |cc|
      @config.config[ NicInfo::CACHE ][ NicInfo::USE_CACHE ] = false if cc =~ /no|false/i
      @config.config[ NicInfo::CACHE ][ NicInfo::USE_CACHE ] = true if cc =~ /yes|true/i
      raise OptionParser::InvalidArgument, cc.to_s unless cc =~ /yes|no|true|false/i
    end

    opts.on("--empty-cache",
            "Empties the cache of all files regardless of eviction policy.") do |cc|
      @config.options.empty_cache = true
      @config.options.require_query = false
    end

    opts.on("--demo",
            "Populates the cache with demonstration results.") do |cc|
      @config.options.demo = true
      @config.options.require_query = false
    end

    opts.separator ""
    opts.separator "Output Options:"

    opts.on( "--messages MESSAGE_LEVEL",
             "Specify the message level",
             "  none - no messages are to be output",
             "  some - some messages but not all",
             "  all  - all messages to be outupt" ) do |m|
      @config.logger.message_level = m.to_s.upcase
      begin
        @config.logger.validate_message_level
      rescue
        raise OptionParser::InvalidArgument, m.to_s
      end
    end

    opts.on( "--messages-out FILE",
             "FILE where messages will be written." ) do |f|
      @config.logger.messages_out = File.open( f, "w+" )
    end

    opts.on( "--data DATA_AMOUNT",
             "Specify the amount of data",
             "  terse  - enough data to identify the object",
             "  normal - normal view of data on objects",
             "  extra  - all data about the object" ) do |d|
      @config.logger.data_amount = d.to_s.upcase
      begin
        @config.logger.validate_data_amount
      rescue
        raise OptionParser::InvalidArgument, d.to_s
      end
    end

    opts.on( "--data-out FILE",
             "FILE where data will be written." ) do |f|
      @config.logger.data_out = File.open( f, "w+" )
    end

    opts.on( "--pager YES|NO|TRUE|FALSE",
             "Turns the pager on and off." ) do |pager|
      @config.logger.pager = false if pager =~ /no|false/i
      @config.logger.pager = true if pager =~ /yes|true/i
      raise OptionParser::InvalidArgument, pager.to_s unless pager =~ /yes|no|true|false/i
    end

    opts.on( "--color-scheme DARK|LIGHT|NONE",
             "Determines color scheme to use:",
             "  dark  - for terminals with dark backgrounds",
             "  light - for terminals with light backgrounds",
             "  none  - turn off colors" ) do |cs|
      @config.logger.color_scheme = cs.to_s.upcase
      raise OptionParser::InvalidArgument, cs.to_s unless cs =~ /dark|light|none/i
    end

    opts.on( "-V",
             "Equivalent to --messages all and --data extra" ) do |v|
      @config.logger.data_amount = NicInfo::DataAmount::EXTRA_DATA
      @config.logger.message_level = NicInfo::MessageLevel::ALL_MESSAGES
    end

    opts.on( "-Q",
             "Equivalent to --messages none and --data extra and --pager false" ) do |q|
      @config.logger.data_amount = NicInfo::DataAmount::EXTRA_DATA
      @config.logger.message_level = NicInfo::MessageLevel::NO_MESSAGES
      @config.logger.pager = false
    end

    opts.on( "--json",
             "Output raw JSON response." ) do |json|
      @config.options.output_json = true
    end

    opts.on( "--jv VALUE",
             "Outputs a specific JSON value." ) do |value|
      unless @config.options.json_values
        @config.options.json_values = Array.new
      end
      @config.options.json_values << value
    end

    opts.separator ""
    opts.separator "Security Options:"

    opts.on( "--try-insecure YES|NO|TRUE|FALSE",
             "Try HTTP if HTTPS fails" ) do |try_insecure|
      @config.config[ NicInfo::SECURITY ][ NicInfo::TRY_INSECURE ] = false if try_insecure =~ /no|false/i
      @config.config[ NicInfo::SECURITY ][ NicInfo::TRY_INSECURE ] = true if try_insecure =~ /yes|true/i
      raise OptionsParser::InvalidArgument, try_insecure.to_s unless try_insecure =~/yes|no|true|false/i
    end

    opts.separator ""
    opts.separator "General Options:"

    opts.on( "-h", "--help",
             "Show this message" ) do
      @config.options.help = true
      @config.options.require_query = false
    end

    opts.on( "--reset",
             "Reset configuration to defaults" ) do
      @config.options.reset_config = true
      @config.options.require_query = false
    end

    opts.on( "--iana",
             "Download RDAP bootstrap files from IANA" ) do
      @config.options.get_iana_files = true
      @config.options.require_query = false
    end

    opts.on( "--jcr STANDARD|STRICT",
             "Validate RDAP response with JCR") do |mode|
      upmode = mode.upcase
      raise OptionParser::InvalidArgument, type.to_s unless JcrMode.has_value?(upmode)
      @config.options.jcr = upmode
      get_jcr_context if upmode == JcrMode::STANDARD_VALIDATION
      get_jcr_strict_context if upmode == JcrMode::STRICT_VALIDATION
    end

  end

  begin
    @opts.parse!(args)
  rescue OptionParser::InvalidOption => e
    puts e.message
    puts "use -h for help"
    exit
  rescue OptionParser::InvalidArgument => e
    puts e.message
    puts "use -h for help"
    exit
  rescue
    puts "Unable to parse command line options"
    puts "use -h for help"
    exit
  end
  @config.options.argv = args

end