accounts/console/db/Migration.php

61 lines
1.8 KiB
PHP

<?php
declare(strict_types=1);
namespace console\db;
use yii\db\Exception;
use yii\db\Migration as YiiMigration;
/**
* @property string $tableOptions
* @method \SamIT\Yii2\MariaDb\ColumnSchemaBuilder json()
*/
class Migration extends YiiMigration {
public function getTableOptions(string $engine = 'InnoDB'): string {
$tableOptions = null;
if ($this->db->driverName === 'mysql') {
$tableOptions = 'CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci ENGINE=' . $engine;
}
return $tableOptions;
}
/**
* @param array<string|\yii\db\ColumnSchemaBuilder>|null $columns
*/
public function createTable($table, $columns, $options = null): void {
if ($options === null) {
$options = $this->getTableOptions();
}
parent::createTable($table, $columns, $options);
}
protected function primary(string ...$columns): string {
foreach ($columns as &$column) {
$column = $this->db->quoteColumnName($column);
}
return ' PRIMARY KEY (' . implode(', ', $columns) . ') ';
}
protected function getPrimaryKeyType(string $table, bool $nullable = false): string {
$primaryKeys = $this->db->getTableSchema($table)->primaryKey;
if (count($primaryKeys) === 0) {
throw new Exception("The table \"{$table}\" have no primary keys.");
}
if (count($primaryKeys) > 1) {
throw new Exception("The table \"{$table}\" have more than one primary key.");
}
return $this->getColumnType($table, $primaryKeys[0], $nullable);
}
protected function getColumnType(string $table, string $column, bool $nullable = false): string {
return $this->db->getTableSchema($table)->getColumn($column)->dbType . ($nullable ? '' : ' NOT NULL');
}
}