123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415
import {decode, encode} from "./dep.ts";
export interface Keypress {    key: string | undefined,    code: string | undefined,    keyCode: number | undefined,    sequence: string,    unicode: string,    ctrlKey: boolean,    metaKey: boolean,    shiftKey: boolean,}
/** * Get unicode characters sequence * @param str */export function toUnicode(str: string) {    let result = '';
    for (let i = 0; i < str.length; i++) {        let unicode = str.charCodeAt(i).toString(16).toUpperCase();
        while (unicode.length < 4) {            unicode = '0' + unicode;        }
        unicode = '\\u' + unicode;        result += unicode;    }
    return result;}
// Regexes used for ansi escape code splittingconst metaKeyCodeRe = /^(?:\x1b)([a-zA-Z0-9])$/;const functionKeyCodeRe =    /^(?:\x1b+)(O|N|\[|\[\[)(?:(\d+)(?:;(\d+))?([~^$])|(?:1;)?(\d+)?([a-zA-Z]))/;
const charsRe = /^[A-zА-яЁё]$/;
/** * Decode control sequence * @param message */export function decodeKeypress(message: Uint8Array): Keypress[] {    let parts;
    let sequence = decode(message);    let event: Keypress = {        key: undefined,        code: undefined,        keyCode: undefined,        sequence,        unicode: toUnicode(sequence),        ctrlKey: false,        metaKey: false,        shiftKey: false,    };
    if (sequence.length === 1) {        event.key = sequence;        event.keyCode = sequence.charCodeAt(0);    }
    if (sequence === '\r') {        // carriage return        event.key = 'return';
    } else if (sequence === '\n') {        // enter, should have been called linefeed        event.key = 'enter';
    } else if (sequence === '\t') {        // tab        event.key = 'tab';
    } else if (sequence === '\b' || sequence === '\x7f' ||        sequence === '\x1b\x7f' || sequence === '\x1b\b') {        // backspace or ctrlKey+h        event.key = 'backspace';        event.metaKey = (sequence.charAt(0) === '\x1b');
    } else if (sequence === '\x1b' || sequence === '\x1b\x1b') {        // escape key        event.key = 'escape';        event.metaKey = (sequence.length === 2);
    } else if (sequence === ' ' || sequence === '\x1b ') {        event.key = 'space';        event.metaKey = (sequence.length === 2);
    } else if (sequence <= '\x1a') {        // ctrlKey+letter        event.key = String.fromCharCode(sequence.charCodeAt(0) + 'a'.charCodeAt(0) - 1);        event.ctrlKey = true;
    } else if (sequence.length === 1 && charsRe.test(sequence)) {        // letter        event.key = sequence;        event.shiftKey = sequence !== sequence.toLowerCase() && sequence === sequence.toUpperCase();    } else if (parts = metaKeyCodeRe.exec(sequence)) {        // metaKey+character key        event.key = parts[1].toLowerCase();        event.metaKey = true;        event.shiftKey = /^[A-Z]$/.test(parts[1]);
    } else if (parts = functionKeyCodeRe.exec(sequence)) {        // ansi escape sequence
        // reassemble the key code leaving out leading \x1b's,        // the modifier key bitflag and any meaningless "1;" sequence        const code = (parts[1] || '') + (parts[2] || '') +            (parts[4] || '') + (parts[6] || '');        const modifier: number = (parts[3] || parts[5] || 1) as number - 1;
        // Parse the key modifier        event.ctrlKey = !!(modifier & 4);        event.metaKey = !!(modifier & 10);        event.shiftKey = !!(modifier & 1);        event.code = code;
        // Parse the key itself        switch (code) {            /* xterm/gnome ESC O letter */            case 'OP':                event.key = 'f1';                break;            case 'OQ':                event.key = 'f2';                break;            case 'OR':                event.key = 'f3';                break;            case 'OS':                event.key = 'f4';                break;
            /* xterm/rxvt ESC [ number ~ */            case '[11~':                event.key = 'f1';                break;            case '[12~':                event.key = 'f2';                break;            case '[13~':                event.key = 'f3';                break;            case '[14~':                event.key = 'f4';                break;
            /* from Cygwin and used in libuv */            case '[[A':                event.key = 'f1';                break;            case '[[B':                event.key = 'f2';                break;            case '[[C':                event.key = 'f3';                break;            case '[[D':                event.key = 'f4';                break;            case '[[E':                event.key = 'f5';                break;
            /* common */            case '[15~':                event.key = 'f5';                break;            case '[17~':                event.key = 'f6';                break;            case '[18~':                event.key = 'f7';                break;            case '[19~':                event.key = 'f8';                break;            case '[20~':                event.key = 'f9';                break;            case '[21~':                event.key = 'f10';                break;            case '[23~':                event.key = 'f11';                break;            case '[24~':                event.key = 'f12';                break;
            /* xterm ESC [ letter */            case '[A':                event.key = 'up';                break;            case '[B':                event.key = 'down';                break;            case '[C':                event.key = 'right';                break;            case '[D':                event.key = 'left';                break;            case '[E':                event.key = 'clear';                break;            case '[F':                event.key = 'end';                break;            case '[H':                event.key = 'home';                break;
            /* xterm/gnome ESC O letter */            case 'OA':                event.key = 'up';                break;            case 'OB':                event.key = 'down';                break;            case 'OC':                event.key = 'right';                break;            case 'OD':                event.key = 'left';                break;            case 'OE':                event.key = 'clear';                break;            case 'OF':                event.key = 'end';                break;            case 'OH':                event.key = 'home';                break;
            /* xterm/rxvt ESC [ number ~ */            case '[1~':                event.key = 'home';                break;            case '[2~':                event.key = 'insert';                break;            case '[3~':                event.key = 'delete';                break;            case '[4~':                event.key = 'end';                break;            case '[5~':                event.key = 'pageup';                break;            case '[6~':                event.key = 'pagedown';                break;
            /* putty */            case '[[5~':                event.key = 'pageup';                break;            case '[[6~':                event.key = 'pagedown';                break;
            /* rxvt */            case '[7~':                event.key = 'home';                break;            case '[8~':                event.key = 'end';                break;
            /* rxvt keys with modifiers */            case '[a':                event.key = 'up';                event.shiftKey = true;                break;            case '[b':                event.key = 'down';                event.shiftKey = true;                break;            case '[c':                event.key = 'right';                event.shiftKey = true;                break;            case '[d':                event.key = 'left';                event.shiftKey = true;                break;            case '[e':                event.key = 'clear';                event.shiftKey = true;                break;
            case '[2$':                event.key = 'insert';                event.shiftKey = true;                break;            case '[3$':                event.key = 'delete';                event.shiftKey = true;                break;            case '[5$':                event.key = 'pageup';                event.shiftKey = true;                break;            case '[6$':                event.key = 'pagedown';                event.shiftKey = true;                break;            case '[7$':                event.key = 'home';                event.shiftKey = true;                break;            case '[8$':                event.key = 'end';                event.shiftKey = true;                break;
            case 'Oa':                event.key = 'up';                event.ctrlKey = true;                break;            case 'Ob':                event.key = 'down';                event.ctrlKey = true;                break;            case 'Oc':                event.key = 'right';                event.ctrlKey = true;                break;            case 'Od':                event.key = 'left';                event.ctrlKey = true;                break;            case 'Oe':                event.key = 'clear';                event.ctrlKey = true;                break;
            case '[2^':                event.key = 'insert';                event.ctrlKey = true;                break;            case '[3^':                event.key = 'delete';                event.ctrlKey = true;                break;            case '[5^':                event.key = 'pageup';                event.ctrlKey = true;                break;            case '[6^':                event.key = 'pagedown';                event.ctrlKey = true;                break;            case '[7^':                event.key = 'home';                event.ctrlKey = true;                break;            case '[8^':                event.key = 'end';                event.ctrlKey = true;                break;
            /* misc. */            case '[Z':                event.key = 'tab';                event.shiftKey = true;                break;            default:                event.key = 'undefined';                break;        }    } else if (sequence.length > 1 && sequence[0] !== '\x1b') {        // Got a longer-than-one string of characters.        // Probably a paste, since it wasn't a control sequence.        const results: Keypress[][] = sequence.split('')            .map(character => decodeKeypress(encode(character)));
        return results.flat();    }
    return [event];}
/** * Read character sequence and decode it as keypress * @param reader - TTY reader * @param bufferLength - used because of opportunity to paste text in terminal */export async function* readKeypress(reader: Deno.Reader & { rid: number } = Deno.stdin, bufferLength: number = 1024): AsyncIterableIterator<Keypress> {    if (!Deno.isatty(reader.rid)) {        throw new Error('Keypress can be read only under TTY.')    }
    while (true) {        const buffer = new Uint8Array(bufferLength);        Deno.setRaw(reader.rid, true);        const length = <number>await reader.read(buffer);        Deno.setRaw(reader.rid, false);
        const events = decodeKeypress(buffer.subarray(0, length));
        for (const event of events) {            yield event;        }    }}