fonecal 0.1.0

data/fonecal.gemspec

@@ -0,0 +1,30 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'fonecal/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "fonecal"
+  spec.version       = Fonecal::VERSION
+  spec.authors       = ["Martin Madsen"]
+  spec.email         = ["martin2madsen@gmail.com"]
+  spec.description   = %q{f1cal generates an iCalendar file for the 2014 Formula 1 season.}
+  spec.summary       = %q{Crawls the Grand Prix list at formula1.com and builds an iCalendar file with events in UTC}
+  spec.homepage      = "https://github.com/Fapper/f1cal"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake", "~> 10.1"
+  spec.add_development_dependency "rspec", "~> 2.14"
+
+  spec.add_dependency "timezone", "~> 0.3"
+  spec.add_dependency "activesupport", "~> 4.0"
+  spec.add_dependency 'nokogiri', "~> 1.6"
+  spec.add_dependency 'geocoder', "~> 1.1"
+  spec.add_dependency 'i18n', "~> 0.6"
+end

data/lib/fonecal/calendar.rb

@@ -0,0 +1,40 @@
+module Fonecal
+  class Calendar
+    attr_accessor :calendar
+
+    def initialize(gps)
+      self.calendar = File.new('f1cal.ical', 'w')
+
+      @calendar.puts "BEGIN:VCALENDAR"
+      @calendar.puts "VERSION:2.0"
+      @calendar.puts "PRODID:-//fonecal//NONSGML v1.0//EN"
+      @calendar.puts ""
+
+      gps.each do |gp|
+        gp.events.each do |event|
+          addEvent(gp, event)
+        end
+      end
+
+      @calendar.puts "END:VCALENDAR"
+      @calendar.close
+    end
+
+    private
+
+    def addEvent(gp, event)
+      @calendar.puts "BEGIN:VEVENT"
+
+      @calendar.puts "SUMMARY:#{gp.grandPrix}: #{event.name}"
+      @calendar.puts "DESCRIPTION:#{gp.raceTitle}: #{event.name}"
+      @calendar.puts "DTSTAMP:#{Util.dtToUTC(DateTime.now)}Z"
+      @calendar.puts "LOCATION:#{gp.city}, #{gp.country}"
+      @calendar.puts "DTSTART:#{Util.dtToUTC(event.start)}Z"
+      puts "#{gp.grandPrix}: #{event.name} starts #{event.start} UTC: #{event.start.utc}"
+      @calendar.puts "DTEND:#{Util.dtToUTC(event.end)}Z"
+
+      @calendar.puts "END:VEVENT"
+      @calendar.puts ""
+    end
+  end
+end

data/lib/fonecal/calendar_crawler.rb

@@ -0,0 +1,12 @@
+module Fonecal
+  class CalendarCrawler < Crawler
+    def eventLinks
+      eventLinks = []
+
+      @site.css('table.raceCalender tr.row').each do |event|
+        eventLinks << @root + event.css('td.raceLocation a').attr('href')
+      end
+      eventLinks
+    end
+  end
+end

data/lib/fonecal/calendar_spec.rb

@@ -0,0 +1,5 @@
+require 'fonecal'
+
+describe Fonecal::Calendar do
+
+end

data/lib/fonecal/circuit_info.rb

@@ -0,0 +1,23 @@
+module Fonecal
+  class CircuitInfo
+    def initialize(info)
+      @info = info
+    end
+
+    def name
+      if @info['Circuit Name:']
+        @info['Circuit Name:'].cap 
+      else
+        ""
+      end
+    end
+
+    def city
+      @info['City:'].cap
+    end
+
+    def laps
+      @info['Number of Laps:'].to_i
+    end
+  end
+end

data/lib/fonecal/crawler.rb

@@ -0,0 +1,13 @@
+require 'open-uri'
+require 'nokogiri'
+
+module Fonecal
+  class Crawler
+    attr_accessor :site, :root
+
+    def initialize(site)
+      self.site = Nokogiri::HTML(open(site))
+      self.root = Util.getRootUrl(site)
+    end
+  end
+end

data/lib/fonecal/event.rb

