mbta 0.1.0 → 0.1.1

data/.gitignore

@@ -3,6 +3,7 @@
 .bundle
 .config
 .yardoc
+.DS_Store
 Gemfile.lock
 InstalledFiles
 _yardoc

data/lib/mbta.rb

@@ -1,191 +1,13 @@
-require 'httparty'
-require 'json'
 require 'csv'
-require 'pp'
 
-require "mbta/version"
+require './mbta/version'
 
-module MBTA
-  module HeavyRail
-    module Announcements
-      class Base
-        attr_reader :trip, :platform, :time
-
-        def initialize(trip, platform, time)
-          @trip = trip
-          @platform = platform
-          @time = time
-        end
-
-        def type
-          self.class.name.split('::').last.downcase.to_sym
-        end
-      end
-
-      class Prediction < Base
-      end
-
-      class Arrival < Base
-      end
-    end
-
-    class Line
-      attr_reader :name, :color
-      attr_writer :stations
-      attr_accessor :trips
-
-      def initialize(name, stations = {})
-        @name = name
-        @color = name.downcase.to_sym
-        @stations = stations
-        @trips = {}
-      end
-
-      def stations(name_or_direction = nil)
-        case name_or_direction
-          when String then self[name_or_direction]
-          when Symbol then nil
-          else @stations
-        end
-      end
-
-      def [](name_or_key)
-        case name_or_key
-          when String then @stations.find { |station| station.name == name_or_key }
-          when Symbol then platforms.find { |platform| platform.key == name_or_key }
-        end
-      end
-
-      def platforms
-        stations.inject(Array.new) { |memo, station| memo << station.platforms }.flatten
-      end
-
-      def parse_announcements(json)
-        data = JSON.parse(json)
-        _trips = data.inject(Array.new) { |memo, announcement| memo << announcement['Trip'].to_i }.uniq
-        _trips.each do |trip|
-          self.trips[trip] = MBTA::HeavyRail::Trip.new(self, trip)
-        end
-
-        data.each do |a|
-          trip = self.trips[a['Trip'].to_i]
-          time = Time.strptime(a['Time'], '%m/%d/%Y %H:%M:%S %p')
-          platform = self[a['PlatformKey'].downcase.to_sym]
-
-          announcement_type = case a['InformationType']
-            when 'Arrived' then MBTA::HeavyRail::Announcements::Arrival
-            when 'Predicted' then MBTA::HeavyRail::Announcements::Prediction
-          end
-
-          trip.announcements << announcement_type.new(trip, platform, time) if announcement_type
-        end
-      end
-    end
-
-    class Trip
-      attr_reader :line, :id
-      attr_accessor :announcements
-
-      def initialize(line, id)
-        @line = line
-        @id = id
-        @announcements = []
-      end
-    end
-
-    class Station
-      attr_reader :line, :name
-      attr_writer :platforms
+require './mbta/commuter_rail'
+require './mbta/heavy_rail'
+require './mbta/bus'
 
-      def initialize(line, name, platforms = [])
-        @line = line
-        @name = name
-        @platforms = platforms
-      end
-
-      def platforms(direction = nil)
-        if (direction)
-          self[direction]
-        else
-          @platforms
-        end
-      end
-
-      def [](direction)
-        @platforms.find { |platform| platform.direction == direction }
-      end
-    end
-
-    class Platform
-      attr_reader :station, :direction, :key
-
-      def initialize(station, direction, order, key)
-        @station = station
-        @direction = direction
-        @order = order
-        @key = key
-      end
-
-      def predictions
-        station.line.trips.inject(Array.new) do |memo, trip|
-          memo << trip[1].announcements.inject(Array.new) do |m, a|
-            m << a if (self.key == a.platform.key && a.type == :prediction)
-            m
-          end
-        end.flatten.compact.sort {|x, y| x.time <=> y.time }
-      end
-    end
-
-    module System
-      class Map
-        attr_reader :lines
-
-        def initialize(lines = {})
-          @lines = lines
-        end
-      end
-
-      class Builder
-        def initialize
-          @data = CSV.read("../data/RealTimeHeavyRailKeys.csv", { headers: true })
-        end
-
-        def map
-          data = @data
-          line_names = data.inject(Array.new) {|memo, platform| memo.push(platform['Line']) }.uniq
-          lines = line_names.map{|name| MBTA::HeavyRail::Line.new(name) }
-
-          lines.each do |line|
-            platforms = data.select {|row| row['Line'] == line.name }
-            station_names = platforms.inject(Array.new) {|memo, platform| memo.push(platform['StationName']) }.uniq
-            line.stations = station_names.map {|name| MBTA::HeavyRail::Station.new(line, name.capitalize) }
-
-            line.stations.each do |station|
-              data.select{|row| row['StationName'].capitalize == station.name }.each do |row|
-                direction = case row['Direction']
-                  when 'NB' then :northbound
-                  when 'SB' then :southbound
-                  when 'EB' then :eastbound
-                  when 'WB' then :westbound
-                end
-                key = row['PlatformKey'].downcase.to_sym
-                station.platforms.push(MBTA::HeavyRail::Platform.new(station, direction, row['PlatformOrder'], key))
-              end
-            end
-          end
-
-          _lines = lines.inject(Hash.new){|lines, line| lines[line.name] = line; lines }
-
-          map = MBTA::HeavyRail::System::Map.new(_lines)
-          return map
-        end
-      end
-
-      MAP = Builder.new.map
-    end
-  end
-
-  BlueLine = MBTA::HeavyRail::System::MAP.lines['Blue']
-  GreenLine = MBTA::HeavyRail::System::MAP.lines['Green']
-  RedLine = MBTA::HeavyRail::System::MAP.lines['Red']
+module MBTA
+  include MBTA::CommuterRail
+  include MBTA::HeavyRail
+  include MBTA::Bus
 end

