Close xdg connection after querying

Added MOUNTED_PARTITIONS bash variable
Added PARTITION<n>_DEVICE bash variable
Added PARTITION<n>_MOUNTPOINT bash variable
Added PARTITION<n>_TOTAL_SIZE bash variable
Added PARTITION<n>_USED_SIZE bash variable
Added PARTITION<n>_FREE_SIZE bash variable

Makefile

@@ -13,7 +13,7 @@ endif
 
 build:
 	mkdir -p build
-	$(GO) build -o build/stormfetch stormfetch
+	$(GO) build -ldflags "-w" -o build/stormfetch stormfetch
 
 install: build/stormfetch config/
 	mkdir -p $(DESTDIR)$(BINDIR)
@@ -29,5 +29,8 @@ compress: build/stormfetch config/
 	tar --owner=root --group=root -czf stormfetch.tar.gz stormfetch
 	rm -r stormfetch
 
+run: build/stormfetch
+	build/stormfetch
+
 clean:
 	rm -r build/

README.md

@@ -9,15 +9,15 @@
 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.
 Stormfetch is still in beta, so distro compatibility is limited. If you would like to contribute ASCII art or add other compatibility features feel free to create a pull request or notify me through GitLab Issues.
 
+### How it looks
+![Stormfetch gif](media/stormfetch.gif)
+
 ### Installation Guide
-#### AUR
-- If you are on Arch Linux, you can use your favorite AUR manager to install the `stormfetch` package.
+#### Using a package manager
+- Arch Linux: You may use your favorite AUR manager to install the `stormfetch` package
 #### Building from source
 - Download `go` from your package manager or from the go website
 - Download `make` from your package manager
