bun run prettier

src/bun.js/event_loop/Task.zig

@@ -490,7 +490,11 @@ pub fn tickQueueWithCount(this: *EventLoop, virtual_machine: *VirtualMachine) u3
                 any.runFromJSThread();
             },
 
-            .@"shell.builtin.yes.YesTask", .@"bun.js.api.Timer.ImmediateObject", .@"bun.js.api.Timer.TimeoutObject" => {
+            .@"shell.builtin.yes.YesTask" => {
+                var yes_task: *YesTask = task.get(YesTask).?;
+                yes_task.runFromMainThread();
+            },
+            .@"bun.js.api.Timer.ImmediateObject", .@"bun.js.api.Timer.TimeoutObject" => {
                 bun.Output.panic("Unexpected tag: {s}", .{@tagName(task.tag())});
             },
             _ => {
@@ -582,6 +586,7 @@ const ShellMvCheckTargetTask = shell.Interpreter.Builtin.Mv.ShellMvCheckTargetTa
 const ShellMvBatchedTask = shell.Interpreter.Builtin.Mv.ShellMvBatchedTask;
 const ShellMkdirTask = shell.Interpreter.Builtin.Mkdir.ShellMkdirTask;
 const ShellTouchTask = shell.Interpreter.Builtin.Touch.ShellTouchTask;
+const YesTask = shell.Interpreter.Builtin.Yes.YesTask;
 const ShellCpTask = shell.Interpreter.Builtin.Cp.ShellCpTask;
 const ShellCondExprStatTask = shell.Interpreter.CondExpr.ShellCondExprStatTask;
 const ShellAsync = shell.Interpreter.Async;

src/shell/builtin/yes.zig

@@ -68,6 +68,10 @@ pub const YesTask = struct {
         const yes: *Yes = @fieldParentPtr("task", this);
 
         // Manually make safeguard since this task should not be created if output does not need IO
+        if (yes.state != .waiting_io) {
+            return;
+        }
+
         yes.bltn().stdout.enqueueFmt(yes, "{s}\n", .{yes.expletive}, .output_needs_io).run();
 
         this.enqueue();

test/js/bun/shell/commands/yes.test.ts

@@ -1,5 +1,7 @@
 import { $ } from "bun";
 import { describe, expect, test } from "bun:test";
+import { createTestBuilder } from "../test_builder";
+const TestBuilder = createTestBuilder(import.meta.path);
 
 $.throws(false);
 
@@ -22,3 +24,11 @@ describe("yes", async () => {
     expect(buffer.toString()).toEqual("ab\nab\nab\nab\nab\nab");
   });
 });
+
+describe("yes command", async () => {
+  TestBuilder.command`yes | head -n 5`.stdout("y\ny\ny\ny\ny\n").runAsTest("default output");
+
+  TestBuilder.command`yes xy | head -n 6`.stdout("xy\nxy\nxy\nxy\nxy\nxy\n").runAsTest("custom expletive");
+
+  TestBuilder.command`yes ab cd ef | head -n 6`.stdout("ab\nab\nab\nab\nab\nab\n").runAsTest("ignores extra args");
+});