GitHubMirror/sshyp
1
0
Fork 0
mirror of https://github.com/rwinkhart/sshyp.git synced 2026-09-05 16:47:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix crash when terminal is not tall enough to display curses radio pretext

Browse Source
This commit is contained in:
RandyTheSilly
2023-12-10 21:49:40 -05:00
parent 2edb1b9b53
commit 27cf610a1f
1 changed files with 34 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/stweak.py
+34 -21
View File
@@ -33,38 +33,51 @@ def curses_radio(_options, _pretext):
# clear curses window # clear curses window
stdscr.clear() stdscr.clear()
# get terminal size # get terminal size
_width = stdscr.getmaxyx()[1] _height, _width = stdscr.getmaxyx()
# track whether to generate interactive buttons (depending on terminal size)
_button_gen = False
# split text based on new lines # split text based on new lines
_pretext_lines = _pretext.split('\n') _pretext_lines = _pretext.split('\n')
# iterate through lines, splitting further (wrapping) as needed, to add to curses window # iterate through lines, splitting further (wrapping) as needed, to add to curses window
_current_line = 0 _current_line = 0
for _line in _pretext_lines: for _line in _pretext_lines:
_remaining = _line if _current_line <= _height-4:
while len(_remaining) > _width: _remaining, _button_gen = _line, True
stdscr.addstr(_current_line, 0, _remaining[:_width]) while len(_remaining) > _width:
_remaining = _remaining[_width:] stdscr.addstr(_current_line, 0, _remaining[:_width])
_remaining = _remaining[_width:]
_current_line += 1
stdscr.addstr(_current_line, 0, _remaining)
_current_line += 1 _current_line += 1
stdscr.addstr(_current_line, 0, _remaining)
_current_line += 1
# create user-interactive options
for _i, _option in enumerate(_options):
_y = _i + _current_line + 1
if _i == _selected:
stdscr.addstr(_y, 0, "[*] " + _option, A_REVERSE)
else: else:
stdscr.addstr(_y, 0, "[ ] " + _option) _button_gen = False
stdscr.refresh() stdscr.addstr(_current_line, 0, '# warning: re-size terminal to see more information'[:_width-1])
_current_line += 1
break
# create user-interactive options if all information has been displayed to the user
if _button_gen:
for _i, _option in enumerate(_options):
_y = _i + _current_line + 1
if _i == _selected:
stdscr.addstr(_y, 0, "[*] " + _option, A_REVERSE)
else:
stdscr.addstr(_y, 0, "[ ] " + _option)
# update _selected based on user input # update _selected based on user input
_key = stdscr.getch() _key = stdscr.getch()
if _key == KEY_UP: if _button_gen:
_selected = (_selected-1) % len(_options) if _key == KEY_UP:
elif _key == KEY_DOWN: _selected = (_selected - 1) % len(_options)
_selected = (_selected+1) % len(_options) elif _key == KEY_DOWN:
elif _key == ord('\n'): _selected = (_selected + 1) % len(_options)
break # ord('\n') == 10
elif _key == 10:
break
stdscr.refresh() stdscr.refresh()
curs_set(1) curs_set(1)