muni 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in muni.gemspec
+gemspec

data/README.md

@@ -0,0 +1 @@
+Hello

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/bin/muni

@@ -0,0 +1,67 @@
+#!/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_options %w( verbose -v ) => :boolean
+  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(' ')
+    
+    # Get the bus route information
+    # TODO Only do this if necessary
+    bus = Muni::Route.find(route)
+
+    # Look up the stop information unless a numerica stop tag is specified
+    stop = bus.send(direction.downcase.to_sym).stop_at(stop).tag unless stop =~ /[1-9][0-9]+/
+    
+    # Look up the direction information if using "inbound" or "outbound" instead
+    # of a direction tag.
+    direction = bus.send(direction.downcase.to_sym).id if direction =~ /(inbound|outbound)/i      
+
+    # Create the stop object from the route, direction, and tag.
+    stop = Muni::Stop.new({:route_tag => route, :direction => direction, :tag => stop})
+
+    # Retrieve the predictions
+    begin
+      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

data/lib/muni.rb

@@ -0,0 +1,7 @@
+require 'muni/base'
+require 'muni/route'
+require "muni/version"
+
+module Muni
+
+end

data/lib/muni/base.rb

@@ -0,0 +1,27 @@
+require 'mash'
+module Muni
+  class NextBusError < StandardError; end
+  class Base < Mash
+    class << self
+      private
+        def fetch(command, options = nil)
+          url = build_url(command, options)
+          xml = Net::HTTP.get(URI.parse(url))
+          doc = XmlSimple.xml_in(xml)
+          fail NextBusError, doc['Error'].first['content'].gsub(/\n/,'') if doc['Error']
+          doc
+        end
+      
+        def build_url(command, options = nil)
+          url = "http://webservices.nextbus.com/service/publicXMLFeed?command=%s&a=sf-muni" %  command
+          if options
+            options.each { |key,value|
+              url << "&#{key}=#{value}"
+            }
+          end
+          url
+        end
+        
+    end
+  end
+end

data/lib/muni/direction.rb

@@ -0,0 +1,12 @@
+require 'amatch'
+
+module Muni
+  class Direction < Base
+    def stop_at(place)
+      pattern = Amatch::Sellers.new(place) 
+      stops.sort_by{ |stop|
+        pattern.match(stop.title)
+      }.first
+    end
+  end
+end    

data/lib/muni/prediction.rb

@@ -0,0 +1,20 @@
+require 'action_view'
+
+module Muni
+  class Prediction < Base
+    include ActionView::Helpers::DateHelper
+    def time
+      Time.at((epochTime.to_i / 1000.0).to_i)
+    end
+    
+    def pretty_time
+      if time > Time.now
+        distance_of_time_in_words_to_now(time)
+      elsif time < Time.now
+        time_ago_in_words(time)
+      else
+        "Arriving"
+      end
+    end
+  end
+end

data/lib/muni/route.rb

@@ -0,0 +1,73 @@
+require 'net/http'
+require 'xmlsimple'
+
+require 'muni/stop'
+require 'muni/direction'
+
+module Muni
+  class Route < Base
+    def outbound
+      directions.select{|dir| dir.name =~ /outbound/i}.first
+    end
+
+    def inbound
+      directions.select{|dir| dir.name =~ /inbound/i}.first
+    end
+
+    class << self
+      def find(tag)
+        if tag == :all
+          find_all
+        else
+          find_by_tag(tag)               
+        end
+      end
+      
+      private
+        def find_all
+          document = fetch(:routeList)
+          document['route'].collect do |el|
+            Route.new(el)
+          end
+        end
+        
+        def find_by_tag(tag)
+          document = fetch(:routeConfig, {:r => tag})
+          route = Route.new({:tag => document['route'].first['tag'], :title => document['route'].first['title']})
+          
+          stops = {}
+          
+          document['route'].first['stop'].each do |stop|
+            st = Stop.new({
+              :tag => stop['tag'],
+              :title => stop['title'],
+              :lat => stop['lat'],
+              :lon => stop['lat'],
+              :stopId => stop['lat'],                            
+            })
+            stops[st[:tag]] = st
+          end
+          
+          directions = []
+          route.directions = document['route'].first['direction'].collect do |direction|
+            direction_stops = direction['stop'].collect do |stop|
+              stops[stop['tag']]
+            end
+            
+            direction_stops.each do |stop|
+              stop.route_tag = route.tag
+              stop.direction = direction['tag']
+            end
+
+            Direction.new({
+                :id => direction['tag'],
+                :name => direction['title'],
+                :stops => direction_stops
+            })
+          end
+          
+          route
+        end
+    end
+  end
+end

