add suspended process error

Author: nicktrn

packages/cli-v3/src/executions/taskRunProcess.ts

@@ -29,6 +29,7 @@ import {
   internalErrorFromUnexpectedExit,
   GracefulExitTimeoutError,
   UnexpectedExitError,
+  SuspendedProcessError,
 } from "@trigger.dev/core/v3/errors";
 
 export type OnWaitForDurationMessage = InferSocketMessageSchema<
@@ -73,6 +74,7 @@ export class TaskRunProcess {
   private _gracefulExitTimeoutElapsed: boolean = false;
   private _isBeingKilled: boolean = false;
   private _isBeingCancelled: boolean = false;
+  private _isBeingSuspended: boolean = false;
   private _stderr: Array<string> = [];
 
   public onTaskRunHeartbeat: Evt<string> = new Evt();
@@ -347,7 +349,11 @@ export class TaskRunProcess {
           // Order matters, this has to be before the graceful exit timeout
           rejecter(new GracefulExitTimeoutError());
         } else if (this._isBeingKilled) {
-          rejecter(new CleanupProcessError());
+          if (this._isBeingSuspended) {
+            rejecter(new SuspendedProcessError());
+          } else {
+            rejecter(new CleanupProcessError());
+          }
         } else {
           rejecter(
             new UnexpectedExitError(
@@ -428,6 +434,11 @@ export class TaskRunProcess {
     }
   }
 
+  async suspend() {
+    this._isBeingSuspended = true;
+    await this.kill("SIGKILL");
+  }
+
   forceExit() {
     try {
       this._isBeingKilled = true;

packages/core/src/v3/errors.ts

@@ -497,6 +497,14 @@ export class CleanupProcessError extends Error {
   }
 }
 
+export class SuspendedProcessError extends Error {
+  constructor() {
+    super("Suspended");
+
+    this.name = "SuspendedProcessError";
+  }
+}
+
 export class CancelledProcessError extends Error {
   constructor() {
     super("Cancelled");