Merge pull request #5 from WarpBuilds/feat/multi-profile-support

Author: guptaankit015

.github/workflows/debug.yaml

@@ -30,6 +30,7 @@ jobs:
           push: false
           # platforms: linux/amd64,linux/arm64
           platforms: linux/amd64
-          profile-name: dev-eph
+          profile-name: test-dev,dev-eph
+          timeout: 40000
 
 

action.yml

@@ -115,7 +115,7 @@ inputs:
     description: "The API key for the WarpBuild API. This is not required in case of using WarpBuild runners"
     required: false
   profile-name:
-    description: "The profile name to use for the WarpBuild Docker Builders"
+    description: "The profile names to use for the WarpBuild Docker Builders, comma separated list of profile names to be attempted in order"
     required: true
   timeout:
     description: "The timeout(in ms) to wait for the Docker Builders to be ready. By default, it is 10 minutes"

dist/index.js

@@ -42,7 +42,7 @@ export interface Inputs {
   ulimit: string[];
   'github-token': string;
   apiKey: string;
-  profileName: string;
+  profileNames: string;
   timeout: string;
 }
 
@@ -82,7 +82,7 @@ export async function getInputs(): Promise<Inputs> {
     ulimit: Util.getInputList('ulimit', {ignoreComma: true}),
     'github-token': core.getInput('github-token'),
     apiKey: core.getInput('api-key'),
-    profileName: core.getInput('profile-name'),
+    profileNames: core.getInput('profile-name'),
     timeout: core.getInput('timeout')
   };
 }

dist/index.js.map

