EnumDev/stormfetch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Removed distro_id config option

Browse Source 
Cleaned up some code
This commit is contained in:
CapCreeperGR 2024-05-05 12:38:02 +03:00
parent 23cdf358b0
commit 0dcb6320af
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
main.go
View File

@ -16,7 +16,6 @@ var configPath string = ""
var config StormfetchConfig = StormfetchConfig{}
type StormfetchConfig struct {
Distro string `yaml:"distro_id"`
Ascii string `yaml:"distro_ascii"`
Fetch string `yaml:"fetch_script"`
}
@ -76,8 +75,7 @@ func readConfig() {
log.Fatal(err)
}
cmd.Dir = workdir
cmd.Environ()
getDistroInfo()
cmd.Env = os.Environ()
cmd.Env = append(cmd.Env, "DISTRO_LONG_NAME="+getDistroInfo().LongName)
cmd.Env = append(cmd.Env, "DISTRO_SHORT_NAME="+getDistroInfo().ShortName)
out, err := cmd.Output()
@ -134,7 +132,6 @@ func readKeyValueFile(filepath string) (map[string]string, error) {
func getDistroInfo() DistroInfo {
distroID := ""
var releaseMap = make(map[string]string)
if config.Distro == "auto" {
if _, err := os.Stat("/etc/os-release"); err == nil {
releaseMap, err = readKeyValueFile("/etc/os-release")
if err != nil {
@ -148,7 +145,6 @@ func getDistroInfo() DistroInfo {
distroID = value
}
}
}
switch distroID {
case "debian":
@ -187,10 +183,16 @@ func getDistroAscii() string {
(| | )
/'\_ _/'\
\___)=(___/ `
if _, err := os.Stat(path.Join(asciiPath, getDistroInfo().ID)); err == nil {
bytes, err := os.ReadFile(path.Join(asciiPath, getDistroInfo().ID))
var id string
if config.Ascii == "auto" {
id = getDistroInfo().ID
} else {
id = config.Ascii
}
if _, err := os.Stat(path.Join(asciiPath, id)); err == nil {
bytes, err := os.ReadFile(path.Join(asciiPath, id))
if err != nil {
return ""
return defaultAscii
}
return string(bytes)
} else {