DoctrineSchemaBuilderFactory.php

Go to the documentation of this file.

<?php
 
namespace Wikimedia\Rdbms\DBAL;
 
use Doctrine\DBAL\Platforms\AbstractPlatform;
use Doctrine\DBAL\Platforms\SqlitePlatform;
use Doctrine\DBAL\Types\Type;
use InvalidArgumentException;
 
class DoctrineSchemaBuilderFactory {
 
    public function getSchemaBuilder( string $platform ) {
        return new DoctrineSchemaBuilder( $this->getPlatform( $platform ) );
    }
 
    public function getSchemaChangeBuilder( $platform ) {
        return new DoctrineSchemaChangeBuilder( $this->getPlatform( $platform ) );
    }
 
    private function getPlatform( string $platform ) {
        $platformObject = match ( $platform ) {
            'mysql' => new MWMySQLPlatform(),
            'postgres' => new MWPostgreSqlPlatform(),
            'sqlite' => new SqlitePlatform(),
            default => throw new InvalidArgumentException( "Unknown platform: $platform" )
        };
 
        $customTypes = [
            'Enum' => [ EnumType::class, EnumType::ENUM ],
            'Tinyint' => [ TinyIntType::class, TinyIntType::TINYINT ],
            'Timestamp' => [ TimestampType::class, TimestampType::TIMESTAMP ],
        ];
 
        foreach ( $customTypes as $type => [ $class, $name ] ) {
            if ( !Type::hasType( $name ) ) {
                Type::addType( $name, $class );
            }
            $platformObject->registerDoctrineTypeMapping( $type, $name );
        }
 
        return $platformObject;
    }
}