lib/github_metadata.rb in github_metadata-0.2.2 vs lib/github_metadata.rb in github_metadata-0.3.0

@@ -64,10 +64,14 @@
   
   def contributors_url
     File.join(github_url, 'contributors')
   end
   
+  def branches_url
+    File.join(github_url, 'branches')
+  end
+  
   def commits_feed_url
     File.join(github_url, "commits/#{default_branch}.atom")
   end
   
   # Returns an array of GithubMetadata::Contributor instances, one for each
@@ -99,11 +103,11 @@
     !!wiki_pages
   end
   
   # Returns the amount of wiki pages or nil when no wiki is present
   def wiki_pages
-    wiki_link = document.at_css('a[highlight="repo_wiki"]')
+    wiki_link = document.at_css('a[highlight="repo_wiki"] .counter')
     return nil unless wiki_link
     wiki_link.text[/\d+/].to_i
   end
   
   # Returns true if the repo has issues enabled
@@ -111,24 +115,26 @@
     !!issues
   end
   
   # Returns issue count or nil if issues are disabled
   def issues
-    issue_link = document.at_css('a[highlight="issues"]')
+    issue_link = document.at_css('a[highlight="issues"] .counter')
     return nil unless issue_link
     issue_link.text[/\d+/].to_i
   end
   
   # Returns amount of pull requests
   def pull_requests
-    pull_request_link = document.at_css('a[highlight="repo_pulls"]')
+    pull_request_link = document.at_css('a[highlight="repo_pulls"] .counter')
     return nil unless pull_request_link
     pull_request_link.text[/\d+/].to_i
   end
   
   # Returns the default branch of the repo
   def default_branch
-    document.at_css('.tabs .contextswitch code').text
+    @default_branch ||= Nokogiri::HTML(open(branches_url)).at_css('tr.base td.name h3').text.strip.chomp
+  rescue OpenURI::HTTPError => err
+    raise GithubMetadata::RepoNotFound, err.to_s
   end
   
   # Returns (at most) the last 20 commits (fetched from atom feed of the default_branch)
   # as instances of GithubMetadata::Commit
   def recent_commits