exhq/neowofetch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
neowofetch/uwufetch.c
zegolem 208c40c19e gpu: forced LANG environement variable to en_US before running lshw 
Before, as my system is configured with LANG=fr_FR
so, lshw would output in french, and the program
couldn't find the right strings in the output,
and so, my GPU wasn't properly detected.

There may be other instances of similar issues
in the code, but for now, everything shows up
correctly for me.
2021-03-19 17:44:35 +01:00

472 lines
17 KiB
C
Raw Blame History

/*
* UwUfetch is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#include <dirent.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/sysinfo.h>
#include <sys/utsname.h>
// COLORS
#define NORMAL "\x1b[0m"
#define BOLD "\x1b[1m"
#define BLACK "\x1b[30m"
#define RED "\x1b[31m"
#define GREEN "\x1b[32m"
#define YELLOW "\x1b[33m"
#define BLUE "\x1b[34m"
#define MAGENTA "\x1b[0;35m"
#define CYAN "\x1b[36m"
#define WHITE "\x1b[37m"
#define PINK "\x1b[38;5;201m"
#define LPINK "\x1b[38;5;213m"
struct utsname sys_var;
struct sysinfo sys;
// initialise the variables to store data, gpu array can hold up to 8 gpus
int ram_max = 0, ram_free = 0, pkgs, a_i_flag = 0;
char user[32], host[256], shell[64], version_name[64], cpu_model[256], gpu_model[8][256] = {{'0'},{'0'},{'0'},{'0'},{'0'},{'0'},{'0'},{'0'}}, pkgman_name[64], image_name[32];
int pkgman();
void get_info();
void list();
void print_ascii();
void print_info();
void print_image();
void usage(char*);
void uwu_name();
int main(int argc, char *argv[]) {
int opt = 0;
get_info();
while((opt = getopt(argc, argv, "ad:hilc:")) != -1) {
switch(opt) {
case 'a':
a_i_flag = 0;
break;
case 'c':
a_i_flag = 1;
sprintf(image_name, "%s", optarg);
break;
case 'd':
if (optarg) sprintf(version_name, "%s", optarg);
break;
case 'h':
usage(argv[0]);
return 0;
case 'i':
a_i_flag = 1;
break;
case 'l':
list(argv[0]);
return 0;
default:
break;
}
}
if (argc == 1 || a_i_flag == 0) print_ascii();
else if (a_i_flag) print_image();
uwu_name();
print_info();
}
int pkgman() { // this is just a function that returns the total of installed packages
int total = 0;
// TODO: should this be at the top of the program? maybe in a config.c file?
// TODO: do we need to `free()` this? I have no idea how to do memory management in C...
struct package_manager {
char command_string[128]; // command to get number of packages installed
char pkgman_name[16]; // name of the package manager
};
struct package_manager pkgmans[] = {
{ "apk info 2> /dev/null | wc -l", "(apk)" },
{ "dnf list installed 2> /dev/null | wc -l", "(dnf)" },
{ "qlist -I 2> /dev/null | wc -l", "(emerge)" },
{ "flatpak list 2> /dev/null | wc -l", "(flatpack)" },
{ "guix package --list-installed 2> /dev/null | wc -l", "(guix)" },
{ "nix-store -q --requisites /run/current-sys_vartem/sw 2> /dev/null | wc -l", "(nix)" },
{ "pacman -Qq 2> /dev/null | wc -l", "(pacman)" },
{ "rpm -qa --last 2> /dev/null | wc -l", "(rpm)" },
{ "xbps-query -l 2> /dev/null | wc -l", "(xbps)" },
};
const unsigned long pkgman_count = sizeof(pkgmans) / sizeof(pkgmans[0]);
for (int i = 0; i < pkgman_count; i++) {
struct package_manager *current = &pkgmans[i];
FILE *fp = popen(current->command_string, "r");
unsigned int pkg_count;
if (fscanf(fp, "%u", &pkg_count) == 3) continue;
fclose(fp);
total += pkg_count;
if (pkg_count > 0) strcat(pkgman_name, current->pkgman_name);
}
return total;
}
void print_info() {
// print collected info - from host to cpu info
printf( "\033[9A\033[18C%s%s%s@%s\n"
"\033[18C%s%sOWOS %s%s\n"
"\033[18C%s%sKEWNEL %s%s %s\n"
"\033[18C%s%sCPUWU %s%s\n",
NORMAL, BOLD, user, host,
NORMAL, BOLD, NORMAL, version_name,
NORMAL, BOLD, NORMAL, sys_var.release, sys_var.machine,
NORMAL, BOLD, NORMAL, cpu_model);
// print the gpus
int gpu_iter = 0;
while(gpu_model[gpu_iter][0] != '0') {
printf( "\033[18C%s%sGPUWU %s%s\n",
NORMAL, BOLD, NORMAL, gpu_model[gpu_iter]);
gpu_iter++;
}
// print ram to uptime and colors
printf( "\033[18C%s%sWAM %s%i MB/%i MB\n"
"\033[18C%s%sSHELL %s%s\n"
"\033[18C%s%sPKGS %s%s%d %s\n"
"\033[18C%s%sUWUPTIME %s"/*"%lid, "*/"%lih, %lim\n"
"\033[18C%s%s\u2587\u2587%s\u2587\u2587%s\u2587\u2587%s\u2587\u2587%s\u2587\u2587%s\u2587\u2587%s\u2587\u2587%s\u2587\u2587%s\n",
NORMAL, BOLD, NORMAL, (ram_max - ram_free), ram_max,
NORMAL, BOLD, NORMAL, shell,
NORMAL, BOLD, NORMAL, NORMAL, pkgs, pkgman_name,
NORMAL, BOLD, NORMAL, /*sys.uptime/60/60/24,*/ sys.uptime/60/60, sys.uptime/60%60,
BOLD, BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE, NORMAL);
}
void get_info() { // get all necessary info
char line[256]; // var to scan file lines
// os version
FILE *os_release = fopen("/etc/os-release", "r");
FILE *cpuinfo = fopen("/proc/cpuinfo", "r");
if (os_release) { // get normal vars
while (fgets(line, sizeof(line), os_release)) if (sscanf(line, "\nID=%s", version_name)) break;
while (fgets(line, sizeof(line), cpuinfo)) if (sscanf(line, "model name : %[^\n]", cpu_model)) break;
sprintf(user, "%s", getenv("USER"));
fclose(os_release);
} else { // try for android vars, or unknown system
DIR *system_app = opendir("/system/app/");
DIR *system_priv_app = opendir("/system/priv-app/");
if (system_app && system_priv_app) { // android
closedir(system_app);
closedir(system_priv_app);
sprintf(version_name, "android");
// android vars
FILE *whoami = popen("whoami", "r");
if (fscanf(whoami, "%s", user) == 3) sprintf(user, "unknown");
fclose(whoami);
while (fgets(line, sizeof(line), cpuinfo)) if (sscanf(line, "Hardware : %[^\n]", cpu_model)) break;
} else sprintf(version_name, "unknown");
}
fclose(cpuinfo);
gethostname(host, 256);
sscanf(getenv("SHELL"), "%s", shell);
if (strlen(shell) > 16) memmove(&shell, &shell[27], strlen(shell)); // android shell was too long, this works only for termux
// system resources
uname(&sys_var);
sysinfo(&sys);
// ram
FILE *meminfo = fopen("/proc/meminfo", "r");
while (fgets(line, sizeof(line), meminfo)) {
sscanf(line, "MemFree: %d kB", &ram_free);
sscanf(line, "MemTotal: %d kB", &ram_max);
if (ram_max > 0 && ram_free > 0) {
ram_max *= 0.001024;
ram_free *= 0.001024;
break;
}
}
fclose(meminfo);
// gpu
int gpun = 0; // number of the gpu that the program is searching for to put in the array
setenv("LANG", "en_US", 1); // force language to english
FILE *gpu;
gpu = popen("lshw -class display 2> /dev/null", "r");
// add all gpus to the array gpu_model (up to 8 gpus)
while (fgets(line, sizeof(line), gpu)) if (sscanf(line, " product: %[^\n]", gpu_model[gpun])) gpun++;
if (strlen(gpu_model[0]) < 1) {
if (strcmp(version_name, "android") != 0) gpu = popen("lspci -mm 2> /dev/null | grep \"VGA\\|00:02\" | cut --fields=4,6 -d '\"' --output-delimiter=\" \" | sed \"s/ Controller.*//\"", "r");
else gpu = popen("getprop ro.hardware.vulkan 2> /dev/null", "r");
while (fgets(line, sizeof(line), gpu)) if (sscanf(line, "%[^\n]", gpu_model[0])) break;
}
fclose(gpu);
// format the strings a bit
for(int i = 0; i < gpun; i++) {
for (int j = 42; j < 256; j++) { //max gpu_name length
gpu_model[i][j] = '\0';
}
}
pkgs = pkgman();
}
void list(char* arg) { // prints distribution list
/* distributions are listed by distribution branch
to make the output easier to understand by the user.*/
printf( "%s -d <options>\n"
" Available distributions:\n"
" %sArch linux %sbased:\n"
" %sarch, artix, %smanjaro, \"manjaro-arm\"\n\n"
" %sDebian/%sUbuntu %sbased:\n"
" %sdebian, %slinuxmint, %spop, %sraspbian\n\n"
" %sOther/spare distributions:\n"
" %salpine, %sfedora, %sgentoo, %s\"void\", android, %sunknown\n\n"
" %sBSD:\n"
" freebsd, %sopenbsd\n",
arg, BLUE, NORMAL, BLUE, GREEN, // Arch based colors
RED, YELLOW, NORMAL, RED, GREEN, BLUE, RED, // Debian based colors
NORMAL, BLUE, BLUE, PINK, GREEN, WHITE, // Other/spare distributions colors
RED, YELLOW); // BSD colors
}
void print_ascii() { // prints logo (as ascii art) of the given system. distributions listed alphabetically.
// linux
if (strcmp(version_name, "alpine") == 0) {
printf("\033[2E\033[4C%s. .___.\n"
" / \\/ \\ /\n"
" /OwO\\ɛU\\/ __\n"
" / \\ \\__/ \\\n"
"/ \\ \\\n\n\n", BLUE);
} else if (strcmp(version_name, "arch") == 0) {
printf( "\033[1E\033[8C%s/\\\n"
" / \\\n"
" /\\ \\\n"
" / > w <\\\n"
" / __ \\\n"
" / __| |__-\\\n"
" /_-'' ''-_\\\n\n", BLUE);
} else if (strcmp(version_name, "artix") == 0) {
printf( "\033[1E\033[8C%s/\\\n"
" / \\\n"
" /`'.,\\\n"
" /\u2022 w \u2022 \\\n"
" / ,`\\\n"
" / ,.'`. \\\n"
" /.,'` `'.\\\n\n", BLUE);
} else if (strcmp(version_name, "debian") == 0) {
printf( "\033[1E\033[6C%s______\n"
" / ___ \\\n"
" | / OwO |\n"
" | \\____-\n"
" -_\n"
" --_\n\n\n", RED);
} else if (strcmp(version_name, "fedora") == 0) {
printf( "\033[1E\033[8C%s_____\n"
" / __)%s\\\n"
" %s> %s| / %s<%s\\ \\\n"
" __%s_| %sw%s|_%s_/ /\n"
" / %s(_ _)%s_/\n"
" / / %s| |\n"
" %s\\ \\%s__/ |\n"
" %s\\%s(_____/\n", BLUE, CYAN, WHITE, BLUE, WHITE, CYAN, BLUE, CYAN, BLUE, CYAN, BLUE, CYAN, BLUE, CYAN, BLUE, CYAN, BLUE);
} else if (strcmp(version_name, "gentoo") == 0) {
printf( "\033[1E\033[3C%s_-----_\n"
" ( \\\n"
" \\ OwO \\\n"
"%s \\ )\n"
" / _/\n"
" ( _-\n"
" \\____-\n\n", MAGENTA, WHITE);
} else if (strcmp(version_name, "manjaro") == 0) {
printf( "\033[0E\033[1C\u25b3 \u25b3 \u25e0\u25e0\u25e0\u25e0\n"
" \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m\e[0;42m\e[1;30m > w < \e[0m\e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n");
} else if (strcmp(version_name, "\"manjaro-arm\"") == 0) {
printf( "\033[0E\033[1C\u25b3 \u25b3 \u25e0\u25e0\u25e0\u25e0\n"
" \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m\e[0;42m\e[1;30m > w < \e[0m\e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n"
" \e[0;42m \e[0m \e[0;42m \e[0m \e[0;42m \e[0m\n");
} else if (strcmp(version_name, "linuxmint") == 0) {
printf( "\033[2E\033[4C%s__/\\____/\\.\n"
" |%s.--. %s|\n"
" %s, %s¯| %s| UwU| %s|\n"
" %s|| %s| %s| | %s|\n"
" %s | %s| %s---- %s|\n"
" %s --%s'--------'\n\n",GREEN, WHITE, GREEN, WHITE, GREEN, WHITE, GREEN, WHITE, GREEN, WHITE, GREEN, WHITE, GREEN, WHITE, GREEN, WHITE, GREEN);
} else if (strcmp(version_name, "\"opensuse-leap\"") == 0) {
printf("\033[2E\033[6C%s_______\n"
" ___| %s__%s \\\n"
" / %s/ o\\%s|\n"
" %s\\__/%s|\n"
" ______/\n"
" \\_______\n"
" \\_________/\n\n", GREEN, NORMAL, GREEN, NORMAL, GREEN, NORMAL, GREEN);
} else if (strcmp(version_name, "pop") == 0) {
printf("\033[2E\033[6C%s|\\.-----./|\n"
" |/ \\|\n"
" | > < |\n"
" | %s~ %sP! %s~ %s|\n"
"_ ---\\ w /\n"
" \\_/ '-----'\n\n", BLUE, LPINK, WHITE, LPINK, BLUE);
} else if (strcmp(version_name, "raspbian") == 0) {
printf("\033[0E\033[6C%s__ __\n"
" (_\\)(/_)\n"
" %s(>(__)<)\n"
" (_(_)(_)_)\n"
" (_(__)_)\n"
" (__)\n\n\n", GREEN, RED);
} else if (strcmp(version_name, "ubuntu") == 0) {
printf( "\033[1E\033[9C%s_\n"
" %s\u25E3%s__(_)%s\u25E2%s\n"
" _/ --- \\\n"
" (_) |>w<| |\n"
" \\ --- _/\n"
" %sC__/%s---(_)\n\n\n", LPINK, PINK, LPINK, PINK, LPINK, PINK, LPINK);
} else if (strcmp(version_name, "\"void\"") == 0){
printf("\033[2E\033[2C%s |\\_____/|\n"
" _\\____ |\n"
" | \\ \\ |\n"
" | | %s\u00d2w\u00d3 %s| | ,\n"
" | \\_____\\_|-, |\n"
" -_______\\ \\_/\n\n", GREEN, WHITE, GREEN);
} else if (strcmp(version_name, "android") == 0) { // android at the end because it could be not considered as an actual distribution of gnu/linux
printf( "\033[2E\033[3C%s\\ _------_ /\n"
" / \\\n"
" | %s~ %s> w < %s~ %s|\n"
" ------------\n\n\n\n", GREEN, RED, GREEN, RED, GREEN);
}
// BSD
else if (strcmp(version_name, "freebsd") == 0) {
printf( "\033[2E\033[1C%s/\\,-'''''-,/\\\n"
" \\_) (_/\n"
" | \\ / |\n"
" | O w O |\n"
" ; ;\n"
" '-_____-'\n\n", RED);
} else if (strcmp(version_name, "openbsd") == 0) {
printf( "\033[1E\033[3C%s ______ \n"
" \\- -/ %s\u2665 \n"
"%s\\_/ \\ \n"
"| %s> < %s| \n"
"|_ < %s// %sW %s// \n"
"%s/ \\ / \n"
" /-________-\\ \n\n", YELLOW, RED, YELLOW, WHITE, YELLOW, LPINK, WHITE, LPINK, YELLOW);
}
else printf( "\033[0E\033[2C%s._.--._.\n"
" \\|>%s_%s< |/\n"
" |%s:_/%s |\n"
" // \\ \\ ?\n"
" (| | ) /\n"
" %s/'\\_ _/`\\%s-\n"
" %s\\___)=(___/\n\n", WHITE, YELLOW, WHITE, YELLOW, WHITE, YELLOW, WHITE, YELLOW);
}
void print_image() { // prints logo (as an image) of the given system. distributions listed alphabetically.
char command[256];
if (strlen(image_name) > 1) sprintf(command, "viu -t -w 18 -h 8 %s 2> /dev/null", image_name);
else {
if (strcmp(version_name, "android") == 0) sprintf(command, "viu -t -w 18 -h 8 /data/data/com.termux/files/usr/lib/uwufetch/%s.png 2> /dev/null", version_name);
else sprintf(command, "viu -t -w 18 -h 8 /usr/lib/uwufetch/%s.png 2> /dev/null", version_name);
}
if (system(command) != 0) { // if viu is not installed or the image is missing
printf( "\033[0E\033[3C%s\n"
" There was an\n"
" error: viu\n"
" is not installed\n"
" or the image\n"
" is not fount\n"
" Read IMAGES.md\n"
" for more info.\n\n", RED);
}
}
void usage(char* arg) {
printf("Usage: %s <args>\n"
" -a, --ascii prints logo as ascii text (default)\n"
" -c, --custom choose a custom image\n"
" -d, --distro lets you choose the logo to print\n"
" -h, --help prints this help page\n"
" -i, --image prints logo as image\n"
" %sworks in most terminals\n"
" read res/IMAGES.md for more info%s\n"
" -l lists all supported distributions\n"
" %sNote that the long options (--opt) are currently not working\n"
" If you are reading this and you know how to implement long opts,\n"
" send a pull request on https://github.com/TheDarkBug/uwufetch%s\n",
arg, BLUE, NORMAL, RED, NORMAL);
}
void uwu_name() { // changes distro name to uwufied(?) name
#define STRING_TO_UWU(original, uwufied) if (strcmp(version_name, original) == 0) sprintf(version_name, "%s", uwufied)
// linux
STRING_TO_UWU("alpine", "Nyalpine");
else STRING_TO_UWU("arch", "Nyarch Linuwu");
else STRING_TO_UWU("artix", "Nyartix Linuwu");
else STRING_TO_UWU("debian", "Debinyan");
else STRING_TO_UWU("endeavour", "endeavOwO");
else STRING_TO_UWU("fedora", "Fedowa");
else STRING_TO_UWU("gentoo", "GentOwO");
else STRING_TO_UWU("guix", "gnUwU gUwUix");
else STRING_TO_UWU("linuxmint", "LinUWU Miwint");
else STRING_TO_UWU("manjaro", "Myanjawo");
else STRING_TO_UWU("\"manjaro-arm\"", "Myanjawo AWM");
else STRING_TO_UWU("neon", "KDE NeOwOn");
else STRING_TO_UWU("nixos", "nixOwOs");
else STRING_TO_UWU("\"opensuse-leap\"", "OwOpenSUSE");
else STRING_TO_UWU("pop", "PopOwOS");
else STRING_TO_UWU("raspbian", "RaspNyan");
else STRING_TO_UWU("slackware", "Swackwawe");
else STRING_TO_UWU("solus", "sOwOlus");
else STRING_TO_UWU("ubuntu", "Uwuntu");
else STRING_TO_UWU("\"void\"", "OwOid");
else STRING_TO_UWU("android", "Nyandroid"); // android at the end because it could be not considered as an actual distribution of gnu/linux
// BSD
else STRING_TO_UWU("freebsd", "FweeBSD");
else STRING_TO_UWU("openbsd", "OwOpenBSD");
else {
sprintf(version_name, "%s", "unknown");
if (a_i_flag == 1) {
print_image();
printf("\n");
}
}
#undef STRING_TO_UWU
}
Reference in a new issue View git blame Copy permalink