@@ -0,0 +1,25 @@
+require 'active_support/all'
+
+module Fonecal
+  class Event
+    attr_accessor :name, :start, :end 
+
+    def initialize(event, timezone)
+      # UTC offset in hours, explicitely add +
+      if (timezone.utc_offset / 60 / 60) > 0
+        offset = "+#{timezone.utc_offset / 60 / 60}"
+      else
+        offset = offset.to_s
+      end
+      
+      self.name = event[:type]
+      self.start = DateTime.parse("#{event[:date]}T#{event[:start]} #{offset}")
+
+      if event[:end]
+        self.end = DateTime.parse("#{event[:date]}T#{event[:end]} #{offset}") 
+      else
+        self.end = self.start + 4.hours
+      end
+    end
+  end
+end

data/lib/fonecal/event_crawler.rb

@@ -0,0 +1,60 @@
+module Fonecal
+  class EventCrawler < Crawler
+    def gp
+      @site.css('div.raceResultsHeading').text.strip
+    end
+
+    def circuitInfo
+      ci = {}
+
+      @site.css('div.circuitInfoBox table tr').each do |row|
+        ci[row.css('td')[0].content.strip] = row.css('td')[1].content.strip
+      end
+
+      city = @site.css('h3.inDetailCircuitName').text
+
+      ci['City:'] = city
+
+      return ci
+    end
+
+    def timeTables
+      tables = []
+      @site.css('div#ctl00_ContentSub_Timetable1_subModuleContentDiv table').each do |tt|
+        tables << parseTimeTable(tt)
+      end
+
+      tables
+    end
+
+    private
+
+    def sessionData(row)
+      inf = {}
+
+      sessData = row.css('td')
+      inf[:type] = sessData[0].text.strip
+      inf[:start] = sessData[1].css('span')[0].text.strip
+      inf[:end] = sessData[1].css('span')[1].text.strip if sessData[1].css('span')[1]
+      inf
+    end
+
+    def parseTimeTable(table)
+      info = {}
+
+      rows = table.css 'tr'
+
+      # The date is the first row in teh table
+      info[:date] = rows[0].text.strip
+
+      # Now extract info from each of the remaining rows
+      # Example: Practice 1 12:30 - 14:00 is stored in a row
+      info[:sessions] ||= []
+      rows[1..-1].each do |row|
+        info[:sessions] << sessionData(row)
+      end
+
+      info
+    end
+  end
+end

data/lib/fonecal/fonecal_spec.rb

@@ -0,0 +1,7 @@
+module Fonecal
+  describe Fonecal::Fonecal do
+    it "creates an icalendar file" do
+      Fonecal.create_ical
+    end
+  end
+end

data/lib/fonecal/grand_prix.rb

@@ -0,0 +1,70 @@
+require 'geocoder'
+require 'timezone'
+
+module Fonecal
+  class GrandPrix
+    attr_accessor :events, :circuit
+
+    def initialize(eventWebsite)
+      @crawler = EventCrawler.new(eventWebsite)
+      self.circuit= CircuitInfo.new(@crawler.circuitInfo)
+      self.events = []
+
+      createEvents
+    end
+
+    def location
+      if @res ||= Geocoder.search("#{@circuit.name}, #{@circuit.city}").first
+        @res
+      elsif @res ||= Geocoder.search("#{@circuit.city}").first
+        @res
+      elsif res ||= Geocoder.search("#{@circuit.name}").first
+        @res
+      else
+        @res
+      end
+
+      #@location ||= Geocoder.search("#{@circuit.name}, #{@circuit.city}").first
+    end
+
+    def city
+      @circuit.city
+    end
+
+    def country
+      location.country
+    end
+
+    def timezone
+      @timezone ||= Timezone::Zone.new :latlon => location.geometry["location"].values
+    end
+
+    def raceTitle 
+      @crawler.gp.gsub(/\w+/).map(&:capitalize).join(' ')
+    end
+
+    def grandPrix
+      "#{country} GP"
+      # Belgian Grand Prix
+    end
+
+    private
+
+    def createEvents
+      @crawler.timeTables.each do |day|
+        day[:sessions].each do |event|
+        type = event[:type]
+          event[:date] = day[:date]
+
+          if(type.include? "Practice")
+            @events << Practice.new(event, timezone)
+          elsif(type.include? "Qualifying")
+            @events << Qualifying.new(event, timezone)
+          elsif(type.include? "Race")
+            @events << Race.new(event, timezone)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fonecal/practice.rb