-- (Optional) Download `xorg-xhost` from your package manager to display DE/WM and monitor information
-- (Optional) Download `xorg-xdpyinfo` from your package manager to display screen resolution
-- (Optional) Download `lshw` from your package manager to display GPU information
 - Run the following command to compile the project
 ```
 make

config/fetch_script.sh

@@ -4,18 +4,22 @@ 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}$(get_shell)"
-echo -e "${C3}CPU: ${C4}$(get_cpu_name) ($(nproc) threads)"
-if command_exists lshw; then
-  echo -e "${C3}GPU: ${C4}$(lshw -class display 2> /dev/null | grep 'product' | cut -d":" -f2 | xargs)"
+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
-echo -e "${C3}Memory: ${C4}$(get_used_mem) MiB / $(get_total_mem) MiB"
-if xhost >& /dev/null ; then
-  if get_de_wm &> /dev/null; then
-    echo -e "${C3}DE/WM: ${C4}$(get_de_wm)"
-  fi
-  if command_exists xdpyinfo ; then
-    echo -e "${C3}Screen Resolution: ${C4}$(get_screen_resolution)"
-  fi
-fi
-
+if [ ! -z "$DE_WM" ]; then echo -e "${C3}DE/WM: ${C4}${DE_WM}"; fi

config/fetch_script_functions.sh

@@ -9,77 +9,6 @@ command_exists() {
     fi
 }
 
-get_shell() {
-    case ${SHELL##*/} in
-    "")
-        echo "Unknown"
-        ;;
-    sh|ash|dash|es)
-        echo "${SHELL##*/} $(${SHELL##*/} --version)"
-        ;;
-    bash)
-        echo "${SHELL##*/} $(${SHELL##*/} -c "echo "'$BASH_VERSION')"
-        ;;
-    *)
-        SHELL_NAME=${SHELL##*/}
-        SHELL_VERSION="$($SHELL --version)"
-        SHELL_VERSION=${SHELL_VERSION//","}
-        SHELL_VERSION=${SHELL_VERSION//" "}
-        SHELL_VERSION=${SHELL_VERSION//"version"}
-        SHELL_VERSION=${SHELL_VERSION//"$SHELL_NAME"}
-        echo "$SHELL_NAME $SHELL_VERSION"
-        unset SHELL_NAME
-        unset SHELL_VERSION
-        ;;
-    esac
-}
-
-get_cpu_name() {
-    grep -m1 "model name" /proc/cpuinfo | cut -d: -f2 | xargs
-}
-
-get_total_mem() {
-     free --mebi -t | grep 'Total' | tr -s ' ' | cut -d" " -f2
-}
-
-get_free_mem() {
-    free --mebi -t | grep 'Total' | tr -s ' ' | cut -d" " -f4
-}
-
-get_used_mem() {
-    free --mebi -t | grep 'Total' | tr -s ' ' | cut -d" " -f3
-}
-
-get_de_wm() {
-    if ps -e | grep "plasmashell" &> /dev/null ; then
-        echo "KDE Plasma $(plasmashell --version | awk '{print $2}')"
-    elif ps -e | grep "gnome-session" &> /dev/null ; then
-        echo "Gnome $(gnome-shell --version | awk '{print $3}')"
-    elif ps -e | grep "xfce4-session" &> /dev/null ; then
-        echo "XFCE $(xfce4-session --version | grep xfce4-session | awk '{print $2}')"
-    elif ps -e | grep "cinnamon" &> /dev/null ; then
-        echo "Cinnamon $(cinnamon --version | awk '{print $2}')"
-    elif ps -e | grep "mate-panel" &> /dev/null ; then
-        echo "Mate $(mate-about --version | awk '{print $4}')"
-    elif ps -e | grep "lxsession" &> /dev/null ; then
-        echo "LXDE"
-    elif ps -e | grep "sway" &> /dev/null ; then
-        echo "Sway $(sway --version | awk '{print $2}')"
-    elif ps -e | grep "icewm-session" &> /dev/null ; then
-        echo "IceWM $(icewm --version | awk '{print $2}' | sed 's/,//g')"
-    elif [ ! -z $DESKTOP_SESSION ]; then
-        echo "$DESKTOP_SESSION"
-    else
-        return 1
-    fi
-}
-
-get_screen_resolution() {
-    if xhost >& /dev/null && command_exists xdpyinfo; then
-        xdpyinfo | grep dimensions | tr -s ' ' | cut -d " " -f3
-    fi
-}
-
 get_packages() {
     ARRAY=()
     if command_exists dpkg; then

go.mod

@@ -2,4 +2,18 @@ module stormfetch
 
 go 1.22
 
-require gopkg.in/yaml.v3 v3.0.1 // indirect
+require (
+	github.com/BurntSushi/xgb v0.0.0-20210121224620-deaf085860bc
+	github.com/jackmordaunt/ghw v1.0.4
+	github.com/mitchellh/go-ps v1.0.0
+	gopkg.in/yaml.v3 v3.0.1
+)
+
+require (
+	github.com/go-ole/go-ole v1.2.6 // indirect
+	github.com/jackmordaunt/pcidb v1.0.1 // indirect
+	github.com/jackmordaunt/wmi v1.2.4 // indirect
+	github.com/kr/pretty v0.1.0 // indirect
+	golang.org/x/sys v0.3.0 // indirect
+	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
+)

go.sum

@@ -1,3 +1,25 @@
+github.com/BurntSushi/xgb v0.0.0-20210121224620-deaf085860bc h1:7D+Bh06CRPCJO3gr2F7h1sriovOZ8BMhca2Rg85c2nk=
+github.com/BurntSushi/xgb v0.0.0-20210121224620-deaf085860bc/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo=
+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=
+github.com/jackmordaunt/ghw v1.0.4/go.mod h1:4dReYvJ36CoAzIxlEx8du25Qi/YqKYvEGE9QJoRXiK8=
+github.com/jackmordaunt/pcidb v1.0.1 h1:uLLZa6kD5P39r2cwMyJJkxmuHfH9Wq19gYQEbYcB0Z4=
+github.com/jackmordaunt/pcidb v1.0.1/go.mod h1:OMmhrZOZVu2hYXhBDZXddypxwKR/dp4DbIgzCkQDxdQ=
+github.com/jackmordaunt/wmi v1.2.4 h1:/XyuMiKby0qXNQp1j0uU0JqTWLM0QGOVok1Hf5Yagtg=
+github.com/jackmordaunt/wmi v1.2.4/go.mod h1:K/+FH4467Vk2fJXhYQTZzhctkCMMR7PwHqKbnizFySo=
+github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
+github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
+github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
+github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
+github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
+github.com/mitchellh/go-ps v1.0.0 h1:i6ampVEEF4wQFF+bkYfwYgY+F/uYJDktmvLPf7qIgjc=
+github.com/mitchellh/go-ps v1.0.0/go.mod h1:J4lOc8z8yJs6vUwklHw2XEIiT4z4C40KtWVN3nvg8Pg=
+golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.3.0 h1:w8ZOecv6NaNa/zC8944JTU3vz4u6Lagfk4RPQxv92NQ=
+golang.org/x/sys v0.3.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY=
+gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

main.go

@@ -31,12 +31,6 @@ type StormfetchConfig struct {
 	DependencyWarning bool   `yaml:"dependency_warning"`
 }
 
-type DistroInfo struct {
-	ID        string
-	LongName  string
-	ShortName string
-}
-
 func main() {
 	readConfig()
 }
@@ -80,7 +74,7 @@ func readConfig() {
 	}
 	// Show Dependency warning if enabled
 	if config.DependencyWarning {
-		dependencies := []string{"lshw", "xhost", "xdpyinfo"}
+		var dependencies []string
 		var missing []string
 		for _, depend := range dependencies {
 			if _, err := os.Stat(path.Join("/usr/bin/", depend)); err != nil {
@@ -108,9 +102,9 @@ func readConfig() {
 		}
 	}
 	setColorMap()
-	ascii := ""
-	if strings.HasPrefix(getDistroAsciiArt(), "#/") {
-		firstLine := strings.Split(getDistroAsciiArt(), "\n")[0]
+	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 {
@@ -126,27 +120,26 @@ func readConfig() {
 			}
 			setColorMap()
 		}
-		ascii = os.Expand(getDistroAsciiArt(), func(s string) string {
+		ascii = os.Expand(ascii, func(s string) string {
 			return colorMap[s]
 		})
 		ascii = strings.TrimPrefix(ascii, firstLine+"\n")
 	} else {
-		ascii = os.Expand(getDistroAsciiArt(), func(s string) string {
+		ascii = os.Expand(ascii, func(s string) string {
 			return colorMap[s]
 		})
 	}
 	asciiSplit := strings.Split(ascii, "\n")
-	asciiNoColor := stripAnsii(ascii)
+	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))
 	}
-	cmd.Env = append(cmd.Env, "DISTRO_LONG_NAME="+getDistroInfo().LongName)
-	cmd.Env = append(cmd.Env, "DISTRO_SHORT_NAME="+getDistroInfo().ShortName)
 	out, err := cmd.Output()
 	if err != nil {
 		log.Fatal(err)
@@ -182,114 +175,48 @@ func readConfig() {
 	fmt.Println(final)
 }
 
-func readKeyValueFile(filepath string) (map[string]string, error) {
-	ret := make(map[string]string)
-	if _, err := os.Stat(filepath); err != nil {
-		return nil, err
+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)
 	}
-	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
+	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)
 		}
 	}
-	return ret, nil
-}
-
-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
+	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()
+	}
 
-	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",
-		}
+	var ret = make([]string, len(env))
+	i := 0
+	for key, value := range env {
+		ret[i] = fmt.Sprintf("%s=%s", key, value)
+		i++
 	}
-}
-
-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 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, "")
+	return ret
 }

utils.go

@@ -0,0 +1,433 @@
+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
+}