Kris
4e07292a81
1. 调整导入顺序和移除多余空行 2. 重构IRC NAMES命令的成员解析逻辑，正确剥离所有前缀 3. 更新配置schema：修改默认消息块大小为350字节，新增disableBlockStreaming配置项 4. 完善CTCP处理：新增ACTION支持，提取CTCP动作文本 5. 新增IRC线程上下文模拟器类 6. 新增SRV记录解析支持，自动解析IRC服务器域名 7. 新增多种IRC通知处理：ACCOUNT、AWAY、INVITE、CAP 8. 重构消息发送逻辑，添加熔断器和重试机制 9. 重写流式消息合并逻辑，新增智能合并缓冲 10. 扩展IRC CAP支持，新增多个常用扩展能力 11. 修复配置读取逻辑，适配新的配置结构
115 lines
4.0 KiB
Python
115 lines
4.0 KiB
Python
from __future__ import annotations
|
|
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
|
|
class ChannelRuntime:
|
|
def __init__(self):
|
|
self._joined_channels: set[str] = set()
|
|
self._channel_members: dict[str, set[str]] = {}
|
|
self._channel_modes: dict[str, str] = {}
|
|
self._channel_topics: dict[str, str] = {}
|
|
|
|
@property
|
|
def joined_channels(self) -> set[str]:
|
|
return self._joined_channels
|
|
|
|
def add_channel(self, channel: str) -> None:
|
|
channel_lower = channel.lower()
|
|
self._joined_channels.add(channel_lower)
|
|
if channel_lower not in self._channel_members:
|
|
self._channel_members[channel_lower] = set()
|
|
|
|
def remove_channel(self, channel: str) -> None:
|
|
self._joined_channels.discard(channel.lower())
|
|
|
|
def handle_names(self, channel: str, names_str: str) -> None:
|
|
channel_lower = channel.lower()
|
|
members = set()
|
|
for raw_name in names_str.split():
|
|
if not raw_name:
|
|
continue
|
|
name = raw_name
|
|
while name and name[0] in "@+%~&!":
|
|
name = name[1:]
|
|
if name:
|
|
members.add(name)
|
|
self._channel_members[channel_lower] = members
|
|
logger.debug(f"IRC NAMES {channel_lower}: {len(members)} members")
|
|
|
|
def handle_join(self, channel: str, nick: str) -> None:
|
|
channel_lower = channel.lower()
|
|
if channel_lower in self._channel_members:
|
|
self._channel_members[channel_lower].add(nick)
|
|
logger.debug(f"IRC JOIN: {nick} -> {channel_lower}")
|
|
|
|
def handle_part(self, channel: str, nick: str) -> None:
|
|
channel_lower = channel.lower()
|
|
if channel_lower in self._channel_members:
|
|
self._channel_members[channel_lower].discard(nick)
|
|
logger.debug(f"IRC PART: {nick} <- {channel_lower}")
|
|
|
|
def handle_kick(self, channel: str, nick: str) -> None:
|
|
channel_lower = channel.lower()
|
|
if channel_lower in self._channel_members:
|
|
self._channel_members[channel_lower].discard(nick)
|
|
logger.debug(f"IRC KICK: {nick} from {channel_lower}")
|
|
|
|
def handle_quit(self, nick: str) -> None:
|
|
for channel in self._channel_members:
|
|
self._channel_members[channel].discard(nick)
|
|
logger.debug(f"IRC QUIT: {nick}")
|
|
|
|
def handle_nick_change(self, old_nick: str, new_nick: str) -> None:
|
|
for channel in self._channel_members:
|
|
members = self._channel_members[channel]
|
|
if old_nick in members:
|
|
members.discard(old_nick)
|
|
members.add(new_nick)
|
|
logger.debug(f"IRC NICK: {old_nick} -> {new_nick}")
|
|
|
|
def get_members(self, channel: str) -> set[str]:
|
|
return self._channel_members.get(channel.lower(), set())
|
|
|
|
def handle_mode(self, channel: str, modes: str) -> None:
|
|
channel_lower = channel.lower()
|
|
current = self._channel_modes.get(channel_lower, "")
|
|
self._channel_modes[channel_lower] = _merge_modes(current, modes)
|
|
logger.debug(f"IRC MODE {channel}: {modes} -> {self._channel_modes[channel_lower]}")
|
|
|
|
def set_topic(self, channel: str, topic: str) -> None:
|
|
self._channel_topics[channel.lower()] = topic
|
|
|
|
def get_topic(self, channel: str) -> str | None:
|
|
return self._channel_topics.get(channel.lower())
|
|
|
|
def get_modes(self, channel: str) -> str:
|
|
return self._channel_modes.get(channel.lower(), "")
|
|
|
|
|
|
def _merge_modes(current: str, incoming: str) -> str:
|
|
if not current:
|
|
return incoming
|
|
if not incoming:
|
|
return current
|
|
adding = True
|
|
mode_set: set[str] = set()
|
|
|
|
for char in current:
|
|
if char in "+-":
|
|
continue
|
|
mode_set.add(char)
|
|
|
|
for char in incoming:
|
|
if char == "+":
|
|
adding = True
|
|
elif char == "-":
|
|
adding = False
|
|
elif adding:
|
|
mode_set.add(char)
|
|
else:
|
|
mode_set.discard(char)
|
|
|
|
sorted_modes = sorted(mode_set)
|
|
return "+" + "".join(sorted_modes)
|