@sophiacyl/dsh-plugin-waves 0.0.0-stage → 0.1.0-beta.2

package/service/rover/api_render.py

@@ -0,0 +1,161 @@
+"""消息段 → 前端 JSON。图片/HTML 落盘走 /api/media, 避免把 MB 级 base64 塞进响应。"""
+
+import base64
+import html as html_lib
+import re
+from typing import Any, Dict, List
+
+import msgspec
+
+from rover.logger import logger
+from rover.media import media_url, save_media
+from rover.models import Message
+
+_IMAGE_MAGIC = {
+    b"\xff\xd8\xff": "jpg",
+    b"\x89PNG": "png",
+    b"GIF8": "gif",
+    b"RIFF": "webp",
+}
+
+
+def _guess_ext(data: bytes) -> str:
+    for magic, ext in _IMAGE_MAGIC.items():
+        if data.startswith(magic):
+            return ext
+    return "jpg"
+
+
+_DROP_TAGS = re.compile(r"<(script|style)\b.*?</\1>", re.S | re.I)
+# 头像格上挂着角色名, 摊平成文字时按原位置补进去
+_NAMED_TAG = re.compile(r'<[^>]*data-ww-name="([^"]*)"[^>]*>')
+_TAGS = re.compile(r"<[^>]+>")
+_BLANK = re.compile(r"[ \t ]+")
+# 卡片里的数值也交给模型: 太长会挤占上下文, 截断即可
+DIGEST_LIMIT = 1600
+
+
+def html_digest(html: str) -> str:
+    """把卡片里的可见文字抽出来, 让模型也能看到面板上的具体数值。"""
+    text = _DROP_TAGS.sub(" ", html)
+    text = _NAMED_TAG.sub(lambda m: m.group(0) + m.group(1) + " ", text)
+    text = _TAGS.sub(" ", text)
+    text = _BLANK.sub(" ", html_lib.unescape(text))
+    words: List[str] = []
+    for line in text.splitlines():
+        line = line.strip()
+        if line:
+            words.append(line)
+    joined = " ".join(words)
+    return joined[:DIGEST_LIMIT] + " …" if len(joined) > DIGEST_LIMIT else joined
+
+
+async def _image_segment(data: Any) -> Dict[str, Any]:
+    if isinstance(data, str):
+        if data.startswith("link://"):
+            return {"type": "image", "url": data[len("link://") :]}
+        if data.startswith("base64://"):
+            raw = base64.b64decode(data[len("base64://") :])
+        else:
+            return {"type": "image", "url": data}
+    elif isinstance(data, (bytes, bytearray, memoryview)):
+        raw = bytes(data)
+    else:
+        return {"type": "text", "text": str(data)}
+    name = save_media(raw, _guess_ext(raw))
+    seg: Dict[str, Any] = {"type": "image", "url": media_url(name), "bytes": len(raw)}
+    from rover.config import core_config
+
+    if core_config.get_config("image_ocr") is not False:
+        from rover.ocr import image_digest
+
+        digest = await image_digest(raw)
+        if digest:
+            seg["digest"] = digest
+    return seg
+
+
+async def render_segments(segments: List[Message]) -> List[Dict[str, Any]]:
+    out: List[Dict[str, Any]] = []
+    for seg in segments:
+        stype = seg.type or "text"
+        data = seg.data
+        try:
+            if stype == "text":
+                text = "" if data is None else str(data)
+                if text.strip():
+                    out.append({"type": "text", "text": text})
+            elif stype == "image":
+                out.append(await _image_segment(data))
+            elif stype == "html":
+                html = data if isinstance(data, str) else str(data)
+                name = save_media(html.encode("utf-8"), "html")
+                out.append(
+                    {
+                        "type": "html",
+                        "url": media_url(name),
+                        "bytes": len(html),
+                        "digest": html_digest(html),
+                    }
+                )
+            elif stype == "node":
+                items = data if isinstance(data, list) else []
+                children = await render_segments(
+                    [i if isinstance(i, Message) else Message(**i) for i in items]
+                )
+                out.append({"type": "node", "items": children})
+            elif stype == "file":
+                raw = str(data or "")
+                file_name, _, payload = raw.partition("|")
+                if payload.startswith("link://"):
+                    out.append(
+                        {"type": "file", "name": file_name, "url": payload[len("link://") :]}
+                    )
+                else:
+                    ext = file_name.rsplit(".", 1)[-1] if "." in file_name else "bin"
+                    name = save_media(base64.b64decode(payload), ext)
+                    out.append({"type": "file", "name": file_name, "url": media_url(name)})
+            elif stype == "buttons":
+                out.append({"type": "buttons", "buttons": msgspec.to_builtins(data)})
+            elif stype == "at":
+                out.append({"type": "at", "target": str(data)})
+            else:
+                out.append({"type": "text", "text": f"[{stype}]"})
+        except Exception as e:
+            logger.warning(f"消息段渲染失败 type={stype}: {e}")
+            out.append({"type": "text", "text": f"[{stype} 渲染失败]"})
+    return out
+
+
+def summarize(rendered: List[Dict[str, Any]]) -> str:
+    """给模型看的纯文本摘要（图片/HTML 只留占位, 不塞内容）。"""
+    parts: List[str] = []
+    for seg in rendered:
+        stype = seg.get("type")
+        if stype == "text":
+            parts.append(seg.get("text", ""))
+        elif stype == "image":
+            digest = seg.get("digest") or ""
+            parts.append(
+                f"[图片已在卡片中展示]\n图片文字(OCR): {digest}" if digest
+                else "[图片已在卡片中展示]"
+            )
+        elif stype == "html":
+            digest = seg.get("digest") or ""
+            parts.append(
+                f"[图文卡片已在界面中展示]\n卡片内容: {digest}" if digest
+                else "[图文卡片已在界面中展示]"
+            )
+        elif stype == "node":
+            parts.append(summarize(seg.get("items", [])))
+        elif stype == "file":
+            parts.append(f"[文件 {seg.get('name')}]")
+        elif stype == "buttons":
+            labels = []
+            for row in seg.get("buttons") or []:
+                for btn in row if isinstance(row, list) else [row]:
+                    if isinstance(btn, dict) and btn.get("text"):
+                        labels.append(str(btn["text"]))
+            if labels:
+                parts.append("可用快捷指令: " + " / ".join(labels))
+    return "\n".join(p for p in parts if p)

