WIP: try to handle cancellation

zama-ai · Dec 4, 2024 · cf52c42 · cf52c42
1 parent 8896570
commit cf52c42
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/src/index.js b/src/index.js
@@ -2,11 +2,22 @@ const slab = require('./slab')
 const config = require('./config')
 const core = require('@actions/core')
 const { waitForRunnerRegistered } = require('./gh')
+const utils = require('./utils')
 
 function setOutput(label) {
   core.setOutput('label', label)
 }
 
+process.on('SIGINT', async function () {
+  core.info('CANCEL caught !!!')
+  core.info('[DEBUG] About to wait 2 seconds async in cleanup')
+  await utils.sleep(2) // DEBUG
+  core.info('[DEBUG] Sleep finished')
+  process.exit()
+})
+
+// TODO Write a cleanup function
+
 async function start() {
   const provider = config.input.backend
 
@@ -33,6 +44,9 @@ async function start() {
     )}`
   )
 
+  core.info('[DEBUG] About to wait 30 seconds for cancelation') // DEBUG
+  await utils.sleep(30) // DEBUG
+
   try {
     const wait_instance_response = await slab.waitForInstance(
       start_instance_response.task_id,