display_file_command.rb in yard-0.9.0

- old
+ new

@@ -3,22 +3,27 @@
     module Commands
       # Displays a README or extra file.
       #
       # @todo Implement better support for detecting binary (image) filetypes
       class DisplayFileCommand < LibraryCommand
+        attr_accessor :index
+
         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)
+            render File.read_binary(filename)
           else
             file = CodeObjects::ExtraFileObject.new(filename)
-            options.update(:object => Registry.root, :type => :layout, :file => file)
+            options.update :object => Registry.root,
+                           :type => :layout,
+                           :file => file,
+                           :index => index ? true : false
             render
           end
         end
       end
     end
   end
-end
\ No newline at end of file
+end