Rakefile in rails-erd-0.3.0 vs Rakefile in rails-erd-0.4.0

@@ -10,13 +10,13 @@
 
   spec.authors = ["Rolf Timmermans"]
   spec.email = "[e-mail]"
   spec.homepage = "http://rails-erd.rubyforge.org/"
 
-  spec.add_runtime_dependency "activerecord", "~> 3.0.0"
+  spec.add_runtime_dependency "activerecord", "~> 3.0"
   spec.add_runtime_dependency "activesupport", "~> 3.0"
-  spec.add_runtime_dependency "ruby-graphviz", "~> 0.9.17"
+  spec.add_runtime_dependency "ruby-graphviz", "~> 0.9.18"
   spec.add_development_dependency "sqlite3-ruby"
   
   # Don't bundle examples or website in gem.
   excluded = Dir["{examples,site}/**/*"]
   spec.files -= excluded
@@ -37,63 +37,23 @@
 task :default => :test
 
 begin
   require "hanna/rdoctask"
   Rake::RDocTask.new do |rdoc|
-    rdoc.rdoc_files = Dir["[A-Z][A-Z]*"] + Dir["lib/**/*.rb"]
-    rdoc.title = "Rails ERD – Entity-Relationship Diagrams for Rails"
+    rdoc.rdoc_files = %w{CHANGES.rdoc LICENSE} + Dir["lib/**/*.rb"]
+    rdoc.title = "Rails ERD – API Documentation"
     rdoc.rdoc_dir = "rdoc"
+    rdoc.main = "RailsERD"
   end
 rescue LoadError
 end
 
 desc "Generate diagrams for bundled examples"
 task :examples do
-  require "rubygems"
-  require "bundler"
-  Bundler.require
-  require "rails_erd/diagram/graphviz"
+  require File.expand_path("examples/generate", File.dirname(__FILE__))
+end
 
-  Dir["examples/*/*"].each do |path|
-    name = File.basename(path)
-    print "==> Generating ERD for #{name.capitalize}... "
-    begin
-      # Load database schema.
-      ActiveRecord::Base.establish_connection :adapter => "sqlite3", :database => ":memory:"
-      ActiveRecord::Migration.suppress_messages do
-        begin
-          require File.expand_path("#{path}/schema.rb", File.dirname(__FILE__))
-        rescue LoadError
-        end
-      end
-      
-      # Load domain models for this example.
-      Dir["#{path}/**/*.rb"].each do |model|
-        require File.expand_path(model, File.dirname(__FILE__))
-      end
-      
-      # Skip empty domain models.
-      next if ActiveRecord::Base.descendants.empty?
-
-      puts "#{ActiveRecord::Base.descendants.length} models"
-      [:simple, :advanced].each do |notation|
-        filename = File.expand_path("examples/#{name}#{notation != :simple ? "-#{notation}" : ""}", File.dirname(__FILE__))
-
-        default_options = { :notation => notation, :filename => filename, :attributes => [:regular],
-          :title => name.classify + " domain model" }
-
-        specific_options = eval((File.read("#{path}/options.rb") rescue "")) || {}
-
-        # Generate ERD.
-        RailsERD::Diagram::Graphviz.create(default_options.merge(specific_options))
-      end
-    ensure
-      # Completely remove all loaded Active Record models.
-      ActiveRecord::Base.descendants.each do |model|
-        Object.send :remove_const, model.name.to_sym rescue nil
-      end
-      ActiveRecord::Base.direct_descendants.clear
-      Arel::Relation.send :class_variable_set, :@@connection_tables_primary_keys, {}
-      ActiveSupport::Dependencies::Reference.clear!
-    end
+namespace :examples do
+  task :sfdp do
+    require File.expand_path("examples/sfdp", File.dirname(__FILE__))
   end
 end