feat: detect windows terminal

termenv_test.go

@@ -2,8 +2,10 @@ package termenv
 
 import (
 	"bytes"
+	"errors"
 	"fmt"
 	"image/color"
+	"io"
 	"io/ioutil"
 	"os"
 	"strings"
@@ -160,6 +162,39 @@ func TestANSIProfile(t *testing.T) {
 	}
 }
 
+func TestWSLDetectioninWindowsTerminal(t *testing.T) {
+	file, err := os.Open("/proc/sys/kernel/osrelease")
+	defer file.Close()
+	if err != nil && !errors.Is(err, os.ErrNotExist) {
+		t.Error("Error reading file proc/sys/kernel/osrelease")
+	}
+
+	tmpFile, err := os.CreateTemp("", "osrelease")
+	defer tmpFile.Close()
+	if err != nil {
+		t.Error("Error making tmp file")
+	}
+
+	io.Copy(tmpFile, file)
+	_, err = io.WriteString(tmpFile, "6.2.6-76060206-microsoft")
+	if err != nil {
+		t.Error("Error writing to tmp file")
+	}
+
+	content, err := os.ReadFile(tmpFile.Name())
+	if err != nil {
+		t.Error("Error reading tmp file")
+	}
+
+	if os.Getenv("WT_SESSION") == "" {
+		os.Setenv("WT_SESSION", "a22023a86b26f5b301335de43994ddc3")
+	}
+	p := ColorProfile()
+	if p != TrueColor && !strings.Contains(strings.ToLower(string(content)), "mircosoft") && os.Getenv("WT_SESSION") == "" {
+		t.Errorf("Expected %d, got %d", TrueColor, p)
+	}
+}
+
 func TestANSI256Profile(t *testing.T) {
 	p := ANSI256
 

termenv_unix.go

@@ -4,8 +4,11 @@
 package termenv
 
 import (
+	"errors"
 	"fmt"
 	"io"
+	"io/ioutil"
+	"os"
 	"strconv"
 	"strings"
 	"time"
@@ -57,6 +60,22 @@ func (o *Output) ColorProfile() Profile {
 	}
 
 	if strings.Contains(term, "256color") {
+		// Check for WSL
+		wsl, err := ioutil.ReadFile("/proc/sys/kernel/osrelease")
+		// 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
 	}
 	if strings.Contains(term, "color") {