174 lines
5.8 KiB
Python
174 lines
5.8 KiB
Python
"""CLI 入口"""
|
||
import click
|
||
import signal
|
||
import os
|
||
import sys
|
||
|
||
from oss import __version__
|
||
from oss.logger.logger import Logger
|
||
from oss.plugin.manager import PluginManager
|
||
from oss.config import init_config, get_config
|
||
|
||
|
||
# 深度隐藏的成就系统导入
|
||
try:
|
||
from oss.core.achievements import init_achievements, get_validator, _cmd_echo, _cmd_help_internal, _cmd_list_all, _cmd_stats, _cmd_reset_progress, _cmd_export, _cmd_import, _cmd_verify, _cmd_debug, _cmd_info
|
||
_ACHIEVEMENTS_ENABLED = True
|
||
except ImportError:
|
||
_ACHIEVEMENTS_ENABLED = False
|
||
|
||
|
||
@click.group()
|
||
@click.option('--config', '-c', type=str, help='配置文件路径')
|
||
@click.pass_context
|
||
def cli(ctx, config):
|
||
"""NebulaShell - 一切皆为插件"""
|
||
# 初始化配置
|
||
ctx.ensure_object(dict)
|
||
ctx.obj['config'] = init_config(config)
|
||
|
||
# 初始化成就系统（如果启用）
|
||
if _ACHIEVEMENTS_ENABLED:
|
||
try:
|
||
init_achievements()
|
||
except Exception:
|
||
pass # 静默失败，不影响主功能
|
||
|
||
|
||
@cli.command()
|
||
@click.option('--host', type=str, default=None, help='监听地址')
|
||
@click.option('--port', type=int, default=None, help='HTTP API 端口')
|
||
@click.option('--tcp-port', type=int, default=None, help='HTTP TCP 端口')
|
||
@click.pass_context
|
||
def serve(ctx, host, port, tcp_port):
|
||
"""启动 NebulaShell 服务端"""
|
||
config = ctx.obj.get('config', get_config())
|
||
|
||
# 命令行参数覆盖配置
|
||
if host:
|
||
config.set('HOST', host)
|
||
if port:
|
||
config.set('HTTP_API_PORT', port)
|
||
if tcp_port:
|
||
config.set('HTTP_TCP_PORT', tcp_port)
|
||
|
||
log = Logger()
|
||
log.info(f"NebulaShell {__version__} 启动")
|
||
log.info(f"监听地址：{config.host}:{config.http_api_port}")
|
||
log.info(f"数据目录：{config.data_dir.absolute()}")
|
||
log.info(f"插件仓库：{config.store_dir.absolute()}")
|
||
|
||
plugin_mgr = PluginManager()
|
||
plugin_mgr.load()
|
||
plugin_mgr.start()
|
||
|
||
log.info("就绪")
|
||
|
||
def shutdown(sig, frame):
|
||
log.info("停止中...")
|
||
plugin_mgr.stop()
|
||
log.info("已停止")
|
||
raise SystemExit(0)
|
||
|
||
signal.signal(signal.SIGINT, shutdown)
|
||
signal.signal(signal.SIGTERM, shutdown)
|
||
|
||
try:
|
||
signal.pause()
|
||
except AttributeError:
|
||
import time
|
||
while True:
|
||
time.sleep(1)
|
||
|
||
|
||
@cli.command()
|
||
def version():
|
||
"""显示版本"""
|
||
click.echo(f"NebulaShell {__version__}")
|
||
|
||
|
||
@cli.command()
|
||
@click.pass_context
|
||
def info(ctx):
|
||
"""显示系统信息"""
|
||
import random
|
||
|
||
config = ctx.obj.get('config', get_config())
|
||
click.echo(f"NebulaShell {__version__}")
|
||
click.echo(f"配置文件：{config._config_file or '无'}")
|
||
click.echo(f"HTTP API 端口：{config.http_api_port}")
|
||
click.echo(f"HTTP TCP 端口：{config.http_tcp_port}")
|
||
click.echo(f"主机地址：{config.host}")
|
||
click.echo(f"数据目录：{config.data_dir.absolute()}")
|
||
click.echo(f"插件仓库：{config.store_dir.absolute()}")
|
||
click.echo(f"日志级别：{config.log_level}")
|
||
click.echo(f"权限检查：{'启用' if config.permission_check else '禁用'}")
|
||
|
||
# 小提示：有些功能需要特殊方式访问 (4675747572654f53)
|
||
# 随机显示隐藏提示（约 10% 概率）
|
||
click.echo("")
|
||
if random.random() < 0.1: # 10% 概率显示彩蛋提示
|
||
click.echo("✨ 奇怪的提示：试试在命令前加两个感叹号会怎样？比如 !!help")
|
||
elif random.random() < 0.05: # 额外 5% 概率显示另一种提示
|
||
click.echo("🤔 听说有人用 !! 开头的命令发现了不得了的东西...")
|
||
|
||
|
||
@cli.command(name="cli")
|
||
@click.option('--connect-host', default='127.0.0.1', help='后端地址（默认 127.0.0.1）')
|
||
@click.option('--connect-port', default=8080, help='后端端口（默认 8080）')
|
||
def cli_command(connect_host, connect_port):
|
||
"""启动 TUI 前端（前后端分离，连接已有后端）"""
|
||
click.echo("NebulaShell TUI 客户端（待实现）")
|
||
click.echo(f"目标后端：{connect_host}:{connect_port}")
|
||
|
||
|
||
def main():
|
||
# 检测是否通过已弃用的 oss 命令调用
|
||
cmd_name = os.path.basename(sys.argv[0])
|
||
if cmd_name in ("oss", "oss.exe"):
|
||
print("╔══════════════════════════════════════════╗")
|
||
print("║ ⚠ oss 命令已弃用，请使用 nebula 替代 ║")
|
||
print("║ 例如: nebula serve ║")
|
||
print("║ nebula info ║")
|
||
print("║ nebula version ║")
|
||
print("╚══════════════════════════════════════════╝")
|
||
sys.exit(1)
|
||
|
||
# 检查隐藏命令前缀
|
||
if len(sys.argv) > 1 and sys.argv[1].startswith("!!"):
|
||
if _ACHIEVEMENTS_ENABLED:
|
||
cmd = sys.argv[1][2:] # 去掉 !! 前缀
|
||
args = sys.argv[2:]
|
||
|
||
# 映射隐藏命令
|
||
cmd_map = {
|
||
"echo": _cmd_echo,
|
||
"help": _cmd_help_internal,
|
||
"list": _cmd_list_all,
|
||
"stats": _cmd_stats,
|
||
"reset": _cmd_reset_progress,
|
||
"export": _cmd_export,
|
||
"import": _cmd_import,
|
||
"verify": _cmd_verify,
|
||
"debug": _cmd_debug,
|
||
"info": _cmd_info,
|
||
}
|
||
|
||
if cmd in cmd_map:
|
||
validator = get_validator()
|
||
validator.use_hidden_command(cmd)
|
||
cmd_map[cmd](args)
|
||
return
|
||
else:
|
||
print(f"未知命令：!!{cmd}")
|
||
return
|
||
else:
|
||
print("成就系统未启用")
|
||
return
|
||
|
||
cli()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|