Current File : //lib/python3/dist-packages/twisted/conch/insults/insults.py
# -*- test-case-name: twisted.conch.test.test_insults -*-
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
"""
VT102 and VT220 terminal manipulation.
@author: Jp Calderone
"""
from zope.interface import Interface, implementer
from twisted.internet import defer, interfaces as iinternet, protocol
from twisted.python.compat import iterbytes, networkString
class ITerminalProtocol(Interface):
def makeConnection(transport):
"""
Called with an L{ITerminalTransport} when a connection is established.
"""
def keystrokeReceived(keyID, modifier):
"""
A keystroke was received.
Each keystroke corresponds to one invocation of this method.
keyID is a string identifier for that key. Printable characters
are represented by themselves. Control keys, such as arrows and
function keys, are represented with symbolic constants on
L{ServerProtocol}.
"""
def terminalSize(width, height):
"""
Called to indicate the size of the terminal.
A terminal of 80x24 should be assumed if this method is not
called. This method might not be called for real terminals.
"""
def unhandledControlSequence(seq):
"""
Called when an unsupported control sequence is received.
@type seq: L{str}
@param seq: The whole control sequence which could not be interpreted.
"""
def connectionLost(reason):
"""
Called when the connection has been lost.
reason is a Failure describing why.
"""
@implementer(ITerminalProtocol)
class TerminalProtocol:
def makeConnection(self, terminal):
# assert ITerminalTransport.providedBy(transport), "TerminalProtocol.makeConnection must be passed an ITerminalTransport implementor"
self.terminal = terminal
self.connectionMade()
def connectionMade(self):
"""
Called after a connection has been established.
"""
def keystrokeReceived(self, keyID, modifier):
pass
def terminalSize(self, width, height):
pass
def unhandledControlSequence(self, seq):
pass
def connectionLost(self, reason):
pass
class ITerminalTransport(iinternet.ITransport):
def cursorUp(n=1):
"""
Move the cursor up n lines.
"""
def cursorDown(n=1):
"""
Move the cursor down n lines.
"""
def cursorForward(n=1):
"""
Move the cursor right n columns.
"""
def cursorBackward(n=1):
"""
Move the cursor left n columns.
"""
def cursorPosition(column, line):
"""
Move the cursor to the given line and column.
"""
def cursorHome():
"""
Move the cursor home.
"""
def index():
"""
Move the cursor down one line, performing scrolling if necessary.
"""
def reverseIndex():
"""
Move the cursor up one line, performing scrolling if necessary.
"""
def nextLine():
"""
Move the cursor to the first position on the next line, performing scrolling if necessary.
"""
def saveCursor():
"""
Save the cursor position, character attribute, character set, and origin mode selection.
"""
def restoreCursor():
"""
Restore the previously saved cursor position, character attribute, character set, and origin mode selection.
If no cursor state was previously saved, move the cursor to the home position.
"""
def setModes(modes):
"""
Set the given modes on the terminal.
"""
def resetModes(mode):
"""
Reset the given modes on the terminal.
"""
def setPrivateModes(modes):
"""
Set the given DEC private modes on the terminal.
"""
def resetPrivateModes(modes):
"""
Reset the given DEC private modes on the terminal.
"""
def applicationKeypadMode():
"""
Cause keypad to generate control functions.
Cursor key mode selects the type of characters generated by cursor keys.
"""
def numericKeypadMode():
"""
Cause keypad to generate normal characters.
"""
def selectCharacterSet(charSet, which):
"""
Select a character set.
charSet should be one of CS_US, CS_UK, CS_DRAWING, CS_ALTERNATE, or
CS_ALTERNATE_SPECIAL.
which should be one of G0 or G1.
"""
def shiftIn():
"""
Activate the G0 character set.
"""
def shiftOut():
"""
Activate the G1 character set.
"""
def singleShift2():
"""
Shift to the G2 character set for a single character.
"""
def singleShift3():
"""
Shift to the G3 character set for a single character.
"""
def selectGraphicRendition(*attributes):
"""
Enabled one or more character attributes.
Arguments should be one or more of UNDERLINE, REVERSE_VIDEO, BLINK, or BOLD.
NORMAL may also be specified to disable all character attributes.
"""
def horizontalTabulationSet():
"""
Set a tab stop at the current cursor position.
"""
def tabulationClear():
"""
Clear the tab stop at the current cursor position.
"""
def tabulationClearAll():
"""
Clear all tab stops.
"""
def doubleHeightLine(top=True):
"""
Make the current line the top or bottom half of a double-height, double-width line.
If top is True, the current line is the top half. Otherwise, it is the bottom half.
"""
def singleWidthLine():
"""
Make the current line a single-width, single-height line.
"""
def doubleWidthLine():
"""
Make the current line a double-width line.
"""
def eraseToLineEnd():
"""
Erase from the cursor to the end of line, including cursor position.
"""
def eraseToLineBeginning():
"""
Erase from the cursor to the beginning of the line, including the cursor position.
"""
def eraseLine():
"""
Erase the entire cursor line.
"""
def eraseToDisplayEnd():
"""
Erase from the cursor to the end of the display, including the cursor position.
"""
def eraseToDisplayBeginning():
"""
Erase from the cursor to the beginning of the display, including the cursor position.
"""
def eraseDisplay():
"""
Erase the entire display.
"""
def deleteCharacter(n=1):
"""
Delete n characters starting at the cursor position.
Characters to the right of deleted characters are shifted to the left.
"""
def insertLine(n=1):
"""
Insert n lines at the cursor position.
Lines below the cursor are shifted down. Lines moved past the bottom margin are lost.
This command is ignored when the cursor is outside the scroll region.
"""
def deleteLine(n=1):
"""
Delete n lines starting at the cursor position.
Lines below the cursor are shifted up. This command is ignored when the cursor is outside
the scroll region.
"""
def reportCursorPosition():
"""
Return a Deferred that fires with a two-tuple of (x, y) indicating the cursor position.
"""
def reset():
"""
Reset the terminal to its initial state.
"""
def unhandledControlSequence(seq):
"""
Called when an unsupported control sequence is received.
@type seq: L{str}
@param seq: The whole control sequence which could not be interpreted.
"""
CSI = b"\x1b"
CST = {b"~": b"tilde"}
class modes:
"""
ECMA 48 standardized modes
"""
# BREAKS YOPUR KEYBOARD MOFO
KEYBOARD_ACTION = KAM = 2
# When set, enables character insertion. New display characters
# move old display characters to the right. Characters moved past
# the right margin are lost.
# When reset, enables replacement mode (disables character
# insertion). New display characters replace old display
# characters at cursor position. The old character is erased.
INSERTION_REPLACEMENT = IRM = 4
# Set causes a received linefeed, form feed, or vertical tab to
# move cursor to first column of next line. RETURN transmits both
# a carriage return and linefeed. This selection is also called
# new line option.
# Reset causes a received linefeed, form feed, or vertical tab to
# move cursor to next line in current column. RETURN transmits a
# carriage return.
LINEFEED_NEWLINE = LNM = 20
class privateModes:
"""
ANSI-Compatible Private Modes
"""
ERROR = 0
CURSOR_KEY = 1
ANSI_VT52 = 2
COLUMN = 3
SCROLL = 4
SCREEN = 5
ORIGIN = 6
AUTO_WRAP = 7
AUTO_REPEAT = 8
PRINTER_FORM_FEED = 18
PRINTER_EXTENT = 19
# Toggle cursor visibility (reset hides it)
CURSOR_MODE = 25
# Character sets
CS_US = b"CS_US"
CS_UK = b"CS_UK"
CS_DRAWING = b"CS_DRAWING"
CS_ALTERNATE = b"CS_ALTERNATE"
CS_ALTERNATE_SPECIAL = b"CS_ALTERNATE_SPECIAL"
# Groupings (or something?? These are like variables that can be bound to character sets)
G0 = b"G0"
G1 = b"G1"
# G2 and G3 cannot be changed, but they can be shifted to.
G2 = b"G2"
G3 = b"G3"
# Character attributes
NORMAL = 0
BOLD = 1
UNDERLINE = 4
BLINK = 5
REVERSE_VIDEO = 7
class Vector:
def __init__(self, x, y):
self.x = x
self.y = y
def log(s):
with open("log", "a") as f:
f.write(str(s) + "\n")
# XXX TODO - These attributes are really part of the
# ITerminalTransport interface, I think.
_KEY_NAMES = (
"UP_ARROW",
"DOWN_ARROW",
"RIGHT_ARROW",
"LEFT_ARROW",
"HOME",
"INSERT",
"DELETE",
"END",
"PGUP",
"PGDN",
"NUMPAD_MIDDLE",
"F1",
"F2",
"F3",
"F4",
"F5",
"F6",
"F7",
"F8",
"F9",
"F10",
"F11",
"F12",
"ALT",
"SHIFT",
"CONTROL",
)
class _const:
"""
@ivar name: A string naming this constant
"""
def __init__(self, name: str) -> None:
self.name = name
def __repr__(self) -> str:
return "[" + self.name + "]"
def __bytes__(self) -> bytes:
return ("[" + self.name + "]").encode("ascii")
FUNCTION_KEYS = [_const(_name).__bytes__() for _name in _KEY_NAMES]
@implementer(ITerminalTransport)
class ServerProtocol(protocol.Protocol):
protocolFactory = None
terminalProtocol = None
TAB = b"\t"
BACKSPACE = b"\x7f"
##
lastWrite = b""
state = b"data"
termSize = Vector(80, 24)
cursorPos = Vector(0, 0)
scrollRegion = None
# Factory who instantiated me
factory = None
def __init__(self, protocolFactory=None, *a, **kw):
"""
@param protocolFactory: A callable which will be invoked with
*a, **kw and should return an ITerminalProtocol implementor.
This will be invoked when a connection to this ServerProtocol
is established.
@param a: Any positional arguments to pass to protocolFactory.
@param kw: Any keyword arguments to pass to protocolFactory.
"""
# assert protocolFactory is None or ITerminalProtocol.implementedBy(protocolFactory), "ServerProtocol.__init__ must be passed an ITerminalProtocol implementor"
if protocolFactory is not None:
self.protocolFactory = protocolFactory
self.protocolArgs = a
self.protocolKwArgs = kw
self._cursorReports = []
def getHost(self):
# ITransport.getHost
raise NotImplementedError("Unimplemented: ServerProtocol.getHost")
def getPeer(self):
# ITransport.getPeer
raise NotImplementedError("Unimplemented: ServerProtocol.getPeer")
def connectionMade(self):
if self.protocolFactory is not None:
self.terminalProtocol = self.protocolFactory(
*self.protocolArgs, **self.protocolKwArgs
)
try:
factory = self.factory
except AttributeError:
pass
else:
self.terminalProtocol.factory = factory
self.terminalProtocol.makeConnection(self)
def dataReceived(self, data):
for ch in iterbytes(data):
if self.state == b"data":
if ch == b"\x1b":
self.state = b"escaped"
else:
self.terminalProtocol.keystrokeReceived(ch, None)
elif self.state == b"escaped":
if ch == b"[":
self.state = b"bracket-escaped"
self.escBuf = []
elif ch == b"O":
self.state = b"low-function-escaped"
else:
self.state = b"data"
self._handleShortControlSequence(ch)
elif self.state == b"bracket-escaped":
if ch == b"O":
self.state = b"low-function-escaped"
elif ch.isalpha() or ch == b"~":
self._handleControlSequence(b"".join(self.escBuf) + ch)
del self.escBuf
self.state = b"data"
else:
self.escBuf.append(ch)
elif self.state == b"low-function-escaped":
self._handleLowFunctionControlSequence(ch)
self.state = b"data"
else:
raise ValueError("Illegal state")
def _handleShortControlSequence(self, ch):
self.terminalProtocol.keystrokeReceived(ch, self.ALT)
def _handleControlSequence(self, buf):
buf = b"\x1b[" + buf
f = getattr(
self.controlSequenceParser,
CST.get(buf[-1:], buf[-1:]).decode("ascii"),
None,
)
if f is None:
self.unhandledControlSequence(buf)
else:
f(self, self.terminalProtocol, buf[:-1])
def unhandledControlSequence(self, buf):
self.terminalProtocol.unhandledControlSequence(buf)
def _handleLowFunctionControlSequence(self, ch):
functionKeys = {b"P": self.F1, b"Q": self.F2, b"R": self.F3, b"S": self.F4}
keyID = functionKeys.get(ch)
if keyID is not None:
self.terminalProtocol.keystrokeReceived(keyID, None)
else:
self.terminalProtocol.unhandledControlSequence(b"\x1b[O" + ch)
class ControlSequenceParser:
def A(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.UP_ARROW, None)
else:
handler.unhandledControlSequence(buf + b"A")
def B(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.DOWN_ARROW, None)
else:
handler.unhandledControlSequence(buf + b"B")
def C(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.RIGHT_ARROW, None)
else:
handler.unhandledControlSequence(buf + b"C")
def D(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.LEFT_ARROW, None)
else:
handler.unhandledControlSequence(buf + b"D")
def E(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.NUMPAD_MIDDLE, None)
else:
handler.unhandledControlSequence(buf + b"E")
def F(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.END, None)
else:
handler.unhandledControlSequence(buf + b"F")
def H(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.HOME, None)
else:
handler.unhandledControlSequence(buf + b"H")
def R(self, proto, handler, buf):
if not proto._cursorReports:
handler.unhandledControlSequence(buf + b"R")
elif buf.startswith(b"\x1b["):
report = buf[2:]
parts = report.split(b";")
if len(parts) != 2:
handler.unhandledControlSequence(buf + b"R")
else:
Pl, Pc = parts
try:
Pl, Pc = int(Pl), int(Pc)
except ValueError:
handler.unhandledControlSequence(buf + b"R")
else:
d = proto._cursorReports.pop(0)
d.callback((Pc - 1, Pl - 1))
else:
handler.unhandledControlSequence(buf + b"R")
def Z(self, proto, handler, buf):
if buf == b"\x1b[":
handler.keystrokeReceived(proto.TAB, proto.SHIFT)
else:
handler.unhandledControlSequence(buf + b"Z")
def tilde(self, proto, handler, buf):
map = {
1: proto.HOME,
2: proto.INSERT,
3: proto.DELETE,
4: proto.END,
5: proto.PGUP,
6: proto.PGDN,
15: proto.F5,
17: proto.F6,
18: proto.F7,
19: proto.F8,
20: proto.F9,
21: proto.F10,
23: proto.F11,
24: proto.F12,
}
if buf.startswith(b"\x1b["):
ch = buf[2:]
try:
v = int(ch)
except ValueError:
handler.unhandledControlSequence(buf + b"~")
else:
symbolic = map.get(v)
if symbolic is not None:
handler.keystrokeReceived(map[v], None)
else:
handler.unhandledControlSequence(buf + b"~")
else:
handler.unhandledControlSequence(buf + b"~")
controlSequenceParser = ControlSequenceParser()
# ITerminalTransport
def cursorUp(self, n=1):
assert n >= 1
self.cursorPos.y = max(self.cursorPos.y - n, 0)
self.write(b"\x1b[%dA" % (n,))
def cursorDown(self, n=1):
assert n >= 1
self.cursorPos.y = min(self.cursorPos.y + n, self.termSize.y - 1)
self.write(b"\x1b[%dB" % (n,))
def cursorForward(self, n=1):
assert n >= 1
self.cursorPos.x = min(self.cursorPos.x + n, self.termSize.x - 1)
self.write(b"\x1b[%dC" % (n,))
def cursorBackward(self, n=1):
assert n >= 1
self.cursorPos.x = max(self.cursorPos.x - n, 0)
self.write(b"\x1b[%dD" % (n,))
def cursorPosition(self, column, line):
self.write(b"\x1b[%d;%dH" % (line + 1, column + 1))
def cursorHome(self):
self.cursorPos.x = self.cursorPos.y = 0
self.write(b"\x1b[H")
def index(self):
# ECMA48 5th Edition removes this
self.cursorPos.y = min(self.cursorPos.y + 1, self.termSize.y - 1)
self.write(b"\x1bD")
def reverseIndex(self):
self.cursorPos.y = max(self.cursorPos.y - 1, 0)
self.write(b"\x1bM")
def nextLine(self):
self.cursorPos.x = 0
self.cursorPos.y = min(self.cursorPos.y + 1, self.termSize.y - 1)
self.write(b"\n")
def saveCursor(self):
self._savedCursorPos = Vector(self.cursorPos.x, self.cursorPos.y)
self.write(b"\x1b7")
def restoreCursor(self):
self.cursorPos = self._savedCursorPos
del self._savedCursorPos
self.write(b"\x1b8")
def setModes(self, modes):
# XXX Support ANSI-Compatible private modes
modesBytes = b";".join(b"%d" % (mode,) for mode in modes)
self.write(b"\x1b[" + modesBytes + b"h")
def setPrivateModes(self, modes):
modesBytes = b";".join(b"%d" % (mode,) for mode in modes)
self.write(b"\x1b[?" + modesBytes + b"h")
def resetModes(self, modes):
# XXX Support ANSI-Compatible private modes
modesBytes = b";".join(b"%d" % (mode,) for mode in modes)
self.write(b"\x1b[" + modesBytes + b"l")
def resetPrivateModes(self, modes):
modesBytes = b";".join(b"%d" % (mode,) for mode in modes)
self.write(b"\x1b[?" + modesBytes + b"l")
def applicationKeypadMode(self):
self.write(b"\x1b=")
def numericKeypadMode(self):
self.write(b"\x1b>")
def selectCharacterSet(self, charSet, which):
# XXX Rewrite these as dict lookups
if which == G0:
which = b"("
elif which == G1:
which = b")"
else:
raise ValueError("`which' argument to selectCharacterSet must be G0 or G1")
if charSet == CS_UK:
charSet = b"A"
elif charSet == CS_US:
charSet = b"B"
elif charSet == CS_DRAWING:
charSet = b"0"
elif charSet == CS_ALTERNATE:
charSet = b"1"
elif charSet == CS_ALTERNATE_SPECIAL:
charSet = b"2"
else:
raise ValueError("Invalid `charSet' argument to selectCharacterSet")
self.write(b"\x1b" + which + charSet)
def shiftIn(self):
self.write(b"\x15")
def shiftOut(self):
self.write(b"\x14")
def singleShift2(self):
self.write(b"\x1bN")
def singleShift3(self):
self.write(b"\x1bO")
def selectGraphicRendition(self, *attributes):
# each member of attributes must be a native string
attrs = []
for a in attributes:
attrs.append(networkString(a))
self.write(b"\x1b[" + b";".join(attrs) + b"m")
def horizontalTabulationSet(self):
self.write(b"\x1bH")
def tabulationClear(self):
self.write(b"\x1b[q")
def tabulationClearAll(self):
self.write(b"\x1b[3q")
def doubleHeightLine(self, top=True):
if top:
self.write(b"\x1b#3")
else:
self.write(b"\x1b#4")
def singleWidthLine(self):
self.write(b"\x1b#5")
def doubleWidthLine(self):
self.write(b"\x1b#6")
def eraseToLineEnd(self):
self.write(b"\x1b[K")
def eraseToLineBeginning(self):
self.write(b"\x1b[1K")
def eraseLine(self):
self.write(b"\x1b[2K")
def eraseToDisplayEnd(self):
self.write(b"\x1b[J")
def eraseToDisplayBeginning(self):
self.write(b"\x1b[1J")
def eraseDisplay(self):
self.write(b"\x1b[2J")
def deleteCharacter(self, n=1):
self.write(b"\x1b[%dP" % (n,))
def insertLine(self, n=1):
self.write(b"\x1b[%dL" % (n,))
def deleteLine(self, n=1):
self.write(b"\x1b[%dM" % (n,))
def setScrollRegion(self, first=None, last=None):
if first is not None:
first = b"%d" % (first,)
else:
first = b""
if last is not None:
last = b"%d" % (last,)
else:
last = b""
self.write(b"\x1b[%b;%br" % (first, last))
def resetScrollRegion(self):
self.setScrollRegion()
def reportCursorPosition(self):
d = defer.Deferred()
self._cursorReports.append(d)
self.write(b"\x1b[6n")
return d
def reset(self):
self.cursorPos.x = self.cursorPos.y = 0
try:
del self._savedCursorPos
except AttributeError:
pass
self.write(b"\x1bc")
# ITransport
def write(self, data):
if data:
if not isinstance(data, bytes):
data = data.encode("utf-8")
self.lastWrite = data
self.transport.write(b"\r\n".join(data.split(b"\n")))
def writeSequence(self, data):
self.write(b"".join(data))
def loseConnection(self):
self.reset()
self.transport.loseConnection()
def connectionLost(self, reason):
if self.terminalProtocol is not None:
try:
self.terminalProtocol.connectionLost(reason)
finally:
self.terminalProtocol = None
# Add symbolic names for function keys
for name, const in zip(_KEY_NAMES, FUNCTION_KEYS):
setattr(ServerProtocol, name, const)
class ClientProtocol(protocol.Protocol):
terminalFactory = None
terminal = None
state = b"data"
_escBuf = None
_shorts = {
b"D": b"index",
b"M": b"reverseIndex",
b"E": b"nextLine",
b"7": b"saveCursor",
b"8": b"restoreCursor",
b"=": b"applicationKeypadMode",
b">": b"numericKeypadMode",
b"N": b"singleShift2",
b"O": b"singleShift3",
b"H": b"horizontalTabulationSet",
b"c": b"reset",
}
_longs = {
b"[": b"bracket-escape",
b"(": b"select-g0",
b")": b"select-g1",
b"#": b"select-height-width",
}
_charsets = {
b"A": CS_UK,
b"B": CS_US,
b"0": CS_DRAWING,
b"1": CS_ALTERNATE,
b"2": CS_ALTERNATE_SPECIAL,
}
# Factory who instantiated me
factory = None
def __init__(self, terminalFactory=None, *a, **kw):
"""
@param terminalFactory: A callable which will be invoked with
*a, **kw and should return an ITerminalTransport provider.
This will be invoked when this ClientProtocol establishes a
connection.
@param a: Any positional arguments to pass to terminalFactory.
@param kw: Any keyword arguments to pass to terminalFactory.
"""
# assert terminalFactory is None or ITerminalTransport.implementedBy(terminalFactory), "ClientProtocol.__init__ must be passed an ITerminalTransport implementor"
if terminalFactory is not None:
self.terminalFactory = terminalFactory
self.terminalArgs = a
self.terminalKwArgs = kw
def connectionMade(self):
if self.terminalFactory is not None:
self.terminal = self.terminalFactory(
*self.terminalArgs, **self.terminalKwArgs
)
self.terminal.factory = self.factory
self.terminal.makeConnection(self)
def connectionLost(self, reason):
if self.terminal is not None:
try:
self.terminal.connectionLost(reason)
finally:
del self.terminal
def dataReceived(self, data):
"""
Parse the given data from a terminal server, dispatching to event
handlers defined by C{self.terminal}.
"""
toWrite = []
for b in iterbytes(data):
if self.state == b"data":
if b == b"\x1b":
if toWrite:
self.terminal.write(b"".join(toWrite))
del toWrite[:]
self.state = b"escaped"
elif b == b"\x14":
if toWrite:
self.terminal.write(b"".join(toWrite))
del toWrite[:]
self.terminal.shiftOut()
elif b == b"\x15":
if toWrite:
self.terminal.write(b"".join(toWrite))
del toWrite[:]
self.terminal.shiftIn()
elif b == b"\x08":
if toWrite:
self.terminal.write(b"".join(toWrite))
del toWrite[:]
self.terminal.cursorBackward()
else:
toWrite.append(b)
elif self.state == b"escaped":
fName = self._shorts.get(b)
if fName is not None:
self.state = b"data"
getattr(self.terminal, fName.decode("ascii"))()
else:
state = self._longs.get(b)
if state is not None:
self.state = state
else:
self.terminal.unhandledControlSequence(b"\x1b" + b)
self.state = b"data"
elif self.state == b"bracket-escape":
if self._escBuf is None:
self._escBuf = []
if b.isalpha() or b == b"~":
self._handleControlSequence(b"".join(self._escBuf), b)
del self._escBuf
self.state = b"data"
else:
self._escBuf.append(b)
elif self.state == b"select-g0":
self.terminal.selectCharacterSet(self._charsets.get(b, b), G0)
self.state = b"data"
elif self.state == b"select-g1":
self.terminal.selectCharacterSet(self._charsets.get(b, b), G1)
self.state = b"data"
elif self.state == b"select-height-width":
self._handleHeightWidth(b)
self.state = b"data"
else:
raise ValueError("Illegal state")
if toWrite:
self.terminal.write(b"".join(toWrite))
def _handleControlSequence(self, buf, terminal):
f = getattr(
self.controlSequenceParser,
CST.get(terminal, terminal).decode("ascii"),
None,
)
if f is None:
self.terminal.unhandledControlSequence(b"\x1b[" + buf + terminal)
else:
f(self, self.terminal, buf)
class ControlSequenceParser:
def _makeSimple(ch, fName):
n = "cursor" + fName
def simple(self, proto, handler, buf):
if not buf:
getattr(handler, n)(1)
else:
try:
m = int(buf)
except ValueError:
handler.unhandledControlSequence(b"\x1b[" + buf + ch)
else:
getattr(handler, n)(m)
return simple
for ch, fName in (
("A", "Up"),
("B", "Down"),
("C", "Forward"),
("D", "Backward"),
):
exec(ch + " = _makeSimple(ch, fName)")
del _makeSimple
def h(self, proto, handler, buf):
# XXX - Handle '?' to introduce ANSI-Compatible private modes.
try:
modes = [int(mode) for mode in buf.split(b";")]
except ValueError:
handler.unhandledControlSequence(b"\x1b[" + buf + b"h")
else:
handler.setModes(modes)
def l(self, proto, handler, buf):
# XXX - Handle '?' to introduce ANSI-Compatible private modes.
try:
modes = [int(mode) for mode in buf.split(b";")]
except ValueError:
handler.unhandledControlSequence(b"\x1b[" + buf + "l")
else:
handler.resetModes(modes)
def r(self, proto, handler, buf):
parts = buf.split(b";")
if len(parts) == 1:
handler.setScrollRegion(None, None)
elif len(parts) == 2:
try:
if parts[0]:
pt = int(parts[0])
else:
pt = None
if parts[1]:
pb = int(parts[1])
else:
pb = None
except ValueError:
handler.unhandledControlSequence(b"\x1b[" + buf + b"r")
else:
handler.setScrollRegion(pt, pb)
else:
handler.unhandledControlSequence(b"\x1b[" + buf + b"r")
def K(self, proto, handler, buf):
if not buf:
handler.eraseToLineEnd()
elif buf == b"1":
handler.eraseToLineBeginning()
elif buf == b"2":
handler.eraseLine()
else:
handler.unhandledControlSequence(b"\x1b[" + buf + b"K")
def H(self, proto, handler, buf):
handler.cursorHome()
def J(self, proto, handler, buf):
if not buf:
handler.eraseToDisplayEnd()
elif buf == b"1":
handler.eraseToDisplayBeginning()
elif buf == b"2":
handler.eraseDisplay()
else:
handler.unhandledControlSequence(b"\x1b[" + buf + b"J")
def P(self, proto, handler, buf):
if not buf:
handler.deleteCharacter(1)
else:
try:
n = int(buf)
except ValueError:
handler.unhandledControlSequence(b"\x1b[" + buf + b"P")
else:
handler.deleteCharacter(n)
def L(self, proto, handler, buf):
if not buf:
handler.insertLine(1)
else:
try:
n = int(buf)
except ValueError:
handler.unhandledControlSequence(b"\x1b[" + buf + b"L")
else:
handler.insertLine(n)
def M(self, proto, handler, buf):
if not buf:
handler.deleteLine(1)
else:
try:
n = int(buf)
except ValueError:
handler.unhandledControlSequence(b"\x1b[" + buf + b"M")
else:
handler.deleteLine(n)
def n(self, proto, handler, buf):
if buf == b"6":
x, y = handler.reportCursorPosition()
proto.transport.write(b"\x1b[%d;%dR" % (x + 1, y + 1))
else:
handler.unhandledControlSequence(b"\x1b[" + buf + b"n")
def m(self, proto, handler, buf):
if not buf:
handler.selectGraphicRendition(NORMAL)
else:
attrs = []
for a in buf.split(b";"):
try:
a = int(a)
except ValueError:
pass
attrs.append(a)
handler.selectGraphicRendition(*attrs)
controlSequenceParser = ControlSequenceParser()
def _handleHeightWidth(self, b):
if b == b"3":
self.terminal.doubleHeightLine(True)
elif b == b"4":
self.terminal.doubleHeightLine(False)
elif b == b"5":
self.terminal.singleWidthLine()
elif b == b"6":
self.terminal.doubleWidthLine()
else:
self.terminal.unhandledControlSequence(b"\x1b#" + b)
__all__ = [
# Interfaces
"ITerminalProtocol",
"ITerminalTransport",
# Symbolic constants
"modes",
"privateModes",
"FUNCTION_KEYS",
"CS_US",
"CS_UK",
"CS_DRAWING",
"CS_ALTERNATE",
"CS_ALTERNATE_SPECIAL",
"G0",
"G1",
"G2",
"G3",
"UNDERLINE",
"REVERSE_VIDEO",
"BLINK",
"BOLD",
"NORMAL",
# Protocol classes
"ServerProtocol",
"ClientProtocol",
]
Mini Shell