lib/railroad/controllers_diagram.rb in drewda-railroad-0.5.0 vs lib/railroad/controllers_diagram.rb in drewda-railroad-0.5.1

@@ -18,14 +18,11 @@
   # Process controller files
   def generate
     STDERR.print "Generating controllers diagram\n" if @options.verbose
 
     files = Dir.glob("app/controllers/**/*_controller.rb") - @options.exclude
-    files << 'app/controllers/application_controller.rb'
     files.each do |f|
       class_name = extract_class_name(f)
-      # ApplicationController's file is 'application.rb'
-      class_name += 'Controller' if class_name == 'Application'
       process_class class_name.constantize
     end 
   end # generate
 
   private