lib/yard/server/commands/display_file_command.rb in yard-0.6.8 vs lib/yard/server/commands/display_file_command.rb in yard-0.7.0

@@ -1,20 +1,21 @@
 module YARD
   module Server
     module Commands
       # Displays a README or extra file.
-      # 
+      #
       # @todo Implement better support for detecting binary (image) filetypes
       class DisplayFileCommand < LibraryCommand
         def run
           ppath = library.source_path
           filename = File.cleanpath(File.join(library.source_path, path))
           raise NotFoundError if !File.file?(filename)
           if filename =~ /\.(jpe?g|gif|png|bmp)$/i
             headers['Content-Type'] = StaticFileCommand::DefaultMimeTypes[$1.downcase] || 'text/html'
             render IO.read(filename)
           else
-            options.update(:object => Registry.root, :type => :layout, :file => filename)
+            file = CodeObjects::ExtraFileObject.new(filename)
+            options.update(:object => Registry.root, :type => :layout, :file => file)
             render
           end
         end
       end
     end
\ No newline at end of file