128 lines
3.7 KiB
Go
128 lines
3.7 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/exhq/neowofetch/data"
|
|
"github.com/exhq/neowofetch/utils"
|
|
)
|
|
|
|
var isuwuified bool = true
|
|
|
|
func checkforconfigfolder() {
|
|
_, folder := os.Stat(data.GetHome() + "/.config")
|
|
if os.IsNotExist(folder) {
|
|
os.Mkdir(data.GetHome()+"/.config", os.ModePerm)
|
|
}
|
|
}
|
|
|
|
func getShellVersion(shellCommand string) string {
|
|
// https://www.youtube.com/watch?v=YPN0qhSyWy8
|
|
cmd := shellCommand + " --version | grep -o -E \"([0-9]\\.?)*\" | head -n1"
|
|
out, _ := exec.Command("bash", "-c", cmd).Output()
|
|
return filepath.Base(shellCommand) + " " + strings.ReplaceAll(string(out), "\n", "")
|
|
}
|
|
|
|
func handleConfig() {
|
|
_, folder := os.Stat(filepath.Dir(data.GetConfigFile()))
|
|
_, file := os.Stat(data.GetConfigFile())
|
|
if os.IsNotExist(folder) {
|
|
os.Mkdir(filepath.Dir(data.GetConfigFile()), os.ModePerm)
|
|
}
|
|
defaultconfig := "println green neOwOfetchh 🔥\ninfo white username\nprint blue @\ninfoln blue hostname\nprint white|space=12 uptime:\ninfoln red uptime\nprint white|space=12 shell:\ninfoln blue shell\nprint white|space=12 distro:\ninfoln blue distro\nprint white|space=12 terminal:\ninfoln blue terminal\nprint white|space=12 WM:\ninfoln blue wm\nprint white|space=12 memory:\ninfo blue memoryUsed\nprint white /\ninfoln blue memoryAll"
|
|
if os.IsNotExist(file) {
|
|
f, _ := os.Create(data.GetConfigFile())
|
|
_, _ = f.WriteString(defaultconfig)
|
|
}
|
|
body, _ := ioutil.ReadFile(data.GetConfigFile())
|
|
sbody := (string(body))
|
|
if utils.Defaultconf {
|
|
sbody = defaultconfig
|
|
}
|
|
fbody := strings.Split(sbody, "\n")
|
|
for _, line := range fbody {
|
|
word := strings.Split(line, " ")
|
|
if len(word) < 3 {
|
|
continue
|
|
}
|
|
action := word[0]
|
|
color := word[1]
|
|
rest := strings.Join(word[2:], " ")
|
|
handlePrint(action, color, rest)
|
|
}
|
|
}
|
|
|
|
func handlePrint(action, format string, rest string) {
|
|
if action == "print" {
|
|
utils.CutePrint(rest, format)
|
|
} else if action == "println" {
|
|
utils.CutePrint(rest, format)
|
|
utils.CuteNewLine()
|
|
} else if action == "info" || action == "infoln" {
|
|
switch rest {
|
|
case "distro":
|
|
if utils.Asciiforced {
|
|
utils.CutePrint(utils.Forceddistro, format)
|
|
} else {
|
|
utils.CutePrint(data.GetDistro(), format)
|
|
}
|
|
case "username":
|
|
utils.CutePrint(data.GetUsername(), format)
|
|
case "uptime":
|
|
no, _ := strconv.Atoi(data.GetUptime())
|
|
utils.CutePrint(data.FormatTime(no), format)
|
|
case "hostname":
|
|
utils.CutePrint(data.Unamebs("-n"), format)
|
|
case "kernelname":
|
|
utils.CutePrint(data.Unamebs("-s"), format)
|
|
case "GPU":
|
|
utils.CutePrint(data.GetGPU(), format)
|
|
case "shell":
|
|
utils.CutePrint(getShellVersion(data.GetShell()), format)
|
|
case "terminal":
|
|
utils.CutePrint(data.GetTerminal(), format)
|
|
case "memoryAll":
|
|
utils.CutePrint(data.GetMemory(false), format)
|
|
case "memoryUsed":
|
|
utils.CutePrint(data.GetMemory(true), format)
|
|
case "wm":
|
|
utils.CutePrint(data.GetWM(), format)
|
|
case "ip":
|
|
utils.CutePrint(data.GetLocalIP(), format)
|
|
default:
|
|
print("{UNKNOWN KEYWORD: " + rest + "}")
|
|
}
|
|
if action == "infoln" {
|
|
utils.CuteNewLine()
|
|
}
|
|
}
|
|
}
|
|
func runpage() {
|
|
if utils.Ishelp {
|
|
println(`neowofetch version x (idk how to implement this)
|
|
--nouwu turns off uwuifying
|
|
--usepng uses a png (only supports arch and ubuntu rn)
|
|
--noascii turns off the ascii
|
|
--nocolor no color.
|
|
--noconf uses builtin config instead of looking for one in the filesystem
|
|
--nocolorconf same thing but for color config :P
|
|
--16color uses escape codes instead of RGB
|
|
--help YOU LITERALLY FUCKING RAN IT RIGHT NOW`)
|
|
os.Exit(0)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
utils.Initargs()
|
|
runpage()
|
|
checkforconfigfolder()
|
|
utils.Initcolor()
|
|
utils.CutePrintInit()
|
|
handleConfig()
|
|
utils.CutePrintEnd()
|
|
}
|