tasks/railroady.rake in railroady-1.2.0 vs tasks/railroady.rake in railroady-1.3.0

@@ -53,10 +53,13 @@
     end
   end
 
   namespace :models do
 
+    desc 'Generated brief and complete class diagrams for all models.'
+    task :all => ['diagram:models:complete', 'diagram:models:brief']
+
     desc 'Generates an class diagram for all models.'
     task :complete do
       f = @MODELS_ALL
       puts "Generating #{f}"
       sh "railroady -lamM | #{@SED} | dot -T#{RailRoady::RakeHelpers.format} > #{f}"
@@ -67,13 +70,31 @@
       f = @MODELS_BRIEF
       puts "Generating #{f}"
       sh "railroady -blamM | #{@SED} | dot -T#{RailRoady::RakeHelpers.format} > #{f}"
     end
 
+    desc 'Generates an class diagram for all models including those in engines'
+    task :complete_with_engines do
+      f = @MODELS_ALL
+      puts "Generating #{f}"
+      sh "railroady -ilamzM | #{@SED} | dot -T#{RailRoady::RakeHelpers.format} > #{f}"
+    end
+    
+    desc 'Generates an abbreviated class diagram for all models including those in engines'
+    task :brief_with_engines do
+      f = @MODELS_BRIEF
+      puts "Generating #{f}"
+      sh "railroady -bilamzM | #{@SED} | dot -T#{RailRoady::RakeHelpers.format} > #{f}"
+    end
+
+
   end
 
   namespace :controllers do
+    
+    desc 'Generated brief and complete class diagrams for all controllers.'
+    task :all => ['diagram:controllers:complete', 'diagram:controllers:brief']
 
     desc 'Generates an class diagram for all controllers.'
     task :complete do
       f = @CONTROLLERS_ALL
       puts "Generating #{f}"
@@ -84,15 +105,42 @@
     task :brief do
       f = @CONTROLLERS_BRIEF
       puts "Generating #{f}"
       sh "railroady -blC | #{@SED} | neato -T#{RailRoady::RakeHelpers.format} > #{f}"
     end
+    
+    desc 'Generates an class diagram for all controllers including those in engines'
+    task :complete_with_engines do 
+      f = @CONTROLLERS_ALL
+      puts "Generating #{f}"
+      sh "railroady -ilC --engine-controllers | #{@SED} | neato -T#{RailRoady::RakeHelpers.format} > #{f}"
+    end
+    
+    desc 'Generates an abbreviated class diagram for all controllers including those in engines.'
+    task :brief_with_engines do
+      f = @CONTROLLERS_BRIEF
+      puts "Generating #{f}"
+      sh "railroady -bilC --engine-controllers | #{@SED} | neato -T#{RailRoady::RakeHelpers.format} > #{f}"
+    end
 
+    
   end
 
   desc 'Generates all class diagrams.'
-  task all: ['diagram:setup:create_new_doc_folder_if_needed',
+  task all: [
+             'diagram:setup:create_new_doc_folder_if_needed',
              'diagram:models:complete',
              'diagram:models:brief',
              'diagram:controllers:complete',
-             'diagram:controllers:brief']
+             'diagram:controllers:brief'
+            ]
+
+  desc 'Generates all class diagrams including those in engines'
+  task all_with_engines: [
+             'diagram:setup:create_new_doc_folder_if_needed',
+             'diagram:models:complete_with_engines',
+             'diagram:models:brief_with_engines',
+             'diagram:controllers:complete_with_engines',
+             'diagram:controllers:brief_with_engines'
+            ]
+
 end