From 97a1adcef10f9a32dbc9b899b56fb96be8095ed2 Mon Sep 17 00:00:00 2001 From: Adam Roben Date: Mon, 10 Nov 2014 14:30:37 -0500 Subject: [PATCH] Allow extensions without a leading dot This makes the API a little easier to use. --- lib/linguist/language.rb | 4 ++++ test/test_language.rb | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/lib/linguist/language.rb b/lib/linguist/language.rb index 5aa1db8c..3f90900f 100644 --- a/lib/linguist/language.rb +++ b/lib/linguist/language.rb @@ -206,8 +206,12 @@ module Linguist # Language.find_by_extension('.rb') # # => [#] # + # Language.find_by_extension('rb') + # # => [#] + # # Returns all matching Languages or [] if none were found. def self.find_by_extension(extname) + extname = ".#{extname}" unless extname.start_with?(".") @extension_index[extname] end diff --git a/test/test_language.rb b/test/test_language.rb index 066f577a..0c7adf81 100644 --- a/test/test_language.rb +++ b/test/test_language.rb @@ -243,8 +243,8 @@ class TestLanguage < Test::Unit::TestCase def test_find_by_extension assert_equal [], Language.find_by_extension('.factor-rc') - assert_equal [], Language.find_by_extension('rb') assert_equal [], Language.find_by_extension('foo.rb') + assert_equal [Language['Ruby']], Language.find_by_extension('rb') assert_equal [Language['Ruby']], Language.find_by_extension('.rb') assert_equal [Language['M'], Language['Mathematica'], Language['Matlab'], Language['Mercury'], Language['Objective-C']], Language.find_by_extension('.m') end