package/service/rover/app_life.py

@@ -0,0 +1,154 @@
+"""服务生命周期。启动顺序: 建库 → 载插件 → 建表迁移 → 启动钩子与调度器。"""
+
+import asyncio
+import importlib
+import pkgutil
+from contextlib import asynccontextmanager
+from pathlib import Path
+
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+from rover.logger import logger
+
+PLUGINS_DIR = Path(__file__).resolve().parents[1] / "plugins"
+
+# 目录名 → 内层包名(上游插件都是同名双层目录)
+PLUGIN_PACKAGES = ["XutheringWavesUID", "RoverSign", "ScoreEcho"]
+
+# 面向原框架自带模型的模块, dsh 有自己的模型, 不加载
+SKIP_MODULES = {"wutheringwaves_ai_rag"}
+
+# 运行时的硬性前提: 出图走本地 HTML, 没有远端渲染器。这两项在设置页里也不展示
+CONFIG_FORCED = {
+    "XutheringWavesUID": {
+        "UseHtmlRender": True,
+        "RemoteRenderEnable": False,
+        # 推送不绑 dsh 会话, 会落到全局收件箱并出现在每张卡片上, 本地关掉
+        "WavesAnnOpen": False,
+    },
+}
+
+# 初值: 只在插件第一次装载时写一次, 之后以用户在设置页里的选择为准
+CONFIG_SEED = {
+    "XutheringWavesUID": {
+        "RankActiveFilterGroup": False,
+        "ActiveUserDays": 0,
+    },
+    "RoverSign": {
+        "UserWavesSignin": True,
+        "UserPGRSignin": True,
+        "UserBBSSchedSignin": True,
+        "SigninMaster": True,
+        "SignActiveUserOnly": False,
+        "SigninMasterSkipInactive": False,
+        "ActiveUserDays": 0,
+    },
+}
+
+_plugins_loaded = False
+plugin_load_errors = []
+
+
+def _inner_package(name: str) -> str:
+    return f"plugins.{name}.{name}"
+
+
+def load_plugins() -> None:
+    """按上游的双层包结构加载, 加载前后各施加一次本地改造。"""
+    global _plugins_loaded
+    if _plugins_loaded:
+        return
+    _plugins_loaded = True
+
+    from rover import compat, hooks
+
+    compat.install([str(PLUGINS_DIR)])
+    # 帮助表在插件 import 时就定型, 扩展模块清单得先按目录里实际有哪些插件对齐
+    hooks.align_help_modules(
+        {p.name for p in PLUGINS_DIR.iterdir() if p.is_dir() and not p.name.startswith(("_", "."))}
+    )
+
+    missing = compat.missing_imports(PLUGINS_DIR)
+    if missing:
+        logger.warning(f"[加载] 运行时缺少上游用到的模块: {missing}")
+
+    for name in PLUGIN_PACKAGES:
+        package_name = _inner_package(name)
+        try:
+            package = importlib.import_module(package_name)
+        except Exception as e:
+            plugin_load_errors.append(name)
+            logger.error(f"[加载] 插件 {name} 载入失败: {e}")
+            logger.exception(e)
+            continue
+
+        hooks.patch_modules(name, package_name)
+
+        loaded = 0
+        for module in pkgutil.iter_modules(package.__path__):
+            if not module.ispkg or module.name.startswith("_") or module.name == "utils":
+                continue
+            if module.name in SKIP_MODULES:
+                continue
+            try:
+                importlib.import_module(f"{package_name}.{module.name}")
+                loaded += 1
+            except Exception as e:
+                plugin_load_errors.append(f"{name}.{module.name}")
+                logger.error(f"[加载] 模块 {name}.{module.name} 载入失败: {e}")
+                logger.exception(e)
+        logger.success(f"[加载] {name} 已载入 {loaded} 个功能模块")
+
+    importlib.import_module("rover.builtin")
+    hooks.apply_defaults(CONFIG_FORCED)
+    hooks.seed_defaults(CONFIG_SEED)
+    hooks.align_local_urls()
+    hooks.drop_commands()
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    from rover.aps import start_scheduler, stop_scheduler
+    from rover.media import register_cleanup_job
+    from rover.server import core_shutdown_execute, core_start_before_execute, core_start_execute
+    from rover.utils.database.base_models import init_database
+    from rover.utils.database.startup import init_all
+
+    await init_database()
+    load_plugins()
+    await init_all()
+    await core_start_before_execute()
+    app.state.resources_ready = False
+    app.state.resource_error = None
+
+    async def prepare_resources():
+        try:
+            await core_start_execute()
+            app.state.resources_ready = True
+        except Exception as error:
+            app.state.resource_error = str(error)
+            logger.exception(error)
+
+    resource_task = asyncio.create_task(prepare_resources())
+    register_cleanup_job()
+    await start_scheduler()
+    logger.success("[服务] 启动完成")
+    try:
+        yield
+    finally:
+        resource_task.cancel()
+        await asyncio.gather(resource_task, return_exceptions=True)
+        await stop_scheduler()
+        await core_shutdown_execute()
+        logger.info("[服务] 已停止")
+
+
+app = FastAPI(title="鸣潮服务", lifespan=lifespan)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=False,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)

