mirror of
https://github.com/zhamao-robot/zhamao-framework.git
synced 2026-03-18 21:24:52 +08:00
240 lines
8.7 KiB
PHP
240 lines
8.7 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace ZM\Command;
|
|
|
|
use Symfony\Component\Console\Attribute\AsCommand;
|
|
use Symfony\Component\Console\Input\InputOption;
|
|
use Symfony\Component\Console\Output\ConsoleSectionOutput;
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
use ZM\Exception\InitException;
|
|
use ZM\Utils\ZMUtil;
|
|
|
|
#[AsCommand(name: 'init', description: '初始化框架运行的基础文件')]
|
|
class InitCommand extends Command
|
|
{
|
|
use NonPharLoadModeOnly;
|
|
|
|
private string $base_path;
|
|
|
|
private bool $force = false;
|
|
|
|
protected function configure(): void
|
|
{
|
|
$this->setDefinition([
|
|
new InputOption('force', 'f', InputOption::VALUE_NONE, '覆盖现有文件'),
|
|
new InputOption('docker', null, InputOption::VALUE_NONE, '启用 Docker 支持'),
|
|
]);
|
|
$this->setHelp('提取框架的基础文件到当前目录，以便于快速开始开发。');
|
|
}
|
|
|
|
protected function handle(): int
|
|
{
|
|
$this->setBasePath();
|
|
$this->force = $this->input->getOption('force');
|
|
|
|
$this->section('提取框架基础文件', function (ConsoleSectionOutput $section) {
|
|
$this->extractFiles($this->getExtractFiles(), $section);
|
|
});
|
|
|
|
if (LOAD_MODE === LOAD_MODE_SRC) {
|
|
$this->section('应用自动加载配置', function (ConsoleSectionOutput $section) {
|
|
$autoload = [
|
|
'psr-4' => [
|
|
'Module\\' => 'src/Module',
|
|
'Custom\\' => 'src/Custom',
|
|
],
|
|
'files' => [
|
|
'src/Custom/global_function.php',
|
|
],
|
|
];
|
|
|
|
$section->write('<fg=gray>更新 composer.json ... </>');
|
|
|
|
if (!file_exists($this->base_path . '/composer.json')) {
|
|
throw new InitException('未找到 composer.json 文件', '请检查当前目录是否为项目根目录', 41);
|
|
}
|
|
|
|
try {
|
|
$composer = json_decode(file_get_contents($this->base_path . '/composer.json'), true, 512, JSON_THROW_ON_ERROR);
|
|
} catch (\JsonException $e) {
|
|
throw new InitException('解析 composer.json 文件失败', '请检查 composer.json 文件是否存在语法错误', 42, $e);
|
|
}
|
|
|
|
if (!isset($composer['autoload'])) {
|
|
$composer['autoload'] = $autoload;
|
|
} else {
|
|
$composer['autoload'] = array_merge_recursive($composer['autoload'], $autoload);
|
|
}
|
|
|
|
try {
|
|
file_put_contents($this->base_path . '/composer.json', json_encode($composer, JSON_THROW_ON_ERROR | JSON_PRETTY_PRINT | JSON_UNESCAPED_UNICODE | JSON_UNESCAPED_SLASHES));
|
|
} catch (\JsonException $e) {
|
|
throw new InitException('写入 composer.json 文件失败', '', 0, $e);
|
|
}
|
|
|
|
$section->writeln('<info>完成</info>');
|
|
|
|
$section->write('<fg=gray>执行 composer dump-autoload ... </>');
|
|
exec('composer dump-autoload');
|
|
|
|
$section->writeln('<info>完成</info>');
|
|
});
|
|
}
|
|
|
|
if ($this->input->getOption('docker')) {
|
|
$this->section('应用 Docker 支持', function (ConsoleSectionOutput $section) {
|
|
$files = $this->getFilesFromPatterns([
|
|
'/docker/*/Dockerfile',
|
|
'/docker/environment.env.example',
|
|
'/docker-compose.yml',
|
|
]);
|
|
$this->extractFiles($files, $section);
|
|
|
|
// 生成 env 文件
|
|
if ($this->shouldExtractFile('/docker/environment.env')) {
|
|
$section->write('<fg=gray>生成环境变量文件 ... </>');
|
|
$env = file_get_contents($this->base_path . '/docker/environment.env.example');
|
|
foreach ($this->getEnvVariables() as $key => $value) {
|
|
$env = $this->injectEnv($env, $key, $value);
|
|
}
|
|
file_put_contents($this->base_path . '/docker/environment.env', $env);
|
|
$section->writeln('<info>完成</info>');
|
|
} else {
|
|
$section->writeln('<fg=gray>生成环境变量文件 ... </><comment>跳过（已存在）</comment>');
|
|
}
|
|
});
|
|
}
|
|
|
|
// 将命令行入口标记为可执行
|
|
chmod($this->base_path . '/zhamao', 0755);
|
|
return self::SUCCESS;
|
|
}
|
|
|
|
private function getExtractFiles(): array
|
|
{
|
|
$patterns = [
|
|
'/zhamao',
|
|
'/.gitignore',
|
|
'/config/*',
|
|
'/src/Globals/*.php',
|
|
];
|
|
|
|
return $this->getFilesFromPatterns($patterns);
|
|
}
|
|
|
|
private function getFilesFromPatterns(array $patterns): array
|
|
{
|
|
$files = [];
|
|
foreach ($patterns as $pattern) {
|
|
// TODO: 优化代码，避免在循环中使用 array_merge 以减少资源消耗
|
|
$files = array_merge($files, glob($this->getVendorPath($pattern), GLOB_BRACE));
|
|
}
|
|
return array_map(fn ($file) => str_replace($this->getVendorPath(''), '', $file), $files);
|
|
}
|
|
|
|
/**
|
|
* 设置基准目录
|
|
*/
|
|
private function setBasePath(): void
|
|
{
|
|
$base_path = WORKING_DIR;
|
|
if (file_exists($base_path . '/vendor/autoload.php')) {
|
|
$this->base_path = $base_path;
|
|
} else {
|
|
$phar_link = new \Phar(__DIR__);
|
|
$current_dir = pathinfo($phar_link->getPath())['dirname'];
|
|
chdir($current_dir);
|
|
$phar_link = 'phar://' . $phar_link->getPath();
|
|
if (file_exists($phar_link . '/vendor/autoload.php')) {
|
|
$this->base_path = $current_dir;
|
|
} else {
|
|
throw new InitException('框架启动模式不是 Composer 模式，无法进行初始化', '如果您是从 Github 下载的框架，请参阅文档进行源码模式启动', 42);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 提取文件
|
|
*
|
|
* @param string $file 文件路径，相对于框架根目录
|
|
* @throws InitException 提取失败时抛出异常
|
|
*/
|
|
private function extractFile(string $file): void
|
|
{
|
|
$info = pathinfo($file);
|
|
// 确保目录存在
|
|
if (
|
|
!file_exists($this->base_path . $info['dirname'])
|
|
&& !mkdir($concurrent_dir = $this->base_path . $info['dirname'], 0777, true)
|
|
&& !is_dir($concurrent_dir)
|
|
) {
|
|
throw new InitException("无法创建目录 {$concurrent_dir}", '请检查目录权限');
|
|
}
|
|
|
|
if (copy($this->getVendorPath($file), $this->base_path . $file) === false) {
|
|
throw new InitException("无法复制文件 {$file}", '请检查目录权限');
|
|
}
|
|
}
|
|
|
|
private function shouldExtractFile(string $file): bool
|
|
{
|
|
return !file_exists($this->base_path . $file) || $this->force;
|
|
}
|
|
|
|
private function getVendorPath(string $file): string
|
|
{
|
|
try {
|
|
$package_name = ZMUtil::getComposerMetadata()['name'];
|
|
} catch (\JsonException) {
|
|
throw new InitException('无法读取框架包的 composer.json', '请检查框架包完整性，或者重新安装框架包');
|
|
}
|
|
return $this->base_path . '/vendor/' . $package_name . $file;
|
|
}
|
|
|
|
private function extractFiles(array $files, OutputInterface $output): void
|
|
{
|
|
foreach ($files as $file) {
|
|
$output->write("<fg=gray>提取 {$file} ... </>");
|
|
if ($this->shouldExtractFile($file)) {
|
|
try {
|
|
$this->extractFile($file);
|
|
$output->write('<info>完成</info>');
|
|
} catch (InitException $e) {
|
|
$output->write('<error>失败</error>');
|
|
throw $e;
|
|
} finally {
|
|
$output->writeln('');
|
|
}
|
|
} else {
|
|
$output->writeln('<comment>跳过（已存在）</comment>');
|
|
}
|
|
}
|
|
}
|
|
|
|
private function injectEnv(string $env, string $key, string $value): string
|
|
{
|
|
$pattern = "/^{$key}=.+$/m";
|
|
if (preg_match($pattern, $env)) {
|
|
return preg_replace($pattern, "{$key}={$value}", $env);
|
|
}
|
|
|
|
return $env . PHP_EOL . "{$key}={$value}";
|
|
}
|
|
|
|
private function getEnvVariables(): array
|
|
{
|
|
return [
|
|
'REDIS_PASSWORD' => bin2hex(random_bytes(8)),
|
|
|
|
'POSTGRES_USER' => 'root',
|
|
'POSTGRES_PASSWORD' => bin2hex(random_bytes(8)),
|
|
|
|
'POSTGRES_APPLICATION_DATABASE' => 'zhamao',
|
|
'POSTGRES_APPLICATION_USER' => 'zhamao',
|
|
'POSTGRES_APPLICATION_USER_PASSWORD' => bin2hex(random_bytes(8)),
|
|
];
|
|
}
|
|
}
|