Bun.$ should keep the Bun process alive while a command is running

oven-sh · Jun 18, 2024 · 0765e76 · 0765e76
1 parent 3a17a2c
commit 0765e76
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 1 deletion.
diff --git a/src/shell/interpreter.zig b/src/shell/interpreter.zig
@@ -843,6 +843,7 @@ pub const Interpreter = struct {
     async_commands_executing: u32 = 0,
 
     globalThis: *JSC.JSGlobalObject,
+    ref: bun.Async.KeepAlive = .{},
 
     flags: packed struct(u8) {
         done: bool = false,
@@ -1235,7 +1236,7 @@ pub const Interpreter = struct {
         interpreter.this_jsvalue = JSC.Codegen.JSShellInterpreter.toJS(interpreter, globalThis);
         JSC.Codegen.JSShellInterpreter.resolveSetCached(interpreter.this_jsvalue, globalThis, resolve);
         JSC.Codegen.JSShellInterpreter.rejectSetCached(interpreter.this_jsvalue, globalThis, reject);
-
+        interpreter.ref.activate(JSC.VirtualMachine.get().uwsLoop());
         bun.Analytics.Features.shell += 1;
         return interpreter.this_jsvalue;
     }
@@ -1656,6 +1657,8 @@ pub const Interpreter = struct {
         defer decrPendingActivityFlag(&this.has_pending_activity);
 
         if (this.event_loop == .js) {
+            this.ref.disable();
+
             defer this.deinitAfterJSRun();
             this.exit_code = exit_code;
             if (this.this_jsvalue != .zero) {
@@ -1676,6 +1679,8 @@ pub const Interpreter = struct {
         defer decrPendingActivityFlag(&this.has_pending_activity);
 
         if (this.event_loop == .js) {
+            this.ref.disable();
+
             if (this.this_jsvalue != .zero) {
                 if (JSC.Codegen.JSShellInterpreter.rejectGetCached(this.this_jsvalue)) |reject| {
                     reject.call(this.globalThis, &[_]JSValue{ JSValue.jsNumberFromChar(1), this.getBufferedStdout(), this.getBufferedStderr() });
@@ -1692,6 +1697,7 @@ pub const Interpreter = struct {
         for (this.jsobjs) |jsobj| {
             jsobj.unprotect();
         }
+        this.ref.disable();
         this.root_io.deref();
         this.root_shell.deinitImpl(false, false);
         this.this_jsvalue = .zero;
@@ -1705,6 +1711,7 @@ pub const Interpreter = struct {
             this.root_shell._buffered_stdout.owned.deinitWithAllocator(bun.default_allocator);
         }
         this.this_jsvalue = .zero;
+        this.ref.disable();
         this.allocator.destroy(this);
     }
 

diff --git a/test/js/bun/shell/shell-keepalive-fixture-1.js b/test/js/bun/shell/shell-keepalive-fixture-1.js
diff --git a/test/js/bun/shell/shell-keepalive-fixture-2.js b/test/js/bun/shell/shell-keepalive-fixture-2.js
diff --git a/test/js/bun/shell/shell-keepalive.test.ts b/test/js/bun/shell/shell-keepalive.test.ts
@@ -0,0 +1,11 @@
+import { expect, test } from "bun:test";
+import { join } from "path";
+import "harness";
+
+test("shell should stay alive while a builtin command is in progress", async () => {
+  expect([join(import.meta.dir, "shell-keepalive-fixture-1.js")]).toRun();
+});
+
+test("shell should stay alive while a non-builtin command is in progress", async () => {
+  expect([join(import.meta.dir, "shell-keepalive-fixture-2.js")]).toRun();
+});