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

feat: colorize --list-languages output

Browse source
This commit is contained in:
CJ van den Berg 2024-09-02 21:43:39 +02:00
parent 72a40a6711
commit 595919c764
Signed by: neurocyte
GPG key ID: 8EB1E1BB660E3FB9
2 changed files with 22 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/list_languages.zig
View file

@ -2,7 +2,7 @@ const std = @import("std");
const syntax = @import("syntax"); const syntax = @import("syntax");
const builtin = @import("builtin"); const builtin = @import("builtin");
pub fn list(allocator: std.mem.Allocator, writer: anytype) !void { pub fn list(allocator: std.mem.Allocator, writer: anytype, tty_config: std.io.tty.Config) !void {
var max_language_len: usize = 0; var max_language_len: usize = 0;
var max_langserver_len: usize = 0; var max_langserver_len: usize = 0;
var max_formatter_len: usize = 0; var max_formatter_len: usize = 0;
@ -32,19 +32,19 @@ pub fn list(allocator: std.mem.Allocator, writer: anytype) !void {
for (syntax.FileType.file_types) |file_type| { for (syntax.FileType.file_types) |file_type| {
try write_string(writer, file_type.name, max_language_len + 1); try write_string(writer, file_type.name, max_language_len + 1);
try write_segmented(writer, file_type.extensions, ",", max_extensions_len + 1); try write_segmented(writer, file_type.extensions, ",", max_extensions_len + 1, tty_config);
if (builtin.os.tag != .windows) if (builtin.os.tag != .windows)
if (file_type.language_server) |language_server| if (file_type.language_server) |language_server|
try write_checkmark(writer, try can_execute(allocator, bin_paths, language_server[0])); try write_checkmark(writer, try can_execute(allocator, bin_paths, language_server[0]), tty_config);
try write_segmented(writer, file_type.language_server, " ", max_langserver_len + 1); try write_segmented(writer, file_type.language_server, " ", max_langserver_len + 1, tty_config);
if (builtin.os.tag != .windows) if (builtin.os.tag != .windows)
if (file_type.formatter) |formatter| if (file_type.formatter) |formatter|
try write_checkmark(writer, try can_execute(allocator, bin_paths, formatter[0])); try write_checkmark(writer, try can_execute(allocator, bin_paths, formatter[0]), tty_config);
try write_segmented(writer, file_type.formatter, " ", max_formatter_len); try write_segmented(writer, file_type.formatter, " ", max_formatter_len, tty_config);
try writer.writeAll("\n"); try writer.writeAll("\n");
} }
} }
@ -65,12 +65,19 @@ fn write_string(writer: anytype, string: []const u8, pad: usize) !void {
try write_padding(writer, string.len, pad); try write_padding(writer, string.len, pad);
} }
fn write_checkmark(writer: anytype, value: bool) !void { fn write_checkmark(writer: anytype, success: bool, tty_config: std.io.tty.Config) !void {
if (value) try writer.writeAll("✓") else try writer.writeAll("✘"); try tty_config.setColor(writer, if (success) .green else .red);
if (success) try writer.writeAll("✓") else try writer.writeAll("✘");
try writer.writeAll(" "); try writer.writeAll(" ");
} }
fn write_segmented(writer: anytype, args_: ?[]const []const u8, sep: []const u8, pad: usize) !void { fn write_segmented(
writer: anytype,
args_: ?[]const []const u8,
sep: []const u8,
pad: usize,
tty_config: std.io.tty.Config,
) !void {
const args = args_ orelse return; const args = args_ orelse return;
var len: usize = 0; var len: usize = 0;
var first: bool = true; var first: bool = true;
@ -82,6 +89,7 @@ fn write_segmented(writer: anytype, args_: ?[]const []const u8, sep: []const u8,
len += arg.len; len += arg.len;
try writer.writeAll(arg); try writer.writeAll(arg);
} }
try tty_config.setColor(writer, .reset);
try write_padding(writer, len, pad); try write_padding(writer, len, pad);
} }

7
src/main.zig
View file

@ -88,8 +88,11 @@ pub fn main() anyerror!void {
if (res.args.version != 0) if (res.args.version != 0)
return std.io.getStdOut().writeAll(@embedFile("version_info")); return std.io.getStdOut().writeAll(@embedFile("version_info"));
if (res.args.@"list-languages" != 0) if (res.args.@"list-languages" != 0) {
return list_languages.list(a, std.io.getStdOut().writer()); const stdout = std.io.getStdOut();
const tty_config = std.io.tty.detectConfig(stdout);
return list_languages.list(a, stdout.writer(), tty_config);
}
if (builtin.os.tag != .windows) if (builtin.os.tag != .windows)
if (std.posix.getenv("JITDEBUG")) |_| thespian.install_debugger(); if (std.posix.getenv("JITDEBUG")) |_| thespian.install_debugger();