feat: Add custom request header zone

Author: LZS911

webapp/packages/core-sdk/src/GraphQLService.ts

@@ -14,6 +14,7 @@ import { uploadDriverLibraryExtension } from './Extensions/uploadDriverLibraryEx
 import { uploadResultDataExtension } from './Extensions/uploadResultDataExtension.js';
 import type { IResponseInterceptor } from './IResponseInterceptor.js';
 import { getSdk, type Sdk } from './sdk.js';
+import { getRecentlySelectedZone } from './LocalStorage.js';
 
 function extendedSDK(
   client: CustomGraphQLClient,
@@ -39,7 +40,11 @@ export class GraphQLService {
 
   constructor(private readonly environmentService: EnvironmentService) {
     const gqlEndpoint = this.environmentService.gqlEndpoint;
-    this.client = new CustomGraphQLClient(gqlEndpoint);
+    this.client = new CustomGraphQLClient(gqlEndpoint, {
+      headers: {
+        zone: getRecentlySelectedZone(),
+      },
+    });
     this.sdk = extendedSDK(this.client);
   }
 

webapp/packages/core-sdk/src/LocalStorage.ts

@@ -0,0 +1,33 @@
+export enum StorageKey {
+  DMS_AVAILABILITY_ZONE = 'DMS_AVAILABILITY_ZONE',
+}
+
+class LocalStorageWrapper {
+  public set<T extends string = string>(key: string, value: T) {
+    return localStorage.setItem(key, value);
+  }
+
+  public get(key: string) {
+    return localStorage.getItem(key);
+  }
+
+  public getOrDefault(key: string, defaultValue: string): string {
+    if (localStorage.getItem(key) === null) {
+      return defaultValue;
+    }
+    return localStorage.getItem(key) as string;
+  }
+}
+
+export const localStorageWrapper = new LocalStorageWrapper();
+
+export const getRecentlySelectedZone = (): string => {
+  const data = localStorageWrapper.get(StorageKey.DMS_AVAILABILITY_ZONE);
+  try {
+    const parsedData = JSON.parse(data || '[]');
+    return parsedData?.[0]?.uid ?? '';
+  } catch (error) {
+    console.error(error);
+    return '';
+  }
+};

webapp/packages/core-sdk/src/index.ts

@@ -26,3 +26,4 @@ export * from './ServiceError.js';
 export * from './SessionError.js';
 export * from './manifest.js';
 export * from './getObjectPropertyValue.js';
+export * from './LocalStorage.js';