Skip to content

Commit

Permalink
Merge pull request #12 from Reokodoku/no-color
Browse files Browse the repository at this point in the history
Support `NO_COLOR` env
  • Loading branch information
Arnau478 authored Apr 29, 2024
2 parents 679c5fd + 8a3b0e9 commit d82c7a5
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion src/main.zig
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,19 @@ inline fn isPrintable(c: u8) bool {
};
}

inline fn useColor(stdout: std.fs.File, color_arg: ?bool) !bool {
var envs = try std.process.getEnvMap(allocator);
defer envs.deinit();

if (color_arg) |val| {
return val;
} else if (envs.get("NO_COLOR")) |val| {
if (!std.mem.eql(u8, val, ""))
return false;
}
return stdout.supportsAnsiEscapeCodes();
}

const DisplayLineOptions = struct {
color: bool,
uppercase: bool,
Expand Down Expand Up @@ -124,7 +137,7 @@ pub fn main() !void {
const stdout = std.io.getStdOut();

try display(file.reader(), stdout.writer(), .{
.color = parsed_args.color orelse stdout.supportsAnsiEscapeCodes(),
.color = try useColor(stdout, parsed_args.color),
.uppercase = parsed_args.uppercase orelse false,
.show_size = parsed_args.show_size orelse true,
.show_offset = parsed_args.show_offset orelse true,
Expand Down

0 comments on commit d82c7a5

Please sign in to comment.