fix color level error on Terminus

gookit · Apr 6, 2021 · 402971f · 402971f
1 parent aaee2b3
commit 402971f
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/detect_env.go b/detect_env.go
@@ -100,6 +100,11 @@ func detectColorLevelFromEnv(termVal string) (terminfo.ColorLevel, error) {
 		return terminfo.ColorLevelBasic, nil
 	case termProg == "Apple_Terminal":
 		return terminfo.ColorLevelHundreds, nil
+	case termProg == "Terminus":
+		if termVal == "screen" { // on TERM=screen: not support true-color
+			return terminfo.ColorLevelHundreds, nil
+		}
+		return terminfo.ColorLevelMillions, nil
 	case termProg == "iTerm.app":
 		if termVal == "screen" { // on TERM=screen: not support true-color
 			return terminfo.ColorLevelHundreds, nil

diff --git a/utils_test.go b/utils_test.go
@@ -68,6 +68,19 @@ ZSH_TMUX_TERM=screen-256color
 func TestIsDetectColorLevel_unix(t *testing.T) {
 	is := assert.New(t)
 
+	// TERM_PROGRAM=Terminus
+	mockOsEnvByText(`
+TERMINUS_PLUGINS=
+TERM=xterm-256color
+TERM_PROGRAM=Terminus
+ZSH_TMUX_TERM=screen-256color
+`, func() {
+		is.Equal(LevelRgb, DetectColorLevel())
+		is.True(IsSupportRGBColor())
+		is.True(IsSupport16Color())
+		is.True(IsSupportColor())
+	})
+
 	// -------- tests on macOS ---------
 
 	// TERM_PROGRAM=Apple_Terminal
@@ -148,6 +161,21 @@ ZSH_TMUX_TERM=screen
 		is.True(IsSupportColor())
 	})
 
+	// TERM_PROGRAM=Terminus use screen
+	mockOsEnvByText(`
+TERM=screen
+TERMCAP=SC|screen|VT 100/ANSI X3.64 virtual terminal:\
+TERMINUS_PLUGINS=
+TERM_PROGRAM=Terminus
+ZSH_TMUX_TERM=screen
+`, func() {
+		is.Equal(Level256, DetectColorLevel())
+		is.False(IsSupportTrueColor())
+		is.True(IsSupport256Color())
+		is.True(IsSupport16Color())
+		is.True(IsSupportColor())
+	})
+
 	// TERMINAL_EMULATOR=JetBrains-JediTerm use screen
 	mockOsEnvByText(`
 TERM=screen