lib/rails_erd.rb in rails-erd-1.5.0 vs lib/rails_erd.rb in rails-erd-1.5.1

@@ -54,9 +54,19 @@
         :only_recursion_depth, nil,
         :prepend_primary, false,
         :cluster, false,
       ]
     end
+
+    def loaded_tasks=(val); @loaded_tasks = val; end
+    def loaded_tasks; return @loaded_tasks; end
+
+    def load_tasks
+      return if(self.loaded_tasks)
+      self.loaded_tasks = true
+
+      Dir[File.join(File.dirname(__FILE__), 'tasks', '**/*.rake')].each { |rake| load rake }
+    end
   end
 
   module Inspectable # @private :nodoc:
     def inspection_attributes(*attributes)
       attribute_inspection = attributes.collect { |attribute|