Merge pull request #2 from katlyn/feat/dynamic-ascii-art

feat: use wildcard to dynamically include ascii art

asciiarts/asciiart.go

@@ -1,167 +1,19 @@
 package asciiarts
 
 import (
-	_ "embed"
+	"embed"
 	"strings"
 )
 
-//go:embed alpine.txt
+//go:embed *.txt
-var alpine string
+var asciifiles embed.FS
 
-//go:embed amog.txt
-var amog string
-
-//go:embed unknown.txt
-var unknown string
-
-//go:embed amogos.txt
-var amogos string
-
-//go:embed android.txt
-var android string
-
-//go:embed arch.txt
-var arch string
-
-//go:embed arcolinux.txt
-var arcolinux string
-
-//go:embed artix.txt
-var artix string
-
-//go:embed debian.txt
-var debian string
-
-//go:embed endeavouros.txt
-var endeavouros string
-
-//go:embed fedora.txt
-var fedora string
-
-//go:embed freebsd.txt
-var freebsd string
-
-//go:embed gentoo.txt
-var gentoo string
-
-//go:embed gnu.txt
-var gnu string
-
-//go:embed guix.txt
-var guix string
-
-//go:embed ios.txt
-var ios string
-
-//go:embed macos.txt
-var macos string
-
-//go:embed manjaro-arm.txt
-var manjaroarm string
-
-//go:embed manjaro.txt
-var manjaro string
-
-//go:embed linuxmint.txt
-var linuxmint string
-
-//go:embed openbsd.txt
-var openbsd string
-
-//go:embed opensuse-leap.txt
-var opensuseleap string
-
-//go:embed opensuse-tumbleweed.txt
-var opensuseweed string
-
-//go:embed pop.txt
-var pop string
-
-//go:embed raspbian.txt
-var raspbian string
-
-//go:embed slackware.txt
-var slackware string
-
-//go:embed solus.txt
-var solus string
-
-//go:embed ubuntu.txt
-var ubuntu string
-
-//go:embed void.txt
-var void string
-
-//go:embed xerolinux.txt
-var xerolinux string
-
-//go:embed windows.txt
-var windows string
-
-func GetAsciiInternal(distroID string) string {
-	switch strings.ToLower(distroID) {
-	case "alpine":
-		return alpine
-	case "arcolinux":
-		return arcolinux
-	case "artix":
-		return artix
-	case "debian":
-		return debian
-	case "endeavouros":
-		return endeavouros
-	case "fedora":
-		return fedora
-	case "freebsd":
-		return freebsd
-	case "gentoo":
-		return gentoo
-	case "gnu":
-		return gnu
-	case "guix":
-		return guix
-	case "ios":
-		return ios
-	case "linuxmint":
-		return linuxmint
-	case "macos":
-		return macos
-	case "manjaro-arm":
-		return manjaroarm
-	case "manjaro":
-		return manjaro
-	case "openbsd":
-		return openbsd
-	case "opensuse-leap":
-		return opensuseleap
-	case "opensuse-tumbleweed":
-		return opensuseweed
-	case "pop":
-		return pop
-	case "raspbian":
-		return raspbian
-	case "slackware":
-		return slackware
-	case "solus":
-		return solus
-	case "ubuntu":
-		return ubuntu
-	case "void":
-		return void
-	case "xerolinux":
-		return xerolinux
-	case "arch", "archcraft":
-		return arch
-	case "windows":
-		return windows
-	case "amog":
-		return amog
-	default:
-		return unknown
-	}
-}
 func GetAscii(distroID string) string {
-	ascii := GetAsciiInternal(distroID)
+	asciiBuff, error := asciifiles.ReadFile(distroID + ".txt")
+	if error != nil {
+		asciiBuff, _ = asciifiles.ReadFile("unknown.txt")
+	}
+	ascii := string(asciiBuff)
 	ascii = strings.ReplaceAll(ascii, "{WHITE}", "")
 	ascii = strings.ReplaceAll(ascii, "{YELLOW}", "")
 	ascii = strings.ReplaceAll(ascii, "{BLUE}", "")