fix: update createSharedSubscription to accept options and improve data handling

Author: HichemTab-tech

demo/app.tsx

@@ -52,7 +52,7 @@ const Comp1 = () => {
 
 
 
-const s = createSharedSubscription<string>((set, onError, onCompletion) => {
+const s = createSharedSubscription((set, onError, onCompletion) => {
 
     return FakeSharedEmitter.subscribe("x", (data: string) => {
         if (data === "do-error") {
@@ -63,22 +63,24 @@ const s = createSharedSubscription<string>((set, onError, onCompletion) => {
         console.log("data loaded...", data);
     }, onError, onCompletion)
 
+}, {
+    initialValue: ""
 })
 
 const use = () => {
     return useSharedSubscription(s);
 }
 
 const Comp2 = () => {
-    const {state, trigger, unsubscribe} = use();
+    const {state: {data, ...state}, trigger, unsubscribe} = use();
 
 
     return (
         <div>
             <h1 className="text-red-600">Comp2 - {state.isLoading && "loading"}</h1>
             <button onClick={() => trigger()}>subscribe</button>
             <button onClick={() => unsubscribe()}>unsubscribe</button>
-            results: {state.data}
+            results: {data}
             <br/>
         </div>
     )

src/hooks/use-shared-subscription.ts

@@ -14,7 +14,7 @@ export namespace SubscriberEvents {
 export type Subscriber<T> = (set: SubscriberEvents.Set<T>, onError: SubscriberEvents.OnError, onCompletion: SubscriberEvents.OnCompletion) => PotentialPromise<Unsubscribe | void | undefined>;
 
 export type SharedSubscriptionValue<T> = {
-    data?: T;
+    data: T;
     isLoading: boolean;
     error?: unknown;
     subscribed?: boolean
@@ -24,11 +24,12 @@ interface SharedSubscription<T> extends SharedSubscriptionValue<T> {
     unsubscribe?: Unsubscribe | void;
 }
 
-const sharedSubscriptionsManager = new SharedValuesManager<SharedSubscription<any>>(() => defaultValue);
+const sharedSubscriptionsManager = new SharedValuesManager<SharedSubscription<any>>();
 export const sharedSubscriptionsApi = new SharedValuesApi<SharedSubscription<any>>(sharedSubscriptionsManager);
 
 interface SharedSubscriptionCreated<T> extends SharedCreated {
-    subscriber: Subscriber<T>
+    subscriber: Subscriber<T>;
+    triggerImmediately?: boolean;
 }
 
 const defaultValue: SharedSubscription<any> = {
@@ -39,8 +40,11 @@ const defaultValue: SharedSubscription<any> = {
     unsubscribe: undefined,
 };
 
-export const createSharedSubscription = <T>(subscriber: Subscriber<T>, scopeName?: Prefix): SharedSubscriptionCreated<T> => {
-    return sharedSubscriptionsManager.createStatic<SharedSubscriptionCreated<T>>({subscriber}, defaultValue, scopeName);
+export const createSharedSubscription = <T>(subscriber: Subscriber<T>, options?: {
+    initialValue?: T,
+    triggerImmediately?: boolean,
+}, scopeName?: Prefix): SharedSubscriptionCreated<T> => {
+    return sharedSubscriptionsManager.createStatic<SharedSubscriptionCreated<T>>({subscriber}, {...defaultValue, data: options?.initialValue as T}, scopeName);
 }
 
 export type SharedSubscriptionStateReturn<T> = {
@@ -69,12 +73,14 @@ export function useSharedSubscription<T, S extends string = string>(
     let keyStr: string;
     let subscriberVal!: Subscriber<T>;
     let scope: Prefix | undefined = scopeName;
+    let triggerImmediately = false;
 
     if (typeof key !== "string") {
-        const {key: key2, subscriber: sub, prefix: prefix2} = key;
+        const {key: key2, subscriber: sub, prefix: prefix2, triggerImmediately: _triggerImmediately} = key;
         keyStr = key2;
         subscriberVal = sub;
         scope = prefix2;
+        triggerImmediately = _triggerImmediately??false;
     } else {
         keyStr = ensureNonEmptyString(key);
         subscriberVal = subscriber as Subscriber<T>;
@@ -147,6 +153,12 @@ export function useSharedSubscription<T, S extends string = string>(
 
     sharedSubscriptionsManager.useEffect(keyStr, prefix);
 
+    useEffect(() => {
+        if (triggerImmediately) {
+            void trigger(false);
+        }
+    }, []);
+
     return {
         state,
         trigger: () => {