Merge pull request #1 from TannerRogalsky/master

Allow gotoState to pass arguments to the new state.

spec/unit.lua

@@ -104,6 +104,17 @@ describe("Unit tests", function()
         assert_true(e.mark)
       end)
 
+      it("passes all additional arguments to enteredState and exitedState", function()
+        local FooEnemy = Enemy:addState("FooEnemy")
+        local testValue = "Bar"
+
+        local validateVarargs = function(self, ...) assert_equal(..., testValue) end
+        FooEnemy.enteredState = validateVarargs
+        FooEnemy.exitedState = validateVarargs
+
+        local e = Enemy:new()
+        e:gotoState("FooEnemy", testValue)
+      end)
     end)
 
     it("raises an error when given an invalid id", function()

stateful.lua

@@ -89,8 +89,8 @@ local function _assertExistingState(self, state, stateName)
   assert(state, "The state " .. stateName .. " was not found in " .. tostring(self.class) )
 end
 
-local function _invokeCallback(self, state, callbackName)
-  if state then state[callbackName](self) end
+local function _invokeCallback(self, state, callbackName, ...)
+  if state then state[callbackName](self, ...) end
 end
 
 local function _getCurrentState(self)
@@ -133,9 +133,9 @@ function Stateful.static:addState(stateName, superState)
   return self.static.states[stateName]
 end
 
-function Stateful:gotoState(stateName)
+function Stateful:gotoState(stateName, ...)
 
-  _invokeCallback(self, _getCurrentState(self), 'exitedState')
+  _invokeCallback(self, _getCurrentState(self), 'exitedState', ...)
 
   if stateName == nil then
     self.__stateStack = { }
@@ -144,7 +144,7 @@ function Stateful:gotoState(stateName)
 
     local newState = _getStateFromClassByName(self, stateName)
     self.__stateStack = { newState }
-    _invokeCallback(self, newState, 'enteredState')
+    _invokeCallback(self, newState, 'enteredState', ...)
   end
 
 end