7 ASCII_printable = ' !"#$%&\'\(\)*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWX'\
8 'YZ[\\]^_\`abcdefghijklmnopqrstuvwxyz{|}~'
11 def recv_loop(server_output):
12 for msg in plom_socket_io.recv(s):
13 while len(server_output) > 0:
15 server_output += [msg]
20 def __init__(self, content, tui, start, size):
22 self.content = content
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
31 return self.win.getmaxyx()
35 """Set window size. Size be y,x tuple. If y or x None, use legal max."""
36 n_lines, n_cols = size
38 n_lines = self.tui.stdscr.getmaxyx()[0] - self.start[0]
40 n_cols = self.tui.stdscr.getmaxyx()[1] - self.start[1]
41 self.win.resize(n_lines, n_cols)
44 return self.win.getmaxyx()[0] * self.win.getmaxyx()[1]
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)
57 if self.content is not None:
58 self.safe_write(''.join(self.content))
60 def draw_and_refresh(self):
66 class LogWidget(Widget):
69 line_width = self.size[1]
71 for line in self.content:
72 to_pad = line_width - (len(line) % line_width)
73 if to_pad == line_width:
75 to_join += [line + ' '*to_pad]
76 self.safe_write(''.join(to_join))
78 class MapWidget(Widget):
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))
89 def __init__(self, server_output):
90 self.server_output = server_output
91 curses.wrapper(self.loop)
93 def setup_screen(self, 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:')
99 def loop(self, stdscr):
100 self.setup_screen(stdscr)
101 curses.curs_set(False) # hide cursor
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]
116 key = self.stdscr.getkey()
118 if len(key) == 1 and key in ASCII_printable and \
119 len(to_send) < len(edit_line):
121 elif key == 'KEY_BACKSPACE':
122 to_send[:] = to_send[:-1]
124 plom_socket_io.send(s, ''.join(to_send))
126 elif key == 'KEY_RESIZE':
128 self.setup_screen(curses.initscr())
135 if len(self.server_output) > 0:
136 log[:0] = [self.server_output[0]]
137 self.server_output[:] = []
142 s = socket.create_connection(('127.0.0.1', 5000))
143 t = threading.Thread(target=recv_loop, args=(server_output,))