module StateMachine
  module YARD
    module Handlers
      # Handles and processes #state
      class State < Base
        handles method_call(:state)
        
        def process
          if owner.is_a?(StateMachine::Machine)
            handler = self
            statement = self.statement
            names = extract_node_names(statement.parameters(false))
            
            names.each do |name|
              owner.state(name) do
                # Parse the block
                handler.parse_block(statement.last.last, :owner => self)
              end
            end
          end
        end
      end
    end
  end
end