Merge pull request #5 from RoadieRich/return-state-in-AddTransitionTo

Adding `return this` to AddTransitionTo to allow chaining methods
RoadieRich · Jan 11, 2024 · 337192a · 337192a
2 parents 46a2fdf + 5a0a0e6
commit 337192a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/Demo/Program.cs b/Demo/Program.cs
@@ -12,11 +12,11 @@
 
 	initFuncState.AddTransitionTo(funcState, (vars) => true);
 
-	incrementFuncState.AddTransitionTo(StateMachine.ExitState, (vars) => vars["x"] > 10);
-	incrementFuncState.AddTransitionTo(funcState, (vars) => true);
+	incrementFuncState.AddTransitionTo(StateMachine.ExitState, (vars) => vars["x"] > 10)
+					  .AddTransitionTo(funcState, (vars) => true);
 
-	funcState.AddTransitionTo(evenFuncState, (vars) => vars["x"] % 2 == 0);
-	funcState.AddTransitionTo(oddFuncState, (vars) => vars["x"] % 2 == 1);
+	funcState.AddTransitionTo(evenFuncState, (vars) => vars["x"] % 2 == 0)
+			 .AddTransitionTo(oddFuncState, (vars) => vars["x"] % 2 == 1);
 
 	evenFuncState.AddTransitionTo(incrementFuncState, (vars) => true);
 

diff --git a/StateMachine/RoadieRichStateMachine.csproj b/StateMachine/RoadieRichStateMachine.csproj
@@ -13,7 +13,7 @@
 	  <RepositoryType>git</RepositoryType>
 	  <Authors>RoadieRich</Authors>
 	  <PackageRequireLicenseAcceptance>True</PackageRequireLicenseAcceptance>
-	  <Version>1.0.4</Version>
+	  <Version>1.0.5</Version>
   </PropertyGroup>
 
   <ItemGroup>

diff --git a/StateMachine/State.cs b/StateMachine/State.cs
@@ -14,9 +14,11 @@ public abstract class State : IDisposable
 		/// <param name="to">state to transition to</param>
 		/// <param name="condition">condition to transition.  Evaluated each time <see cref="Inner(Dictionary{string, dynamic})"/> is run.  If true, the state machine moves to the associated state.  Use <c>null</c> to always transition.</param>
 		/// <remarks>Transition conditions are evaulated in the order they are added.</remarks>
-		public void AddTransitionTo(State to, TransitionConditionDelegate? condition)
+		/// <returns>The state this method was called on</returns>
+		public State AddTransitionTo(State to, TransitionConditionDelegate? condition)
 		{
 			transitions.Add(new Transition(to, condition));
+			return this;
 		}
 
 		internal State RunAndGetNextState(int delay, IDictionary<string, dynamic> vars)