data/lib/muni/stop.rb

@@ -0,0 +1,13 @@
+require 'muni/base'
+require 'muni/prediction'
+
+module Muni
+  class Stop < Base
+    def predictions
+      document = Stop.send(:fetch, :predictions, {:r => route_tag, :d => direction, :s => tag})
+      document['predictions'].first['direction'].first['prediction'].collect do |pred|
+        Prediction.new(pred)
+      end
+    end
+  end
+end

data/lib/muni/version.rb

@@ -0,0 +1,3 @@
+module Muni
+  VERSION = "0.0.1"
+end

data/muni.gemspec

@@ -0,0 +1,28 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "muni/version"
+
+Gem::Specification.new do |s|
+  s.name        = "muni"
+  s.version     = Muni::VERSION
+  s.authors     = ["Jeff Remer"]
+  s.email       = ["jeff@threestarchina.com"]
+  s.homepage    = "http://jeffremer.com/muni"
+  s.summary     = %q{NextBus San Francisco Muni API Client}
+  s.description = %q{A simple NextBus API client for retrieving San Francisco Muni bus route and stop prediction information}
+
+  s.rubyforge_project = "muni"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency "mash"
+  s.add_dependency "xml-simple"
+  s.add_dependency "actionpack"
+  s.add_dependency "amatch"
+  # For the executable
+  s.add_dependency "thor"   
+  s.add_development_dependency "rspec", "~> 2.0.0.beta.22"
+end

data/spec/route_spec.rb

@@ -0,0 +1,12 @@
+require 'muni/base'
+require 'muni/route'
+
+describe "Muni::Route" do
+  it "fetches all available routes" do   
+    Muni::Route.find(:all)
+  end
+  
+  it "fetches routes by tag" do
+    Muni::Route.find(21)
+  end
+end

data/spec/stop_spec.rb

@@ -0,0 +1,14 @@
+require 'muni/base'
+require 'muni/route'
+
+describe "Muni::Stop" do
+  it "#times fetches predictions" do   
+    route = Muni::Route.find(21)
+    direction = route.directions.first
+    stop = direction.stops.first
+    times = stop.predictions
+    times.each do |time|
+      p time.pretty_time
+    end        
+  end
+end

metadata

@@ -0,0 +1,132 @@
+--- !ruby/object:Gem::Specification
+name: muni
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Jeff Remer
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-05-31 00:00:00.000000000 -07:00
+default_executable: 
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mash
+  requirement: &2152729760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2152729760
+- !ruby/object:Gem::Dependency
+  name: xml-simple
+  requirement: &2152729320 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2152729320
+- !ruby/object:Gem::Dependency
+  name: actionpack
+  requirement: &2152728900 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2152728900
+- !ruby/object:Gem::Dependency
+  name: amatch
+  requirement: &2152728460 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2152728460
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: &2152728000 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2152728000
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &2152727420 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.0.beta.22
+  type: :development
+  prerelease: false
+  version_requirements: *2152727420
+description: A simple NextBus API client for retrieving San Francisco Muni bus route
+  and stop prediction information
+email:
+- jeff@threestarchina.com
+executables:
+- muni
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- bin/muni
+- lib/muni.rb
+- lib/muni/base.rb
+- lib/muni/direction.rb
+- lib/muni/prediction.rb
+- lib/muni/route.rb
+- lib/muni/stop.rb
+- lib/muni/version.rb
+- muni.gemspec
+- spec/route_spec.rb
+- spec/stop_spec.rb
+has_rdoc: true
+homepage: http://jeffremer.com/muni
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: muni
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: NextBus San Francisco Muni API Client
+test_files:
+- spec/route_spec.rb
+- spec/stop_spec.rb