From 48abe2ad7ebafae60558895e737b2295decdfcb2 Mon Sep 17 00:00:00 2001 From: mathieui Date: Sat, 10 Apr 2021 13:13:12 +0200 Subject: feature: add a log loader class --- poezio/log_loader.py | 194 ++++++++++++++++++++++++++++++++++++++++++++++++ poezio/tabs/basetabs.py | 6 +- poezio/tabs/muctab.py | 6 +- 3 files changed, 202 insertions(+), 4 deletions(-) create mode 100644 poezio/log_loader.py (limited to 'poezio') diff --git a/poezio/log_loader.py b/poezio/log_loader.py new file mode 100644 index 00000000..92a0306c --- /dev/null +++ b/poezio/log_loader.py @@ -0,0 +1,194 @@ +import logging +from datetime import datetime, timedelta +from typing import List, Dict, Any +from poezio import tabs +from poezio.logger import iterate_messages_reverse, Logger +from poezio.mam import ( + fetch_history, + NoMAMSupportException, + MAMQueryException, + DiscoInfoException, + make_line, +) +from poezio.common import to_utc +from poezio.ui.types import EndOfArchive, Message, BaseMessage +from poezio.text_buffer import HistoryGap +from slixmpp import JID + + +log = logging.getLogger(__name__) + + +def make_line_local(tab: tabs.ChatTab, msg: Dict[str, Any]) -> Message: + if isinstance(tab, tabs.MucTab): + jid = JID(tab.jid) + jid.resource = msg['nickname'] + else: + jid = JID(tab.jid) + return make_line(tab, msg['txt'], msg['time'], jid, '') + + +STATUS = {'mam_only', 'local_only', 'local_mam_completed'} + + +class LogLoader: + """ + An ephemeral class that loads history in a tab + """ + load_status: str = 'mam_only' + logger: Logger + tab: tabs.ChatTab + + def __init__(self, logger: Logger, tab: tabs.ChatTab, + load_status: str = 'local_only'): + if load_status not in STATUS: + self.load_status = 'mam_only' + else: + self.load_status = load_status + self.logger = logger + self.tab = tab + + async def tab_open(self): + """Called on a tab opening or a MUC join""" + amount = 2 * self.tab.text_win.height + gap = self.tab._text_buffer.find_last_gap_muc() + if gap is not None: + if self.load_status == 'local_only': + messages = await self.local_fill_gap(gap) + else: + messages = await self.mam_fill_gap(gap) + else: + if self.load_status == 'mam_only': + messages = await self.mam_tab_open(amount) + else: + messages = await self.local_tab_open(amount) + + if messages: + self.tab._text_buffer.add_history_messages(messages) + self.tab.core.refresh_window() + + async def mam_tab_open(self, nb: int) -> List[BaseMessage]: + tab = self.tab + end = datetime.now() + for message in tab._text_buffer.messages: + time_ok = to_utc(message.time) < to_utc(end) + if isinstance(message, Message) and time_ok: + end = message.time + break + end = end - timedelta(microseconds=1) + try: + return await fetch_history(tab, end=end, amount=nb) + except (NoMAMSupportException, MAMQueryException, DiscoInfoException): + return [] + finally: + tab.query_status = False + + async def local_tab_open(self, nb: int) -> List[BaseMessage]: + results: List[BaseMessage] = [] + filepath = self.logger.get_file_path(self.tab.jid) + for msg in iterate_messages_reverse(filepath): + typ_ = msg.pop('type') + if typ_ == 'message': + results.append(make_line_local(self.tab, msg)) + if len(results) >= nb: + break + return results[::-1] + + async def mam_fill_gap(self, gap: HistoryGap) -> List[BaseMessage]: + tab = self.tab + + start = gap.last_timestamp_before_leave + end = gap.first_timestamp_after_join + if start: + start = start + timedelta(seconds=1) + if end: + end = end - timedelta(seconds=1) + try: + return await fetch_history(tab, start=start, end=end, amount=999) + except (NoMAMSupportException, MAMQueryException, DiscoInfoException): + return [] + finally: + tab.query_status = False + + async def local_fill_gap(self, gap: HistoryGap) -> List[BaseMessage]: + start = gap.last_timestamp_before_leave + end = gap.first_timestamp_after_join + + results: List[BaseMessage] = [] + filepath = self.logger.get_file_path(self.tab.jid) + for msg in iterate_messages_reverse(filepath): + typ_ = msg.pop('type') + if start and msg['time'] < start: + break + if typ_ == 'message' and (not end or msg['time'] < end): + results.append(make_line_local(self.tab, msg)) + return results[::-1] + + async def scroll_requested(self): + """When a scroll up is requested in a chat tab. + + Try to load more history if there are no more messages in the buffer. + """ + tab = self.tab + tw = tab.text_win + + # If position in the tab is < two screen pages, then fetch MAM, so that + # wa keep some prefetched margin. A first page should also be + # prefetched on join if not already available. + total, pos, height = len(tw.built_lines), tw.pos, tw.height + rest = (total - pos) // height + + if rest > 1: + return None + + if self.load_status == 'mam_only': + messages = await self.mam_scroll_requested(height) + else: + messages = await self.local_scroll_requested(height) + log.debug('%s %s', messages[0].txt, messages[0].time) + tab._text_buffer.add_history_messages(messages) + if messages: + tab.core.refresh_window() + + async def local_scroll_requested(self, nb: int) -> List[BaseMessage]: + tab = self.tab + last_message_time = None + if tab._text_buffer.messages: + last_message_time = to_utc(tab._text_buffer.messages[0].time) + last_message_time -= timedelta(microseconds=1) + + results: List[BaseMessage] = [] + filepath = self.logger.get_file_path(self.tab.jid) + for msg in iterate_messages_reverse(filepath): + typ_ = msg.pop('type') + if last_message_time is None or msg['time'] < last_message_time: + if typ_ == 'message': + results.append(make_line_local(self.tab, msg)) + if len(results) >= nb: + break + return results[::-1] + + async def mam_scroll_requested(self, nb: int) -> List[BaseMessage]: + tab = self.tab + try: + messages = await fetch_history(tab, amount=nb) + last_message_exists = False + if tab._text_buffer.messages: + last_message = tab._text_buffer.messages[0] + last_message_exists = True + if (not messages and + last_message_exists + and not isinstance(last_message, EndOfArchive)): + time = tab._text_buffer.messages[0].time + messages = [EndOfArchive('End of archive reached', time=time)] + return messages + except NoMAMSupportException: + return await self.local_scroll_requested(nb) + except (MAMQueryException, DiscoInfoException): + tab.core.information( + f'An error occured when fetching MAM for {tab.jid}', + 'Error' + ) + return await self.local_scroll_requested(nb) + finally: + tab.query_status = False diff --git a/poezio/tabs/basetabs.py b/poezio/tabs/basetabs.py index 2f221afe..306c79f9 100644 --- a/poezio/tabs/basetabs.py +++ b/poezio/tabs/basetabs.py @@ -965,8 +965,10 @@ class ChatTab(Tab): def on_scroll_up(self): if not self.query_status: - from poezio import mam - mam.schedule_scroll_up(tab=self) + from poezio.log_loader import LogLoader + asyncio.ensure_future( + LogLoader(logger, self).scroll_requested() + ) return self.text_win.scroll_up(self.text_win.height - 1) def on_scroll_down(self): diff --git a/poezio/tabs/muctab.py b/poezio/tabs/muctab.py index a39a0234..54e78c72 100644 --- a/poezio/tabs/muctab.py +++ b/poezio/tabs/muctab.py @@ -39,7 +39,6 @@ from slixmpp.exceptions import IqError, IqTimeout from poezio.tabs import ChatTab, Tab, SHOW_NAME from poezio import common -from poezio import mam from poezio import multiuserchat as muc from poezio import timed_events from poezio import windows @@ -49,6 +48,7 @@ from poezio.config import config from poezio.core.structs import Command from poezio.decorators import refresh_wrapper, command_args_parser from poezio.logger import logger +from poezio.log_loader import LogLoader from poezio.roster import roster from poezio.theming import get_theme, dump_tuple from poezio.user import User @@ -604,7 +604,9 @@ class MucTab(ChatTab): }, ), ) - mam.schedule_tab_open(self) + asyncio.ensure_future( + LogLoader(logger, self).tab_open(), + ) def handle_presence_joined(self, presence: Presence, status_codes: Set[int]) -> None: """ -- cgit v1.2.3