From 9edc1338b92d1ead4c5f2fad8c0516037963f7b6 Mon Sep 17 00:00:00 2001 From: Matthias Andreas Benkard Date: Sat, 16 Aug 2025 12:50:23 +0200 Subject: Unify output handling. Now every command always uses an OutputWriter instead of conditionally writing directly to stdout using std.debug.print. --- src/cmd/lib/types.zig | 25 ++++++++++++++++++++++--- 1 file changed, 22 insertions(+), 3 deletions(-) (limited to 'src/cmd/lib') diff --git a/src/cmd/lib/types.zig b/src/cmd/lib/types.zig index c1ce3c7..f663a64 100644 --- a/src/cmd/lib/types.zig +++ b/src/cmd/lib/types.zig @@ -32,6 +32,13 @@ pub const OutputCapture = struct { } }; +pub const StdoutOutputCapture = struct { + pub fn write(self: *StdoutOutputCapture, data: []const u8) !void { + _ = self; + try std.io.getStdOut().writeAll(data); + } +}; + pub const InputSource = struct { data: []const u8, position: usize, @@ -65,18 +72,30 @@ pub const InputSource = struct { } }; +pub const OutputWriter = union(enum) { + capture: *OutputCapture, + stdout: *StdoutOutputCapture, + + pub fn write(self: *OutputWriter, data: []const u8) !void { + switch (self.*) { + .capture => |capture| try capture.write(data), + .stdout => |stdout| try stdout.write(data), + } + } +}; + pub const ExecuteCommandFn = *const fn (Command, Allocator, ?*OutputCapture, ?*InputSource) anyerror!CommandStatus; pub const CommandContext = struct { allocator: Allocator, - output_capture: ?*OutputCapture, + output_writer: OutputWriter, input_source: ?*InputSource, execute_command: ExecuteCommandFn, - pub fn init(allocator: Allocator, output_capture: ?*OutputCapture, input_source: ?*InputSource, execute_command: ExecuteCommandFn) CommandContext { + pub fn init(allocator: Allocator, output_writer: OutputWriter, input_source: ?*InputSource, execute_command: ExecuteCommandFn) CommandContext { return CommandContext{ .allocator = allocator, - .output_capture = output_capture, + .output_writer = output_writer, .input_source = input_source, .execute_command = execute_command, }; -- cgit v1.2.1