Merge pull request #7 from WarpBuilds/feat/builder-v2

Author: guptaankit015

.github/workflows/debug.yaml

@@ -12,10 +12,10 @@ env:
 jobs:
   test-multi-warp:
     name: Debug Run ${{ matrix.run_number }}
-    runs-on: warpdev-ubuntu-latest-x64-2x
+    runs-on: warpdev-ubuntu-latest-x64-8x
     strategy:
       matrix:
-        run_number: [1]
+        run_number: [1,2,3]
       fail-fast: false
     steps:
       - name: Checkout repository
@@ -30,7 +30,6 @@ jobs:
           push: false
           # platforms: linux/amd64,linux/arm64
           platforms: linux/amd64
-          profile-name: test-dev,dev-eph
-          timeout: 40000
+          profile-name: test-dev
 
 

__tests__/Dockerfile

@@ -4,4 +4,9 @@ RUN apt-get update && apt-get install -y \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
+
+RUN echo "Sleeping..." && \
+    sleep 60 && \
+    echo "Done sleeping!"   
+
 CMD ["python3", "--version"]

dist/index.js

@@ -16,6 +16,7 @@ interface BuilderConfig {
 
 interface BuilderInstance {
   id: string;
+  request_id: string;
   arch: string;
   metadata: {
     host: string;
@@ -485,13 +486,16 @@ export class WarpBuildRemoteBuilders {
   public async removeBuilderInstances(): Promise<void> {
     for (const builderInstance of this.builderInstances) {
       try {
-        const removeBuilderEndpoint = `${this.apiDomain}/api/v1/builders/${builderInstance.id}/teardown`;
+        core.info(`Removing builder instance ${builderInstance.id} request ${builderInstance.request_id}`);
+        const removeBuilderEndpoint = `${this.apiDomain}/api/v1/builder-session-requests/complete`;
         const authHeader = this.isWarpBuildRunner ? `Bearer ${process.env.WARPBUILD_RUNNER_VERIFICATION_TOKEN}` : `Bearer ${this.apiKey}`;
 
         const response = await fetch(removeBuilderEndpoint, {
-          method: 'DELETE',
-          headers: {Authorization: authHeader}
-        });
+          method: 'POST',
+          headers: {Authorization: authHeader},
+          body: JSON.stringify({request_id: builderInstance.request_id})
+        },
+      );
 
         if (!response.ok) {
           const errorData = await response.json().catch(() => ({message: 'Unknown error'}));
@@ -500,7 +504,7 @@ export class WarpBuildRemoteBuilders {
 
         core.info(`Builder instance ${builderInstance.id} removed successfully`);
       } catch (error) {
-        core.warning(`Failed to remove builder instance: ${error.message}`);
+        core.warning(`Failed to remove builder request ${builderInstance.request_id}: ${error.message}`);
       }
     }
   }