feat: add update methods to SharedFunctionsManager, SharedStatesManager, and SharedSubscriptionsManager for modifying shared data

Author: HichemTab-tech

src/hooks/use-shared-function.ts

@@ -55,6 +55,22 @@ export class SharedFunctionsApi extends SharedValuesApi<SharedFunction<unknown>,
         }
         super.set(key, value, scopeName);
     }
+    update<T, Args extends unknown[], S extends string = string>(key: S, updater: (prev: SharedFunctionValue<T>) => { fnState: SharedFunctionValue<T> }, scopeName?: Prefix): void;
+    update<T, Args extends unknown[]>(sharedFunctionCreated: SharedFunctionCreated<T, Args>, updater: (prev: SharedFunctionValue<T>) => { fnState: SharedFunctionValue<T> }): void;
+    update<T, Args extends unknown[], S extends string = string>(key: S | SharedFunctionCreated<T, Args>, updater: (prev: SharedFunctionValue<T>) => { fnState: SharedFunctionValue<T> }, scopeName: Prefix = "_global") {
+        let prev: SharedFunctionValue<T>;
+        if (typeof key === "string") {
+            prev = this.get(key, scopeName);
+        } else {
+            prev = this.get(key);
+        }
+        const newValue = updater(prev);
+        if (typeof key === "string") {
+            this.set(key, newValue, scopeName);
+        } else {
+            this.set(key, newValue);
+        }
+    }
 }
 
 const sharedFunctionsManager = new SharedFunctionsManager();

src/hooks/use-shared-state.ts

@@ -47,6 +47,22 @@ export class SharedStatesApi extends SharedValuesApi<SharedState<unknown>, { val
         }
         super.set(key, {value}, scopeName);
     }
+    update<T, S extends string = string>(key: S, updater: (prev: T) => T, scopeName?: Prefix): void;
+    update<T>(sharedStateCreated: SharedStateCreated<T>, updater: (prev: T) => T): void;
+    update<T, S extends string = string>(key: S | SharedStateCreated<T>, updater: (prev: T) => T, scopeName: Prefix = "_global") {
+        let prev: T;
+        if (typeof key === "string") {
+            prev = this.get(key, scopeName);
+        } else {
+            prev = this.get(key);
+        }
+        const newValue = updater(prev);
+        if (typeof key === "string") {
+            this.set(key, newValue, scopeName);
+        } else {
+            this.set(key, newValue);
+        }
+    }
 }
 
 const sharedStatesManager = new SharedStatesManager();

src/hooks/use-shared-subscription.ts

@@ -92,6 +92,22 @@ export class SharedSubscriptionsApi extends SharedValuesApi<SharedSubscription<u
         }
         super.set(key, value, scopeName);
     }
+    update<T, S extends string = string>(key: S, updater: (prev: SharedSubscriptionValue<T>) => { fnState: SharedSubscriptionValue<T> }, scopeName?: Prefix): void;
+    update<T>(sharedSubscriptionCreated: SharedSubscriptionCreated<T>, updater: (prev: SharedSubscriptionValue<T>) => { fnState: SharedSubscriptionValue<T> }): void;
+    update<T, S extends string = string>(key: S | SharedSubscriptionCreated<T>, updater: (prev: SharedSubscriptionValue<T>) => { fnState: SharedSubscriptionValue<T> }, scopeName: Prefix = "_global") {
+        let prev: SharedSubscriptionValue<T>;
+        if (typeof key === "string") {
+            prev = this.get(key, scopeName);
+        } else {
+            prev = this.get(key);
+        }
+        const newValue = updater(prev);
+        if (typeof key === "string") {
+            this.set(key, newValue, scopeName);
+        } else {
+            this.set(key, newValue);
+        }
+    }
 }
 
 const sharedSubscriptionsManager = new SharedSubscriptionsManager();