stormfetch/src/user.go

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 ""
}