package/service/rover/aps.py

@@ -0,0 +1,47 @@
+"""定时任务调度器。插件通过 `@scheduler.scheduled_job` / `scheduler.add_job` 注册。"""
+
+from apscheduler.executors.pool import ThreadPoolExecutor
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+
+from rover.logger import logger
+
+# default 执行器保持 AsyncIOExecutor（协程任务必须在主事件循环跑），
+# 阻塞型任务显式 add_job(..., executor="threadpool")
+executor = ThreadPoolExecutor(10)
+def _misfire_grace_time() -> int:
+    from rover.config import core_config
+
+    try:
+        return int(core_config.get_config("misfire_grace_time") or 90)
+    except Exception:
+        return 90
+
+
+job_defaults = {"misfire_grace_time": _misfire_grace_time(), "coalesce": True}
+options = {
+    "executors": {"threadpool": executor},
+    "job_defaults": job_defaults,
+    "timezone": "Asia/Shanghai",
+}
+
+
+class _Scheduler(AsyncIOScheduler):
+    def scheduled_job(self, *args, **kwargs):
+        kwargs.pop("replace_existing", None)
+        return super().scheduled_job(*args, **kwargs)
+
+
+scheduler = _Scheduler()
+scheduler.configure(**options)
+
+
+async def start_scheduler() -> None:
+    if not scheduler.running:
+        scheduler.start()
+        logger.info("[调度器] 启动完成")
+
+
+async def stop_scheduler() -> None:
+    if scheduler.running:
+        scheduler.shutdown(wait=False)
+        logger.info("[调度器] 已关闭定时任务")

