Adds REGION_DOMAINS for domain mapping to region

Author: cadwallion

lib/bnet_scraper/starcraft2.rb

@@ -15,7 +15,16 @@ module Starcraft2
       'eu'  => { domain: 'eu.battle.net', dir: 'eu', label: 'Europe' },
       'cn'  => { domain: 'www.battlenet.com.cn', dir: 'zh', label: 'China' },
       'sea' => { domain: 'sea.battle.net', dir: 'en', label: 'South-East Asia' },
-      'fea' => { domain: 'tw.battle.net', dir: 'zh', label: 'Korea' } 
+      'fea' => { domain: 'tw.battle.net', dir: 'zh', label: 'Korea' }
+    }
+
+    REGION_DOMAINS = {
+      'us.battle.net' => 'na',
+      'eu.battle.net' => 'eu',
+      'www.battlenet.com.cn' => 'cn',
+      'sea.battle.net' => 'sea',
+      'kr.battle.net' => 'fea',
+      'tw.battle.net' => 'fea'
     }
 
     # This is a convenience method that chains calls to ProfileScraper,

lib/bnet_scraper/starcraft2/base_scraper.rb

@@ -20,7 +20,7 @@ class BaseScraper
       def initialize options = {}
         if options[:url]
           extracted_data = options[:url].match(/http:\/\/(.+)\/sc2\/(.+)\/profile\/(.+)\/(\d{1})\/(.[^\/]+)\//)
-          @region = REGIONS.select { |k,v| v[:domain] == extracted_data[1] && v[:dir] == extracted_data[2] }.first.first
+          @region     = REGION_DOMAINS[extracted_data[1]]
           @bnet_id    = extracted_data[3]
           @bnet_index = extracted_data[4]
           @account    = extracted_data[5]
@@ -58,7 +58,7 @@ def profile_url bnet_index = @bnet_index
       # converts region short-code to region-based URL information
       #   'na' => { domain: 'us.battle.net', :dir: 'en' }
       def region_info
-        REGIONS[region] 
+        REGIONS[region]
       end
 
       def valid?

lib/bnet_scraper/starcraft2/status_scraper.rb

@@ -30,9 +30,11 @@ def method_missing sym
         private
         def status? region
           @status ||= fetch
-          @status.select do |r|
-            r[:region] == REGIONS.select { |k,v| k == region.to_s }.first.last[:label]
-          end.first[:status]
+          region_status = @status.find do |r|
+            r[:region] == REGIONS[region.to_s][:label]
+          end
+
+          region_status[:status]
         end
       end
     end