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

package/service/rover/compat.py

@@ -0,0 +1,215 @@
+"""让上游插件源码零改动运行。
+
+插件按它们原本的框架命名空间 import; 这里把那些名字全部指向本运行时的实现,
+因此插件目录可以直接 git pull 上游, 不需要改一行源码。
+"""
+
+import importlib
+import sys
+from types import ModuleType
+from typing import Dict, List
+
+from rover.logger import logger
+
+# 上游框架的顶层包名与插件命名空间
+UPSTREAM = "gsuid_core"
+UPSTREAM_PLUGINS = f"{UPSTREAM}.plugins"
+
+# 上游模块路径 → 本运行时模块路径
+MODULE_MAP: Dict[str, str] = {
+    "": "rover",
+    "bot": "rover.bot",
+    "models": "rover.models",
+    "segment": "rover.segment",
+    "sv": "rover.sv",
+    "trigger": "rover.trigger",
+    "handler": "rover.handler",
+    "logger": "rover.logger",
+    "config": "rover.config",
+    "data_store": "rover.data_store",
+    "aps": "rover.aps",
+    "pool": "rover.pool",
+    "gss": "rover.gss",
+    "server": "rover.server",
+    "subscribe": "rover.subscribe",
+    "i18n": "rover.i18n",
+    "message_models": "rover.message_models",
+    "web_app": "rover.web_app",
+    "app_life": "rover.app_life",
+    "utils": "rover.utils",
+    "utils.database": "rover.utils.database",
+    "utils.database.base_models": "rover.utils.database.base_models",
+    "utils.database.models": "rover.utils.database.models",
+    "utils.database.startup": "rover.utils.database.startup",
+    "utils.plugins_config": "rover.utils.plugins_config",
+    "utils.plugins_config.models": "rover.utils.plugins_config.models",
+    "utils.plugins_config.gs_config": "rover.utils.plugins_config.gs_config",
+    "utils.image": "rover.utils.image",
+    "utils.image.convert": "rover.utils.image.convert",
+    "utils.image.image_tools": "rover.utils.image.image_tools",
+    "utils.image.utils": "rover.utils.image.utils",
+    "utils.fonts": "rover.utils.fonts",
+    "utils.fonts.fonts": "rover.utils.fonts.fonts",
+    "utils.download_resource": "rover.utils.download_resource",
+    "utils.download_resource.download_file": "rover.utils.download_resource.download_file",
+    "utils.download_resource.download_core": "rover.utils.download_resource.download_core",
+    "utils.cookie_manager": "rover.utils.cookie_manager",
+    "utils.cookie_manager.qrlogin": "rover.utils.cookie_manager.qrlogin",
+    "utils.boardcast": "rover.utils.boardcast",
+    "utils.boardcast.models": "rover.utils.boardcast.models",
+    "help": "rover.help",
+    "help.model": "rover.help.model",
+    "help.utils": "rover.help.utils",
+    "help.draw_new_plugin_help": "rover.help.draw_new_plugin_help",
+    "status": "rover.status",
+    "status.plugin_status": "rover.status.plugin_status",
+    "webconsole": "rover.webconsole",
+    "webconsole.mount_app": "rover.webconsole.mount_app",
+    "ai_core": "rover.ai_core",
+    "ai_core.models": "rover.ai_core.models",
+    "ai_core.register": "rover.ai_core.register",
+    "ai_core.resource": "rover.ai_core.resource",
+    "ai_core.rag": "rover.ai_core.rag",
+    "ai_core.rag.knowledge": "rover.ai_core.rag.knowledge",
+    "ai_core.skills": "rover.ai_core.skills",
+    "ai_core.skills.operations": "rover.ai_core.skills.operations",
+}
+
+# 上游的重启入口在 buildin 插件里, 这里造一个同路径模块转发过去
+RESTART_PATH = f"{UPSTREAM}.buildin_plugins.core_command.core_restart.restart"
+
+# 上游配置模型的类名, 与本运行时的对应关系
+CONFIG_CLASS_ALIASES = {
+    "GsStrConfig": "StrConfig",
+    "GsBoolConfig": "BoolConfig",
+    "GsDictConfig": "DictConfig",
+    "GsListStrConfig": "ListStrConfig",
+    "GsListConfig": "ListConfig",
+    "GsIntConfig": "IntConfig",
+    "GsFloatConfig": "FloatConfig",
+    "GsImageConfig": "ImageConfig",
+    "GsTimeRConfig": "TimeRConfig",
+    "GsDivider": "Divider",
+    "GsFileUploadConfig": "FileUploadConfig",
+    "GsFilesUploadConfig": "FilesUploadConfig",
+    "GsDateConfig": "DateConfig",
+    "GsTimeRangeConfig": "TimeRangeConfig",
+    "GsColorConfig": "ColorConfig",
+    "GsRepeatGroupConfig": "RepeatGroupConfig",
+    "GsTimeConfig": "TimeConfig",
+    "GSC": "ConfigItem",
+}
+
+# 其它沿用上游名字的符号: 运行时模块 → {上游名: 本地名}
+SYMBOL_ALIASES = {
+    "rover.webconsole.mount_app": {"GsAdminModel": "AdminModel"},
+}
+
+_installed = False
+
+
+def _alias_module(upstream_name: str, target_name: str) -> None:
+    module = importlib.import_module(target_name)
+    sys.modules[upstream_name] = module
+
+
+def _make_namespace(name: str, path: List[str]) -> ModuleType:
+    module = ModuleType(name)
+    module.__path__ = path  # type: ignore[attr-defined]
+    sys.modules[name] = module
+    return module
+
+
+def _install_config_aliases() -> None:
+    models = importlib.import_module("rover.utils.plugins_config.models")
+    for upstream_name, ours in CONFIG_CLASS_ALIASES.items():
+        target = getattr(models, ours, None)
+        if target is None:
+            logger.warning(f"[兼容层] 配置模型缺少 {ours}")
+            continue
+        setattr(models, upstream_name, target)
+
+
+def _install_symbol_aliases() -> None:
+    for module_name, mapping in SYMBOL_ALIASES.items():
+        try:
+            module = importlib.import_module(module_name)
+        except ModuleNotFoundError as e:
+            logger.warning(f"[兼容层] 跳过 {module_name}: {e}")
+            continue
+        for upstream_name, ours in mapping.items():
+            target = getattr(module, ours, None)
+            if target is None:
+                logger.warning(f"[兼容层] {module_name} 缺少 {ours}")
+                continue
+            setattr(module, upstream_name, target)
+
+
+def _install_restart_shim() -> None:
+    import rover.restart as restart
+
+    module = ModuleType(RESTART_PATH)
+    module.restart_genshinuid = restart.restart_service  # type: ignore[attr-defined]
+    for attr in dir(restart):
+        if not attr.startswith("_"):
+            setattr(module, attr, getattr(restart, attr))
+    parts = RESTART_PATH.split(".")
+    for i in range(2, len(parts)):
+        parent = ".".join(parts[:i])
+        if parent not in sys.modules:
+            _make_namespace(parent, [])
+    sys.modules[RESTART_PATH] = module
+    setattr(sys.modules[".".join(parts[:-1])], parts[-1], module)
+
+
+def install(plugins_path: List[str]) -> None:
+    """建立命名空间映射。plugins_path 是存放上游插件包的目录列表。"""
+    global _installed
+    if _installed:
+        return
+    _installed = True
+
+    for suffix, target in MODULE_MAP.items():
+        upstream_name = UPSTREAM if not suffix else f"{UPSTREAM}.{suffix}"
+        try:
+            _alias_module(upstream_name, target)
+        except ModuleNotFoundError as e:
+            logger.warning(f"[兼容层] 跳过 {upstream_name}: {e}")
+
+    _make_namespace(UPSTREAM_PLUGINS, list(plugins_path))
+    setattr(sys.modules[UPSTREAM], "plugins", sys.modules[UPSTREAM_PLUGINS])
+
+    _install_config_aliases()
+    _install_symbol_aliases()
+    _install_restart_shim()
+    logger.debug(f"[兼容层] 已映射 {len(MODULE_MAP)} 个模块")
+
+
+def missing_imports(source_root) -> List[str]:
+    """扫描插件源码里本运行时尚未提供的框架模块, 供上游更新后自检。"""
+    import ast
+    from pathlib import Path
+
+    known = {UPSTREAM if not s else f"{UPSTREAM}.{s}" for s in MODULE_MAP}
+    known.add(UPSTREAM_PLUGINS)
+    known.add(RESTART_PATH)
+    missing = set()
+
+    for py in Path(source_root).rglob("*.py"):
+        try:
+            tree = ast.parse(py.read_text(encoding="utf-8"))
+        except (OSError, SyntaxError):
+            continue
+        for node in ast.walk(tree):
+            names = []
+            if isinstance(node, ast.ImportFrom) and node.module and node.level == 0:
+                names = [node.module]
+            elif isinstance(node, ast.Import):
+                names = [a.name for a in node.names]
+            for name in names:
+                if name == UPSTREAM or name.startswith(f"{UPSTREAM}."):
+                    if name.startswith(UPSTREAM_PLUGINS):
+                        continue
+                    if name not in known:
+                        missing.add(name)
+    return sorted(missing)