data/lib/mbta/bus.rb

@@ -0,0 +1,4 @@
+module MBTA
+  module Bus
+  end
+end

data/lib/mbta/commuter_rail.rb

@@ -0,0 +1,4 @@
+module MBTA
+  module CommuterRail
+  end
+end

data/lib/mbta/heavy_rail.rb

@@ -0,0 +1,11 @@
+require './mbta/heavy_rail/system'
+
+module MBTA
+  module HeavyRail
+    include MBTA::HeavyRail::System
+
+    BlueLine = MBTA::HeavyRail::System::MAP.lines['Blue']
+    OrangeLine = MBTA::HeavyRail::System::MAP.lines['Orange']
+    RedLine = MBTA::HeavyRail::System::MAP.lines['Red']
+  end
+end

data/lib/mbta/heavy_rail/announcements/arrival.rb

@@ -0,0 +1,8 @@
+module MBTA
+  module HeavyRail
+    module Announcements
+      class Arrival < Base
+      end
+    end
+  end
+end

data/lib/mbta/heavy_rail/announcements/base.rb

@@ -0,0 +1,19 @@
+module MBTA
+  module HeavyRail
+    module Announcements
+      class Base
+        attr_reader :trip, :platform, :time
+
+        def initialize(trip, platform, time)
+          @trip = trip
+          @platform = platform
+          @time = time
+        end
+
+        def type
+          self.class.name.split('::').last.downcase.to_sym
+        end
+      end
+    end
+  end
+end

data/lib/mbta/heavy_rail/announcements/prediction.rb

@@ -0,0 +1,8 @@
+module MBTA
+  module HeavyRail
+    module Announcements
+      class Prediction < Base
+      end
+    end
+  end
+end

data/lib/mbta/heavy_rail/line.rb

@@ -0,0 +1,56 @@
+module MBTA
+  module HeavyRail
+    class Line
+      attr_reader :name, :color
+      attr_writer :stations
+      attr_accessor :trips
+
+      def initialize(name, stations = {})
+        @name = name
+        @color = name.downcase.to_sym
+        @stations = stations
+        @trips = {}
+      end
+
+      def stations(name_or_direction = nil)
+        case name_or_direction
+          when String then self[name_or_direction]
+          when Symbol then nil
+          else @stations
+        end
+      end
+
+      def [](name_or_key)
+        case name_or_key
+          when String then @stations.find { |station| station.name == name_or_key }
+          when Symbol then platforms.find { |platform| platform.key == name_or_key }
+        end
+      end
+
+      def platforms
+        stations.inject(Array.new) { |memo, station| memo << station.platforms }.flatten
+      end
+
+      def parse_announcements(json)
+        data = JSON.parse(json)
+        _trips = data.inject(Array.new) { |memo, announcement| memo << announcement['Trip'].to_i }.uniq
+        _trips.each do |trip|
+          self.trips[trip] = MBTA::HeavyRail::Trip.new(self, trip)
+        end
+
+        data.each do |a|
+          trip = self.trips[a['Trip'].to_i]
+          time = Time.strptime(a['Time'], '%m/%d/%Y %H:%M:%S %p')
+          platform = self[a['PlatformKey'].downcase.to_sym]
+
+          announcement_type = case a['InformationType']
+            when 'Arrived' then MBTA::HeavyRail::Announcements::Arrival
+            when 'Predicted' then MBTA::HeavyRail::Announcements::Prediction
+          end
+
+          trip.announcements << announcement_type.new(trip, platform, time) if announcement_type
+        end
+      end
+    end
+  end
+end

data/lib/mbta/heavy_rail/platform.rb

@@ -0,0 +1,23 @@
+module MBTA
+  module HeavyRail
+    class Platform
+      attr_reader :station, :direction, :key
+
+      def initialize(station, direction, order, key)
+        @station = station
+        @direction = direction
+        @order = order
+        @key = key
+      end
+
+      def predictions
+        station.line.trips.inject(Array.new) do |memo, trip|
+          memo << trip[1].announcements.inject(Array.new) do |m, a|
+            m << a if (self.key == a.platform.key && a.type == :prediction)
+            m
+          end
+        end.flatten.compact.sort {|x, y| x.time <=> y.time }
+      end
+    end
+  end
+end

data/lib/mbta/heavy_rail/station.rb

