ispusage 0.2.0 → 0.3.0

data/Rakefile

@@ -11,6 +11,10 @@ begin
     gem.homepage = "http://github.com/aussiegeek/ispusage"
     gem.authors = ["Alan Harper"]
     gem.add_development_dependency "rspec", ">= 1.2.9"
+    gem.add_dependency 'json'
+    gem.add_dependency 'nokogiri'
+    gem.add_dependency 'mechanize'
+    
     # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
   end
   Jeweler::GemcutterTasks.new

data/VERSION

@@ -1 +1 @@
-0.2.0
+0.3.0

data/lib/ispusage/fetchers/au_optus.rb

@@ -0,0 +1,48 @@
+require 'rubygems'
+require 'mechanize'
+
+class IspUsage
+  class Fetchers
+    class AUOptus < Fetcher
+      def initialize(options)
+        super(options)
+      end
+
+      def fetch_usage
+        agent = Mechanize.new
+
+        page = agent.get('https://my.optus.com.au/web/oscportal.portal?site=personal')
+        form = page.form('Login')
+        form['USER'] = self.username
+        form['PASSWORD'] = self.password
+
+        page = agent.submit(form, form.buttons.first)
+
+        account_no = page.search('h2').text.match(/\d+/)
+        services = page.search('ol.service_list')
+        services.each do |service|
+          phone_no = service.search('div>a').text.strip
+          page = agent.get("https://my.optus.com.au/web/oscportal.portal?_nfpb=true&_pageLabel=redirectPage&redirectTo=deeplink_myusage_postpaid&site=personal&pageName=unbilledUsage&virAcctNum=#{phone_no}&accNum=#{account_no}&_activeBA=#{account_no}&_activeSI=#{phone_no}")
+
+          plan_text = page.search("html>body>div>div>div:nth-of-type(3)>div>div:nth-of-type(2)>div:nth-of-type(4)>div>table tr:nth-of-type(5)>td>table tr:nth-of-type(2)>td>div>table tr:nth-of-type(2)>td:nth-of-type(2)").text
+
+          
+          downloads = UsagePeriod.new
+          self.usage_periods << downloads
+          
+          downloads.label = phone_no
+          downloads.quota = plan_text.match(/(\d+)MB/)[1].to_i
+          
+          usagesummary = page.search('table.usagesummary')
+          usagesummary.search('tr').each do |row|
+            row_name =  row.search('td[1]').text 
+            case row_name
+            when 'Data - Mobile Internet'
+              downloads.used = row.search('td[2]').text.match(/\d+\./)[0].to_i
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ispusage/fetchers/au_three.rb

@@ -0,0 +1,34 @@
+require 'rubygems'
+require 'mechanize'
+
+class IspUsage
+  class Fetchers
+    class AUThree < Fetcher
+      def initialize(options)
+        super(options)
+      end
+
+      def fetch_usage
+        agent = Mechanize.new
+        page = agent.get('https://www.my.three.com.au/My3/jfn')
+        login_form = page.form('login')
+        login_form.login = self.username
+        login_form.password = self.password
+        page = agent.submit(login_form, login_form.buttons.first)
+        pin_form = page.form('myForm')
+        if pin_form.nil?
+          puts "Authentication failure"
+          exit(1)
+        end
+        pin_form.pin = options[:pin]
+        page = agent.submit(pin_form, pin_form.buttons.first)
+        
+        downloads = UsagePeriod.new
+        self.usage_periods << downloads
+
+        downloads.quota = page.search("td[width='190px;']").text.strip.to_i
+        downloads.used = downloads.quota - page.search('br+strong.largeclass').text.strip.to_i
+      end
+    end
+  end
+end

data/lib/ispusage/fetchers/fetcher.rb

@@ -4,12 +4,12 @@ require 'json'
 class IspUsage
   class Fetchers
     class Fetcher
-      attr_accessor :usage_periods, :username, :password
+      attr_accessor :usage_periods, :username, :password, :options
       
       def initialize(options)
         @username = options[:username]
         @password = options[:password]
-        
+        self.options = options
         self.usage_periods = []
       end
       

data/lib/ispusage/fetchers.rb

@@ -1,3 +1,5 @@
 require 'ispusage/fetchers/fetcher'
+require 'ispusage/fetchers/au_optus'
+require 'ispusage/fetchers/au_three'
 require 'ispusage/fetchers/iinet'
 require 'ispusage/fetchers/internode'

