kiri-gii/GiiModel.php

<?php
declare(strict_types=1);


namespace Gii;

use Database\Db;
use Exception;
use Kiri;
use ReflectionException;
use function logger;

/**
 * Class GiiModel
 * @package Gii
 */
class GiiModel extends GiiBase
{

	public ?string $classFileName;
	public ?array $visible;
	public ?array $res;
	public ?array $fields;

	/**
	 * GiiModel constructor.
	 * @param string $classFileName
	 * @param string $tableName
	 * @param array $visible
	 * @param array $res
	 * @param array $fields
	 */
	public function __construct(string $classFileName, string $tableName, array $visible, array $res, array $fields)
	{
		$this->classFileName = $classFileName;
		$this->tableName = $tableName;
		$this->visible = $visible;
		$this->res = $res;
		$this->fields = $fields;
	}

	/**
	 * @throws ReflectionException
	 * @throws Exception
	 */
	public function generate(): string
	{
		$class = '';
		$modelPath = $this->getModelPath();
		$managerName = $this->classFileName;

		$namespace = rtrim($modelPath['namespace'], '\\');

		if (file_exists($modelPath['path'] . '/' . $managerName . '.php')) {
			try {
				$className = str_replace('\\\\', '\\', "{$modelPath['namespace']}\\{$managerName}");

				$class = Kiri::getDi()->getReflectionClass($className);

				$html = '<?php
namespace ' . $namespace . ';

';
				$imports = $this->getImports($modelPath['path'] . '/' . $managerName . '.php', $class);
				if (!empty($imports)) {
					$html .= $imports . PHP_EOL;
				}
			} catch (\Throwable $e) {
				error($e);
			}
		}


		if (empty($html)) {
			$html = '<?php
namespace ' . $namespace . ';


use Exception;
use Kiri\Core\Json;
use Database\Connection;
use Database\Model;
' . PHP_EOL;
		}

		$createSql = $this->setCreateSql($this->tableName);

		if (!str_contains($html, $createSql)) {
			$html .= '
' . $this->setCreateSql($this->tableName);
		}

		$html .= '

/**
 * Class ' . $managerName . '
 *
 *' . implode('', $this->visible) . '
 */
class ' . $managerName . ' extends Model
{

';

		if (!empty($class)) {
			$html .= $this->getClassProperty($class);
		}

		$primary = $this->createPrimary($this->fields);
		if (!empty($primary)) {
			$html .= $primary . "\n";
		}

		$html .= $this->createTableName($this->tableName) . "\n";

		$html .= $this->createDatabaseSource();

		$html .= $this->createRules($this->fields);

		if (is_object($class)) {
			$html .= $this->getClassMethods($class, ['rules', 'tableName', 'attributes', 'getDb']);
		} else {
			$other = $this->generate_json_function($html, $this->fields);
			if (!empty($other)) {
				$html .= implode($other);
			}
		}

		$html .= '
}';

		$file = rtrim($modelPath['path'], '/') . '/' . $managerName . '.php';
		if (file_exists($file)) {
			unlink($file);
		}

		Kiri::writeFile($file, $html);
		return $managerName . '.php';
	}


	/**
	 * @param $html
	 * @param $fields
	 * @return array
	 */
	private function generate_json_function($html, $fields): array
	{
		$strings = [];
		foreach ($fields as $field) {
			if ($field['Type'] === 'json') {
				$function = '
	/**
	 * @param string|null $value
	 * @return int|bool|string
	 * @throws Exception
    */
	public function set' . ucfirst($field['Field']) . 'Attribute(?array $value): int|bool|string
	{
		return \json_encode($value, JSON_UNESCAPED_UNICODE); 
	}
	';

				$get_function = '
	/**
	 * @param $value
	 * @return array|null|bool
	 */
	public function get' . ucfirst($field['Field']) . 'Attribute(?string $value): array|null|bool
	{
		return \json_decode($value, true); 
	}
	';

				if (!str_contains($html, 'set' . ucfirst($field['Field']) . 'Attribute')) {
					$strings[] = $function;
				}

				if (!str_contains($html, 'get' . ucfirst($field['Field']) . 'Attribute')) {
					$strings[] = $get_function;
				}
			}
		}

		return $strings;
	}


	/**
	 * @param $field
	 * @return string
	 * 创建表名称
	 */
	private function createTableName($field): string
	{

		$prefixed = $this->db->tablePrefix;
		if (!empty($prefixed)) {
			if (str_starts_with($field, $prefixed)) {
				$field = str_replace($prefixed, '', $field);
			}
		}

		return '
    /**
     * @inheritdoc
     */
    protected string $table = \'' . $field . '\';
';
	}

