plomlompom.com Git - plomrogue2-experiments/blob - client-curses.py

   1 #!/usr/bin/env python3
   2 import curses
   3 import plom_socket_io
   4 import socket
   5 import threading
   6
   7 ASCII_printable = ' !"#$%&\'\(\)*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWX'\
   8                   'YZ[\\]^_\`abcdefghijklmnopqrstuvwxyz{|}~'
   9
  10
  11 def recv_loop(server_output):
  12     for msg in plom_socket_io.recv(s):
  13         while len(server_output) > 0:
  14             pass
  15         server_output += [msg]
  16
  17
  18 class Widget:
  19
  20     def __init__(self, content, tui, start, size):
  21         self.tui = tui
  22         self.content = content
  23         self.start = start
  24         self.win = curses.newwin(1, 1, self.start[0], self.start[1])
  25         self.size_def = size  # store for re-calling .size on SIGWINCH
  26         self.size = size
  27         self.update = True
  28
  29     @property
  30     def size(self):
  31         return self.win.getmaxyx()
  32
  33     @size.setter
  34     def size(self, size):
  35         """Set window size. Size be y,x tuple. If y or x None, use legal max."""
  36         n_lines, n_cols = size
  37         if n_lines is None:
  38             n_lines = self.tui.stdscr.getmaxyx()[0] - self.start[0]
  39         if n_cols is None:
  40             n_cols = self.tui.stdscr.getmaxyx()[1] - self.start[1]
  41         self.win.resize(n_lines, n_cols)
  42
  43     def __len__(self):
  44         return self.win.getmaxyx()[0] * self.win.getmaxyx()[1]
  45
  46     def safe_write(self, string):
  47         if len(string) < len(self):
  48             self.win.addstr(0, 0, string)
  49         else:  # workaround to <https://stackoverflow.com/q/7063128>
  50             cut = string[:len(self) - 1]
  51             self.win.addch(self.size[0] - 1, self.size[1] - 2,
  52                            string[len(self) - 1])
  53             self.win.insstr(self.size[0] - 1, self.size[1] - 2, " ")
  54             self.win.addstr(0, 0, cut)
  55
  56     def draw(self):
  57         if self.content is not None:
  58             self.safe_write(''.join(self.content))
  59
  60     def draw_and_refresh(self):
  61         self.win.erase()
  62         self.draw()
  63         self.win.refresh()
  64
  65
  66 class LogWidget(Widget):
  67
  68     def draw(self):
  69         line_width = self.size[1]
  70         to_join = []
  71         for line in self.content:
  72             to_pad = line_width - (len(line) % line_width)
  73             if to_pad == line_width:
  74                 to_pad = 0
  75             to_join += [line + ' '*to_pad]
  76         self.safe_write(''.join(to_join))
  77
  78 class MapWidget(Widget):
  79
  80     def draw(self):
  81         from datetime import datetime
  82         with open('log', 'a') as f:
  83             f.write(str(datetime.now()) + ' TRIGGERED ' + str(len(self)) + '\n')
  84         self.safe_write('#'*len(self))
  85
  86
  87 class TUI:
  88
  89     def __init__(self, server_output):
  90         self.server_output = server_output
  91         curses.wrapper(self.loop)
  92
  93     def setup_screen(self, stdscr):
  94         self.stdscr = stdscr
  95         self.stdscr.refresh()  # will be called by getkey else, clearing screen
  96         self.stdscr.timeout(10)
  97         self.stdscr.addstr(0, 0, 'SEND:')
  98
  99     def loop(self, stdscr):
 100         self.setup_screen(stdscr)
 101         curses.curs_set(False)  # hide cursor
 102         to_send = []
 103         log = []
 104         edit_line = Widget(to_send, self, (0, 6), (1, 14))
 105         log_display = LogWidget(log, self, (1, 0), (None, 20))
 106         map_view = MapWidget(None, self, (0, 20), (None, None))
 107         map_view.update = True
 108         widgets = [edit_line, log_display, map_view]
 109         do_update = True
 110         while True:
 111             if do_update:
 112                 for w in widgets:
 113                     w.draw_and_refresh()
 114                 do_update = False
 115             try:
 116                 key = self.stdscr.getkey()
 117                 do_update = True
 118                 if len(key) == 1 and key in ASCII_printable and \
 119                         len(to_send) < len(edit_line):
 120                     to_send += [key]
 121                 elif key == 'KEY_BACKSPACE':
 122                     to_send[:] = to_send[:-1]
 123                 elif key == '\n':
 124                     plom_socket_io.send(s, ''.join(to_send))
 125                     to_send[:] = []
 126                 elif key == 'KEY_RESIZE':
 127                     curses.endwin()
 128                     self.setup_screen(curses.initscr())
 129                     for w in widgets:
 130                         w.size = w.size_def
 131                 else:
 132                     do_update = False
 133             except curses.error:
 134                 pass
 135             if len(self.server_output) > 0:
 136                 log[:0] = [self.server_output[0]]
 137                 self.server_output[:] = []
 138                 do_update = True
 139
 140
 141 server_output = []
 142 s = socket.create_connection(('127.0.0.1', 5000))
 143 t = threading.Thread(target=recv_loop, args=(server_output,))
 144 t.start()
 145 TUI(server_output)