data/lib/ispusage.rb

@@ -7,9 +7,10 @@ require 'json'
 class IspUsage
 
   ISPS = [
-    'AUBigpond',
     'AUIinet',
-    'AUInternode'
+    'AUInternode',
+    'AUOptus',
+    'AUThree'
   ]
 
   class Cli
@@ -20,6 +21,7 @@ class IspUsage
           opts.on('-i', '--isp isp', 'ISP') {|i| options[:isp] = i}
           opts.on('-u', '--username username', 'Username') {|u| options[:username] = u}
           opts.on('-p', '--password password', 'Password') {|p| options[:password] = p}
+          opts.on('-q', '--pin ', 'PIN') {|pin| options[:pin] = pin}
           opts.on('-j', '--json', 'JSON Output') {|j| options[:output] = :json}
         end
         optparse.parse!
@@ -35,7 +37,9 @@ class IspUsage
         fetcher.fetch_usage
         case options[:output]
         when :text
-          puts fetcher.used.to_s + '/' + fetcher.quota.to_s
+          fetcher.usage_periods.each do |usage_period|
+            puts usage_period.label + ':' + usage_period.used.to_s + '/' + usage_period.quota.to_s
+          end
         when :json
           puts fetcher.to_json
         end

data/spec/fetchers/au_optus_spec.rb

@@ -0,0 +1,34 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe IspUsage::Fetchers::AUOptus do
+  describe "single usage period" do
+    before do
+      signon = File.read(File.dirname(__FILE__) + '/../fixtures/au_optus/signon.html')
+      FakeWeb.register_uri(:get, 'https://my.optus.com.au/web/oscportal.portal?site=personal', :response => signon)
+      my_account = File.read(File.dirname(__FILE__) + '/../fixtures/au_optus/my_account.html')
+      FakeWeb.register_uri(:post, 'https://my.optus.com.au/signon/Optus/login_ext.sec', :response => my_account)
+      my_usage = File.read(File.dirname(__FILE__) + '/../fixtures/au_optus/my_usage.html')
+      FakeWeb.register_uri(:get, 'https://my.optus.com.au/web/oscportal.portal?_nfpb=true&_pageLabel=redirectPage&redirectTo=deeplink_myusage_postpaid&site=personal&pageName=unbilledUsage&virAcctNum=0123456789&accNum=01234567890123&_activeBA=01234567890123&_activeSI=0123456789', :response => my_usage)
+
+      options = {:username => 'user', :password => 'password'}
+      @usage = IspUsage::Fetchers::AUOptus.new(options)
+      @usage.fetch_usage
+    end
+
+    it "should return usage correctly" do
+      @usage.usage_periods.first.used.should == 151
+    end
+
+    it "should return quota correctly" do
+      @usage.usage_periods.first.quota.should == 500
+    end
+
+    it "should return total correctly" do
+      @usage.usage_periods.first.total.should == 500
+    end
+    
+    it "should set label to phone number" do
+      @usage.usage_periods.first.label.should == "0123456789"
+    end
+  end
+end

data/spec/fetchers/au_three_spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe IspUsage::Fetchers::AUThree do
+  describe "single usage period" do
+    before do
+      signon = File.read(File.dirname(__FILE__) + '/../fixtures/au_three/signon.html')
+      FakeWeb.register_uri(:get, 'https://www.my.three.com.au/My3/jfn', :response => signon)
+      pin = File.read(File.dirname(__FILE__) + '/../fixtures/au_three/pin.html')
+      my_usage = File.read(File.dirname(__FILE__) + '/../fixtures/au_three/usage.html')
+      FakeWeb.register_uri(:post, 'https://www.my.three.com.au/My3/jfn', [{:response => pin, :times => 1}, {:response => my_usage}])
+
+      options = {:username => 'user', :password => 'password', :pin => '1234'}
+      @usage = IspUsage::Fetchers::AUThree.new(options)
+      @usage.fetch_usage
+    end
+
+    it "should return usage correctly" do
+      @usage.usage_periods.first.used.should == 61
+    end
+
+    it "should return quota correctly" do
+      @usage.usage_periods.first.quota.should == 500
+    end
+
+    it "should return total correctly" do
+      @usage.usage_periods.first.total.should == 500
+    end
+  end
+end