feat(os): allow overriding distro name

resolves #5011
JanDeDobbeleer · Jun 6, 2024 · 523cd98 · 523cd98
1 parent 2b47173
commit 523cd98
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/segments/os.go b/src/segments/os.go
@@ -38,7 +38,7 @@ func (oi *Os) Enabled() bool {
 		pf := oi.env.Platform()
 		displayDistroName := oi.props.GetBool(DisplayDistroName, false)
 		if displayDistroName {
-			oi.Icon = pf
+			oi.Icon = oi.props.GetString(properties.Property(pf), pf)
 			break
 		}
 		oi.Icon = oi.getDistroIcon(pf)

diff --git a/src/segments/os_test.go b/src/segments/os_test.go
@@ -76,6 +76,14 @@ func TestOSInfo(t *testing.T) {
 			GOOS:           "linux",
 			Platform:       "crazy",
 		},
+		{
+			Case:              "show distro name, mapped",
+			ExpectedString:    "<3",
+			DisplayDistroName: true,
+			GOOS:              "linux",
+			Icon:              "<3",
+			Platform:          "love",
+		},
 	}
 	for _, tc := range cases {
 		env := new(mock.MockedEnvironment)