ispusage 0.4.0 → 0.4.1

data/VERSION

@@ -1 +1 @@
-0.4.0
+0.4.1

data/lib/ispusage/fetchers/au_exetel.rb

@@ -0,0 +1,36 @@
+require 'rubygems'
+require 'mechanize'
+
+class IspUsage
+  class Fetchers
+    class AUExetel < Fetcher
+      def initialize(options)
+        super(options)
+      end
+
+      def fetch_usage
+        agent = Mechanize.new
+
+        page = agent.get('http://www.exetel.com.au/index_alt.php')
+        form = page.form('loginform')
+        form.login_name = self.username
+        form.password = self.password
+
+        page = agent.submit(form, form.buttons.first)
+
+        page.search('.new_usage_content').each do |usage_div|
+          usage_period = UsagePeriod.new
+          matches = usage_div.text.match(/^(\d+\.\d+)\D+(\d+)/)
+          next if matches.nil?
+          usage_period.used = matches[1].to_f * 1000
+          usage_period.quota = matches[2].to_f * 1000
+
+          usage_periods << usage_period
+        end
+
+        usage_periods.first.label = 'Peak'
+        usage_periods.last.label = 'Off Peak'
+      end
+    end
+  end
+end

data/lib/ispusage/fetchers/au_tpg.rb

@@ -0,0 +1,39 @@
+require 'rubygems'
+require 'mechanize'
+
+class IspUsage
+  class Fetchers
+    class AUTpg < Fetcher
+      def initialize(options)
+        super(options)
+      end
+
+      def fetch_usage
+        agent = Mechanize.new
+
+        page = agent.get('https://cyberstore.tpg.com.au/your_account/')
+        form = page.form('form')
+        form.check_username = self.username
+        form.password = self.password
+
+        page = agent.submit(form, form.buttons.first)
+        if page.body =~ /Invalid username/
+          self.error = "Error: Invalid username or password"
+          return
+        end
+
+        page = agent.get('https://cyberstore.tpg.com.au/your_account/index.php?function=checkaccountusage')
+
+        usage_text = page.search('table.light_box table tr').text
+        peak = UsagePeriod.new(:label => 'Peak', :type => :balance)
+        usage_periods << peak
+        peak.used = usage_text.match(/\dPeak Downloads used: (\d+)/)[1].to_i
+
+        offpeak = UsagePeriod.new(:label => 'Off Peak', :type => :balance)
+        usage_periods << offpeak
+        offpeak.used = usage_text.match(/Off-Peak Downloads used: (\d+)/)[1].to_i
+
+      end
+    end
+  end
+end

data/lib/ispusage/fetchers/fetcher.rb

@@ -4,7 +4,7 @@ require 'json'
 class IspUsage
   class Fetchers
     class Fetcher
-      attr_accessor :usage_periods, :username, :password, :options
+      attr_accessor :usage_periods, :username, :password, :options, :error
       
       def initialize(options)
         @username = options[:username]
@@ -14,11 +14,17 @@ class IspUsage
       end
       
       def to_hash
