This commit is contained in:
2023-10-07 20:40:59 +08:00
parent 38ae3d2031
commit ce9d172016
+279 -279
View File
@@ -21,320 +21,320 @@ use Symfony\Component\Console\Input\InputInterface;
*/ */
class Gii class Gii
{ {
private ?string $tableName = NULL; private ?string $tableName = NULL;
/** @var null|Connection */ /** @var null|Connection */
private ?Connection $db; private ?Connection $db;
private InputInterface $input; private InputInterface $input;
public string $modelPath = APP_PATH . 'app/Model/'; public string $modelPath = APP_PATH . 'app/Model/';
public string $modelNamespace = 'App\Model\\'; public string $modelNamespace = 'App\Model\\';
public string $controllerPath = APP_PATH . 'app/Controller/'; public string $controllerPath = APP_PATH . 'app/Controller/';
public string $controllerNamespace = 'App\\Controller\\'; public string $controllerNamespace = 'App\\Controller\\';
public static array $createSqls = []; public static array $createSqls = [];
public array $keyword = [ public array $keyword = [
'ADD', 'ALL', 'ALTER', 'AND', 'AS', 'ASC', 'ASENSITIVE', 'BEFORE', 'BETWEEN', 'BIGINT', 'BINARY', 'BLOB', 'BOTH', 'BY', 'CALL', 'CASCADE', 'CASE', 'CHANGE', 'CHAR', 'CHARACTER', 'CHECK', 'COLLATE', 'COLUMN', 'CONDITION', 'CONNECTION', 'CONSTRAINT', 'CONTINUE', 'CONVERT', 'CREATE', 'CROSS', 'CURRENT_DATE', 'CURRENT_TIME', 'CURRENT_TIMESTAMP', 'CURRENT_USER', 'CURSOR', 'DATABASE', 'DATABASES', 'DAY_HOUR', 'DAY_MICROSECOND', 'DAY_MINUTE', 'DAY_SECOND', 'DEC', 'DECIMAL', 'DECLARE', 'DEFAULT', 'DELAYED', 'DELETE', 'DESC', 'DESCRIBE', 'DETERMINISTIC', 'DISTINCT', 'DISTINCTROW', 'DIV', 'DOUBLE', 'DROP', 'DUAL', 'EACH', 'ELSE', 'ELSEIF', 'ENCLOSED', 'ESCAPED', 'EXISTS', 'EXIT', 'EXPLAIN', 'FALSE', 'FETCH', 'FLOAT', 'FLOAT4', 'FLOAT8', 'FOR', 'FORCE', 'FOREIGN', 'FROM', 'FULLTEXT', 'GOTO', 'GRANT', 'GROUP', 'HAVING', 'HIGH_PRIORITY', 'HOUR_MICROSECOND', 'HOUR_MINUTE', 'HOUR_SECOND', 'IF', 'IGNORE', 'IN', 'INDEX', 'INFILE', 'INNER', 'INOUT', 'INSENSITIVE', 'INSERT', 'INT', 'INT1', 'INT2', 'INT3', 'INT4', 'INT8', 'INTEGER', 'INTERVAL', 'INTO', 'IS', 'ITERATE', 'JOIN', 'KEY', 'KEYS', 'KILL', 'LABEL', 'LEADING', 'LEAVE', 'LEFT', 'LIKE', 'LIMIT', 'LINEAR', 'LINES', 'LOAD', 'LOCALTIME', 'LOCALTIMESTAMP', 'LOCK', 'LONG', 'LONGBLOB', 'LONGTEXT', 'LOOP', 'LOW_PRIORITY', 'MATCH', 'MEDIUMBLOB', 'MEDIUMINT', 'MEDIUMTEXT', 'MIDDLEINT', 'MINUTE_MICROSECOND', 'MINUTE_SECOND', 'MOD', 'MODIFIES', 'NATURAL', 'NOT', 'NO_WRITE_TO_BINLOG', 'NULL', 'NUMERIC', 'ON', 'OPTIMIZE', 'OPTION', 'OPTIONALLY', 'OR', 'ORDER', 'OUT', 'OUTER', 'OUTFILE', 'PRECISION', 'PRIMARY', 'PROCEDURE', 'PURGE', 'RAID0', 'RANGE', 'READ', 'READS', 'REAL', 'REFERENCES', 'REGEXP', 'RELEASE', 'RENAME', 'REPEAT', 'REPLACE', 'REQUIRE', 'RESTRICT', 'RETURN', 'REVOKE', 'RIGHT', 'RLIKE', 'SCHEMA', 'SCHEMAS', 'SECOND_MICROSECOND', 'SELECT', 'SENSITIVE', 'SEPARATOR', 'SET', 'SHOW', 'SMALLINT', 'SPATIAL', 'SPECIFIC', 'SQL', 'SQLEXCEPTION', 'SQLSTATE', 'SQLWARNING', 'SQL_BIG_RESULT', 'SQL_CALC_FOUND_ROWS', 'SQL_SMALL_RESULT', 'SSL', 'STARTING', 'STRAIGHT_JOIN', 'TABLE', 'TERMINATED', 'THEN', 'TINYBLOB', 'TINYINT', 'TINYTEXT', 'TO', 'TRAILING', 'TRIGGER', 'TRUE', 'UNDO', 'UNION', 'UNIQUE', 'UNLOCK', 'UNSIGNED', 'UPDATE', 'USAGE', 'USE', 'USING', 'UTC_DATE', 'UTC_TIME', 'UTC_TIMESTAMP', 'VALUES', 'VARBINARY', 'VARCHAR', 'VARCHARACTER', 'VARYING', 'WHEN', 'WHERE', 'WHILE', 'WITH', 'WRITE', 'X509', 'XOR', 'YEAR_MONTH', 'ZEROFILL' 'ADD', 'ALL', 'ALTER', 'AND', 'AS', 'ASC', 'ASENSITIVE', 'BEFORE', 'BETWEEN', 'BIGINT', 'BINARY', 'BLOB', 'BOTH', 'BY', 'CALL', 'CASCADE', 'CASE', 'CHANGE', 'CHAR', 'CHARACTER', 'CHECK', 'COLLATE', 'COLUMN', 'CONDITION', 'CONNECTION', 'CONSTRAINT', 'CONTINUE', 'CONVERT', 'CREATE', 'CROSS', 'CURRENT_DATE', 'CURRENT_TIME', 'CURRENT_TIMESTAMP', 'CURRENT_USER', 'CURSOR', 'DATABASE', 'DATABASES', 'DAY_HOUR', 'DAY_MICROSECOND', 'DAY_MINUTE', 'DAY_SECOND', 'DEC', 'DECIMAL', 'DECLARE', 'DEFAULT', 'DELAYED', 'DELETE', 'DESC', 'DESCRIBE', 'DETERMINISTIC', 'DISTINCT', 'DISTINCTROW', 'DIV', 'DOUBLE', 'DROP', 'DUAL', 'EACH', 'ELSE', 'ELSEIF', 'ENCLOSED', 'ESCAPED', 'EXISTS', 'EXIT', 'EXPLAIN', 'FALSE', 'FETCH', 'FLOAT', 'FLOAT4', 'FLOAT8', 'FOR', 'FORCE', 'FOREIGN', 'FROM', 'FULLTEXT', 'GOTO', 'GRANT', 'GROUP', 'HAVING', 'HIGH_PRIORITY', 'HOUR_MICROSECOND', 'HOUR_MINUTE', 'HOUR_SECOND', 'IF', 'IGNORE', 'IN', 'INDEX', 'INFILE', 'INNER', 'INOUT', 'INSENSITIVE', 'INSERT', 'INT', 'INT1', 'INT2', 'INT3', 'INT4', 'INT8', 'INTEGER', 'INTERVAL', 'INTO', 'IS', 'ITERATE', 'JOIN', 'KEY', 'KEYS', 'KILL', 'LABEL', 'LEADING', 'LEAVE', 'LEFT', 'LIKE', 'LIMIT', 'LINEAR', 'LINES', 'LOAD', 'LOCALTIME', 'LOCALTIMESTAMP', 'LOCK', 'LONG', 'LONGBLOB', 'LONGTEXT', 'LOOP', 'LOW_PRIORITY', 'MATCH', 'MEDIUMBLOB', 'MEDIUMINT', 'MEDIUMTEXT', 'MIDDLEINT', 'MINUTE_MICROSECOND', 'MINUTE_SECOND', 'MOD', 'MODIFIES', 'NATURAL', 'NOT', 'NO_WRITE_TO_BINLOG', 'NULL', 'NUMERIC', 'ON', 'OPTIMIZE', 'OPTION', 'OPTIONALLY', 'OR', 'ORDER', 'OUT', 'OUTER', 'OUTFILE', 'PRECISION', 'PRIMARY', 'PROCEDURE', 'PURGE', 'RAID0', 'RANGE', 'READ', 'READS', 'REAL', 'REFERENCES', 'REGEXP', 'RELEASE', 'RENAME', 'REPEAT', 'REPLACE', 'REQUIRE', 'RESTRICT', 'RETURN', 'REVOKE', 'RIGHT', 'RLIKE', 'SCHEMA', 'SCHEMAS', 'SECOND_MICROSECOND', 'SELECT', 'SENSITIVE', 'SEPARATOR', 'SET', 'SHOW', 'SMALLINT', 'SPATIAL', 'SPECIFIC', 'SQL', 'SQLEXCEPTION', 'SQLSTATE', 'SQLWARNING', 'SQL_BIG_RESULT', 'SQL_CALC_FOUND_ROWS', 'SQL_SMALL_RESULT', 'SSL', 'STARTING', 'STRAIGHT_JOIN', 'TABLE', 'TERMINATED', 'THEN', 'TINYBLOB', 'TINYINT', 'TINYTEXT', 'TO', 'TRAILING', 'TRIGGER', 'TRUE', 'UNDO', 'UNION', 'UNIQUE', 'UNLOCK', 'UNSIGNED', 'UPDATE', 'USAGE', 'USE', 'USING', 'UTC_DATE', 'UTC_TIME', 'UTC_TIMESTAMP', 'VALUES', 'VARBINARY', 'VARCHAR', 'VARCHARACTER', 'VARYING', 'WHEN', 'WHERE', 'WHILE', 'WITH', 'WRITE', 'X509', 'XOR', 'YEAR_MONTH', 'ZEROFILL'
]; ];
/** /**
* @param Connection|null $db * @param Connection|null $db
* *
* @param InputInterface $input * @param InputInterface $input
* @return array * @return array
* @throws Exception * @throws Exception
*/ */
public function run(?Connection $db, InputInterface $input): array public function run(?Connection $db, InputInterface $input): array
{ {
return $this->gen($input, $db); return $this->gen($input, $db);
} }
/** /**
* @param InputInterface $input * @param InputInterface $input
* @param $db * @param $db
* @return array * @return array
* @throws Exception * @throws Exception
*/ */
public function gen(InputInterface $input, $db): array public function gen(InputInterface $input, $db): array
{ {
$this->input = $input; $this->input = $input;
$this->db = $db; $this->db = $db;
$make = $this->input->getOption('make'); $make = $this->input->getOption('make');
if (empty($make)) { if (empty($make)) {
throw new Exception('构建类型不能为空~'); throw new Exception('构建类型不能为空~');
} }
switch (strtolower($make)) { switch (strtolower($make)) {
case 'task': case 'task':
$task = new GiiTask(); $task = new GiiTask();
$task->setInput($this->input); $task->setInput($this->input);
return $task->generate(); return $task->generate();
case 'middleware': case 'middleware':
$task = new GiiMiddleware(); $task = new GiiMiddleware();
$task->setInput($this->input); $task->setInput($this->input);
return $task->generate(); return $task->generate();
case 'rpc-client': case 'rpc-client':
$task = new GiiRpcClient(); $task = new GiiRpcClient();
$task->setInput($this->input); $task->setInput($this->input);
return $task->generate(); return $task->generate();
case 'rpc-service': case 'rpc-service':
$task = new GiiRpcService(); $task = new GiiRpcService();
$task->setInput($this->input); $task->setInput($this->input);
return $task->generate(); return $task->generate();
case 'json-rpc': case 'json-rpc':
$task = new GiiJsonRpc(); $task = new GiiJsonRpc();
$task->setInput($this->input); $task->setInput($this->input);
return $task->create(); return $task->create();
default: default:
return $this->getModel($make, $input); return $this->getModel($make, $input);
} }
} }
/** /**
* @param $make * @param $make
* @param $input * @param $input
* @return array * @return array
* @throws Exception * @throws Exception
*/ */
private function getModel($make, $input): array private function getModel($make, $input): array
{ {
return $this->makeByDatabases($make, $input); return $this->makeByDatabases($make, $input);
} }
/** /**
* @param $make * @param $make
* @param InputInterface $input * @param InputInterface $input
* @return array * @return array
* @throws Exception * @throws Exception
*/ */
private function makeByDatabases($make, InputInterface $input): array private function makeByDatabases($make, InputInterface $input): array
{ {
if ($input->hasOption('table')) { if ($input->hasOption('table')) {
$this->tableName = $input->getOption('table'); $this->tableName = $input->getOption('table');
} }
return match ($make) { return match ($make) {
'controller' => $this->getTable(1, 0), 'controller' => $this->getTable(1, 0),
'model' => $this->getTable(0, 1), 'model' => $this->getTable(0, 1),
default => [], default => [],
}; };
} }
/** /**
* @param $controller * @param $controller
* @param $model * @param $model
* @return array * @return array
* *
* @throws Exception * @throws Exception
*/ */
private function getTable($controller, $model): array private function getTable($controller, $model): array
{ {
$tables = $this->getFields($this->getTables()); $tables = $this->getFields($this->getTables());
if (empty($tables)) { if (empty($tables)) {
return []; return [];
} }
$fileList = []; $fileList = [];
foreach ($tables as $key => $val) { foreach ($tables as $key => $val) {
$data = $this->createModelFile($key, $val); $data = $this->createModelFile($key, $val);
if ($controller == 1) { if ($controller == 1) {
$fileList[] = $this->generateController($data); $fileList[] = $this->generateController($data);
} }
if ($model == 1) { if ($model == 1) {
$fileList[] = $this->generateModel($data); $fileList[] = $this->generateModel($data);
} }
} }
return $fileList; return $fileList;
} }
/** /**
* @param array $data * @param array $data
* @return string * @return string
* @throws Exception * @throws Exception
*/ */
private function generateModel(array $data): string private function generateModel(array $data): string
{ {
$controller = new GiiModel($data['classFileName'], $data['tableName'], $data['visible'], $data['res'], $data['fields']); $controller = new GiiModel($data['classFileName'], $data['tableName'], $data['visible'], $data['res'], $data['fields']);
$controller->setConnection($this->db); $controller->setConnection($this->db);
$controller->setModelPath($this->modelPath); $controller->setModelPath($this->modelPath);
$controller->setModelNamespace($this->modelNamespace); $controller->setModelNamespace($this->modelNamespace);
$controller->setInput($this->input); $controller->setInput($this->input);
// $controller->setModule($this->input->getArgument('module')); // $controller->setModule($this->input->getArgument('module'));
$controller->setControllerPath($this->controllerPath); $controller->setControllerPath($this->controllerPath);
$controller->setControllerNamespace($this->controllerNamespace); $controller->setControllerNamespace($this->controllerNamespace);
return $controller->generate(); return $controller->generate();
} }
/** /**
* @param array $data * @param array $data
* @return string * @return string
* @throws Exception * @throws Exception
*/ */
private function generateController(array $data): string private function generateController(array $data): string
{ {
$controller = new GiiController($data['classFileName'], $data['fields'], $data['tableName']); $controller = new GiiController($data['classFileName'], $data['fields'], $data['tableName']);
$controller->setConnection($this->db); $controller->setConnection($this->db);
$controller->setModelPath($this->modelPath); $controller->setModelPath($this->modelPath);
$controller->setInput($this->input); $controller->setInput($this->input);
$controller->setModelNamespace($this->modelNamespace); $controller->setModelNamespace($this->modelNamespace);
$controller->setControllerPath($this->controllerPath); $controller->setControllerPath($this->controllerPath);
// $controller->setModule($this->input->getArgument('module')); // $controller->setModule($this->input->getArgument('module'));
$controller->setControllerNamespace($this->controllerNamespace); $controller->setControllerNamespace($this->controllerNamespace);
return $controller->generate(); return $controller->generate();
} }
/** /**
* @return array|string|null * @return array|string|null
* @throws Exception * @throws Exception
*/ */
private function getTables(): array|string|null private function getTables(): array|string|null
{ {
if (empty($this->tableName)) { if (empty($this->tableName)) {
return $this->showAll(); return $this->showAll();
} }
$res = $this->tableName; $res = $this->tableName;
if (is_string($res)) { if (is_string($res)) {
$res = explode(',', $this->tableName); $res = explode(',', $this->tableName);
} }
if (empty($res)) { if (empty($res)) {
return []; return [];
} }
return $res; return $res;
} }
/** /**
* @return array * @return array
* @throws Exception * @throws Exception
*/ */
private function showAll(): array private function showAll(): array
{ {
$res = []; $res = [];
$_tables = Db::findAllBySql('show tables from `' . $this->db->database . '`', [], $this->db); $_tables = Db::connect($this->db)->query('SHOW TABLES FROM `' . $this->db->database . '`');
if (empty($_tables)) { if (empty($_tables)) {
return $res; return $res;
} }
foreach ($_tables as $key => $val) { foreach ($_tables as $key => $val) {
$res[] = array_shift($val); $res[] = array_shift($val);
} }
return $res; return $res;
} }
/** /**
* @param $table * @param $table
* @return bool|int|null * @return bool|int|null
* @throws Exception * @throws Exception
*/ */
private function getIndex($table): bool|int|null private function getIndex($table): bool|int|null
{ {
$data = Db::findAllBySql('SHOW INDEX FROM ' . $table, [], $this->db); $data = Db::connect($this->db)->query('SHOW INDEX FROM `' . $this->db->database . '`.`' . $table . '`', []);
return empty($data) ? NULL : $data[0]; return empty($data) ? NULL : $data[0];
} }
/** /**
* @param $tables * @param $tables
* *
* @return array * @return array
* @throws * @throws
*/ */
private function getFields($tables): array private function getFields($tables): array
{ {
$res = []; $res = [];
if (!is_array($tables)) { if (!is_array($tables)) {
$tables = [$tables]; $tables = [$tables];
} }
foreach ($tables as $key => $val) { foreach ($tables as $key => $val) {
if (empty($val)) continue; if (empty($val)) continue;
$_tmp = Db::connect($this->db)->query('SHOW FULL FIELDS FROM `' . $this->db->database . '`.' . $val, []); $_tmp = Db::connect($this->db)->query('SHOW FULL FIELDS FROM `' . $this->db->database . '`.' . $val, []);
if (empty($_tmp)) { if (empty($_tmp)) {
continue; continue;
} }
$res[$val] = $_tmp; $res[$val] = $_tmp;
} }
return $res; return $res;
} }
/** /**
* @param $tableName * @param $tableName
* @param $tables * @param $tables
* *
* @return array * @return array
* @throws Exception * @throws Exception
*/ */
public function createModelFile($tableName, $tables): array public function createModelFile($tableName, $tables): array
{ {
$res = $visible = $fields = $keys = []; $res = $visible = $fields = $keys = [];
foreach ($tables as $_key => $_val) { foreach ($tables as $_key => $_val) {
$keys = $tableName; $keys = $tableName;
if ($_val['Extra'] == 'auto_increment' || $_val['Key'] == 'PRI') { if ($_val['Extra'] == 'auto_increment' || $_val['Key'] == 'PRI') {
$keys = $tableName; $keys = $tableName;
} }
if (!isset($keys) && !($index = $this->getIndex($tableName))) { if (!isset($keys) && !($index = $this->getIndex($tableName))) {
$keys = $index['Column_name']; $keys = $index['Column_name'];
} }
if (in_array(strtoupper($_val['Field']), $this->keyword)) { if (in_array(strtoupper($_val['Field']), $this->keyword)) {
throw new Exception('You can not use keyword "' . $_val['Field'] . '" as field at table "' . $tableName . '"'); throw new Exception('You can not use keyword "' . $_val['Field'] . '" as field at table "' . $tableName . '"');
} }
$visible[] = $this->createVisible($_val['Field']); $visible[] = $this->createVisible($_val['Field']);
$fields[] = $_val; $fields[] = $_val;
$res[] = $this->createSetFunc($_val['Field'], $_val['Comment']); $res[] = $this->createSetFunc($_val['Field'], $_val['Comment']);
} }
$classFileName = $this->getClassName($tableName); $classFileName = $this->getClassName($tableName);
return [ return [
'classFileName' => $classFileName, 'classFileName' => $classFileName,
'tableName' => $keys, 'tableName' => $keys,
'visible' => $visible, 'visible' => $visible,
'fields' => $fields, 'fields' => $fields,
'res' => $res, 'res' => $res,
]; ];
} }
/** /**
* @param $field * @param $field
* @return string * @return string
* 创建变量注释 * 创建变量注释
*/ */
private function createVisible($field): string private function createVisible($field): string
{ {
return ' return '
* @property $' . $field; * @property $' . $field;
} }
/** /**
* @param $field * @param $field
* @param $comment * @param $comment
* @return string * @return string
* 暂时不知道干嘛用的 * 暂时不知道干嘛用的
*/ */
private function createSetFunc($field, $comment): string private function createSetFunc($field, $comment): string
{ {
return ' return '
' . str_pad('\'' . $field . '\'', 20, ' ', STR_PAD_RIGHT) . '=> \'' . (empty($comment) ? ucfirst($field) : $comment) . '\','; ' . str_pad('\'' . $field . '\'', 20, ' ', STR_PAD_RIGHT) . '=> \'' . (empty($comment) ? ucfirst($field) : $comment) . '\',';
} }
/** /**
* @param $tableName * @param $tableName
* @return string * @return string
* 构建类名称 * 构建类名称
*/ */
private function getClassName($tableName): string private function getClassName($tableName): string
{ {
$res = []; $res = [];
$tableName = str_replace($this->db->tablePrefix,'', $tableName); $tableName = str_replace($this->db->tablePrefix, '', $tableName);
foreach (explode('_', $tableName) as $n => $val) { foreach (explode('_', $tableName) as $n => $val) {
$res[] = ucfirst($val); $res[] = ucfirst($val);
} }
return implode('', $res); return implode('', $res);
} }
} }