Mini Shell

"""Functions for obtaining CLI input"""

import termios
import sys
import os
from . import color


def get_keypress() -> str:
    """Get a single keypress, the ugly POSIX way"""
    fileno = sys.stdin.fileno()
    old = termios.tcgetattr(fileno)
    new = termios.tcgetattr(fileno)
    new[3] = new[3] & ~termios.ICANON & ~termios.ECHO
    new[6][termios.VMIN] = 1
    new[6][termios.VTIME] = 0
    termios.tcsetattr(fileno, termios.TCSANOW, new)
    key = None
    try:
        key = str(os.read(fileno, 4), 'ascii')
    finally:
        termios.tcsetattr(fileno, termios.TCSAFLUSH, old)
    keylist = {
        '\t': 'TAB',
        '\n': 'ENTER',
        '\x7f': 'BACKSPACE',
        '\x1b': 'ESC',
        '\x1b[Z': 'BACKTAB',
        '\x1bOH': 'HOME',
        '\x1bOP': 'F1',
        '\x1bOQ': 'F2',
        '\x1bOR': 'F3',
        '\x1bOS': 'F4',
        '\x1b[15': 'F5',
        '\x1b[17': 'F6',
        '\x1b[18': 'F7',
        '\x1b[19': 'F8',
        '\x1b[20': 'F9',
        '\x1b[21': 'F10',
        '\x1b[23': 'F11',
        '\x1b[24': 'F12',
        '\x1b[A': 'ARROW_UP',
        '\x1b[B': 'ARROW_DN',
        '\x1b[C': 'ARROW_RT',
        '\x1b[D': 'ARROW_LT',
    }
    return key if key not in keylist else keylist[key]


get_keypress.__module__ = 'rads'


def prompt_y_n(question: str) -> bool:
    """Prompt for a yes or no answer

    Args:
        question: text to display before [y/n] prompt

    Raises:
        SystemExit: if SIGINT is received
    """
    valid = {"yes": True, "y": True, "no": False, "n": False}
    while True:
        print(question, '[y/n]')
        try:
            choice = input().lower()
        except KeyboardInterrupt:
            print("\nCancelled")
            sys.exit(1)
        if choice in valid:
            return valid[choice]
        print('Invalid answer. Try again:', color.bold('[y/n]'))


prompt_y_n.__module__ = 'rads'