package/service/rover/config.py

@@ -0,0 +1,163 @@
+"""核心配置: 数据根下的 config.json。"""
+
+import json
+import time
+from typing import Any, Dict, List, Union, Literal, overload
+
+from boltons.fileutils import atomic_save
+
+from rover.data_store import get_res_path
+
+CONFIG_PATH = get_res_path() / "config.json"
+
+CONFIG_DEFAULT: Dict[str, Any] = {
+    # 把插件作为工具暴露给模型(占少量上下文); 关闭后只能用 /ww 命令
+    "expose_to_model": True,
+    # 用户消息命中游戏常识关键词时自动附带知识条目; 关闭只停自动注入, 主动检索不受影响
+    "knowledge_enhance": True,
+    # 图片卡片(评分/面板等)本地 OCR 抽文字进 summary, 纯文本模型才看得到卡上数值
+    "image_ocr": True,
+    # 卡片头部头像: 本地图片路径 / http(s) 链接 / QQ 号; 置空用插件自带的
+    "avatar": "",
+    # 服务监听地址; 登录页与字体链接也按它拼
+    "HOST": "127.0.0.1",
+    "PORT": "9777",
+    # 命令前缀, 多个用逗号分隔, 置空则不需要前缀
+    "command_prefix": "ww",
+    # 本机使用者标识, 数据都归属它, 一般不用改
+    "masters": ["local"],
+    # 定时任务允许的迟到秒数
+    "misfire_grace_time": 90,
+    # 单条命令最长执行秒数
+    "command_timeout": 300,
+    # 竖图(高>宽)在会话区里的最大宽度百分比
+    "image_portrait_max_percent": 50,
+    # 横图(宽>=高)的最大宽度百分比
+    "image_landscape_max_percent": 75,
+    # 渲染产物目录容量上限(MB), 超出时按最旧优先淘汰; 0 表示不限
+    "media_max_mb": 512,
+    # 日志级别; 输出固定为 stdout + data/logs 下的滚动文件(有上限, 不会涨爆)
+    "log_level": "INFO",
+    # 已写过初值的插件, 写过就不再覆盖用户的选择
+    "seeded": [],
+    # 单个功能项的覆盖: {"功能名": {"enabled": false, "pm": 0}}
+    "sv": {},
+}
+
+STR_CONFIG = Literal["HOST", "PORT", "command_prefix", "avatar", "log_level"]
+INT_CONFIG = Literal[
+    "misfire_grace_time",
+    "command_timeout",
+    "media_max_mb",
+    "image_portrait_max_percent",
+    "image_landscape_max_percent",
+]
+LIST_CONFIG = Literal["masters", "seeded"]
+DICT_CONFIG = Literal["sv"]
+BOOL_CONFIG = Literal["expose_to_model"]
+
+
+class CoreConfig:
+    def __init__(self) -> None:
+        if not CONFIG_PATH.exists():
+            with open(CONFIG_PATH, "w", encoding="UTF-8") as file:
+                json.dump(CONFIG_DEFAULT, file, indent=4, ensure_ascii=False)
+
+        self.update_config()
+
+    def write_config(self):
+        max_retries = 3
+        retry_delay = 0.5
+
+        for attempt in range(max_retries):
+            try:
+                with atomic_save(
+                    str(CONFIG_PATH),
+                    text_mode=False,
+                    overwrite=True,
+                    overwrite_part=True,
+                    file_perms=0o644,
+                ) as file:
+                    if file:
+                        json_str = json.dumps(
+                            self.config,
+                            indent=4,
+                            ensure_ascii=False,
+                        )
+                        file.write(json_str.encode("utf-8"))
+                    else:
+                        raise RuntimeError("写入配置文件失败!")
+                return
+            except OSError as e:
+                if attempt < max_retries - 1:
+                    time.sleep(retry_delay)
+                else:
+                    raise RuntimeError(f"写入配置文件失败，已重试 {max_retries} 次: {str(e)}")
+
+    def update_config(self):
+        try:
+            with open(CONFIG_PATH, "r", encoding="UTF-8") as f:
+                self.config: Dict[str, Any] = json.load(f)
+        except UnicodeDecodeError:
+            with open(CONFIG_PATH, "r") as f:
+                self.config = json.load(f)
+
+        # 旧版把日志写成 {level,output,module} 一整块, 只保留级别
+        legacy = self.config.pop("log", None)
+        if isinstance(legacy, dict) and "log_level" not in self.config:
+            self.config["log_level"] = legacy.get("level") or "INFO"
+
+        # 对没有的值，添加默认值
+        for key in CONFIG_DEFAULT:
+            if key not in self.config:
+                self.config[key] = CONFIG_DEFAULT[key]
+            if isinstance(CONFIG_DEFAULT[key], Dict):
+                for sub_key in CONFIG_DEFAULT[key]:
+                    if sub_key not in self.config[key]:
+                        self.config[key][sub_key] = CONFIG_DEFAULT[key][sub_key]
+
+    @overload
+    def get_config(self, key: STR_CONFIG) -> str: ...
+
+    @overload
+    def get_config(self, key: LIST_CONFIG) -> List: ...
+
+    @overload
+    def get_config(self, key: INT_CONFIG) -> int: ...
+
+    def get_config(self, key: str) -> Union[str, Dict, List, int, bool]:
+        if key in self.config:
+            return self.config[key]
+        elif key in CONFIG_DEFAULT:
+            self.update_config()
+            return self.config[key]
+        else:
+            return {}
+
+    @overload
+    def set_config(self, key: STR_CONFIG, value: str) -> bool: ...
+
+    @overload
+    def set_config(self, key: LIST_CONFIG, value: List) -> bool: ...
+
+    @overload
+    def set_config(self, key: INT_CONFIG, value: int) -> bool: ...
+
+    def set_config(self, key: str, value: Union[str, List, Dict, int, bool]) -> bool:
+        if key in CONFIG_DEFAULT:
+            self.config[key] = value
+            self.write_config()
+            return True
+        else:
+            return False
+
+    def lazy_write_config(self):
+        self.write_config()
+
+    def lazy_set_config(self, key: str, value: Union[str, List, Dict, int, bool]):
+        """只改内存, 由 lazy_write_config 统一落盘。"""
+        if key in CONFIG_DEFAULT:
+            self.config[key] = value
+
+
+core_config: CoreConfig = CoreConfig()

