276 lines
8.0 KiB
Python
276 lines
8.0 KiB
Python
import json
|
||
import shutil
|
||
from datetime import date
|
||
from io import StringIO
|
||
from pathlib import Path
|
||
from typing import Any, List, Optional
|
||
|
||
from mcdreforged.api.all import *
|
||
|
||
try:
|
||
from ruamel.yaml import YAML
|
||
except ImportError: # pragma: no cover
|
||
YAML = None # type: ignore
|
||
|
||
|
||
def _today() -> str:
|
||
return date.today().strftime("%Y-%m-%d")
|
||
|
||
|
||
class Config(Serializable):
|
||
# 使用 !!share / !!hat / !!head / !!c 所需的最低权限等级
|
||
permission: int = 1
|
||
# !!head 冷却时间（秒），0 表示不限制
|
||
head_cooldown_seconds: int = 60
|
||
# !!hat 冷却时间（秒），0 表示不限制
|
||
hat_cooldown_seconds: int = 3
|
||
# !!share 是否播放提示音
|
||
share_sound: bool = True
|
||
# 强制指定 MC 版本号（如 "1.12.2"），留空则自动检测
|
||
force_version: str = ""
|
||
# 玩家加入时是否提示可用命令
|
||
join_tip: bool = True
|
||
# 加入提示延迟（秒），避免和进服消息挤在一起
|
||
join_tip_delay_seconds: float = 1.5
|
||
# 是否发送进服 MOTD（开服天数 + 自定义行）
|
||
motd_enabled: bool = True
|
||
# 开服日期，格式 YYYY-MM-DD；默认为配置生成当天（当天算第 1 天）
|
||
motd_start_day: str = _today()
|
||
# MOTD 文本行
|
||
motd_lines: List[str] = [
|
||
"§6欢迎 §e{player}§6 加入服务器！",
|
||
"§7服务器已开服 §b{days}§7 天",
|
||
"§7官网: {link:§b点击打开官网|https://www.skymirror.top}",
|
||
"§7QQ群: {link:§f985402607|https://skymirror.top/qq}",
|
||
]
|
||
|
||
|
||
# 当前配置：MCDR 数据目录 → config/iridium/iridium.yml
|
||
CONFIG_FILE_NAME = "iridium.yml"
|
||
|
||
|
||
def build_default_config_text() -> str:
|
||
"""带注释的默认 YAML 配置；开服日期为生成当天。"""
|
||
today = _today()
|
||
return f"""# Iridium 配置文件 (YAML)
|
||
# 修改后执行: !!MCDR reload plugin
|
||
# This file uses standard YAML. '#' starts a comment.
|
||
|
||
# 使用 !!share / !!hat / !!head / !!c 所需的最低权限等级
|
||
# 0=控制台 1=普通玩家 2=OP/管理员
|
||
permission: 1
|
||
|
||
# !!head 冷却时间（秒），0 表示不限制
|
||
head_cooldown_seconds: 60
|
||
|
||
# !!hat 冷却时间（秒），0 表示不限制
|
||
hat_cooldown_seconds: 3
|
||
|
||
# !!share 是否播放提示音
|
||
share_sound: true
|
||
|
||
# 强制指定 Minecraft 版本（如 "1.12.2"、"1.20.1"）
|
||
# 留空则自动检测；检测失败时可手动填写
|
||
force_version: ""
|
||
|
||
# 玩家进服时是否提示可用命令
|
||
join_tip: true
|
||
|
||
# 进服消息延迟（秒），0 为立即发送
|
||
join_tip_delay_seconds: 1.5
|
||
|
||
# 是否发送进服 MOTD（开服天数 + 自定义文案）
|
||
motd_enabled: true
|
||
|
||
# 开服日期，格式 YYYY-MM-DD（当天算第 1 天）
|
||
# 默认为本配置文件生成当天，可自行修改为真实开服日
|
||
motd_start_day: "{today}"
|
||
|
||
# MOTD 文本行
|
||
# 占位符: {{player}} 玩家名 {{days}} 开服天数 {{online}} 在线人数
|
||
# 链接: {{link:显示文字|https://url}} 或 {{link:https://url}}
|
||
# 颜色: 支持原版 § 颜色代码
|
||
motd_lines:
|
||
- "§6欢迎 §e{{player}}§6 加入服务器！"
|
||
- "§7服务器已开服 §b{{days}}§7 天"
|
||
- "§7官网: {{link:§b点击打开官网|https://www.skymirror.top}}"
|
||
- "§7QQ群: {{link:§f985402607|https://skymirror.top/qq}}"
|
||
"""
|
||
|
||
|
||
# 模块级单例；on_load 时原地更新字段，避免各模块持有过期引用
|
||
config = Config()
|
||
|
||
|
||
def apply_config(new_cfg: Config) -> None:
|
||
config.permission = new_cfg.permission
|
||
config.head_cooldown_seconds = new_cfg.head_cooldown_seconds
|
||
config.hat_cooldown_seconds = new_cfg.hat_cooldown_seconds
|
||
config.share_sound = new_cfg.share_sound
|
||
config.force_version = new_cfg.force_version
|
||
config.join_tip = new_cfg.join_tip
|
||
config.join_tip_delay_seconds = new_cfg.join_tip_delay_seconds
|
||
config.motd_enabled = new_cfg.motd_enabled
|
||
config.motd_start_day = new_cfg.motd_start_day
|
||
config.motd_lines = list(new_cfg.motd_lines)
|
||
|
||
|
||
def _yaml():
|
||
if YAML is None:
|
||
raise RuntimeError("ruamel.yaml is required")
|
||
return YAML(typ="safe")
|
||
|
||
|
||
def _load_yaml_text(text: str) -> Any:
|
||
return _yaml().load(StringIO(text))
|
||
|
||
|
||
def _dump_yaml(data: dict) -> str:
|
||
buf = StringIO()
|
||
y = _yaml()
|
||
y.default_flow_style = False
|
||
y.dump(data, buf)
|
||
return buf.getvalue()
|
||
|
||
|
||
def _strip_json_comments(text: str) -> str:
|
||
"""Remove // line comments outside of strings (legacy JSON only)."""
|
||
out = []
|
||
i = 0
|
||
n = len(text)
|
||
in_string = False
|
||
while i < n:
|
||
c = text[i]
|
||
if in_string:
|
||
out.append(c)
|
||
if c == "\\" and i + 1 < n:
|
||
out.append(text[i + 1])
|
||
i += 2
|
||
continue
|
||
if c == '"':
|
||
in_string = False
|
||
i += 1
|
||
continue
|
||
if c == '"':
|
||
in_string = True
|
||
out.append(c)
|
||
i += 1
|
||
continue
|
||
if c == "/" and i + 1 < n and text[i + 1] == "/":
|
||
while i < n and text[i] != "\n":
|
||
i += 1
|
||
continue
|
||
out.append(c)
|
||
i += 1
|
||
return "".join(out)
|
||
|
||
|
||
def config_file_path(server: PluginServerInterface) -> Path:
|
||
"""Canonical path: MCDR data folder → config/iridium/iridium.yml"""
|
||
return Path(server.get_data_folder()) / CONFIG_FILE_NAME
|
||
|
||
|
||
def _working_dir(server: PluginServerInterface) -> Path:
|
||
try:
|
||
wd = server.get_mcdr_config().get("working_directory") or "."
|
||
return Path(wd).expanduser()
|
||
except Exception:
|
||
return Path.cwd()
|
||
|
||
|
||
def _legacy_json_paths(server: PluginServerInterface) -> List[Path]:
|
||
data_dir = Path(server.get_data_folder())
|
||
wd = _working_dir(server)
|
||
return [
|
||
data_dir / "iridium.json",
|
||
wd / "config" / "iridium.json",
|
||
]
|
||
|
||
|
||
def _write_default(path: Path, server: PluginServerInterface) -> bool:
|
||
try:
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
path.write_text(build_default_config_text(), encoding="utf-8")
|
||
server.logger.info(f"已生成配置文件: {path}")
|
||
return True
|
||
except Exception as e:
|
||
server.logger.error(f"写入配置文件失败 {path}: {e}")
|
||
return False
|
||
|
||
|
||
def _try_load_dict_from_text(text: str) -> Optional[dict]:
|
||
text = text.strip()
|
||
if not text:
|
||
return None
|
||
if text.startswith("{"):
|
||
return json.loads(_strip_json_comments(text))
|
||
data = _load_yaml_text(text)
|
||
return data if isinstance(data, dict) else None
|
||
|
||
|
||
def _migrate_legacy_json(server: PluginServerInterface, dest: Path) -> None:
|
||
"""Convert old JSON config (with optional // comments) to YAML if dest is missing."""
|
||
if dest.is_file():
|
||
return
|
||
for src in _legacy_json_paths(server):
|
||
try:
|
||
if not src.is_file():
|
||
continue
|
||
raw = src.read_text(encoding="utf-8")
|
||
data = _try_load_dict_from_text(raw)
|
||
if not data:
|
||
continue
|
||
dest.parent.mkdir(parents=True, exist_ok=True)
|
||
# keep comments in default template + overlay known keys via dump
|
||
cfg = Config.deserialize(data)
|
||
overlay = {
|
||
"permission": cfg.permission,
|
||
"head_cooldown_seconds": cfg.head_cooldown_seconds,
|
||
"hat_cooldown_seconds": cfg.hat_cooldown_seconds,
|
||
"share_sound": cfg.share_sound,
|
||
"force_version": cfg.force_version,
|
||
"join_tip": cfg.join_tip,
|
||
"join_tip_delay_seconds": cfg.join_tip_delay_seconds,
|
||
"motd_enabled": cfg.motd_enabled,
|
||
"motd_start_day": cfg.motd_start_day,
|
||
"motd_lines": list(cfg.motd_lines),
|
||
}
|
||
# Prefer commented default template when values match defaults is complex;
|
||
# dump parsed values with a short header for clarity.
|
||
header = (
|
||
"# Iridium 配置 (从旧 JSON 迁移)\n"
|
||
"# Migrated from legacy JSON config\n\n"
|
||
)
|
||
dest.write_text(header + _dump_yaml(overlay), encoding="utf-8")
|
||
server.logger.info(f"已从旧 JSON 迁移配置: {src} -> {dest}")
|
||
return
|
||
except Exception as e:
|
||
server.logger.warning(f"迁移旧配置失败 {src}: {e}")
|
||
|
||
|
||
def load_config(server: PluginServerInterface) -> Config:
|
||
"""
|
||
Load config/iridium/iridium.yml (MCDR data folder).
|
||
Create commented default if missing; migrate legacy JSON if needed.
|
||
"""
|
||
path = config_file_path(server)
|
||
_migrate_legacy_json(server, path)
|
||
|
||
if not path.is_file():
|
||
_write_default(path, server)
|
||
|
||
if not path.is_file():
|
||
server.logger.warning(f"配置文件不存在且无法创建，使用内存默认值: {path}")
|
||
return Config()
|
||
|
||
try:
|
||
raw = path.read_text(encoding="utf-8")
|
||
data = _load_yaml_text(raw)
|
||
if not isinstance(data, dict):
|
||
raise ValueError("YAML root must be a mapping")
|
||
server.logger.info(f"配置已加载: {path}")
|
||
return Config.deserialize(data)
|
||
except Exception as e:
|
||
server.logger.error(f"读取/解析 YAML 配置失败 {path}: {e}，使用默认值")
|
||
return Config()
|