| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836 |
- package input
- import (
- "encoding/base64"
- "strings"
- "unicode/utf8"
- "github.com/charmbracelet/x/ansi"
- "github.com/charmbracelet/x/ansi/parser"
- "github.com/erikgeiser/coninput"
- )
- // Flags to control the behavior of the parser.
- const (
- // When this flag is set, the driver will treat both Ctrl+Space and Ctrl+@
- // as the same key sequence.
- //
- // Historically, the ANSI specs generate NUL (0x00) on both the Ctrl+Space
- // and Ctrl+@ key sequences. This flag allows the driver to treat both as
- // the same key sequence.
- FlagCtrlAt = 1 << iota
- // When this flag is set, the driver will treat the Tab key and Ctrl+I as
- // the same key sequence.
- //
- // Historically, the ANSI specs generate HT (0x09) on both the Tab key and
- // Ctrl+I. This flag allows the driver to treat both as the same key
- // sequence.
- FlagCtrlI
- // When this flag is set, the driver will treat the Enter key and Ctrl+M as
- // the same key sequence.
- //
- // Historically, the ANSI specs generate CR (0x0D) on both the Enter key
- // and Ctrl+M. This flag allows the driver to treat both as the same key
- FlagCtrlM
- // When this flag is set, the driver will treat Escape and Ctrl+[ as
- // the same key sequence.
- //
- // Historically, the ANSI specs generate ESC (0x1B) on both the Escape key
- // and Ctrl+[. This flag allows the driver to treat both as the same key
- // sequence.
- FlagCtrlOpenBracket
- // When this flag is set, the driver will send a BS (0x08 byte) character
- // instead of a DEL (0x7F byte) character when the Backspace key is
- // pressed.
- //
- // The VT100 terminal has both a Backspace and a Delete key. The VT220
- // terminal dropped the Backspace key and replaced it with the Delete key.
- // Both terminals send a DEL character when the Delete key is pressed.
- // Modern terminals and PCs later readded the Delete key but used a
- // different key sequence, and the Backspace key was standardized to send a
- // DEL character.
- FlagBackspace
- // When this flag is set, the driver will recognize the Find key instead of
- // treating it as a Home key.
- //
- // The Find key was part of the VT220 keyboard, and is no longer used in
- // modern day PCs.
- FlagFind
- // When this flag is set, the driver will recognize the Select key instead
- // of treating it as a End key.
- //
- // The Symbol key was part of the VT220 keyboard, and is no longer used in
- // modern day PCs.
- FlagSelect
- // When this flag is set, the driver will use Terminfo databases to
- // overwrite the default key sequences.
- FlagTerminfo
- // When this flag is set, the driver will preserve function keys (F13-F63)
- // as symbols.
- //
- // Since these keys are not part of today's standard 20th century keyboard,
- // we treat them as F1-F12 modifier keys i.e. ctrl/shift/alt + Fn combos.
- // Key definitions come from Terminfo, this flag is only useful when
- // FlagTerminfo is not set.
- FlagFKeys
- )
- var flags int
- // SetFlags sets the flags for the parser.
- // This will control the behavior of ParseSequence.
- func SetFlags(f int) {
- flags = f
- }
- // ParseSequence finds the first recognized event sequence and returns it along
- // with its length.
- //
- // It will return zero and nil no sequence is recognized or when the buffer is
- // empty. If a sequence is not supported, an UnknownEvent is returned.
- func ParseSequence(buf []byte) (n int, e Event) {
- if len(buf) == 0 {
- return 0, nil
- }
- switch b := buf[0]; b {
- case ansi.ESC:
- if len(buf) == 1 {
- // Escape key
- return 1, KeyPressEvent{Sym: KeyEscape}
- }
- switch b := buf[1]; b {
- case 'O': // Esc-prefixed SS3
- return parseSs3(buf)
- case 'P': // Esc-prefixed DCS
- return parseDcs(buf)
- case '[': // Esc-prefixed CSI
- return parseCsi(buf)
- case ']': // Esc-prefixed OSC
- return parseOsc(buf)
- case '_': // Esc-prefixed APC
- return parseApc(buf)
- default:
- n, e := ParseSequence(buf[1:])
- if k, ok := e.(KeyPressEvent); ok && !k.Mod.HasAlt() {
- k.Mod |= ModAlt
- return n + 1, k
- }
- // Not a key sequence, nor an alt modified key sequence. In that
- // case, just report a single escape key.
- return 1, KeyPressEvent{Sym: KeyEscape}
- }
- case ansi.SS3:
- return parseSs3(buf)
- case ansi.DCS:
- return parseDcs(buf)
- case ansi.CSI:
- return parseCsi(buf)
- case ansi.OSC:
- return parseOsc(buf)
- case ansi.APC:
- return parseApc(buf)
- default:
- if b <= ansi.US || b == ansi.DEL || b == ansi.SP {
- return 1, parseControl(b)
- } else if b >= ansi.PAD && b <= ansi.APC {
- // C1 control code
- // UTF-8 never starts with a C1 control code
- // Encode these as Ctrl+Alt+<code - 0x40>
- return 1, KeyPressEvent{Rune: rune(b) - 0x40, Mod: ModCtrl | ModAlt}
- }
- return parseUtf8(buf)
- }
- }
- func parseCsi(b []byte) (int, Event) {
- if len(b) == 2 && b[0] == ansi.ESC {
- // short cut if this is an alt+[ key
- return 2, KeyPressEvent{Rune: rune(b[1]), Mod: ModAlt}
- }
- var csi ansi.CsiSequence
- var params [parser.MaxParamsSize]int
- var paramsLen int
- var i int
- if b[i] == ansi.CSI || b[i] == ansi.ESC {
- i++
- }
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == '[' {
- i++
- }
- // Initial CSI byte
- if i < len(b) && b[i] >= '<' && b[i] <= '?' {
- csi.Cmd |= int(b[i]) << parser.MarkerShift
- }
- // Scan parameter bytes in the range 0x30-0x3F
- var j int
- for j = 0; i < len(b) && paramsLen < len(params) && b[i] >= 0x30 && b[i] <= 0x3F; i, j = i+1, j+1 {
- if b[i] >= '0' && b[i] <= '9' {
- if params[paramsLen] == parser.MissingParam {
- params[paramsLen] = 0
- }
- params[paramsLen] *= 10
- params[paramsLen] += int(b[i]) - '0'
- }
- if b[i] == ':' {
- params[paramsLen] |= parser.HasMoreFlag
- }
- if b[i] == ';' || b[i] == ':' {
- paramsLen++
- if paramsLen < len(params) {
- // Don't overflow the params slice
- params[paramsLen] = parser.MissingParam
- }
- }
- }
- if j > 0 && paramsLen < len(params) {
- // has parameters
- paramsLen++
- }
- // Scan intermediate bytes in the range 0x20-0x2F
- var intermed byte
- for ; i < len(b) && b[i] >= 0x20 && b[i] <= 0x2F; i++ {
- intermed = b[i]
- }
- // Set the intermediate byte
- csi.Cmd |= int(intermed) << parser.IntermedShift
- // Scan final byte in the range 0x40-0x7E
- if i >= len(b) || b[i] < 0x40 || b[i] > 0x7E {
- // Special case for URxvt keys
- // CSI <number> $ is an invalid sequence, but URxvt uses it for
- // shift modified keys.
- if b[i-1] == '$' {
- n, ev := parseCsi(append(b[:i-1], '~'))
- if k, ok := ev.(KeyPressEvent); ok {
- k.Mod |= ModShift
- return n, k
- }
- }
- return i, UnknownEvent(b[:i-1])
- }
- // Add the final byte
- csi.Cmd |= int(b[i])
- i++
- csi.Params = params[:paramsLen]
- marker, cmd := csi.Marker(), csi.Command()
- switch marker {
- case '?':
- switch cmd {
- case 'y':
- switch intermed {
- case '$':
- // Report Mode (DECRPM)
- if paramsLen != 2 {
- return i, UnknownCsiEvent(b[:i])
- }
- return i, ReportModeEvent{Mode: csi.Param(0), Value: csi.Param(1)}
- }
- case 'c':
- // Primary Device Attributes
- return i, parsePrimaryDevAttrs(&csi)
- case 'u':
- // Kitty keyboard flags
- if param := csi.Param(0); param != -1 {
- return i, KittyKeyboardEvent(param)
- }
- case 'R':
- // This report may return a third parameter representing the page
- // number, but we don't really need it.
- if paramsLen >= 2 {
- return i, CursorPositionEvent{Row: csi.Param(0), Column: csi.Param(1)}
- }
- }
- return i, UnknownCsiEvent(b[:i])
- case '<':
- switch cmd {
- case 'm', 'M':
- // Handle SGR mouse
- if paramsLen != 3 {
- return i, UnknownCsiEvent(b[:i])
- }
- return i, parseSGRMouseEvent(&csi)
- default:
- return i, UnknownCsiEvent(b[:i])
- }
- case '>':
- switch cmd {
- case 'm':
- // XTerm modifyOtherKeys
- if paramsLen != 2 || csi.Param(0) != 4 {
- return i, UnknownCsiEvent(b[:i])
- }
- return i, ModifyOtherKeysEvent(csi.Param(1))
- default:
- return i, UnknownCsiEvent(b[:i])
- }
- case '=':
- // We don't support any of these yet
- return i, UnknownCsiEvent(b[:i])
- }
- switch cmd := csi.Command(); cmd {
- case 'R':
- // Cursor position report OR modified F3
- if paramsLen == 0 {
- return i, KeyPressEvent{Sym: KeyF3}
- } else if paramsLen != 2 {
- break
- }
- // XXX: We cannot differentiate between cursor position report and
- // CSI 1 ; <mod> R (which is modified F3) when the cursor is at the
- // row 1. In this case, we report a modified F3 event since it's more
- // likely to be the case than the cursor being at the first row.
- //
- // For a non ambiguous cursor position report, use
- // [ansi.RequestExtendedCursorPosition] (DECXCPR) instead.
- if csi.Param(0) != 1 {
- return i, CursorPositionEvent{Row: csi.Param(0), Column: csi.Param(1)}
- }
- fallthrough
- case 'a', 'b', 'c', 'd', 'A', 'B', 'C', 'D', 'E', 'F', 'H', 'P', 'Q', 'S', 'Z':
- var k KeyPressEvent
- switch cmd {
- case 'a', 'b', 'c', 'd':
- k = KeyPressEvent{Sym: KeyUp + KeySym(cmd-'a'), Mod: ModShift}
- case 'A', 'B', 'C', 'D':
- k = KeyPressEvent{Sym: KeyUp + KeySym(cmd-'A')}
- case 'E':
- k = KeyPressEvent{Sym: KeyBegin}
- case 'F':
- k = KeyPressEvent{Sym: KeyEnd}
- case 'H':
- k = KeyPressEvent{Sym: KeyHome}
- case 'P', 'Q', 'R', 'S':
- k = KeyPressEvent{Sym: KeyF1 + KeySym(cmd-'P')}
- case 'Z':
- k = KeyPressEvent{Sym: KeyTab, Mod: ModShift}
- }
- if paramsLen > 1 && csi.Param(0) == 1 {
- // CSI 1 ; <modifiers> A
- if paramsLen > 1 {
- k.Mod |= KeyMod(csi.Param(1) - 1)
- }
- }
- return i, k
- case 'M':
- // Handle X10 mouse
- if i+3 > len(b) {
- return i, UnknownCsiEvent(b[:i])
- }
- return i + 3, parseX10MouseEvent(append(b[:i], b[i:i+3]...))
- case 'y':
- // Report Mode (DECRPM)
- if paramsLen != 2 {
- return i, UnknownCsiEvent(b[:i])
- }
- return i, ReportModeEvent{Mode: csi.Param(0), Value: csi.Param(1)}
- case 'u':
- // Kitty keyboard protocol & CSI u (fixterms)
- if paramsLen == 0 {
- return i, UnknownCsiEvent(b[:i])
- }
- return i, parseKittyKeyboard(&csi)
- case '_':
- // Win32 Input Mode
- if paramsLen != 6 {
- return i, UnknownCsiEvent(b[:i])
- }
- rc := uint16(csi.Param(5))
- if rc == 0 {
- rc = 1
- }
- event := parseWin32InputKeyEvent(
- coninput.VirtualKeyCode(csi.Param(0)), // Vk wVirtualKeyCode
- coninput.VirtualKeyCode(csi.Param(1)), // Sc wVirtualScanCode
- rune(csi.Param(2)), // Uc UnicodeChar
- csi.Param(3) == 1, // Kd bKeyDown
- coninput.ControlKeyState(csi.Param(4)), // Cs dwControlKeyState
- rc, // Rc wRepeatCount
- )
- if event == nil {
- return i, UnknownCsiEvent(b[:])
- }
- return i, event
- case '@', '^', '~':
- if paramsLen == 0 {
- return i, UnknownCsiEvent(b[:i])
- }
- param := csi.Param(0)
- switch cmd {
- case '~':
- switch param {
- case 27:
- // XTerm modifyOtherKeys 2
- if paramsLen != 3 {
- return i, UnknownCsiEvent(b[:i])
- }
- return i, parseXTermModifyOtherKeys(&csi)
- case 200:
- // bracketed-paste start
- return i, PasteStartEvent{}
- case 201:
- // bracketed-paste end
- return i, PasteEndEvent{}
- }
- }
- switch param {
- case 1, 2, 3, 4, 5, 6, 7, 8:
- fallthrough
- case 11, 12, 13, 14, 15:
- fallthrough
- case 17, 18, 19, 20, 21, 23, 24, 25, 26:
- fallthrough
- case 28, 29, 31, 32, 33, 34:
- var k KeyPressEvent
- switch param {
- case 1:
- if flags&FlagFind != 0 {
- k = KeyPressEvent{Sym: KeyFind}
- } else {
- k = KeyPressEvent{Sym: KeyHome}
- }
- case 2:
- k = KeyPressEvent{Sym: KeyInsert}
- case 3:
- k = KeyPressEvent{Sym: KeyDelete}
- case 4:
- if flags&FlagSelect != 0 {
- k = KeyPressEvent{Sym: KeySelect}
- } else {
- k = KeyPressEvent{Sym: KeyEnd}
- }
- case 5:
- k = KeyPressEvent{Sym: KeyPgUp}
- case 6:
- k = KeyPressEvent{Sym: KeyPgDown}
- case 7:
- k = KeyPressEvent{Sym: KeyHome}
- case 8:
- k = KeyPressEvent{Sym: KeyEnd}
- case 11, 12, 13, 14, 15:
- k = KeyPressEvent{Sym: KeyF1 + KeySym(param-11)}
- case 17, 18, 19, 20, 21:
- k = KeyPressEvent{Sym: KeyF6 + KeySym(param-17)}
- case 23, 24, 25, 26:
- k = KeyPressEvent{Sym: KeyF11 + KeySym(param-23)}
- case 28, 29:
- k = KeyPressEvent{Sym: KeyF15 + KeySym(param-28)}
- case 31, 32, 33, 34:
- k = KeyPressEvent{Sym: KeyF17 + KeySym(param-31)}
- }
- // modifiers
- if paramsLen > 1 {
- k.Mod |= KeyMod(csi.Param(1) - 1)
- }
- // Handle URxvt weird keys
- switch cmd {
- case '^':
- k.Mod |= ModCtrl
- case '@':
- k.Mod |= ModCtrl | ModShift
- }
- return i, k
- }
- }
- return i, UnknownCsiEvent(b[:i])
- }
- // parseSs3 parses a SS3 sequence.
- // See https://vt100.net/docs/vt220-rm/chapter4.html#S4.4.4.2
- func parseSs3(b []byte) (int, Event) {
- if len(b) == 2 && b[0] == ansi.ESC {
- // short cut if this is an alt+O key
- return 2, KeyPressEvent{Rune: rune(b[1]), Mod: ModAlt}
- }
- var i int
- if b[i] == ansi.SS3 || b[i] == ansi.ESC {
- i++
- }
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == 'O' {
- i++
- }
- // Scan numbers from 0-9
- var mod int
- for ; i < len(b) && b[i] >= '0' && b[i] <= '9'; i++ {
- mod *= 10
- mod += int(b[i]) - '0'
- }
- // Scan a GL character
- // A GL character is a single byte in the range 0x21-0x7E
- // See https://vt100.net/docs/vt220-rm/chapter2.html#S2.3.2
- if i >= len(b) || b[i] < 0x21 || b[i] > 0x7E {
- return i, UnknownEvent(b[:i])
- }
- // GL character(s)
- gl := b[i]
- i++
- var k KeyPressEvent
- switch gl {
- case 'a', 'b', 'c', 'd':
- k = KeyPressEvent{Sym: KeyUp + KeySym(gl-'a'), Mod: ModCtrl}
- case 'A', 'B', 'C', 'D':
- k = KeyPressEvent{Sym: KeyUp + KeySym(gl-'A')}
- case 'E':
- k = KeyPressEvent{Sym: KeyBegin}
- case 'F':
- k = KeyPressEvent{Sym: KeyEnd}
- case 'H':
- k = KeyPressEvent{Sym: KeyHome}
- case 'P', 'Q', 'R', 'S':
- k = KeyPressEvent{Sym: KeyF1 + KeySym(gl-'P')}
- case 'M':
- k = KeyPressEvent{Sym: KeyKpEnter}
- case 'X':
- k = KeyPressEvent{Sym: KeyKpEqual}
- case 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y':
- k = KeyPressEvent{Sym: KeyKpMultiply + KeySym(gl-'j')}
- default:
- return i, UnknownSs3Event(b[:i])
- }
- // Handle weird SS3 <modifier> Func
- if mod > 0 {
- k.Mod |= KeyMod(mod - 1)
- }
- return i, k
- }
- func parseOsc(b []byte) (int, Event) {
- if len(b) == 2 && b[0] == ansi.ESC {
- // short cut if this is an alt+] key
- return 2, KeyPressEvent{Rune: rune(b[1]), Mod: ModAlt}
- }
- var i int
- if b[i] == ansi.OSC || b[i] == ansi.ESC {
- i++
- }
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == ']' {
- i++
- }
- // Parse OSC command
- // An OSC sequence is terminated by a BEL, ESC, or ST character
- var start, end int
- cmd := -1
- for ; i < len(b) && b[i] >= '0' && b[i] <= '9'; i++ {
- if cmd == -1 {
- cmd = 0
- } else {
- cmd *= 10
- }
- cmd += int(b[i]) - '0'
- }
- if i < len(b) && b[i] == ';' {
- // mark the start of the sequence data
- i++
- start = i
- }
- for ; i < len(b); i++ {
- // advance to the end of the sequence
- if b[i] == ansi.BEL || b[i] == ansi.ESC || b[i] == ansi.ST {
- break
- }
- }
- if i >= len(b) {
- return i, UnknownEvent(b[:i])
- }
- end = i // end of the sequence data
- i++
- // Check 7-bit ST (string terminator) character
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == '\\' {
- i++
- }
- if end <= start {
- return i, UnknownOscEvent(b[:i])
- }
- data := string(b[start:end])
- switch cmd {
- case 10:
- return i, ForegroundColorEvent{xParseColor(data)}
- case 11:
- return i, BackgroundColorEvent{xParseColor(data)}
- case 12:
- return i, CursorColorEvent{xParseColor(data)}
- case 52:
- parts := strings.Split(data, ";")
- if len(parts) == 0 {
- return i, ClipboardEvent("")
- }
- b64 := parts[len(parts)-1]
- bts, err := base64.StdEncoding.DecodeString(b64)
- if err != nil {
- return i, ClipboardEvent("")
- }
- return i, ClipboardEvent(bts)
- default:
- return i, UnknownOscEvent(b[:i])
- }
- }
- // parseStTerminated parses a control sequence that gets terminated by a ST character.
- func parseStTerminated(intro8, intro7 byte) func([]byte) (int, Event) {
- return func(b []byte) (int, Event) {
- var i int
- if b[i] == intro8 || b[i] == ansi.ESC {
- i++
- }
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == intro7 {
- i++
- }
- // Scan control sequence
- // Most common control sequence is terminated by a ST character
- // ST is a 7-bit string terminator character is (ESC \)
- // nolint: revive
- for ; i < len(b) && b[i] != ansi.ST && b[i] != ansi.ESC; i++ {
- }
- if i >= len(b) {
- switch intro8 {
- case ansi.DCS:
- return i, UnknownDcsEvent(b[:i])
- case ansi.APC:
- return i, UnknownApcEvent(b[:i])
- default:
- return i, UnknownEvent(b[:i])
- }
- }
- i++
- // Check 7-bit ST (string terminator) character
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == '\\' {
- i++
- }
- switch intro8 {
- case ansi.DCS:
- return i, UnknownDcsEvent(b[:i])
- case ansi.APC:
- return i, UnknownApcEvent(b[:i])
- default:
- return i, UnknownEvent(b[:i])
- }
- }
- }
- func parseDcs(b []byte) (int, Event) {
- if len(b) == 2 && b[0] == ansi.ESC {
- // short cut if this is an alt+P key
- return 2, KeyPressEvent{Rune: rune(b[1]), Mod: ModAlt}
- }
- var params [16]int
- var paramsLen int
- var dcs ansi.DcsSequence
- // DCS sequences are introduced by DCS (0x90) or ESC P (0x1b 0x50)
- var i int
- if b[i] == ansi.DCS || b[i] == ansi.ESC {
- i++
- }
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == 'P' {
- i++
- }
- // initial DCS byte
- if i < len(b) && b[i] >= '<' && b[i] <= '?' {
- dcs.Cmd |= int(b[i]) << parser.MarkerShift
- }
- // Scan parameter bytes in the range 0x30-0x3F
- var j int
- for j = 0; i < len(b) && paramsLen < len(params) && b[i] >= 0x30 && b[i] <= 0x3F; i, j = i+1, j+1 {
- if b[i] >= '0' && b[i] <= '9' {
- if params[paramsLen] == parser.MissingParam {
- params[paramsLen] = 0
- }
- params[paramsLen] *= 10
- params[paramsLen] += int(b[i]) - '0'
- }
- if b[i] == ':' {
- params[paramsLen] |= parser.HasMoreFlag
- }
- if b[i] == ';' || b[i] == ':' {
- paramsLen++
- if paramsLen < len(params) {
- // Don't overflow the params slice
- params[paramsLen] = parser.MissingParam
- }
- }
- }
- if j > 0 && paramsLen < len(params) {
- // has parameters
- paramsLen++
- }
- // Scan intermediate bytes in the range 0x20-0x2F
- var intermed byte
- for j := 0; i < len(b) && b[i] >= 0x20 && b[i] <= 0x2F; i, j = i+1, j+1 {
- intermed = b[i]
- }
- // set intermediate byte
- dcs.Cmd |= int(intermed) << parser.IntermedShift
- // Scan final byte in the range 0x40-0x7E
- if i >= len(b) || b[i] < 0x40 || b[i] > 0x7E {
- return i, UnknownEvent(b[:i])
- }
- // Add the final byte
- dcs.Cmd |= int(b[i])
- i++
- start := i // start of the sequence data
- for ; i < len(b); i++ {
- if b[i] == ansi.ST || b[i] == ansi.ESC {
- break
- }
- }
- if i >= len(b) {
- return i, UnknownEvent(b[:i])
- }
- end := i // end of the sequence data
- i++
- // Check 7-bit ST (string terminator) character
- if i < len(b) && b[i-1] == ansi.ESC && b[i] == '\\' {
- i++
- }
- dcs.Params = params[:paramsLen]
- switch cmd := dcs.Command(); cmd {
- case 'r':
- switch dcs.Intermediate() {
- case '+':
- // XTGETTCAP responses
- switch param := dcs.Param(0); param {
- case 0, 1:
- tc := parseTermcap(b[start:end])
- // XXX: some terminals like KiTTY report invalid responses with
- // their queries i.e. sending a query for "Tc" using "\x1bP+q5463\x1b\\"
- // returns "\x1bP0+r5463\x1b\\".
- // The specs says that invalid responses should be in the form of
- // DCS 0 + r ST "\x1bP0+r\x1b\\"
- //
- // See: https://invisible-island.net/xterm/ctlseqs/ctlseqs.html#h3-Operating-System-Commands
- tc.IsValid = param == 1
- return i, tc
- }
- }
- }
- return i, UnknownDcsEvent(b[:i])
- }
- func parseApc(b []byte) (int, Event) {
- if len(b) == 2 && b[0] == ansi.ESC {
- // short cut if this is an alt+_ key
- return 2, KeyPressEvent{Rune: rune(b[1]), Mod: ModAlt}
- }
- // APC sequences are introduced by APC (0x9f) or ESC _ (0x1b 0x5f)
- return parseStTerminated(ansi.APC, '_')(b)
- }
- func parseUtf8(b []byte) (int, Event) {
- r, rw := utf8.DecodeRune(b)
- if r <= ansi.US || r == ansi.DEL || r == ansi.SP {
- // Control codes get handled by parseControl
- return 1, parseControl(byte(r))
- } else if r == utf8.RuneError {
- return 1, UnknownEvent(b[0])
- }
- return rw, KeyPressEvent{Rune: r}
- }
- func parseControl(b byte) Event {
- switch b {
- case ansi.NUL:
- if flags&FlagCtrlAt != 0 {
- return KeyPressEvent{Rune: '@', Mod: ModCtrl}
- }
- return KeyPressEvent{Rune: ' ', Sym: KeySpace, Mod: ModCtrl}
- case ansi.BS:
- return KeyPressEvent{Rune: 'h', Mod: ModCtrl}
- case ansi.HT:
- if flags&FlagCtrlI != 0 {
- return KeyPressEvent{Rune: 'i', Mod: ModCtrl}
- }
- return KeyPressEvent{Sym: KeyTab}
- case ansi.CR:
- if flags&FlagCtrlM != 0 {
- return KeyPressEvent{Rune: 'm', Mod: ModCtrl}
- }
- return KeyPressEvent{Sym: KeyEnter}
- case ansi.ESC:
- if flags&FlagCtrlOpenBracket != 0 {
- return KeyPressEvent{Rune: '[', Mod: ModCtrl}
- }
- return KeyPressEvent{Sym: KeyEscape}
- case ansi.DEL:
- if flags&FlagBackspace != 0 {
- return KeyPressEvent{Sym: KeyDelete}
- }
- return KeyPressEvent{Sym: KeyBackspace}
- case ansi.SP:
- return KeyPressEvent{Sym: KeySpace, Rune: ' '}
- default:
- if b >= ansi.SOH && b <= ansi.SUB {
- // Use lower case letters for control codes
- return KeyPressEvent{Rune: rune(b + 0x60), Mod: ModCtrl}
- } else if b >= ansi.FS && b <= ansi.US {
- return KeyPressEvent{Rune: rune(b + 0x40), Mod: ModCtrl}
- }
- return UnknownEvent(b)
- }
- }
|