package/service/rover/custom_panel.py

@@ -0,0 +1,220 @@
+"""AI 自定义面板: 按传入 ctx 重组角色声骸后渲完整面板。
+
+只渲染: 不写 players/、不进上传队列(总排行)、不改任何用户数据。
+"""
+
+import copy
+from typing import Any, Dict, List, Optional, Tuple
+
+from rover.bot import Bot
+from rover.handler import build_message, msg_process
+from rover.logger import logger
+from rover.models import Message
+
+VALID_COSTS = {1, 3, 4}
+MAX_TOTAL_COST = 12
+
+
+def _props(pairs: Any) -> List[Dict[str, Any]]:
+    """[["攻击","18.0%"], ...] 或 [{"attributeName":...}] → Props 列表。"""
+    out: List[Dict[str, Any]] = []
+    for item in pairs or []:
+        if isinstance(item, dict):
+            name = item.get("attributeName") or item.get("name")
+            value = item.get("attributeValue") or item.get("value")
+        elif isinstance(item, (list, tuple)) and len(item) >= 2:
+            name, value = item[0], item[1]
+        else:
+            raise ValueError(f"词条格式不对: {item!r}")
+        if not name or value is None:
+            raise ValueError(f"词条缺名字或数值: {item!r}")
+        out.append({"attributeName": str(name), "attributeValue": str(value)})
+    return out
+
+
+def _build_custom_echo(spec: Dict[str, Any]) -> Dict[str, Any]:
+    from plugins.XutheringWavesUID.XutheringWavesUID.utils.name_convert import (
+        alias_to_echo_name,
+        alias_to_sonata_name,
+        echo_name_to_echo_id,
+    )
+
+    raw_name = str(spec.get("name") or "").strip()
+    if not raw_name:
+        raise ValueError("自定义声骸缺 name")
+    cost = int(spec.get("cost") or 0)
+    if cost not in VALID_COSTS:
+        raise ValueError(f"声骸 cost 只能是 1/3/4: {raw_name} cost={cost}")
+
+    echo_name = alias_to_echo_name(raw_name) or raw_name
+    echo_id = echo_name_to_echo_id(echo_name)
+    sonata = alias_to_sonata_name(str(spec.get("sonata") or "")) or str(
+        spec.get("sonata") or ""
+    )
+    quality = int(spec.get("quality") or 5)
+    main = _props(spec.get("main"))
+    subs = _props(spec.get("subs"))
+    if not main:
+        raise ValueError(f"声骸 {echo_name} 缺主词条 main")
+
+    return {
+        "phantomProp": {
+            "phantomPropId": 0,
+            "name": echo_name,
+            "phantomId": int(echo_id) if echo_id else 0,
+            "quality": quality,
+            "cost": cost,
+            "iconUrl": "",
+            "skillDescription": "",
+        },
+        "cost": cost,
+        "quality": quality,
+        "level": int(spec.get("level") or 25),
+        "fetterDetail": {"groupId": 0, "name": sonata, "num": int(spec.get("num") or 5)},
+        "mainProps": main,
+        "subProps": subs,
+    }
+
+
+async def _ref_echo(uid: str, ref: Dict[str, Any]):
+    """{"char": "洛可可", "slot": 2, "uid": 可选} → 那个角色身上装着的那件。"""
+    from plugins.XutheringWavesUID.XutheringWavesUID.utils.char_info_utils import (
+        get_char_detail_for_id,
+    )
+    from plugins.XutheringWavesUID.XutheringWavesUID.utils.name_convert import (
+        alias_to_char_name,
+        char_name_to_char_id,
+    )
+
+    char = str(ref.get("char") or "").strip()
+    slot = int(ref.get("slot") or 0)
+    if not char or slot < 1 or slot > 5:
+        raise ValueError(f"引用格式不对(需要 char + slot 1-5): {ref!r}")
+    char_name = alias_to_char_name(char)
+    char_id = char_name_to_char_id(char_name)
+    if not char_id:
+        raise ValueError(f"引用的角色不认识: {char}")
+    detail = await get_char_detail_for_id(str(ref.get("uid") or uid), char_id)
+    if detail is None:
+        raise ValueError(f"{char_name} 没有本地面板数据, 先 ww刷新{char_name}面板")
+    lst = detail.phantomData.equipPhantomList if detail.phantomData else None
+    echo = lst[slot - 1] if lst and len(lst) >= slot else None
+    if echo is None:
+        raise ValueError(f"{char_name} 的 {slot} 号位没有声骸")
+    return copy.deepcopy(echo)
+
+
+async def _apply_echoes(uid: str, role_detail, echoes: List[Dict[str, Any]]) -> None:
+    from plugins.XutheringWavesUID.XutheringWavesUID.utils.api.model import (
+        EquipPhantom,
+        EquipPhantomData,
+    )
+
+    if role_detail.phantomData is None:
+        role_detail.phantomData = EquipPhantomData(cost=0, equipPhantomList=None)
+    current = list(role_detail.phantomData.equipPhantomList or [None] * 5)
+    while len(current) < 5:
+        current.append(None)
+
+    for i, spec in enumerate(echoes):
+        if not isinstance(spec, dict):
+            raise ValueError(f"声骸项要是对象: {spec!r}")
+        slot = int(spec.get("slot") or i + 1)
+        if slot < 1 or slot > 5:
+            raise ValueError(f"slot 只能是 1-5: {slot}")
+        if spec.get("from"):
+            current[slot - 1] = await _ref_echo(uid, spec["from"])
+        else:
+            current[slot - 1] = EquipPhantom(**_build_custom_echo(spec))
+
+    total = sum(e.cost for e in current if e)
+    if total > MAX_TOTAL_COST:
+        raise ValueError(f"总 cost {total} 超过 {MAX_TOTAL_COST}, 检查各件 cost")
+    role_detail.phantomData.cost = total
+    role_detail.phantomData.equipPhantomList = current
+
+
+async def render_custom_panel(
+    payload: Dict[str, Any],
+) -> Tuple[Optional[str], List[Message]]:
+    """(错误文案, 消息段)。成功时错误为 None。"""
+    from plugins.XutheringWavesUID.XutheringWavesUID.utils.char_info_utils import (
+        get_char_detail_for_id,
+    )
+    from plugins.XutheringWavesUID.XutheringWavesUID.utils.database.models import (
+        WavesBind,
+    )
+    from plugins.XutheringWavesUID.XutheringWavesUID.utils.name_convert import (
+        alias_to_char_name,
+        char_name_to_char_id,
+    )
+    from plugins.XutheringWavesUID.XutheringWavesUID.wutheringwaves_charinfo.draw_char_card import (
+        draw_char_detail_img,
+    )
+
+    char = str(payload.get("char") or "").strip()
+    if not char:
+        return "缺角色名 char", []
+    char_name = alias_to_char_name(char)
+    char_id = char_name_to_char_id(char_name)
+    if not char_id or len(char_id) != 4 or not char_id.isdigit():
+        return f"未找到角色: {char}", []
+
+    from rover.config import core_config
+
+    masters = core_config.get_config("masters") or []
+    default_user = str(masters[0]) if masters else "dsh-user"
+    ev = msg_process(build_message(f"{char_name}面板", user_id=default_user))
+    ev.user_pm = 0
+
+    uid = str(payload.get("uid") or "").strip()
+    if not uid:
+        uid = await WavesBind.get_uid_by_game(ev.user_id, ev.bot_id) or ""
+    if not uid:
+        return "没有可用 UID: 传 uid, 或先 ww绑定<特征码>", []
+
+    base = await get_char_detail_for_id(uid, char_id)
+    if base is None:
+        return f"{char_name} 没有本地面板数据, 先跑一次 ww刷新{char_name}面板", []
+    role_detail = copy.deepcopy(base)
+
+    echoes = payload.get("echoes")
+    if echoes:
+        if not isinstance(echoes, list):
+            return "echoes 要是数组", []
+        try:
+            await _apply_echoes(uid, role_detail, echoes)
+        except ValueError as e:
+            return str(e), []
+
+    change = str(payload.get("change") or "").strip() or None
+
+    bot = Bot(ev)
+    try:
+        im = await draw_char_detail_img(
+            ev,
+            uid,
+            char_name,
+            ev.user_id,
+            change_list_regex=change,
+            role_detail_override=role_detail,
+        )
+        # 账号上下文取不到时降级为无账号渲染
+        if isinstance(im, str) and not change:
+            im2 = await draw_char_detail_img(
+                ev,
+                uid,
+                char_name,
+                ev.user_id,
+                is_limit_query=True,
+                role_detail_override=role_detail,
+            )
+            if not isinstance(im2, str):
+                im = im2
+    except Exception as e:
+        logger.exception(f"[自定义面板] 渲染失败 {char_name}: {e}")
+        return f"渲染失败: {e}", []
+    if isinstance(im, str):
+        return im, []
+    await bot.send(im)
+    return None, bot.sink

