Mastodon
/
Toot-Mastodon-CLI-TUI-client-Linux
mirror of https://github.com/ihabunek/toot
1
0
Fork 0
Code Issues Releases Wiki Activity

Extract status list window

This commit is contained in:
Ivan Habunek 2018-01-08 15:19:02 +01:00
parent 554618db4b
commit e204722425
No known key found for this signature in database
GPG Key ID: CDBD63C43A30BB95
1 changed files with 117 additions and 98 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

215
toot/ui/app.py
View File

@ -64,10 +64,98 @@ class FooterWindow:
self.window.refresh() self.window.refresh()
def clear_message(self): def clear_message(self):
self.window.addstr(0, 0, " " * self.width) self.window.addstr(1, 0, "".ljust(self.width - 1))
self.window.refresh() self.window.refresh()
class StatusListWindow:
"""Window which shows the scrollable list of statuses (left side)."""
def __init__(self, height, width, top, left):
# Dimensions and position of region in stdscr which will contain the pad
self.region_height = height
self.region_width = width
self.region_top = top
self.region_left = left
# How many statuses fit on one page (excluding border, at 3 lines per status)
self.page_size = (height - 2) // 3
# Initially, size the pad to the dimensions of the region, will be
# increased later to accomodate statuses
self.pad = curses.newpad(10, width)
self.pad.box()
self.scroll_pos = 0
def draw_statuses(self, statuses, selected, starting=0):
# Resize window to accomodate statuses if required
height, width = self.pad.getmaxyx()
new_height = len(statuses) * 3 + 1
if new_height > height:
self.pad.resize(new_height, width)
self.pad.box()
last_idx = len(statuses) - 1
for index, status in enumerate(statuses):
if index >= starting:
highlight = selected == index
draw_divider = index < last_idx
self.draw_status_row(status, index, highlight, draw_divider)
def draw_status_row(self, status, index, highlight=False, draw_divider=True):
offset = 3 * index
height, width = self.pad.getmaxyx()
color = Color.GREEN if highlight else Color.WHITE
date, time = status['created_at']
self.pad.addstr(offset + 1, 1, " " + date.ljust(14), color)
self.pad.addstr(offset + 2, 1, " " + time.ljust(14), color)
trunc_width = width - 15
acct = trunc(status['account']['acct'], trunc_width).ljust(trunc_width)
display_name = trunc(status['account']['display_name'], trunc_width).ljust(trunc_width)
self.pad.addstr(offset + 1, 14, display_name, color)
self.pad.addstr(offset + 2, 14, acct, color)
if draw_divider:
draw_horizontal_divider(self.pad, offset + 3)
self.refresh()
def refresh(self):
self.pad.refresh(
self.scroll_pos * 3, # top
0, # left
self.region_top,
self.region_left,
self.region_height,
self.region_width,
)
def scroll_to(self, index):
self.scroll_pos = index
self.refresh()
def scroll_up(self):
if self.scroll_pos > 0:
self.scroll_to(self.scroll_pos - 1)
def scroll_down(self):
self.scroll_to(self.scroll_pos + 1)
def scroll_if_required(self, new_index):
if new_index < self.scroll_pos:
self.scroll_up()
elif new_index >= self.scroll_pos + self.page_size:
self.scroll_down()
else:
self.refresh()
class StatusDetailWindow: class StatusDetailWindow:
"""Window which shows details of a status (right side)""" """Window which shows details of a status (right side)"""
def __init__(self, height, width, y, x): def __init__(self, height, width, y, x):
@ -110,6 +198,9 @@ class StatusDetailWindow:
self.window.erase() self.window.erase()
self.window.box() self.window.box()
if not status:
return
content = self.content_lines(status) content = self.content_lines(status)
footer = self.footer_lines(status) footer = self.footer_lines(status)
@ -120,14 +211,11 @@ class StatusDetailWindow:
self.window.refresh() self.window.refresh()
class TimelineApp: class TimelineApp:
def __init__(self, status_generator): def __init__(self, status_generator):
self.status_generator = status_generator self.status_generator = status_generator
self.statuses = [] self.statuses = []
self.selected = None
self.stdscr = None self.stdscr = None
self.scroll_pos = 0
def run(self): def run(self):
curses.wrapper(self._wrapped_run) curses.wrapper(self._wrapped_run)
@ -135,8 +223,8 @@ class TimelineApp:
def _wrapped_run(self, stdscr): def _wrapped_run(self, stdscr):
self.stdscr = stdscr self.stdscr = stdscr
self.setup_windows()
Color.setup_palette() Color.setup_palette()
self.setup_windows()
# Load some data and redraw # Load some data and redraw
self.fetch_next() self.fetch_next()
@ -151,17 +239,17 @@ class TimelineApp:
if screen_width < 60: if screen_width < 60:
raise ConsoleError("Terminal screen is too narrow, toot curses requires at least 60 columns to display properly.") raise ConsoleError("Terminal screen is too narrow, toot curses requires at least 60 columns to display properly.")
self.left_width = max(min(screen_width // 3, 60), 30) left_width = max(min(screen_width // 3, 60), 30)
self.right_width = screen_width - self.left_width right_width = screen_width - left_width
self.header = HeaderWindow(2, screen_width, 0, 0) self.header = HeaderWindow(2, screen_width, 0, 0)
self.footer = FooterWindow(2, screen_width, screen_height - 2, 0) self.footer = FooterWindow(2, screen_width, screen_height - 2, 0)
self.left = curses.newpad(screen_height - 4, self.left_width) self.left = StatusListWindow(screen_height - 4, left_width, 2, 0)
self.right = StatusDetailWindow(screen_height - 4, self.right_width, 2, self.left_width) self.right = StatusDetailWindow(screen_height - 4, right_width, 2, left_width)
def loop(self): def loop(self):
while True: while True:
key = self.left.getkey() key = self.left.pad.getkey()
if key.lower() == 'q': if key.lower() == 'q':
return return
@ -181,22 +269,6 @@ class TimelineApp:
self.setup_windows() self.setup_windows()
self.full_redraw() self.full_redraw()
def scroll_to(self, index):
self.scroll_pos = index
height, width = self.stdscr.getmaxyx()
self.left.refresh(3 * index, 0, 2, 0, height - 4, self.left_width)
def scroll_up(self):
if self.scroll_pos > 0:
self.scroll_to(self.scroll_pos - 1)
def scroll_down(self):
self.scroll_to(self.scroll_pos + 1)
def scroll_refresh(self):
self.scroll_to(self.scroll_pos)
def select_previous(self): def select_previous(self):
"""Move to the previous status in the timeline.""" """Move to the previous status in the timeline."""
self.footer.clear_message() self.footer.clear_message()
@ -211,49 +283,22 @@ class TimelineApp:
self.selected = new_index self.selected = new_index
self.redraw_after_selection_change(old_index, new_index) self.redraw_after_selection_change(old_index, new_index)
# Scroll if required
if new_index < self.scroll_pos:
self.scroll_up()
else:
self.scroll_refresh()
def select_next(self): def select_next(self):
"""Move to the next status in the timeline.""" """Move to the next status in the timeline."""
self.footer.clear_message() self.footer.clear_message()
# Load more statuses if no more are available
if self.selected + 1 >= len(self.statuses):
self.fetch_next()
self.draw_statuses(self.left, self.selected + 1)
self.draw_footer_status()
old_index = self.selected old_index = self.selected
new_index = self.selected + 1 new_index = self.selected + 1
# Load more statuses if no more are available
if self.selected + 1 >= len(self.statuses):
self.fetch_next()
self.left.draw_statuses(self.statuses, self.selected, new_index - 1)
self.draw_footer_status()
self.selected = new_index self.selected = new_index
self.redraw_after_selection_change(old_index, new_index) self.redraw_after_selection_change(old_index, new_index)
# Scroll if required
if new_index >= self.scroll_pos + self.get_page_size():
self.scroll_down()
else:
self.scroll_refresh()
def get_page_size(self):
"""Calculate how many statuses fit on one page (3 lines per status)"""
height = self.right.window.getmaxyx()[0] - 2 # window height - borders
return height // 3
def redraw_after_selection_change(self, old_index, new_index):
old_status = self.statuses[old_index]
new_status = self.statuses[new_index]
# Perform a partial redraw
self.draw_status_row(self.left, old_status, old_index, False)
self.draw_status_row(self.left, new_status, new_index, True)
self.right.draw(new_status)
self.draw_footer_status()
def fetch_next(self): def fetch_next(self):
try: try:
self.footer.draw_message("Loading toots...", Color.BLUE) self.footer.draw_message("Loading toots...", Color.BLUE)
@ -273,52 +318,26 @@ class TimelineApp:
self.header.draw() self.header.draw()
self.draw_footer_status() self.draw_footer_status()
self.left.clear() self.left.draw_statuses(self.statuses, self.selected)
self.left.box()
self.draw_statuses(self.left)
self.right.draw(self.get_selected_status()) self.right.draw(self.get_selected_status())
self.header.draw() self.header.draw()
self.scroll_refresh()
def redraw_after_selection_change(self, old_index, new_index):
old_status = self.statuses[old_index]
new_status = self.statuses[new_index]
# Perform a partial redraw
self.left.draw_status_row(old_status, old_index, highlight=False, draw_divider=False)
self.left.draw_status_row(new_status, new_index, highlight=True, draw_divider=False)
self.left.scroll_if_required(new_index)
self.right.draw(new_status)
self.draw_footer_status()
def get_selected_status(self): def get_selected_status(self):
if len(self.statuses) > self.selected: if len(self.statuses) > self.selected:
return self.statuses[self.selected] return self.statuses[self.selected]
def draw_status_row(self, window, status, index, highlight=False):
offset = 3 * index
height, width = window.getmaxyx()
color = Color.BLUE if highlight else 0
date, time = status['created_at']
window.addstr(offset + 1, 2, date, color)
window.addstr(offset + 2, 2, time, color)
trunc_width = width - 16
acct = trunc(status['account']['acct'], trunc_width).ljust(trunc_width)
display_name = trunc(status['account']['display_name'], trunc_width).ljust(trunc_width)
window.addstr(offset + 1, 14, acct, color)
window.addstr(offset + 2, 14, display_name, color)
window.addstr(offset + 3, 1, '─' * (width - 2))
screen_height, screen_width = self.stdscr.getmaxyx()
window.refresh(0, 0, 2, 0, screen_height - 4, self.left_width)
def draw_statuses(self, window, starting=0):
# Resize window to accomodate statuses if required
height, width = window.getmaxyx()
new_height = len(self.statuses) * 3 + 1
if new_height > height:
window.resize(new_height, width)
window.box()
for index, status in enumerate(self.statuses):
if index >= starting:
highlight = self.selected == index
self.draw_status_row(window, status, index, highlight)
def draw_footer_status(self): def draw_footer_status(self):
self.footer.draw_status(self.selected, len(self.statuses)) self.footer.draw_status(self.selected, len(self.statuses))