Adds ProfileScraper, starts LeagueScraper

Author: cadwallion

.rspec

@@ -0,0 +1 @@
+--color

lib/bnet_scraper/starcraft2.rb

@@ -0,0 +1,78 @@
+module BnetScraper
+  module Starcraft2
+    REGIONS = {
+      'na'  => { domain: 'us.battle.net', dir: 'en' },
+      'eu'  => { domain: 'eu.battle.net', dir: 'eu' },
+      'cn'  => { domain: 'www.battlenet.com.cn', dir: 'zh' },
+      'sea' => { domain: 'sea.battle.net', dir: 'en' },
+      'fea' => { domain: 'tw.battle.net', dir: 'zh' } 
+    }
+
+    class ProfileScraper
+      attr_reader :bnet_id, :account, :region, :agent, :bnet_index
+
+      def initialize bnet_id, account, region = 'na'
+        @bnet_id  = bnet_id
+        @account  = account
+        @region   = region
+        @agent    = Mechanize.new
+        set_bnet_index
+      end
+
+      def set_bnet_index
+        [1,2].each do |idx|
+          res = Net::HTTP.get_response URI profile_url idx 
+          if res.is_a? Net::HTTPSuccess
+            @bnet_index = idx
+            return
+          end
+        end
+      end
+
+      def scrape
+        @response = @agent.get(profile_url)
+
+        @race = @response.search("#season-snapshot .module-footer a").first().inner_html()
+        @wins = @response.search("#career-stats h2").inner_html()
+        @achievements = @response.search("#profile-header h3").inner_html()
+
+        parse_response
+      end
+
+      def parse_response
+         {
+          bnet_id: @bnet_id,
+          account: @account,
+          bnet_index: @bnet_index,
+          race: @race,
+          wins: @wins,
+          achievements: @achievements 
+        }  
+      end
+
+      def profile_url bnet_index = @bnet_index
+        "http://#{region_info[:domain]}/sc2/#{region_info[:dir]}/profile/#{bnet_id}/#{bnet_index}/#{account}/"
+      end
+
+      def get_bnet_index
+        unless @bnet_index
+          [1,2].each do |idx|
+                        
+          end
+        end
+      end
+
+      def region_info
+        REGIONS[region] 
+      end
+    end
+
+    class LeagueScraper
+      attr_reader :url
+
+      def initialize(url)
+        @url = url 
+      end
+    end
+  end
+end

spec/spec_helper.rb

@@ -0,0 +1,5 @@
+$:.push File.join(File.dirname(__FILE__), '..', 'lib')
+
+require 'bnet_scraper'
+Dir[File.dirname(__FILE__) + "/support/**/*.rb"].each {|f| require f }
+

spec/starcraft2/league_scraper_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+describe BnetScraper::Starcraft2::LeagueScraper do
+  describe '#initialize' do
+    it 'should take a league URL parameter' do
+      url = "http://us.battle.net/sc2/en/profile/2377239/1/Demon/ladder/12345"
+      scraper = BnetScraper::Starcraft2::LeagueScraper.new(url)
+      scraper.url.should == url
+    end
+  end
+end

spec/starcraft2/profile_scraper_spec.rb

@@ -0,0 +1,81 @@
+require 'spec_helper'
+
+describe BnetScraper::Starcraft2::ProfileScraper do
+  subject { BnetScraper::Starcraft2::ProfileScraper.new('2377239', 'Demon') }
+
+  describe '#initialize' do
+    it 'should take bnet_id and account parameters' do
+      subject.bnet_id.should == '2377239'
+      subject.account.should == 'Demon'
+    end
+
+    it 'should default the region to na' do
+      subject.region.should == 'na'
+    end
+
+    it 'should assign region if passed' do
+      BnetScraper::Starcraft2::ProfileScraper.any_instance.should_receive(:set_bnet_index)
+      scraper = BnetScraper::Starcraft2::ProfileScraper.new('2377239', 'Demon', 'fea')
+      scraper.region.should == 'fea'
+    end
+  end
+
+  describe '#region_info' do
+    it 'should return information based on the set region' do
+      subject.region_info.should == { domain: 'us.battle.net', dir: 'en' }
+    end
+  end
+
+  describe '#set_bnet_index' do
+    it 'should return the valid integer needed for a proper URL parse from bnet' do
+      subject.set_bnet_index 
+      subject.bnet_index.should == 1
+    end
+  end
+
+  describe '#profile_url' do
+    it 'should return a string URL for bnet' do
+      subject.profile_url.should == 'http://us.battle.net/sc2/en/profile/2377239/1/Demon/'
+    end
+
+    it 'should optionally take a bnet_index to use instead of saved bnet_index' do
+      subject.profile_url(2).should == 'http://us.battle.net/sc2/en/profile/2377239/2/Demon/'
+    end
+  end
+
+  describe '#scrape' do
+    it 'should set the race, wins, and achievements attributes' do
+      subject.instance_variable_get(:@race).should be_nil
+      subject.instance_variable_get(:@achievements).should be_nil
+      subject.instance_variable_get(:@wins).should be_nil
+
+      subject.scrape
+      
+      subject.instance_variable_get(:@race).should == 'Protoss'
+      subject.instance_variable_get(:@achievements).should == '3630'
+      subject.instance_variable_get(:@wins).should == '684'
+    end
+
+    it 'should call parse_response' do
+      subject.should_receive(:parse_response)
+      subject.scrape
+    end
+  end
+
+  describe '#parse_response' do
+    it 'should extract profile data from the response' do
+      expected = {
+        bnet_id: '2377239',
+        account: 'Demon',
+        bnet_index: 1,
+        race: 'Protoss',
+        wins: '684',
+        achievements: '3630'
+      }
+
+      subject.parse_response.should == { bnet_id: '2377239', account: 'Demon', bnet_index: 1, race: nil, wins: nil, achievements: nil }
+      subject.scrape
+      subject.parse_response.should == expected
+    end
+  end
+end