fix: detect windows terminal environment variable

And add test to look for WT_SESSION

termenv_test.go

@@ -4,7 +4,6 @@ import (
 	"bytes"
 	"fmt"
 	"image/color"
-	"io"
 	"io/ioutil"
 	"os"
 	"strings"
@@ -161,28 +160,6 @@ func TestANSIProfile(t *testing.T) {
 	}
 }
 
-func TestWSLDetectioninWindowsTerminal(t *testing.T) {
-	tmpFile, err := os.CreateTemp("", "osrelease")
-	if err != nil {
-		t.Error("Error making tmp file")
-	}
-	defer tmpFile.Close()
-
-	_, err = io.WriteString(tmpFile, "6.2.6-76060206-microsoft")
-	if err != nil {
-		t.Error("Error writing to tmp file")
-	}
-
-	if os.Getenv("WT_SESSION") == "" {
-		os.Setenv("WT_SESSION", "a22023a86b26f5b301335de43994ddc3")
-	}
-
-	p := checkWSLinWindowsTerminal(os.Getenv("TERM"), tmpFile.Name())
-	if p != TrueColor {
-		t.Errorf("Expected %d, got %d", TrueColor, p)
-	}
-}
-
 func TestANSI256Profile(t *testing.T) {
 	p := ANSI256
 
@@ -438,3 +415,12 @@ func TestWithTTY(t *testing.T) {
 		}
 	}
 }
+
+func TestWindowsTerminal(t *testing.T) {
+	t.Setenv("TERM", "xterm-256color")
+	t.Setenv("WT_SESSION", "1")
+	o := NewOutput(ioutil.Discard, WithTTY(true))
+	if cp := o.ColorProfile(); cp != TrueColor {
+		t.Fatalf("expected color profile to be %d, got %d", TrueColor, cp)
+	}
+}

termenv_unix.go

@@ -4,11 +4,8 @@
 package termenv
 
 import (
-	"errors"
 	"fmt"
 	"io"
-	"io/ioutil"
-	"os"
 	"strconv"
 	"strings"
 	"time"
@@ -60,7 +57,11 @@ func (o *Output) ColorProfile() Profile {
 	}
 
 	if strings.Contains(term, "256color") {
-		return checkWSLinWindowsTerminal(term, "/proc/sys/kernel/osrelease")
+		// Check if we're running Windows Terminal
+		if o.environ.Getenv("WT_SESSION") != "" {
+			return TrueColor
+		}
+		return ANSI256
 	}
 	if strings.Contains(term, "color") {
 		return ANSI
@@ -72,26 +73,6 @@ func (o *Output) ColorProfile() Profile {
 	return Ascii
 }
 
-func checkWSLinWindowsTerminal(term, filename string) Profile {
-	// Check for WSL
-	wsl, err := ioutil.ReadFile(filename)
-	// Additional check for Mac OS as it doesn't have "/proc/" folder
-	if err != nil && !errors.Is(err, os.ErrNotExist) {
-		return ANSI256
-	}
-	if string(wsl) != "" {
-		// Lowercasing every content inside "/proc/sys/kernal/osrelease"
-		// because it gives "Microsoft" for WSL and "microsoft" for WSL 2
-		// so no need of checking twice
-		wslLower := strings.ToLower(string(wsl))
-		// Also check if the terminal used is Windows Terminal
-		if strings.Contains(wslLower, "microsoft") && os.Getenv("WT_SESSION") != "" {
-			return TrueColor
-		}
-	}
-	return ANSI256
-}
-
 func (o Output) foregroundColor() Color {
 	s, err := o.termStatusReport(10)
 	if err == nil {