Allow to enumerate distro types

4 years ago · e91af205f4
parent 7040fb7a0b
commit e91af205f4
1 changed files with 3 additions and 1 deletions
--- a/main.go
+++ b/main.go
@ -93,8 +93,10 @@ const (
 	Debian
 )

+var distroTypeStrings = [...]string{"Ubuntu", "CentOS", "Debian"}
+
 func (dt distroType) String() string {
-	return [...]string{"Ubuntu", "CentOS", "Debian"}[dt]
+	return distroTypeStrings[dt]
 }

 func (dt *distroType) UnmarshalTOML(data []byte) (err error) {