gnu-social/src/Util/SchemaDefDriver.php

135 lines
4.5 KiB
PHP
Raw Normal View History

<?php
namespace App\Util;
use Doctrine\Persistence\Mapping\ClassMetadata;
use Doctrine\Persistence\Mapping\Driver\StaticPHPDriver;
use Functional as F;
class SchemaDefDriver extends StaticPHPDriver
{
/**
* PEAR DB type => Doctrine type
*/
private const types = [
'varchar' => 'string',
'int' => 'integer',
'tinyint' => 'smallint', // no portable tinyint
'bigint' => 'bigint',
'bool' => 'boolean',
'numeric' => 'decimal',
'text' => 'text',
'datetime' => 'datetime',
// Unused in V2, but might start being used
'date' => 'date',
'time' => 'time',
'datetimez' => 'datetimez',
'object' => 'object',
'array' => 'array',
'simplearray' => 'simplearray',
'json_array' => 'json_array',
'float' => 'float',
'guid' => 'guid',
'blob' => 'blob',
];
/**
* Fill in the database $metadata for $className
*
* @param string $className
* @param ClassMetadata $metadata
*/
public function loadMetadataForClass($className, ClassMetadata $metadata)
{
$schema = $className::schemaDef();
$metadata->setPrimaryTable(['name' => $schema['name'],
'indexes' => self::kv_to_name_col($schema['indexes']),
'uniqueConstraints' => self::kv_to_name_col($schema['unique keys']),
'options' => ['comment' => $schema['description']], ]);
foreach ($schema['fields'] as $name => $opts) {
// Convert old to new types
$type = // $name === 'date'
// // Old date fields were stored as int, store as datetime/timestamp
// ? 'datetime'
// // For ints, prepend the size (smallint)
// // The size field doesn't exist otherwise
// :
self::types[($opts['size'] ?? '') . $opts['type']];
$unique = null;
foreach ($schema['unique keys'] as $key => $uniq_arr) {
if (in_array($name, $uniq_arr)) {
$unique = $key;
break;
}
}
$field = [
// boolean, optional
'id' => in_array($name, $schema['primary key']),
// string
'fieldName' => $name,
// string
'type' => $type,
// stringn, optional
'unique' => $unique,
// String length, ignored if not a string
// int, optional
'length' => $opts['length'] ?? null,
// boolean, optional
'nullable' => !($opts['not null'] ?? false),
// Numeric precision and scale, ignored if not a number
// integer, optional
'precision' => $opts['precision'] ?? null,
// integer, optional
'scale' => $opts['scale'] ?? null,
'options' => [
'comment' => $opts['description'],
'default' => $opts['default'] ?? null,
'unsigned' => $opts['unsigned'] ?? null,
// 'fixed' => bool, unused
// 'collation' => string, unused
// 'check', unused
],
// 'columnDefinition', unused
];
// The optional feilds from earlier were populated with null, remove them
$field = array_filter($field, F\not('is_null'));
$field['options'] = array_filter($field['options'], F\not('is_null'));
$metadata->mapField($field);
}
// TODO foreign keys
}
/**
* Override StaticPHPDriver's method,
* we care about classes that have the method `schemaDef`,
* instead of `loadMetadata`.
*
* @param string $className
*/
public function isTransient($className)
{
return !method_exists($className, 'schemaDef');
}
/**
* Convert [$key => $val] to ['name' => $key, 'columns' => $val]
*
* @param array
*
* @return array
*/
private static function kv_to_name_col(array $arr): array
{
$res = [];
foreach ($arr as $name => $cols) {
$res[] = ['name' => $name, 'columns' => $cols];
}
return $res;
}
}