143 lines
3.8 KiB
Go
143 lines
3.8 KiB
Go
package tui
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"kitty"
|
|
)
|
|
|
|
const (
|
|
SAVE_CURSOR = "\0337"
|
|
RESTORE_CURSOR = "\0338"
|
|
S7C1T = "\033 F"
|
|
SAVE_PRIVATE_MODE_VALUES = "\033[?s"
|
|
RESTORE_PRIVATE_MODE_VALUES = "\033[?r"
|
|
SAVE_COLORS = "\033[#P"
|
|
RESTORE_COLORS = "\033[#Q"
|
|
DECSACE_DEFAULT_REGION_SELECT = "\033[*x"
|
|
CLEAR_SCREEN = "\033[H\033[2J"
|
|
)
|
|
|
|
type Mode uint32
|
|
|
|
const private Mode = 1 << 31
|
|
|
|
const (
|
|
LNM Mode = 20
|
|
IRM = 4
|
|
DECKM = 1 | private
|
|
DECSCNM = 5 | private
|
|
DECOM = 6 | private
|
|
DECAWM = 7 | private
|
|
DECARM = 8 | private
|
|
DECTCEM = 25 | private
|
|
MOUSE_BUTTON_TRACKING = 1000 | private
|
|
MOUSE_MOTION_TRACKING = 1002 | private
|
|
MOUSE_MOVE_TRACKING = 1003 | private
|
|
FOCUS_TRACKING = 1004 | private
|
|
MOUSE_UTF8_MODE = 1005 | private
|
|
MOUSE_SGR_MODE = 1006 | private
|
|
MOUSE_URXVT_MODE = 1015 | private
|
|
MOUSE_SGR_PIXEL_MODE = 1016 | private
|
|
ALTERNATE_SCREEN = 1049 | private
|
|
BRACKETED_PASTE = 2004 | private
|
|
PENDING_UPDATE = 2026 | private
|
|
HANDLE_TERMIOS_SIGNALS = kitty.HandleTermiosSignals | private
|
|
)
|
|
|
|
func (self *Mode) escape_code(which string) string {
|
|
num := *self
|
|
priv := ""
|
|
if num&private > 0 {
|
|
priv = "?"
|
|
num &^= private
|
|
}
|
|
return fmt.Sprintf("\033[%s%d%s", priv, uint32(num), which)
|
|
}
|
|
|
|
func (self *Mode) EscapeCodeToSet() string {
|
|
return self.escape_code("h")
|
|
}
|
|
|
|
func (self *Mode) EscapeCodeToReset() string {
|
|
return self.escape_code("h")
|
|
}
|
|
|
|
type MouseTracking uint8
|
|
|
|
const (
|
|
NO_MOUSE_TRACKING MouseTracking = iota
|
|
BUTTONS_ONLY_MOUSE_TRACKING
|
|
BUTTONS_AND_DRAG_MOUSE_TRACKING
|
|
FULL_MOUSE_TRACKING
|
|
)
|
|
|
|
type TerminalState struct {
|
|
alternate_screen, kitty_keyboard_mode bool
|
|
mouse_tracking MouseTracking
|
|
}
|
|
|
|
func set_modes(sb *strings.Builder, modes ...Mode) {
|
|
for _, m := range modes {
|
|
sb.WriteString(m.EscapeCodeToSet())
|
|
}
|
|
}
|
|
|
|
func reset_modes(sb *strings.Builder, modes ...Mode) {
|
|
for _, m := range modes {
|
|
sb.WriteString(m.EscapeCodeToReset())
|
|
}
|
|
}
|
|
|
|
func (self *TerminalState) SetStateEscapeCodes() []byte {
|
|
var sb strings.Builder
|
|
sb.Grow(256)
|
|
sb.WriteString(S7C1T)
|
|
if self.alternate_screen {
|
|
sb.WriteString(SAVE_CURSOR)
|
|
}
|
|
sb.WriteString(SAVE_PRIVATE_MODE_VALUES)
|
|
sb.WriteString(SAVE_COLORS)
|
|
sb.WriteString(DECSACE_DEFAULT_REGION_SELECT)
|
|
reset_modes(&sb, IRM, DECKM, DECSCNM, MOUSE_BUTTON_TRACKING, MOUSE_MOTION_TRACKING,
|
|
MOUSE_MOVE_TRACKING, FOCUS_TRACKING, MOUSE_UTF8_MODE, MOUSE_SGR_MODE, BRACKETED_PASTE)
|
|
set_modes(&sb, DECARM, DECAWM, DECTCEM)
|
|
if self.alternate_screen {
|
|
set_modes(&sb, ALTERNATE_SCREEN)
|
|
sb.WriteString(CLEAR_SCREEN)
|
|
}
|
|
if self.kitty_keyboard_mode {
|
|
sb.WriteString("\033[>31u")
|
|
} else {
|
|
sb.WriteString("\033[>u")
|
|
}
|
|
if self.mouse_tracking != NO_MOUSE_TRACKING {
|
|
sb.WriteString(MOUSE_SGR_PIXEL_MODE.EscapeCodeToSet())
|
|
switch self.mouse_tracking {
|
|
case BUTTONS_ONLY_MOUSE_TRACKING:
|
|
sb.WriteString(MOUSE_BUTTON_TRACKING.EscapeCodeToSet())
|
|
case BUTTONS_AND_DRAG_MOUSE_TRACKING:
|
|
sb.WriteString(MOUSE_MOTION_TRACKING.EscapeCodeToSet())
|
|
case FULL_MOUSE_TRACKING:
|
|
sb.WriteString(MOUSE_MOVE_TRACKING.EscapeCodeToSet())
|
|
}
|
|
}
|
|
return []byte(sb.String())
|
|
}
|
|
|
|
func (self *TerminalState) ResetStateData() []byte {
|
|
var sb strings.Builder
|
|
sb.Grow(64)
|
|
sb.WriteString("\033[<u")
|
|
if self.alternate_screen {
|
|
sb.WriteString(ALTERNATE_SCREEN.EscapeCodeToReset())
|
|
} else {
|
|
sb.WriteString(SAVE_CURSOR)
|
|
}
|
|
sb.WriteString(RESTORE_PRIVATE_MODE_VALUES)
|
|
sb.WriteString(RESTORE_CURSOR)
|
|
sb.WriteString(RESTORE_COLORS)
|
|
return []byte(sb.String())
|
|
}
|