Add image fetching logic

Author: cmaitchison

lib/rails-file-icons.rb

@@ -1,14 +1,27 @@
-require "rails_file_icons/version"
+require "rails-file-icons/engine"
 
-module RailsFileIcons
-  require "rails_file_icons/engine"
+module Icon
 
-  def self.icon_for_mime_type mime_type
-    'file_extension_doc.png'
+  KNOWN_EXTENSIONS = %w{
+    3gp 7z ace ai aif aiff amr asf asx bat bin bmp bup cab cbr cda cdl cdr chm
+    dat divx dll dmg doc dss dvf dwg eml eps exe fla flv gif gz hqx htm html
+    ifo indd iso jar jpeg jpg lnk log m4a m4b m4p m4v mcd mdb mid mov mp2 mp4
+    mpeg mpg msi mswmm ogg pdf png pps ps psd pst ptb pub qbb qbw qxd ram
+    rar rm rmvb rtf sea ses sit sitx ss swf tgz thm tif tmp torrent ttf txt
+    vcd vob wav wma wmv wps xls xpi zip
+    }.inject({}) do |known_extensions, ext|
+      known_extensions[ext] = "file_extension_#{ext}.png"
+      known_extensions
+    end
+  
+ 
+  
+  def self.for_filename filename
+    for_ext File.extname(filename)
   end
 
-  def self.icon_for_file_extension file_extension
-    'file_extension_ace.png'
+  def self.for_ext file_extension
+    ext = file_extension.start_with?('.') ? file_extension[1..-1] : file_extension
+    KNOWN_EXTENSIONS[ext.downcase] || 'file_extension_unknown.png'
   end
-
-end
+end

rails-file-icons.gemspec

@@ -1,7 +1,7 @@
 # -*- encoding: utf-8 -*-
 lib = File.expand_path('../lib', __FILE__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
-require 'rails_file_icons/version'
+require 'rails-file-icons/version'
 
 Gem::Specification.new do |gem|
   gem.name          = "rails-file-icons"