class AasmDiagram

# 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

# 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

# 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

# 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

# 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

# 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

# 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