tobias-railroad 0.10.0

data/README.rdoc

@@ -0,0 +1,169 @@
+= RailRoad 
+
+RailRoad generates models and controllers diagrams in DOT language for a 
+Rails application.
+
+This is a patched version based on the original v0.5.0.
+
+I (Peter Hoeg) am not trying to hijack Javier's project, but since he hasn't released any
+new versions since May '08, I figured I'd better put one out in order to make
+railroad work with rails v2.3.
+
+= Usage
+
+Run RailRoad on the Rails application's root directory. You can redirect its
+output to a .dot file or pipe it to the dot or neato utilities to produce a
+graphic. Model diagrams are intended to be processed using dot and
+controller diagrams are best processed using neato.
+
+    railroad [options] command
+ 
+== Options
+
+Common options:
+    -b, --brief                      Generate compact diagram
+                                       (no attributes nor methods)
+    -s, --specify file1[,fileN]      Specify given files only for the diagram
+                                       (can take a glob pattern)
+    -e, --exclude file1[,fileN]      Exclude given files
+                                       (can take a glob pattern)
+    -i, --inheritance                Include inheritance relations
+    -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
+                                       (not only ActiveRecord::Base derived)
+        --hide-magic                 Hide magic field names
+        --hide-types                 Hide attributes type
+    -j, --join                       Concentrate edges
+    -m, --modules                    Include modules
+    -p, --plugins-models             Include plugins models
+    -t, --transitive                 Include transitive associations
+                                     (through inheritance)
+
+Controllers diagram options:
+        --hide-public                Hide public methods
+        --hide-protected             Hide protected methods
+        --hide-private               Hide private methods
+
+Other options:
+    -h, --help                       Show this message
+        --version                    Show version and copyright
+
+== Commands
+
+    -M, --models                     Generate models diagram
+    -C, --controllers                Generate controllers diagram
+    -A, --aasm                       Generate "acts as state machine" diagram
+
+
+== Examples
+
+    railroad -o models.dot -M 
+      Produces a models diagram to the file 'models.dot'
+    railroad -a -i -o full_models.dot -M 
+      Models diagram with all classes showing inheritance relations
+    railroad -M | dot -Tsvg > models.svg
+      Model diagram in SVG format
+    railroad -C | neato -Tpng > controllers.png
+      Controller diagram in PNG format
+    railroad -h
+      Shows usage help
+
+
+= Processing DOT files
+
+To produce a PNG image from model diagram generated by RailRoad you can
+issue the following command:
+
+    dot -Tpng models.dot > models.png
+
+If you want to do the same with a controller diagram, use neato instead of
+dot:
+
+    neato -Tpng controllers.dot > controllers.png
+
+If you want to produce SVG (vectorial, scalable, editable) files, you can do
+the following:
+
+    dot -Tsvg models.dot > models.svg
+    neato -Tsvg controllers.dot > controllers.svg
+
+Important: There is a bug in Graphviz tools when generating SVG files that
+cause a text overflow. You can solve this problem editing (with a text
+editor, not a graphical SVG editor) the file and replacing around line 12
+"font-size:14.00;" by "font-size:11.00;", or by issuing the following command 
+(see "man sed"):
+
+    sed -i 's/font-size:14.00/font-size:11.00/g' file.svg
+
+Note: For viewing and editing SVG there is an excellent opensource tool
+called Inkscape (similar to Adobe Illustrator. For DOT processing you can
+also use Omnigraffle (on Mac OS X).
+
+= RailRoad as a rake task
+
+(Thanks to Thomas Ritz, http://www.galaxy-ritz.de ,for the code.)
+
+In your Rails application, put the following rake tasks into 'lib/task/diagrams.rake':
+
+  namespace :doc do
+    namespace :diagram do
+      task :models do
+        sh "railroad -i -l -a -m -M | dot -Tsvg | sed 's/font-size:14.00/font-size:11.00/g' > doc/models.svg"
+      end
+  
+      task :controllers do
+        sh "railroad -i -l -C | neato -Tsvg | sed 's/font-size:14.00/font-size:11.00/g' > doc/controllers.svg"
+      end
+    end
+  
+    task :diagrams => %w(diagram:models diagram:controllers)
+  end
+  
+Then, 'rake doc:diagrams' produces 'doc/models.svg' and 'doc/controllers.svg'.
+
+= Requirements
+
+RailRoad has been tested with the following Ruby and Rails versions
+
+== Ruby
+* 1.8.5
+* 1.8.7
+
+== Rails
+* 1.1.6 to 1.2.3
+* 2.3.2 to 2.3.5
+
+There are no additional requirements (nevertheless, all your Rails application
+requirements must be installed).
+
+In order to view/export the DOT diagrams, you'll need the processing tools
+from Graphviz.
+
+= Website and Project Home
+
+http://railroad.rubyforge.org
+
+= License
+
+RailRoad is distributed  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.
+
+See LICENSE for details.
+
+== Copyright
+
+Copyright (c) 2007-2008 Javier Smaldone
+Copyright (c) 2009 Peter Hoeg
+
+See LICENSE for details.
+
+== Authors
+
+See AUTHORS for details.

data/Rakefile

@@ -0,0 +1,46 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "tobias-railroad"
+    gem.executables = "railroad"
+    gem.email = ['tcrawley@gmail.com', 'peter@hoeg.com', 'p.hoeg@northwind.sg', 'javier@smaldone.com.ar']
+    gem.homepage = "http://github.com/tobias/RailRoad"
+    gem.authors = ["Tobias Crawley", "Peter Hoeg", "Javier Smaldone"]
+    gem.summary = "A DOT diagram generator for Ruby on Rail applications"
+    gem.description = gem.summary
+    gem.files = FileList["[A-Z]*", "{autotest,bin,lib,spec}/**/*", ".document"]
+    gem.extra_rdoc_files = FileList["*.rdoc"]
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :default => :spec
+
+require 'rake/rdoctask'
+require 'lib/railroad/version'
+Rake::RDocTask.new do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "railroad #{RailRoad::VERSION::STRING}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('CHANGELOG*')
+  rdoc.rdoc_files.include('AUTHORS*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION.yml

@@ -0,0 +1,5 @@
+--- 
+:minor: 10
+:build: 
+:patch: 0
+:major: 0

data/bin/railroad

@@ -0,0 +1,53 @@
+#!/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-2008 - 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.
+#
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'railroad'
+
+APP_NAME       = "railroad"
+APP_HUMAN_NAME = "RailRoad"
+# APP_VERSION    = [RailRoad::VERSION::MAJOR, RailRoad::VERSION::MINOR, RailRoad::VERSION::PATCH]
+APP_VERSION    = RailRoad::VERSION::STRING
+COPYRIGHT      = "Copyright (C) 2007-2008 Javier Smaldone, 2009 Peter Hoeg"
+
+options = OptionsStruct.new({:app_name => APP_NAME, :app_human_name => APP_HUMAN_NAME, :app_version =>APP_VERSION, :copyright =>COPYRIGHT})
+
+options.parse ARGV
+
+old_dir = Dir.pwd
+
+Dir.chdir(options.root) if options.root != ''
+
+if options.command == 'models'
+  diagram = ModelsDiagram.new options
+elsif options.command == 'controllers'
+  diagram = ControllersDiagram.new options
+elsif options.command == 'aasm'
+  diagram = AasmDiagram.new(options)
+else
+  STDERR.print "#{APP_HUMAN_NAME} v#{RailRoad::VERSION::STRING}\n" +
+    "Error: You must supply a command\n" +
+    "  (try #{APP_NAME} -h)\n\n"
+  exit 1
+end
+
+diagram.process
+diagram.generate
+
+Dir.chdir(old_dir)
+
+diagram.print
+

data/lib/railroad.rb

@@ -0,0 +1,2 @@
+['version', 'options_struct', 'models_diagram', 'controllers_diagram', 'aasm_diagram'].each { |f| require "railroad/#{f}" }
+

data/lib/railroad/aasm_diagram.rb

@@ -0,0 +1,109 @@
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+# Copyright 2007-2008 - Javier Smaldone (http://www.smaldone.com.ar)
+# See COPYING for more details
+
+# AASM code provided by Ana Nelson (http://ananelson.com/)
+
+require 'railroad/app_diagram'
+
+# Diagram for Acts As State Machine
+class AasmDiagram < AppDiagram
+
+  def initialize(options = OptionsStruct.new)
+    #options.exclude.map! {|e| e = "app/models/" + e}
+    super options
+    @graph.diagram_type = 'Models'
+    # Processed habtm associations
+    @habtm = []
+  end
+
+  # Process model files
+  def generate
+    STDERR.print "Generating AASM diagram\n" if @options.verbose
+    get_files.each do |f|
+      process_class extract_class_name(f).constantize
+    end
+  end
+
+  def get_files(prefix ='')
+    files = !@options.specify.empty? ? Dir.glob(@options.specify) : Dir.glob(prefix << "app/models/**/*.rb")
+    files += Dir.glob("vendor/plugins/**/app/models/*.rb") if @options.plugins_models
+    files -= Dir.glob(@options.exclude)
+    files
+  end
+
+  private
+
+  # Load model classes
+  def load_classes
+    begin
+      disable_stdout
+      get_files.each {|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
+
+    # Only interested in acts_as_state_machine models.
+    process_acts_as_state_machine_class(current_class)  if current_class.respond_to?(:states)
+    process_aasm_class(current_class)  if current_class.respond_to?(:aasm_states)
+  end # process_class
+
+  def process_acts_as_state_machine_class(current_class)
+    node_attribs = []
+    node_type = 'aasm'
+    
+    STDERR.print "\t\tprocessing as acts_as_state_machine\n" if @options.verbose
+    current_class.states.each do |state_name|
+      state = current_class.read_inheritable_attribute(:states)[state_name]
+      node_shape = (current_class.initial_state === state_name) ? ", peripheries = 2" : ""
+      node_attribs << "#{current_class.name.downcase}_#{state_name} [label=#{state_name} #{node_shape}];"
+    end
+    @graph.add_node [node_type, current_class.name, node_attribs]
+
+    current_class.read_inheritable_attribute(:transition_table).each do |event_name, event|
+      event.each do |transition|
+        @graph.add_edge [
+                         'event',
+                         current_class.name.downcase + "_" + transition.from.to_s,
+                         current_class.name.downcase + "_" + transition.to.to_s,
+                         event_name.to_s
+                        ]
+      end
+    end
+  end
+
+  def process_aasm_class(current_class)
+    node_attribs = []
+    node_type = 'aasm'
+
+    STDERR.print "\t\tprocessing as aasm\n" if @options.verbose
+    current_class.aasm_states.each do |state|
+      node_shape = (current_class.aasm_initial_state === state.name) ? ", peripheries = 2" : ""
+      node_attribs << "#{current_class.name.downcase}_#{state.name} [label=#{state.name} #{node_shape}];"
+    end
+    @graph.add_node [node_type, current_class.name, node_attribs]
+
+    current_class.aasm_events.each do |event_name, event|
+      event.all_transitions.each do |transition|
+        @graph.add_edge [
+                         'event',
+                         current_class.name.downcase + "_" + transition.from.to_s,
+                         current_class.name.downcase + "_" + transition.to.to_s,
+                         event_name.to_s
+                        ]
+      end
+    end
+  end
+
+end # class AasmDiagram

data/lib/railroad/app_diagram.rb

@@ -0,0 +1,112 @@
+# RailRoad - RoR diagrams generator
+# http://railroad.rubyforge.org
+#
+# Copyright 2007-2008 - Javier Smaldone (http://www.smaldone.com.ar)
+# See COPYING for more details
+
+require 'railroad/diagram_graph'
+
+# camelize and constantize methods brought over from active_support
+class String
+  def camelize(first_letter_in_uppercase = true)
+    if first_letter_in_uppercase
+      self.to_s.gsub(/\/(.?)/) { "::" + $1.upcase }.gsub(/(^|_)(.)/) { $2.upcase }
+    else
+      self.first + self.camelize[1..-1]
+    end
+  end
+  def constantize
+     names = self.split('::')
+     names.shift if names.empty? || names.first.empty?
+  
+     constant = Object
+     names.each do |name|
+       constant = constant.const_defined?(name) ? constant.const_get(name) : constant.const_missing(name)
+     end
+     constant
+   end
+end
+
+# Root class for RailRoad diagrams
+class AppDiagram
+
+  def initialize(options = OptionsStruct.new)
+    @options = options
+    @graph = DiagramGraph.new
+    @graph.show_label = @options.label
+  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
+    
+    if @options.xmi 
+        STDERR.print "Generating XMI diagram\n" if @options.verbose
+    	STDOUT.print @graph.to_xmi
+    else
+        STDERR.print "Generating DOT graph\n" if @options.verbose
+        STDOUT.print @graph.to_dot 
+    end
+
+    if @options.output
+      STDOUT.reopen(old_stdout)
+    end
+  end # print
+
+  def process
+    load_environment
+  end
+
+  private 
+  
+  # Load Rails application's environment
+  def load_environment
+    STDERR.print "Loading application environment\n" if @options.verbose
+    begin
+      disable_stdout
+      require "config/environment"
+      enable_stdout
+    rescue LoadError
+      enable_stdout
+      print_error "application environment"
+      raise
+    end
+    STDERR.print "Loading application classes as we go\n" if @options.verbose
+  end
+
+  # Prevents Rails application from writing to STDOUT
+  def disable_stdout
+    @old_stdout = STDOUT.dup
+    STDOUT.reopen(::RUBY_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 " +
+                 "#{@options.app_name} on the aplication's root directory?)\n\n"
+  end
+
+  # Extract class name from filename
+  def extract_class_name(filename)
+    #filename.split('/')[2..-1].join('/').split('.').first.camelize
+    # Fixed by patch from ticket #12742
+    # File.basename(filename).chomp(".rb").camelize
+    filename.split('/')[2..-1].collect { |i| i.camelize }.join('::').chomp(".rb")
+  end
+
+end # class AppDiagram