Rakefile in rails-erd-0.2.0 vs Rakefile in rails-erd-0.3.0

@@ -11,13 +11,18 @@
   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 "activesupport", "~> 3.0.0"
+  spec.add_runtime_dependency "activesupport", "~> 3.0"
   spec.add_runtime_dependency "ruby-graphviz", "~> 0.9.17"
   spec.add_development_dependency "sqlite3-ruby"
+  
+  # Don't bundle examples or website in gem.
+  excluded = Dir["{examples,site}/**/*"]
+  spec.files -= excluded
+  spec.test_files -= excluded
 end
 
 Jeweler::GemcutterTasks.new
 
 Jeweler::RubyforgeTasks.new do |rubyforge|
@@ -37,6 +42,58 @@
     rdoc.rdoc_files = Dir["[A-Z][A-Z]*"] + Dir["lib/**/*.rb"]
     rdoc.title = "Rails ERD – Entity-Relationship Diagrams for Rails"
     rdoc.rdoc_dir = "rdoc"
   end
 rescue LoadError
+end
+
+desc "Generate diagrams for bundled examples"
+task :examples do
+  require "rubygems"
+  require "bundler"
+  Bundler.require
+  require "rails_erd/diagram/graphviz"
+
+  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
+  end
 end