move color to top level, add windows support
This commit is contained in:
parent
4077ac03f1
commit
bca4eef166
@ -3,12 +3,9 @@ package color
|
||||
import (
|
||||
"os"
|
||||
"strings"
|
||||
"syscall"
|
||||
"unsafe"
|
||||
"sync"
|
||||
)
|
||||
|
||||
const ioctlReadTermios = 0x5401
|
||||
|
||||
// ANSI color codes
|
||||
const (
|
||||
resetCode = "\033[0m"
|
||||
@ -22,7 +19,10 @@ const (
|
||||
grayCode = "\033[90m"
|
||||
)
|
||||
|
||||
var useColors = true
|
||||
var (
|
||||
useColors bool
|
||||
colorMu sync.RWMutex
|
||||
)
|
||||
|
||||
// Color function; makes a call to makeColorFunc with the associated color code
|
||||
var (
|
||||
@ -37,9 +37,17 @@ var (
|
||||
Gray = makeColorFunc(grayCode)
|
||||
)
|
||||
|
||||
func init() {
|
||||
useColors = DetectShellColors()
|
||||
}
|
||||
|
||||
func makeColorFunc(code string) func(string) string {
|
||||
return func(text string) string {
|
||||
if useColors {
|
||||
colorMu.RLock()
|
||||
enabled := useColors
|
||||
colorMu.RUnlock()
|
||||
|
||||
if enabled {
|
||||
return code + text + resetCode
|
||||
}
|
||||
return text
|
||||
@ -74,23 +82,20 @@ func DetectShellColors() bool {
|
||||
strings.Contains(term, "xterm") ||
|
||||
strings.Contains(term, "screen") ||
|
||||
strings.Contains(term, "tmux") ||
|
||||
term == "linux"
|
||||
term == "linux" ||
|
||||
isWindowsTerminal()
|
||||
}
|
||||
|
||||
// SetColors enables or disables colors globally
|
||||
func SetColors(enabled bool) {
|
||||
colorMu.Lock()
|
||||
useColors = enabled
|
||||
colorMu.Unlock()
|
||||
}
|
||||
|
||||
// ColorsEnabled returns current global color setting
|
||||
func ColorsEnabled() bool {
|
||||
colorMu.RLock()
|
||||
defer colorMu.RUnlock()
|
||||
return useColors
|
||||
}
|
||||
|
||||
// isTerminal checks if the file is a terminal
|
||||
func isTerminal(f *os.File) bool {
|
||||
fd := f.Fd()
|
||||
var termios syscall.Termios
|
||||
_, _, err := syscall.Syscall6(syscall.SYS_IOCTL, fd, ioctlReadTermios, uintptr(unsafe.Pointer(&termios)), 0, 0, 0)
|
||||
return err == 0
|
||||
}
|
33
color/color_unix.go
Normal file
33
color/color_unix.go
Normal file
@ -0,0 +1,33 @@
|
||||
//go:build !windows
|
||||
// +build !windows
|
||||
|
||||
package color
|
||||
|
||||
import (
|
||||
"os"
|
||||
"syscall"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
const ioctlReadTermios = 0x5401
|
||||
|
||||
// isTerminal checks if the file is a terminal
|
||||
func isTerminal(f *os.File) bool {
|
||||
fd := f.Fd()
|
||||
var termios syscall.Termios
|
||||
|
||||
r1, _, errno := syscall.Syscall6(
|
||||
syscall.SYS_IOCTL,
|
||||
fd,
|
||||
ioctlReadTermios,
|
||||
uintptr(unsafe.Pointer(&termios)),
|
||||
0, 0, 0,
|
||||
)
|
||||
|
||||
return r1 == 0 && errno == 0
|
||||
}
|
||||
|
||||
// isWindowsTerminal always returns false on Unix
|
||||
func isWindowsTerminal() bool {
|
||||
return false
|
||||
}
|
29
color/color_windows.go
Normal file
29
color/color_windows.go
Normal file
@ -0,0 +1,29 @@
|
||||
//go:build windows
|
||||
// +build windows
|
||||
|
||||
package color
|
||||
|
||||
import (
|
||||
"os"
|
||||
"syscall"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
var (
|
||||
kernel32 = syscall.NewLazyDLL("kernel32.dll")
|
||||
procGetConsoleMode = kernel32.NewProc("GetConsoleMode")
|
||||
)
|
||||
|
||||
// isTerminal checks if the file is a terminal on Windows
|
||||
func isTerminal(f *os.File) bool {
|
||||
handle := syscall.Handle(f.Fd())
|
||||
var mode uint32
|
||||
|
||||
r1, _, _ := procGetConsoleMode.Call(uintptr(handle), uintptr(unsafe.Pointer(&mode)))
|
||||
return r1 != 0
|
||||
}
|
||||
|
||||
// isWindowsTerminal checks for Windows Terminal
|
||||
func isWindowsTerminal() bool {
|
||||
return os.Getenv("WT_SESSION") != ""
|
||||
}
|
@ -8,11 +8,11 @@ import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"Moonshark/color"
|
||||
"Moonshark/router"
|
||||
"Moonshark/runner"
|
||||
"Moonshark/sessions"
|
||||
"Moonshark/utils"
|
||||
"Moonshark/utils/color"
|
||||
"Moonshark/utils/config"
|
||||
"Moonshark/utils/logger"
|
||||
"Moonshark/utils/metadata"
|
||||
|
@ -13,12 +13,12 @@ import (
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"Moonshark/color"
|
||||
"Moonshark/http"
|
||||
"Moonshark/router"
|
||||
"Moonshark/runner"
|
||||
"Moonshark/runner/lualibs"
|
||||
"Moonshark/sessions"
|
||||
"Moonshark/utils/color"
|
||||
"Moonshark/utils/config"
|
||||
"Moonshark/utils/logger"
|
||||
"Moonshark/utils/metadata"
|
||||
|
@ -10,7 +10,7 @@ import (
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
"Moonshark/utils/color"
|
||||
"Moonshark/color"
|
||||
"Moonshark/utils/logger"
|
||||
|
||||
luajit "git.sharkk.net/Sky/LuaJIT-to-Go"
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"Moonshark/utils/color"
|
||||
"Moonshark/color"
|
||||
"Moonshark/utils/logger"
|
||||
|
||||
lru "git.sharkk.net/Go/LRU"
|
||||
|
@ -11,7 +11,7 @@ import (
|
||||
sqlite "zombiezen.com/go/sqlite"
|
||||
"zombiezen.com/go/sqlite/sqlitex"
|
||||
|
||||
"Moonshark/utils/color"
|
||||
"Moonshark/color"
|
||||
"Moonshark/utils/logger"
|
||||
|
||||
luajit "git.sharkk.net/Sky/LuaJIT-to-Go"
|
||||
|
@ -9,7 +9,7 @@ import (
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"Moonshark/utils/color"
|
||||
"Moonshark/color"
|
||||
)
|
||||
|
||||
// Log levels
|
||||
|
@ -5,9 +5,9 @@ import (
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
"Moonshark/color"
|
||||
"Moonshark/router"
|
||||
"Moonshark/runner"
|
||||
"Moonshark/utils/color"
|
||||
"Moonshark/utils/logger"
|
||||
)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user