Trim last newline character from ascii art

Fixed bug which would cause partitions not to show up if /dev/disk/by-partlabel/ did not exist
Updated fetch_script.sh to use labels if available

.gitignore

@@ -1,4 +0,0 @@
-.idea
-build
-stormfetch
-stormfetch.tar.gz

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 CapCreeperGR
+Copyright (c) 2024 EnumDev
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

Makefile

@@ -1,36 +1,35 @@
-ifeq ($(PREFIX),)
+# Installation paths
-    PREFIX := /usr/local
+PREFIX ?= /usr/local
-endif
+BINDIR ?= $(PREFIX)/bin
-ifeq ($(BINDIR),)
+SYSCONFDIR := $(PREFIX)/etc
-    BINDIR := $(PREFIX)/bin
+
-endif
+# Compilers and tools
-ifeq ($(SYSCONFDIR),)
+GO ?= go
-    SYSCONFDIR := $(PREFIX)/etc
+
-endif
+# Build-time variables
-ifeq ($(GO),)
+VERSION ?= $(shell git describe --tags --dirty)
-    GO := /usr/bin/go
-endif
 
 build:
 	mkdir -p build
-	$(GO) build -ldflags "-w" -o build/stormfetch stormfetch
+	cd src; $(GO) build -ldflags "-w -X 'main.StormfetchVersion=$(VERSION)' -X 'main.SystemConfigDir=$(SYSCONFDIR)'" -o ../build/stormfetch stormfetch
 
 install: build/stormfetch config/
-	mkdir -p $(DESTDIR)$(BINDIR)
+	# Create directoriy
-	mkdir -p $(DESTDIR)$(SYSCONFDIR)/stormfetch/
+	install -dm755 $(DESTDIR)$(BINDIR)
-	cp build/stormfetch $(DESTDIR)$(BINDIR)/stormfetch
+	#  Install binary
-	cp -r config/. $(DESTDIR)$(SYSCONFDIR)/stormfetch/
+	install -m755 build/stormfetch $(DESTDIR)$(BINDIR)/stormfetch
 
-compress: build/stormfetch config/
+install-config:
-	mkdir -p stormfetch/$(BINDIR)
+	# Create directory
-	mkdir -p stormfetch/$(SYSCONFDIR)/stormfetch/
+	install -dm755 $(DESTDIR)$(SYSCONFDIR)
-	cp build/stormfetch stormfetch/$(BINDIR)/stormfetch
+	# Install files
-	cp -r config/. stormfetch/$(SYSCONFDIR)/stormfetch/
+	cp -r config $(DESTDIR)$(SYSCONFDIR)/stormfetch/
-	tar --owner=root --group=root -czf stormfetch.tar.gz stormfetch
-	rm -r stormfetch
 
-run: build/stormfetch
+uninstall:
-	build/stormfetch
+	-rm -f $(DESTDIR)$(BINDIR)/stormfetch
+	-rm -rf $(DESTDIR)$(SYSCONFDIR)/stormfetch
 
 clean:
-	rm -r build/
+	rm -r build/
+
+.PHONY: build install install-config uninstall clean

README.md

@@ -1,9 +1,6 @@
 
 # Stormfetch
