neurocyte/zat
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: act more like standard cat if stdout is not a tty

Browse source
This commit is contained in:
CJ van den Berg 2024-11-18 17:40:05 +01:00
parent e3cef17b63
commit ec79f693a5
Signed by: neurocyte
GPG key ID: 8EB1E1BB660E3FB9
1 changed files with 34 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
src/main.zig
View file

@ -53,8 +53,9 @@ pub fn main() !void {
};
defer res.deinit();
const stdout_file = std.io.getStdOut().writer();
var bw = std.io.bufferedWriter(stdout_file);
const stdout_file = std.io.getStdOut();
const stdout_writer = stdout_file.writer();
var bw = std.io.bufferedWriter(stdout_writer);
const writer = bw.writer();
defer bw.flush() catch {};
@ -67,6 +68,9 @@ pub fn main() !void {
if (res.args.@"list-languages" != 0)
return list_langs(writer);
if (!stdout_file.supportsAnsiEscapeCodes())
return plain_cat(res.positionals);
var conf_buf: ?[]const u8 = null;
const conf = config_loader.read_config(a, &conf_buf);
const theme_name = if (res.args.theme) |theme| theme else conf.theme;
@ -103,7 +107,10 @@ pub fn main() !void {
if (res.positionals.len > 0) {
for (res.positionals) |arg| {
const file = try std.fs.cwd().openFile(arg, .{ .mode = .read_only });
const file = if (std.mem.eql(u8, arg, "-"))
std.io.getStdIn()
else
try std.fs.cwd().openFile(arg, .{ .mode = .read_only });
defer file.close();
const content = try file.readToEndAlloc(a, std.math.maxInt(u32));
defer a.free(content);
@ -478,3 +485,27 @@ fn render_file_type(writer: Writer, file_type: *const syntax.FileType, theme: *c
try set_ansi_style(writer, plain);
try writer.writeAll("\n");
}
fn plain_cat(files: []const []const u8) !void {
const stdout = std.io.getStdOut();
if (files.len == 0) {
try plain_cat_file(stdout, "-");
} else {
for (files) |file| try plain_cat_file(stdout, file);
}
}
fn plain_cat_file(out_file: std.fs.File, in_file_name: []const u8) !void {
var in_file = if (std.mem.eql(u8, in_file_name, "-"))
std.io.getStdIn()
else
try std.fs.cwd().openFile(in_file_name, .{});
defer in_file.close();
var buf: [std.mem.page_size]u8 = undefined;
while (true) {
const bytes_read = try in_file.read(&buf);
if (bytes_read == 0) return;
try out_file.writeAll(buf[0..bytes_read]);
}
}