allow multiple states to be defined on one line

README.md

@@ -28,8 +28,7 @@ class Job
 
   aasm do
     state :sleeping, :initial => true
-    state :running
+    state :running, :cleaning
-    state :cleaning
 
     event :run do
       transitions :from => :sleeping, :to => :running

lib/aasm/base.rb

@@ -63,7 +63,24 @@ module AASM
     end
 
     # define a state
-    def state(name, options={})
+    # args
+    # [0] state
+    # [1] options (or nil)
+    # or
+    # [0] state
+    # [1..] state
+    def state(*args)
+      if args.last.is_a?(Hash) && args.size == 2
+        names = [args.first]
+        options = args.last
+      elsif
+        names = args
+        options = {}
+      else
+        raise "count not parse states: #{args}"
+      end
+
+      names.each do |name|
         @state_machine.add_state(name, @klass, options)
 
         if @klass.instance_methods.include?("#{name}?")
@@ -80,6 +97,7 @@ module AASM
           @klass.const_set("STATE_#{name.upcase}", name)
         end
       end
+    end
 
     # define an event
     def event(name, options={}, &block)

spec/models/states_on_one_line_example.rb

@@ -0,0 +1,8 @@
+class StatesOnOneLineExample
+  include AASM
+
+  aasm :one_line do
+    state :initial, :initial => true
+    state :first, :second
+  end
+end

spec/unit/readme_spec.rb

@@ -8,8 +8,7 @@ describe 'testing the README examples' do
 
       aasm do
         state :sleeping, :initial => true
-        state :running
+        state :running, :cleaning
-        state :cleaning
 
         event :run do
           transitions :from => :sleeping, :to => :running

spec/unit/states_on_one_line_example_spec.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+
+describe StatesOnOneLineExample do
+  let(:example) { StatesOnOneLineExample.new }
+  describe 'on initialize' do
+    it 'should be in the initial state' do
+      expect(example.aasm(:one_line).current_state).to eql :initial
+    end
+  end
+
+  describe 'states' do
+    it 'should have all 3 states defined' do
+      expect(example.aasm(:one_line).states.map(&:name)).to eq [:initial, :first, :second]
+    end
+  end
+end