-        {
+        hash = {
           :isp => self.class.to_s.split('::').last,
           :username => @username,
           :usage_periods => self.usage_periods.map(&:to_hash)
         }
+
+        unless error.nil?
+          hash[:error] = error
+        end
+
+        hash
       end
       
       def to_json

data/lib/ispusage/fetchers.rb

@@ -1,6 +1,8 @@
 require 'ispusage/fetchers/fetcher'
 require 'ispusage/fetchers/au_optus'
+require 'ispusage/fetchers/au_exetel'
 require 'ispusage/fetchers/au_three'
+require 'ispusage/fetchers/au_tpg'
 require 'ispusage/fetchers/iinet'
 require 'ispusage/fetchers/internode'
 require 'ispusage/fetchers/au_translink'

data/lib/ispusage/usage_period.rb

@@ -3,9 +3,8 @@ class IspUsage
     attr_accessor :quota, :used, :label
     attr_reader :type
     def initialize(options = {})
-      self.label = 'No Label Set'
       self.type = :meter
-      
+      self.label = ''
       auto_setters = [:quota, :used, :label, :type]
       options.each do |key, value|
         if auto_setters.include?(key)

data/lib/ispusage.rb

@@ -10,8 +10,10 @@ class IspUsage
   ISPS = [
     {:class => 'AUIinet', :name => 'iiNet'},
     {:class => 'AUInternode', :name => 'Internode'},
+    {:class => 'AUExetel', :name => 'Exetel'},
     {:class => 'AUOptus', :name => 'Optus'},
     {:class => 'AUThree', :name => 'Three'},
+    {:class => 'AUTpg', :name => 'TPG'},
     {:class => 'AUTranslink', :name => 'Translink'}
   ]
 

data/spec/fetchers/au_exetel_spec.rb

@@ -0,0 +1,49 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe IspUsage::Fetchers::AUExetel do
+  it "should be included in the ISP list" do
+    IspUsage::ISPS.map{|i| i[:class]}.should include('AUExetel')  
+  end
+
+  describe "single usage period" do
+    before do
+      signon = File.read(File.dirname(__FILE__) + '/../fixtures/au_exetel/login.html')
+      FakeWeb.register_uri(:get, 'http://www.exetel.com.au/index_alt.php', :response => signon)
+      usage = File.read(File.dirname(__FILE__) + '/../fixtures/au_exetel/usage.html')
+      FakeWeb.register_uri(:post, 'https://www.exetel.com.au/login/redirect.php', :response => usage)
+
+      options = {:username => 'user', :password => 'password'}
+      @usage = IspUsage::Fetchers::AUExetel.new(options)
+      @usage.fetch_usage
+    end
+
+    describe "peak usage" do
+      it "should return usage correctly" do
+        @usage.usage_periods.first.used.should == 13150
+      end
+
+      it "should return quota correctly" do
+        @usage.usage_periods.first.quota.should == 70000
+      end
+
+      it "should set label to phone number" do
+        @usage.usage_periods.first.label.should == "Peak"
+      end
+    end
+
+    describe "off peak usage" do
+      it "should return usage correctly" do
+        @usage.usage_periods.last.used.should == 6920
+      end
+
+      it "should return quota correctly" do
+        @usage.usage_periods.last.quota.should == 60000
+      end
+
+      it "should set label to phone number" do
+        @usage.usage_periods.last.label.should == "Off Peak"
+      end
+    end
+
+  end
+end

data/spec/fetchers/au_tpg_spec.rb

@@ -0,0 +1,62 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe IspUsage::Fetchers::AUTpg do
+  it "should be included in the ISP list" do
+    IspUsage::ISPS.map{|i| i[:class]}.should include('AUTpg')  
+  end
+
+  describe "single usage period" do
+    before do
+      signon = File.read(File.dirname(__FILE__) + '/../fixtures/au_tpg/login.html')
+      FakeWeb.register_uri(:get, 'https://cyberstore.tpg.com.au/your_account/', :response => signon)
+      FakeWeb.register_uri(:post, 'https://cyberstore.tpg.com.au/your_account/index.php', :response => signon)
+      usage = File.read(File.dirname(__FILE__) + '/../fixtures/au_tpg/usage.html')
+      FakeWeb.register_uri(:get, 'https://cyberstore.tpg.com.au/your_account/index.php?function=checkaccountusage', :response => usage)
+
+      options = {:username => 'user', :password => 'password'}
+      @usage = IspUsage::Fetchers::AUTpg.new(options)
+      @usage.fetch_usage
+    end
+
+    describe "peak usage" do
+      it "should return usage correctly" do
+        @usage.usage_periods.first.used.should == 921
+      end
+
+      it "should be a balance" do
+        @usage.usage_periods.first.type.should == :balance
+      end
+
+      it "should set label to phone number" do
+        @usage.usage_periods.first.label.should == "Peak"
+      end
+    end
+
+    describe "off peak usage" do
+      it "should return usage correctly" do
+        @usage.usage_periods.last.used.should == 1402
+      end
+
+      it "should be a balance" do
+        @usage.usage_periods.last.type.should == :balance
+      end
+
+      it "should set label to phone number" do
+        @usage.usage_periods.last.label.should == "Off Peak"
+      end
+    end
+
+  end
+
+  it "should handle a logon failure" do
+    signon = File.read(File.dirname(__FILE__) + '/../fixtures/au_tpg/login.html')
+    FakeWeb.register_uri(:get, 'https://cyberstore.tpg.com.au/your_account/', :response => signon)
+    failure = File.read(File.dirname(__FILE__) + '/../fixtures/au_tpg/failure.html')
+    FakeWeb.register_uri(:post, 'https://cyberstore.tpg.com.au/your_account/index.php', :response => failure)
+
+    options = {:username => 'user', :password => 'password'}
+    @usage = IspUsage::Fetchers::AUTpg.new(options)
+    @usage.fetch_usage
+    @usage.error.should == "Error: Invalid username or password"
+  end
+end

data/spec/fetchers/fetcher_spec.rb

@@ -10,4 +10,16 @@ describe IspUsage::Fetchers::Fetcher do
       :username => 'bob'
     }
   end