-## A simple linux fetch program written in go and bash
+## A simple linux fetch program written in go
-
-### Developers:
-- [CapCreeperGR ](https://gitlab.com/CapCreeperGR)
 
 ### Project Information
 Stormfetch is a program that can read your system's information and display it in the terminal along with the ASCII art of the Linux distribution you are running.
@@ -20,7 +17,7 @@ Stormfetch is still in beta, so distro compatibility is limited. If you would li
 - Download `make` from your package manager
 - Run the following command to compile the project
 ```
-make
+make SYSCONFDIR=/etc
 ```
 - Run the following command to install stormfetch into your system. You may also append a DESTDIR variable at the end of this line if you wish to install in a different location
 ```

config/ascii/arch

@@ -1,5 +1,5 @@
 #/4;27;4;11
-${C1}                   -`
+%1                   -`
                   .o+`
                  `ooo/
                 `+oooo:
@@ -8,9 +8,9 @@ ${C1}                   -`
              `/:-:++oooo+:
             `/++++/+++++++:
            `/++++++++++++++:
-          `/+++o${C2}oooooooo${C1}oooo/`
+          `/+++o%2oooooooo%1oooo/`
-${C2}         ${C1}./${C2}ooosssso++osssssso${C1}+`
+%2         %1./%2ooosssso++osssssso%1+`
-${C2}        .oossssso-````/ossssss+`
+%2        .oossssso-````/ossssss+`
        -osssssso.      :ssssssso.
       :osssssss/        osssso+++.
      /ossssssss/        +ssssooo/-

config/ascii/artix

@@ -0,0 +1,20 @@
+#/45;27;45;7
+%1                  'o'
+                 'ooo'
+                'ooxoo'
+               'ooxxxoo'
+              'oookkxxoo'
+             'oiioxkkxxoo'
+            ':;:iiiioxxxoo'
+               `'.;::ioxxoo'
+          '-.      `':;jiooo'
+         'oooio-..     `'i:io'
+        'ooooxxxxoio:,.   `'-;'
+       'ooooxxxxxkkxoooIi:-.  `'
+      'ooooxxxxxkkkkxoiiiiiji'
+     'ooooxxxxxkxxoiiii:'`     .i'
+    'ooooxxxxxoi:::'`       .;ioxo'
+   'ooooxooi::'`         .:iiixkxxo'
+  'ooooi:'`                `'';ioxxo'
+ 'i:'`                          '':io'
+'`                                   `'

config/ascii/bedrock

@@ -0,0 +1,12 @@
+#/7;8;8;7
+%1   \\\\\\\\\\\\
+    \\\      \\\
+     \\\      \\\
+      \\\      \\\\\\\\\\\\\\\\\
+       \\\                    \\\
+        \\\                    \\\
+         \\\        %2______      \\\
+          \\\                   ///
+           \\\                 ///
+            \\\               ///
+             \\\////////////////

config/ascii/cachyos

@@ -0,0 +1,23 @@
+#/43;45;45;15
+           %3.%1-------------------------:
+          .%2+=%1========================.
+         :%2++%1===%2++===%1===============-       :%2++%1-
+        :%2*++%1====%2+++++==%1===========-        .==:
+       -%2*+++%1=====%2+***++=%1=========:
+      =%2*++++=%1=======------------:
+     =%2*+++++=%1====-                     %3...%1
+   .%2+*+++++%1=-===:                    .%2=+++=%1:
+  :%2++++%1=====-==:                     -***%2**%1+
+ :%2++=%1=======-=.                      .=+**+%3.%1
+.%2+%1==========-.                          %3.%1
+ :%2+++++++%1====-                                %3.%1--==-%3.%1
+  :%2++%1==========.                             %3:%2+++++++%1%3:
+   %1.-===========.                            =*****+*+
+    %1.-===========:                           .+*****+:
+      %1-=======%2++++%1:::::::::::::::::::::::::-:  %3.%1---:
+       :======%2++++%1====%2+++******************=.
+        %1:=====%2+++%1==========%2++++++++++++++*-
+         %1.====%2++%1==============%2++++++++++*-
+          %1.===%2+%1==================%2+++++++:
+           %1.-=======================%2+++:
+             %3..........................

config/ascii/debian

@@ -1,16 +1,16 @@
 #/1;7;9;15
-${C2}       _,met&&&&&gg.
+%2       _,met&&&&&gg.
     ,g&&&&&&&&&&&&&&&P.
   ,g&&P"        """Y&&.".
  ,&&P'              `&&&.
 ',&&P       ,ggs.     `&&b:
-`d&&'     ,&P"'   ${C1}.${C2}    &&&
+`d&&'     ,&P"'   %1.%2    &&&
- &&P      d&'     ${C1},${C2}    &&P
+ &&P      d&'     %1,%2    &&P
- &&:      &&.   ${C1}-${C2}    ,d&&'
+ &&:      &&.   %1-%2    ,d&&'
  &&;      Y&b._   _,d&P'
- Y&&.    ${C1}`.${C2}`"Y&&&&P"'
+ Y&&.    %1`.%2`"Y&&&&P"'
-${C2} `&&b      ${C1}"-.__
+%2 `&&b      %1"-.__
-${C2}  `Y&&
+%2  `Y&&
    `Y&&.
      `&&b.
        `Y&&b.

config/ascii/fedora

@@ -1,20 +1,20 @@
 #/12;7;4;15
-${C1}             .',;::::;,'.
+%1             .',;::::;,'.
          .';:cccccccccccc:;,.
       .;cccccccccccccccccccccc;.
     .:cccccccccccccccccccccccccc:.
-  .;ccccccccccccc;${C2}.:dddl:.${C1};ccccccc;.
+  .;ccccccccccccc;%2.:dddl:.%1;ccccccc;.
- .:ccccccccccccc;${C2}OWMKOOXMWd${C1};ccccccc:.
+ .:ccccccccccccc;%2OWMKOOXMWd%1;ccccccc:.
-.:ccccccccccccc;${C2}KMMc${C1};cc;${C2}xMMc${C1};ccccccc:.
+.:ccccccccccccc;%2KMMc%1;cc;%2xMMc%1;ccccccc:.
-,cccccccccccccc;${C2}MMM.${C1};cc;${C2};WW:${C1};cccccccc,
+,cccccccccccccc;%2MMM.%1;cc;%2;WW:%1;cccccccc,
-:cccccccccccccc;${C2}MMM.${C1};cccccccccccccccc:
+:cccccccccccccc;%2MMM.%1;cccccccccccccccc:
-:ccccccc;${C2}oxOOOo${C1};${C2}MMM0OOk.${C1};cccccccccccc:
+:ccccccc;%2oxOOOo%1;%2MMM0OOk.%1;cccccccccccc:
-cccccc;${C2}0MMKxdd:${C1};${C2}MMMkddc.${C1};cccccccccccc;
+cccccc;%20MMKxdd:%1;%2MMMkddc.%1;cccccccccccc;
-ccccc;${C2}XM0'${C1};cccc;${C2}MMM.${C1};cccccccccccccccc'
+ccccc;%2XM0'%1;cccc;%2MMM.%1;cccccccccccccccc'
-ccccc;${C2}MMo${C1};ccccc;${C2}MMW.${C1};ccccccccccccccc;
+ccccc;%2MMo%1;ccccc;%2MMW.%1;ccccccccccccccc;
-ccccc;${C2}0MNc.${C1}ccc${C2}.xMMd${C1};ccccccccccccccc;
+ccccc;%20MNc.%1ccc%2.xMMd%1;ccccccccccccccc;
-cccccc;${C2}dNMWXXXWM0:${C1};cccccccccccccc:,
+cccccc;%2dNMWXXXWM0:%1;cccccccccccccc:,
-cccccccc;${C2}.:odl:.${C1};cccccccccccccc:,.
+cccccccc;%2.:odl:.%1;cccccccccccccc:,.
 :cccccccccccccccccccccccccccc:'.
 .:cccccccccccccccccccccc:;,..
   '::cccccccccccccc::;,.

config/ascii/gentoo

@@ -1,19 +1,19 @@
 #/5;7;13;69
-${C1}         -/oyddmdhs+:.
+%1         -/oyddmdhs+:.
-     -o${C2}dNMMMMMMMMNNmhy+${C1}-`
+     -o%2dNMMMMMMMMNNmhy+%1-`
-   -y${C2}NMMMMMMMMMMMNNNmmdhy${C1}+-
+   -y%2NMMMMMMMMMMMNNNmmdhy%1+-
- `o${C2}mMMMMMMMMMMMMNmdmmmmddhhy${C1}/`
+ `o%2mMMMMMMMMMMMMNmdmmmmddhhy%1/`
- om${C2}MMMMMMMMMMMN${C1}hhyyyo${C2}hmdddhhhd${C1}o`
+ om%2MMMMMMMMMMMN%1hhyyyo%2hmdddhhhd%1o`
-.y${C2}dMMMMMMMMMMd${C1}hs++so/s${C2}mdddhhhhdm${C1}+`
+.y%2dMMMMMMMMMMd%1hs++so/s%2mdddhhhhdm%1+`
- oy${C2}hdmNMMMMMMMN${C1}dyooy${C2}dmddddhhhhyhN${C1}d.
+ oy%2hdmNMMMMMMMN%1dyooy%2dmddddhhhhyhN%1d.
-  :o${C2}yhhdNNMMMMMMMNNNmmdddhhhhhyym${C1}Mh
+  :o%2yhhdNNMMMMMMMNNNmmdddhhhhhyym%1Mh
-    .:${C2}+sydNMMMMMNNNmmmdddhhhhhhmM${C1}my
+    .:%2+sydNMMMMMNNNmmmdddhhhhhhmM%1my
-       /m${C2}MMMMMMNNNmmmdddhhhhhmMNh${C1}s:
+       /m%2MMMMMMNNNmmmdddhhhhhmMNh%1s:
-    `o${C2}NMMMMMMMNNNmmmddddhhdmMNhs${C1}+`
+    `o%2NMMMMMMMNNNmmmddddhhdmMNhs%1+`
-  `s${C2}NMMMMMMMMNNNmmmdddddmNMmhs${C1}/.
+  `s%2NMMMMMMMMNNNmmmdddddmNMmhs%1/.
- /N${C2}MMMMMMMMNNNNmmmdddmNMNdso${C1}:`
+ /N%2MMMMMMMMNNNNmmmdddmNMNdso%1:`
-+M${C2}MMMMMMNNNNNmmmmdmNMNdso${C1}/-
++M%2MMMMMMNNNNNmmmmdmNMNdso%1/-
-yM${C2}MNNNNNNNmmmmmNNMmhs+/${C1}-`
+yM%2MNNNNNNNmmmmmNNMmhs+/%1-`
-/h${C2}MMNNNNNNNNMNdhs++/${C1}-`
+/h%2MMNNNNNNNNMNdhs++/%1-`
-`/${C2}ohdmmddhys+++/:${C1}.`
+`/%2ohdmmddhys+++/:%1.`
   `-//////:--.

config/ascii/linuxmint

@@ -0,0 +1,20 @@
+#/34;15;46;252
+%2             ...-:::::-...
+%2          .-MMMMMMMMMMMMMMM-.
+      .-MMMM%1`..-:::::::-..`%2MMMM-.
+    .:MMMM%1.:MMMMMMMMMMMMMMM:.%2MMMM:.
+   -MMM%1-M---MMMMMMMMMMMMMMMMMMM.%2MMM-
+ `:MMM%1:MM`  :MMMM:....::-...-MMMM:%2MMM:`
+ :MMM%1:MMM`  :MM:`  ``    ``  `:MMM:%2MMM:
+.MMM%1.MMMM`  :MM.  -MM.  .MM-  `MMMM.%2MMM.
+:MMM%1:MMMM`  :MM.  -MM-  .MM:  `MMMM-%2MMM:
+:MMM%1:MMMM`  :MM.  -MM-  .MM:  `MMMM:%2MMM:
+:MMM%1:MMMM`  :MM.  -MM-  .MM:  `MMMM-%2MMM:
+.MMM%1.MMMM`  :MM:--:MM:--:MM:  `MMMM.%2MMM.
+ :MMM%1:MMM-  `-MMMMMMMMMMMM-`  -MMM-%2MMM:
+  :MMM%1:MMM:`                `:MMM:%2MMM:
+   .MMM%1.MMMM:--------------:MMMM.%2MMM.
+     '-MMMM%1.-MMMMMMMMMMMMMMM-.%2MMMM-'
+       '.-MMMM%1``--:::::--``%2MMMM-.'
+%2            '-MMMMMMMMMMMMM-'
+%2               ``-:::::-``

config/ascii/opensuse

@@ -1,18 +1,18 @@
 #/2;7;10;15
-${C2}           .;ldkO0000Okdl;.
+%2           .;ldkO0000Okdl;.
        .;d00xl:^''''''^:ok00d;.
      .d00l'                'o00d.
-   .d0Kd'${C1}  Okxol:;,.          ${C2}:O0d.
+   .d0Kd'%1  Okxol:;,.          %2:O0d.
-  .OK${C1}KKK0kOKKKKKKKKKKOxo:,      ${C2}lKO.
+  .OK%1KKK0kOKKKKKKKKKKOxo:,      %2lKO.
- ,0K${C1}KKKKKKKKKKKKKKK0P^${C2},,,${C1}^dx:${C2}    ;00,
+ ,0K%1KKKKKKKKKKKKKKK0P^%2,,,%1^dx:%2    ;00,
-.OK${C1}KKKKKKKKKKKKKKKk'${C2}.oOPPb.${C1}'0k.${C2}   cKO.
+.OK%1KKKKKKKKKKKKKKKk'%2.oOPPb.%1'0k.%2   cKO.
-:KK${C1}KKKKKKKKKKKKKKK: ${C2}kKx..dd ${C1}lKd${C2}   'OK:
+:KK%1KKKKKKKKKKKKKKK: %2kKx..dd %1lKd%2   'OK:
-dKK${C1}KKKKKKKKKOx0KKKd ${C2}^0KKKO' ${C1}kKKc${C2}   dKd
+dKK%1KKKKKKKKKOx0KKKd %2^0KKKO' %1kKKc%2   dKd
-dKK${C1}KKKKKKKKKK;.;oOKx,..${C2}^${C1}..;kKKK0.${C2}  dKd
+dKK%1KKKKKKKKKK;.;oOKx,..%2^%1..;kKKK0.%2  dKd
-:KK${C1}KKKKKKKKKK0o;...^cdxxOK0O/^^'  ${C2}.0K:
+:KK%1KKKKKKKKKK0o;...^cdxxOK0O/^^'  %2.0K:
- kKK${C1}KKKKKKKKKKKKK0x;,,......,;od  ${C2}lKk
+ kKK%1KKKKKKKKKKKKK0x;,,......,;od  %2lKk
- '0K${C1}KKKKKKKKKKKKKKKKKKKK00KKOo^  ${C2}c00'
+ '0K%1KKKKKKKKKKKKKKKKKKKK00KKOo^  %2c00'
-  'kK${C1}KKOxddxkOO00000Okxoc;''   ${C2}.dKk'
+  'kK%1KKOxddxkOO00000Okxoc;''   %2.dKk'
     l0Ko.                    .c00l'
      'l0Kk:.              .;xK0l'
         'lkK0xl:;,,,,;:ldO0kl'

config/ascii/opensuse-tumbleweed

@@ -1,5 +1,5 @@
 #/2;7;10;15
-${C1}                                     ......
+%1                                     ......
      .,cdxxxoc,.               .:kKMMMNWMMMNk:.
     cKMMN0OOOKWMMXo. ;        ;0MWk:.      .:OMMk.
   ;WMK;.       .lKMMNM,     :NMK,             .OMW;

config/ascii/pop

@@ -0,0 +1,21 @@
+#/33;7;11;15
+%1             /////////////
+         /////////////////////
+      ///////%2*767%1////////////////
+    //////%27676767676*%1//////////////
+   /////%276767%1//%27676767%1//////////////
+  /////%2767676%1///%2*76767%1///////////////
+ ///////%2767676%1///%276767%1.///%27676*%1///////
+/////////%2767676%1//%276767%1///%2767676%1////////
+//////////%276767676767%1////%276767%1/////////
+///////////%276767676%1//////%27676%1//////////
+////////////,%27676%1,///////%2767%1///////////
+/////////////*%27676%1///////%276%1////////////
+///////////////%27676%1////////////////////
+ ///////////////%27676%1///%2767%1////////////
+  //////////////////////%2'%1////////////
+   //////%2.7676767676767676767,%1//////
+    /////%2767676767676767676767%1/////
+      ///////////////////////////
+         /////////////////////
+             /////////////

config/ascii/tide

@@ -1,18 +1,15 @@
-#/6;10;5;12
+#/27;39;5;12
-${C1}                 @@@@@@@@@
+            %1@@@@               %1@@@%2*
-              @@@~~~~~~~~~@@
+          %1@@%2~~~~%1@@           %1@@%2~~~
-            @@~~~         ~~@
+        %1@@%2~~    %2~~%1@        %1@@%2~~
-           @~~              @
+      %1@@%2~~        %2~%1@    %1@@@%2~~
-          @~          @@@  @
+   %2*%1@@%2~~   %1@@@@    %2~%1@@@@%2~~~   %1@@@%2*
-        @@~          @   @@
+    %2~~   %1@@%2~~~~%1@@   %2~~~~    %1@@%2~~~
-      @@~~           @
+       %1@@%2~~    %2~~%1@        %1@@%2~~
-   @@@~~             ~@@           @@@
+     %1@@%2~~        %2~%1@    %1@@@%2~~
-   ~~~                ~~@@@    @@@@~~~
+  %2*%1@@%2~~   %1@@@@    %2~%1@@@@%2~~~   %1@@@%2*
-                        ~~~@@@@~~~~
+   %2~~   %1@@%2~~~~%1@@   %2~~~~    %1@@%2~~~
-   @@@@          @@@@@@@   ~~~~   @@@@
+      %1@@%2~~    %2~~%1@        %1@@%2~~
-   ~~~~@@@@    @@~~~~~~~@@    @@@@
+    %1@@%2~~        %2~%1@    %1@@@%2~~
-       ~~~~@@@@~~       ~~@@@@~~~~
+ %2*%1@@%2~~           %2~%1@@@@%2~~~
-    @@@    ~~~~   @@@@@   ~~~~    @@@
+  %2~~              %2~~~~
-    ~~~@@@@   @@@@~~~~~@@@@   @@@@~~~
-       ~~~~@@@~~~~     ~~~~@@@~~~
-           ~~~             ~~~

config/ascii/ubuntu

@@ -1,21 +1,21 @@
 #/1;7;3;15
-${C1}            .-/+oossssoo+\-.
+%1            .-/+oossssoo+\-.
-        ´:+ssssssssssssssssss+:`
+        ':+ssssssssssssssssss+:'
       -+ssssssssssssssssssyyssss+-
-    .ossssssssssssssssss${C2}dMMMNy${C1}sssso.
+    .ossssssssssssssssss%2dMMMNy%1sssso.
-   /sssssssssss${C2}hdmmNNmmyNMMMMh${C1}ssssss\
+   /sssssssssss%2hdmmNNmmyNMMMMh%1ssssss\
-  +sssssssss${C2}hm${C1}yd${C2}MMMMMMMNddddy${C1}ssssssss+
+  +sssssssss%2hm%1yd%2MMMMMMMNddddy%1ssssssss+
- /ssssssss${C2}hNMMM${C1}yh${C2}hyyyyhmNMMMNh${C1}ssssssss\
+ /ssssssss%2hNMMM%1yh%2hyyyyhmNMMMNh%1ssssssss\
-.ssssssss${C2}dMMMNh${C1}ssssssssss${C2}hNMMMd${C1}ssssssss.
+.ssssssss%2dMMMNh%1ssssssssss%2hNMMMd%1ssssssss.
-+ssss${C2}hhhyNMMNy${C1}ssssssssssss${C2}yNMMMy${C1}sssssss+
++ssss%2hhhyNMMNy%1ssssssssssss%2yNMMMy%1sssssss+
-oss${C2}yNMMMNyMMh${C1}ssssssssssssss${C2}hmmmh${C1}ssssssso
+oss%2yNMMMNyMMh%1ssssssssssssss%2hmmmh%1ssssssso
-oss${C2}yNMMMNyMMh${C1}sssssssssssssshmmmh${C1}ssssssso
+oss%2yNMMMNyMMh%1sssssssssssssshmmmh%1ssssssso
-+ssss${C2}hhhyNMMNy${C1}ssssssssssss${C2}yNMMMy${C1}sssssss+
++ssss%2hhhyNMMNy%1ssssssssssss%2yNMMMy%1sssssss+
-.ssssssss${C2}dMMMNh${C1}ssssssssss${C2}hNMMMd${C1}ssssssss.
+.ssssssss%2dMMMNh%1ssssssssss%2hNMMMd%1ssssssss.
- \ssssssss${C2}hNMMM${C1}yh${C2}hyyyyhdNMMMNh${C1}ssssssss/
+ \ssssssss%2hNMMM%1yh%2hyyyyhdNMMMNh%1ssssssss/
-  +sssssssss${C2}dm${C1}yd${C2}MMMMMMMMddddy${C1}ssssssss+
+  +sssssssss%2dm%1yd%2MMMMMMMMddddy%1ssssssss+
-   \sssssssssss${C2}hdmNNNNmyNMMMMh${C1}ssssss/
+   \sssssssssss%2hdmNNNNmyNMMMMh%1ssssss/
-    .ossssssssssssssssss${C2}dMMMNy${C1}sssso.
+    .ossssssssssssssssss%2dMMMNy%1sssso.
-      -+sssssssssssssssss${C2}yyy${C1}ssss+-
+      -+sssssssssssssssss%2yyy%1ssss+-
-        `:+ssssssssssssssssss+:`
+        ':+ssssssssssssssssss+:'
             .-\+oossssoo+/-.

config/ascii/void

@@ -0,0 +1,19 @@
+#/2;36;72;15
+          %2:::::----:::::
+       %2::----------------::=
+        %2:-------------------::
+%1   :      %2:::.      ..:-------:
+%1  =*=.                  %2.:-----:
+%1 =****-                   %2:-----:
+%1=*****-       %2::::::       :-----:
+%1=****+      %2:--------:      :----:
+%1+****-     %2.----------.     :----:
+%1+****-     %2.----------.     :-----
+%1=****+      %2:--------:      :----:
+%1=*****-       %2::::::       :-----:
+%1 =*****-                   %2:----:
+%1  =*****+:                   %2:-:
+%1   =*******=::      ::=-.     %2.
+%1    ==*******************-
+%1      ===****************=-
+%1          ===+=****++===

config/config.yaml

@@ -1,5 +0,0 @@
-distro_ascii: auto
-fetch_script: auto
-ansii_colors: []
-force_config_ansii: false
-dependency_warning: true

config/config.yml

@@ -0,0 +1,16 @@
+distro_ascii: auto
+modules:
+  - name: distribution
+  - name: hostname
+  - name: kernel
+  - name: packages
+  - name: shell
+  - name: init
+  - name: motherboard
+  - name: cpus
+  - name: gpus
+  - name: memory
+  - name: partitions
+  - name: local_ip
+  - name: de_wm
+  - name: monitors

config/fetch_script.sh

@@ -1,25 +0,0 @@
-source fetch_script_functions.sh
-
-echo -e "${C3}Distribution: ${C4}${DISTRO_LONG_NAME} ($(uname -m))"
-echo -e "${C3}Hostname: ${C4}$(cat /etc/hostname)"
-echo -e "${C3}Kernel: ${C4}$(uname -s) $(uname -r)"
-echo -e "${C3}Packages: ${C4}$(get_packages)"
-echo -e "${C3}Shell: ${C4}${USER_SHELL}"
-if [ ! -z "$CPU_MODEL" ]; then echo -e "${C3}CPU: ${C4}${CPU_MODEL} (${CPU_THREADS} threads)"; fi
-if [ ! -z "$GPU_MODEL" ]; then echo -e "${C3}GPU: ${C4}${GPU_MODEL}"; fi
-if [ ! -z "$MEM_TOTAL" ] && [ ! -z "$MEM_USED" ]; then echo -e "${C3}Memory: ${C4}${MEM_USED} MiB / ${MEM_TOTAL} MiB"; fi
-for i in $(seq ${MOUNTED_PARTITIONS}); do
-    device="PARTITION${i}_DEVICE"
-    mountpoint="PARTITION${i}_MOUNTPOINT"
-    total="PARTITION${i}_TOTAL_SIZE"
-    used="PARTITION${i}_USED_SIZE"
-    echo -e "${C3}${!mountpoint}: ${C4}${!used}/${!total}"
-  done
-if [ ! -z "$DISPLAY_PROTOCOL" ]; then
-  echo -e "${C3}Display Protocol: ${C4}${DISPLAY_PROTOCOL}"
-  for i in $(seq ${CONNECTED_MONITORS}); do
-    monitor="MONITOR$i"
-    echo -e "${C3}Screen $i: ${C4}${!monitor}"
-  done
-fi
-if [ ! -z "$DE_WM" ]; then echo -e "${C3}DE/WM: ${C4}${DE_WM}"; fi

config/fetch_script_functions.sh

@@ -1,37 +0,0 @@
-command_exists() {
-    if [ -z "$1" ]; then
-        return 1
-    fi
-    if command -v "$1" &> /dev/null; then
-        return 0
-    else
-        return 1
-    fi
-}
-
-get_packages() {
-    ARRAY=()
-    if command_exists dpkg; then
-        ARRAY+=("$(dpkg-query -f '.\n' -W | wc -l) (dpkg)")
-    fi
-    if command_exists pacman; then
-        ARRAY+=("$(pacman -Q | wc -l) (pacman)")
-    fi
-    if command_exists rpm; then
-        ARRAY+=("$(rpm -qa | wc -l) (rpm)")
-    fi
-    if command_exists bpm; then
-        ARRAY+=("$(bpm list -n) (bpm)")
-    fi
-    if command_exists emerge; then
-        ARRAY+=("$(ls -l /var/db/pkg/ | wc -l) (emerge)")
-    fi
-    if command_exists flatpak; then
-        ARRAY+=("$(flatpak list | wc -l) (flatpak)")
-    fi
-    if command_exists snap; then
-        ARRAY+=("$(snap list | wc -l) (snap)")
-    fi
-    echo "${ARRAY[@]}"
-    unset ARRAY
-}

main.go

@@ -1,222 +0,0 @@
-package main
-
-import (
-	"fmt"
-	yaml "gopkg.in/yaml.v3"
-	"log"
-	"os"
-	"os/exec"
-	"path"
-	"regexp"
-	"strconv"
-	"strings"
-)
-
-var configPath = ""
-var fetchScriptPath = ""
-
-var config = StormfetchConfig{
-	Ascii:             "auto",
-	FetchScript:       "auto",
-	AnsiiColors:       make([]int, 0),
-	ForceConfigAnsii:  false,
-	DependencyWarning: true,
-}
-
-type StormfetchConfig struct {
-	Ascii             string `yaml:"distro_ascii"`
-	FetchScript       string `yaml:"fetch_script"`
-	AnsiiColors       []int  `yaml:"ansii_colors"`
-	ForceConfigAnsii  bool   `yaml:"force_config_ansii"`
-	DependencyWarning bool   `yaml:"dependency_warning"`
-}
-
-func main() {
-	readConfig()
-}
-
-func readConfig() {
-	// Get home directory
-	configDir, _ := os.UserConfigDir()
-	// Find valid config directory
-	if _, err := os.Stat(path.Join(configDir, "stormfetch/config.yaml")); err == nil {
-		configPath = path.Join(configDir, "stormfetch/config.yaml")
-	} else if _, err := os.Stat("/etc/stormfetch/config.yaml"); err == nil {
-		configPath = "/etc/stormfetch/config.yaml"
-	} else {
-		log.Fatalf("Config file not found: %s", err.Error())
-	}
-	// Parse config
-	bytes, err := os.ReadFile(configPath)
-	if err != nil {
-		log.Fatal(err)
-	}
-	err = yaml.Unmarshal(bytes, &config)
-	if err != nil {
-		log.Fatal(err)
-	}
-	if config.FetchScript == "" {
-		log.Fatalf("Fetch script path is empty")
-	} else if config.FetchScript != "auto" {
-		stat, err := os.Stat(config.FetchScript)
-		if err != nil {
-			log.Fatalf("Fetch script file not found: %s", err.Error())
-		} else if stat.IsDir() {
-			log.Fatalf("Fetch script path points to a directory")
-		}
-	}
-	if _, err := os.Stat(path.Join(configDir, "stormfetch/fetch_script.sh")); err == nil {
-		fetchScriptPath = path.Join(configDir, "stormfetch/fetch_script.sh")
-	} else if _, err := os.Stat("/etc/stormfetch/fetch_script.sh"); err == nil {
-		fetchScriptPath = "/etc/stormfetch/fetch_script.sh"
-	} else {
-		log.Fatalf("Fetch script file not found: %s", err.Error())
-	}
-	// Show Dependency warning if enabled
-	if config.DependencyWarning {
-		var dependencies []string
-		var missing []string
-		for _, depend := range dependencies {
-			if _, err := os.Stat(path.Join("/usr/bin/", depend)); err != nil {
-				missing = append(missing, depend)
-			}
-		}
-		if len(missing) != 0 {
-			fmt.Println("[WARNING] Stormfetch functionality may be limited due to the following dependencies not being installed:")
-			for _, depend := range missing {
-				fmt.Println(depend)
-			}
-			fmt.Println("You can disable this warning through your stormfetch config")
-		}
-	}
-	// Fetch ascii art and apply colors
-	colorMap := make(map[string]string)
-	colorMap["C0"] = "\033[0m"
-	setColorMap := func() {
-		for i := 0; i < 6; i++ {
-			if i > len(config.AnsiiColors)-1 {
-				colorMap["C"+strconv.Itoa(i+1)] = "\033[0m"
-				continue
-			}
-			colorMap["C"+strconv.Itoa(i+1)] = fmt.Sprintf("\033[1m\033[38;5;%dm", config.AnsiiColors[i])
-		}
-	}
-	setColorMap()
-	ascii := getDistroAsciiArt()
-	if strings.HasPrefix(ascii, "#/") {
-		firstLine := strings.Split(ascii, "\n")[0]
-		if !config.ForceConfigAnsii {
-			ansiiColors := strings.Split(strings.TrimPrefix(firstLine, "#/"), ";")
-			for i, color := range ansiiColors {
-				atoi, err := strconv.Atoi(color)
-				if err != nil {
-					log.Fatal(err)
-				}
-				if i < len(config.AnsiiColors) {
-					config.AnsiiColors[i] = atoi
-				} else {
-					config.AnsiiColors = append(config.AnsiiColors, atoi)
-				}
-			}
-			setColorMap()
-		}
-		ascii = os.Expand(ascii, func(s string) string {
-			return colorMap[s]
-		})
-		ascii = strings.TrimPrefix(ascii, firstLine+"\n")
-	} else {
-		ascii = os.Expand(ascii, func(s string) string {
-			return colorMap[s]
-		})
-	}
-	asciiSplit := strings.Split(ascii, "\n")
-	asciiNoColor := StripAnsii(ascii)
-	//Execute fetch script
-	cmd := exec.Command("/bin/bash", fetchScriptPath)
-	cmd.Dir = path.Dir(fetchScriptPath)
-	cmd.Env = os.Environ()
-	cmd.Env = append(cmd.Env, SetupFetchEnv()...)
-	cmd.Env = append(cmd.Env, "C0=\033[0m")
-	for key, value := range colorMap {
-		cmd.Env = append(cmd.Env, fmt.Sprintf("%s=%s", key, value))
-	}
-	out, err := cmd.Output()
-	if err != nil {
-		log.Fatal(err)
-	}
-	// Print Distro Information
-	maxWidth := 0
-	for _, line := range strings.Split(asciiNoColor, "\n") {
-		if len(line) > maxWidth {
-			maxWidth = len(line)
-		}
-	}
-	final := ""
-	for lineIndex, line := range asciiSplit {
-		lineVisibleLength := len(strings.Split(asciiNoColor, "\n")[lineIndex])
-		lastAsciiColor := ""
-		if lineIndex != 0 {
-			r := regexp.MustCompile("\033[38;5;[0-9]+m")
-			matches := r.FindAllString(asciiSplit[lineIndex-1], -1)
-			if len(matches) != 0 {
-				lastAsciiColor = r.FindAllString(asciiSplit[lineIndex-1], -1)[len(matches)-1]
-			}
-		}
-		for i := lineVisibleLength; i < maxWidth+5; i++ {
-			line = line + " "
-		}
-		asciiSplit[lineIndex] = lastAsciiColor + line
-		str := string(out)
-		if lineIndex < len(strings.Split(str, "\n")) {
-			line = line + colorMap["C0"] + strings.Split(str, "\n")[lineIndex]
-		}
-		final += lastAsciiColor + line + "\n"
-	}
-	fmt.Println(final)
-}
-
-func SetupFetchEnv() []string {
-	var env = make(map[string]string)
-	env["DISTRO_LONG_NAME"] = getDistroInfo().LongName
-	env["DISTRO_SHORT_NAME"] = getDistroInfo().ShortName
-	env["CPU_MODEL"] = getCPUName()
-	env["CPU_THREADS"] = strconv.Itoa(getCPUThreads())
-	memory := GetMemoryInfo()
-	if memory != nil {
-		env["MEM_TOTAL"] = strconv.Itoa(memory.MemTotal)
-		env["MEM_USED"] = strconv.Itoa(memory.MemTotal - memory.MemAvailable)
-		env["MEM_FREE"] = strconv.Itoa(memory.MemAvailable)
-	}
-	partitions := getMountedPartitions()
-	if len(partitions) != 0 {
-		env["MOUNTED_PARTITIONS"] = strconv.Itoa(len(partitions))
-		for i, part := range partitions {
-			env["PARTITION"+strconv.Itoa(i+1)+"_DEVICE"] = part.Device
-			env["PARTITION"+strconv.Itoa(i+1)+"_MOUNTPOINT"] = part.MountPoint
-			env["PARTITION"+strconv.Itoa(i+1)+"_TOTAL_SIZE"] = FormatBytes(part.TotalSize)
-			env["PARTITION"+strconv.Itoa(i+1)+"_USED_SIZE"] = FormatBytes(part.UsedSize)
-			env["PARTITION"+strconv.Itoa(i+1)+"_FREE_SIZE"] = FormatBytes(part.FreeSize)
-		}
-	}
-	env["DE_WM"] = GetDEWM()
-	env["USER_SHELL"] = GetShell()
-	env["DISPLAY_PROTOCOL"] = GetDisplayProtocol()
-	monitors := getMonitorResolution()
-	if len(monitors) != 0 {
-		env["CONNECTED_MONITORS"] = strconv.Itoa(len(monitors))
-		for i, monitor := range monitors {
-			env["MONITOR"+strconv.Itoa(i+1)] = monitor
-		}
-	}
-	if getGPUName() != "" {
-		env["GPU_MODEL"] = getGPUName()
-	}
-
-	var ret = make([]string, len(env))
-	i := 0
-	for key, value := range env {
-		ret[i] = fmt.Sprintf("%s=%s", key, value)
-		i++
-	}
-	return ret
-}

src/color.go

@@ -0,0 +1,28 @@
+package main
+
+import (
+	"fmt"
+	"strings"
+)
+
+func setupColorMap(asciiArtHeader string) []string {
+	colorMap := make([]string, 10)
+
+	// Set default color map values
+	for i := range 10 {
+		colorMap[i] = "\033[0m"
+	}
+
+	// Return if header is empty
+	if asciiArtHeader == "" {
+		return colorMap
+	}
+
+	// Read colors from ascii art header
+	ansiColors := strings.Split(strings.TrimPrefix(asciiArtHeader, "#/"), ";")
+	for i := 0; i < 9 && i < len(ansiColors); i++ {
+		colorMap[i+1] = fmt.Sprintf("\033[38;5;%sm", ansiColors[i])
+	}
+
+	return colorMap
+}

src/config.go

@@ -0,0 +1,47 @@
+package main
+
+import (
+	"log"
+	"os"
+	"path"
+
+	"gopkg.in/yaml.v3"
+)
+
+type StormfetchConfig struct {
+	Ascii            string                   `yaml:"distro_ascii"`
+	Modules          []stormfetchModuleConfig `yaml:"modules"`
+	AnsiiColors      []int                    `yaml:"ansii_colors"`
+	ForceConfigAnsii bool                     `yaml:"force_config_ansii"`
+}
+
+var config = StormfetchConfig{
+	Ascii:   "auto",
+	Modules: make([]stormfetchModuleConfig, 0),
+}
+
+func readConfig() {
+	if ConfigPath == "" {
+		// Get home directory
+		userConfigDir, _ := os.UserConfigDir()
+
+		// Find valid config directory
+		if _, err := os.Stat(path.Join(userConfigDir, "stormfetch/config.yml")); err == nil {
+			ConfigPath = path.Join(userConfigDir, "stormfetch/config.yml")
+		} else if _, err := os.Stat(path.Join(SystemConfigDir, "stormfetch/config.yml")); err == nil {
+			ConfigPath = path.Join(SystemConfigDir, "stormfetch/config.yml")
+		} else {
+			log.Fatalf("Config file not found: %s", err.Error())
+		}
+	}
+
+	// Parse config
+	bytes, err := os.ReadFile(ConfigPath)
+	if err != nil {
+		log.Fatal(err)
+	}
+	err = yaml.Unmarshal(bytes, &config)
+	if err != nil {
+		log.Fatal(err)
+	}
+}

src/go.mod

@@ -3,7 +3,7 @@ module stormfetch
 go 1.22
 
 require (
-	github.com/BurntSushi/xgb v0.0.0-20210121224620-deaf085860bc
+	github.com/go-gl/glfw/v3.3/glfw v0.0.0-20240506104042-037f3cc74f2a
 	github.com/jackmordaunt/ghw v1.0.4
 	github.com/mitchellh/go-ps v1.0.0
 	gopkg.in/yaml.v3 v3.0.1

src/go.sum

@@ -1,5 +1,5 @@
-github.com/BurntSushi/xgb v0.0.0-20210121224620-deaf085860bc h1:7D+Bh06CRPCJO3gr2F7h1sriovOZ8BMhca2Rg85c2nk=
+github.com/go-gl/glfw/v3.3/glfw v0.0.0-20240506104042-037f3cc74f2a h1:vxnBhFDDT+xzxf1jTJKMKZw3H0swfWk9RpWbBbDK5+0=
-github.com/BurntSushi/xgb v0.0.0-20210121224620-deaf085860bc/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo=
+github.com/go-gl/glfw/v3.3/glfw v0.0.0-20240506104042-037f3cc74f2a/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=
 github.com/go-ole/go-ole v1.2.6 h1:/Fpf6oFPoeFik9ty7siob0G6Ke8QvQEuVcuChpwXzpY=
 github.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=
 github.com/jackmordaunt/ghw v1.0.4 h1:as+COFuPuXaNQC3WqzoHS/E2JYWZU7gN8ompNTUxNxs=

src/hardware.go

@@ -0,0 +1,99 @@
+package main
+
+import (
+	"os"
+	"os/exec"
+	"slices"
+	"strings"
+
+	"github.com/go-gl/glfw/v3.3/glfw"
+	"github.com/jackmordaunt/ghw"
+)
+
+type CPU struct {
+	Vendor  string
+	Model   string
+	Cores   int
+	Threads int
+}
+
+type Monitor struct {
+	Width       int
+	Height      int
+	RefreshRate int
+}
+
+func GetCPUs(hiddenCPUs []int) []CPU {
+	ret := make([]CPU, 0)
+
+	cpus, err := ghw.CPU()
+	if err != nil {
+		return ret
+	}
+
+	for i, cpu := range cpus.Processors {
+		if slices.Contains(hiddenCPUs, i+1) {
+			continue
+		}
+
+		ret = append(ret, CPU{
+			Vendor:  cpu.Vendor,
+			Model:   cpu.Model,
+			Cores:   int(cpu.NumCores),
+			Threads: int(cpu.NumThreads),
+		})
+	}
+
+	return ret
+}
+
+func GetGPUModels(hiddenGPUS []int) (ret []string) {
+	cmd := exec.Command("sh", "-c", "lspci -v -m | grep 'VGA' -A6 | grep '^Device:'")
+	bytes, err := cmd.Output()
+	if err != nil {
+		return nil
+	}
+
+	for i, gpu := range strings.Split(string(bytes), "\n") {
+		if slices.Contains(hiddenGPUS, i+1) {
+			continue
+		}
+		if gpu == "" {
+			continue
+		}
+		gpu = strings.TrimPrefix(strings.TrimSpace(gpu), "Device:\t")
+		ret = append(ret, gpu)
+	}
+
+	return ret
+}
+
+func GetMotherboardModel() string {
+	bytes, err := os.ReadFile("/sys/devices/virtual/dmi/id/board_name")
+	if err != nil {
+		return ""
+	}
+	return strings.TrimSpace(string(bytes))
+}
+
+func GetMonitors() []Monitor {
+	var monitors []Monitor
+	if GetDisplayProtocol() != "" {
+		err := glfw.Init()
+		if err != nil {
+			panic(err)
+		}
+
+		for _, monitor := range glfw.GetMonitors() {
+			mode := monitor.GetVideoMode()
+
+			monitors = append(monitors, Monitor{
+				Width:       mode.Width,
+				Height:      mode.Height,
+				RefreshRate: mode.RefreshRate,
+			})
+		}
+		defer glfw.Terminate()
+	}
+	return monitors
+}

src/main.go

@@ -0,0 +1,159 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"strconv"
+	"strings"
+	"time"
+	"unicode"
+)
+
+// Build-time variables
+var StormfetchVersion = "dev"
+var SystemConfigDir = "/etc/"
+
+// Flag variables
+var ShowVersion = false
+var ConfigPath = ""
+var Ascii = ""
+var ShowModuleTimeTaken = false
+
+func main() {
+	parseFlags()
+	readConfig()
+	initializeModuleMap()
+	run()
+}
+
+func parseFlags() {
+	flag.BoolVar(&ShowVersion, "version", false, "Show Stormfetch version")
+	flag.StringVar(&ConfigPath, "config", "", "Use the specified config file")
+	flag.BoolVar(&ShowModuleTimeTaken, "time-taken", false, "Show time taken to execute each module")
+	flag.StringVar(&Ascii, "ascii", "", "Set distro ascii")
+	flag.Parse()
+}
+
+func run() {
+	if ShowVersion {
+		fmt.Printf("Stormfetch version %s\n", StormfetchVersion)
+		return
+	}
+
+	// Fetch ascii art and remove header
+	asciiArt := GetDistroAsciiArt()
+	asciiArtHeader := ""
+	if strings.HasPrefix(asciiArt, "#/") {
+		asciiArtHeader = strings.SplitN(asciiArt, "\n", 2)[0]
+		asciiArt = strings.SplitN(asciiArt, "\n", 2)[1]
+	}
+	asciiArtNoColor := asciiArt
+
+	// Rewrite last color code to the start of the next line
+	lastColor := "0"
+	for i := 1; i < len(asciiArt); i++ {
+		rune := rune(asciiArt[i])
+
+		if unicode.IsDigit(rune) && asciiArt[i-1] == '%' {
+			lastColor = string(rune)
+		}
+
+		if rune == '\n' {
+			asciiArt = asciiArt[:i+1] + "%" + lastColor + asciiArt[i+1:]
+		}
+	}
+
+	// Setup color map and replace colors in ascii art
+	colorMap := setupColorMap(asciiArtHeader)
+	for key, value := range colorMap {
+		asciiArt = strings.ReplaceAll(asciiArt, "%"+strconv.Itoa(key), value)
+		asciiArtNoColor = strings.ReplaceAll(asciiArtNoColor, "%"+strconv.Itoa(key), "")
+	}
+
+	// Execute modules in order
+	modulesText := make([]string, 0)
+	for _, moduleConfig := range config.Modules {
+		module, ok := Modules[moduleConfig.Name]
+		if !ok {
+			continue
+		}
+
+		// Set module config options
+		if moduleConfig.Format != "" {
+			module.Format = moduleConfig.Format
+		}
+		if moduleConfig.Data != nil {
+			module.Data = moduleConfig.Data
+		}
+
+		// Execute module
+		start := time.Now().UnixMilli()
+		text := module.Execute(module)
+		end := time.Now().UnixMilli()
+
+		// Insert default color at the start of the module's text
+		text = colorMap[0] + text
+
+		// Show time taken
+		if ShowModuleTimeTaken {
+			fmt.Printf("Module '%s' took %d milliseconds\n", module.Name, end-start)
+		}
+
+		// Replace colors in returned string
+		textNoColor := text
+		for key, value := range colorMap {
+			text = strings.ReplaceAll(text, "%"+strconv.Itoa(key), value)
+			textNoColor = strings.ReplaceAll(textNoColor, "%"+strconv.Itoa(key), value)
+		}
+
+		// Continue if text length is 0
+		if len(textNoColor) == 0 {
+			continue
+		}
+
+		// Add text to slice
+		for _, line := range strings.Split(strings.TrimSpace(text), "\n") {
+			modulesText = append(modulesText, line)
+		}
+	}
+
+	// Get longest line in ascii art
+	maxWidth := 0
+	for _, line := range strings.Split(asciiArtNoColor, "\n") {
+		if len(line) > maxWidth {
+			maxWidth = len(line)
+		}
+	}
+
+	// Split ascii art into each line
+	asciiArtSplit := strings.Split(asciiArt, "\n")
+	asciiArtNoColorSplit := strings.Split(asciiArtNoColor, "\n")
+
+	// Get amount of lines to print
+	lineCount := max(len(asciiArtSplit), len(modulesText))
+
+	// Combine ascii art and module text
+	final := strings.Builder{}
+	for i := range lineCount {
+		// Write ascii art
+		currentLineLength := 0
+		if i < len(asciiArtSplit) {
+			final.WriteString(asciiArtSplit[i])
+			currentLineLength += len(asciiArtNoColorSplit[i])
+		}
+
+		// Write blank space between ascii art and module text
+		for i := currentLineLength; i < maxWidth+3; i++ {
+			final.WriteString(" ")
+		}
+
+		// Write module text
+		if i < len(modulesText) {
+			final.WriteString(modulesText[i])
+		}
+
+		final.WriteString("\n")
+	}
+
+	fmt.Println(strings.TrimRight(final.String(), "\n") + "\033[0m")
+}

src/memory.go

@@ -0,0 +1,57 @@
+package main
+
+import (
+	"bufio"
+	"os"
+	"strconv"
+	"strings"
+)
+
+type Memory struct {
+	MemTotal     int
+	MemFree      int
+	MemAvailable int
+}
+
+func GetMemoryInfo() *Memory {
+	toInt := func(raw string) int {
+		if raw == "" {
+			return 0
+		}
+		res, err := strconv.Atoi(raw)
+		if err != nil {
+			panic(err)
+		}
+		return res
+	}
+
+	parseLine := func(raw string) (key string, value int) {
+		text := strings.ReplaceAll(raw[:len(raw)-2], " ", "")
+		keyValue := strings.Split(text, ":")
+		return keyValue[0], toInt(keyValue[1])
+	}
+
+	if _, err := os.Stat("/proc/meminfo"); err != nil {
+		return nil
+	}
+	file, err := os.Open("/proc/meminfo")
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+	bufio.NewScanner(file)
+	scanner := bufio.NewScanner(file)
+	res := Memory{}
+	for scanner.Scan() {
+		key, value := parseLine(scanner.Text())
+		switch key {
+		case "MemTotal":
+			res.MemTotal = value / 1024
+		case "MemFree":
+			res.MemFree = value / 1024
+		case "MemAvailable":
+			res.MemAvailable = value / 1024
+		}
+	}
+	return &res
+}

src/modules.go

@@ -0,0 +1,435 @@
+package main
+
+import (
+	"os"
+	"reflect"
+	"strconv"
+	"strings"
+)
+
+// Used to declare modules in config files
+type stormfetchModuleConfig struct {
+	Name   string         `yaml:"name"`
+	Format string         `yaml:"format"`
+	Data   map[string]any `yaml:"data"`
+}
+
+type StormfetchModule struct {
+	Execute func(StormfetchModule) string
+	stormfetchModuleConfig
+}
+
+var Modules map[string]StormfetchModule = make(map[string]StormfetchModule)
+
+func (sm StormfetchModule) GetData(key string, defaultValue any) (any, bool) {
+	if sm.Data == nil {
+		return defaultValue, false
+	}
+
+	data, ok := sm.Data[key]
+	if !ok {
+		return defaultValue, false
+	}
+
+	if reflect.ValueOf(data).Kind() != reflect.ValueOf(defaultValue).Kind() {
+		return defaultValue, false
+	}
+
+	return data, true
+}
+
+func RegisterModule(module StormfetchModule) bool {
+	if _, ok := Modules[module.Name]; ok {
+		return false
+	}
+
+	Modules[module.Name] = module
+	return true
+}
+
+func initializeModuleMap() {
+	// Distribution Module
+	distributionModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "distribution", Format: "%3Distribution: %4$DISTRO_SHORT ($ARCH)"}, Execute: func(sm StormfetchModule) string {
+		distroInfo := GetDistroInfo()
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "DISTRO_ID":
+				return distroInfo.ID
+			case "DISTRO_SHORT":
+				return distroInfo.ShortName
+			case "DISTRO_LONG":
+				return distroInfo.LongName
+			case "ARCH":
+				return GetArch()
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(distributionModule)
+
+	// Hostname module
+	hostnameModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "hostname", Format: "%3Hostname: %4$HOSTNAME"}, Execute: func(sm StormfetchModule) string {
+		hostname, _ := os.Hostname()
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "HOSTNAME":
+				return hostname
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(hostnameModule)
+
+	// Kernel module
+	kernelModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "kernel", Format: "%3Kernel: %4$KERNEL_NAME $KERNEL_RELEASE"}, Execute: func(sm StormfetchModule) string {
+		kernelName, kernelRelease := GetKernel()
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "KERNEL_NAME":
+				return kernelName
+			case "KERNEL_RELEASE":
+				return kernelRelease
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(kernelModule)
+
+	// Packages module
+	packagesModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "packages", Format: "%3Packages: %4$PACKAGES"}, Execute: func(sm StormfetchModule) string {
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "PACKAGES":
+				return GetInstalledPackages()
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(packagesModule)
+
+	// Shell module
+	shellModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "shell", Format: "%3Shell: %4$SHELL"}, Execute: func(sm StormfetchModule) string {
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "SHELL":
+				return GetShell()
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(shellModule)
+
+	// Init system module
+	initSystemModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "init_system", Format: "%3Init: %4$INIT"}, Execute: func(sm StormfetchModule) string {
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "INIT":
+				return GetInitSystem()
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(initSystemModule)
+
+	// Libc module
+	libcModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "libc", Format: "%3Libc: %4$LIBC"}, Execute: func(sm StormfetchModule) string {
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "LIBC":
+				return GetLibc()
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(libcModule)
+
+	// Motherboard module
+	MotherboardModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "motherboard", Format: "%3Motherboard: %4$MOTHERBOARD"}, Execute: func(sm StormfetchModule) string {
+		motherboard := GetMotherboardModel()
+
+		// Return empty string if can't detect motherboard model
+		if motherboard == "" {
+			return ""
+		}
+
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "MOTHERBOARD":
+				return motherboard
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(MotherboardModule)
+
+	// Motherboard module
+	cpusModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "cpus", Format: "%3CPU: %4$CPU_MODEL ($CPU_THREADS threads)"}, Execute: func(sm StormfetchModule) string {
+		hiddenCPUsInterface, _ := sm.GetData("hidden_cpus", make([]any, 0))
+
+		// Convert interface slices to string slices
+		hiddenCPUs := make([]int, 0)
+		for _, value := range hiddenCPUsInterface.([]any) {
+			hiddenCPUs = append(hiddenCPUs, value.(int))
+		}
+
+		builder := strings.Builder{}
+		cpus := GetCPUs(hiddenCPUs)
+
+		for i, cpu := range cpus {
+			expanded := os.Expand(sm.Format, func(s string) string {
+				switch s {
+				case "CPU_NUM":
+					return strconv.Itoa(i + 1)
+				case "CPU_VENDOR":
+					return cpu.Vendor
+				case "CPU_MODEL":
+					return cpu.Model
+				case "CPU_CORES":
+					return strconv.Itoa(cpu.Cores)
+				case "CPU_THREADS":
+					return strconv.Itoa(cpu.Threads)
+				default:
+					return ""
+				}
+			})
+
+			builder.WriteString(expanded + "\n")
+		}
+
+		return builder.String()
+	}}
+	RegisterModule(cpusModule)
+
+	// GPUs module
+	gpusModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "gpus", Format: "%3GPU: %4$GPU_MODEL"}, Execute: func(sm StormfetchModule) string {
+		hiddenGPUsInterface, _ := sm.GetData("hidden_gpus", make([]any, 0))
+
+		// Convert interface slices to string slices
+		hiddenGPUs := make([]int, 0)
+		for _, value := range hiddenGPUsInterface.([]any) {
+			hiddenGPUs = append(hiddenGPUs, value.(int))
+		}
+
+		builder := strings.Builder{}
+		gpus := GetGPUModels(hiddenGPUs)
+
+		for i, gpu := range gpus {
+			expanded := os.Expand(sm.Format, func(s string) string {
+				switch s {
+				case "GPU_NUM":
+					return strconv.Itoa(i + 1)
+				case "GPU_MODEL":
+					return gpu
+				default:
+					return ""
+				}
+			})
+
+			builder.WriteString(expanded + "\n")
+		}
+
+		return builder.String()
+	}}
+	RegisterModule(gpusModule)
+
+	// Memory module
+	memoryModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "memory", Format: "%3Memory: %4$MEM_USED MiB / $MEM_TOTAL MiB"}, Execute: func(sm StormfetchModule) string {
+		memoryInfo := GetMemoryInfo()
+
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "MEM_TOTAL":
+				return strconv.Itoa(memoryInfo.MemTotal)
+			case "MEM_AVAILABLE":
+				return strconv.Itoa(memoryInfo.MemAvailable)
+			case "MEM_FREE":
+				return strconv.Itoa(memoryInfo.MemFree)
+			case "MEM_USED":
+				return strconv.Itoa(memoryInfo.MemTotal - memoryInfo.MemAvailable)
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(memoryModule)
+
+	// Partitions module
+	partitionsModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "partitions", Format: "%3Partition ${PART_AUTONAME} (${PART_FS}): %4${PART_USED} / ${PART_TOTAL}"}, Execute: func(sm StormfetchModule) string {
+		hiddenPartitionsInterface, _ := sm.GetData("hidden_partitions", make([]any, 0))
+		hiddenFilesystemsInterface, _ := sm.GetData("hidden_filesystems", make([]any, 0))
+		alternativeNamesInterface, _ := sm.GetData("alternative_names", make(map[string]any, 0))
+
+		// Convert interface slices to string slices
+		hiddenPartitions := make([]string, 0)
+		for _, value := range hiddenPartitionsInterface.([]any) {
+			hiddenPartitions = append(hiddenPartitions, value.(string))
+		}
+		hiddenFilesystems := make([]string, 0)
+		for _, value := range hiddenFilesystemsInterface.([]any) {
+			hiddenFilesystems = append(hiddenFilesystems, value.(string))
+		}
+
+		// Convert interface map to map[string]string
+		alternativeNames := make(map[string]string)
+		for key, value := range alternativeNamesInterface.(map[string]any) {
+			alternativeNames[key] = value.(string)
+		}
+
+		builder := strings.Builder{}
+		partitions := GetMountedPartitions(hiddenPartitions, hiddenFilesystems)
+
+		for i, partition := range partitions {
+			partitionAutoname := ""
+			if altName, ok := alternativeNames[partition.Device]; ok {
+				partitionAutoname = altName
+			} else if partition.Label != "" {
+				partitionAutoname = partition.Label
+			} else {
+				partitionAutoname = partition.MountPoint
+			}
+
+			expanded := os.Expand(sm.Format, func(s string) string {
+				switch s {
+				case "PART_NUM":
+					return strconv.Itoa(i + 1)
+				case "PART_FS":
+					return partition.FileystemType
+				case "PART_DEVICE":
+					return partition.Device
+				case "PART_AUTONAME":
+					return partitionAutoname
+				case "PART_LABEL":
+					return partition.Label
+				case "PART_MOUNTPOINT":
+					return partition.MountPoint
+				case "PART_FREE":
+					return FormatBytes(partition.FreeSize)
+				case "PART_USED":
+					return FormatBytes(partition.UsedSize)
+				case "PART_TOTAL":
+					return FormatBytes(partition.TotalSize)
+				default:
+					return ""
+				}
+			})
+
+			builder.WriteString(expanded + "\n")
+		}
+
+		return builder.String()
+	}}
+	RegisterModule(partitionsModule)
+
+	// Local IP module
+	localIpModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "local_ip", Format: "%3Local IP: %4$LOCAL_IP"}, Execute: func(sm StormfetchModule) string {
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "LOCAL_IP":
+				return GetLocalIP()
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(localIpModule)
+
+	// DEWM module
+	dewmModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "de_wm", Format: "%3${DEWM_TYPE}: %4${DEWM_NAME} ${DEWM_VERSION} ($DISPLAY_PROTOCOL)"}, Execute: func(sm StormfetchModule) string {
+		// Return empty string if currently in TTY
+		if os.Getenv("XDG_SESSION_TYPE") == "" || os.Getenv("XDG_SESSION_TYPE") == "tty" {
+			return ""
+		}
+
+		dewm := GetDEWM()
+
+		// Return empty string if can't detect DE/WM
+		if dewm.Name == "Unknown" {
+			return ""
+		}
+
+		return os.Expand(sm.Format, func(s string) string {
+			switch s {
+			case "DEWM_NAME":
+				return dewm.Name
+			case "DEWM_TYPE":
+				return dewm.Type
+			case "DEWM_VERSION":
+				return dewm.Version
+			case "DISPLAY_PROTOCOL":
+				return GetDisplayProtocol()
+			default:
+				return ""
+			}
+		})
+	}}
+	RegisterModule(dewmModule)
+
+	// Monitors module
+	monitorsModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "monitors", Format: "%3Monitor: %4${MONITOR_WIDTH}x${MONITOR_HEIGHT} ${MONITOR_REFRESH_RATE}Hz"}, Execute: func(sm StormfetchModule) string {
+		builder := strings.Builder{}
+		monitors := GetMonitors()
+
+		for i, monitor := range monitors {
+			expanded := os.Expand(sm.Format, func(s string) string {
+				switch s {
+				case "MONITOR_NUM":
+					return strconv.Itoa(i + 1)
+				case "MONITOR_WIDTH":
+					return strconv.Itoa(monitor.Width)
+				case "MONITOR_HEIGHT":
+					return strconv.Itoa(monitor.Height)
+				case "MONITOR_REFRESH_RATE":
+					return strconv.Itoa(monitor.RefreshRate)
+				default:
+					return ""
+				}
+			})
+
+			builder.WriteString(expanded + "\n")
+		}
+
+		return builder.String()
+	}}
+	RegisterModule(monitorsModule)
+
+	// Custom module
+	customModule := StormfetchModule{stormfetchModuleConfig: stormfetchModuleConfig{Name: "custom"}, Execute: func(sm StormfetchModule) string {
+		shell, _ := sm.GetData("shell", "/bin/sh")
+		commandList, _ := sm.GetData("commands", make([]any, 0))
+
+		// Exeucte all commands
+		commandOutput := make(map[int]string)
+		for i, value := range commandList.([]any) {
+			command, ok := value.(string)
+			if !ok {
+				continue
+			}
+
+			commandOutput[i+1] = runCommand(command, shell.(string))
+		}
+
+		return os.Expand(sm.Format, func(s string) string {
+			if len(s) <= 4 || !strings.HasPrefix(s, "CMD_") {
+				return ""
+			}
+
+			commandIndexStr := strings.Split(s, "CMD_")[1]
+
+			commandIndex, err := strconv.Atoi(commandIndexStr)
+			if err != nil {
+				return ""
+			}
+
+			return commandOutput[commandIndex]
+		})
+	}}
+	RegisterModule(customModule)
+}

src/network.go

@@ -0,0 +1,15 @@
+package main
+
+import "net"
+
+func GetLocalIP() string {
+	conn, err := net.Dial("udp", "8.8.8.8:80")
+	if err != nil {
+		return ""
+	}
+	defer conn.Close()
+
+	localAddr := conn.LocalAddr().(*net.UDPAddr)
+
+	return localAddr.IP.String()
+}

src/partitions.go

@@ -0,0 +1,124 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"slices"
+	"strings"
+	"syscall"
+)
+
+type partition struct {
+	Device        string
+	MountPoint    string
+	Label         string
+	FileystemType string
+	TotalSize     uint64
+	UsedSize      uint64
+	FreeSize      uint64
+}
+
+func GetMountedPartitions(hiddenPartitions, hiddenFilesystems []string) []partition {
+	// Get all filesystem and partition labels
+	fslabels, err := os.ReadDir("/dev/disk/by-label")
+	if err != nil && !os.IsNotExist(err) {
+		return nil
+	}
+	partlabels, err := os.ReadDir("/dev/disk/by-partlabel")
+	if err != nil && !os.IsNotExist(err) {
+		return nil
+	}
+	labels := make(map[string]string)
+	for _, entry := range partlabels {
+		link, err := filepath.EvalSymlinks(filepath.Join("/dev/disk/by-partlabel/", entry.Name()))
+		if err != nil {
+			continue
+		}
+		labels[link] = entry.Name()
+	}
+	for _, entry := range fslabels {
+		link, err := filepath.EvalSymlinks(filepath.Join("/dev/disk/by-label/", entry.Name()))
+		if err != nil {
+			continue
+		}
+		labels[link] = entry.Name()
+	}
+
+	// Get all mounted partitions
+	file, err := os.ReadFile("/proc/mounts")
+	if err != nil {
+		return nil
+	}
+
+	var partitions []partition
+	for _, entry := range strings.Split(string(file), "\n") {
+		fields := strings.Fields(entry)
+		if entry == "" {
+			continue
+		}
+
+		// Skip virtual partitions not under /dev
+		if !strings.HasPrefix(fields[0], "/dev") {
+			continue
+		}
+
+		// Skip partition if explicitly hidden
+		if slices.Contains(hiddenPartitions, fields[0]) {
+			continue
+		}
+
+		// Skip filesystem if explicitely hidden
+		if slices.Contains(hiddenFilesystems, fields[2]) {
+			continue
+		}
+
+		device, err := filepath.EvalSymlinks(fields[0])
+		if err != nil {
+			device = fields[0]
+		}
+
+		p := partition{
+			Device:        device,
+			MountPoint:    fields[1],
+			Label:         "",
+			FileystemType: fields[2],
+			TotalSize:     0,
+			UsedSize:      0,
+			FreeSize:      0,
+		}
+
+		// Skip already added partitions
+		skip := false
+		for _, part := range partitions {
+			if part.Device == p.Device {
+				skip = true
+			}
+		}
+		if skip {
+			continue
+		}
+
+		// Set partition label if available
+		if value, ok := labels[p.Device]; ok {
+			p.Label = strings.ReplaceAll(value, "\\x20", " ")
+		}
+
+		// Get partition total, used and free space
+		buf := new(syscall.Statfs_t)
+		err = syscall.Statfs(p.MountPoint, buf)
+		if err != nil {
+			continue
+		}
+		totalBlocks := buf.Blocks
+		freeBlocks := buf.Bfree
+		usedBlocks := totalBlocks - freeBlocks
+		blockSize := uint64(buf.Bsize)
+
+		p.TotalSize = totalBlocks * blockSize
+		p.FreeSize = freeBlocks * blockSize
+		p.UsedSize = usedBlocks * blockSize
+
+		partitions = append(partitions, p)
+	}
+	return partitions
+}

src/pms.go

@@ -0,0 +1,53 @@
+package main
+
+import (
+	"fmt"
+	"os/exec"
+	"strings"
+)
+
+type PackageManager struct {
+	Name               string
+	ExecutableName     string
+	PackageListCommand string
+}
+
+var PackageManagers = []PackageManager{
+	{Name: "dpkg", ExecutableName: "dpkg", PackageListCommand: "dpkg-query -f '${Package}\\n' -W"},
+	{Name: "pacman", ExecutableName: "pacman", PackageListCommand: "pacman -Q"},
+	{Name: "rpm", ExecutableName: "rpm", PackageListCommand: "rpm -qa"},
+	{Name: "xbps", ExecutableName: "xbps-query", PackageListCommand: "xbps-query -l"},
+	{Name: "bpm", ExecutableName: "bpm", PackageListCommand: "ls /var/lib/bpm/installed/"},
+	{Name: "portage", ExecutableName: "emerge", PackageListCommand: "find /var/db/pkg/*/ -mindepth 1 -maxdepth 1"},
+	{Name: "flatpak", ExecutableName: "flatpak", PackageListCommand: "flatpak list"},
+	{Name: "snap", ExecutableName: "snap", PackageListCommand: "snap list | tail +2"},
+}
+
+func (pm *PackageManager) CountPackages() int {
+	// Return 0 if package manager is not found
+	if _, err := exec.LookPath(pm.ExecutableName); err != nil {
+		return 0
+	}
+
+	output, err := exec.Command("/bin/sh", "-c", pm.PackageListCommand).Output()
+	if err != nil {
+		return 0
+	}
+
+	return strings.Count(string(output), "\n")
+}
+
+func GetInstalledPackages() (ret string) {
+	for _, pm := range PackageManagers {
+		count := pm.CountPackages()
+		if count > 0 {
+			if ret == "" {
+				ret += fmt.Sprintf("%d (%s)", count, pm.Name)
+			} else {
+				ret += fmt.Sprintf(" %d (%s)", count, pm.Name)
+			}
+		}
+	}
+
+	return ret
+}

src/system.go

@@ -0,0 +1,183 @@
+package main
+
+import (
+	"os"
+	"os/exec"
+	"path"
+	"strings"
+	"syscall"
+
+	"github.com/mitchellh/go-ps"
+)
+
+type DistroInfo struct {
+	ID        string
+	LongName  string
+	ShortName string
+}
+
+func GetDistroInfo() DistroInfo {
+	info := DistroInfo{
+		ID:        "unknown",
+		LongName:  "Unknown",
+		ShortName: "Unknown",
+	}
+
+	// Detect release file location
+	var releaseFile string
+	if _, err := os.Stat("/bedrock/etc/os-release"); os.Getenv("BEDROCK_RESTRICT") == "" && err == nil {
+		// Using Bedrock Linux
+		releaseFile = "/bedrock/etc/os-release"
+	} else if _, err := os.Stat("/etc/os-release"); err == nil {
+		// Using a regular linux distribution
+		releaseFile = "/etc/os-release"
+	} else {
+		return info
+	}
+
+	// Read release file
+	releaseMap, err := ReadKeyValueFile(releaseFile)
+	if err != nil {
+		return info
+	}
+
+	// Set struct fields
+	if id, ok := releaseMap["ID"]; ok {
+		info.ID = id
+	}
+	if longName, ok := releaseMap["PRETTY_NAME"]; ok {
+		info.LongName = longName
+	}
+	if shortName, ok := releaseMap["NAME"]; ok {
+		info.ShortName = shortName
+	}
+
+	return info
+}
+
+func GetDistroAsciiArt() string {
+	defaultAscii :=
+		`    .--.
+   |o_o |
+   |:_/ |
+  //   \ \
+ (|     | )
+/'\_   _/'\
+\___)=(___/`
+
+	// Get ascii name to use
+	var asciiName string
+	if Ascii != "" {
+		asciiName = Ascii
+	} else if config.Ascii == "auto" {
+		asciiName = GetDistroInfo().ID
+	} else {
+		asciiName = config.Ascii
+	}
+
+	// Check for ascii art in home directory
+	userConfDir, err := os.UserConfigDir()
+	if err == nil {
+		if _, err := os.Stat(path.Join(userConfDir, "stormfetch/ascii/", asciiName)); err == nil {
+			if bytes, err := os.ReadFile(path.Join(userConfDir, "stormfetch/ascii/", asciiName)); err == nil {
+				return strings.TrimRight(string(bytes), "\n")
+			}
+		}
+	}
+
+	// Check for ascii art in system config directory
+	if _, err := os.Stat(path.Join(SystemConfigDir, "stormfetch/ascii/", asciiName)); err == nil {
+		if bytes, err := os.ReadFile(path.Join(SystemConfigDir, "stormfetch/ascii/", asciiName)); err == nil {
+			return strings.TrimRight(string(bytes), "\n")
+		}
+	}
+
+	return defaultAscii
+}
+
+func GetArch() string {
+	uname := syscall.Utsname{}
+	err := syscall.Uname(&uname)
+	if err != nil {
+		return "unknown"
+	}
+
+	var byteString [65]byte
+	var indexLength int
+	for ; uname.Machine[indexLength] != 0; indexLength++ {
+		byteString[indexLength] = uint8(uname.Machine[indexLength])
+	}
+	return string(byteString[:indexLength])
+}
+
+func GetKernel() (string, string) {
+	uname := syscall.Utsname{}
+	err := syscall.Uname(&uname)
+	if err != nil {
+		return "unknown", "unknown"
+	}
+
+	var kernelNameByteString [65]byte
+	var kernelNameLength int
+	for ; uname.Sysname[kernelNameLength] != 0; kernelNameLength++ {
+		kernelNameByteString[kernelNameLength] = uint8(uname.Sysname[kernelNameLength])
+	}
+
+	var kernelReleaseByteString [65]byte
+	var kernelReleaseLength int
+	for ; uname.Release[kernelReleaseLength] != 0; kernelReleaseLength++ {
+		kernelReleaseByteString[kernelReleaseLength] = uint8(uname.Release[kernelReleaseLength])
+	}
+
+	return string(kernelNameByteString[:kernelNameLength]), string(kernelReleaseByteString[:kernelReleaseLength])
+}
+
+func GetInitSystem() string {
+	process, err := ps.FindProcess(1)
+	if err != nil {
+		return ""
+	}
+
+	// Special cases
+	// OpenRC check
+	if _, err := os.Stat("/usr/sbin/openrc"); err == nil {
+		return "OpenRC " + runCommand("openrc --version | awk '{print $3}'", "/bin/sh")
+	}
+
+	// Default PID 1 process name checking
+	switch process.Executable() {
+	case "systemd":
+		return "Systemd " + runCommand("systemctl --version | head -n1 | awk '{print $2}'", "/bin/sh")
+	case "runit":
+		return "Runit"
+	case "dinit":
+		return "Dinit " + runCommand("dinit --version | head -n1 | awk '{print substr($3, 1, length($3)-1)}'", "/bin/sh")
+	case "enit":
+		return "Enit " + runCommand("enit --version | awk '{print $3}'", "/bin/sh")
+	default:
+		return process.Executable()
+	}
+}
+
+func GetLibc() string {
+	checkLibcOutput, err := exec.Command("ldd", "/usr/bin/ls").Output()
+	if err != nil {
+		return "Unknown"
+	}
+
+	if strings.Contains(string(checkLibcOutput), "ld-musl") {
+		// Using Musl Libc
+		output, _ := exec.Command("ldd").CombinedOutput()
+		return "Musl " + strings.TrimPrefix(strings.Split(strings.TrimSpace(string(output)), "\n")[1], "Version ")
+	} else {
+		// Using Glibc
+		cmd := exec.Command("ldd", "--version")
+		output, err := cmd.Output()
+		if err != nil {
+			return "Glibc"
+		}
+		outputSplit := strings.Split(strings.Split(strings.TrimSpace(string(output)), "\n")[0], " ")
+		ver := outputSplit[len(outputSplit)-1]
+		return "Glibc " + ver
+	}
+}

src/user.go

@@ -0,0 +1,172 @@
+package main
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+	"slices"
+	"strconv"
+	"strings"
+
+	"github.com/mitchellh/go-ps"
+)
+
+type DEWM struct {
+	Name    string
+	Type    string
+	Version string
+}
+
+func GetShell() string {
+	file, err := os.ReadFile("/etc/passwd")
+	if err != nil {
+		return ""
+	}
+	str := string(file)
+	shell := ""
+
+	for _, line := range strings.Split(str, "\n") {
+		if strings.TrimSpace(line) == "" {
+			continue
+		}
+		userInfo := strings.Split(line, ":")
+		if userInfo[2] == strconv.Itoa(os.Getuid()) {
+			shell = userInfo[6]
+		}
+	}
+	shellName := filepath.Base(shell)
+	switch shellName {
+	case "dash":
+		return "Dash"
+	case "bash":
+		return "Bash " + runCommand("echo $BASH_VERSION", "/bin/sh")
+	case "zsh":
+		return "Zsh " + runCommand("$SHELL --version | awk '{print $2}'", "/bin/sh")
+	case "fish":
+		return "Fish " + runCommand("$SHELL --version | awk '{print $3}'", "/bin/sh")
+	case "nu":
+		return "Nushell " + runCommand("$SHELL --version", "/bin/sh")
+	default:
+		return "Unknown"
+	}
+}
+
+func GetDEWM() DEWM {
+	processes, err := ps.Processes()
+	if err != nil {
+		log.Fatalf("Error: could not get processes: %s", err)
+	}
+	var executables []string
+	for _, process := range processes {
+		executables = append(executables, process.Executable())
+	}
+
+	processExists := func(process string) bool {
+		return slices.Contains(executables, process)
+	}
+	if processExists("plasmashell") {
+		dewm := DEWM{
+			Name:    "KDE Plasma",
+			Type:    "DE",
+			Version: runCommand("plasmashell --version | awk '{print $2}'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("gnome-session") {
+		dewm := DEWM{
+			Name:    "Gnome",
+			Type:    "DE",
+			Version: runCommand("gnome-shell --version | awk '{print $3}'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("xfce4-session") {
+		dewm := DEWM{
+			Name:    "XFCE",
+			Type:    "DE",
+			Version: runCommand("xfce4-session --version | head -n1 | awk '{print $2}'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("cinnamon") {
+		dewm := DEWM{
+			Name:    "Cinnamon",
+			Type:    "DE",
+			Version: runCommand("cinnamon --version | awk '{print $3}'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("mate-panel") {
+		dewm := DEWM{
+			Name:    "MATE",
+			Type:    "DE",
+			Version: runCommand("mate-about --version | awk '{print $4}'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("lxsession") {
+		dewm := DEWM{
+			Name:    "LXDE",
+			Type:    "DE",
+			Version: "",
+		}
+		return dewm
+	} else if processExists("lxqt-session") {
+		dewm := DEWM{
+			Name:    "LXQt",
+			Type:    "DE",
+			Version: runCommand("lxqt-session --version | head -n1 | awk '{print $2}'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("i3") || processExists("i3-with-shmlog") {
+		dewm := DEWM{
+			Name:    "i3",
+			Type:    "WM",
+			Version: runCommand("i3 --version | awk '{print $3}'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("sway") {
+		dewm := DEWM{
+			Name:    "Sway",
+			Type:    "WM",
+			Version: runCommand("sway --version | awk '{print $3}'", "/bin/sh"),
+		}
+		if runCommand("sway --version | awk '{print $1}'", "/bin/sh") == "swayfx" {
+			dewm.Name = "SwayFX"
+		} else {
+			dewm.Name = "Sway"
+		}
+		return dewm
+	} else if processExists("bspwm") {
+		dewm := DEWM{
+			Name:    "Bspwm",
+			Type:    "WM",
+			Version: runCommand("bspwm -v", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("Hyprland") {
+		dewm := DEWM{
+			Name:    "Hyprland",
+			Type:    "WM",
+			Version: runCommand("hyprctl version | sed -n 3p | awk '{print $2}' | tr -d 'v,'", "/bin/sh"),
+		}
+		return dewm
+	} else if processExists("icewm-session") {
+		dewm := DEWM{
+			Name:    "IceWM",
+			Type:    "WM",
+			Version: runCommand("icewm --version | awk '{print $2}'", "/bin/sh"),
+		}
+		return dewm
+	}
+	return DEWM{
+		Name:    "Unknown",
+		Type:    "Unknown",
+		Version: "Unknown",
+	}
+}
+
+func GetDisplayProtocol() string {
+	protocol := os.Getenv("XDG_SESSION_TYPE")
+	if protocol == "x11" {
+		return "X11"
+	} else if protocol == "wayland" {
+		return "Wayland"
+	}
+	return ""
+}

src/utils.go

@@ -0,0 +1,74 @@
+package main
+
+import (
+	"fmt"
+	"math"
+	"os"
+	"os/exec"
+	"regexp"
+	"strings"
+)
+
+func FormatBytes(bytes uint64) string {
+	var suffixes [6]string
+	suffixes[0] = "B"
+	suffixes[1] = "KiB"
+	suffixes[2] = "MiB"
+	suffixes[3] = "GiB"
+	suffixes[4] = "TiB"
+	suffixes[5] = "PiB"
+
+	bf := float64(bytes)
+	for _, unit := range suffixes {
+		if math.Abs(bf) < 1024.0 {
+			return fmt.Sprintf("%3.1f %s", bf, unit)
+		}
+		bf /= 1024.0
+	}
+	return fmt.Sprintf("%.1fYiB", bf)
+}
+
+func StripAnsii(str string) string {
+	const ansi = "[\u001B\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[a-zA-Z\\d]*)*)?\u0007)|(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PRZcf-ntqry=><~]))"
+	var re = regexp.MustCompile(ansi)
+	return re.ReplaceAllString(str, "")
+}
+
+func ReadKeyValueFile(filepath string) (map[string]string, error) {
+	ret := make(map[string]string)
+	if _, err := os.Stat(filepath); err != nil {
+		return nil, err
+	}
+	bytes, err := os.ReadFile(filepath)
+	if err != nil {
+		return nil, err
+	}
+	str := string(bytes)
+	lines := strings.Split(str, "\n")
+	for _, line := range lines {
+		if len(strings.Split(line, "=")) >= 2 {
+			key := strings.SplitN(line, "=", 2)[0]
+			value := strings.SplitN(line, "=", 2)[1]
+			if strings.HasPrefix(value, "\"") && strings.HasSuffix(value, "\"") {
+				value = value[1 : len(value)-1]
+			}
+			ret[key] = value
+		}
+	}
+	return ret, nil
+}
+
+func runCommand(command string, shell string) string {
+	cmd := exec.Command(shell, "-c", command)
+	workdir, err := os.Getwd()
+	if err != nil {
+		return ""
+	}
+	cmd.Dir = workdir
+	cmd.Env = os.Environ()
+	out, err := cmd.Output()
+	if err != nil {
+		return ""
+	}
+	return strings.TrimSpace(string(out))
+}

utils.go

@@ -1,433 +0,0 @@
-package main
-
-import (
-	"bufio"
-	"fmt"
-	"github.com/BurntSushi/xgb"
-	"github.com/BurntSushi/xgb/xinerama"
-	"github.com/jackmordaunt/ghw"
-	"github.com/mitchellh/go-ps"
-	"log"
-	"math"
-	"os"
-	"os/exec"
-	"path"
-	"path/filepath"
-	"regexp"
-	"slices"
-	"strconv"
-	"strings"
-	"syscall"
-)
-
-type DistroInfo struct {
-	ID        string
-	LongName  string
-	ShortName string
-}
-
-func getDistroInfo() DistroInfo {
-	distroID := ""
-	var releaseMap = make(map[string]string)
-	if _, err := os.Stat("/etc/os-release"); err == nil {
-		releaseMap, err = ReadKeyValueFile("/etc/os-release")
-		if err != nil {
-			return DistroInfo{
-				ID:        "unknown",
-				LongName:  "Unknown",
-				ShortName: "Unknown",
-			}
-		}
-		if value, ok := releaseMap["ID"]; ok {
-			distroID = value
-		}
-	}
-
-	switch distroID {
-	default:
-		if id, ok := releaseMap["ID"]; ok {
-			if longName, ok := releaseMap["PRETTY_NAME"]; ok {
-				if shortName, ok := releaseMap["NAME"]; ok {
-					return DistroInfo{
-						ID:        id,
-						LongName:  longName,
-						ShortName: shortName,
-					}
-				}
-			}
-		}
-		return DistroInfo{
-			ID:        "unknown",
-			LongName:  "Unknown",
-			ShortName: "Unknown",
-		}
-	}
-}
-
-func getDistroAsciiArt() string {
-	defaultAscii :=
-		`    .--.
-   |o_o |
-   |:_/ |
-  //   \ \
- (|     | )
-/'\_   _/'\
-\___)=(___/ `
-	var id string
-	if config.Ascii == "auto" {
-		id = getDistroInfo().ID
-	} else {
-		id = config.Ascii
-	}
-	userConfDir, err := os.UserConfigDir()
-	if err != nil {
-		if _, err := os.Stat(path.Join("/etc/stormfetch/ascii/", id)); err == nil {
-			bytes, err := os.ReadFile(path.Join("/etc/stormfetch/ascii/", id))
-			if err != nil {
-				return defaultAscii
-			}
-			return string(bytes)
-		} else {
-			return defaultAscii
-		}
-	}
-	if _, err := os.Stat(path.Join(userConfDir, "stormfetch/ascii/", id)); err == nil {
-		bytes, err := os.ReadFile(path.Join(userConfDir, "stormfetch/ascii/", id))
-		if err != nil {
-			return defaultAscii
-		}
-		return string(bytes)
-	} else if _, err := os.Stat(path.Join("/etc/stormfetch/ascii/", id)); err == nil {
-		bytes, err := os.ReadFile(path.Join("/etc/stormfetch/ascii/", id))
-		if err != nil {
-			return defaultAscii
-		}
-		return string(bytes)
-	} else {
-		return defaultAscii
-	}
-}
-
-func getCPUName() string {
-	cpu, err := ghw.CPU()
-	if err != nil {
-		return ""
-	}
-	if len(cpu.Processors) == 0 {
-		return ""
-	}
-	return cpu.Processors[0].Model
-}
-
-func getCPUThreads() int {
-	cpu, err := ghw.CPU()
-	if err != nil {
-		return 0
-	}
-	return int(cpu.TotalThreads)
-}
-
-func getGPUName() string {
-	null, _ := os.Open(os.DevNull)
-	serr := os.Stderr
-	os.Stderr = null
-	gpu, err := ghw.GPU()
-	defer null.Close()
-	os.Stderr = serr
-	if err != nil {
-		return ""
-	}
-	for _, graphics := range gpu.GraphicsCards {
-		if graphics.DeviceInfo != nil {
-			return graphics.DeviceInfo.Product.Name
-		}
-	}
-	return ""
-}
-
-type Memory struct {
-	MemTotal     int
-	MemFree      int
-	MemAvailable int
-}
-
-func GetMemoryInfo() *Memory {
-	toInt := func(raw string) int {
-		if raw == "" {
-			return 0
-		}
-		res, err := strconv.Atoi(raw)
-		if err != nil {
-			panic(err)
-		}
-		return res
-	}
-
-	parseLine := func(raw string) (key string, value int) {
-		text := strings.ReplaceAll(raw[:len(raw)-2], " ", "")
-		keyValue := strings.Split(text, ":")
-		return keyValue[0], toInt(keyValue[1])
-	}
-
-	if _, err := os.Stat("/proc/meminfo"); err != nil {
-		return nil
-	}
-	file, err := os.Open("/proc/meminfo")
-	if err != nil {
-		panic(err)
-	}
-	defer file.Close()
-	bufio.NewScanner(file)
-	scanner := bufio.NewScanner(file)
-	res := Memory{}
-	for scanner.Scan() {
-		key, value := parseLine(scanner.Text())
-		switch key {
-		case "MemTotal":
-			res.MemTotal = value / 1024
-		case "MemFree":
-			res.MemFree = value / 1024
-		case "MemAvailable":
-			res.MemAvailable = value / 1024
-		}
-	}
-	return &res
-}
-
-func GetShell() string {
-	runCommand := func(command string) string {
-		cmd := exec.Command("/bin/bash", "-c", command)
-		workdir, err := os.Getwd()
-		if err != nil {
-			return ""
-		}
-		cmd.Dir = workdir
-		cmd.Env = os.Environ()
-		out, err := cmd.Output()
-		if err != nil {
-			return ""
-		}
-		return strings.TrimSpace(string(out))
-	}
-	file, err := os.ReadFile("/etc/passwd")
-	if err != nil {
-		return ""
-	}
-	str := string(file)
-	shell := ""
-
-	for _, line := range strings.Split(str, "\n") {
-		if strings.TrimSpace(line) == "" {
-			continue
-		}
-		userInfo := strings.Split(line, ":")
-		if userInfo[2] == strconv.Itoa(os.Getuid()) {
-			shell = userInfo[6]
-		}
-	}
-	shellName := filepath.Base(shell)
-	switch shellName {
-	case "dash":
-		return "Dash"
-	case "bash":
-		return "Bash " + runCommand("echo $BASH_VERSION")
-	case "zsh":
-		return "Zsh " + runCommand("$SHELL --version | awk '{print $2}'")
-	case "fish":
-		return "Fish " + runCommand("$SHELL --version | awk '{print $3}'")
-	default:
-		return "Unknown"
-	}
-}
-
-func GetDEWM() string {
-	processes, err := ps.Processes()
-	if err != nil {
-		log.Fatal(err)
-	}
-	var executables []string
-	for _, process := range processes {
-		executables = append(executables, process.Executable())
-	}
-
-	processExists := func(process string) bool {
-		return slices.Contains(executables, process)
-	}
-	runCommand := func(command string) string {
-		cmd := exec.Command("/bin/bash", "-c", command)
-		workdir, err := os.Getwd()
-		if err != nil {
-			return ""
-		}
-		cmd.Dir = workdir
-		cmd.Env = os.Environ()
-		out, err := cmd.Output()
-		if err != nil {
-			return ""
-		}
-		return strings.TrimSpace(string(out))
-	}
-
-	if processExists("plasmashell") {
-		return "KDE Plasma " + runCommand("plasmashell --version | awk '{print $2}'")
-	} else if processExists("gnome-session") {
-		return "Gnome " + runCommand("gnome-shell --version | awk '{print $3}'")
-	} else if processExists("xfce4-session") {
-		return "XFCE " + runCommand("xfce4-session --version | grep xfce4-session | awk '{print $2}'")
-	} else if processExists("cinnamon") {
-		return "Cinnamon " + runCommand("cinnamon --version | awk '{print $3}'")
-	} else if processExists("mate-panel") {
-		return "MATE " + runCommand("mate-about --version | awk '{print $4}'")
-	} else if processExists("lxsession") {
-		return "LXDE"
-	} else if processExists("sway") {
-		return "Sway " + runCommand("sway --version | awk '{print $3}'")
-	} else if processExists("bspwm") {
-		return "Bspwm " + runCommand("bspwm -v")
-	} else if processExists("icewm-session") {
-		return "IceWM " + runCommand("icewm --version | awk '{print $2}'")
-	}
-	return ""
-}
-
-func GetDisplayProtocol() string {
-	protocol := os.Getenv("XDG_SESSION_TYPE")
-	if protocol == "x11" {
-		return "X11"
-	} else if protocol == "wayland" {
-		return "Wayland"
-	}
-	return ""
-}
-
-func getMonitorResolution() []string {
-	var monitors []string
-	if GetDisplayProtocol() == "X11" {
-		conn, err := xgb.NewConn()
-		if err != nil {
-			return nil
-		}
-		err = xinerama.Init(conn)
-		if err != nil {
-			return nil
-		}
-		reply, _ := xinerama.QueryScreens(conn).Reply()
-		conn.Close()
-		for _, screen := range reply.ScreenInfo {
-			monitors = append(monitors, strconv.Itoa(int(screen.Width))+"x"+strconv.Itoa(int(screen.Height)))
-		}
-	}
-	return monitors
-}
-
-type partition struct {
-	Device     string
-	MountPoint string
-	TotalSize  uint64
-	UsedSize   uint64
-	FreeSize   uint64
-}
-
-func getMountedPartitions() []partition {
-	entries, err := os.ReadDir("/dev/disk/by-partuuid")
-	if err != nil {
-		return nil
-	}
-
-	bytes, err := os.ReadFile("/proc/mounts")
-	if err != nil {
-		return nil
-	}
-	mountsRaw := strings.Split(string(bytes), "\n")
-	mounts := make(map[string]string)
-	for i := 0; i < len(mountsRaw); i++ {
-		split := strings.Split(mountsRaw[i], " ")
-		if len(split) <= 2 {
-			continue
-		}
-		mounts[split[0]] = split[1]
-	}
-
-	var partitions []partition
-	for _, entry := range entries {
-		link, err := filepath.EvalSymlinks(filepath.Join("/dev/disk/by-partuuid/", entry.Name()))
-		if err != nil {
-			continue
-		}
-		if _, ok := mounts[link]; !ok {
-			continue
-		}
-		p := partition{
-			link,
-			mounts[link],
-			0,
-			0,
-			0,
-		}
-		buf := new(syscall.Statfs_t)
-		err = syscall.Statfs(p.MountPoint, buf)
-		if err != nil {
-			log.Fatal(err)
-		}
-		totalBlocks := buf.Blocks
-		freeBlocks := buf.Bfree
-		usedBlocks := totalBlocks - freeBlocks
-		blockSize := uint64(buf.Bsize)
-
-		p.TotalSize = totalBlocks * blockSize
-		p.FreeSize = freeBlocks * blockSize
-		p.UsedSize = usedBlocks * blockSize
-
-		partitions = append(partitions, p)
-	}
-	return partitions
-}
-
-func FormatBytes(bytes uint64) string {
-	var suffixes [6]string
-	suffixes[0] = "B"
-	suffixes[1] = "KiB"
-	suffixes[2] = "MiB"
-	suffixes[3] = "GiB"
-	suffixes[4] = "TiB"
-	suffixes[5] = "PiB"
-
-	bf := float64(bytes)
-	for _, unit := range suffixes {
-		if math.Abs(bf) < 1024.0 {
-			return fmt.Sprintf("%3.1f %s", bf, unit)
-		}
-		bf /= 1024.0
-	}
-	return fmt.Sprintf("%.1fYiB", bf)
-}
-
-func StripAnsii(str string) string {
-	const ansi = "[\u001B\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[a-zA-Z\\d]*)*)?\u0007)|(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PRZcf-ntqry=><~]))"
-	var re = regexp.MustCompile(ansi)
-	return re.ReplaceAllString(str, "")
-}
-
-func ReadKeyValueFile(filepath string) (map[string]string, error) {
-	ret := make(map[string]string)
-	if _, err := os.Stat(filepath); err != nil {
-		return nil, err
-	}
-	bytes, err := os.ReadFile(filepath)
-	if err != nil {
-		return nil, err
-	}
-	str := string(bytes)
-	lines := strings.Split(str, "\n")
-	for _, line := range lines {
-		if len(strings.Split(line, "=")) >= 2 {
-			key := strings.SplitN(line, "=", 2)[0]
-			value := strings.SplitN(line, "=", 2)[1]
-			if strings.HasPrefix(value, "\"") && strings.HasSuffix(value, "\"") {
-				value = value[1 : len(value)-1]
-			}
-			ret[key] = value
-		}
-	}
-	return ret, nil
-}