feat(slab): using slab new routes to handle async tasks

This ensures to wait for individual asynchronous tasks since there
are GitHub-related operations and backend providers ones.

Author: soonum

CHANGELOG.md

@@ -2,10 +2,10 @@
 
 ## [1.4.2](https://github.com/zama-ai/slab-github-runner/compare/v1.4.1...v1.4.2) (2025-11-28)
 
-
 ### Bug Fixes
 
-* **gh:** print out any error on the runner list query ([38175e5](https://github.com/zama-ai/slab-github-runner/commit/38175e59c2c30e5cd4c7fe81d16ecfad751d0db5))
+- **gh:** print out any error on the runner list query
+  ([38175e5](https://github.com/zama-ai/slab-github-runner/commit/38175e59c2c30e5cd4c7fe81d16ecfad751d0db5))
 
 ## [1.4.1](https://github.com/zama-ai/slab-github-runner/compare/v1.4.0...v1.4.1) (2024-12-20)
 

dist/index.js

@@ -7,7 +7,6 @@ function setOutput(label) {
   core.setOutput('label', label)
 }
 
-// This variable should only be defined for cleanup purpose.
 let runnerName
 
 async function cleanup() {
@@ -30,28 +29,31 @@ async function start() {
   for (let i = 1; i <= 3; i++) {
     try {
       startInstanceResponse = await slab.startInstanceRequest()
-      runnerName = startInstanceResponse.runner_name
       break
     } catch (error) {
       core.info('Retrying request now...')
     }
 
     if (i === 3) {
       core.setFailed(
-        `${provider} instance start request has failed after 3 attempts`
+        `${provider} instance start request has failed after 3 attempts (reason: configuration fetching has failed)`
       )
     }
   }
 
-  setOutput(startInstanceResponse.runner_name)
+  try {
+    const waitGithubResponse = await slab.waitForGithub(
+      startInstanceResponse.task_id,
+      'configuration_fetching'
+    )
 
-  core.info(
-    `${provider} instance details: ${JSON.stringify(
-      startInstanceResponse.details
-    )}`
-  )
+    runnerName = waitGithubResponse.configuration_fetching.runner_name
+    setOutput(runnerName)
+
+    core.info(
+      `${provider} instance details: ${waitGithubResponse.configuration_fetching.details}`
+    )
 
-  try {
     const waitInstanceResponse = await slab.waitForInstance(
       startInstanceResponse.task_id,
       'start'
@@ -60,10 +62,10 @@ async function start() {
     const instanceId = waitInstanceResponse.start.instance_id
     core.info(`${provider} instance started with ID: ${instanceId}`)
 
-    await waitForRunnerRegistered(startInstanceResponse.runner_name)
+    await waitForRunnerRegistered(runnerName)
   } catch (error) {
     core.info(`Clean up after error, stop ${provider} instance`)
-    await slab.stopInstanceRequest(startInstanceResponse.runner_name)
+    await slab.stopInstanceRequest(runnerName)
     core.setFailed(`${provider} instance start has failed`)
   }
 }
@@ -84,6 +86,22 @@ async function stop() {
     }
   }
 
+  try {
+    const waitGithubResponse = await slab.waitForGithub(
+      stopInstanceResponse.task_id,
+      'runner_unregister'
+    )
+    const taskStatus = waitGithubResponse.runner_unregister.status.toLowerCase()
+    if (taskStatus === 'done') {
+      core.info(
+        `Runner ${config.input.label} unregistered from GitHub successfully`
+      )
+    }
+  } catch (error) {
+    // Unregistration failure is not critical, so we just log it and continue.
+    core.warning('An error occurred while unregistering runner, check job logs')
+  }
+
   await slab.waitForInstance(stopInstanceResponse.task_id, 'stop')
 
   core.info('Instance successfully stopped')