Class: AasmDiagram

Inherits:
AppDiagram show all
Defined in:
lib/railroad/aasm_diagram.rb

Overview

Diagram for Acts As State Machine

Instance Method Summary collapse

Methods inherited from AppDiagram

#print

Constructor Details

#initialize(options) ⇒ AasmDiagram

Returns a new instance of AasmDiagram.



14
15
16
17
18
19
20
 
# File 'lib/railroad/aasm_diagram.rb', line 14

def initialize(options)
  #options.exclude.map! {|e| e = "app/models/" + e}
  super options 
  @graph.diagram_type = 'Models'
  # Processed habtm associations
  @habtm = []
end

Instance Method Details

#generate ⇒ Object

Process model files



23
24
25
26
27
28
29
30
31
 
# File 'lib/railroad/aasm_diagram.rb', line 23

def generate
  STDERR.print "Generating AASM diagram\n" if @options.verbose
  files = Dir.glob("app/models/**/*.rb") 
  files += Dir.glob("vendor/plugins/**/app/models/*.rb") if @options.plugins_models
  files -= @options.exclude
  files.each do |f| 
    process_class extract_class_name(f).constantize
  end
end