Contents

#!/usr/bin/env ruby
# -*- mode: ruby -*-

require 'muni'
require 'thor'

class MUNI < Thor
  desc "list", "Lists all routes"
  def list
    begin
      Muni::Route.find(:all).each do |route|
        say_status route.tag, route.title, :green
      end
    rescue Muni::NextBusError => e
      say_status "ERROR", e.message, :red
    end    
  end
  
  desc "show [ROUTE_TAG] [DIRECTION]", "Shows a specifc route by name"
  method_option :verbose, :type => :boolean, :aliases => %w(-v --verbose)
  def show(tag, direction = nil)
    begin
      route = Muni::Route.find(tag)
      say_status route.tag, route.title, :green
      dirs = direction ? [route.send(direction.downcase.to_sym)] : route.directions
      dirs.each do |direction|
        say_status direction.id, direction.name, :yellow
        if options.verbose            
          print_table direction.stops.collect{|stop| [stop.tag, stop.title]}, {:ident => 8}
        end
      end
    rescue Muni::NextBusError => e
      say_status "ERROR", e.message, :red
    end      
  end
  
  desc "predict [ROUTE] [DIRECTION] [STOP]", "Retrieve predictions for a route at a specific stop"
  def predict(route, direction, *stop)
    # Join the remaining arguments so that predict can be called without
    # having to quote the stop name.
    stop = stop.join(' ')
    
    if stop.empty?
      say_status "ERROR", "A stop is required", :red
      exit!
    end
    
    # Get the bus route information
    # TODO Only do this if necessary
    bus = Muni::Route.find(route)

    # Look up the directon information
    direction = bus.direction_at(direction)
    
    # Look up the stop information
    stop = direction.stop_at(stop)
    
    # Retrieve the predictions
    begin
      say "Route #{bus.title} going #{direction.name} at #{stop.title}:"
      print_table stop.predictions.collect{|time| [time.vehicle, time.pretty_time]}, {:ident => 8}
    rescue Muni::NextBusError => e
      say_status "ERROR", e.message, :red
    end
  end

end

MUNI.start

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
muni-0.0.8	bin/muni
muni-0.0.7	bin/muni
muni-0.0.6	bin/muni
muni-0.0.5	bin/muni
muni-0.0.4	bin/muni
muni-0.0.3	bin/muni
muni-0.0.2	bin/muni