Andrew Hamilton
5341d91592
- termstr no longer depends on terminal. - Moved some code from terminal to termstr.
92 lines
2.1 KiB
Python
92 lines
2.1 KiB
Python
|
|
|
|
import contextlib
|
|
import sys
|
|
import termios
|
|
|
|
|
|
ESC = "\x1b"
|
|
MOUSE = ESC + "[M"
|
|
|
|
UP_KEY = ESC + "[A"
|
|
DOWN_KEY = ESC + "[B"
|
|
RIGHT_KEY = ESC + "[C"
|
|
LEFT_KEY = ESC + "[D"
|
|
PAGE_UP_KEY = ESC + "[5~"
|
|
PAGE_DOWN_KEY = ESC + "[6~"
|
|
HOME_KEY = ESC + "[H"
|
|
END_KEY = ESC + "[F"
|
|
|
|
|
|
def move(x, y):
|
|
return ESC + f"[{y + 1:d};{x + 1:d}H"
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def terminal_title(title):
|
|
sys.stdout.write(ESC + "7") # save
|
|
sys.stdout.write(f"\033]0;{title}\007") # set title
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout.write(ESC + "8") # restore
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def mouse_tracking():
|
|
sys.stdout.write(ESC + "[?1000h" + ESC + "[?1002h") # tracking on
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout.write(ESC + "[?1002l" + ESC + "[?1000l") # tracking off
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def alternate_buffer():
|
|
sys.stdout.write(ESC + "[?1049h") # switch to alternate buffer
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout.write(ESC + "[?1049l") # restore normal buffer
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def interactive():
|
|
old_termios_settings = termios.tcgetattr(sys.stdin)
|
|
new_settings = termios.tcgetattr(sys.stdin)
|
|
new_settings[3] = new_settings[3] & ~termios.ECHO & ~termios.ICANON
|
|
new_settings[6][termios.VMIN] = 0
|
|
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, new_settings)
|
|
sys.stdout.write(ESC + "[?1l") # Ensure normal cursor key codes
|
|
try:
|
|
yield
|
|
finally:
|
|
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_termios_settings)
|
|
|
|
|
|
RELEASE_MOUSE = 0
|
|
DRAG_MOUSE = 1
|
|
CLICK_MOUSE = 2
|
|
PRESS_MOUSE = 3
|
|
|
|
WHEEL_UP_MOUSE = 4
|
|
WHEEL_DOWN_MOUSE = 5
|
|
|
|
|
|
def decode_mouse_input(term_code):
|
|
keys = [ord(byte) for byte in term_code]
|
|
b = keys[0] - 32
|
|
x, y = (keys[1] - 33) % 256, (keys[2] - 33) % 256
|
|
button = ((b & 64) / 64 * 3) + (b & 3) + 1
|
|
if b & 3 == 3:
|
|
action = RELEASE_MOUSE
|
|
button = 0
|
|
elif b & 2048:
|
|
action = RELEASE_MOUSE
|
|
elif b & 32:
|
|
action = DRAG_MOUSE
|
|
elif b & 1536:
|
|
action = CLICK_MOUSE
|
|
else:
|
|
action = PRESS_MOUSE
|
|
return (action, button, x, y)
|