Compare commits
33 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e492443a2c | ||
|
|
f1bebb08e9 | ||
|
|
f3af8712d2 | ||
|
|
67a2e19ac4 | ||
|
|
2de5047eb7 | ||
|
|
e88586c47a | ||
|
|
28f94d87ed | ||
|
|
0d7f606b7d | ||
|
|
ec0e3ffd18 | ||
|
|
f73b1be023 | ||
|
|
3a5a268bbe | ||
|
|
bdf4a3f7d7 | ||
|
|
68cc541575 | ||
|
|
e28a3b96e5 | ||
|
|
01ad39c0b5 | ||
|
|
a18fd7ade8 | ||
|
|
3e14691e14 | ||
|
|
4eab0e4487 | ||
|
|
3b737ed4d7 | ||
|
|
073d512b67 | ||
|
|
707f94bd8b | ||
|
|
de214a4a89 | ||
|
|
5b63b30a4f | ||
|
|
9af26702cc | ||
|
|
c660fc903f | ||
|
|
90abbcb1fd | ||
|
|
a52dd145da | ||
|
|
5faf60e53d | ||
|
|
7993cab4df | ||
|
|
b4ec1b6d2d | ||
|
|
cd40fe34d2 | ||
|
|
5e71eaf3f6 | ||
|
|
7ac43a2694 |
@@ -14,7 +14,11 @@ Zigfetch is a minimal [neofetch](https://github.com/dylanaraps/neofetch)/[fastfe
|
||||
|
||||
## Requirements
|
||||
|
||||
- [zig v0.14.0](https://ziglang.org/)
|
||||
- \>= [zig v0.14.0](https://ziglang.org/)
|
||||
|
||||
### Linux only
|
||||
|
||||
- [libpci](https://github.com/pciutils/pciutils)
|
||||
|
||||
## Installation
|
||||
|
||||
@@ -71,4 +75,5 @@ $ cp /path/to/zigfetch/config.json ~/.config/zigfetch/config.json
|
||||
- [ ] Add support for Windows
|
||||
|
||||
## Contributing
|
||||
|
||||
If you would like to contribute to this project just create a pull request which I will try to review as soon as possible.
|
||||
|
||||
@@ -29,6 +29,7 @@ pub fn build(b: *std.Build) void {
|
||||
|
||||
if (target.result.os.tag == .linux) {
|
||||
exe.linkLibC();
|
||||
exe.linkSystemLibrary("pci");
|
||||
}
|
||||
|
||||
// This declares intent for the executable to be installed into the
|
||||
|
||||
@@ -10,7 +10,7 @@
|
||||
|
||||
// This is a [Semantic Version](https://semver.org/).
|
||||
// In a future version of Zig it will be used for package deduplication.
|
||||
.version = "0.5.1",
|
||||
.version = "0.7.2",
|
||||
|
||||
// Together with name, this represents a globally unique package
|
||||
// identifier. This field is generated by the Zig toolchain when the
|
||||
|
||||
@@ -105,11 +105,13 @@ pub fn printAscii(allocator: std.mem.Allocator, sys_info_list: std.ArrayList([]u
|
||||
if (i < sys_info_len) {
|
||||
try stdout.print("{s}\n", .{sys_info_items[i]});
|
||||
} else if (i == sys_info_len + 1) {
|
||||
// Print the first row of colors
|
||||
for (0..8) |j| {
|
||||
try stdout.print("\x1b[48;5;{d}m \x1b[0m", .{j});
|
||||
}
|
||||
try stdout.print("\n", .{});
|
||||
} else if (i == sys_info_len + 2) {
|
||||
// Print the second row of colors
|
||||
for (8..16) |j| {
|
||||
try stdout.print("\x1b[48;5;{d}m \x1b[0m", .{j});
|
||||
}
|
||||
|
||||
@@ -23,9 +23,9 @@ pub const ModuleType = enum {
|
||||
ram,
|
||||
swap,
|
||||
disk,
|
||||
net,
|
||||
terminal,
|
||||
locale,
|
||||
net, // It must always be the last element of this array
|
||||
};
|
||||
|
||||
pub fn getModulesTypes(allocator: std.mem.Allocator, config: ?std.json.Parsed(Config)) !std.ArrayList(ModuleType) {
|
||||
|
||||
@@ -3,7 +3,12 @@ const std = @import("std");
|
||||
const ascii = @import("ascii.zig");
|
||||
const detection = @import("detection.zig").os_module;
|
||||
|
||||
pub const formatters = [_]*const fn (allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) anyerror![]u8{
|
||||
const Result = union(enum) {
|
||||
string: []u8,
|
||||
string_arraylist: std.ArrayList([]u8),
|
||||
};
|
||||
|
||||
pub const formatters = [_]*const fn (allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) anyerror!Result{
|
||||
&getFormattedOsInfo,
|
||||
&getFormattedKernelInfo,
|
||||
&getFormattedUptimeInfo,
|
||||
@@ -14,11 +19,12 @@ pub const formatters = [_]*const fn (allocator: std.mem.Allocator, key: []const
|
||||
&getFormattedRamInfo,
|
||||
&getFormattedSwapInfo,
|
||||
&getFormattedDiskInfo,
|
||||
&getFormattedNetInfo,
|
||||
&getFormattedTerminalNameInfo,
|
||||
&getFormattedLocaleInfo,
|
||||
};
|
||||
|
||||
pub const default_formatters = [_]*const fn (allocator: std.mem.Allocator) anyerror![]u8{
|
||||
pub const default_formatters = [_]*const fn (allocator: std.mem.Allocator) anyerror!Result{
|
||||
&getDefaultFormattedOsInfo,
|
||||
&getDefaultFormattedKernelInfo,
|
||||
&getDefaultFormattedUptimeInfo,
|
||||
@@ -29,145 +35,166 @@ pub const default_formatters = [_]*const fn (allocator: std.mem.Allocator) anyer
|
||||
&getDefaultFormattedRamInfo,
|
||||
&getDefaultFormattedSwapInfo,
|
||||
&getDefaultFormattedDiskInfo,
|
||||
&getDefaultFormattedNetInfo,
|
||||
&getDefaultFormattedTerminalNameInfo,
|
||||
&getDefaultFormattedLocaleInfo,
|
||||
};
|
||||
|
||||
pub fn getDefaultFormattedKernelInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedKernelInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedKernelInfo(allocator, "Kernel", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedKernelInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedKernelInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const kernel_info = try detection.system.getKernelInfo(allocator);
|
||||
defer allocator.free(kernel_info.kernel_name);
|
||||
defer allocator.free(kernel_info.kernel_release);
|
||||
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s} {s}", .{ key_color, key, ascii.Reset, kernel_info.kernel_name, kernel_info.kernel_release });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s} {s}", .{ key_color, key, ascii.Reset, kernel_info.kernel_name, kernel_info.kernel_release }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedOsInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedOsInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedOsInfo(allocator, "OS", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedOsInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedOsInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const os_info = try detection.system.getOsInfo(allocator);
|
||||
defer allocator.free(os_info);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, os_info });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, os_info }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedLocaleInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedLocaleInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedLocaleInfo(allocator, "Locale", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedLocaleInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedLocaleInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const locale = try detection.system.getLocale(allocator);
|
||||
defer allocator.free(locale);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, locale });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, locale }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedUptimeInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedUptimeInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedUptimeInfo(allocator, "Uptime", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedUptimeInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedUptimeInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const uptime = try detection.system.getSystemUptime();
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {} days, {} hours, {} minutes ", .{ key_color, key, ascii.Reset, uptime.days, uptime.hours, uptime.minutes });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {} days, {} hours, {} minutes ", .{ key_color, key, ascii.Reset, uptime.days, uptime.hours, uptime.minutes }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedPackagesInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedPackagesInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedPackagesInfo(allocator, "Packages", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedPackagesInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedPackagesInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
if (builtin.os.tag == .macos) {
|
||||
const packages_info = try detection.packages.getPackagesInfo(allocator);
|
||||
defer allocator.free(packages_info);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s}{s}", .{ key_color, key, ascii.Reset, packages_info });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s}{s}", .{ key_color, key, ascii.Reset, packages_info }) };
|
||||
} else if (builtin.os.tag == .linux) {
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} WIP", .{ key_color, key, ascii.Reset });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} WIP", .{ key_color, key, ascii.Reset }) };
|
||||
}
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedShellInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedShellInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedShellInfo(allocator, "Shell", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedShellInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedShellInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const shell = try detection.user.getShell(allocator);
|
||||
defer allocator.free(shell);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, shell[0..(shell.len - 1)] });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, shell[0..(shell.len - 1)] }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedCpuInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedCpuInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedCpuInfo(allocator, "Cpu", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedCpuInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedCpuInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const cpu_info = try detection.hardware.getCpuInfo(allocator);
|
||||
defer allocator.free(cpu_info.cpu_name);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s} ({}) @ {d:.2} GHz", .{ key_color, key, ascii.Reset, cpu_info.cpu_name, cpu_info.cpu_cores, cpu_info.cpu_max_freq });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s} ({}) @ {d:.2} GHz", .{ key_color, key, ascii.Reset, cpu_info.cpu_name, cpu_info.cpu_cores, cpu_info.cpu_max_freq }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedGpuInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedGpuInfo(allocator: std.mem.Allocator) !Result {
|
||||
if (builtin.os.tag == .macos) {
|
||||
return try getFormattedGpuInfo(allocator, "Gpu", ascii.Yellow);
|
||||
} else if (builtin.os.tag == .linux) {
|
||||
return try getFormattedGpuInfo(allocator, "Gpu", ascii.Yellow);
|
||||
}
|
||||
}
|
||||
|
||||
pub fn getFormattedGpuInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedGpuInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
if (builtin.os.tag == .macos) {
|
||||
const gpu_info = try detection.hardware.getGpuInfo(allocator);
|
||||
defer allocator.free(gpu_info.gpu_name);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s} ({}) @ {d:.2} GHz", .{ key_color, key, ascii.Reset, gpu_info.gpu_name, gpu_info.gpu_cores, gpu_info.gpu_freq });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s} ({}) @ {d:.2} GHz", .{ key_color, key, ascii.Reset, gpu_info.gpu_name, gpu_info.gpu_cores, gpu_info.gpu_freq }) };
|
||||
} else if (builtin.os.tag == .linux) {
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} WIP", .{ key_color, key, ascii.Reset });
|
||||
var formatted_gpu_info_list = std.ArrayList([]u8).init(allocator);
|
||||
|
||||
const gpu_info_list = try detection.hardware.getGpuInfo(allocator);
|
||||
|
||||
for (gpu_info_list.items) |g| {
|
||||
var formatted_gpu_info: []u8 = undefined;
|
||||
if ((g.gpu_cores == 0) or (g.gpu_freq == 0.0)) {
|
||||
formatted_gpu_info = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, g.gpu_name });
|
||||
} else {
|
||||
formatted_gpu_info = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s} ({}) @ {d:.2} GHz", .{ key_color, key, ascii.Reset, g.gpu_name, g.gpu_cores, g.gpu_freq });
|
||||
}
|
||||
try formatted_gpu_info_list.append(formatted_gpu_info);
|
||||
allocator.free(g.gpu_name);
|
||||
}
|
||||
gpu_info_list.deinit();
|
||||
|
||||
return Result{ .string_arraylist = formatted_gpu_info_list };
|
||||
}
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedRamInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedRamInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedRamInfo(allocator, "Ram", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedRamInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedRamInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const ram_info = if (builtin.os.tag == .macos) try detection.hardware.getRamInfo() else if (builtin.os.tag == .linux) try detection.hardware.getRamInfo(allocator);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {d:.2} / {d:.2} GiB ({}%)", .{ key_color, key, ascii.Reset, ram_info.ram_usage, ram_info.ram_size, ram_info.ram_usage_percentage });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {d:.2} / {d:.2} GiB ({}%)", .{ key_color, key, ascii.Reset, ram_info.ram_usage, ram_info.ram_size, ram_info.ram_usage_percentage }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedSwapInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedSwapInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedSwapInfo(allocator, "Swap", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedSwapInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedSwapInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const swap_info = if (builtin.os.tag == .macos) try detection.hardware.getSwapInfo() else if (builtin.os.tag == .linux) try detection.hardware.getSwapInfo(allocator);
|
||||
if (swap_info) |s| {
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {d:.2} / {d:.2} GiB ({}%)", .{ key_color, key, ascii.Reset, s.swap_usage, s.swap_usage, s.swap_usage_percentage });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {d:.2} / {d:.2} GiB ({}%)", .{ key_color, key, ascii.Reset, s.swap_usage, s.swap_size, s.swap_usage_percentage }) };
|
||||
} else {
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} Disabled", .{ key_color, key, ascii.Reset });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} Disabled", .{ key_color, key, ascii.Reset }) };
|
||||
}
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedDiskInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedDiskInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedDiskInfo(allocator, "Disk", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedDiskInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedDiskInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const disk_info = try detection.hardware.getDiskSize("/");
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s} ({s}):{s} {d:.2} / {d:.2} GB ({}%)", .{ key_color, key, disk_info.disk_path, ascii.Reset, disk_info.disk_usage, disk_info.disk_size, disk_info.disk_usage_percentage });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s} ({s}):{s} {d:.2} / {d:.2} GB ({}%)", .{ key_color, key, disk_info.disk_path, ascii.Reset, disk_info.disk_usage, disk_info.disk_size, disk_info.disk_usage_percentage }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedTerminalNameInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedTerminalNameInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedTerminalNameInfo(allocator, "Terminal", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedTerminalNameInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) ![]u8 {
|
||||
pub fn getFormattedTerminalNameInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
const terminal_name = try detection.user.getTerminalName(allocator);
|
||||
defer allocator.free(terminal_name);
|
||||
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, terminal_name });
|
||||
return Result{ .string = try std.fmt.allocPrint(allocator, "{s}{s}:{s} {s}", .{ key_color, key, ascii.Reset, terminal_name }) };
|
||||
}
|
||||
|
||||
pub fn getDefaultFormattedNetInfo(allocator: std.mem.Allocator) ![]u8 {
|
||||
pub fn getDefaultFormattedNetInfo(allocator: std.mem.Allocator) !Result {
|
||||
return try getFormattedNetInfo(allocator, "Local IP", ascii.Yellow);
|
||||
}
|
||||
|
||||
pub fn getFormattedNetInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !std.ArrayList([]u8) {
|
||||
pub fn getFormattedNetInfo(allocator: std.mem.Allocator, key: []const u8, key_color: []const u8) !Result {
|
||||
var formatted_net_info_list = std.ArrayList([]u8).init(allocator);
|
||||
|
||||
var net_info_list = try detection.network.getNetInfo(allocator);
|
||||
@@ -178,5 +205,5 @@ pub fn getFormattedNetInfo(allocator: std.mem.Allocator, key: []const u8, key_co
|
||||
}
|
||||
net_info_list.deinit();
|
||||
|
||||
return formatted_net_info_list;
|
||||
return Result{ .string_arraylist = formatted_net_info_list };
|
||||
}
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
const std = @import("std");
|
||||
const c_unistd = @cImport(@cInclude("unistd.h"));
|
||||
const c_statvfs = @cImport(@cInclude("sys/statvfs.h"));
|
||||
const c_libpci = @cImport(@cInclude("pci/pci.h"));
|
||||
|
||||
/// Struct representing CPU informations
|
||||
pub const CpuInfo = struct {
|
||||
@@ -9,6 +10,13 @@ pub const CpuInfo = struct {
|
||||
cpu_max_freq: f32,
|
||||
};
|
||||
|
||||
/// Struct representing GPU informations
|
||||
pub const GpuInfo = struct {
|
||||
gpu_name: []u8,
|
||||
gpu_cores: i32,
|
||||
gpu_freq: f64,
|
||||
};
|
||||
|
||||
/// Struct representing RAM usage informations
|
||||
pub const RamInfo = struct {
|
||||
ram_size: f64,
|
||||
@@ -43,6 +51,7 @@ pub fn getCpuInfo(allocator: std.mem.Allocator) !CpuInfo {
|
||||
|
||||
// Parsing /proc/cpuinfo
|
||||
var model_name: ?[]const u8 = null;
|
||||
var cpu_max_freq_mhz_str: ?[]const u8 = null;
|
||||
|
||||
var lines = std.mem.splitScalar(u8, cpuinfo_data, '\n');
|
||||
while (lines.next()) |line| {
|
||||
@@ -52,14 +61,36 @@ pub fn getCpuInfo(allocator: std.mem.Allocator) !CpuInfo {
|
||||
_ = parts.next(); // discards the key
|
||||
if (parts.next()) |value| {
|
||||
model_name = std.mem.trim(u8, value, " ");
|
||||
break;
|
||||
}
|
||||
} else if (std.mem.startsWith(u8, trimmed, "cpu MHz") and cpu_max_freq_mhz_str == null) {
|
||||
var parts = std.mem.splitScalar(u8, trimmed, ':');
|
||||
_ = parts.next(); // discard the key
|
||||
if (parts.next()) |value| {
|
||||
cpu_max_freq_mhz_str = std.mem.trim(u8, value, " ");
|
||||
}
|
||||
}
|
||||
|
||||
// Reads /sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq
|
||||
if ((model_name != null) and (cpu_max_freq_mhz_str != null)) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
var cpu_max_freq: f32 = 0.0;
|
||||
|
||||
const cpuinfo_max_freq_path = "/sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq";
|
||||
var cmf_exists: bool = true;
|
||||
|
||||
// Checks if /sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq exists
|
||||
_ = std.fs.accessAbsolute(cpuinfo_max_freq_path, .{ .mode = .read_only }) catch |err| {
|
||||
if (err == std.posix.AccessError.FileNotFound) {
|
||||
cmf_exists = false;
|
||||
}
|
||||
};
|
||||
|
||||
if (cmf_exists) {
|
||||
// Reads /sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq
|
||||
var file2 = try std.fs.cwd().openFile(cpuinfo_max_freq_path, .{});
|
||||
|
||||
defer file2.close();
|
||||
const cpuinfo_max_freq_data = try file2.readToEndAlloc(allocator, std.math.maxInt(usize));
|
||||
defer allocator.free(cpuinfo_max_freq_data);
|
||||
@@ -67,7 +98,13 @@ pub fn getCpuInfo(allocator: std.mem.Allocator) !CpuInfo {
|
||||
// Parsing /sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq
|
||||
const trimmed = std.mem.trim(u8, cpuinfo_max_freq_data, " \n\r");
|
||||
const cpu_max_freq_khz: f32 = try std.fmt.parseFloat(f32, trimmed);
|
||||
const cpu_max_freq: f32 = cpu_max_freq_khz / 1_000_000;
|
||||
cpu_max_freq = cpu_max_freq_khz / 1_000_000;
|
||||
} else {
|
||||
if (cpu_max_freq_mhz_str != null) {
|
||||
const cpu_max_freq_mhz: f32 = try std.fmt.parseFloat(f32, cpu_max_freq_mhz_str.?);
|
||||
cpu_max_freq = cpu_max_freq_mhz / 1_000;
|
||||
}
|
||||
}
|
||||
|
||||
return CpuInfo{
|
||||
.cpu_name = try allocator.dupe(u8, model_name orelse "Unknown"),
|
||||
@@ -76,6 +113,86 @@ pub fn getCpuInfo(allocator: std.mem.Allocator) !CpuInfo {
|
||||
};
|
||||
}
|
||||
|
||||
pub fn getGpuInfo(allocator: std.mem.Allocator) !std.ArrayList(GpuInfo) {
|
||||
var gpu_info_list = std.ArrayList(GpuInfo).init(allocator);
|
||||
|
||||
const display_controller = 0x03;
|
||||
|
||||
const pacc = c_libpci.pci_alloc();
|
||||
defer c_libpci.pci_cleanup(pacc);
|
||||
c_libpci.pci_init(pacc);
|
||||
c_libpci.pci_scan_bus(pacc);
|
||||
|
||||
var devices = pacc.*.devices;
|
||||
while (devices != null) : (devices = devices.*.next) {
|
||||
// NOTE: for references: https://github.com/pciutils/pciutils/blob/3ec74c71c01878f92e751f15bb8febe720c3ab40/lib/access.c#L194
|
||||
const known_fields = c_libpci.pci_fill_info(devices, c_libpci.PCI_FILL_IDENT | c_libpci.PCI_FILL_CLASS);
|
||||
if (known_fields <= 0) {
|
||||
return error.NoLibpciFieldsFound;
|
||||
}
|
||||
|
||||
if ((devices.*.device_class >> 8) == display_controller) {
|
||||
var name_buffer: [256]u8 = undefined;
|
||||
|
||||
const name = c_libpci.pci_lookup_name(
|
||||
pacc,
|
||||
&name_buffer,
|
||||
name_buffer.len,
|
||||
c_libpci.PCI_LOOKUP_VENDOR | c_libpci.PCI_LOOKUP_DEVICE,
|
||||
devices.*.vendor_id,
|
||||
devices.*.device_id,
|
||||
);
|
||||
|
||||
const gpu_name = try allocator.dupe(u8, std.mem.span(name));
|
||||
defer allocator.free(gpu_name);
|
||||
|
||||
const parsed_gpu_name = try parseGpuName(allocator, gpu_name);
|
||||
|
||||
try gpu_info_list.append(GpuInfo{
|
||||
.gpu_name = parsed_gpu_name,
|
||||
.gpu_cores = 0,
|
||||
.gpu_freq = 0.0,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
if (gpu_info_list.items.len == 0) {
|
||||
try gpu_info_list.append(GpuInfo{
|
||||
.gpu_name = try allocator.dupe(u8, "Unknown"),
|
||||
.gpu_cores = 0,
|
||||
.gpu_freq = 0.0,
|
||||
});
|
||||
}
|
||||
|
||||
return gpu_info_list;
|
||||
}
|
||||
|
||||
fn parseGpuName(allocator: std.mem.Allocator, name: []u8) ![]u8 {
|
||||
// NOTE: for references: https://github.com/pciutils/pciutils/blob/master/pci.ids
|
||||
|
||||
if (std.mem.startsWith(u8, name, "Advanced Micro Devices, Inc. [AMD/ATI]")) {
|
||||
const size = std.mem.replacementSize(u8, name, "Advanced Micro Devices, Inc. [AMD/ATI]", "AMD");
|
||||
const parsed_gpu_name = try allocator.alloc(u8, size);
|
||||
_ = std.mem.replace(u8, name, "Advanced Micro Devices, Inc. [AMD/ATI]", "AMD", parsed_gpu_name);
|
||||
|
||||
return parsed_gpu_name;
|
||||
} else if (std.mem.startsWith(u8, name, "Intel Corporation")) {
|
||||
const size = std.mem.replacementSize(u8, name, "Intel Corporation", "Intel");
|
||||
const parsed_gpu_name = try allocator.alloc(u8, size);
|
||||
_ = std.mem.replace(u8, name, "Intel Corporation", "Intel", parsed_gpu_name);
|
||||
|
||||
return parsed_gpu_name;
|
||||
} else if (std.mem.startsWith(u8, name, "NVIDIA Corporation")) {
|
||||
const size = std.mem.replacementSize(u8, name, "NVIDIA Corporation", "NVIDIA");
|
||||
const parsed_gpu_name = try allocator.alloc(u8, size);
|
||||
_ = std.mem.replace(u8, name, "NVIDIA Corporation", "NVIDIA", parsed_gpu_name);
|
||||
|
||||
return parsed_gpu_name;
|
||||
}
|
||||
|
||||
return name;
|
||||
}
|
||||
|
||||
pub fn getRamInfo(allocator: std.mem.Allocator) !RamInfo {
|
||||
// Reads /proc/meminfo
|
||||
const meminfo_path = "/proc/meminfo";
|
||||
@@ -203,7 +320,7 @@ pub fn getDiskSize(disk_path: []const u8) !DiskInfo {
|
||||
}
|
||||
|
||||
const total_size = stat.f_blocks * stat.f_frsize;
|
||||
const free_size = stat.f_bavail * stat.f_frsize;
|
||||
const free_size = stat.f_bfree * stat.f_frsize;
|
||||
const used_size = total_size - free_size;
|
||||
|
||||
const used_size_percentage = (used_size * 100) / total_size;
|
||||
|
||||
@@ -24,8 +24,8 @@ pub fn getShell(allocator: std.mem.Allocator) ![]u8 {
|
||||
}
|
||||
|
||||
pub fn getTerminalName(allocator: std.mem.Allocator) ![]u8 {
|
||||
const term_progrm = std.process.getEnvVarOwned(allocator, "TERM_PROGRAM") catch |err| if (err == error.EnvironmentVariableNotFound) {
|
||||
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_progrm;
|
||||
return term_program;
|
||||
}
|
||||
|
||||
@@ -172,7 +172,6 @@ fn getCpuFreqAppleSilicon() !f64 {
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: test on intel machine
|
||||
pub fn getCpuFreqIntel() f64 {
|
||||
var freq: f64 = 0;
|
||||
var size: usize = @sizeOf(f64);
|
||||
@@ -419,7 +418,7 @@ pub fn getDiskSize(disk_path: []const u8) !DiskInfo {
|
||||
}
|
||||
|
||||
const total_size = stat.f_blocks * stat.f_frsize;
|
||||
const free_size = stat.f_bavail * stat.f_frsize;
|
||||
const free_size = stat.f_bfree * stat.f_frsize;
|
||||
const used_size = total_size - free_size;
|
||||
|
||||
const used_size_percentage = (used_size * 100) / total_size;
|
||||
|
||||
@@ -27,8 +27,8 @@ pub fn getShell(allocator: std.mem.Allocator) ![]u8 {
|
||||
}
|
||||
|
||||
pub fn getTerminalName(allocator: std.mem.Allocator) ![]u8 {
|
||||
const term_progrm = std.process.getEnvVarOwned(allocator, "TERM_PROGRAM") catch |err| if (err == error.EnvironmentVariableNotFound) {
|
||||
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_progrm;
|
||||
return term_program;
|
||||
}
|
||||
|
||||
27
src/main.zig
27
src/main.zig
@@ -43,8 +43,15 @@ pub fn main() !void {
|
||||
try sys_info_list.append(separtor_buffer);
|
||||
|
||||
if (modules_types.items.len == 0) {
|
||||
inline for (0..formatters.formatters.len) |i| {
|
||||
try sys_info_list.append(try formatters.default_formatters[i](allocator));
|
||||
inline for (0..formatters.default_formatters.len) |i| {
|
||||
const result = try formatters.default_formatters[i](allocator);
|
||||
switch (result) {
|
||||
.string => |r| try sys_info_list.append(r),
|
||||
.string_arraylist => |r| {
|
||||
defer r.deinit();
|
||||
try sys_info_list.appendSlice(r.items);
|
||||
},
|
||||
}
|
||||
}
|
||||
} else if (conf) |c| {
|
||||
for (modules_types.items, c.value.modules) |module_type, module| {
|
||||
@@ -52,16 +59,14 @@ pub fn main() !void {
|
||||
const rgb = try ascii.hexColorToRgb(module.key_color);
|
||||
const key_color = try std.fmt.bufPrint(&buf, "\x1b[38;2;{d};{d};{d}m", .{ rgb.r, rgb.g, rgb.b });
|
||||
|
||||
if (module_type == config.ModuleType.net) {
|
||||
var formatted_net_info_list = try formatters.getFormattedNetInfo(allocator, module.key, key_color);
|
||||
defer formatted_net_info_list.deinit();
|
||||
|
||||
try sys_info_list.appendSlice(formatted_net_info_list.items);
|
||||
|
||||
continue;
|
||||
const result = try formatters.formatters[@intFromEnum(module_type)](allocator, module.key, key_color);
|
||||
switch (result) {
|
||||
.string => |r| try sys_info_list.append(r),
|
||||
.string_arraylist => |r| {
|
||||
defer r.deinit();
|
||||
try sys_info_list.appendSlice(r.items);
|
||||
},
|
||||
}
|
||||
|
||||
try sys_info_list.append(try formatters.formatters[@intFromEnum(module_type)](allocator, module.key, key_color));
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user