@@ -0,0 +1,4 @@
+module Fonecal
+  class Practice < Event
+  end
+end

data/lib/fonecal/qualifying.rb

@@ -0,0 +1,4 @@
+module Fonecal
+  class Qualifying < Event 
+  end
+end

data/lib/fonecal/race.rb

@@ -0,0 +1,4 @@
+module Fonecal
+  class Race < Event
+  end
+end

data/lib/fonecal/session.rb

@@ -0,0 +1,10 @@
+module Fonecal
+  class Session
+    attr_accessor :start, :length
+
+    def initilaize(start, length)
+      @start = start
+      @length = length
+    end
+  end
+end

data/lib/fonecal/util.rb

@@ -0,0 +1,26 @@
+require 'timezone'
+require 'active_support/all'
+
+module Fonecal
+  class Util
+    def self.getRootUrl(site)
+      site.match('.+\.com').to_s
+    end
+
+    def self.dtToUTC(dt)
+      #dt.with_offset(0).strftime("%Y%m%dT%H%M%S")
+      dt.utc.strftime("%Y%m%dT%H%M%S")
+    end
+  end
+
+end
+
+class String
+  def cap
+    self.gsub(/\w+/).map(&:capitalize).join(' ')
+  end
+end
+
+Timezone::Configure.begin do |c|
+    c.username = 'fonecal'
+end

data/lib/fonecal/version.rb

@@ -0,0 +1,3 @@
+module Fonecal
+  VERSION = "0.1.0"
+end

data/lib/fonecal.rb

@@ -0,0 +1,25 @@
+require 'fonecal/version'
+require 'fonecal/util'
+require 'fonecal/crawler'
+require 'fonecal/event_crawler'
+require 'fonecal/calendar_crawler'
+require 'fonecal/calendar'
+require 'fonecal/grand_prix'
+require 'fonecal/event'
+require 'fonecal/session'
+require 'fonecal/practice'
+require 'fonecal/qualifying'
+require 'fonecal/race'
+require 'fonecal/circuit_info'
+
+module Fonecal
+  def self.create_ical
+    gps = []
+
+    Fonecal::CalendarCrawler.new("http://www.formula1.com/races/calendar.html").eventLinks.each do |l|
+      gps << Fonecal::GrandPrix.new(l)
+    end
+
+    Fonecal::Calendar.new(gps)
+  end
+end

data/spec/calendar_crawler_spec.rb

@@ -0,0 +1,19 @@
+require 'fonecal'
+
+describe Fonecal::CalendarCrawler do
+  let(:crawler) { Fonecal::CalendarCrawler.new("http://www.formula1.com/races/calendar.html") }
+
+  it "successfully connects to the website" do
+    crawler.site.should_not be_nil
+  end
+
+  describe "#eventLinks" do
+    it "returns 19 grand prix links" do
+      crawler.eventLinks.count.should eql(19)
+    end
+
+    it "has the full url" do
+      crawler.eventLinks.first.should eql('http://www.formula1.com/races/in_detail/australia_914/circuit_diagram.html')
+    end
+  end
+end

data/spec/event_crawler_spec.rb

