feat: Enhanced withCallState to accept mutliple collection configuration

Author: abhijit-chikane

libs/ngrx-toolkit/src/lib/with-call-state.spec.ts

@@ -15,7 +15,7 @@ describe('withCallState', () => {
   it('should use the callState for a collection', () => {
     const DataStore = signalStore(
       { protectedState: false },
-      withCallState({ collection: 'entities' })
+      withCallState({ collections: ['entities'] })
     );
     const dataStore = new DataStore();
 

libs/ngrx-toolkit/src/lib/with-call-state.ts

@@ -31,18 +31,17 @@ export type SetCallState<Prop extends string | undefined> = Prop extends string
   ? NamedCallStateSlice<Prop>
   : CallStateSlice;
 
-export function getCallStateKeys(config?: { collection?: string }) {
-  const prop = config?.collection;
+export function getCallStateKeys(collection?: string) {
   return {
-    callStateKey: prop ? `${config.collection}CallState` : 'callState',
-    loadingKey: prop ? `${config.collection}Loading` : 'loading',
-    loadedKey: prop ? `${config.collection}Loaded` : 'loaded',
-    errorKey: prop ? `${config.collection}Error` : 'error',
+    callStateKey: collection ? `${collection}CallState` : 'callState',
+    loadingKey: collection ? `${collection}Loading` : 'loading',
+    loadedKey: collection ? `${collection}Loaded` : 'loaded',
+    errorKey: collection ? `${collection}Error` : 'error',
   };
 }
 
 export function withCallState<Collection extends string>(config: {
-  collection: Collection;
+  collections: Collection[];
 }): SignalStoreFeature<
   EmptyFeatureResult,
   EmptyFeatureResult & {
@@ -58,16 +57,39 @@ export function withCallState(): SignalStoreFeature<
   }
 >;
 export function withCallState<Collection extends string>(config?: {
-  collection: Collection;
+  collections: Collection[];
 }): SignalStoreFeature {
-  const { callStateKey, errorKey, loadedKey, loadingKey } =
-    getCallStateKeys(config);
-
   return signalStoreFeature(
-    withState({ [callStateKey]: 'init' }),
+    withState(
+      config
+        ? config.collections.reduce(
+            (acc, cur) => ({
+              ...acc,
+              ...{ [cur ? `${cur}CallState` : 'callState']: 'init' },
+            }),
+            {}
+          )
+        : { callStateKey: 'init' }
+    ),
     withComputed((state: Record<string, Signal<unknown>>) => {
+      if (config) {
+        return config.collections.reduce<Record<string, Signal<unknown>>>((acc, cur: string) => {
+          const { callStateKey, errorKey, loadedKey, loadingKey } =
+            getCallStateKeys(cur);
+          const callState = state[callStateKey] as Signal<CallState>;
+          return {
+            ...acc,
+            [loadingKey]: computed(() => callState() === 'loading'),
+            [loadedKey]: computed(() => callState() === 'loaded'),
+            [errorKey]: computed(() => {
+              const v = callState();
+              return typeof v === 'object' ? v.error : null;
+            }),
+          };
+        }, {});
+      } 
+      const { callStateKey, errorKey, loadedKey, loadingKey } = getCallStateKeys();
       const callState = state[callStateKey] as Signal<CallState>;
-
       return {
         [loadingKey]: computed(() => callState() === 'loading'),
         [loadedKey]: computed(() => callState() === 'loaded'),