feat: add a utility function to retrieve the terminal size

This commit is contained in:
utox39
2025-07-31 01:43:40 +02:00
parent 3b3c29e22d
commit 9df0845530

51
src/utils.zig Normal file
View File

@@ -0,0 +1,51 @@
const std = @import("std");
const builtin = @import("builtin");
pub const TermSize = struct {
width: u16,
height: u16,
};
pub fn getTerminalSize() !TermSize {
// https://github.com/softprops/zig-termsize (https://github.com/softprops/zig-termsize/blob/main/src/main.zig)
const stdout = std.io.getStdIn();
switch (builtin.os.tag) {
.windows => {
var buf: std.os.windows.CONSOLE_SCREEN_BUFFER_INFO = undefined;
switch (std.os.windows.kernel32.GetConsoleScreenBufferInfo(stdout.handle, &buf)) {
std.os.windows.TRUE => return TermSize{
.width = @intCast(buf.srWindow.Right - buf.srWindow.Left + 1),
.height = @intCast(buf.srWindow.Bottom - buf.srWindow.Top + 1),
},
else => return error.GetConsoleScreenBufferInfoFailed,
}
},
.linux, .macos => {
var buf: std.posix.winsize = undefined;
switch (std.posix.errno(
std.posix.system.ioctl(
stdout.handle,
std.posix.T.IOCGWINSZ,
@intFromPtr(&buf),
),
)) {
std.posix.E.SUCCESS => return TermSize{
.width = buf.col,
.height = buf.row,
},
else => return error.IoctlFailed,
}
},
else => return error.UnsupportedOperatingSystem,
}
}
test "getTerminalSize" {
const terminal_size = try getTerminalSize();
std.debug.print("Height: {}, Width {}\n", .{ terminal_size.height, terminal_size.width });
try std.testing.expect((terminal_size.height > 0) and (terminal_size.width > 0));
}