fix(signals): patch state methods do not correctly resolve generic co…

…llection names withEntities

Ensures type-safety of named entity collections with patchState helpers

fixes ngrx#4638

modules/signals/entities/spec/types/models.types.spec.ts

@@ -0,0 +1,36 @@
+import { expecter } from 'ts-snippet';
+import { compilerOptions } from './helpers';
+
+describe('models', () => {
+  const expectSnippet = expecter(
+    (code) => `
+        import { patchState, signalStore, type, withMethods } from '@ngrx/signals';
+        import { addEntity, entityConfig, withEntities } from '@ngrx/signals/entities';
+        type User = { id: number; name: string };
+
+        ${code}
+      `,
+    compilerOptions()
+  );
+
+  it('succeeds when entity collection name is passed as the type parameter', () => {
+    const snippet = `
+    const storeFactory = <Collection extends string>(
+      collection: Collection
+    ) => {
+      const Store = signalStore(
+        withEntities({ entity: type<User>(), collection: collection }),
+        withMethods((store) => ({
+          addUsers(user: User): void {
+            patchState(store, addEntity(user, { collection: collection }));
+          },
+        }))
+      );
+
+      return new Store();
+    };
+`;
+
+    expectSnippet(snippet).toSucceed();
+  });
+});

modules/signals/entities/src/models.ts

@@ -1,4 +1,5 @@
 import { Signal } from '@angular/core';
+import { Prettify } from '@ngrx/signals';
 
 export type EntityId = string | number;
 
@@ -9,9 +10,13 @@ export type EntityState<Entity> = {
   ids: EntityId[];
 };
 
-export type NamedEntityState<Entity, Collection extends string> = {
-  [K in keyof EntityState<Entity> as `${Collection}${Capitalize<K>}`]: EntityState<Entity>[K];
-};
+export type NamedEntityState<Entity, Collection extends string> = Prettify<
+  {
+    [K in `${Collection}EntityMap`]: EntityMap<Entity>;
+  } & {
+    [K in `${Collection}Ids`]: EntityId[];
+  }
+>;
 
 export type EntityProps<Entity> = {
   entities: Signal<Entity[]>;