Files
math_shell/src/main.zig
2025-03-25 21:49:32 -04:00

99 lines
3.2 KiB
Zig

const std = @import("std");
const Linenoise = @import("linenoise").Linenoise;
fn clear() !void {
const stdout = std.io.getStdOut().writer();
try stdout.print("\u{001b}[H\u{001b}[J", .{});
}
const Pipe = packed struct {
const fd_t = std.posix.fd_t;
read: fd_t,
write: fd_t,
};
fn pipe() !Pipe {
return @bitCast(try std.posix.pipe());
}
fn getPrompt() ![]const u8 {
const p = try pipe();
const close = std.posix.close;
const dup2 = std.posix.dup2;
const STDIN_FILENO = std.posix.STDIN_FILENO;
const STDOUT_FILENO = std.posix.STDOUT_FILENO;
const STDERR_FILENO = std.posix.STDERR_FILENO;
const script = @embedFile("./get_prompt.bash");
if (std.posix.fork()) |pid| {
if (pid == 0) {
close(p.read);
try dup2(p.write, STDOUT_FILENO);
close(STDIN_FILENO);
close(STDERR_FILENO); // Do not print errors to the screen
// Command-line arguments (must include program name as first argument)
// const args = [_:null]?[*:0]const u8{ "bash", "-i", "-c", script, null };
var args = [_:null]?[*:0]const u8{ "bash", "-i", "-c", script, null };
try std.posix.execvpeZ("bash", &args, std.c.environ);
// std.posix.execvpeZ("bash", @as([*:null]const ?[*:0]const u8, &args[0]), std.os.environ);
_ = std.posix.execvpeZ("bash", @as([*:null]const ?[*:0]const u8, &args[0]), std.os.environ);
// kill the child
std.process.exit(0);
} else {}
return "";
} else |err| {
return err;
}
}
pub fn main() !void {
try clear();
const allocator = std.heap.page_allocator;
var ln = Linenoise.init(allocator);
defer ln.deinit();
_ = try getPrompt();
while (try ln.linenoise("hello> ")) |input| {
defer allocator.free(input);
std.debug.print("input: {s}\n", .{input});
try ln.history.add(input);
}
// // Prints to stderr (it's a shortcut based on `std.io.getStdErr()`)
// std.debug.print("All your {s} are belong to us.\n", .{"codebase"});
// // stdout is for the actual output of your application, for example if you
// // are implementing gzip, then only the compressed bytes should be sent to
// // stdout, not any debugging messages.
// const stdout_file = std.io.getStdOut().writer();
// var bw = std.io.bufferedWriter(stdout_file);
// const stdout = bw.writer();
// try stdout.print("Run `zig build test` to run the tests.\n", .{});
// try bw.flush(); // Don't forget to flush!
}
test "simple test" {
var list = std.ArrayList(i32).init(std.testing.allocator);
defer list.deinit(); // Try commenting this out and see if zig detects the memory leak!
try list.append(42);
try std.testing.expectEqual(@as(i32, 42), list.pop());
}
test "fuzz example" {
const Context = struct {
fn testOne(context: @This(), input: []const u8) anyerror!void {
_ = context;
// Try passing `--fuzz` to `zig build test` and see if it manages to fail this test case!
try std.testing.expect(!std.mem.eql(u8, "canyoufindme", input));
}
};
try std.testing.fuzz(Context{}, Context.testOne, .{});
}