323 lines
9.9 KiB
PHP
323 lines
9.9 KiB
PHP
|
<?php
|
|||
|
|
declare(strict_types=1);
|
|||
|
|
/**
|
|||
|
|
*+------------------
|
|||
|
|
* madong
|
|||
|
|
*+------------------
|
|||
|
|
* Copyright (c) https://gitee.com/motion-code All rights reserved.
|
|||
|
|
*+------------------
|
|||
|
|
* Author: Mr. April (405784684@qq.com)
|
|||
|
|
*+------------------
|
|||
|
|
* Official Website: http://www.madong.tech
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
namespace app\command;
|
|||
|
|
|
|||
|
|
use Symfony\Component\Console\Command\Command;
|
|||
|
|
use Symfony\Component\Console\Style\SymfonyStyle;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 命令基类
|
|||
|
|
*
|
|||
|
|
* 提供通用的工具方法供所有命令使用
|
|||
|
|
* - SSE 事件处理：适用于需要流式输出的命令（如插件安装/卸载）
|
|||
|
|
* - 通用工具方法：目录操作、确认、输出等
|
|||
|
|
*
|
|||
|
|
* @author Mr.April
|
|||
|
|
* @since 1.0.0
|
|||
|
|
*/
|
|||
|
|
abstract class BaseCommand extends Command
|
|||
|
|
{
|
|||
|
|
/**
|
|||
|
|
* 解析SSE事件
|
|||
|
|
*
|
|||
|
|
* @param string $event SSE事件字符串
|
|||
|
|
* @return array|null 解析后的事件数据
|
|||
|
|
*/
|
|||
|
|
protected function parseSseEvent(string $event): ?array
|
|||
|
|
{
|
|||
|
|
// 提取所有data行的内容
|
|||
|
|
if (preg_match_all('/^data: (.+)$/m', $event, $matches)) {
|
|||
|
|
// 合并所有data行
|
|||
|
|
$data = implode("\n", $matches[1]);
|
|||
|
|
return json_decode($data, true);
|
|||
|
|
}
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理SSE事件输出
|
|||
|
|
*
|
|||
|
|
* @param string $event SSE事件字符串
|
|||
|
|
* @param SymfonyStyle $io SymfonyStyle 实例
|
|||
|
|
* @param string|null $pluginName 插件名称（可选）
|
|||
|
|
* @return int|null 返回状态码，如果需要提前退出
|
|||
|
|
*/
|
|||
|
|
protected function handleSseEvent(string $event, SymfonyStyle $io, ?string $pluginName = null): ?int
|
|||
|
|
{
|
|||
|
|
try {
|
|||
|
|
// 直接输出原始事件内容，以便调试
|
|||
|
|
if (strpos($event, 'data:') !== false) {
|
|||
|
|
// 提取data部分
|
|||
|
|
$lines = explode("\n", $event);
|
|||
|
|
foreach ($lines as $line) {
|
|||
|
|
if (strpos($line, 'data:') === 0) {
|
|||
|
|
$data = substr($line, 5);
|
|||
|
|
if (!empty(trim($data))) {
|
|||
|
|
// 尝试解析JSON
|
|||
|
|
$jsonData = json_decode($data, true);
|
|||
|
|
if ($jsonData) {
|
|||
|
|
$eventType = $jsonData['event'] ?? '';
|
|||
|
|
$data = $jsonData['data'] ?? [];
|
|||
|
|
|
|||
|
|
switch ($eventType) {
|
|||
|
|
case 'progress':
|
|||
|
|
$progress = $data['progress'] ?? 0;
|
|||
|
|
$message = $data['message'] ?? '';
|
|||
|
|
$io->text(sprintf("[%d%%] %s", $progress, $message));
|
|||
|
|
break;
|
|||
|
|
|
|||
|
|
case 'completed':
|
|||
|
|
$message = $data['message'] ?? 'Operation completed successfully';
|
|||
|
|
if ($pluginName) {
|
|||
|
|
$io->success("Plugin '{$pluginName}' {$message}");
|
|||
|
|
} else {
|
|||
|
|
$io->success($message);
|
|||
|
|
}
|
|||
|
|
return Command::SUCCESS;
|
|||
|
|
|
|||
|
|
case 'error':
|
|||
|
|
$io->error($data['message'] ?? 'Operation failed');
|
|||
|
|
return Command::FAILURE;
|
|||
|
|
|
|||
|
|
case 'warning':
|
|||
|
|
$io->warning($data['message'] ?? 'Warning');
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 不是JSON，直接输出
|
|||
|
|
$io->text(trim($data));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 不是SSE事件，直接输出
|
|||
|
|
$io->text(trim($event));
|
|||
|
|
}
|
|||
|
|
} catch (Exception $e) {
|
|||
|
|
// 忽略错误，继续执行
|
|||
|
|
$io->error('处理SSE事件时出错: ' . $e->getMessage());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 执行流式操作（SSE）
|
|||
|
|
*
|
|||
|
|
* @param iterable $generator 迭代器
|
|||
|
|
* @param SymfonyStyle $io SymfonyStyle 实例
|
|||
|
|
* @param string|null $pluginName 插件名称
|
|||
|
|
* @return int 命令退出码
|
|||
|
|
*/
|
|||
|
|
protected function executeStream(iterable $generator, SymfonyStyle $io, ?string $pluginName = null): int
|
|||
|
|
{
|
|||
|
|
foreach ($generator as $event) {
|
|||
|
|
$result = $this->handleSseEvent($event, $io, $pluginName);
|
|||
|
|
if ($result !== null) {
|
|||
|
|
return $result;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return Command::SUCCESS;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 递归删除目录
|
|||
|
|
*
|
|||
|
|
* @param string $directory 目录路径
|
|||
|
|
* @return bool 删除是否成功
|
|||
|
|
*/
|
|||
|
|
protected function deleteDirectory(string $directory): bool
|
|||
|
|
{
|
|||
|
|
if (!is_dir($directory)) {
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$files = scandir($directory);
|
|||
|
|
foreach ($files as $file) {
|
|||
|
|
if ($file === '.' || $file === '..') {
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$path = $directory . '/' . $file;
|
|||
|
|
if (is_dir($path)) {
|
|||
|
|
if (!$this->deleteDirectory($path)) {
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
if (!unlink($path)) {
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return rmdir($directory);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 确认操作
|
|||
|
|
*
|
|||
|
|
* @param SymfonyStyle $io SymfonyStyle 实例
|
|||
|
|
* @param string $message 确认消息
|
|||
|
|
* @param bool $default 默认值
|
|||
|
|
* @return bool 是否确认
|
|||
|
|
*/
|
|||
|
|
protected function confirm(SymfonyStyle $io, string $message, bool $default = false): bool
|
|||
|
|
{
|
|||
|
|
return $io->confirm($message, $default);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 密码输入（带掩码显示）
|
|||
|
|
*
|
|||
|
|
* @param SymfonyStyle $io SymfonyStyle 实例
|
|||
|
|
* @param string $message 提示消息
|
|||
|
|
* @param string $mask 掩码字符，默认 *
|
|||
|
|
* @return string 输入的密码
|
|||
|
|
*/
|
|||
|
|
protected function askPassword(SymfonyStyle $io, string $message, string $mask = '*'): string
|
|||
|
|
{
|
|||
|
|
$io->text($message);
|
|||
|
|
|
|||
|
|
$password = '';
|
|||
|
|
while (true) {
|
|||
|
|
$char = $this->readCharacter();
|
|||
|
|
|
|||
|
|
if ($char === "\n" || $char === "\r") {
|
|||
|
|
$io->newLine();
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if ($char === "\x7f" || $char === "\b") { // Backspace
|
|||
|
|
if (strlen($password) > 0) {
|
|||
|
|
$password = substr($password, 0, -1);
|
|||
|
|
$io->write("\b \b");
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
$password .= $char;
|
|||
|
|
$io->write($mask);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return $password;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 读取单个字符（支持 Windows/Linux）
|
|||
|
|
*/
|
|||
|
|
private function readCharacter(): string
|
|||
|
|
{
|
|||
|
|
// Windows 环境
|
|||
|
|
if (DIRECTORY_SEPARATOR === '\\') {
|
|||
|
|
return $this->readCharacterWindows();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Linux/Mac 环境
|
|||
|
|
$tty = '/dev/tty';
|
|||
|
|
if (!is_file($tty) || !is_readable($tty)) {
|
|||
|
|
// 回退到标准输入
|
|||
|
|
return fgets(STDIN) ?: '';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$ttySpec = trim(`stty -g`);
|
|||
|
|
system('stty -icanon -echo');
|
|||
|
|
$char = fgetc(fopen($tty, 'r'));
|
|||
|
|
system("stty {$ttySpec}");
|
|||
|
|
|
|||
|
|
return $char ?: '';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* Windows 环境下读取单个字符
|
|||
|
|
*/
|
|||
|
|
private function readCharacterWindows(): string
|
|||
|
|
{
|
|||
|
|
$input = stream_get_contents(STDIN, 1);
|
|||
|
|
|
|||
|
|
// 处理 Windows 特殊键
|
|||
|
|
if ($input === "\xe0" || $input === "\0") {
|
|||
|
|
// 功能键前缀，跳过下一个字节
|
|||
|
|
stream_get_contents(STDIN, 1);
|
|||
|
|
return "\x7f"; // 作为退格键处理
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return $input;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 确认密码输入
|
|||
|
|
*
|
|||
|
|
* @param SymfonyStyle $io SymfonyStyle 实例
|
|||
|
|
* @param string $message 提示消息
|
|||
|
|
* @param int $minLength 最小长度
|
|||
|
|
* @return string 确认后的密码
|
|||
|
|
*/
|
|||
|
|
protected function askPasswordWithConfirm(SymfonyStyle $io, string $message, int $minLength = 6): string
|
|||
|
|
{
|
|||
|
|
$io->text($message);
|
|||
|
|
|
|||
|
|
while (true) {
|
|||
|
|
$password = $this->askPassword($io, '请输入密码: ');
|
|||
|
|
|
|||
|
|
if (strlen($password) < $minLength) {
|
|||
|
|
$io->warning("密码长度不能少于 {$minLength} 位");
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 确认密码
|
|||
|
|
$confirm = $this->askPassword($io, '请再次输入密码: ');
|
|||
|
|
|
|||
|
|
if ($password !== $confirm) {
|
|||
|
|
$io->error('两次输入的密码不一致，请重新输入');
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return $password;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 输出错误信息并记录
|
|||
|
|
*
|
|||
|
|
* @param SymfonyStyle $io SymfonyStyle 实例
|
|||
|
|
* @param string $message 错误消息
|
|||
|
|
* @param \Throwable|null $exception 异常对象
|
|||
|
|
* @return int 返回失败状态码
|
|||
|
|
*/
|
|||
|
|
protected function outputError(SymfonyStyle $io, string $message, ?\Throwable $exception = null): int
|
|||
|
|
{
|
|||
|
|
$io->error($message);
|
|||
|
|
if ($exception) {
|
|||
|
|
$io->error(sprintf("File: %s Line: %d", $exception->getFile(), $exception->getLine()));
|
|||
|
|
if ($io->isVerbose()) {
|
|||
|
|
$io->text($exception->getTraceAsString());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return Command::FAILURE;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 输出成功信息
|
|||
|
|
*
|
|||
|
|
* @param SymfonyStyle $io SymfonyStyle 实例
|
|||
|
|
* @param string $message 成功消息
|
|||
|
|
* @return int 返回成功状态码
|
|||
|
|
*/
|
|||
|
|
protected function outputSuccess(SymfonyStyle $io, string $message): int
|
|||
|
|
{
|
|||
|
|
$io->success($message);
|
|||
|
|
return Command::SUCCESS;
|
|||
|
|
}
|
|||
|
|
}
|