diff options
Diffstat (limited to 'slack/slack_buffer.py')
| -rw-r--r-- | slack/slack_buffer.py | 20 |
1 files changed, 15 insertions, 5 deletions
diff --git a/slack/slack_buffer.py b/slack/slack_buffer.py index 881e56d..9554db1 100644 --- a/slack/slack_buffer.py +++ b/slack/slack_buffer.py @@ -3,7 +3,7 @@ from __future__ import annotations import time from abc import ABC, abstractmethod from contextlib import contextmanager -from typing import TYPE_CHECKING, Dict, List, Mapping, Optional, Tuple +from typing import TYPE_CHECKING, Dict, List, Mapping, Optional, Set, Tuple import weechat @@ -127,8 +127,11 @@ class SlackBuffer(ABC): # TODO: buffer_pointer may be accessed by buffer_switch before it's initialized self.buffer_pointer: str = "" self.is_loading = False - self.history_filled = False self.history_pending = False + self.history_pending_messages: List[SlackMessage] = [] + self.history_needs_refresh = False + self.last_printed_ts: Optional[SlackTs] = None + self.hotlist_tss: Set[SlackTs] = set() self.completion_context: Literal[ "NO_COMPLETION", @@ -185,9 +188,8 @@ class SlackBuffer(ABC): async def get_name_and_buffer_props(self) -> Tuple[str, Dict[str, str]]: raise NotImplementedError() - @abstractmethod async def buffer_switched_to(self) -> None: - raise NotImplementedError() + self.hotlist_tss.clear() def get_full_name(self, name: str) -> str: return f"{shared.SCRIPT_NAME}.{self.workspace.name}.{name}" @@ -234,6 +236,10 @@ class SlackBuffer(ABC): for key, value in buffer_props.items(): weechat.buffer_set(self.buffer_pointer, key, value) + @abstractmethod + async def set_hotlist(self) -> None: + raise NotImplementedError() + async def rerender_message(self, message: SlackMessage): modify_buffer_line( self.buffer_pointer, @@ -258,6 +264,9 @@ class SlackBuffer(ABC): weechat.prnt_date_tags(self.buffer_pointer, message.ts.major, tags, rendered) if backlog: weechat.buffer_set(self.buffer_pointer, "unread", "") + else: + self.hotlist_tss.add(message.ts) + self.last_printed_ts = message.ts def last_read_line_ts(self) -> Optional[SlackTs]: if self.buffer_pointer: @@ -287,6 +296,7 @@ class SlackBuffer(ABC): # TODO: Move unread marker to correct position according to last_read for WeeChat >= 4.0.0 weechat.buffer_set(self.buffer_pointer, "unread", "") weechat.buffer_set(self.buffer_pointer, "hotlist", "-1") + self.hotlist_tss.clear() def _buffer_input_cb(self, data: str, buffer: str, input_data: str) -> int: weechat.prnt(buffer, "Text: %s" % input_data) @@ -296,5 +306,5 @@ class SlackBuffer(ABC): if self.buffer_pointer in shared.buffers: del shared.buffers[self.buffer_pointer] self.buffer_pointer = "" - self.history_filled = False + self.last_printed_ts = None return weechat.WEECHAT_RC_OK |