cursed.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

import curses
import asyncio
import sys

class Console():
    def __init__(self, messages):
        self.stdscr = curses.initscr()
        self.messages = messages
        self.typed_message = ""

    def __enter__(self):
        curses.noecho()
        curses.cbreak()
        self.stdscr.keypad(True)
        return self

    def __exit__(self, *_):
        curses.nocbreak()
        self.stdscr.keypad(False)
        curses.echo()
        curses.endwin()

    def redraw(self):
        self.stdscr.clear()
        for msg in self.messages:
            self.stdscr.addstr(msg + '\n')
        if self.typed_message != "":
            self.stdscr.addstr(self.typed_message)
        self.stdscr.refresh()


async def start(c, on_str, on_tab):
    while True:
        c.redraw()
        char = await asyncio.to_thread(c.stdscr.get_wch)
        if isinstance(char, str):
            if char == '\n':
                if c.typed_message != "":
                    await on_str(c.typed_message)
                    c.messages.append(c.typed_message)
                    c.typed_message = ""
            elif char == '\t':
                await on_tab()
            else:
                c.typed_message += char
        else:
            if char == 263:
                c.typed_message = c.typed_message[:-1]