1 files changed, 9 insertions, 10 deletions

diff --git a/src/cmd/sort.zig b/src/cmd/sort.zig
index 9c04c1e..8b797d1 100644
--- a/src/cmd/sort.zig
+++ b/src/cmd/sort.zig
@@ -5,31 +5,30 @@ const print = std.debug.print;
 
 const types = @import("./lib/types.zig");
 const CommandStatus = types.CommandStatus;
-const OutputCapture = types.OutputCapture;
-const InputSource = types.InputSource;
+const CommandContext = types.CommandContext;
 
 pub const Sort = struct {
-    pub fn eval(sort: Sort, allocator: Allocator, output_capture: ?*OutputCapture, input_source: ?*InputSource) !CommandStatus {
+    pub fn eval(sort: Sort, ctx: CommandContext) !CommandStatus {
         _ = sort;
 
-        var lines = ArrayList([]const u8).init(allocator);
+        var lines = ArrayList([]const u8).init(ctx.allocator);
         defer {
             for (lines.items) |line| {
-                allocator.free(line);
+                ctx.allocator.free(line);
             }
             lines.deinit();
         }
 
         // Read input lines
-        if (input_source) |source| {
+        if (ctx.input_source) |source| {
             // Read from input redirection
-            while (try source.readLine(allocator)) |line| {
+            while (try source.readLine(ctx.allocator)) |line| {
                 try lines.append(line);
             }
         } else {
             // Read from stdin (simplified - just show message)
             const msg = "SORT: Use input redirection (< file.txt) to sort file contents\n";
-            if (output_capture) |capture| {
+            if (ctx.output_capture) |capture| {
                 try capture.write(msg);
             } else {
                 print("{s}", .{msg});
@@ -45,14 +44,14 @@ pub const Sort = struct {
         }.lessThan);
 
         // Output sorted lines
-        var output_buffer = ArrayList(u8).init(allocator);
+        var output_buffer = ArrayList(u8).init(ctx.allocator);
         defer output_buffer.deinit();
 
         for (lines.items) |line| {
             try output_buffer.writer().print("{s}\n", .{line});
         }
 
-        if (output_capture) |capture| {
+        if (ctx.output_capture) |capture| {
             try capture.write(output_buffer.items);
         } else {
             print("{s}", .{output_buffer.items});