feat: make setKeyInfoForResource public (#9352)

* fix: make setKeyInfoForResource public

* fix export

Author: runspired

packages/store/src/-private.ts

@@ -8,13 +8,7 @@ export { recordIdentifierFor } from './-private/caches/instance-cache';
 
 export { CacheHandler, type LifetimesService } from './-private/cache-handler';
 
-export {
-  setIdentifierGenerationMethod,
-  setIdentifierUpdateMethod,
-  setIdentifierForgetMethod,
-  setIdentifierResetMethod,
-  isStableIdentifier,
-} from './-private/caches/identifier-cache';
+export { isStableIdentifier } from './-private/caches/identifier-cache';
 
 export { default as constructResource } from './-private/utils/construct-resource';
 

packages/store/src/-private/caches/identifier-cache.ts

@@ -25,6 +25,8 @@ import type { ExistingResourceObject, ResourceIdentifierObject } from '@warp-dri
 import type {
   ForgetMethod,
   GenerationMethod,
+  KeyInfo,
+  KeyInfoMethod,
   ResetMethod,
   ResourceData,
   UpdateMethod,
@@ -76,19 +78,14 @@ type TypeMap = { [key: string]: KeyOptions };
 // type IdentifierTypeLookup = { all: Set<StableRecordIdentifier>; id: Map<string, StableRecordIdentifier> };
 // type IdentifiersByType = Map<string, IdentifierTypeLookup>;
 type IdentifierMap = Map<string, StableRecordIdentifier>;
-type KeyInfo = {
-  id: string | null;
-  type: string;
-};
+
 type StableCache = {
   resources: IdentifierMap;
   documents: Map<string, StableDocumentIdentifier>;
   resourcesByType: TypeMap;
   polymorphicLidBackMap: Map<string, string[]>;
 };
 
-export type KeyInfoMethod = (resource: unknown, known: StableRecordIdentifier | null) => KeyInfo;
-
 export type MergeMethod = (
   targetIdentifier: StableRecordIdentifier,
   matchedIdentifier: StableRecordIdentifier,

packages/store/src/-types/q/identifier.ts

@@ -171,3 +171,28 @@ export type ForgetMethod = (identifier: StableIdentifier | StableRecordIdentifie
   @static
 */
 export type ResetMethod = () => void;
+
+/**
+ Configure a callback for when the identifier cache is generating a new
+ StableRecordIdentifier for a resource.
+
+ This method controls the `type` and `id` that will be assigned to the
+ `StableRecordIdentifier` that is created.
+
+ This configuration MUST occur prior to the store instance being created.
+
+ ```js
+ import { setKeyInfoForResource } from '@ember-data/store';
+ ```
+
+  @method setKeyInfoForResource
+  @for @ember-data/store
+  @param method
+  @public
+  @static
+ */
+export type KeyInfo = {
+  id: string | null;
+  type: string;
+};
+export type KeyInfoMethod = (resource: unknown, known: StableRecordIdentifier | null) => KeyInfo;

packages/store/src/index.ts

@@ -187,10 +187,14 @@ export {
   CacheHandler,
   type LifetimesService,
   type StoreRequestInput,
+  recordIdentifierFor,
+  storeFor,
+} from './-private';
+
+export {
   setIdentifierGenerationMethod,
   setIdentifierUpdateMethod,
   setIdentifierForgetMethod,
   setIdentifierResetMethod,
-  recordIdentifierFor,
-  storeFor,
-} from './-private';
+  setKeyInfoForResource,
+} from './-private/caches/identifier-cache';

tests/docs/fixtures/expected.js

@@ -467,6 +467,7 @@ module.exports = {
     '(public) @ember-data/store @ember-data/store#setIdentifierGenerationMethod',
     '(public) @ember-data/store @ember-data/store#setIdentifierResetMethod',
     '(public) @ember-data/store @ember-data/store#setIdentifierUpdateMethod',
+    '(public) @ember-data/store @ember-data/store#setKeyInfoForResource',
     '(public) @ember-data/store CacheManager#changedAttrs',
     '(public) @ember-data/store CacheManager#clientDidCreate',
     '(public) @ember-data/store CacheManager#commitWasRejected',