Merge pull request #15 from techno-dwarf-works/feature/clean-up

Version 0.1.8
techno-dwarf-works · May 19, 2024 · 9378096 · 9378096
1 parent 97e7287
commit 9378096
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/Runtime/Modules/CacheModule.cs b/Runtime/Modules/CacheModule.cs
@@ -90,8 +90,20 @@ public T GetState<T>()
             DebugUtility.LogException<InvalidOperationException>(message);
             return state;
         }
-
-        public bool RemoveState<T>() where T : TState
+
+        public T GetOrAddState<T>()
+            where T : TState, new()
+        {
+            if (!TryGetState(out T state))
+            {
+                state = CacheState<T>();
+            }
+
+            return state;
+        }
+
+        public bool RemoveState<T>()
+            where T : TState
         {
             var type = typeof(T);
             return _stateLocator.Remove(type);

diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "com.tdw.better.statemachine",
   "displayName": "Better State Machine",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "unity": "2021.3",
   "description": " ",
   "dependencies": {