Merge pull request #3 from utox39/feat-linux/gpu-info

Feat linux/gpu info
This commit is contained in:
utox39
2025-07-12 01:56:51 +02:00
committed by GitHub
5 changed files with 160 additions and 72 deletions

View File

@@ -1,12 +1,12 @@
# Zigfetch # Zigfetch
- [Description](#description) - [Description](#description)
- [Requirements](#requirements) - [Requirements](#requirements)
- [Installation](#installation) - [Installation](#installation)
- [Usage](#usage) - [Usage](#usage)
- [Configuration](#configuration) - [Configuration](#configuration)
- [Roadtrip](#roadtrip) - [Roadtrip](#roadtrip)
- [Contributing](#contributing) - [Contributing](#contributing)
## Description ## Description
@@ -14,7 +14,11 @@ Zigfetch is a minimal [neofetch](https://github.com/dylanaraps/neofetch)/[fastfe
## Requirements ## Requirements
- [zig v0.14.0](https://ziglang.org/) - \>= [zig v0.14.0](https://ziglang.org/)
### Linux only
- [libpci](https://github.com/pciutils/pciutils) (linux only)
## Installation ## Installation
@@ -43,20 +47,20 @@ $ zigfetch
> [!IMPORTANT] > [!IMPORTANT]
> Currently, Zig does not have a built-in library for JSON validation via JSON schema, so it is very important to follow the pattern shown in the default configuration file ([config.json](https://github.com/utox39/zigfetch/blob/main/config.json)) to avoid errors > Currently, Zig does not have a built-in library for JSON validation via JSON schema, so it is very important to follow the pattern shown in the default configuration file ([config.json](https://github.com/utox39/zigfetch/blob/main/config.json)) to avoid errors
- Create the config folder - Create the config folder
```console ```console
$ mkdir -p ~/.config/zigfetch $ mkdir -p ~/.config/zigfetch
``` ```
- Create the config file - Create the config file
```console ```console
$ cd ~/.config/zigfetch $ cd ~/.config/zigfetch
$ touch config.json $ touch config.json
``` ```
- Or copy the default config (preferred way) - Or copy the default config (preferred way)
```console ```console
$ cp /path/to/zigfetch/config.json ~/.config/zigfetch/config.json $ cp /path/to/zigfetch/config.json ~/.config/zigfetch/config.json
@@ -64,11 +68,12 @@ $ cp /path/to/zigfetch/config.json ~/.config/zigfetch/config.json
## Roadtrip ## Roadtrip
- [ ] Add ASCII art for each operating system and Linux distro - [ ] Add ASCII art for each operating system and Linux distro
- [ ] Add GPU info for Linux - [ ] Add GPU info for Linux
- [ ] Add packages info for Linux - [ ] Add packages info for Linux
- [x] Add user customization - [x] Add user customization
- [ ] Add support for Windows - [ ] Add support for Windows
## Contributing ## 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. If you would like to contribute to this project just create a pull request which I will try to review as soon as possible.

View File

@@ -29,6 +29,7 @@ pub fn build(b: *std.Build) void {
if (target.result.os.tag == .linux) { if (target.result.os.tag == .linux) {
exe.linkLibC(); exe.linkLibC();
exe.linkSystemLibrary("pci");
} }
// This declares intent for the executable to be installed into the // This declares intent for the executable to be installed into the

View File

@@ -3,7 +3,12 @@ const std = @import("std");
const ascii = @import("ascii.zig"); const ascii = @import("ascii.zig");
const detection = @import("detection.zig").os_module; 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, &getFormattedOsInfo,
&getFormattedKernelInfo, &getFormattedKernelInfo,
&getFormattedUptimeInfo, &getFormattedUptimeInfo,
@@ -14,11 +19,12 @@ pub const formatters = [_]*const fn (allocator: std.mem.Allocator, key: []const
&getFormattedRamInfo, &getFormattedRamInfo,
&getFormattedSwapInfo, &getFormattedSwapInfo,
&getFormattedDiskInfo, &getFormattedDiskInfo,
&getFormattedNetInfo,
&getFormattedTerminalNameInfo, &getFormattedTerminalNameInfo,
&getFormattedLocaleInfo, &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, &getDefaultFormattedOsInfo,
&getDefaultFormattedKernelInfo, &getDefaultFormattedKernelInfo,
&getDefaultFormattedUptimeInfo, &getDefaultFormattedUptimeInfo,
@@ -29,145 +35,159 @@ pub const default_formatters = [_]*const fn (allocator: std.mem.Allocator) anyer
&getDefaultFormattedRamInfo, &getDefaultFormattedRamInfo,
&getDefaultFormattedSwapInfo, &getDefaultFormattedSwapInfo,
&getDefaultFormattedDiskInfo, &getDefaultFormattedDiskInfo,
&getDefaultFormattedNetInfo,
&getDefaultFormattedTerminalNameInfo, &getDefaultFormattedTerminalNameInfo,
&getDefaultFormattedLocaleInfo, &getDefaultFormattedLocaleInfo,
}; };
pub fn getDefaultFormattedKernelInfo(allocator: std.mem.Allocator) ![]u8 { pub fn getDefaultFormattedKernelInfo(allocator: std.mem.Allocator) !Result {
return try getFormattedKernelInfo(allocator, "Kernel", ascii.Yellow); 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); const kernel_info = try detection.system.getKernelInfo(allocator);
defer allocator.free(kernel_info.kernel_name); defer allocator.free(kernel_info.kernel_name);
defer allocator.free(kernel_info.kernel_release); 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); 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); const os_info = try detection.system.getOsInfo(allocator);
defer allocator.free(os_info); 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); 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); const locale = try detection.system.getLocale(allocator);
defer allocator.free(locale); 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); 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(); 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); 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) { if (builtin.os.tag == .macos) {
const packages_info = try detection.packages.getPackagesInfo(allocator); const packages_info = try detection.packages.getPackagesInfo(allocator);
defer allocator.free(packages_info); 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) { } 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); 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); const shell = try detection.user.getShell(allocator);
defer allocator.free(shell); 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); 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); const cpu_info = try detection.hardware.getCpuInfo(allocator);
defer allocator.free(cpu_info.cpu_name); 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 {
return try getFormattedGpuInfo(allocator, "Gpu", ascii.Yellow); 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 {
var formatted_gpu_info = std.ArrayList([]u8).init(allocator);
if (builtin.os.tag == .macos) { if (builtin.os.tag == .macos) {
const gpu_info = try detection.hardware.getGpuInfo(allocator); const gpu_info = try detection.hardware.getGpuInfo(allocator);
defer allocator.free(gpu_info.gpu_name); 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) { } else if (builtin.os.tag == .linux) {
return try std.fmt.allocPrint(allocator, "{s}{s}:{s} WIP", .{ key_color, key, ascii.Reset }); const gpu_info_list = try detection.hardware.getGpuInfo(allocator);
for (gpu_info_list.items) |g| {
try formatted_gpu_info.append(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 }));
allocator.free(g.gpu_name);
}
gpu_info_list.deinit();
return Result{ .string_arraylist = formatted_gpu_info };
} }
} }
pub fn getDefaultFormattedRamInfo(allocator: std.mem.Allocator) ![]u8 { pub fn getDefaultFormattedRamInfo(allocator: std.mem.Allocator) !Result {
return try getFormattedRamInfo(allocator, "Ram", ascii.Yellow); 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); 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); 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); 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| { 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_size, 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 { } 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); 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("/"); 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); 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); const terminal_name = try detection.user.getTerminalName(allocator);
defer allocator.free(terminal_name); 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); 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 formatted_net_info_list = std.ArrayList([]u8).init(allocator);
var net_info_list = try detection.network.getNetInfo(allocator); var net_info_list = try detection.network.getNetInfo(allocator);
@@ -178,5 +198,5 @@ pub fn getFormattedNetInfo(allocator: std.mem.Allocator, key: []const u8, key_co
} }
net_info_list.deinit(); net_info_list.deinit();
return formatted_net_info_list; return Result{ .string_arraylist = formatted_net_info_list };
} }