package/service/rover/avatar.py

@@ -0,0 +1,85 @@
+"""头像配置解析。
+
+配置项 avatar 支持三种写法: 本地图片路径 / http(s) 链接 / 纯数字 QQ 号,
+统一解析成能直接放进 <img src> 的地址; 置空则不覆盖, 沿用插件自己取的头像。
+"""
+
+import base64
+from pathlib import Path
+from typing import Dict, Optional, Set, Tuple
+
+from rover.logger import logger
+
+QQ_AVATAR_URL = "https://q1.qlogo.cn/g?b=qq&nk={}&s=640"
+
+# 按文件头认图片类型, 不信扩展名
+MAGICS = (
+    (b"\x89PNG\r\n\x1a\n", "image/png"),
+    (b"\xff\xd8\xff", "image/jpeg"),
+    (b"GIF8", "image/gif"),
+    (b"BM", "image/bmp"),
+    (b"\x00\x00\x01\x00", "image/x-icon"),
+)
+
+# 本地文件的 data:URI 缓存: 路径 → (修改时间, 大小, data:URI)
+_file_cache: Dict[str, Tuple[float, int, str]] = {}
+# 同一个坏配置只提示一次
+_warned: Set[str] = set()
+
+
+def _warn(raw: str, reason: str) -> None:
+    if raw in _warned:
+        return
+    _warned.add(raw)
+    logger.warning(f"[头像] 配置无效, 回退默认头像: {raw} ({reason})")
+
+
+def _mime(data: bytes) -> Optional[str]:
+    if data[:4] == b"RIFF" and data[8:12] == b"WEBP":
+        return "image/webp"
+    for magic, mime in MAGICS:
+        if data.startswith(magic):
+            return mime
+    if b"<svg" in data[:512].lower():
+        return "image/svg+xml"
+    return None
+
+
+def _data_uri(raw: str) -> str:
+    path = Path(raw).expanduser()
+    try:
+        stat = path.stat()
+    except OSError as e:
+        _warn(raw, f"读不到文件: {e}")
+        return ""
+    key = str(path)
+    cached = _file_cache.get(key)
+    if cached and cached[0] == stat.st_mtime and cached[1] == stat.st_size:
+        return cached[2]
+    try:
+        data = path.read_bytes()
+    except OSError as e:
+        _warn(raw, f"读文件失败: {e}")
+        return ""
+    mime = _mime(data)
+    if mime is None:
+        _warn(raw, "不是图片文件")
+        return ""
+    uri = f"data:{mime};base64,{base64.b64encode(data).decode()}"
+    _file_cache[key] = (stat.st_mtime, stat.st_size, uri)
+    _warned.discard(raw)
+    return uri
+
+
+def get_avatar_url() -> str:
+    """返回配置的头像地址; 未配置或配置无效时返回空串。"""
+    from rover.config import core_config
+
+    raw = str(core_config.get_config("avatar") or "").strip()
+    if not raw:
+        return ""
+    if raw.startswith(("http://", "https://", "data:")):
+        return raw
+    if raw.isascii() and raw.isdigit():
+        return QQ_AVATAR_URL.format(raw)
+    return _data_uri(raw)

package/service/rover/bot.py

