|
|
@@ -1,6 +1,7 @@ |
| 1 |
1 |
#include <stdio.h> |
| 2 |
2 |
#include <sys/utsname.h> |
| 3 |
3 |
#include <string.h> |
|
4 |
+#include <unistd.h> |
| 4 |
5 |
#include <stdlib.h> |
| 5 |
6 |
|
| 6 |
7 |
#define C_RESET "\033[0m" |
|
|
@@ -12,23 +13,27 @@ |
| 12 |
13 |
#define C_RED "\033[1;31m" |
| 13 |
14 |
#define C_PURPLE "\033[1;35m" |
| 14 |
15 |
|
| 15 |
|
-static char *version = "0.1.3"; |
|
16 |
+#define ENABLE_ASCII 1 |
| 16 |
17 |
|
| 17 |
|
-static const char asciiArt[] = |
| 18 |
|
-" ████████ \n" |
| 19 |
|
-" ██████████ \n" |
| 20 |
|
-" █████████ \n" |
| 21 |
|
-" ███████ \n" |
| 22 |
|
-" ██████ ██ ██ ██ ██ ██ ██ ██ \n" |
| 23 |
|
-" ███████ \n" |
| 24 |
|
-" █████████ \n" |
| 25 |
|
-" ██████████ \n" |
| 26 |
|
-" ████████ "; |
|
18 |
+const char *version = "0.1.5"; |
| 27 |
19 |
|
| 28 |
20 |
void printAscii(void) { |
|
21 |
+#if ENABLE_ASCII |
|
22 |
+ static const char asciiArt[] = |
|
23 |
+ " ████████ \n" |
|
24 |
+ " ██████████ \n" |
|
25 |
+ " █████████ \n" |
|
26 |
+ " ███████ \n" |
|
27 |
+ " ██████ ██ ██ ██ ██ ██ ██ ██ \n" |
|
28 |
+ " ███████ \n" |
|
29 |
+ " █████████ \n" |
|
30 |
+ " ██████████ \n" |
|
31 |
+ " ████████ "; |
| 29 |
32 |
printf(C_YELLOW "%s" C_RESET "\n\n", asciiArt); |
|
33 |
+#endif |
| 30 |
34 |
} |
| 31 |
35 |
void printOS(void) { |
|
36 |
+#ifdef __linux__ |
| 32 |
37 |
FILE *fp = fopen("/etc/os-release", "r"); |
| 33 |
38 |
if (!fp) { |
| 34 |
39 |
printf(C_CYAN "OS: " C_RESET "Generic Linux\n"); |
|
|
@@ -49,6 +54,14 @@ void printOS(void) { |
| 49 |
54 |
} |
| 50 |
55 |
fclose(fp); |
| 51 |
56 |
printf(C_CYAN "OS: " C_RESET "Generic Linux\n"); |
|
57 |
+#else |
|
58 |
+ struct utsname otherOS; |
|
59 |
+ if (uname(&otherOS) != 0) { |
|
60 |
+ perror("uname"); |
|
61 |
+ return; |
|
62 |
+ } |
|
63 |
+ printf(C_CYAN "OS: " C_RESET "%s\n", otherOS.sysname); |
|
64 |
+#endif |
| 52 |
65 |
} |
| 53 |
66 |
int printStats(void) { |
| 54 |
67 |
char *term = getenv("TERM_PROGRAM"); |
|
|
@@ -59,22 +72,31 @@ int printStats(void) { |
| 59 |
72 |
if (!wm) { |
| 60 |
73 |
wm = getenv("DESKTOP_SESSION"); |
| 61 |
74 |
} |
|
75 |
+ char *session = getenv("XDG_SESSION_TYPE"); |
| 62 |
76 |
struct utsname info; |
| 63 |
77 |
if (uname(&info) != 0) { |
| 64 |
78 |
perror("uname"); |
| 65 |
79 |
return 1; |
| 66 |
80 |
} |
| 67 |
81 |
printOS(); |
| 68 |
|
- printf(C_GREEN "Kernel: " C_RESET "%s\n", info.release); |
|
82 |
+ printf(C_GREEN "Kernel: " C_RESET "%s %s\n",info.sysname, info.release); |
| 69 |
83 |
printf(C_BLUE "Host: " C_RESET "%s\n", info.nodename); |
| 70 |
|
- printf(C_RED "WM: " C_RESET "%s\n", wm ? wm : "Unknown / TTY"); |
| 71 |
|
- printf(C_PURPLE "Term: " C_RESET "%s\n", term ? term : "couldn't read terminal name"); |
|
84 |
+ printf(C_RED "WM: " C_RESET "%s (%s)\n", wm ? wm : "unknown / tty", session ? session : "unknown"); |
|
85 |
+ printf(C_PURPLE "Term: " C_RESET "%s\n", term ? term : "unknown"); |
| 72 |
86 |
return 0; |
| 73 |
87 |
} |
| 74 |
88 |
int main(int argc, char **argv) { |
| 75 |
|
- if (argc > 1 && strcmp("-v", argv[1]) == 0) { |
| 76 |
|
- printf("sfetch-%s\n", version); |
| 77 |
|
- return 0; |
|
89 |
+ for (int i = 1; i < argc; i++) { |
|
90 |
+ if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--help") == 0) { |
|
91 |
+ printf("usage: %s [-v/-h]\n", argv[0]); |
|
92 |
+ return 0; |
|
93 |
+ } |
|
94 |
+ if (strcmp(argv[i], "-v") == 0 || strcmp(argv[i], "--version") == 0) { |
|
95 |
+ printf("sfetch-%s\n", version); |
|
96 |
+ return 0; |
|
97 |
+ } |
|
98 |
+ fprintf(stderr, "Unknown argument: %s\nUsage: %s [-v/-h]\n", argv[i], argv[0]); |
|
99 |
+ return 1; |
| 78 |
100 |
} |
| 79 |
101 |
printAscii(); |
| 80 |
102 |
printStats(); |