lib/rails_erd/tasks.rake in rails-erd-0.1.1 vs lib/rails_erd/tasks.rake in rails-erd-0.2.0

@@ -12,26 +12,26 @@
       end
     end
   end
   
   task :load_models do
-    say "Loading ActiveRecord models..."
+    say "Loading Active Record models..."
 
     Rake::Task[:environment].invoke
     Rails.application.config.paths.app.models.paths.each do |model_path|
       Dir["#{model_path}/**/*.rb"].sort.each do |file|
         require_dependency file
       end
     end
   end
   
   task :generate => [:options, :load_models] do
-    say "Generating ERD diagram..."
+    say "Generating Entity-Relationship Diagram..."
 
-    require "rails_erd/diagram"
-    diagram = RailsERD::Diagram.generate
+    require "rails_erd/diagram/graphviz"
+    file = RailsERD::Diagram::Graphviz.create
 
-    say "Done! Saved diagram to #{diagram.file_name}."
+    say "Done! Saved diagram to #{file}."
   end
 end
 
 desc "Generate an Entity-Relationship Diagram based on your models"
 task :erd => "erd:generate"