@@ -0,0 +1,133 @@
+"""会话对象。插件调用 bot.send 产出的消息段被收集起来, 由上层一次性返回前端。"""
+
+from typing import Any, Dict, List, Optional, Union
+
+import msgspec
+
+from rover.logger import logger
+from rover.message_models import Button
+from rover.models import Event, Message
+from rover.segment import MessageSegment, convert_message
+
+msgjson = msgspec.json
+
+
+class _AsyncLogger:
+    """插件里以 `await bot.logger.info(...)` 形式使用。"""
+
+    def __init__(self, prefix: str = ""):
+        self.prefix = prefix
+
+    async def _log(self, level: str, message: Any, *args: Any) -> None:
+        getattr(logger, level)(f"{self.prefix}{message}", *args)
+
+    async def info(self, message: Any, *args: Any) -> None:
+        await self._log("info", message, *args)
+
+    async def debug(self, message: Any, *args: Any) -> None:
+        await self._log("debug", message, *args)
+
+    async def trace(self, message: Any, *args: Any) -> None:
+        await self._log("trace", message, *args)
+
+    async def success(self, message: Any, *args: Any) -> None:
+        await self._log("success", message, *args)
+
+    async def warning(self, message: Any, *args: Any) -> None:
+        await self._log("warning", message, *args)
+
+    async def error(self, message: Any, *args: Any) -> None:
+        await self._log("error", message, *args)
+
+    async def exception(self, message: Any, *args: Any) -> None:
+        logger.exception(message)
+
+
+def _log_dropped_push(segments, target_type, target_id) -> None:
+    """主动推送在本地没有去处: 只有一个使用者, 也没有群会话。记一行日志就够了。"""
+    from rover.logger import logger
+
+    text = " ".join(str(s.data) for s in segments if getattr(s, "type", "") == "text")
+    logger.info(f"[推送] 丢弃一条 {target_type}/{target_id or '-'} 的主动推送: {text[:80]}")
+
+
+class Bot:
+    def __init__(self, ev: Event, sink: Optional[List[Message]] = None):
+        self.ev = ev
+        self.bot_id = ev.bot_id
+        self.bot_self_id = ev.bot_self_id
+        self.logger = _AsyncLogger()
+        self.sink: List[Message] = sink if sink is not None else []
+
+    async def send(
+        self,
+        message: Union[Message, List[Message], List[str], str, bytes, None],
+        at_sender: bool = False,
+        extra_metadata: Optional[Dict] = None,
+        wait_recall: bool = False,
+    ) -> None:
+        if message is None:
+            return None
+        segments = await convert_message(message, self.bot_id, self.bot_self_id)
+        self.sink.extend(segments)
+        return None
+
+    async def send_option(
+        self,
+        reply: Union[Message, List[Message], List[str], str, bytes, None],
+        option_list: Optional[Union[List[Button], List[List[Button]]]] = None,
+        unsuported_platform: bool = False,
+        is_mutiply: bool = False,
+        timeout: int = 60,
+    ) -> None:
+        await self.send(reply)
+        if option_list:
+            self.sink.append(MessageSegment.buttons(option_list))
+        return None
+
+    async def target_send(
+        self,
+        message: Union[Message, List[Message], List[str], str, bytes],
+        target_type: str,
+        target_id: Optional[str],
+        bot_id: str = "",
+        bot_self_id: str = "",
+        msg_id: str = "",
+        at_sender: bool = False,
+        sender_id: str = "",
+        group_id: Optional[str] = None,
+        **kwargs: Any,
+    ) -> None:
+        segments = await convert_message(message, bot_id or self.bot_id, bot_self_id)
+        _log_dropped_push(segments, target_type, target_id)
+        return None
+
+
+class PushBot:
+    """给定时任务/订阅推送用的发送端, 语义与 Bot.target_send 一致。"""
+
+    def __init__(self, bot_id: str = "dsh", bot_self_id: str = "dsh"):
+        self.bot_id = bot_id
+        self.bot_self_id = bot_self_id
+        self.logger = _AsyncLogger()
+
+    async def target_send(
+        self,
+        message: Union[Message, List[Message], List[str], str, bytes],
+        target_type: str,
+        target_id: Optional[str],
+        bot_id: str = "",
+        bot_self_id: str = "",
+        msg_id: str = "",
+        at_sender: bool = False,
+        sender_id: str = "",
+        group_id: Optional[str] = None,
+        **kwargs: Any,
+    ) -> None:
+        segments = await convert_message(message, bot_id or self.bot_id, bot_self_id)
+        _log_dropped_push(segments, target_type, target_id)
+        return None
+
+    async def send(self, message: Any, *args: Any, **kwargs: Any) -> None:
+        await self.target_send(message, "group", "dsh")
+        return None