Index by language object

lib/linguist/repository.rb

@@ -35,7 +35,7 @@ module Linguist
     def initialize(paths)
       @paths = paths
 
-      @stats = nil
+      @language = @size = nil
       @sizes = Hash.new { 0 }
     end
 
@@ -44,20 +44,18 @@ module Linguist
     end
 
     def language
-      Language[stats[:primary]]
+      compute_stats
+      @language
     end
 
     def languages
-      stats
+      compute_stats
       @sizes
     end
 
     def size
-      stats[:total_size]
+      compute_stats
-    end
+      @size
-
-    def stats
-      @stats ||= compute_stats
     end
 
     def compute_stats
@@ -67,29 +65,17 @@ module Linguist
         language = blob.language
 
         if language.common?
-          @sizes[language.name] += blob.size
+          @sizes[language] += blob.size
         end
       end
 
-      total_size = @sizes.inject(0) { |s,(k,v)| s + v }
+      @size = @sizes.inject(0) { |s,(k,v)| s + v }
 
-      results = {
+      if primary = @sizes.max_by { |(_, size)| size }
-        :total_size => total_size
+        @language = primary[0]
-      }
-
-      @sizes.each do |language, size|
-        results[language] = size
       end
 
-      primary = @sizes.max_by { |(language, size)|
+      nil
-        size
-      }
-
-      if primary
-        results[:primary] = primary[0]
-      end
-
-      results
     end
   end
 end

test/test_repository.rb

@@ -49,8 +49,8 @@ class TestRepository < Test::Unit::TestCase
   end
 
   def test_linguist_languages
-    assert linguist_repo.languages['Ruby'] > 30_000
+    assert linguist_repo.languages[Language['Ruby']] > 30_000
-    assert linguist_repo.languages['Python'] < 1000
+    assert linguist_repo.languages[Language['Python']] < 1000
   end
 
   def test_linguist_size