railroad 0.3.3 → 0.3.4

data/ChangeLog

@@ -1,6 +1,17 @@
+Version 0.3.4 (apr 12 2007)
+- Add support for model abstract classes.
+  (don't try to get content columns, bug #10033)
+- Add verbose mode
+- More code cleanup.
+- Using an internal representation and then 
+  generating the DOT output. This will allow to
+  add more output formats in the future.
+
+
 Version 0.3.3 (apr 10 2007)
 - Code cleanup 
 
+
 Version 0.3.2 (apr 9 2007)
 - Disable STDOUT when loading applications classes, avoiding 
   messing up the DOT output.

data/README

@@ -22,6 +22,8 @@ Common options:
     -l, --label                      Add a label with diagram information
                                        (type, date, migration, version)
     -o, --output FILE                Write diagram to file FILE
+    -v, --verbose                    Enable verbose output
+                                       (produce messages to STDOUT)
 
 Models diagram options:
     -a, --all                        Include all models
@@ -98,7 +100,7 @@ from Graphviz.
 
 = Website and Project Home
 
-http://railroad.rubyforge.org/
+http://railroad.rubyforge.org
 
 = License
 

data/bin/railroad

@@ -0,0 +1,41 @@
+#!/usr/bin/env ruby
+
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+#   RailRoad generates models and controllers diagrams in DOT language
+#   for a Rails application.
+#
+# Copyright 2007 - Javier Smaldone (http://www.smaldone.com.ar)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+
+APP_NAME       = "railroad"
+APP_HUMAN_NAME = "RailRoad"
+APP_VERSION    = [0,3,4]
+COPYRIGHT      = "Copyright (C) 2007 Javier Smaldone"
+
+require 'railroad/options_struct'
+require 'railroad/models_diagram'
+require 'railroad/controllers_diagram'
+
+options = OptionsStruct.new
+
+options.parse ARGV
+
+if options.command == 'models'
+  diagram = ModelsDiagram.new options
+elsif options.command == 'controllers'
+  diagram = ControllersDiagram.new options
+else
+  STDERR.print "Error: You must supply a command\n" +
+               "  (try #{APP_NAME} -h)\n\n"
+  exit 1
+end
+
+diagram.generate
+diagram.print

data/lib/railroad/app_diagram.rb

@@ -0,0 +1,83 @@
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+# Copyright 2007 - Javier Smaldone (http://www.smaldone.com.ar)
+# See COPYING for more details
+
+require 'railroad/diagram_graph'
+
+# Root class for RailRoad diagrams
+class AppDiagram
+
+  def initialize(options)
+    @options = options
+    @graph = DiagramGraph.new
+    @graph.show_label = @options.label
+
+    STDERR.print "Loading application environment\n" if @options.verbose
+    load_environment
+
+    STDERR.print "Loading application classes\n" if @options.verbose
+    load_classes
+  end
+
+  # Print diagram
+  def print
+    if @options.output
+      old_stdout = STDOUT.dup
+      begin
+        STDOUT.reopen(@options.output)
+      rescue
+        STDERR.print "Error: Cannot write diagram to #{@options.output}\n\n"
+        exit 2
+      end
+    end
+ 
+    STDERR.print "Generating DOT graph\n" if @options.verbose
+
+    STDOUT.print @graph.to_dot 
+
+    if @options.output
+      STDOUT.reopen(old_stdout)
+    end
+  end # print
+
+  private 
+
+  # Prevents Rails application from writing to STDOUT
+  def disable_stdout
+    @old_stdout = STDOUT.dup
+    STDOUT.reopen(PLATFORM =~ /mswin/ ? "NUL" : "/dev/null")
+  end
+
+  # Restore STDOUT  
+  def enable_stdout
+    STDOUT.reopen(@old_stdout)
+  end
+
+
+  # Print error when loading Rails application
+  def print_error(type)
+    STDERR.print "Error loading #{type}.\n  (Are you running " +
+                 "#{APP_NAME} on the aplication's root directory?)\n\n"
+  end
+
+  # Load Rails application's environment
+  def load_environment
+    begin
+      disable_stdout
+      require "config/environment"
+      enable_stdout
+    rescue LoadError
+      enable_stdout
+      print_error "application environment"
+      raise
+    end
+  end
+
+  # Extract class name from filename
+  def extract_class_name(filename)
+    filename.split('/')[2..-1].join('/').split('.').first.camelize
+  end
+
+end # class AppDiagram

data/lib/railroad/controllers_diagram.rb

@@ -0,0 +1,81 @@
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+# Copyright 2007 - Javier Smaldone (http://www.smaldone.com.ar)
+# See COPYING for more details
+
+require 'railroad/app_diagram'
+
+# RailRoad controllers diagram
+class ControllersDiagram < AppDiagram
+ 
+  def initialize(options)
+    super options
+    @graph.diagram_type = 'Controllers'
+  end
+
+  # Process controller files
+  def generate
+    STDERR.print "Generating controllers diagram\n" if @options.verbose
+
+    files = Dir.glob("app/controllers/**/*_controller.rb")
+    files << 'app/controllers/application.rb'
+    files.each do |f|
+      class_name = extract_class_name(f)
+      # ApplicationController's file is 'application.rb'
+      class_name += 'Controller' if class_name == 'Application'
+      process_class class_name.constantize
+    end 
+  end # generate
+
+  private
+
+  # Load controller classes
+  def load_classes
+    begin
+      disable_stdout
+      # ApplicationController must be loaded first
+      require "app/controllers/application.rb" 
+      Dir.glob("app/controllers/**/*_controller.rb") {|c| require c }
+      enable_stdout
+    rescue LoadError
+      enable_stdout
+      print_error "controller classes"
+      raise
+    end
+  end # load_classes
+
+  # Proccess a controller class
+  def process_class(current_class)
+
+    STDERR.print "\tProcessing #{current_class}\n" if @options.verbose
+
+    if @options.brief
+      @graph.add_node ['controller-brief', current_class.name]
+    elsif current_class.is_a? Class 
+      # Collect controller's methods
+      node_attribs = {:public    => [], 
+                      :protected => [], 
+                      :private   => []}
+      current_class.public_instance_methods(false).sort.each { |m|
+        node_attribs[:public] << m
+      } unless @options.hide_public
+      current_class.protected_instance_methods(false).sort.each { |m|
+        node_attribs[:protected] << m
+      } unless @options.hide_protected
+      current_class.private_instance_methods(false).sort.each { |m|
+        node_attribs[:private] << m 
+      } unless @options.hide_private
+      @graph.add_node ['controller', current_class.name, node_attribs]
+    elsif @options.modules && current_class.is_a?(Module)
+      @graph.add_node ['module', current_class.name]
+    end
+
+    # Generate the inheritance edge (only for ApplicationControllers)
+    if @options.inheritance && 
+       (ApplicationController.subclasses.include? current_class.name)
+      @graph.add_edge ['is-a', current_class.name, current_class.superclass.name]
+    end
+  end # process_class
+
+end # class ControllersDiagram

data/lib/railroad/diagram_graph.rb

@@ -0,0 +1,119 @@
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+# Copyright 2007 - Javier Smaldone (http://www.smaldone.com.ar)
+# See COPYING for more details
+
+
+# RailRoad diagram structure
+class DiagramGraph
+
+  def initialize
+    @diagram_type = ''
+    @show_label = false
+    @nodes = []
+    @edges = []
+  end 
+
+  def add_node(node)
+    @nodes << node
+  end
+
+  def add_edge(edge)
+    @edges << edge
+  end
+  
+  def diagram_type= (type)
+    @diagram_type = type
+  end
+
+  def show_label= (value)
+    @show_label = value
+  end
+
+
+  # Generate DOT graph
+  def to_dot
+    return dot_header +
+           @nodes.map{|n| dot_node n[0], n[1], n[2]}.join +
+           @edges.map{|e| dot_edge e[0], e[1], e[2], e[3]}.join +
+           dot_footer
+  end
+
+  private
+
+  # Build DOT diagram header
+  def dot_header
+    result = "digraph #{@diagram_type.downcase}_diagram {\n" +
+             "\tgraph[overlap=false, splines=true]\n"
+    result += dot_label if @show_label
+    return result
+  end
+
+  # Build DOT diagram footer
+  def dot_footer
+    return "}\n"
+  end
+  
+  # Build diagram label
+  def dot_label
+    return "\t_diagram_info [shape=\"plaintext\", " +
+           "label=\"Diagram: #{@diagram_type}\\l" +
+           "Date: #{Time.now.strftime "%b %d %Y - %H:%M"}\\l" + 
+           "Migration version: " +
+           "#{ActiveRecord::Migrator.current_version}\\l" +
+           "Generated by #{APP_HUMAN_NAME} #{APP_VERSION.join('.')}"+
+           "\\l\", fontsize=14]\n"
+  end
+
+  # Build a DOT graph node
+  def dot_node(type, name, attributes=nil)
+    case type
+      when 'model'
+           options = 'shape=Mrecord, label="{' + name + '|'
+           options += attributes.join('\l')
+           options += '\l}"'
+      when 'model-brief'
+           options = ''
+      when 'class'
+           options = 'shape=record, label="{' + name + '|}"' 
+      when 'class-brief'
+           options = 'shape=box' 
+      when 'controller'
+           options = 'shape=Mrecord, label="{' + name + '|'
+           public_methods    = attributes[:public].join('\l')
+           protected_methods = attributes[:protected].join('\l')
+           private_methods   = attributes[:private].join('\l')
+           options += public_methods + '\l|' + protected_methods + '\l|' + 
+                      private_methods + '\l'
+           options += '}"'
+      when 'controller-brief'
+           options = '' 
+      when 'module'
+           options = 'shape=box, style=dotted, label="' + name + '"'
+    end # case
+    return "\t#{quote(name)} [#{options}]\n"
+  end # dot_node
+
+  # Build a DOT graph edge
+  def dot_edge(type, from, to, name = '')
+    options =  name != '' ? "label=\"#{name}\", " : ''
+    case type
+      when 'one-one'
+           options += 'taillabel="1"'
+      when 'one-many'
+           options += 'taillabel="n"'                    
+      when 'many-many'
+           options += 'taillabel="n", headlabel="n", arrowtail="normal"'
+      when 'is-a'
+           options += 'arrowhead="onormal"'
+    end
+    return "\t#{quote(from)} -> #{quote(to)} [#{options}]\n"
+  end # dot_edge
+
+  # Quotes a class name
+  def quote(name)
+    '"' + name + '"'
+  end
+  
+end # class DiagramGraph

data/lib/railroad/models_diagram.rb

@@ -0,0 +1,115 @@
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+# Copyright 2007 - Javier Smaldone (http://www.smaldone.com.ar)
+# See COPYING for more details
+
+require 'railroad/app_diagram'
+
+# RailRoad models diagram
+class ModelsDiagram < AppDiagram
+
+  def initialize(options)
+    super options 
+    @graph.diagram_type = 'Models'
+    # Processed habtm associations
+    @habtm = []
+  end
+
+  # Process model files
+  def generate
+    STDERR.print "Generating models diagram\n" if @options.verbose
+    files = Dir.glob("app/models/**/*.rb")
+    files.each do |f| 
+      process_class extract_class_name(f).constantize
+    end
+  end 
+
+  private
+
+  # Load model classes
+  def load_classes
+    begin
+      disable_stdout
+      Dir.glob("app/models/**/*.rb") {|m| require m }
+      enable_stdout
+    rescue LoadError
+      enable_stdout
+      print_error "model classes"
+      raise
+    end
+  end  # load_classes
+
+  # Process a model class
+  def process_class(current_class)
+
+    STDERR.print "\tProcessing #{current_class}\n" if @options.verbose
+
+    generated = false
+    
+    # Is current_clas derived from ActiveRecord::Base?
+    if current_class.respond_to?'reflect_on_all_associations'
+      node_attribs = []
+      if @options.brief || current_class.abstract_class?
+        node_type = 'model-brief'
+      else 
+        node_type = 'model'
+        # Collect model's content columns
+        current_class.content_columns.each do |a|
+          content_column = a.name
+          content_column += ' :' + a.type.to_s unless @options.hide_types
+          node_attribs << content_column
+        end
+      end
+      @graph.add_node [node_type, current_class.name, node_attribs]
+      generated = true
+      # Process class associations
+      current_class.reflect_on_all_associations.each do |a|
+        process_association current_class.name, a
+      end
+    elsif @options.all && (current_class.is_a? Class)
+      # Not ActiveRecord::Base model
+      node_type = @options.brief ? 'class-brief' : 'class'
+      @graph.add_node [node_type, current_class.name]
+      generated = true
+    elsif @options.modules && (current_class.is_a? Module)
+        @graph.add_node ['module', current_class.name]
+    end
+
+    # Only consider meaningful inheritance relations for generated classes
+    if @options.inheritance && generated && 
+       (current_class.superclass != ActiveRecord::Base) &&
+       (current_class.superclass != Object)
+      @graph.add_edge ['is-a', current_class.name, current_class.superclass.name]
+    end      
+
+  end # process_class
+
+  # Process a model association
+  def process_association(class_name, assoc)
+
+    STDERR.print "\t\tProcessing model association #{assoc.name.to_s}\n" if @options.verbose
+
+    # Skip "belongs_to" associations
+    return if assoc.macro.to_s == 'belongs_to'
+
+    # Only non standard association names needs a label
+    if assoc.class_name == assoc.name.to_s.singularize.camelize
+      assoc_name = ''
+    else
+      assoc_name = assoc.name.to_s
+    end 
+
+    if assoc.macro.to_s == 'has_one' 
+      assoc_type = 'one-one'
+    elsif assoc.macro.to_s == 'has_many' && (! assoc.options[:through])
+      assoc_type = 'one-many'
+    else # habtm or has_many, :through
+      return if @habtm.include? [assoc.class_name, class_name, assoc_name]
+      assoc_type = 'many-many'
+      @habtm << [class_name, assoc.class_name, assoc_name]
+    end  
+    @graph.add_edge [assoc_type, class_name, assoc.class_name, assoc_name]
+  end # process_association
+
+end # class ModelsDiagram

data/lib/railroad/options_struct.rb

@@ -0,0 +1,138 @@
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+# Copyright 2007 - Javier Smaldone (http://www.smaldone.com.ar)
+# See COPYING for more details
+
+require 'ostruct'
+
+# RailRoad command line options parser
+class OptionsStruct < OpenStruct
+
+  require 'optparse'
+
+  def initialize
+    init_options = { :all => false,
+                     :brief => false,
+                     :inheritance => false,
+                     :join => false,
+                     :label => false,
+                     :modules => false,
+                     :hide_types => false,
+                     :hide_public => false,
+                     :hide_protected => false,
+                     :hide_private => false,
+                     :verbose => false,
+                     :command => '' }
+    super(init_options)
+  end # initialize
+
+  def parse(args)
+    @opt_parser = OptionParser.new do |opts|
+      opts.banner = "Usage: #{APP_NAME} [options] command"
+      opts.separator ""
+      opts.separator "Common options:"
+      opts.on("-b", "--brief", "Generate compact diagram", 
+              "  (no attributes nor methods)") do |b|
+        self.brief = b
+      end
+      opts.on("-i", "--inheritance", "Include inheritance relations") do |i|
+        self.inheritance = i
+      end
+      opts.on("-l", "--label", "Add a label with diagram information", 
+              "  (type, date, migration, version)") do |l|
+        self.label = l
+      end
+      opts.on("-o", "--output FILE", "Write diagram to file FILE") do |f|
+        self.output = f
+      end
+      opts.on("-v", "--verbose", "Enable verbose output", 
+              "  (produce messages to STDOUT)") do |v|
+        self.verbose = v
+      end
+      opts.separator ""
+      opts.separator "Models diagram options:"
+      opts.on("-a", "--all", "Include all models", 
+              "  (not only ActiveRecord::Base derived)") do |a|
+        self.all = a
+      end
+      opts.on("--hide-types", "Hide attributes type") do |h|
+        self.hide_types = h
+      end
+      opts.on("-j", "--join", "Concentrate edges") do |j|
+        self.join = j
+      end
+      opts.on("-m", "--modules", "Include modules") do |m|
+        self.modules = m
+      end
+      opts.separator ""
+      opts.separator "Controllers diagram options:"
+      opts.on("--hide-public", "Hide public methods") do |h|
+        self.hide_public = h
+      end
+      opts.on("--hide-protected", "Hide protected methods") do |h|
+        self.hide_protected = h
+      end
+      opts.on("--hide-private", "Hide private methods") do |h|
+        self.hide_private = h
+      end
+      opts.separator ""
+      opts.separator "Other options:"
+      opts.on("-h", "--help", "Show this message") do
+        STDOUT.print "#{opts}\n"
+        exit
+      end
+      opts.on("--version", "Show version and copyright") do
+        STDOUT.print "#{APP_HUMAN_NAME} version #{APP_VERSION.join('.')}\n\n" +
+                     "#{COPYRIGHT}\nThis is free software; see the source " +
+                     "for copying conditions.\n\n"
+        exit
+      end
+      opts.separator ""
+      opts.separator "Commands (you must supply one of these):"
+      opts.on("-M", "--models", "Generate models diagram") do |c|
+        if self.command == 'controllers'
+          STDERR.print "Error: Can't generate models AND " + 
+                       "controllers diagram\n\n"
+          exit 1
+        else 
+          self.command = 'models'        
+        end
+      end 
+      opts.on("-C", "--controllers", "Generate controllers diagram") do |c|
+        if self.command == 'models'
+          STDERR.print "Error: Can't generate models AND " +
+                       "controllers diagram\n\n"
+          exit 1
+        else 
+          self.command = 'controllers'        
+        end
+      end
+      opts.separator ""
+      opts.separator "For bug reporting and additional information, please see:"
+      opts.separator "http://railroad.rubyforge.org"
+    end # do
+
+    begin
+      @opt_parser.parse!(args)
+    rescue OptionParser::AmbiguousOption
+      option_error "Ambiguous option"
+    rescue OptionParser::InvalidOption
+      option_error "Invalid option"
+    rescue OptionParser::InvalidArgument
+      option_error "Invalid argument"
+    rescue OptionParser::MissingArgument
+      option_error "Missing argument"
+    rescue
+      option_error "Unknown error"
+    end
+  end  # parse
+
+  private 
+
+  def option_error(msg)
+    STDERR.print "Error: #{msg}\n\n #{@opt_parser}\n"
+    exit 1
+  end
+
+end  # class OptionsStruct

data/rake.gemspec

@@ -1,16 +1,17 @@
 require 'rubygems'
 SPEC = Gem::Specification.new do |s|
   s.name         = "railroad"
-  s.version      = "0.3.3"
+  s.version      = "0.3.4"
   s.author       = "Javier Smaldone"
   s.email        = "javier@smaldone.com.ar"
   s.homepage     = "http://railroad.rubyforge.org"
   s.platform     = Gem::Platform::RUBY
   s.summary      = "A DOT diagram generator for Ruby on Rail applications"
-  s.files        = ["lib/railroad", "ChangeLog", "COPYING", "rake.gemspec"]
-  s.bindir       = "lib"
+  s.files        = Dir.glob("lib/railroad/*.rb") + 
+                   ["ChangeLog", "COPYING", "rake.gemspec"]
+  s.bindir       = "bin"
   s.executables  = ["railroad"]
   s.default_executable = "railroad"
   s.has_rdoc     = true
-  s.extra_rdoc_files = ["README", "lib/railroad"]
+  s.extra_rdoc_files = ["README", "COPYING"]
 end

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.2
 specification_version: 1
 name: railroad
 version: !ruby/object:Gem::Version 
-  version: 0.3.3
-date: 2007-04-10 00:00:00 -03:00
+  version: 0.3.4
+date: 2007-04-12 00:00:00 -03:00
 summary: A DOT diagram generator for Ruby on Rail applications
 require_paths: 
 - lib
@@ -14,7 +14,7 @@ rubyforge_project:
 description: 
 autorequire: 
 default_executable: railroad
-bindir: lib
+bindir: bin
 has_rdoc: true
 required_ruby_version: !ruby/object:Gem::Version::Requirement 
   requirements: 
@@ -29,7 +29,11 @@ post_install_message:
 authors: 
 - Javier Smaldone
 files: 
-- lib/railroad
+- lib/railroad/models_diagram.rb
+- lib/railroad/app_diagram.rb
+- lib/railroad/controllers_diagram.rb
+- lib/railroad/diagram_graph.rb
+- lib/railroad/options_struct.rb
 - ChangeLog
 - COPYING
 - rake.gemspec
@@ -40,7 +44,7 @@ rdoc_options: []
 
 extra_rdoc_files: 
 - README
-- lib/railroad
+- COPYING
 executables: 
 - railroad
 extensions: []

data/lib/railroad

@@ -1,459 +0,0 @@
-#!/usr/bin/env ruby
-
-# RailRoad - RoR diagrams generator
-# http://railroad.rubyforge.org
-#
-#   RailRoad generates models and controllers diagrams in DOT language
-#   for a Rails application.
-#
-# Copyright 2007 - Javier Smaldone (http://www.smaldone.com.ar)
-#
-# This program is free software; you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation; either version 2 of the License, or
-# (at your option) any later version.
-#
-
-APP_NAME  = "railroad"
-APP_HUMAN_NAME = "RailRoad"
-APP_VERSION   = [0,3,3]
-COPYRIGHT = "Copyright (C) 2007 Javier Smaldone"
-
-require 'ostruct'
-
-# Command line options structure
-class OptionsStruct < OpenStruct
-
-  require 'optparse'
-
-  def initialize
-    init_options = { :all => false,
-                     :brief => false,
-                     :inheritance => false,
-                     :join => false,
-                     :label => false,
-                     :modules => false,
-                     :hide_types => false,
-                     :hide_public => false,
-                     :hide_protected => false,
-                     :hide_private => false,
-                     :command => '' }
-    super(init_options)
-  end # initialize
-
-  def parse(args)
-    @opt_parser = OptionParser.new do |opts|
-      opts.banner = "Usage: #{APP_NAME} [options] command"
-      opts.separator ""
-      opts.separator "Common options:"
-      opts.on("-b", "--brief", "Generate compact diagram", 
-              "  (no attributes nor methods)") do |b|
-        self.brief = b
-      end
-      opts.on("-i", "--inheritance", "Include inheritance relations") do |i|
-        self.inheritance = i
-      end
-      opts.on("-l", "--label", "Add a label with diagram information", 
-              "  (type, date, migration, version)") do |l|
-        self.label = l
-      end
-      opts.on("-o", "--output FILE", "Write diagram to file FILE") do |f|
-        self.output = f
-      end
-      opts.separator ""
-      opts.separator "Models diagram options:"
-      opts.on("-a", "--all", "Include all models", 
-              "  (not only ActiveRecord::Base derived)") do |a|
-        self.all = a
-      end
-      opts.on("--hide-types", "Hide attributes type") do |h|
-        self.hide_types = h
-      end
-      opts.on("-j", "--join", "Concentrate edges") do |j|
-        self.join = j
-      end
-      opts.on("-m", "--modules", "Include modules") do |m|
-        self.modules = m
-      end
-      opts.separator ""
-      opts.separator "Controllers diagram options:"
-      opts.on("--hide-public", "Hide public methods") do |h|
-        self.hide_public = h
-      end
-      opts.on("--hide-protected", "Hide protected methods") do |h|
-        @options.hide_protected = h
-      end
-      opts.on("--hide-private", "Hide private methods") do |h|
-        @options.hide_private = h
-      end
-      opts.separator ""
-      opts.separator "Other options:"
-      opts.on("-h", "--help", "Show this message") do
-        print "#{opts}\n"
-        exit
-      end
-      opts.on("--version", "Show version and copyright") do
-        print "#{APP_HUMAN_NAME} version #{APP_VERSION.join('.')}\n\n" +
-              "#{COPYRIGHT}\n" +
-              "This is free software; see the source for copying conditions.\n\n"
-        exit
-      end
-      opts.separator ""
-      opts.separator "Commands (you must supply one of these):"
-      opts.on("-M", "--models", "Generate models diagram") do |c|
-         if self.command == 'controllers'
-           STDERR.print "Error: Can't generate models AND " + 
-                        "controllers diagram\n\n"
-           exit 1
-         else 
-           self.command = 'models'        
-         end
-      end
-      opts.on("-C", "--controllers", "Generate controllers diagram") do |c|
-         if self.command == 'models'
-           STDERR.print "Error: Can't generate models AND " +
-                        "controllers diagram\n\n"
-           exit 1
-         else 
-           self.command = 'controllers'        
-         end
-      end
-      opts.separator ""
-      opts.separator "For bug reporting and additional information, please see:"
-      opts.separator "http://railroad.rubyforge.org"
-    end
-
-    begin
-      @opt_parser.parse!(args)
-    rescue OptionParser::AmbiguousOption
-      option_error "Ambiguous option"
-    rescue OptionParser::InvalidOption
-      option_error "Invalid option"
-    rescue OptionParser::InvalidArgument
-      option_error "Invalid argument"
-    rescue OptionParser::MissingArgument
-      option_error "Missing argument"
-    rescue
-      option_error "Unknown error"
-    end
-  end  # initialize
-
-  private 
-
-  def option_error(msg)
-    STDERR.print "Error: #{msg}\n\n #{@opt_parser}\n"
-    exit 1
-  end
-
-end  # class OptionsStruct
-
-
-# Root class for RailRoad diagrams
-class AppDiagram
-
-  def initialize(options)
-    @options = options
-    load_environment
-    load_classes
-  end
-
-  private 
-
-  # Quotes a class name
-  def node(name)
-    '"' + name + '"'
-  end
-
-  # Prevents Rails application from writing to STDOUT
-  def disable_stdout
-    @old_stdout = STDOUT.dup
-    STDOUT.reopen( PLATFORM =~ /mswin/ ? "NUL" : "/dev/null" )
-  end
-
-  # Restore STDOUT  
-  def enable_stdout
-    STDOUT.reopen(@old_stdout)
-  end
-
-  # Print diagram header
-  def print_header(type)
-    print "digraph #{type.downcase}_diagram {\n" +
-          "\tgraph[overlap=false, splines=true]\n"
-    print_info(type) if @options.label
-  end
-  
-
-  # Print diagram label
-  def print_info(type)
-    print "\t_diagram_info [shape=\"plaintext\", label=\"Diagram: #{type}\\l" +
-          "Date: #{Time.now.strftime "%b %d %Y - %H:%M"}\\l" + 
-          "Migration version: #{ActiveRecord::Migrator.current_version}\\l" +
-          "Generated by #{APP_HUMAN_NAME} #{APP_VERSION.join('.')}\\l\""+
-          ", fontsize=14]\n"
-  end
-
-  # Print an edge
-  def print_edge(from, to, attributes)
-    print "\t#{node(from)} -> #{node(to)} [#{attributes}]\n"
-  end
-
-  # Print a node
-  def print_node(name, attributes=nil)
-    print "\t#{node(name)}"
-    print " [#{attributes}]" if attributes && attributes != ''
-    print "\n"
-  end
-
-  # Print error when loading Rails application
-  def print_error(type)
-    STDERR.print "Error loading #{type}.\n  (Are you running " +
-                 "#{APP_NAME} on the aplication's root directory?)\n\n"
-  end
-
-  # Load Rails application's environment
-  def load_environment
-    begin
-      disable_stdout
-      require "config/environment"
-      enable_stdout
-    rescue LoadError
-      enable_stdout
-      print_error "application environment"
-      raise
-    end
-  end
-
-end # class AppDiagram
-
-
-# RailRoad models diagram
-class ModelsDiagram < AppDiagram
-
-  def initialize(options)
-    super options 
-    # Processed habtm associations
-    @habtm = []
-  end
-
-  # Generate models diagram
-  def generate
-    print_header 'Models'
-    models_files = Dir.glob("app/models/**/*.rb")
-    models_files.each do |m|
-      # Extract the class name from the file name
-      class_name = m.split('/')[2..-1].join('/').split('.').first.camelize
-      process_class class_name.constantize  
-    end 
-    print "}\n"
-  end # generate
-
-  private
-
-  # Load model classes
-  def load_classes
-    begin
-      disable_stdout
-      Dir.glob("app/models/**/*.rb") {|m| require m }
-      enable_stdout
-    rescue LoadError
-      enable_stdout
-      print_error "model classes"
-      raise
-    end
-  end  # load_classes
-
-  # Process model class
-  def process_class(current_class)
-
-    class_printed = false
-    
-    # Is current_clas derived from ActiveRecord::Base?
-    if current_class.respond_to?'reflect_on_all_associations'
-
-      # Print the node
-      if @options.brief
-        node_attrib = ''
-      else 
-        node_attrib = 'shape=Mrecord, label="{' + current_class.name + '|'
-        current_class.content_columns.each do |a|
-          node_attrib += a.name
-          node_attrib += ' :' + a.type.to_s unless @options.hide_types
-          node_attrib += '\l'
-        end
-        node_attrib += '}"'
-      end
-      print_node current_class.name, node_attrib
-      class_printed = true
-      # Iterate over the class associations
-      current_class.reflect_on_all_associations.each do |a|
-        process_association(current_class, a)
-      end
-    elsif @options.all && (current_class.is_a? Class)
-      # Not ActiveRecord::Base model
-      if @options.brief
-        node_attrib = 'shape=box'
-      else 
-        node_attrib = 'shape=record, label="{' + current_class.name + '|}"'
-      end
-      print_node current_class.name, node_attrib
-      class_printed = true
-    elsif @options.modules && (current_class.is_a? Module)
-        print_node current_class.name, 
-                   'shape=box, style=dotted, label="' + current_class.name + '"'        
-    end
-
-    # Only consider meaningful inheritance relations for printed classes
-    if @options.inheritance && class_printed && 
-       (current_class.superclass != ActiveRecord::Base) &&
-       (current_class.superclass != Object)
-      print_edge(current_class.name, current_class.superclass.name, 
-                 'arrowhead="onormal"')
-    end      
-
-  end # process_class
-
-  # Process model association
-  def process_association(current_class, association)
-
-    # Skip "belongs_to" associations
-    return if association.macro.to_s == 'belongs_to'
- 
-    assoc_attrib = ""
-    assoc_name = ""
-    # Only non standard association names needs a label
-    if association.class_name != association.name.to_s.singularize.camelize
-      association_name = association.name.to_s
-      assoc_attrib += 'label="' + association_name + '", '
-    end
-
-    # Tail label with the association arity
-    assoc_attrib += association.macro.to_s == 'has_one' ? 'taillabel="1"' : 'taillabel="n"'
-
-    # Use double-headed arrows for habtm and has_many, :through
-    if association.macro.to_s == 'has_and_belongs_to_many' ||
-       (association.macro.to_s == 'has_many' && association.options[:through])
-      
-      # Skip habtm associations already considered
-      return if @habtm.include? [association.class_name, current_class.name, 
-                          association_name]
-      @habtm << [current_class.name, association.class_name,association_name]
-      assoc_attrib += ', headlabel="n", arrowtail="normal"'
-    end  
-    print_edge(current_class.name, association.class_name, assoc_attrib)
-  end # process_association
-
-end # class ModelsDiagram
-
-
-# RailRoad controllers diagram
-class ControllersDiagram < AppDiagram
- 
-  def initialize(options)
-    super options
-    @app_controller = ApplicationController
-  end
-
-  # Generate controllers diagram
-  def generate
-
-    print_header 'Controllers'
-
-    controllers_files = Dir.glob("app/controllers/**/*_controller.rb")
-    controllers_files << 'app/controllers/application.rb'
-    controllers_files.each do |c|
-      # Extract the class name from the file name
-      class_name = c.split('/')[2..-1].join('/').split('.').first.camelize
-      # ApplicationController's file is 'application.rb'
-      class_name += 'Controller' if class_name == 'Application'
-      process_class class_name.constantize
-    end 
-    print "}\n"
-  end # generate
-
-  private
-
-  # Load controller classes
-  def load_classes
-    begin
-      disable_stdout
-      # ApplicationController must be loaded first
-      require "app/controllers/application.rb" 
-      Dir.glob("app/controllers/**/*_controller.rb") {|c| require c }
-      enable_stdout
-    rescue LoadError
-      enable_stdout
-      print_error "controller classes"
-      raise
-    end
-  end # load_classes
-
-  # Proccess controller class
-  def process_class(current_class)
-
-    if @options.brief
-      print_node  current_class.name
-
-    elsif current_class.is_a? Class 
-      node_attrib = 'shape=Mrecord, label="{' + current_class.name + '|'
-      current_class.public_instance_methods(false).sort.each { |m|
-        node_attrib += m + '\l' 
-      } unless @options.hide_public
-      node_attrib += '|'
-      current_class.protected_instance_methods(false).sort.each { |m|
-        node_attrib += m + '\l'
-      } unless @options.hide_protected
-      node_attrib += '|'
-      current_class.private_instance_methods(false).sort.each { |m|
-        node_attrib += m + '\l'
-      } unless @options.hide_private
-      node_attrib += '}"' 
-      print_node current_class.name, node_attrib 
-
-    elsif @options.modules && current_class.is_a?(Module)
-        print_node current_class.name, 
-                   'shape=box, style=dotted, label="' + current_class.name + '"'
-    end
-
-    # Print the inheritance edge (only for ApplicationControllers)
-    if @options.inheritance && 
-       (@app_controller.subclasses.include? current_class.name)
-      print_edge(current_class.name, current_class.superclass.name, 
-                 'arrowhead="onormal"')
-    end
-
-  end # process_class
-
-end # class ControllersDiagram
-
-
-# Main program
-
-options = OptionsStruct.new
-
-options.parse ARGV
-
-if options.command == 'models'
-  diagram = ModelsDiagram.new options
-elsif options.command == 'controllers'
-  diagram = ControllersDiagram.new options
-else
-  STDERR.print "Error: You must supply a command\n" +
-               "  (try #{APP_NAME} -h)\n\n"
-  exit 1
-end
-
-if options.output
-  old_stdout = STDOUT.dup
-  begin
-     STDOUT.reopen(options.output)
-  rescue
-     STDERR.print "Error: Cannot write diagram to #{options.output}\n\n"
-     exit 2
-  end
-end
-
-diagram.generate
-
-if options.output
-  STDOUT.reopen(old_stdout)
-end