Files
MrleeAdmin/backend/app/command/make/DaoCommand.php
T

118 lines
3.5 KiB
PHP
Raw Normal View History

2026-08-31 11:33:59 +08:00
<?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\make;
use app\command\BaseCommand;
use Symfony\Component\Console\Attribute\AsCommand;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\Style\SymfonyStyle;
/**
* 生成Dao命令
* 使用方法:
* - 生成Dao:php webman make:dao TestDao
* - 生成带完整路径的Dao:php webman make:dao system/TestDao
*
* @author Mr.April
* @since 1.0.0
*/
#[AsCommand(
name: 'madong-make:dao',
description: 'Generate a new DAO class',
aliases: ['madong-make:dao'],
hidden: false
)]
class DaoCommand extends BaseCommand
{
/**
* 配置命令
*/
protected function configure(): void
{
$this->addArgument('name', InputArgument::REQUIRED, 'DAO name');
}
/**
* 执行命令
*/
public function __invoke(InputInterface $input, OutputInterface $output): int
{
$io = new SymfonyStyle($input, $output);
$io->title('DAO Generator');
$name = $input->getArgument('name');
$io->info("Generating DAO: {$name}");
// 处理路径和命名空间
$name = str_replace('\\', '/', $name);
$parts = explode('/', $name);
$className = ucfirst(array_pop($parts));
// 构建文件路径和命名空间
if (empty($parts)) {
// 如果没有提供路径,默认创建到app/dao目录
$file = app_path() . DIRECTORY_SEPARATOR . 'dao' . DIRECTORY_SEPARATOR . "{$className}.php";
$namespace = 'app\dao';
} else {
// 使用用户提供的完整路径
$relativePath = implode(DIRECTORY_SEPARATOR, $parts);
$file = app_path() . DIRECTORY_SEPARATOR . 'dao' . DIRECTORY_SEPARATOR . $relativePath . DIRECTORY_SEPARATOR . "{$className}.php";
$namespace = 'app\dao\\' . str_replace('/', '\\', $relativePath);
}
if (is_file($file)) {
if (!$this->confirm($io, "$file already exists. Do you want to override it?")) {
$io->warning('Operation cancelled.');
return Command::SUCCESS;
}
}
// 提取模型名称
$modelName = str_replace('Dao', '', $className);
if ($modelName === $className) {
$modelName .= 'Model';
}
$this->createDao($className, $namespace, $modelName, $file);
$io->success("DAO {$className} generated successfully!");
return Command::SUCCESS;
}
/**
* 创建Dao文件
*/
protected function createDao(string $className, string $namespace, string $modelName, string $file): void
{
$path = pathinfo($file, PATHINFO_DIRNAME);
if (!is_dir($path)) {
mkdir($path, 0777, true);
}
// 读取模板文件
$template = file_get_contents(__DIR__ . '/stubs/dao.stub');
// 替换占位符
$content = str_replace(
['DummyNamespace', 'DummyClass', 'DummyModel'],
[$namespace, $className, $modelName],
$template
);
file_put_contents($file, $content);
}
}