#
# Modified by EmmanuelOga.blogspot.com, converted to a task
# Original in http://www.hackdiary.com/archives/000093.html
#
# Requirements: dot (http://www.graphviz.org/)
#
# Usage: Drop in lib/tasks of your rails app. rake doc:reflex
#        See reflections.png in doc
#
namespace :doc do

  desc "Graph model reflections"
  task(:reflex => :environment) do

    filename= "#{RAILS_ROOT}/doc/reflections.dot"

    File.open(filename, "w+") do |file|
      file << "digraph x {\n"

      Dir.glob("#{RAILS_ROOT}/app/models/*rb") do |f|

        f.match(/\/([a-z_]+).rb/)
        classname = $1.camelize
        klass = Kernel.const_get classname

        if klass.superclass == ActiveRecord::Base
          file << classname << "\n"
          klass.reflect_on_all_associations.each do |a|
            file << classname << " -> " << a.name.to_s.camelize.singularize 
            file << " [label=#{a.macro.to_s}]\n"
          end
        end
      end

      file << "}"
    end

    system("dot -Tpng -O #{filename}")

  end 
end