Add helper for MimeMagic.by_path

chrugail · Oct 3, 2010 · ddcbcf3 · ddcbcf3
1 parent 266129f
commit ddcbcf3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/mimemagic.rb b/lib/mimemagic.rb
@@ -78,6 +78,11 @@ def self.by_extension(ext)
     mime ? new(mime) : nil
   end
 
+  # Lookup mime type by filename
+  def self.by_path(path)
+    by_extension(File.extname(path))
+  end
+
   # Lookup mime type by magic content analysis.
   # This is a slow operation.
   def self.by_magic(io)

diff --git a/test/mimemagic_test.rb b/test/mimemagic_test.rb
@@ -34,6 +34,14 @@
     MimeMagic.by_extension('').should.equal nil
   end
 
+  it 'should recognize by a path' do
+    MimeMagic.by_path('/adsjkfa/kajsdfkadsf/kajsdfjasdf.html').should.equal 'text/html'
+    MimeMagic.by_path('something.html').should.equal 'text/html'
+    MimeMagic.by_path('wtf.rb').should.equal 'application/x-ruby'
+    MimeMagic.by_path('where/am.html/crazy').should.equal nil
+    MimeMagic.by_path('').should.equal nil
+  end
+
   it 'should recognize by magic' do
     Dir['test/files/*'].each do |file|
       mime = file[11..-1].gsub('.', '/')