@@ -0,0 +1,68 @@
+require 'fonecal'
+
+describe Fonecal::EventCrawler do
+  let(:crawler) { Fonecal::EventCrawler.new("http://www.formula1.com/races/in_detail/australia_914/circuit_diagram.html") }
+
+  it "successfully connects to the website" do
+    crawler.site.should_not be_nil
+  end
+
+  describe "2014 Australian GP" do
+    describe "#circuitInfo" do
+      it "fetches a circuit name" do
+        crawler.circuitInfo['Circuit Name:'].should eq 'Albert Park'
+      end
+
+      it "fetches a date" do
+        crawler.circuitInfo['Race Date:'].should_not be_nil
+      end
+
+      it "fetches a city" do
+        crawler.circuitInfo['City:'].should eq 'Melbourne'
+      end
+    end
+
+    describe "#gp" do
+      it "should fetch the GP name" do
+        crawler.gp.should eq '2014 FORMULA 1 ROLEX AUSTRALIAN GRAND PRIX'
+      end
+    end
+
+    describe "#timeTables" do
+      it "should get three timetables" do
+        crawler.timeTables.count.should eq 3
+      end
+
+      it "should have data about practice 1" do
+        crawler.timeTables.first[:date].should eq 'Fri 14 March 2014'
+        crawler.timeTables.first[:sessions].first[:type].should eq 'Practice 1'
+        crawler.timeTables.first[:sessions].first[:start].should eq '12:30'
+      end
+
+      it "should have data about practice 2" do
+        crawler.timeTables.first[:date].should eq 'Fri 14 March 2014'
+        crawler.timeTables.first[:sessions].last[:type].should eq 'Practice 2'
+        crawler.timeTables.first[:sessions].last[:start].should eq '16:30'
+      end
+
+      it "should have data about practice 3" do
+        crawler.timeTables[1][:date].should eq 'Sat 15 March 2014'
+        crawler.timeTables[1][:sessions].first[:type].should eq 'Practice 3'
+        crawler.timeTables[1][:sessions].first[:start].should eq '14:00'
+      end
+
+      it "should have data about qualifying" do
+        crawler.timeTables[1][:date].should eq 'Sat 15 March 2014'
+        crawler.timeTables[1][:sessions].last[:type].should eq 'Qualifying'
+        crawler.timeTables[1][:sessions].last[:start].should eq '17:00'
+      end
+
+      it "should have data about the race" do
+        crawler.timeTables.last[:date].should eq 'Sun 16 March 2014'
+        crawler.timeTables.last[:sessions].first[:type].should eq 'Race'
+        crawler.timeTables.last[:sessions].first[:start].should eq '17:00'
+      end
+    end
+  end
+end
+

data/spec/event_spec.rb

@@ -0,0 +1,4 @@
+require 'fonecal'
+
+describe Fonecal::Event do
+end

data/spec/fonecal_spec.rb

@@ -0,0 +1,9 @@
+require 'fonecal'
+
+describe Fonecal do
+  describe "#create_ical" do
+    it "creates the file" do
+      Fonecal.create_ical
+    end
+  end
+end

data/spec/grand_prix_spec.rb

@@ -0,0 +1,31 @@
+require 'fonecal'
+
+describe Fonecal::GrandPrix do
+  describe "Getting Belgian GP info" do
+    let(:belgianGP) { Fonecal::GrandPrix.new("http://www.formula1.com/races/in_detail/belgium_927/circuit_diagram.html") }
+
+    it "should have the correct race title" do
+      belgianGP.raceTitle.should eql "2014 Formula 1 Shell Belgian Grand Prix"
+    end
+
+    it "should get a country" do
+      belgianGP.country.should eql "Belgium"
+    end
+
+    it "has location object" do
+      belgianGP.location.should_not be_nil
+    end
+
+    describe "its events" do
+      describe "Practice 1" do
+        it "should have the right name" do
+          belgianGP.events.first.name.should eql "Practice 1"
+        end
+
+        it "should have the correct start time" do
+          belgianGP.events.first.start.should eql DateTime.new(2014, 8, 22, 10, 0, 0, '+1')
+        end
+      end
+    end
+  end
+end

data/spec/race_spec.rb

@@ -0,0 +1,6 @@
+require 'fonecal'
+
+#describe Fonecal::Race do
+  #let(:event) { Fonecal::Race.new()  }
+
+#end

data/spec/spec_helper.rb

@@ -0,0 +1,17 @@
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+end

data/spec/util_test.rb

@@ -0,0 +1,10 @@
+require 'fonecal'
+
+describe Fonecal::Util do
+  describe "#getRootURL" do
+    it "extracts formula1.com from any url" do
+      calSite = "http://www.formula1.com/races/calendar.html"
+      Fonecal::Util.getRootUrl(calSite).should eql('http://www.formula1.com')
+    end
+  end
+end