add after_setup.host

kla · Nov 30, 2024 · 78d7543 · 78d7543
1 parent 3f49bb2
commit 78d7543
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/src/container.ts b/src/container.ts
@@ -194,13 +194,16 @@ export default class Container {
   }
 
   async runHook(type) {
-    let hook = this.labels[`stax.${type}`]
-    if (!hook) return
-
-    if (existsSync(hook))
-      run(`cat ${hook} | docker container exec --interactive ${this.containerName} /bin/sh`)
-    else
-      run(hook)
+    [ '', '.host' ].forEach((location) => {
+      let hook = this.labels[`stax.${type}${location}`]
+      if (!hook) return
+
+      if (existsSync(hook)) {
+        const command = location === '.host' ? hook : `cat ${hook} | docker container exec --interactive ${this.containerName} /bin/sh`
+        run(command)
+      } else
+        run(hook)
+    })
   }
 
   async copy(source: string, destination: string, options: { dontOverwrite?: boolean } = {}) {