From 328beaba3549e4aa7c87b3d1a00fe014c56ba401 Mon Sep 17 00:00:00 2001
From: J H <dragondef@gmail.com>
Date: Thu, 7 Mar 2024 13:36:38 -0700
Subject: [PATCH] chore: Add in the possibility to get the status code from the
 executed health check

---
 .../DockerProcedureContainer.ts               |  9 ++++
 .../Systems/SystemForEmbassy/MainLoop.ts      | 52 ++++++++++++++++---
 2 files changed, 54 insertions(+), 7 deletions(-)

diff --git a/container-runtime/src/Adapters/Systems/SystemForEmbassy/DockerProcedureContainer.ts b/container-runtime/src/Adapters/Systems/SystemForEmbassy/DockerProcedureContainer.ts
index 3129ce45d..3506575bc 100644
--- a/container-runtime/src/Adapters/Systems/SystemForEmbassy/DockerProcedureContainer.ts
+++ b/container-runtime/src/Adapters/Systems/SystemForEmbassy/DockerProcedureContainer.ts
@@ -73,6 +73,15 @@ export class DockerProcedureContainer {
     }
   }
 
+  async execSpawn(commands: string[]) {
+    try {
+      const spawned = await this.overlay.spawn(commands)
+      return spawned
+    } finally {
+      await this.overlay.destroy()
+    }
+  }
+
   async spawn(commands: string[]): Promise<cp.ChildProcessWithoutNullStreams> {
     return await this.overlay.spawn(commands)
   }
diff --git a/container-runtime/src/Adapters/Systems/SystemForEmbassy/MainLoop.ts b/container-runtime/src/Adapters/Systems/SystemForEmbassy/MainLoop.ts
index 8142a17df..c5e6644af 100644
--- a/container-runtime/src/Adapters/Systems/SystemForEmbassy/MainLoop.ts
+++ b/container-runtime/src/Adapters/Systems/SystemForEmbassy/MainLoop.ts
@@ -3,6 +3,7 @@ import { DockerProcedureContainer } from "./DockerProcedureContainer"
 import { SystemForEmbassy } from "."
 import { HostSystemStartOs } from "../../HostSystemStartOs"
 import { util, Daemons, types as T } from "@start9labs/start-sdk"
+import { exec } from "child_process"
 
 const EMBASSY_HEALTH_INTERVAL = 15 * 1000
 const EMBASSY_PROPERTIES_LOOP = 30 * 1000
@@ -117,17 +118,54 @@ export class MainLoop {
               actionProcedure,
               manifest.volumes,
             )
-            const executed = await container.exec([
+            const executed = await container.execSpawn([
               actionProcedure.entrypoint,
               ...actionProcedure.args,
               JSON.stringify(timeChanged),
             ])
-            const stderr = executed.stderr.toString()
-            if (stderr)
-              console.error(
-                `Error running health check ${value.name}: ${stderr}`,
-              )
-            return executed.stdout.toString()
+            if (executed.exitCode === 59) {
+              await effects.setHealth({
+                name: healthId,
+                status: "disabled",
+                message:
+                  executed.stderr.toString() || executed.stdout.toString(),
+              })
+              return
+            }
+            if (executed.exitCode === 60) {
+              await effects.setHealth({
+                name: healthId,
+                status: "starting",
+                message:
+                  executed.stderr.toString() || executed.stdout.toString(),
+              })
+              return
+            }
+            if (executed.exitCode === 61) {
+              await effects.setHealth({
+                name: healthId,
+                status: "warning",
+                message:
+                  executed.stderr.toString() || executed.stdout.toString(),
+              })
+              return
+            }
+            const errorMessage = executed.stderr.toString()
+            const message = executed.stdout.toString()
+            if (!!errorMessage) {
+              await effects.setHealth({
+                name: healthId,
+                status: "failure",
+                message: errorMessage,
+              })
+              return
+            }
+            await effects.setHealth({
+              name: healthId,
+              status: "passing",
+              message,
+            })
+            return
           } else {
             actionProcedure
             const moduleCode = await this.system.moduleCode