169 lines
5.3 KiB
Python
169 lines
5.3 KiB
Python
"""NapCat 禁言状态存储。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import asdict, dataclass
|
|
from pathlib import Path
|
|
from typing import Any, Dict, List, Mapping, Optional
|
|
|
|
import asyncio
|
|
import json
|
|
|
|
|
|
_PROJECT_ROOT = Path(__file__).resolve().parents[2]
|
|
_DEFAULT_STORAGE_PATH = _PROJECT_ROOT / "data" / "napcat_adapter" / "ban_state.json"
|
|
|
|
|
|
@dataclass
|
|
class NapCatBanRecord:
|
|
"""NapCat 禁言记录。"""
|
|
|
|
group_id: str
|
|
user_id: str
|
|
lift_time: int
|
|
|
|
@property
|
|
def record_key(self) -> str:
|
|
"""返回当前记录的稳定键。
|
|
|
|
Returns:
|
|
str: 由群号和用户号拼接得到的稳定键。
|
|
"""
|
|
return f"{self.group_id}:{self.user_id}"
|
|
|
|
@classmethod
|
|
def from_mapping(cls, payload: Mapping[str, Any]) -> Optional["NapCatBanRecord"]:
|
|
"""从字典构造禁言记录。
|
|
|
|
Args:
|
|
payload: 原始记录字典。
|
|
|
|
Returns:
|
|
Optional[NapCatBanRecord]: 构造成功时返回记录对象，否则返回 ``None``。
|
|
"""
|
|
group_id = str(payload.get("group_id") or "").strip()
|
|
user_id = str(payload.get("user_id") or "").strip()
|
|
if not group_id or not user_id:
|
|
return None
|
|
|
|
try:
|
|
lift_time = int(payload.get("lift_time", -1))
|
|
except (TypeError, ValueError):
|
|
lift_time = -1
|
|
return cls(group_id=group_id, user_id=user_id, lift_time=lift_time)
|
|
|
|
def to_dict(self) -> Dict[str, Any]:
|
|
"""将记录转换为可序列化字典。
|
|
|
|
Returns:
|
|
Dict[str, Any]: 可直接写入 JSON 的记录字典。
|
|
"""
|
|
return asdict(self)
|
|
|
|
|
|
class NapCatBanStateStore:
|
|
"""NapCat 禁言状态持久化仓库。"""
|
|
|
|
def __init__(self, logger: Any, storage_path: Path = _DEFAULT_STORAGE_PATH) -> None:
|
|
"""初始化禁言状态仓库。
|
|
|
|
Args:
|
|
logger: 插件日志对象。
|
|
storage_path: 持久化文件路径。
|
|
"""
|
|
self._logger = logger
|
|
self._storage_path = storage_path
|
|
self._records: Dict[str, NapCatBanRecord] = {}
|
|
self._records_lock = asyncio.Lock()
|
|
|
|
async def load(self) -> None:
|
|
"""从本地文件加载禁言记录。"""
|
|
if not self._storage_path.exists():
|
|
return
|
|
|
|
try:
|
|
raw_payload = json.loads(self._storage_path.read_text(encoding="utf-8"))
|
|
except Exception as exc:
|
|
self._logger.warning(f"NapCat 禁言状态文件读取失败，将忽略旧记录: {exc}")
|
|
return
|
|
|
|
if not isinstance(raw_payload, list):
|
|
self._logger.warning("NapCat 禁言状态文件格式非法，将忽略旧记录")
|
|
return
|
|
|
|
loaded_records: Dict[str, NapCatBanRecord] = {}
|
|
for item in raw_payload:
|
|
if not isinstance(item, Mapping):
|
|
continue
|
|
record = NapCatBanRecord.from_mapping(item)
|
|
if record is not None:
|
|
loaded_records[record.record_key] = record
|
|
|
|
async with self._records_lock:
|
|
self._records = loaded_records
|
|
if loaded_records:
|
|
self._logger.info(f"NapCat 禁言状态已加载 {len(loaded_records)} 条记录")
|
|
|
|
async def snapshot(self) -> List[NapCatBanRecord]:
|
|
"""返回当前记录快照。
|
|
|
|
Returns:
|
|
List[NapCatBanRecord]: 当前内存中的记录列表副本。
|
|
"""
|
|
async with self._records_lock:
|
|
return list(self._records.values())
|
|
|
|
async def upsert(self, record: NapCatBanRecord) -> None:
|
|
"""新增或更新一条禁言记录。
|
|
|
|
Args:
|
|
record: 待写入的禁言记录。
|
|
"""
|
|
async with self._records_lock:
|
|
self._records[record.record_key] = record
|
|
await self.persist()
|
|
|
|
async def remove(self, group_id: str, user_id: str) -> Optional[NapCatBanRecord]:
|
|
"""删除指定禁言记录。
|
|
|
|
Args:
|
|
group_id: 群号。
|
|
user_id: 用户号。
|
|
|
|
Returns:
|
|
Optional[NapCatBanRecord]: 被移除的记录；不存在时返回 ``None``。
|
|
"""
|
|
record_key = f"{group_id}:{user_id}"
|
|
return await self.pop(record_key)
|
|
|
|
async def pop(self, record_key: str) -> Optional[NapCatBanRecord]:
|
|
"""按稳定键移除一条记录。
|
|
|
|
Args:
|
|
record_key: 记录稳定键。
|
|
|
|
Returns:
|
|
Optional[NapCatBanRecord]: 被移除的记录；不存在时返回 ``None``。
|
|
"""
|
|
async with self._records_lock:
|
|
removed_record = self._records.pop(record_key, None)
|
|
if removed_record is not None:
|
|
await self.persist()
|
|
return removed_record
|
|
|
|
async def persist(self) -> None:
|
|
"""将当前禁言记录持久化到本地文件。"""
|
|
async with self._records_lock:
|
|
serialized_records = [
|
|
record.to_dict() for record in sorted(self._records.values(), key=lambda item: item.record_key)
|
|
]
|
|
|
|
try:
|
|
self._storage_path.parent.mkdir(parents=True, exist_ok=True)
|
|
self._storage_path.write_text(
|
|
json.dumps(serialized_records, ensure_ascii=False, indent=2),
|
|
encoding="utf-8",
|
|
)
|
|
except Exception as exc:
|
|
self._logger.warning(f"NapCat 禁言状态持久化失败: {exc}")
|