package/service/rover/data_store.py

@@ -0,0 +1,31 @@
+"""数据目录。根为 service/data。"""
+
+from pathlib import Path
+from typing import List, Optional, Union
+
+core_path = Path(__file__).parent
+data_path = Path(__file__).parents[1] / "data"
+
+
+def get_res_path(_path: Optional[Union[str, List[str], Path]] = None) -> Path:
+    if _path is None:
+        path = data_path
+    elif isinstance(_path, Path):
+        path = _path
+    elif isinstance(_path, str):
+        path = data_path / _path
+    else:
+        path = data_path.joinpath(*_path)
+
+    if not path.exists():
+        path.mkdir(parents=True)
+
+    return path
+
+
+RES = get_res_path()
+
+# 下面这些按需创建: 取用时才落盘, 避免一上来就铺一堆空目录
+image_res = data_path / "IMAGE_TEMP"
+data_cache_path = data_path / "DATA_CACHE_PATH"
+CONFIGS_PATH = data_path / "configs"

package/service/rover/gss.py

@@ -0,0 +1,19 @@
+"""连接注册表。定时推送与订阅回发都从这里取发送端。"""
+
+from typing import Dict
+
+from rover.bot import PushBot
+
+DEFAULT_BOT_ID = "dsh"
+
+
+class ServerState:
+    def __init__(self):
+        self.active_bot: Dict[str, PushBot] = {DEFAULT_BOT_ID: PushBot(DEFAULT_BOT_ID, DEFAULT_BOT_ID)}
+        self.bot_users: Dict[str, list] = {}
+
+    def get_bot(self, bot_id: str = DEFAULT_BOT_ID) -> PushBot:
+        return self.active_bot.get(bot_id) or self.active_bot[DEFAULT_BOT_ID]
+
+
+gss = ServerState()