1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
const std = @import("std");
const Allocator = std.mem.Allocator;
const ArrayList = std.ArrayList;
const print = std.debug.print;
const types = @import("./lib/types.zig");
const CommandStatus = types.CommandStatus;
const OutputCapture = types.OutputCapture;
const InputSource = types.InputSource;
pub const External = struct {
program: []const u8,
args: ArrayList([]const u8),
pub fn eval(external: External, allocator: Allocator, output_capture: ?*OutputCapture, input_source: ?*InputSource) !CommandStatus {
// Try to execute external command
var child_args = ArrayList([]const u8).init(allocator);
defer child_args.deinit();
try child_args.append(external.program);
for (external.args.items) |arg| {
try child_args.append(arg);
}
var child = std.process.Child.init(child_args.items, allocator);
// Set up pipes for capturing output
child.stdin_behavior = if (input_source != null) .Pipe else .Inherit;
child.stdout_behavior = if (output_capture != null) .Pipe else .Inherit;
child.stderr_behavior = if (output_capture != null) .Pipe else .Inherit;
const spawn_result = child.spawn();
if (spawn_result) |_| {
// Spawn succeeded, continue with execution
} else |err| switch (err) {
error.FileNotFound => {
const error_msg = try std.fmt.allocPrint(allocator, "'{s}' is not recognized as an internal or external command,\noperable program or batch file.\n", .{external.program});
defer allocator.free(error_msg);
if (output_capture) |capture| {
try capture.write(error_msg);
} else {
print("{s}", .{error_msg});
}
return CommandStatus{ .Code = 1 };
},
error.AccessDenied => {
const error_msg = "Access is denied.\n";
if (output_capture) |capture| {
try capture.write(error_msg);
} else {
print("{s}", .{error_msg});
}
return CommandStatus{ .Code = 1 };
},
else => {
const error_msg = try std.fmt.allocPrint(allocator, "Cannot execute '{s}': {}\n", .{ external.program, err });
defer allocator.free(error_msg);
if (output_capture) |capture| {
try capture.write(error_msg);
} else {
print("{s}", .{error_msg});
}
return CommandStatus{ .Code = 1 };
},
}
// Handle input redirection
if (input_source) |source| {
if (child.stdin) |stdin| {
const writer = stdin.writer();
// Reset source position for reading
var temp_source = source.*;
temp_source.position = 0;
while (try temp_source.readLine(allocator)) |line| {
defer allocator.free(line);
try writer.print("{s}\n", .{line});
}
child.stdin.?.close();
child.stdin = null;
}
}
// Handle output capture
if (output_capture) |capture| {
// Read stdout
if (child.stdout) |stdout| {
var buffer: [4096]u8 = undefined;
while (true) {
const bytes_read = stdout.read(&buffer) catch break;
if (bytes_read == 0) break;
try capture.write(buffer[0..bytes_read]);
}
}
// Read stderr
if (child.stderr) |stderr| {
var buffer: [4096]u8 = undefined;
while (true) {
const bytes_read = stderr.read(&buffer) catch break;
if (bytes_read == 0) break;
try capture.write(buffer[0..bytes_read]);
}
}
}
// Wait for process to complete
const term = child.wait() catch |err| {
const error_msg = switch (err) {
error.FileNotFound => try std.fmt.allocPrint(allocator, "'{s}' is not recognized as an internal or external command,\noperable program or batch file.\n", .{external.program}),
else => try std.fmt.allocPrint(allocator, "Error waiting for command: {}\n", .{err}),
};
defer allocator.free(error_msg);
if (output_capture) |capture| {
try capture.write(error_msg);
} else {
print("{s}", .{error_msg});
}
return CommandStatus{ .Code = 1 };
};
// Return exit code
switch (term) {
.Exited => |code| return CommandStatus{ .Code = @intCast(code) },
.Signal => |_| return CommandStatus{ .Code = 1 },
.Stopped => |_| return CommandStatus{ .Code = 1 },
.Unknown => |_| return CommandStatus{ .Code = 1 },
}
}
};
|
