check file type

app/controllers/desktop_publications_controller.rb

@@ -1,3 +1,5 @@
+require 'mime/types'
+
 class DesktopPublicationsController< ApplicationController
   def journal_p
     render "desktop/journal_p", :layout => false
@@ -42,12 +44,34 @@ class DesktopPublicationsController< ApplicationController
     render :json=>b.to_json
   end
 
+  def check_file_type file
+    if not file.nil?
+      file_type = MIME::Types.type_for(file).first.to_s.split("/")[1]
+
+      case file_type
+      when "jpg", "jpeg"
+        type = "jpg" 
+      when "text", "txt"
+        type = "txt"
+      when "pdf"
+        type = "pdf"
+      when "png"
+        type = "png"
+      else "readme"
+      end
+
+      file_type = "/assets/ft-icons/#{type}/#{type}-48_32.png"
+    else
+      file_type = ""
+    end
+  end
+
   def getjournals
     publications = WritingJournal.where(:create_user_id => current_user.id)
 
     sort_publications= Hash.new
     data = Array.new
-    icon_size = "128_32"
+
     publications.each do |publication|
       if sort_publications[publication.journal_title].nil?
         sort_publications[publication.journal_title] = Array.new
@@ -60,7 +84,7 @@ class DesktopPublicationsController< ApplicationController
           coauthors: publication.authors,
           year: publication.year,
           files: publication.writing_journal_files.collect{|file|
-            {title: file.title, url: file.file.url, icon: "/assets/ft-icons/jpg/jpg-48_32.png"}
+            {title: file.title, url: file.file.url, icon: check_file_type(file.file.url)}
           }
         }
     end