	/**
	 * @param $fields
	 * @return string
	 * 创建效验规则
	 */
	private function createRules($fields): string
	{
		$data = [];
		foreach ($fields as $key => $val) {
			if ($val['Extra'] == 'auto_increment') continue;
			$type = preg_replace('/\(.*?\)|\s+\w+/', '', $val['Type']);
			foreach ($this->type as $_key => $_val) {
				if (in_array($type, $_val)) {
					$type = lcfirst(str_replace('get', '', $_key));
					break;
				}
			}
			$data[$type][] = $val;
		}

		$_field_one = '';
		$required = $this->getRequired($fields);
		if (!empty($required)) {
			$_field_one .= $required;
		}
		foreach ($data as $key => $val) {
			$field = '[\'' . implode('\', \'', array_column($val, 'Field')) . '\']';
			if (count($val) == 1) {
				$field = '\'' . current($val)['Field'] . '\'';
			}
			$_field_one .= '
			[' . $field . ', \'' . $key . '\'],';
		}
		foreach ($data as $key => $val) {
			$length = $this->getLength($val);
			if (!empty($length)) {
				$_field_one .= $length . ',';
			}
		}
		$required = $this->getUnique($fields);
		if (!empty($required)) {
			$_field_one .= $required;
		}
		return '
	/**
	 * @return array
	 */
    public function rules(): array
    {
        return [' . $_field_one . '
        ];
    }
';
	}

	/**
	 * @param $val
	 * @return string
	 */
	public function getLength($val): string
	{
		$data = [];
		foreach ($val as $key => $_val) {
			$preg = preg_match('/(\w+)\((.*?)\)/', $_val['Type'], $results);
			if ($preg && isset($results[2])) {
				$results[] = $_val['Field'];

				$data[$results[2]][] = $results;
			}
		}
		if (empty($data)) return '';
		$string = [];
		foreach ($data as $key => $_val) {
			if (in_array($_val[0][1], $this->type['float'])) {
				$e_x = explode(',', $key);
				$key = '\'round\' => ' . $e_x[1] . ', \'maxLength\' => ' . ((int)$e_x[0] + 1);
			} else if (is_string($key) && str_contains($key, ',')) {
			} else {
				$key = '\'maxLength\' => ' . $key;
			}
			if (count($_val) == 1) {
				$_tmp = '
			[\'' . $_val[0][3] . '\', ' . ($_val[0][1] == 'enum' ? '\'enum\' => [' . $key .']' : $key) . ']';
			} else {
				$_tmp = '
			[[\'' . implode('\', \'', array_column($_val, 3)) . '\'], ' . $key . ']';
			}
			$string[] = $_tmp;
		}
		return implode(',', $string);
	}

	/**
	 * @param $fields
	 * @return string
	 */
	public function getUnique($fields): string
	{
		$data = [];
		foreach ($fields as $_val) {
			if ($_val['Extra'] == 'auto_increment') continue;
			if (str_contains($_val['Type'], 'unique')) {
				$data[] = $_val['Field'];
			}
		}
		if (empty($data)) {
			return '';
		}
		return '
			[[\'' . implode('\', \'', $data) . '\'], \'unique\'],';
	}

	/**
	 * @param $val
	 * @return string
	 */
	public function getRequired($val): string
	{
		$data = [];
		foreach ($val as $_key => $_val) {
			if ($_val['Extra'] == 'auto_increment') continue;
			if ($_val['Key'] == 'PRI' || $_val['Key'] == 'UNI' || $this->checkIsRequired($_val) === 'true') {
				$data[] = $_val['Field'];
			}
		}
		if (empty($data)) {
			return '';
		}
		return '
			[[\'' . implode('\', \'', $data) . '\'], \'required\'],';
	}

	/**
	 * 用来生成文档的
	 * 格式
	 * @param $fields
	 * @return null|string
	 * array(
	 *      'field' ,'字段類型' ,'是否必填' ,'字段长度' , '字段解释',
	 *      'field' ,'字段類型' ,'是否必填' ,'字段长度' , '字段解释',
	 *      'field' ,'字段類型' ,'是否必填' ,'字段长度' , '字段解释',
	 *      'field' ,'字段類型' ,'是否必填' ,'字段长度' , '字段解释',
	 *      'field' ,'字段類型' ,'是否必填' ,'字段长度' , '字段解释',
	 *      'field' ,'字段類型' ,'是否必填' ,'字段长度' , '字段解释',
	 * )
	 */
	private function createPrimary($fields): ?string
	{
		$field = $this->getPrimaryKey($fields);
		if (empty($field)) {
			return null;
		}
		return '
	/**
	 * @var string|null 
	 */
	protected ?string $primary = \'' . $field . '\';';
	}

	/**
	 * @return string
	 */
	private function createDatabaseSource(): string
	{
		return '
    /**
	 * @var string
	 */
    protected string $connection = \'' . $this->db->id . '\';
';
	}

	/**
	 * @param $table
	 * @return string
	 * @throws Exception
	 */
	private function setCreateSql($table): string
	{
		if (isset(Gii::$createSqls[$table])) {
			return Gii::$createSqls[$table];
		}

		$text = Db::showCreateSql($table, $this->db)['Create Table'] ?? '';

		$_tmp = [];
		foreach (explode(PHP_EOL, $text) as $val) {
			$_tmp[] = '// ' . $val;
		}

		return Gii::$createSqls[$table] = implode(PHP_EOL, $_tmp);
	}


}