Fix auto diagram generation

README.md

@@ -66,6 +66,11 @@ cluster: false
 splines: spline
 ```
 
+Auto generation
+---------------
+
+* Run <tt>rails generate erd:install</tt>
+* Run <tt>rails db:migrate</tt>, then the diagram is generated
 
 Learn more
 ----------

lib/generators/erd/templates/auto_generate_diagram.rake

@@ -2,5 +2,5 @@
 # NOTE: are sensitive to local FS writes, and besides -- it's just not proper
 # NOTE: to have a dev-mode tool do its thing in production.
 if Rails.env.development?
-  Erd.load_tasks
+  RailsERD.load_tasks
 end

lib/rails_erd.rb

@@ -56,6 +56,16 @@ def default_options
         :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: