check_late 0.0.1 → 0.0.5

data/bin/check_late

@@ -1,4 +1,15 @@
 #!/usr/bin/env ruby
 
 require 'check_late'
-CheckLate.hi
+require 'optparse'
+
+options = {}
+OptionParser.new do |opts|
+  opts.banner = "Usage: check_late -t train_number"
+
+  opts.on("-t num", "--train num", "Insert train number") do |num|
+    options[:train] = num
+  end
+end.parse!
+
+CheckLate.new(options[:train]).start!

data/lib/check_late/notifier.rb

@@ -1,5 +1,21 @@
-class CheckLate::Notifier
-  def notify
-    puts "notification!"
+require 'rubygems'
+require 'synack'
+
+
+class Notifier
+
+  def notify!(message)
+    if message and not message.empty?
+      puts message
+    end
   end
+
+  def synack!(message)
+    if message and not message.empty?
+      Synack::Client.new(:host => 'localhost', :port => '1234').say(message)
+      #`synack -p 1234 #{message}`
+    end
+  end
+
+
 end

data/lib/check_late/scraper.rb

@@ -0,0 +1,54 @@
+require 'rubygems'
+require 'mechanize'
+
+module Scraper
+  URL          = "http://mobile.viaggiatreno.it/viaggiatreno/mobile/numero/"
+  SEARCH_TYPE  = "numero"
+  LANG         = "IT"
+
+  def generate_request(train_number)
+    agent = Mechanize.new
+    response = agent.post( URL,
+                           {
+                             :numeroTreno => train_number,
+                             :tipoRicerca => SEARCH_TYPE,
+                             :lang        => LANG
+                           }
+                           )
+    return response.parser.xpath('//div[@class="evidenziato"]/strong')
+  end
+
+  def scrape!(train_number)
+    xpath = generate_request(train_number)
+    begin
+
+      if xpath.to_a.first.nil?
+        raise Exception
+      else
+        sanitize_text xpath.to_a.first.to_html
+      end
+
+    rescue Exception
+      puts "Invalid Train Number"
+    end
+
+  end
+
+  def sanitize_text(text)
+    text = text.gsub(/\r|\n/, '')
+    text = text.gsub(/\t+/, ' ')
+    text = text.gsub(/ +/, ' ')
+    text = text.gsub(/<!--.*-->/, '')
+    text = text.gsub(/<br>/, '')
+    text = text.gsub(/<strong>|<\/strong>/, '')
+    text = text.strip
+  end
+
+end
+
+
+# # Some useless variables to store data
+# delay        = text.match(/(\d+) minuti di ritardo/)
+# early        = text.match(/(\d+) minuti di anticipo/)
+# arrived      = text.match(/arrivato/)
+# not_departed = text.match(/non e' ancora partito/)

data/lib/check_late/version.rb

@@ -0,0 +1,3 @@
+module CheckLate
+  VERSION = "0.0.5"
+end

data/lib/check_late.rb

@@ -1,9 +1,29 @@
+require 'check_late/notifier'
+require 'check_late/scraper'
+
 class CheckLate
-  def self.hi
-    notifier = Notifier.new
-    notifier.notify
+  include Scraper
+  attr_accessor :train_number, :status, :old_status
+
+  def initialize(train_number)
+    @train_number = train_number
+    @notifier     = Notifier.new
   end
-end
 
+  def status
+    new = scrape!(train_number)
+    if @status != new
+      @status = new
+      return new
+    end
+  end
 
-require 'check_late/notifier'
+  def start!
+    while true
+      #@notifier.notify!(status)
+      @notifier.synack!(status)
+      sleep(60)
+    end
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: check_late
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -19,6 +19,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/check_late/notifier.rb
+- lib/check_late/scraper.rb
+- lib/check_late/version.rb
 - lib/check_late.rb
 - bin/check_late
 - README.md