@@ -0,0 +1,30 @@
+module MBTA
+  module HeavyRail
+    class Station
+      attr_reader :line, :name
+      attr_writer :platforms
+
+      def initialize(line, name, platforms = [])
+        @line = line
+        @name = name
+        @platforms = platforms
+      end
+
+      def inspect
+        %("#{@name}")
+      end
+
+      def platforms(direction = nil)
+        if (direction)
+          self[direction]
+        else
+          @platforms
+        end
+      end
+
+      def [](direction)
+        @platforms.find { |platform| platform.direction == direction }
+      end
+    end
+  end
+end

data/lib/mbta/heavy_rail/system.rb

@@ -0,0 +1,55 @@
+require './mbta/heavy_rail/line'
+require './mbta/heavy_rail/station'
+require './mbta/heavy_rail/platform'
+
+module MBTA
+  module HeavyRail
+    module System
+      class Map
+        attr_reader :lines
+
+        def initialize(lines = {})
+          @lines = lines
+        end
+      end
+
+      class Builder
+        def initialize
+          @data = ::CSV.read("../data/RealTimeHeavyRailKeys.csv", { headers: true })
+        end
+
+        def map
+          data = @data
+          line_names = data.inject(Array.new) {|memo, platform| memo.push(platform['Line']) }.uniq
+          lines = line_names.map{|name| MBTA::HeavyRail::Line.new(name) }
+
+          lines.each do |line|
+            platforms = data.select {|row| row['Line'] == line.name }
+            station_names = platforms.inject(Array.new) {|memo, platform| memo.push(platform['StationName']) }.uniq
+            line.stations = station_names.map {|name| MBTA::HeavyRail::Station.new(line, name.capitalize) }
+
+            line.stations.each do |station|
+              data.select{|row| row['StationName'].capitalize == station.name }.each do |row|
+                direction = case row['Direction']
+                  when 'NB' then :northbound
+                  when 'SB' then :southbound
+                  when 'EB' then :eastbound
+                  when 'WB' then :westbound
+                end
+                key = row['PlatformKey'].downcase.to_sym
+                station.platforms.push(MBTA::HeavyRail::Platform.new(station, direction, row['PlatformOrder'], key))
+              end
+            end
+          end
+
+          _lines = lines.inject(Hash.new){|lines, line| lines[line.name] = line; lines }
+
+          map = MBTA::HeavyRail::System::Map.new(_lines)
+          return map
+        end
+      end
+
+      MAP = Builder.new.map
+    end
+  end
+end

data/lib/mbta/heavy_rail/trip.rb

@@ -0,0 +1,14 @@
+module MBTA
+  module HeavyRail
+    class Trip
+      attr_reader :line, :id
+      attr_accessor :announcements
+
+      def initialize(line, id)
+        @line = line
+        @id = id
+        @announcements = []
+      end
+    end
+  end
+end

data/lib/mbta/version.rb

@@ -1,3 +1,3 @@
 module MBTA
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/mbta.gemspec

@@ -2,16 +2,18 @@
 require File.expand_path('../lib/mbta/version', __FILE__)
 
 Gem::Specification.new do |gem|
-  gem.authors       = ["Chris Kalafarski"]
-  gem.email         = ["chris@farski.com"]
+  gem.authors       = ['Chris Kalafarski']
+  gem.email         = ['chris@farski.com']
   gem.description   = %q{Library for accessing real-time MBTA data}
   gem.summary       = %q{Not a product of the MBTA. See mbta.com/developers for more info about their API}
-  gem.homepage      = "https://github.com/farski/mbta"
+  gem.homepage      = 'https://github.com/farski/mbta'
 
   gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
   gem.files         = `git ls-files`.split("\n")
   gem.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
-  gem.name          = "mbta"
-  gem.require_paths = ["lib"]
+  gem.name          = 'mbta'
+  gem.require_paths = ['lib']
   gem.version       = MBTA::VERSION
+
+  gem.add_dependency('httparty', '>= 0.8.0')
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mbta
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,8 +9,19 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-01-24 00:00:00.000000000 Z
-dependencies: []
+date: 2012-02-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: &70218718288260 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70218718288260
 description: Library for accessing real-time MBTA data
 email:
 - chris@farski.com
@@ -18,15 +29,25 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- .DS_Store
 - .gitignore
 - Gemfile
 - LICENSE
 - README.md
 - Rakefile
-- data/.DS_Store
 - data/RealTimeHeavyRailKeys.csv
 - lib/mbta.rb
+- lib/mbta/bus.rb
+- lib/mbta/commuter_rail.rb
+- lib/mbta/heavy_rail.rb
+- lib/mbta/heavy_rail/announcements.rb
+- lib/mbta/heavy_rail/announcements/arrival.rb
+- lib/mbta/heavy_rail/announcements/base.rb
+- lib/mbta/heavy_rail/announcements/prediction.rb
+- lib/mbta/heavy_rail/line.rb
+- lib/mbta/heavy_rail/platform.rb
+- lib/mbta/heavy_rail/station.rb
+- lib/mbta/heavy_rail/system.rb
+- lib/mbta/heavy_rail/trip.rb
 - lib/mbta/version.rb
 - mbta.gemspec
 homepage: https://github.com/farski/mbta