mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-16 01:40:15 +08:00
98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
# =============================================================================
|
|
# 新旧对比 - cli.py
|
|
# =============================================================================
|
|
#
|
|
# 【旧版 src/astrbot_sdk/cli/main.py】
|
|
# - 位于 cli/ 文件夹中
|
|
# - 有 setup_logger() 函数带 docstring
|
|
# - run 命令有 @click.pass_context 装饰器
|
|
# - plugins-dir 参数类型为 str
|
|
# - run 命令有 logger.info 输出
|
|
# - websocket 命令有 help 参数和 logger.info 输出
|
|
#
|
|
# 【新版 src-new/astrbot_sdk/cli.py】
|
|
# - 位于第一层，单文件
|
|
# - setup_logger() 无 docstring
|
|
# - run 命令无 @click.pass_context
|
|
# - plugins-dir 参数类型为 Path
|
|
# - 无日志输出
|
|
#
|
|
# =============================================================================
|
|
# TODO: 功能缺失
|
|
# =============================================================================
|
|
#
|
|
# 1. CLI 命令缺少 docstring
|
|
# - 旧版 cli() 有 """AstrBot SDK CLI""" docstring
|
|
# - 旧版 run() 有 """Start the plugin supervisor over stdio.""" docstring
|
|
# - 旧版 worker() 有 """Internal command used by the supervisor to start a worker.""" docstring
|
|
# - 旧版 websocket() 有 """Legacy websocket runtime entrypoint.""" docstring
|
|
# - 新版所有命令都缺少 docstring
|
|
#
|
|
# 2. 缺少日志输出
|
|
# - 旧版 run() 有 logger.info(f"Starting plugin supervisor with plugins dir: {plugins_dir}")
|
|
# - 旧版 websocket() 有 logger.info(f"Starting WebSocket server on port {port}...")
|
|
# - 新版无对应日志输出
|
|
#
|
|
# 3. websocket 命令缺少 help 参数
|
|
# - 旧版: @click.option("--port", default=8765, help="WebSocket server port", type=int)
|
|
# - 新版: @click.option("--port", default=8765, type=int)
|
|
#
|
|
# =============================================================================
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import click
|
|
from loguru import logger
|
|
|
|
from .runtime.bootstrap import run_plugin_worker, run_supervisor, run_websocket_server
|
|
|
|
|
|
def setup_logger(verbose: bool = False) -> None:
|
|
logger.remove()
|
|
logger.add(
|
|
sys.stderr,
|
|
format="<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <level>{message}</level>",
|
|
level="DEBUG" if verbose else "INFO",
|
|
colorize=True,
|
|
)
|
|
|
|
|
|
@click.group()
|
|
@click.option("-v", "--verbose", is_flag=True, help="Enable verbose output")
|
|
@click.pass_context
|
|
def cli(ctx, verbose: bool) -> None:
|
|
ctx.ensure_object(dict)
|
|
ctx.obj["verbose"] = verbose
|
|
setup_logger(verbose)
|
|
|
|
|
|
@cli.command()
|
|
@click.option(
|
|
"--plugins-dir",
|
|
default="plugins",
|
|
type=click.Path(file_okay=False, dir_okay=True, path_type=Path),
|
|
help="Directory containing plugin folders",
|
|
)
|
|
def run(plugins_dir: Path) -> None:
|
|
asyncio.run(run_supervisor(plugins_dir=plugins_dir))
|
|
|
|
|
|
@cli.command(hidden=True)
|
|
@click.option(
|
|
"--plugin-dir",
|
|
required=True,
|
|
type=click.Path(file_okay=False, dir_okay=True, path_type=Path),
|
|
)
|
|
def worker(plugin_dir: Path) -> None:
|
|
asyncio.run(run_plugin_worker(plugin_dir=plugin_dir))
|
|
|
|
|
|
@cli.command(hidden=True)
|
|
@click.option("--port", default=8765, type=int)
|
|
def websocket(port: int) -> None:
|
|
asyncio.run(run_websocket_server(port=port))
|