@contentstack/cli-variants 0.0.1-alpha → 1.1.0

package/src/utils/variant-api-adapter.ts

@@ -7,6 +7,7 @@ import {
   ContentstackClient,
   ContentstackConfig,
   managementSDKClient,
+  authenticationHandler,
 } from '@contentstack/cli-utilities';
 
 import {
@@ -37,6 +38,16 @@ export class VariantHttpClient<C> extends AdapterHelper<C, HttpClient> implement
     this.apiClient.baseUrl(this.baseURL);
   }
 
+  async init(): Promise<void> {
+    await authenticationHandler.getAuthDetails();
+    const token = authenticationHandler.accessToken;
+    if (authenticationHandler.isOauthEnabled) {
+      this.apiClient.headers({ authorization: token });
+    } else {
+      this.apiClient.headers({ authtoken: token });
+    }
+  }
+
   async variantEntry(options: VariantOptions) {
     // TODO single entry variant
     return { entry: {} };
@@ -124,7 +135,7 @@ export class VariantHttpClient<C> extends AdapterHelper<C, HttpClient> implement
     }
 
     const data = await this.apiClient.get(endpoint);
-    const response = this.handleVariantAPIRes(data) as { entries: VariantEntryStruct[]; count: number };
+    const response = (await this.handleVariantAPIRes(data)) as { entries: VariantEntryStruct[]; count: number };
 
     if (callback) {
       callback(response.entries);
@@ -184,6 +195,7 @@ export class VariantHttpClient<C> extends AdapterHelper<C, HttpClient> implement
       });
 
     try {
+      this.apiClient.headers({ api_version: undefined });
       const res = await this.apiClient.put<VariantEntryStruct>(endpoint, { entry: input });
       const data = this.handleVariantAPIRes(res);
 
@@ -214,8 +226,7 @@ export class VariantHttpClient<C> extends AdapterHelper<C, HttpClient> implement
     const { entry_uid, content_type_uid } = options;
     let endpoint = `content_types/${content_type_uid}/entries/${entry_uid}/publish`;
 
-    const onSuccess = (response: any) =>
-      resolve({ response, apiData: { entryUid: entry_uid, variantUid }, log });
+    const onSuccess = (response: any) => resolve({ response, apiData: { entryUid: entry_uid, variantUid }, log });
     const onReject = (error: any) =>
       reject({
         error,
@@ -244,15 +255,18 @@ export class VariantHttpClient<C> extends AdapterHelper<C, HttpClient> implement
    * @returns The variant API response data.
    * @throws If the API response status is not within the success range, an error message is thrown.
    */
-  handleVariantAPIRes(
+  async handleVariantAPIRes(
     res: APIResponse,
-  ): VariantEntryStruct | { entries: VariantEntryStruct[]; count: number } | string | any {
+  ): Promise<VariantEntryStruct | { entries: VariantEntryStruct[]; count: number } | string | any> {
     const { status, data } = res;
 
     if (status >= 200 && status < 300) {
       return data;
     }
 
+    // Refresh the access token if the response status is 401
+    await authenticationHandler.refreshAccessToken(res);
+
     const errorMsg = data?.errors
       ? formatErrors(data.errors)
       : data?.error_message || data?.message || 'Something went wrong while processing entry variant request!';
@@ -290,9 +304,9 @@ export class VariantManagementSDK<T>
     return Promise.resolve({} as VariantEntryStruct);
   }
 
-  handleVariantAPIRes(
+  async handleVariantAPIRes(
     res: APIResponse,
-  ): VariantEntryStruct | { entries: VariantEntryStruct[]; count: number } | string {
+  ): Promise<VariantEntryStruct | { entries: VariantEntryStruct[]; count: number } | string> {
     return res.data;
   }