View File

@@ -1,6 +1,7 @@
const std = @import("std"); const std = @import("std");
const c_unistd = @cImport(@cInclude("unistd.h")); const c_unistd = @cImport(@cInclude("unistd.h"));
const c_statvfs = @cImport(@cInclude("sys/statvfs.h")); const c_statvfs = @cImport(@cInclude("sys/statvfs.h"));
const c_libpci = @cImport(@cInclude("pci/pci.h"));
/// Struct representing CPU informations /// Struct representing CPU informations
pub const CpuInfo = struct { pub const CpuInfo = struct {
@@ -9,6 +10,13 @@ pub const CpuInfo = struct {
cpu_max_freq: f32, 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 /// Struct representing RAM usage informations
pub const RamInfo = struct { pub const RamInfo = struct {
ram_size: f64, ram_size: f64,
@@ -76,6 +84,55 @@ 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,
);
try gpu_info_list.append(GpuInfo{
.gpu_name = try allocator.dupe(u8, std.mem.span(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;
}
pub fn getRamInfo(allocator: std.mem.Allocator) !RamInfo { pub fn getRamInfo(allocator: std.mem.Allocator) !RamInfo {
// Reads /proc/meminfo // Reads /proc/meminfo
const meminfo_path = "/proc/meminfo"; const meminfo_path = "/proc/meminfo";

View File

@@ -43,8 +43,15 @@ pub fn main() !void {
try sys_info_list.append(separtor_buffer); try sys_info_list.append(separtor_buffer);
if (modules_types.items.len == 0) { if (modules_types.items.len == 0) {
inline for (0..formatters.formatters.len) |i| { inline for (0..formatters.default_formatters.len) |i| {
try sys_info_list.append(try formatters.default_formatters[i](allocator)); 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| { } else if (conf) |c| {
for (modules_types.items, c.value.modules) |module_type, module| { 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 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 }); 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) { const result = try formatters.formatters[@intFromEnum(module_type)](allocator, module.key, key_color);
var formatted_net_info_list = try formatters.getFormattedNetInfo(allocator, module.key, key_color); switch (result) {
defer formatted_net_info_list.deinit(); .string => |r| try sys_info_list.append(r),
.string_arraylist => |r| {
try sys_info_list.appendSlice(formatted_net_info_list.items); defer r.deinit();
try sys_info_list.appendSlice(r.items);
continue; },
} }
try sys_info_list.append(try formatters.formatters[@intFromEnum(module_type)](allocator, module.key, key_color));
} }
} }