mirror of
https://github.com/dnslin/aria2bot.git
synced 2026-01-12 04:22:21 +08:00
feat: init project
This commit is contained in:
dnslin
212
src/telegram/handlers.py
Normal file
212
src/telegram/handlers.py
Normal file
@@ -0,0 +1,212 @@
|
||||
"""Telegram bot command handlers."""
|
||||
from __future__ import annotations
|
||||
|
||||
from telegram import Update
|
||||
from telegram.ext import ContextTypes, CommandHandler
|
||||
|
||||
from src.core import (
|
||||
Aria2Config,
|
||||
Aria2Error,
|
||||
NotInstalledError,
|
||||
ServiceError,
|
||||
DownloadError,
|
||||
ConfigError,
|
||||
is_aria2_installed,
|
||||
get_aria2_version,
|
||||
ARIA2_CONF,
|
||||
)
|
||||
from src.aria2 import Aria2Installer, Aria2ServiceManager
|
||||
|
||||
|
||||
class Aria2BotAPI:
|
||||
def __init__(self, config: Aria2Config | None = None):
|
||||
self.config = config or Aria2Config()
|
||||
self.installer = Aria2Installer(self.config)
|
||||
self.service = Aria2ServiceManager()
|
||||
|
||||
async def _reply(self, update: Update, context: ContextTypes.DEFAULT_TYPE, text: str, **kwargs):
|
||||
if update.effective_message:
|
||||
return await update.effective_message.reply_text(text, **kwargs)
|
||||
if update.effective_chat:
|
||||
return await context.bot.send_message(chat_id=update.effective_chat.id, text=text, **kwargs)
|
||||
return None
|
||||
|
||||
def _get_rpc_secret(self) -> str:
|
||||
if self.config.rpc_secret:
|
||||
return self.config.rpc_secret
|
||||
if ARIA2_CONF.exists():
|
||||
try:
|
||||
for line in ARIA2_CONF.read_text(encoding="utf-8", errors="ignore").splitlines():
|
||||
stripped = line.strip()
|
||||
if stripped.startswith("rpc-secret="):
|
||||
secret = stripped.split("=", 1)[1].strip()
|
||||
if secret:
|
||||
self.config.rpc_secret = secret
|
||||
return secret
|
||||
except OSError:
|
||||
return ""
|
||||
return ""
|
||||
|
||||
def _get_rpc_port(self) -> int | None:
|
||||
if ARIA2_CONF.exists():
|
||||
try:
|
||||
for line in ARIA2_CONF.read_text(encoding="utf-8", errors="ignore").splitlines():
|
||||
stripped = line.strip()
|
||||
if stripped.startswith("rpc-listen-port="):
|
||||
port_str = stripped.split("=", 1)[1].strip()
|
||||
if port_str.isdigit():
|
||||
return int(port_str)
|
||||
except OSError:
|
||||
return None
|
||||
return self.config.rpc_port
|
||||
|
||||
async def install(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
await self._reply(update, context, "正在安装 aria2，处理中，请稍候...")
|
||||
try:
|
||||
result = await self.installer.install()
|
||||
version = get_aria2_version() or result.get("version") or "未知"
|
||||
rpc_secret = self._get_rpc_secret() or "未设置"
|
||||
rpc_port = self._get_rpc_port() or self.config.rpc_port
|
||||
await self._reply(
|
||||
update,
|
||||
context,
|
||||
"\n".join(
|
||||
[
|
||||
"安装完成 ✅",
|
||||
f"版本：{version}",
|
||||
f"二进制：{result.get('binary')}",
|
||||
f"配置目录：{result.get('config_dir')}",
|
||||
f"配置文件：{result.get('config')}",
|
||||
f"RPC 端口：{rpc_port}",
|
||||
f"RPC 密钥：{rpc_secret}",
|
||||
]
|
||||
),
|
||||
)
|
||||
except (DownloadError, ConfigError, Aria2Error) as exc:
|
||||
await self._reply(update, context, f"安装失败：{exc}")
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"安装失败，发生未知错误：{exc}")
|
||||
|
||||
async def uninstall(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
await self._reply(update, context, "正在卸载 aria2，处理中，请稍候...")
|
||||
try:
|
||||
try:
|
||||
self.service.stop()
|
||||
except ServiceError:
|
||||
pass
|
||||
self.installer.uninstall()
|
||||
await self._reply(update, context, "卸载完成 ✅")
|
||||
except Aria2Error as exc:
|
||||
await self._reply(update, context, f"卸载失败：{exc}")
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"卸载失败，发生未知错误：{exc}")
|
||||
|
||||
async def start_service(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
try:
|
||||
if not is_aria2_installed():
|
||||
await self._reply(update, context, "aria2 未安装，请先运行 /install")
|
||||
return
|
||||
self.service.start()
|
||||
await self._reply(update, context, "aria2 服务已启动 ✅")
|
||||
except NotInstalledError:
|
||||
await self._reply(update, context, "aria2 未安装，请先运行 /install")
|
||||
except ServiceError as exc:
|
||||
await self._reply(update, context, f"启动失败：{exc}")
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"启动失败，发生未知错误：{exc}")
|
||||
|
||||
async def stop_service(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
try:
|
||||
self.service.stop()
|
||||
await self._reply(update, context, "aria2 服务已停止 ✅")
|
||||
except ServiceError as exc:
|
||||
await self._reply(update, context, f"停止失败：{exc}")
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"停止失败，发生未知错误：{exc}")
|
||||
|
||||
async def restart_service(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
try:
|
||||
self.service.restart()
|
||||
await self._reply(update, context, "aria2 服务已重启 ✅")
|
||||
except ServiceError as exc:
|
||||
await self._reply(update, context, f"重启失败：{exc}")
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"重启失败，发生未知错误：{exc}")
|
||||
|
||||
async def status(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
try:
|
||||
info = self.service.status()
|
||||
version = get_aria2_version() or "未知"
|
||||
rpc_secret = self._get_rpc_secret() or "未设置"
|
||||
rpc_port = self._get_rpc_port() or self.config.rpc_port or "未知"
|
||||
except ServiceError as exc:
|
||||
await self._reply(update, context, f"获取状态失败：{exc}")
|
||||
return
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"获取状态失败，发生未知错误：{exc}")
|
||||
return
|
||||
|
||||
text = (
|
||||
"*Aria2 状态*\n"
|
||||
f"- 安装状态：{'已安装 ✅' if info.get('installed') or is_aria2_installed() else '未安装 ❌'}\n"
|
||||
f"- 运行状态：{'运行中 ✅' if info.get('running') else '未运行 ❌'}\n"
|
||||
f"- PID：`{info.get('pid') or 'N/A'}`\n"
|
||||
f"- 版本：`{version}`\n"
|
||||
f"- RPC 端口：`{rpc_port}`\n"
|
||||
f"- RPC 密钥：`{rpc_secret}`"
|
||||
)
|
||||
await self._reply(update, context, text, parse_mode="Markdown")
|
||||
|
||||
async def view_logs(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
try:
|
||||
logs = self.service.view_log(lines=30)
|
||||
except ServiceError as exc:
|
||||
await self._reply(update, context, f"读取日志失败：{exc}")
|
||||
return
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"读取日志失败，发生未知错误：{exc}")
|
||||
return
|
||||
|
||||
if not logs.strip():
|
||||
await self._reply(update, context, "暂无日志内容。")
|
||||
return
|
||||
|
||||
await self._reply(update, context, f"最近 30 行日志：\n{logs}")
|
||||
|
||||
async def clear_logs(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
try:
|
||||
self.service.clear_log()
|
||||
await self._reply(update, context, "日志已清空 ✅")
|
||||
except ServiceError as exc:
|
||||
await self._reply(update, context, f"清空日志失败：{exc}")
|
||||
except Exception as exc: # noqa: BLE001
|
||||
await self._reply(update, context, f"清空日志失败，发生未知错误：{exc}")
|
||||
|
||||
async def help_command(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
commands = [
|
||||
"/install - 安装 aria2",
|
||||
"/uninstall - 卸载 aria2",
|
||||
"/start - 启动 aria2 服务",
|
||||
"/stop - 停止 aria2 服务",
|
||||
"/restart - 重启 aria2 服务",
|
||||
"/status - 查看 aria2 状态",
|
||||
"/logs - 查看最近日志",
|
||||
"/clear_logs - 清空日志",
|
||||
"/help - 显示此帮助",
|
||||
]
|
||||
await self._reply(update, context, "可用命令：\n" + "\n".join(commands))
|
||||
|
||||
|
||||
def build_handlers(api: Aria2BotAPI) -> list[CommandHandler]:
|
||||
"""构建 CommandHandler 列表"""
|
||||
return [
|
||||
CommandHandler("install", api.install),
|
||||
CommandHandler("uninstall", api.uninstall),
|
||||
CommandHandler("start", api.start_service),
|
||||
CommandHandler("stop", api.stop_service),
|
||||
CommandHandler("restart", api.restart_service),
|
||||
CommandHandler("status", api.status),
|
||||
CommandHandler("logs", api.view_logs),
|
||||
CommandHandler("clear_logs", api.clear_logs),
|
||||
CommandHandler("help", api.help_command),
|
||||
]
|
||||
Reference in New Issue
Block a user