class AasmDiagram
Diagram for Acts As State Machine
Public Class Methods
new(options = OptionsStruct.new)
click to toggle source
Calls superclass method
AppDiagram::new
# File lib/prailroady/aasm_diagram.rb, line 13 def initialize(options = OptionsStruct.new) # options.exclude.map! {|e| e = "app/models/" + e} super options @graph.diagram_type = 'Models' # Processed habtm associations @habtm = [] end
Public Instance Methods
generate()
click to toggle source
Process model files
# File lib/prailroady/aasm_diagram.rb, line 22 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
get_files(prefix = '')
click to toggle source
# File lib/prailroady/aasm_diagram.rb, line 29 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(prefix + 'app/models/concerns/**/*.rb') unless @options.include_concerns files -= Dir.glob(@options.exclude) files end
Private Instance Methods
load_classes()
click to toggle source
Load model classes
# File lib/prailroady/aasm_diagram.rb, line 40 def load_classes disable_stdout get_files.each { |m| require m } enable_stdout rescue LoadError enable_stdout print_error 'model classes' raise end
process_aasm_class(current_class)
click to toggle source
# File lib/prailroady/aasm_diagram.rb, line 83 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| event.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
process_acts_as_state_machine_class(current_class)
click to toggle source
# File lib/prailroady/aasm_diagram.rb, line 59 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.aasm.states.each do |state_name| 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| event_name = event.name event.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
process_class(current_class)
click to toggle source
Process a model class
# File lib/prailroady/aasm_diagram.rb, line 51 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) || current_class.respond_to?(:aasm) end