From 128f482a9b9120566869da3281ef9009f679ef5c Mon Sep 17 00:00:00 2001
From: utox39 <francescomoccaldi39@gmail.com>
Date: Thu, 6 Mar 2025 17:51:22 +0100
Subject: [PATCH] feat(linux): add cpu info

---
 src/linux/linux.zig | 52 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/src/linux/linux.zig b/src/linux/linux.zig
index 50bc69b..71f50cf 100644
--- a/src/linux/linux.zig
+++ b/src/linux/linux.zig
@@ -1,5 +1,6 @@
 const std = @import("std");
 const c_sysinfo = @cImport(@cInclude("sys/sysinfo.h"));
+const c_unistd = @cImport(@cInclude("unistd.h"));
 
 /// Structure representing system uptime in days, hours, and minutes.
 pub const SystemUptime = struct {
@@ -8,6 +9,12 @@ pub const SystemUptime = struct {
     minutes: i8,
 };
 
+pub const CpuInfo = struct {
+    cpu_name: []u8,
+    cpu_cores: i32,
+    cpu_max_freq: f32,
+};
+
 pub fn getUsername(allocator: std.mem.Allocator) ![]u8 {
     const username = try std.process.getEnvVarOwned(allocator, "USER");
     return username;
@@ -70,3 +77,48 @@ pub fn getShell(allocator: std.mem.Allocator) ![]u8 {
 
     return output;
 }
+
+pub fn getCpuInfo(allocator: std.mem.Allocator) !CpuInfo {
+    const cpu_cores = c_unistd.sysconf(c_unistd._SC_NPROCESSORS_ONLN);
+
+    // Reads /proc/cpuinfo
+    const cpuinfo_path = "/proc/cpuinfo";
+    var file = try std.fs.cwd().openFile(cpuinfo_path, .{});
+    defer file.close();
+    const cpuinfo_data = try file.readToEndAlloc(allocator, std.math.maxInt(usize));
+    defer allocator.free(cpuinfo_data);
+
+    // Parsing /proc/cpuinfo
+    var model_name: ?[]const u8 = null;
+
+    var lines = std.mem.split(u8, cpuinfo_data, "\n");
+    while (lines.next()) |line| {
+        const trimmed = std.mem.trim(u8, line, " \t");
+        if (std.mem.startsWith(u8, trimmed, "model name") and model_name == null) {
+            var parts = std.mem.split(u8, trimmed, ":");
+            _ = parts.next(); // discards the key
+            if (parts.next()) |value| {
+                model_name = std.mem.trim(u8, value, " ");
+                break;
+            }
+        }
+    }
+
+    // Reads /sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq
+    const cpuinfo_max_freq_path = "/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);
+
+    // 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;
+
+    return CpuInfo{
+        .cpu_name = try allocator.dupe(u8, model_name orelse "Unknown"),
+        .cpu_cores = @as(i32, @intCast(cpu_cores)),
+        .cpu_max_freq = cpu_max_freq,
+    };
+}