tmiller/urban

require "optparse"
require "ostruct"
require "socket"

module Urban
  class CLI

    attr_accessor :dictionary

    def initialize
      @dictionary = Urban::Dictionary.new
    end

    def run(args = ARGV)

      options = parse(args)
      results = lookup(options)

      case
      when results.definitions
        generate_output results, options
      when results.phrase
        error "no definitions found for #{results.phrase.upcase}."
      else
        $stdout.puts options.version ? version : usage
      end

    rescue SocketError
      error "no internet connection available."
    rescue OptionParser::InvalidOption => e
      error "#{e.message}\nTry `urban --help' for more information."
    rescue Object => e
      error e.message
    end

    private

    def error(message)
      $stderr.puts "urban: #{message}"
    end

    def generate_output(entry, options)
      output = "\n#{entry.phrase.upcase}\n\n"
      if options.all
        output << "#{entry.definitions.join("\n\n")}\n\n"
      else
        output << "#{entry.definitions.first}\n\n"
      end
      output << "URL: #{entry.url}\n\n" if options.url

      $stdout.puts output
    end

    def parse(args)
      options = OpenStruct.new

      options_parser = OptionParser.new do |o|
        o.on("-a", "--all") { options.all = true }
        o.on("-r", "--random") { options.random = true }
        o.on("-u", "--url") { options.url = true }
        o.on("-h", "--help")
        o.on("-v", "--version") { options.version = true }
      end

      options_parser.parse!(args)
      options.phrase = args.join(" ")
      options.search = !options.phrase.empty?
      options
    end

    def lookup(options)
      case
      when options.random then dictionary.random
      when options.search then dictionary.search(options.phrase)
      else OpenStruct.new
      end
    end

    def version
      "Urban #{Urban::VERSION} (c) Thomas Miller"
    end

    def usage
      <<-EOS
Usage: urban [OPTION]... [PHRASE]
Search http://urbandictionary.com for definitions of phrases

Options:
    -a, --all                   List all definitions
    -r, --random                Return a random phrase and definition
    -u, --url                   Print the definition's url after the definition
    -h, --help                  Show this message
    -v, --version               Show version information

Examples:
    urban cookie monster        Search for "cookie monster" and print its
                                first definition
    urban -a cookie monster     Search for "cookie monster" and print all of
                                its available definitions
    urban -r                    Print a random phrase and its first definition
    urban -ra                   Print a random phrase and all of its available
                                definitions

      EOS
    end
  end
end