1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
from __future__ import annotations
import time
from dataclasses import dataclass
from enum import IntEnum
from typing import List, Set
import weechat
from slack.error import store_and_format_exception
from slack.shared import shared
from slack.util import get_callback_name
class LogLevel(IntEnum):
TRACE = 1
DEBUG = 2
INFO = 3
WARN = 4
ERROR = 5
FATAL = 6
class DebugMessageType(IntEnum):
WEBSOCKET_SEND = 1
WEBSOCKET_RECV = 2
HTTP_REQUEST = 3
LOG = 4
@dataclass
class DebugMessage:
time: float
level: LogLevel
message_type: DebugMessageType
message: str
debug_messages: List[DebugMessage] = []
printed_exceptions: Set[BaseException] = set()
# TODO: Figure out what to do with print_error vs log
def print_error(message: str):
weechat.prnt("", f"{weechat.prefix('error')}{shared.SCRIPT_NAME}: {message}")
def print_exception_once(e: BaseException):
if e not in printed_exceptions:
print_error(store_and_format_exception(e))
printed_exceptions.add(e)
def log(level: LogLevel, message_type: DebugMessageType, message: str):
if level >= LogLevel.INFO:
prefix = weechat.prefix("error") if level >= LogLevel.ERROR else "\t"
weechat.prnt("", f"{prefix}{shared.SCRIPT_NAME} {level.name}: {message}")
debug_message = DebugMessage(time.time(), level, message_type, message)
debug_messages.append(debug_message)
print_debug_buffer(debug_message)
def _close_debug_buffer_cb(data: str, buffer: str):
shared.debug_buffer_pointer = None
return weechat.WEECHAT_RC_OK
def open_debug_buffer():
if shared.debug_buffer_pointer:
weechat.buffer_set(shared.debug_buffer_pointer, "display", "1")
return
name = f"{shared.SCRIPT_NAME}.debug"
shared.debug_buffer_pointer = weechat.buffer_new_props(
name,
{"display": "1"},
"",
"",
get_callback_name(_close_debug_buffer_cb),
"",
)
for message in debug_messages:
print_debug_buffer(message)
def print_debug_buffer(debug_message: DebugMessage):
if shared.debug_buffer_pointer:
message = f"{debug_message.level.name} - {debug_message.message_type.name}\t{debug_message.message}"
weechat.prnt_date_tags(
shared.debug_buffer_pointer, int(debug_message.time), "", message
)
|
