box-o-sand/redis-with-ruby/eq-info.rb

#!/usr/bin/env ruby

require 'json'
require 'slop'

require File.expand_path('../earthquakes', __FILE__)


def main
  opts = Slop.parse(:help => true, :strict => true) do
    banner "#{File.basename(__FILE__)} [options]"
    on :L, :listall, 'List all available earthquake ids.'
    on :R, :listall_regions, 'List available earthquake regions.'
    on :M, :listall_magnitudes, 'List available earthquakes magnitudes.'
    on :r, :region=, 'Show earthquakes for region.'
    on :m, :magnitude=, 'Show earthquakes of magnitude.'
    on :i, :eqid=, 'Show all available info for earthquake with id.'
  end

  if ARGV.length < 1
    puts opts
    return 2
  end

  earthquakes = Earthquakes.new

  if opts.listall?
    earthquakes.all_ids.each { |i| puts i }
    return 0
  end

  if opts.listall_regions?
    earthquakes.all_regions.sort.each { |r| puts r }
    return 0
  end

  if opts.listall_magnitudes?
    earthquakes.all_magnitudes.sort.each { |m| puts m }
    return 0
  end

  if opts[:region]
    puts "/* Earthquakes in region #{opts[:region]} */"

    results = []
    earthquakes.in_region(opts[:region]).each do |i|
      results << earthquakes.get_info(i)
    end

    puts JSON.pretty_generate(results)
    return 0
  end

  if opts[:magnitude]
    puts "/* Earthquakes with magnitude #{opts[:magnitude]} */"

    results = []
    earthquakes.with_magnitude(opts[:magnitude]).each do |i|
      results << earthquakes.get_info(i)
    end

    puts JSON.pretty_generate(results)
    return 0
  end

  if opts[:eqid]
    puts "/* Earthquake with id of #{opts[:eqid]} */"
    puts JSON.pretty_generate(earthquakes.get_info(opts[:eqid]))
    return 0
  end

  return 1
end


if $0 == __FILE__
  exit(main)
end