feat : 接入了noname的插件到bot(多线程运行)
This commit is contained in:
Wbucket
@@ -0,0 +1,359 @@
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
from types import (FunctionType,MethodType)
|
||||
import re
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
from typing import Dict, Any, Optional, Set
|
||||
import time
|
||||
import httpx
|
||||
from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
|
||||
|
||||
import logging
|
||||
|
||||
# from astrbot.api.all import (
|
||||
# register, command, Context, Star, AstrMessageEvent,
|
||||
# MessageChain, Plain, logger
|
||||
# )
|
||||
|
||||
# from astrbot.api.star import StarTools
|
||||
|
||||
|
||||
MANIFEST_URL = "https://piston-meta.mojang.com/mc/game/version_manifest_v2.json"
|
||||
|
||||
|
||||
def _ts():
|
||||
return datetime.now().strftime("[%Y-%m-%d %H:%M:%S]")
|
||||
|
||||
|
||||
# ========== 已改进版本的文章链接生成函数 ==========
|
||||
def format_article_url(lang: str, version_id: str) -> str:
|
||||
vid = (version_id or "").lower()
|
||||
|
||||
def _join_parts(match: re.Match) -> str:
|
||||
parts = [match.group("major"), match.group("minor")]
|
||||
patch = match.group("patch")
|
||||
if patch is not None:
|
||||
parts.append(patch)
|
||||
return "-".join(parts)
|
||||
|
||||
# pre-release: minecraft-1-a-b-pre-release-c
|
||||
pre_match = re.match(
|
||||
r"^(?P<major>\d+)\.(?P<minor>\d+)(?:\.(?P<patch>\d+))?-pre(?P<build>\d+)$",
|
||||
vid,
|
||||
)
|
||||
if pre_match:
|
||||
slug = _join_parts(pre_match)
|
||||
build = pre_match.group("build")
|
||||
return f"https://www.minecraft.net/{lang}/article/minecraft-{slug}-pre-release-{build}"
|
||||
|
||||
# RC: minecraft-1-a-b-release-candidate-c
|
||||
rc_match = re.match(
|
||||
r"^(?P<major>\d+)\.(?P<minor>\d+)(?:\.(?P<patch>\d+))?-rc(?P<build>\d+)$",
|
||||
vid,
|
||||
)
|
||||
if rc_match:
|
||||
slug = _join_parts(rc_match)
|
||||
build = rc_match.group("build")
|
||||
return f"https://www.minecraft.net/{lang}/article/minecraft-{slug}-release-candidate-{build}"
|
||||
|
||||
# Snapshot，例如 25w46a
|
||||
if "w" in vid:
|
||||
slug = vid.replace(" ", "").replace("_", "-")
|
||||
return f"https://www.minecraft.net/{lang}/article/minecraft-snapshot-{slug}"
|
||||
|
||||
return ""
|
||||
|
||||
|
||||
# ===================================================
|
||||
|
||||
|
||||
class State:
|
||||
def __init__(self, path: Path):
|
||||
self.path = path
|
||||
self.last_snapshot_id: Optional[str] = None
|
||||
self.last_release_id: Optional[str] = None
|
||||
self.targets: Set[str] = set()
|
||||
self.etag: Optional[str] = None
|
||||
|
||||
def load(self):
|
||||
if not self.path.exists():
|
||||
return
|
||||
try:
|
||||
data = json.loads(self.path.read_text("utf-8"))
|
||||
self.last_snapshot_id = data.get("last_snapshot_id")
|
||||
self.last_release_id = data.get("last_release_id")
|
||||
self.targets = set(data.get("targets", []))
|
||||
self.etag = data.get("etag")
|
||||
except (json.JSONDecodeError, KeyError, OSError) as e:
|
||||
logger.warning(f"{_ts()} state 文件异常：{e}")
|
||||
|
||||
def save(self):
|
||||
data = {
|
||||
"last_snapshot_id": self.last_snapshot_id,
|
||||
"last_release_id": self.last_release_id,
|
||||
"targets": list(self.targets),
|
||||
"etag": self.etag,
|
||||
}
|
||||
try:
|
||||
self.path.write_text(json.dumps(data, ensure_ascii=False, indent=2), "utf-8")
|
||||
except OSError as e:
|
||||
logger.warning(f"{_ts()} state 写入失败：{e}")
|
||||
|
||||
|
||||
# @register(
|
||||
# "astrbot_plugin_mcwatcher",
|
||||
# "noname2309-bot",
|
||||
# "MCWatcher 是一个基于 AstrBot 的插件，用于 自动监测 Mojang 的最新 Minecraft 版本",
|
||||
# "0.2.4",
|
||||
# "https://github.com/wadadawsd/astrbot_plugin_mcwatcher"
|
||||
# )
|
||||
class MCWatcher:
|
||||
|
||||
# def __init__(self, context: Context, config: Optional[Dict[str, Any]] = None, **kwargs):
|
||||
def __init__( self,
|
||||
config: Optional[Dict[str, Any]] = None,
|
||||
hookfunction:FunctionType | MethodType = print,
|
||||
logging =logging,
|
||||
**kwargs):
|
||||
# super().__init__(context)
|
||||
|
||||
self.config: Dict[str, Any] = config or {}
|
||||
# self.ctx = context
|
||||
|
||||
def cfg(key: str, default):
|
||||
return self.config.get(key, default)
|
||||
global logger
|
||||
logger = logging
|
||||
self.poll_seconds = int(cfg("interval_seconds", cfg("poll_seconds", 120)))
|
||||
self.tz_name = cfg("timezone", "Asia/Shanghai")
|
||||
# self.watch_channels = set(cfg("watch_channels", ["snapshot"]))
|
||||
self.watch_channels = set(cfg("watch_channels", ["snapshot",'release']))
|
||||
self.article_lang = cfg("mc_article_lang", "en-us")
|
||||
|
||||
#设置自建回调
|
||||
self.hookfunction = hookfunction
|
||||
# data_dir = Path(StarTools.get_data_dir("astrbot_plugin_mcwatcher"))
|
||||
data_dir = Path("MCWatcher")
|
||||
data_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
self.state = State(data_dir / "state.json")
|
||||
self.state.load()
|
||||
|
||||
self._stop = False
|
||||
self._task = asyncio.run(self._poll_loop())
|
||||
# self._task = asyncio.create_task(self._poll_loop())
|
||||
|
||||
logger.info(
|
||||
f"{_ts()} MCWatcher started. interval={self.poll_seconds}s tz={self.tz_name} watch={self.watch_channels}"
|
||||
)
|
||||
|
||||
async def terminate(self):
|
||||
self._stop = True
|
||||
if self._task:
|
||||
self._task.cancel()
|
||||
# try:
|
||||
# await self._task
|
||||
# except asyncio.CancelledError:
|
||||
# pass
|
||||
self.state.save()
|
||||
|
||||
# ========== 工具方法 ==========
|
||||
|
||||
# def _get_plain_text(self, event: AstrMessageEvent) -> str:
|
||||
# try:
|
||||
# mc = getattr(event, "message_chain", None)
|
||||
# if mc:
|
||||
# parts = []
|
||||
# for seg in mc:
|
||||
# if isinstance(seg, Plain):
|
||||
# parts.append(getattr(seg, "text", "") or getattr(seg, "content", ""))
|
||||
# return "".join(parts)
|
||||
# except (AttributeError, TypeError) as e:
|
||||
# logger.warning(f"Failed to parse plain text: {e}", exc_info=True)
|
||||
# return ""
|
||||
|
||||
def _parse_fake_version(self, raw: str, keyword: str, default_vid: str) -> str:
|
||||
toks = raw.replace("\u3000", " ").split()
|
||||
tokens_lower = [t.lower() for t in toks]
|
||||
|
||||
if keyword in tokens_lower:
|
||||
idx = tokens_lower.index(keyword)
|
||||
if idx + 1 < len(toks):
|
||||
return toks[idx + 1]
|
||||
return default_vid
|
||||
|
||||
if len(toks) <= 2:
|
||||
return default_vid
|
||||
|
||||
return toks[2]
|
||||
|
||||
# ========== 指令 ==========
|
||||
|
||||
# # @command("mcwatch bind", alias={"mcwatch on", "mc订阅"})
|
||||
# async def bind_here(self, event: AstrMessageEvent):
|
||||
# sid = event.unified_msg_origin
|
||||
# self.state.targets.add(sid)
|
||||
# self.state.save()
|
||||
# yield event.plain_result("已绑定本会话。")
|
||||
|
||||
# # @command("mcwatch unbind", alias={"mcwatch off"})
|
||||
# async def unbind_here(self, event: AstrMessageEvent):
|
||||
# sid = event.unified_msg_origin
|
||||
# if sid in self.state.targets:
|
||||
# self.state.targets.remove(sid)
|
||||
# self.state.save()
|
||||
# yield event.plain_result("已取消绑定。")
|
||||
# else:
|
||||
# yield event.plain_result("本会话未绑定。")
|
||||
|
||||
# # @command("mcwatch list")
|
||||
# async def list_targets(self, event: AstrMessageEvent):
|
||||
# if not self.state.targets:
|
||||
# yield event.plain_result("暂无绑定会话。")
|
||||
# else:
|
||||
# yield event.plain_result("已绑定：\n" + "\n".join(sorted(self.state.targets)))
|
||||
|
||||
# # @command("mcwatch now")
|
||||
# async def check_now(self, event: AstrMessageEvent):
|
||||
# await self._check_once(force_push=True)
|
||||
# yield event.plain_result("OK，已主动检查一次。")
|
||||
|
||||
# ========== fake 推送（仅当前会话，不广播） ==========
|
||||
|
||||
# @command("mcwatch fake")
|
||||
# async def fake_snapshot(self, event: AstrMessageEvent):
|
||||
# raw = self._get_plain_text(event).strip()
|
||||
# vid = self._parse_fake_version(raw, "fake", "25w45a")
|
||||
# msg = self._build_message("snapshot", vid, datetime.now().isoformat())
|
||||
|
||||
# await self.ctx.send_message(event.unified_msg_origin, MessageChain([Plain(msg)]))
|
||||
# yield event.plain_result(f"已模拟 snapshot 推送：{vid}")
|
||||
|
||||
# # @command("mcwatch fake_release")
|
||||
# async def fake_release(self, event: AstrMessageEvent):
|
||||
# raw = self._get_plain_text(event).strip()
|
||||
# vid = self._parse_fake_version(raw, "fake_release", "1.21.3")
|
||||
# msg = self._build_message("release", vid, datetime.now().isoformat())
|
||||
|
||||
# await self.ctx.send_message(event.unified_msg_origin, MessageChain([Plain(msg)]))
|
||||
# yield event.plain_result(f"已模拟 release 推送：{vid}")
|
||||
|
||||
# ========== 轮询 ==========
|
||||
|
||||
async def _poll_loop(self):
|
||||
logger.info("等待bot完毕~")
|
||||
# 防止bot未初始化
|
||||
time.sleep(10)
|
||||
try:
|
||||
await self._check_once(force_push=False)
|
||||
except Exception as e:
|
||||
logger.warning(f"MCWatcher 首次检查异常：{e}", exc_info=True)
|
||||
|
||||
while not self._stop:
|
||||
try:
|
||||
await asyncio.sleep(self.poll_seconds)
|
||||
await self._check_once(force_push=False)
|
||||
except asyncio.CancelledError:
|
||||
break
|
||||
except Exception as e:
|
||||
logger.warning(f"MCWatcher loop error: {e}", exc_info=True)
|
||||
|
||||
async def _fetch_manifest(self, ignore_cache: bool = False):
|
||||
headers = {}
|
||||
if not ignore_cache and self.state.etag:
|
||||
headers["If-None-Match"] = self.state.etag
|
||||
|
||||
async with httpx.AsyncClient(timeout=15.0) as client:
|
||||
resp = await client.get(MANIFEST_URL, headers=headers)
|
||||
|
||||
if resp.status_code == 304:
|
||||
return None
|
||||
|
||||
resp.raise_for_status()
|
||||
|
||||
etag = resp.headers.get("ETag")
|
||||
if etag:
|
||||
self.state.etag = etag
|
||||
self.state.save()
|
||||
|
||||
return resp.json()
|
||||
|
||||
async def _check_once(self, force_push: bool):
|
||||
data = await self._fetch_manifest(ignore_cache=force_push)
|
||||
if data is None and not force_push:
|
||||
return None
|
||||
|
||||
latest = (data or {}).get("latest", {})
|
||||
msgs = []
|
||||
|
||||
if "snapshot" in self.watch_channels:
|
||||
sid = latest.get("snapshot")
|
||||
if sid and sid != self.state.last_snapshot_id:
|
||||
t = self._lookup_release_time(data, sid)
|
||||
msgs.append(self._build_message("snapshot", sid, t))
|
||||
self.state.last_snapshot_id = sid
|
||||
|
||||
if "release" in self.watch_channels:
|
||||
rid = latest.get("release")
|
||||
if rid and rid != self.state.last_release_id:
|
||||
t = self._lookup_release_time(data, rid)
|
||||
msgs.append(self._build_message("release", rid, t))
|
||||
self.state.last_release_id = rid
|
||||
|
||||
if msgs:
|
||||
self.state.save()
|
||||
await self._broadcast("\n\n".join(msgs), self.state.targets)
|
||||
|
||||
def _lookup_release_time(self, data: Optional[Dict[str, Any]], vid: str):
|
||||
if not data:
|
||||
return None
|
||||
for v in data.get("versions", []):
|
||||
if v.get("id") == vid:
|
||||
return v.get("releaseTime")
|
||||
return None
|
||||
|
||||
def _build_message(self, vtype: str, vid: str, release_iso: Optional[str]):
|
||||
dt_str = "未知时间"
|
||||
if release_iso:
|
||||
try:
|
||||
dt = datetime.fromisoformat(
|
||||
release_iso.replace("Z", "+00:00")
|
||||
).astimezone(ZoneInfo(self.tz_name))
|
||||
dt_str = dt.strftime("%Y-%m-%d %H:%M:%S %z")
|
||||
except (ValueError, ZoneInfoNotFoundError):
|
||||
dt_str = release_iso
|
||||
|
||||
article = format_article_url(self.article_lang, vid) if vtype == "snapshot" else ""
|
||||
|
||||
lines = [
|
||||
f"MC 更新：{vid} ({vtype})",
|
||||
f"时间：{dt_str}"
|
||||
]
|
||||
if article:
|
||||
lines.append(f"日志：{article}")
|
||||
|
||||
return "\n".join(lines)
|
||||
|
||||
# ========== 高性能并发广播 ==========
|
||||
# async def _broadcast(self, text: str, targets: Set[str]):
|
||||
# if not targets:
|
||||
# return
|
||||
|
||||
# mc = MessageChain([Plain(text)])
|
||||
# sid_list = list(targets)
|
||||
|
||||
# tasks = [self.ctx.send_message(sid, mc) for sid in sid_list]
|
||||
# results = await asyncio.gather(*tasks, return_exceptions=True)
|
||||
|
||||
# ok = 0
|
||||
# for sid, result in zip(sid_list, results):
|
||||
# if isinstance(result, Exception):
|
||||
# logger.warning(f"{_ts()} 推送失败 {sid}: {result}", exc_info=True)
|
||||
# elif result:
|
||||
# ok += 1
|
||||
|
||||
# logger.info(f"{_ts()} MCWatcher 已推送 {ok}/{len(targets)} 个会话。")
|
||||
async def _broadcast(self, text: str, targets: Set[str]):
|
||||
self.hookfunction(text)
|
||||
Reference in New Issue
Block a user