415 lines
17 KiB
Python
415 lines
17 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""Project configuration model."""
|
||
|
||
from util.runtime import *
|
||
|
||
def _empty_config_data():
|
||
"""无项目时的默认配置结构，避免 get() 报错"""
|
||
return {
|
||
"workspace": {"server_root": "", "run_dir": ""},
|
||
"database": {"host": "", "user": "", "password": "", "port": 3306},
|
||
"server": {
|
||
"prefix": "ddxq2", "login_server_node": "", "center_server_node": "",
|
||
"default_server_id": "1", "server_type": "game_server", "server_name": "",
|
||
"ip": get_local_ip(), "game_host": get_local_ip(), "cookie": "ddxq2-node",
|
||
},
|
||
"erlang": {"r25_path": "", "erl": "erl", "werl": "werl", "escript": "escript"},
|
||
"login_db": {
|
||
"name": "", "server_id": 900, "host": "", "port": 0,
|
||
"user": "", "password": "",
|
||
},
|
||
"commands": {"rebar": REBAR3_CMD, "svn": "svn"},
|
||
"terminal": {"launch_mode": "auto", "exe_path": ""},
|
||
}
|
||
|
||
|
||
class Config:
|
||
"""项目配置管理类：读写当前打开项目下的 .server_manager/config/tool.config（与项目一起走）。
|
||
打开项目时会将软件目录下受管模板同步到 .server_manager/config，并保留项目自己的 tool.config。
|
||
支持“无项目”启动：project_root=None 时仅显示欢迎页；通过 open_project(path) 打开项目后加载该项目配置。
|
||
"""
|
||
|
||
def __init__(self, project_root: Optional[Path] = None):
|
||
if getattr(sys, 'frozen', False):
|
||
self._exe_dir = Path(sys.executable).parent
|
||
self._script_dir = self._exe_dir
|
||
else:
|
||
self._script_dir = SRC_DIR
|
||
self._exe_dir = self._script_dir.parent.parent
|
||
|
||
self.tool_dir = None
|
||
self.tool_config_path = None
|
||
self.config_error = None
|
||
self.data = _empty_config_data()
|
||
|
||
if project_root is not None:
|
||
self.open_project(Path(project_root))
|
||
else:
|
||
self.config_error = "未打开项目"
|
||
|
||
def has_project(self) -> bool:
|
||
"""是否已打开有效项目（存在 .server_manager/config）"""
|
||
if self.tool_dir is None:
|
||
return False
|
||
return get_server_manager_config_dir(self.tool_dir).is_dir()
|
||
|
||
def open_project(self, project_root: Path) -> bool:
|
||
"""打开项目目录并同步受管配置模板到 .server_manager/config。成功返回 True。"""
|
||
project_root = Path(project_root).resolve()
|
||
cfg_dir, err = ensure_and_get_server_manager_config_dir(project_root)
|
||
if not cfg_dir:
|
||
self.config_error = err
|
||
return False
|
||
self.tool_dir = project_root
|
||
self.tool_config_path = cfg_dir / "tool.config"
|
||
self.config_error = None
|
||
self.data = self._load_config()
|
||
return True
|
||
|
||
def _find_server_root(self) -> Path:
|
||
"""查找有效的服务器根目录
|
||
|
||
查找顺序：优先运行目录（当前工作目录），再考虑 exe/脚本所在位置。
|
||
|
||
Returns:
|
||
有效的服务器根目录路径，如果都找不到返回 None
|
||
"""
|
||
candidates = [
|
||
Path.cwd(), # 运行目录优先
|
||
self._exe_dir,
|
||
self._script_dir,
|
||
]
|
||
|
||
for path in candidates:
|
||
if (get_server_manager_config_dir(path) / "default.kv").exists():
|
||
return path
|
||
if (path / "config" / "default.kv").exists():
|
||
return path
|
||
|
||
return None
|
||
|
||
def _load_tool_config(self) -> dict:
|
||
"""加载 tool.config 文件"""
|
||
if self.tool_config_path is None or not self.tool_config_path.exists():
|
||
return {}
|
||
|
||
config = {}
|
||
try:
|
||
with open(self.tool_config_path, 'r', encoding='utf-8') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if not line or line.startswith('#'):
|
||
continue
|
||
if '=' in line:
|
||
key, value = line.split('=', 1)
|
||
config[key.strip()] = value.strip()
|
||
except Exception:
|
||
return {}
|
||
return config
|
||
|
||
def _save_tool_config(self, config: dict):
|
||
"""保存 tool.config 文件"""
|
||
lines = [
|
||
"# Server Manager config",
|
||
"# 此文件配置工具运行所需的环境路径",
|
||
"",
|
||
"# === 工作目录配置 ===",
|
||
f"server_root={config.get('server_root', '')}",
|
||
f"run_dir={config.get('run_dir', '')}",
|
||
"",
|
||
"# === 启动窗口配置 ===",
|
||
f"terminal_launch_mode={config.get('terminal_launch_mode', 'auto')}",
|
||
f"terminal_exe_path={config.get('terminal_exe_path', '')}",
|
||
"",
|
||
"# === Erlang 配置 ===",
|
||
f"r25_path={config.get('r25_path', '')}",
|
||
"",
|
||
"# === 数据库配置 ===",
|
||
f"db_host={config.get('db_host', '')}",
|
||
f"db_port={config.get('db_port', '3306')}",
|
||
f"db_user={config.get('db_user', '')}",
|
||
f"db_pass={config.get('db_pass', '')}",
|
||
"",
|
||
"# === 登录服数据库配置（远程连接用）===",
|
||
f"login_db_name={config.get('login_db_name', '')}",
|
||
f"login_db_server_id={config.get('login_db_server_id', '900')}",
|
||
f"login_db_host={config.get('login_db_host', '')}",
|
||
f"login_db_port={config.get('login_db_port', '0')}",
|
||
f"login_db_user={config.get('login_db_user', '')}",
|
||
f"login_db_pass={config.get('login_db_pass', '')}",
|
||
"",
|
||
"# === 服务器基础配置 ===",
|
||
f"prefix={config.get('prefix', 'ddxq2')}",
|
||
f"default_server_id={config.get('default_server_id', '1')}",
|
||
f"server_type={config.get('server_type', 'game_server')}",
|
||
f"server_name={config.get('server_name', '')}",
|
||
f"login_node={config.get('login_node', '')}",
|
||
f"center_node={config.get('center_node', '')}",
|
||
f"cookie={config.get('cookie', 'ddxq2-node')}",
|
||
]
|
||
|
||
self.tool_config_path.parent.mkdir(parents=True, exist_ok=True)
|
||
with open(self.tool_config_path, 'w', encoding='utf-8') as f:
|
||
f.write('\n'.join(lines))
|
||
|
||
def _load_default_kv(self, server_root: str) -> dict:
|
||
"""从 default.kv 读取默认配置"""
|
||
if not server_root:
|
||
return {}
|
||
|
||
default_kv_path = get_server_manager_config_dir(server_root) / 'default.kv'
|
||
if not default_kv_path.exists():
|
||
return {}
|
||
|
||
config = {}
|
||
try:
|
||
with open(default_kv_path, 'r', encoding='utf-8') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if not line or line.startswith('#'):
|
||
continue
|
||
if '=' in line:
|
||
key, value = line.split('=', 1)
|
||
config[key.strip()] = value.strip()
|
||
except Exception:
|
||
return {}
|
||
return config
|
||
|
||
def _load_config(self):
|
||
"""加载配置，从 tool.config 读取，空值从 default.kv 补充。
|
||
以当前已打开的项目目录（self.tool_dir）为 server_root 默认值，tool.config 可覆盖。
|
||
"""
|
||
if self.tool_dir is None or self.tool_config_path is None:
|
||
return self.data
|
||
tool_cfg = self._load_tool_config()
|
||
is_first_run = not self.tool_config_path.exists()
|
||
|
||
# 以当前项目目录为默认，tool.config 可覆盖
|
||
server_root = tool_cfg.get('server_root', '') or str(self.tool_dir)
|
||
run_dir = tool_cfg.get('run_dir', '')
|
||
if not run_dir and server_root:
|
||
run_dir = str(Path(server_root) / "run")
|
||
|
||
# 首次运行时，创建运行目录
|
||
if is_first_run and run_dir:
|
||
run_path = Path(run_dir)
|
||
if not run_path.exists():
|
||
try:
|
||
run_path.mkdir(parents=True, exist_ok=True)
|
||
except Exception:
|
||
pass
|
||
|
||
# 如果有 server_root，尝试从 default.kv 读取默认值
|
||
default_kv = self._load_default_kv(server_root)
|
||
|
||
# 辅助函数：优先使用 tool.config 值，为空则使用 default.kv 值
|
||
def get_value(tool_key, default_key, fallback=''):
|
||
val = tool_cfg.get(tool_key, '')
|
||
if not val and default_kv:
|
||
val = default_kv.get(default_key, fallback)
|
||
# 去掉可能的引号
|
||
if isinstance(val, str) and val.startswith("'") and val.endswith("'"):
|
||
val = val[1:-1]
|
||
return val if val else fallback
|
||
|
||
db_host = get_value('db_host', 'db_host', '')
|
||
db_user = get_value('db_user', 'db_user', '')
|
||
db_pass = get_value('db_pass', 'db_pass', '')
|
||
db_port = get_value('db_port', 'db_port', '3306')
|
||
cookie = get_value('cookie', 'cookie', 'ddxq2-node')
|
||
login_db_name = get_value('login_db_name', 'login_db_name', '')
|
||
login_db_server_id = get_value('login_db_server_id', 'login_db_server_id', '900')
|
||
login_db_host = get_value('login_db_host', 'login_db_host', '')
|
||
login_db_port = get_value('login_db_port', 'login_db_port', '0')
|
||
login_db_user = get_value('login_db_user', 'login_db_user', '')
|
||
login_db_pass = get_value('login_db_pass', 'login_db_pass', '')
|
||
r25_path = get_value('r25_path', 'r25_path', '')
|
||
terminal_launch_mode = (tool_cfg.get('terminal_launch_mode', '') or 'auto').strip().lower()
|
||
terminal_exe_path = tool_cfg.get('terminal_exe_path', '')
|
||
if not terminal_exe_path:
|
||
legacy_exe = tool_cfg.get('terminal_custom_command', '')
|
||
legacy_cwd = tool_cfg.get('terminal_custom_cwd', '')
|
||
if legacy_exe and legacy_cwd and not any(sep in legacy_exe for sep in ('/', '\\')):
|
||
legacy_path = Path(os.path.expandvars(os.path.expanduser(legacy_cwd))) / legacy_exe
|
||
if legacy_path.is_file():
|
||
terminal_exe_path = str(legacy_path.resolve())
|
||
if not terminal_exe_path:
|
||
terminal_exe_path = legacy_exe
|
||
|
||
# 服务器基础配置
|
||
prefix = get_value('prefix', 'prefix', 'ddxq2')
|
||
default_server_id = get_value('default_server_id', 'server_id', '1')
|
||
server_type = get_value('server_type', 'server_type', 'game_server')
|
||
server_name = get_value('server_name', 'server_name', '')
|
||
local_ip = get_local_ip()
|
||
login_node = get_value('login_node', 'login_node', '')
|
||
center_node = get_value('center_node', 'center_node', '')
|
||
|
||
config_data = {
|
||
"workspace": {
|
||
"server_root": server_root,
|
||
"run_dir": run_dir,
|
||
},
|
||
"database": {
|
||
"host": db_host,
|
||
"user": db_user,
|
||
"password": db_pass,
|
||
"port": int(db_port) if db_port else 3306,
|
||
},
|
||
"server": {
|
||
"prefix": prefix,
|
||
"login_server_node": login_node,
|
||
"center_server_node": center_node,
|
||
"default_server_id": default_server_id,
|
||
"server_type": server_type,
|
||
"server_name": server_name,
|
||
"ip": local_ip,
|
||
"game_host": local_ip,
|
||
"cookie": cookie,
|
||
},
|
||
"erlang": {
|
||
"r25_path": r25_path,
|
||
"erl": "erl",
|
||
"werl": "werl",
|
||
"escript": "escript"
|
||
},
|
||
"login_db": {
|
||
"name": login_db_name,
|
||
"server_id": int(login_db_server_id) if login_db_server_id else 900,
|
||
"host": login_db_host,
|
||
"port": int(login_db_port) if login_db_port else 0,
|
||
"user": login_db_user,
|
||
"password": login_db_pass,
|
||
},
|
||
"commands": {
|
||
"rebar": REBAR3_CMD,
|
||
"svn": "svn"
|
||
},
|
||
"terminal": {
|
||
"launch_mode": terminal_launch_mode,
|
||
"exe_path": terminal_exe_path,
|
||
},
|
||
}
|
||
|
||
# 每次启动都检查配置完整性，从 default.kv 补充缺失的值
|
||
need_save = is_first_run
|
||
if default_kv:
|
||
# 检查是否有从 default.kv 补充的值
|
||
fields_to_check = [
|
||
('db_host', db_host),
|
||
('db_user', db_user),
|
||
('db_pass', db_pass),
|
||
('prefix', prefix),
|
||
('login_node', login_node),
|
||
('center_node', center_node),
|
||
('cookie', cookie),
|
||
]
|
||
for field_name, new_value in fields_to_check:
|
||
old_value = tool_cfg.get(field_name, '')
|
||
if not old_value and new_value:
|
||
need_save = True
|
||
break
|
||
|
||
if need_save:
|
||
self.data = config_data
|
||
self.save()
|
||
|
||
return config_data
|
||
|
||
def has_default_kv(self) -> bool:
|
||
"""检查 default.kv 是否存在"""
|
||
server_root = self.get('workspace', 'server_root', default='')
|
||
if not server_root:
|
||
return False
|
||
default_kv_path = get_server_manager_config_dir(server_root) / 'default.kv'
|
||
return default_kv_path.exists()
|
||
|
||
def is_config_complete(self) -> bool:
|
||
"""检查工具配置是否完整"""
|
||
# 检查必要的配置项
|
||
db_host = self.get('database', 'host', default='')
|
||
db_user = self.get('database', 'user', default='')
|
||
prefix = self.get('server', 'prefix', default='')
|
||
return bool(db_host and db_user and prefix)
|
||
|
||
def save(self):
|
||
"""保存配置到 tool.config"""
|
||
if self.tool_config_path is None:
|
||
return
|
||
local_ip = get_local_ip()
|
||
self.set(local_ip, 'server', 'ip')
|
||
self.set(local_ip, 'server', 'game_host')
|
||
tool_cfg = {
|
||
'server_root': self.get('workspace', 'server_root', default=''),
|
||
'run_dir': self.get('workspace', 'run_dir', default=''),
|
||
'r25_path': self.get('erlang', 'r25_path', default=''),
|
||
'db_host': self.get('database', 'host', default=''),
|
||
'db_port': str(self.get('database', 'port', default=3306)),
|
||
'db_user': self.get('database', 'user', default=''),
|
||
'db_pass': self.get('database', 'password', default=''),
|
||
'login_db_name': self.get('login_db', 'name', default=''),
|
||
'login_db_server_id': str(self.get('login_db', 'server_id', default=900)),
|
||
'login_db_host': self.get('login_db', 'host', default=''),
|
||
'login_db_port': str(self.get('login_db', 'port', default=0)),
|
||
'login_db_user': self.get('login_db', 'user', default=''),
|
||
'login_db_pass': self.get('login_db', 'password', default=''),
|
||
'prefix': self.get('server', 'prefix', default='ddxq2'),
|
||
'default_server_id': self.get('server', 'default_server_id', default='1'),
|
||
'server_type': self.get('server', 'server_type', default='game_server'),
|
||
'server_name': self.get('server', 'server_name', default=''),
|
||
'login_node': self.get('server', 'login_server_node', default=''),
|
||
'center_node': self.get('server', 'center_server_node', default=''),
|
||
'cookie': self.get('server', 'cookie', default='ddxq2-node'),
|
||
'terminal_launch_mode': self.get('terminal', 'launch_mode', default='auto'),
|
||
'terminal_exe_path': self.get('terminal', 'exe_path', default=''),
|
||
}
|
||
self._save_tool_config(tool_cfg)
|
||
|
||
def get(self, *keys, default=None):
|
||
"""获取配置项"""
|
||
result = self.data
|
||
for key in keys:
|
||
if isinstance(result, dict) and key in result:
|
||
result = result[key]
|
||
else:
|
||
return default
|
||
return result
|
||
|
||
def set(self, value, *keys):
|
||
"""设置配置项"""
|
||
target = self.data
|
||
for key in keys[:-1]:
|
||
if key not in target:
|
||
target[key] = {}
|
||
target = target[key]
|
||
target[keys[-1]] = value
|
||
|
||
def is_configured(self) -> bool:
|
||
"""检查必要配置是否已设置"""
|
||
server_root = self.get('workspace', 'server_root', default='')
|
||
return bool(server_root and Path(server_root).exists())
|
||
|
||
def get_missing_config(self) -> list:
|
||
"""获取缺失的必要配置项"""
|
||
missing = []
|
||
server_root = self.get('workspace', 'server_root', default='')
|
||
if not server_root:
|
||
missing.append("服务器根目录 (server_root)")
|
||
elif not Path(server_root).exists():
|
||
missing.append(f"服务器根目录不存在: {server_root}")
|
||
return missing
|
||
|
||
|
||
def apply_terminal_launch_config(config: Config) -> None:
|
||
"""Apply project terminal launch settings to the process environment."""
|
||
mode = (config.get('terminal', 'launch_mode', default='auto') or 'auto').strip().lower() or 'auto'
|
||
terminal_exe = (config.get('terminal', 'exe_path', default='') or '').strip()
|
||
|
||
os.environ['SERVER_MANAGER_TERMINAL'] = mode
|
||
if terminal_exe:
|
||
os.environ['SERVER_MANAGER_TERMINAL_EXE'] = terminal_exe
|
||
else:
|
||
os.environ.pop('SERVER_MANAGER_TERMINAL_EXE', None)
|
||
os.environ.pop('SERVER_MANAGER_TERMINAL_COMMAND', None)
|
||
os.environ.pop('SERVER_MANAGER_TERMINAL_CWD', None)
|