Merge pull request #247 from CarterLi/master

[macOS] support cpuUsage and other improvements

Author: LinusDierheimer

CMakeLists.txt

@@ -256,6 +256,7 @@ endif()
 
 if(LINUX OR BSD)
     list(APPEND LIBFASTFETCH_SRC
+        src/detection/cpuUsage/cpuUsage_linux.c
         src/detection/host/host_linux.c
         src/detection/os/os_linux.c
         src/detection/gpu/gpu_linux.c
@@ -272,6 +273,7 @@ endif()
 
 if(APPLE)
     list(APPEND LIBFASTFETCH_SRC
+        src/detection/cpuUsage/cpuUsage_apple.c
         src/util/apple/cf_helpers.c
         src/util/apple/osascript.m
         src/detection/host/host_apple.c
@@ -295,6 +297,7 @@ endif()
 
 if(ANDROID)
     list(APPEND LIBFASTFETCH_SRC
+        src/detection/cpuUsage/cpuUsage_linux.c
         src/detection/host/host_android.c
         src/detection/os/os_android.c
         src/detection/gpu/gpu_android.c

README.md

@@ -71,7 +71,7 @@ Pacman, dpkg, rpm, emerge, xbps, nix, Flatpak, Snap, apk, pkg, brew, MacPorts
 
 ##### WM themes
 ```
-KWin, Mutter, Muffin, Marco, XFWM, Openbox (LXDE, LXQT & without DE)
+KWin, Mutter, Muffin, Marco, XFWM, Openbox (LXDE, LXQT & without DE), Quartz Compositor (macOS)
 ```
 
 ##### DE versions
@@ -81,7 +81,7 @@ KDE Plasma, Gnome, Cinnamon, Mate, XFCE4, LXQt
 
 ##### Terminal fonts
 ```
-Konsole, Gnome Terminal, Tilix, XFCE4 Terminal, Alacritty, LXTerminal, ITerm2, Apple Terminal, TTY
+Konsole, Gnome Terminal, Tilix, XFCE4 Terminal, Alacritty, LXTerminal, iTerm2, Apple Terminal, TTY
 ```
 
 ## Building

src/common/processing.c

@@ -6,16 +6,16 @@
 #include <unistd.h>
 #include <sys/wait.h>
 
-void ffProcessAppendStdOut(FFstrbuf* buffer, char* const argv[])
+const char* ffProcessAppendStdOut(FFstrbuf* buffer, char* const argv[])
 {
     int pipes[2];
 
     if(pipe(pipes) == -1)
-        return;
+        return "pipe() failed";
 
     pid_t childPid = fork();
     if(childPid == -1)
-        return;
+        return "fork() failed";
 
     //Child
     if(childPid == 0)
@@ -33,4 +33,6 @@ void ffProcessAppendStdOut(FFstrbuf* buffer, char* const argv[])
     waitpid(childPid, NULL, 0);
     ffAppendFDBuffer(pipes[0], buffer);
     close(pipes[0]);
+
+    return NULL;
 }

src/common/processing.h

@@ -5,6 +5,6 @@
 
 #include "util/FFstrbuf.h"
 
-void ffProcessAppendStdOut(FFstrbuf* buffer, char* const argv[]);
+const char* ffProcessAppendStdOut(FFstrbuf* buffer, char* const argv[]);
 
 #endif

src/detection/cpuUsage/cpuUsage.h

@@ -0,0 +1,8 @@
+#pragma once
+
+#ifndef FF_INCLUDED_detection_cpu_cpuUsage
+#define FF_INCLUDED_detection_cpu_cpuUsage
+
+const char* ffGetCpuUsagePercent(double* result);
+
+#endif

src/detection/cpuUsage/cpuUsage_apple.c

@@ -0,0 +1,46 @@
+#include "fastfetch.h"
+#include "cpuUsage.h"
+
+#include <mach/processor_info.h>
+#include <mach/mach_host.h>
+#include <unistd.h>
+
+static const char* getCpuUsageInfo(long* inUseAll, long* totalAll)
+{
+    natural_t numCPUs = 0U;
+    processor_info_array_t cpuInfo;
+    mach_msg_type_number_t numCpuInfo;
+
+    if (host_processor_info(mach_host_self(), PROCESSOR_CPU_LOAD_INFO, &numCPUs, &cpuInfo, &numCpuInfo) != KERN_SUCCESS)
+        return "host_processor_info() failed";
+    if (numCPUs * CPU_STATE_MAX != numCpuInfo)
+        return "Unexpected host_processor_info() result";
+
+    for (natural_t i = 0U; i < numCPUs; ++i) {
+        integer_t inUse = cpuInfo[CPU_STATE_MAX * i + CPU_STATE_USER]
+            + cpuInfo[CPU_STATE_MAX * i + CPU_STATE_SYSTEM]
+            + cpuInfo[CPU_STATE_MAX * i + CPU_STATE_NICE];
+        integer_t total = inUse + cpuInfo[CPU_STATE_MAX * i + CPU_STATE_IDLE];
+        *inUseAll += inUse;
+        *totalAll += total;
+    }
+    return NULL;
+}
+
+const char* ffGetCpuUsagePercent(double* result)
+{
+    long inUseAll1 = 0, totalAll1 = 0;
+    const char* error = getCpuUsageInfo(&inUseAll1, &totalAll1);
+    if(error)
+        return error;
+
+    sleep(1);
+
+    long inUseAll2 = 0, totalAll2 = 0;
+    error = getCpuUsageInfo(&inUseAll2, &totalAll2);
+    if(error)
+        return error;
+
+    *result = (double)(inUseAll2 - inUseAll1) / (double)(totalAll2 - totalAll1) * 100;
+    return NULL;
+}

src/detection/cpuUsage/cpuUsage_linux.c

@@ -0,0 +1,46 @@
+#include "fastfetch.h"
+#include "cpuUsage.h"
+
+#include <unistd.h>
+#include <stdio.h>
+
+static const char* getCpuUsageInfo(FILE* procStat, long* inUseAll, long* totalAll)
+{
+    long user, nice, system, idle, iowait, irq, softirq;
+
+    if (fscanf(procStat, "cpu%ld%ld%ld%ld%ld%ld%ld", &user, &nice, &system, &idle, &iowait, &irq, &softirq) < 0)
+    {
+        fclose(procStat);
+        return "fscanf() failed";
+    }
+    *inUseAll = user + nice + system;
+    *totalAll = *inUseAll + idle + iowait + irq + softirq;
+    return NULL;
+}
+
+const char* ffGetCpuUsagePercent(double* result)
+{
+    FILE* procStat = fopen("/proc/stat", "r");
+    if(procStat == NULL)
+        return "fopen(\"""/proc/stat\", \"r\") == NULL";
+
+    const char* error = NULL;
+    long inUseAll1 = 0, totalAll1 = 0;
+    error = getCpuUsageInfo(procStat, &inUseAll1, &totalAll1);
+    if(error)
+        goto exit;
+
+    sleep(1);
+    rewind(procStat);
+
+    long inUseAll2 = 0, totalAll2 = 0;
+    error = getCpuUsageInfo(procStat, &inUseAll2, &totalAll2);
+    if(error)
+        goto exit;
+
+    *result = (double)(inUseAll2 - inUseAll1) / (double)(totalAll2 - totalAll1) * 100;
+
+exit:
+    fclose(procStat);
+    return error;
+}

src/detection/memory/memory.c

@@ -5,7 +5,7 @@ void ffDetectMemoryImpl(FFMemoryResult* memory);
 
 static void calculatePercentage(FFMemoryStorage* storage)
 {
-    if(storage->error.length == 0)
+    if(storage->error.length != 0)
         return;
 
     if(storage->bytesTotal == 0)

src/detection/os/os_apple.c

@@ -3,6 +3,7 @@
 #include "common/sysctl.h"
 
 #include <stdlib.h>
+#include <string.h>
 
 typedef enum PListKey
 {
@@ -12,7 +13,7 @@ typedef enum PListKey
     PLIST_KEY_OTHER
 } PListKey;
 
-static void parseFile(FFOSResult* os)
+static void parseSystemVersion(FFOSResult* os)
 {
     FILE* plist = fopen("/System/Library/CoreServices/SystemVersion.plist", "r");
     if(plist == NULL)
@@ -58,6 +59,33 @@ static void parseFile(FFOSResult* os)
     fclose(plist);
 }
 
+void parseOSXSoftwareLicense(FFOSResult* os)
+{
+    FILE* rtf = fopen("/System/Library/CoreServices/Setup Assistant.app/Contents/Resources/en.lproj/OSXSoftwareLicense.rtf", "r");
+    if(rtf == NULL)
+        return;
+
+    char* line = NULL;
+    size_t len = 0;
+    const char* searchStr = "\\f0\\b SOFTWARE LICENSE AGREEMENT FOR macOS ";
+    const size_t searchLen = strlen(searchStr);
+    while(getline(&line, &len, rtf) != EOF)
+    {
+        if (strncmp(line, searchStr, searchLen) == 0)
+        {
+            ffStrbufAppendS(&os->codename, line + searchLen);
+            ffStrbufTrimRight(&os->codename, '\n');
+            ffStrbufTrimRight(&os->codename, '\\');
+            break;
+        }
+    }
+
+    if(line != NULL)
+        free(line);
+
+    fclose(rtf);
+}
+
 void ffDetectOSImpl(FFOSResult* os, const FFinstance* instance)
 {
     FF_UNUSED(instance);
@@ -76,17 +104,21 @@ void ffDetectOSImpl(FFOSResult* os, const FFinstance* instance)
     ffStrbufInitA(&os->variant, 0);
     ffStrbufInitA(&os->variantID, 0);
 
-    parseFile(os);
+    parseSystemVersion(os);
 
     if(ffStrbufStartsWithIgnCaseS(&os->name, "MacOS"))
         ffStrbufAppendS(&os->id, "macos");
 
     if(os->version.length == 0)
         ffSysctlGetString("kern.osproductversion", &os->version);
 
-    //TODO map version to pretty name
+    if(os->buildID.length == 0)
+        ffSysctlGetString("kern.osversion", &os->buildID);
+
     ffStrbufAppend(&os->prettyName, &os->name);
     ffStrbufAppend(&os->versionID, &os->version);
     ffSysctlGetString("kern.ostype", &os->systemName);
     ffSysctlGetString("hw.machine", &os->architecture);
+
+    parseOSXSoftwareLicense(os);
 }

src/detection/terminalShell.c

@@ -130,6 +130,13 @@ static void getTerminalFromEnv(FFTerminalShellResult* result)
         getenv("WT_PROFILE_ID") != NULL
     )) term = "Windows Terminal";
 
+    //Alacritty
+    if(!ffStrSet(term) && (
+        getenv("ALACRITTY_SOCKET") != NULL ||
+        getenv("ALACRITTY_LOG") != NULL ||
+        getenv("ALACRITTY_WINDOW_ID") != NULL
+    )) term = "Alacritty";
+
     //Termux
     if(!ffStrSet(term) && (
         getenv("TERMUX_VERSION") != NULL ||