Move azure token logic into seperate file (#1606)

Part of OPS-2860

Author: bigfluffycookie

packages/blocks/azure/src/lib/actions/custom-azure-api-action.ts

@@ -1,31 +1,9 @@
 import { createCustomApiCallAction } from '@openops/blocks-common';
 import { Property } from '@openops/blocks-framework';
-import {
-  authenticateUserWithAzure,
-  azureAuth,
-  getUseHostSessionProperty,
-} from '@openops/common';
-import { runCommand } from '../azure-cli';
+import { azureAuth, getUseHostSessionProperty } from '@openops/common';
+import { getAzureAccessToken } from '../auth/get-azure-access-token';
 import { getSubscriptionsDropdownForHostSession } from '../common-properties';
 
-const getHostAccessToken = async (
-  auth: unknown,
-  subscription: string,
-): Promise<string> => {
-  const output = await runCommand(
-    'account get-access-token --resource https://management.azure.com --output json',
-    auth,
-    true,
-    subscription,
-  );
-  const parsed = JSON.parse(output ?? '{}');
-  const token = parsed?.accessToken;
-  if (!token) {
-    throw new Error('Failed to obtain Azure access token');
-  }
-  return token as string;
-};
-
 export const customAzureApiCallAction = createCustomApiCallAction({
   auth: azureAuth,
   name: 'custom_azure_api_call',
@@ -65,9 +43,11 @@ export const customAzureApiCallAction = createCustomApiCallAction({
     const selectedSubscription =
       context.propsValue?.subscriptions?.['subDropdown'];
 
-    const token = shouldUseHostCredentials
-      ? await getHostAccessToken(context.auth, selectedSubscription)
-      : (await authenticateUserWithAzure(context.auth)).access_token;
+    const token = await getAzureAccessToken(
+      context.auth,
+      !!shouldUseHostCredentials,
+      selectedSubscription,
+    );
 
     return {
       Authorization: `Bearer ${token}`,

packages/blocks/azure/src/lib/auth/get-azure-access-token.ts

@@ -0,0 +1,36 @@
+import { authenticateUserWithAzure } from '@openops/common';
+import { runCommand } from '../azure-cli';
+
+function parseAzAccessToken(raw?: string | null): string {
+  const parsed = JSON.parse(raw ?? '{}');
+  const token = parsed?.accessToken;
+  if (!token) {
+    throw new Error('Failed to obtain Azure access token');
+  }
+  return token;
+}
+
+export const getHostAccessToken = async (
+  auth: unknown,
+  subscription?: string,
+): Promise<string> => {
+  const output = await runCommand(
+    'account get-access-token --resource https://management.azure.com --output json',
+    auth,
+    true,
+    subscription,
+  );
+  return parseAzAccessToken(output);
+};
+
+export const getAzureAccessToken = async (
+  auth: unknown,
+  useHost: boolean,
+  subscription?: string,
+): Promise<string> => {
+  if (useHost) {
+    return getHostAccessToken(auth, subscription);
+  }
+  const { access_token } = await authenticateUserWithAzure(auth);
+  return access_token;
+};