+
+  it "should return an error" do
+    @usage = IspUsage::Fetchers::Fetcher.new(:username => 'bob', :password => 'xkcd')
+    @usage.error = "Invalid username/password"
+
+    @usage.to_hash.should == {
+            :isp => 'Fetcher',
+            :username => 'bob',
+            :error => "Invalid username/password",
+            :usage_periods => []
+    }
+  end
 end

data/spec/fixtures/au_exetel/login.html

@@ -0,0 +1,70 @@
+HTTP/1.1 200 OK
+Content-Type: text/html; charset=iso-8859-1
+
+<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
+<html xmlns="http://www.w3.org/1999/xhtml">
+<head><title>Exetel Pty Ltd</title>
+<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
+<meta name="Keywords" content="ADSL, ADSL2, VoIP, HSPA, HSDPA, Mobile, Hosting, Value, X Calling Card, Exetel, Residential, Business, Corporate, Agents, Country Broadband, SMS via ADSL, Wireless, Wireline, Voice, Ethernet, SHDSL, Fibre, Business Grade, VPN, VISP, Telephony, Cheap, Telephony" />
+<meta name="Description" content="Exetel have been providing a range of Internet and phone services to customers since Feb' 2004. There are now over 100,000 customers using a wide range of different Exetel services in every State and Territory." />
+<link href="css/index_alt.css" rel="stylesheet" type="text/css" />
+<script type="text/javascript" src="js/index_alt.js"></script>
+</head>
+<body>
+
+<div id="container">
+	<div id="login">
+		<form action="https://www.exetel.com.au/login/redirect.php" method="post" name="loginform">
+			<input type="text" name="login_name" accesskey="u" tabindex="1" class="input_text" autocomplete="on" />
+			<input type="password" name="password" accesskey="p" tabindex="2" class="input_text" />
+			<input name="doLogin" type="hidden" value="1" />
+			<input type="image" name="submit" src="images/front_btn_login.gif" class="btn_login" />
+		</form>
+	</div>
+
+	<div id="button_resi">
+		<a class="button_resi" href="res_main.php"><!-- --></a>
+	</div>
+	
+	<div id="button_busi">
+		<a class="button_busi" href="small_main.php"><!-- --></a>
+	</div>
+	
+	<div id="button_corp">
+		<a class="button_corp" href="large_main.php"><!-- --></a>
+	</div>
+	
+	<div id="button_agen">
+		<a class="button_agen" href="agents_main.php"><!-- --></a>
+	</div>
+	
+	<div id="button_country">
+		<a class="button_country" href="http://www.countrybroadband.com.au"><!-- --></a>
+	</div>
+	
+	<div id="button_wireless">
+		<a class="button_wireless" href="residential-hspa-information.php"><!-- --></a>
+	</div>
+
+	<div id="bottom_left_logo">
+		<a href="http://www.australiangeographic.com.au/society/2009-conservationist-of-the-year-awardee-ray-thomas.htm" target="_blank"><img src="images/ausGeo_web_logo.gif" alt="Australian Geographic" /></a>
+	</div>	
+	
+	<div id="links">
+		<ul>
+			<li><a href="https://helpdesk.exetel.com.au/">Helpdesk</a></li>
+			<li><a href="http://mirror.exetel.com.au/">Mirror</a></li>
+			<li><a href="http://games.exetel.com.au">Games</a></li>
+			<li><a href="https://webmail.exetel.com.au/src/login.php">Webmail</a></li>
+			<li><a href="http://forum.exetel.com.au/">Forums</a></li>
+			<li><a href="news_main.php">About</a></li>
+			<li><a href="a_network.php">Network Status</a></li>
+			<li><a href="contact.php">Contact</a></li>
+		</ul>
+	</div>
+</div>
+			
+<div id="noflash"><a href="index.php">Click here for the <b>Flash</b> version</a></div>
+
+</body>
+</html>