47 lines
1.8 KiB
Zig
47 lines
1.8 KiB
Zig
const std = @import("std");
|
|
|
|
/// Returns the current logged-in user's username.
|
|
/// Uses the environment variable `USER`.
|
|
/// The caller is responsible for freeing the allocated memory.
|
|
pub fn getUsername(allocator: std.mem.Allocator) ![]u8 {
|
|
const username = try std.process.getEnvVarOwned(allocator, "USER");
|
|
return username;
|
|
}
|
|
|
|
pub fn getShell(allocator: std.mem.Allocator) ![]u8 {
|
|
const shell = std.process.getEnvVarOwned(allocator, "SHELL") catch |err| if (err == error.EnvironmentVariableNotFound) {
|
|
return allocator.dupe(u8, "Unknown");
|
|
} else return err;
|
|
|
|
defer allocator.free(shell);
|
|
|
|
const result = try std.process.Child.run(.{ .allocator = allocator, .argv = &[_][]const u8{ shell, "--version" } });
|
|
const result_stdout = result.stdout;
|
|
|
|
if (std.mem.indexOf(u8, shell, "bash") != null) {
|
|
const bash_version = parseBashVersion(result_stdout);
|
|
defer allocator.free(result_stdout);
|
|
return try std.fmt.allocPrint(allocator, "{s} {s}", .{ "bash", bash_version.? });
|
|
}
|
|
|
|
return result_stdout;
|
|
}
|
|
|
|
fn parseBashVersion(shell_version_output: []u8) ?[]u8 {
|
|
const end_index = std.mem.indexOf(u8, shell_version_output, "(");
|
|
if (end_index == null) return null;
|
|
|
|
const version_keyword = "version ";
|
|
const version_keyword_index = std.mem.indexOf(u8, shell_version_output[0..end_index.?], version_keyword);
|
|
if (version_keyword_index == null) return null;
|
|
|
|
return shell_version_output[version_keyword_index.? + version_keyword.len .. end_index.?];
|
|
}
|
|
|
|
pub fn getTerminalName(allocator: std.mem.Allocator) ![]u8 {
|
|
const term_program = std.process.getEnvVarOwned(allocator, "TERM_PROGRAM") catch |err| if (err == error.EnvironmentVariableNotFound) {
|
|
return allocator.dupe(u8, "Unknown");
|
|
} else return err;
|
|
return term_program;
|
|
}
|