@@ -36,7 +36,7 @@ actionsToolkit.run(
 
     remoteBuilders = new WarpBuildRemoteBuilders({
       apiKey: inputs.apiKey,
-      profileName: inputs.profileName,
+      profileNames: inputs.profileNames,
       timeout: parsedTimeout
     });
 
@@ -255,7 +255,7 @@ actionsToolkit.run(
 
     remoteBuilders = new WarpBuildRemoteBuilders({
       apiKey: inputs.apiKey,
-      profileName: inputs.profileName,
+      profileNames: inputs.profileNames,
       timeout: parsedTimeout
     });
 

dist/licenses.txt

@@ -9,7 +9,7 @@ import * as core from '@actions/core';
 const execAsync = promisify(exec);
 
 interface BuilderConfig {
-  profileName: string;
+  profileNames: string;
   apiKey?: string;
   timeout?: number;
 }
@@ -39,19 +39,19 @@ interface CleanupState {
 }
 
 export class WarpBuildRemoteBuilders {
-  private readonly profileName: string;
+  private readonly profileNames: string[];
   private readonly apiKey?: string;
   private readonly timeout: number;
   private readonly isWarpBuildRunner: boolean;
-  private readonly scriptStartTime: number;
   private readonly apiDomain: string;
+  private scriptStartTime: number;
   private builderName: string;
   private builderInstances: BuilderInstance[] = [];
   private certDirs: string[] = [];
   private assignmentPromise: Promise<void> | null = null;
 
   constructor(config: BuilderConfig) {
-    this.profileName = config.profileName;
+    this.profileNames = config.profileNames.split(',');
     this.apiKey = config.apiKey;
     this.timeout = config.timeout || 600000; // Default timeout 600 seconds
     this.scriptStartTime = Date.now();
@@ -146,64 +146,71 @@ export class WarpBuildRemoteBuilders {
     let retryCount = 0;
     const staticWait = 10000; // 10 seconds
 
-    await core.group(`Assigning WarpBuild builders for profile ${this.profileName}`, async () => {
-      while (this.checkGlobalTimeoutNotExceeded()) {
-        retryCount++;
-
-        try {
-          core.info(`Making API request to assign builder (attempt ${retryCount})...`);
-
-          const authHeader = this.isWarpBuildRunner ? `Bearer ${process.env.WARPBUILD_RUNNER_VERIFICATION_TOKEN}` : `Bearer ${this.apiKey}`;
-
-          const response = await fetch(assignBuilderEndpoint, {
-            method: 'POST',
-            headers: {
-              'Content-Type': 'application/json',
-              Authorization: authHeader
-            },
-            body: JSON.stringify({profile_name: this.profileName})
-          });
-
-          if (!response.ok) {
-            const statusCode = response.status;
-
-            const errorData = await response.json().catch(() => ({message: 'Unknown error'}));
-
-            // Determine if we should retry based on status code
-            if (statusCode >= 500 || statusCode === 409 || statusCode === 429) {
-              core.warning(`${errorData.description || errorData.message || 'Assign builder failed'}`);
-              core.info(`Waiting ${staticWait / 1000} seconds before next attempt...`);
-              await new Promise(resolve => setTimeout(resolve, staticWait));
-              continue;
+    await core.group(`Assigning WarpBuild builders for profile ${this.profileNames}`, async () => {
+      for (const profileName of this.profileNames) {
+        core.info(`Assigning builder for profile ${profileName}`);
+        while (this.checkGlobalTimeoutNotExceeded()) {
+          retryCount++;
+
+          try {
+            core.info(`Making API request to assign builder (attempt ${retryCount})...`);
+
+            const authHeader = this.isWarpBuildRunner ? `Bearer ${process.env.WARPBUILD_RUNNER_VERIFICATION_TOKEN}` : `Bearer ${this.apiKey}`;
+
+            const response = await fetch(assignBuilderEndpoint, {
+              method: 'POST',
+              headers: {
+                'Content-Type': 'application/json',
+                Authorization: authHeader
+              },
+              body: JSON.stringify({profile_name: profileName.trim()})
+            });
+
+            if (!response.ok) {
+              const statusCode = response.status;
+
+              const errorData = await response.json().catch(() => ({message: 'Unknown error'}));
+
+              // Determine if we should retry based on status code
+              if (statusCode >= 500 || statusCode === 409 || statusCode === 429) {
+                core.warning(`${errorData.description || errorData.message || 'Assign builder failed'}`);
+                core.info(`Waiting ${staticWait / 1000} seconds before next attempt...`);
+                await new Promise(resolve => setTimeout(resolve, staticWait));
+                continue;
+              }
+
+              // Not a retriable error
+              core.debug(`Error data: ${JSON.stringify(errorData)}`);
+              throw new Error(`HTTP ${statusCode}: ${errorData.description || errorData.message || 'Unknown error'}`);
             }
 
-            // Not a retriable error
-            core.debug(`Error data: ${JSON.stringify(errorData)}`);
-            throw new Error(`HTTP ${statusCode}: ${errorData.description || errorData.message || 'Unknown error'}`);
-          }
+            const data = (await response.json()) as AssignBuilderResponse;
 
-          const data = (await response.json()) as AssignBuilderResponse;
+            if (!data.builder_instances || data.builder_instances.length === 0) {
+              throw new Error('No builder instances assigned');
+            }
 
-          if (!data.builder_instances || data.builder_instances.length === 0) {
-            throw new Error('No builder instances assigned');
-          }
+            core.info(`✓ Successfully assigned ${data.builder_instances.length} builder(s) after ${retryCount} attempts`);
+            this.builderInstances = data.builder_instances;
+            this.saveState();
+            return;
+          } catch (error) {
+            if (error instanceof Error && error.message.startsWith('API Error:')) {
+              throw error; // Re-throw non-retriable errors
+            }
 
-          core.info(`✓ Successfully assigned ${data.builder_instances.length} builder(s) after ${retryCount} attempts`);
-          this.builderInstances = data.builder_instances;
-          this.saveState();
-          return;
-        } catch (error) {
-          if (error instanceof Error && error.message.startsWith('API Error:')) {
-            throw error; // Re-throw non-retriable errors
+            core.warning(`Request error: ${error}`);
+            core.info(`Waiting ${staticWait / 1000} seconds before next attempt...`);
+            await new Promise(resolve => setTimeout(resolve, staticWait));
           }
-
-          core.warning(`Request error: ${error}`);
-          core.info(`Waiting ${staticWait / 1000} seconds before next attempt...`);
-          await new Promise(resolve => setTimeout(resolve, staticWait));
         }
+
+        //Reset script start time after each profile
+        this.scriptStartTime = Date.now();
+        core.info(`Timeout reached for profile ${profileName}`);
       }
 
-      throw new Error('Exceeded global timeout waiting for builder assignment');
+      throw new Error('Exceeded global timeout waiting for builder assignment for all profiles');
     });
   }
 
@@ -267,7 +274,7 @@ export class WarpBuildRemoteBuilders {
    * Validate required inputs
    */
   private validateInputs(): void {
-    if (!this.profileName) {
+    if (!this.profileNames) {
       throw new Error('Profile name is required');
     }