Minor refactor

Author: rkodev

packages/abstractions/src/apiClientBuilder.ts

@@ -4,7 +4,6 @@
  * See License in the project root for license information.
  * -------------------------------------------------------------------------------------------
  */
-export * from "./apiClientBuilder";
 export * from "./apiClientProxifier";
 export * from "./apiError";
 export * from "./authentication";

packages/abstractions/src/index.ts

@@ -9,6 +9,7 @@ import type { SerializationWriterFactory } from "./serializationWriterFactory";
 import type { Parsable } from "./parsable";
 import type { ModelSerializerFunction } from "./serializationFunctionTypes";
 import { BackingStoreSerializationWriterProxyFactory } from "../store";
+import { ParseNodeFactoryRegistry } from "./parseNodeFactoryRegistry";
 
 /** This factory holds a list of all the registered factories for the various types of nodes. */
 export class SerializationWriterFactoryRegistry implements SerializationWriterFactory {
@@ -177,4 +178,23 @@ export class SerializationWriterFactoryRegistry implements SerializationWriterFa
 		const decoder = new TextDecoder();
 		return decoder.decode(buffer);
 	}
+
+	/**
+	 * Enables the backing store on default serialization writers and the given serialization writer.
+	 * @param parseNodeFactoryRegistry The parse node factory registry to enable the backing store on.
+	 * @param original The serialization writer to enable the backing store on.
+	 * @returns A new serialization writer with the backing store enabled.
+	 */
+	public enableBackingStoreForSerializationWriterFactory = (parseNodeFactoryRegistry: ParseNodeFactoryRegistry, original: SerializationWriterFactory): SerializationWriterFactory => {
+		if (!original) throw new Error("Original must be specified");
+		let result = original;
+		if (original instanceof SerializationWriterFactoryRegistry) {
+			original.enableBackingStoreForSerializationRegistry();
+		} else {
+			result = new BackingStoreSerializationWriterProxyFactory(original);
+		}
+		this.enableBackingStoreForSerializationRegistry();
+		parseNodeFactoryRegistry.enableBackingStoreForParseNodeRegistry();
+		return result;
+	};
 }

packages/abstractions/src/serialization/serializationWriterFactoryRegistry.ts

@@ -5,7 +5,7 @@
  * -------------------------------------------------------------------------------------------
  */
 
-import { type ApiError, type AuthenticationProvider, type BackingStoreFactory, BackingStoreFactorySingleton, type DateOnly, DefaultApiError, type Duration, enableBackingStoreForSerializationWriterFactory, type ErrorMappings, type Parsable, type ParsableFactory, type ParseNode, type ParseNodeFactory, ParseNodeFactoryRegistry, type PrimitiveTypesForDeserialization, type PrimitiveTypesForDeserializationType, type RequestAdapter, type RequestInformation, type ResponseHandler, type ResponseHandlerOption, ResponseHandlerOptionKey, type SerializationWriterFactory, SerializationWriterFactoryRegistry, type TimeOnly } from "@microsoft/kiota-abstractions";
+import { type ApiError, type AuthenticationProvider, type BackingStoreFactory, BackingStoreFactorySingleton, type DateOnly, DefaultApiError, type Duration, type ErrorMappings, type Parsable, type ParsableFactory, type ParseNode, type ParseNodeFactory, ParseNodeFactoryRegistry, type PrimitiveTypesForDeserialization, type PrimitiveTypesForDeserializationType, type RequestAdapter, type RequestInformation, type ResponseHandler, type ResponseHandlerOption, ResponseHandlerOptionKey, type SerializationWriterFactory, SerializationWriterFactoryRegistry, type TimeOnly } from "@microsoft/kiota-abstractions";
 import { type Span, SpanStatusCode, trace } from "@opentelemetry/api";
 
 import { HttpClient } from "./httpClient";
@@ -365,7 +365,7 @@ export class FetchRequestAdapter implements RequestAdapter {
 			throw new Error("parseNodeFactory is not a ParseNodeFactoryRegistry");
 		}
 		if (this.serializationWriterFactory instanceof SerializationWriterFactoryRegistry && this.parseNodeFactory instanceof ParseNodeFactoryRegistry) {
-			this.serializationWriterFactory = enableBackingStoreForSerializationWriterFactory(this.serializationWriterFactory, this.parseNodeFactory, this.serializationWriterFactory);
+			this.serializationWriterFactory = this.serializationWriterFactory.enableBackingStoreForSerializationWriterFactory(this.parseNodeFactory, this.serializationWriterFactory);
 		} else {
 			throw new Error("serializationWriterFactory is not a SerializationWriterFactoryRegistry or parseNodeFactory is not a ParseNodeFactoryRegistry");
 		}