126 lines
3.9 KiB
PHP
126 lines
3.9 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Kimai time-tracking app.
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace App\DependencyInjection\Compiler;
|
|
|
|
use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
|
|
use Symfony\Component\DependencyInjection\ContainerBuilder;
|
|
use Symfony\Component\Yaml\Yaml;
|
|
|
|
/**
|
|
* Dynamically loads additional doctrine functions for the configured database engine.
|
|
*/
|
|
class DoctrineCompilerPass implements CompilerPassInterface
|
|
{
|
|
|
|
/**
|
|
* @var string[]
|
|
*/
|
|
protected $allowedEngines = [
|
|
'mysql',
|
|
'oracle',
|
|
'postgres',
|
|
'sqlite'
|
|
];
|
|
|
|
/**
|
|
* @param ContainerBuilder $container
|
|
* @return array|false|null|string
|
|
* @throws \Exception
|
|
*/
|
|
protected function findEngine(ContainerBuilder $container)
|
|
{
|
|
$engine = null;
|
|
|
|
// TODO - this does return the wrong connection. it used to be mysql, even if
|
|
// TODO - getenv('DATABASE_URL') returned an sqlite:// connection string
|
|
/*
|
|
$dbConfig = $container->getExtensionConfig('doctrine');
|
|
if (isset($dbConfig[0]['dbal']['driver'])) {
|
|
$engine = str_replace('pdo_', '', $dbConfig[0]['dbal']['driver']);
|
|
}
|
|
*/
|
|
|
|
if ($engine === null) {
|
|
$dbConfig = explode('://', getenv('DATABASE_URL'));
|
|
$engine = $dbConfig['0'] ?: null;
|
|
}
|
|
|
|
if ($engine === null) {
|
|
$engine = getenv('DATABASE_ENGINE');
|
|
}
|
|
|
|
if ($engine === null) {
|
|
throw new \Exception(
|
|
'Could not detect database engine. Please set the environment config DATABASE_ENGINE ' .
|
|
'to one of: "' . implode(', ', $this->allowedEngines) . '" in your .env file: DATABASE_ENGINE=sqlite'
|
|
);
|
|
}
|
|
|
|
if (!in_array($engine, $this->allowedEngines)) {
|
|
throw new \Exception(
|
|
'Unsupported database engine: ' . $engine . '. Kimai only supports one of: ' .
|
|
implode(', ', $this->allowedEngines)
|
|
);
|
|
}
|
|
|
|
return $engine;
|
|
}
|
|
|
|
/**
|
|
* @param ContainerBuilder $container
|
|
* @return string
|
|
* @throws \Exception
|
|
*/
|
|
protected function getConfigFile(ContainerBuilder $container)
|
|
{
|
|
$engine = $this->findEngine($container);
|
|
|
|
$configDir = realpath(
|
|
$container->getParameter('kernel.project_dir') . '/vendor/beberlei/DoctrineExtensions/config/'
|
|
);
|
|
|
|
$configFile = $configDir . '/' . $engine . '.yml';
|
|
|
|
if (!file_exists($configFile)) {
|
|
throw new \Exception('Could not find config file for database engine. Looked at ' . $configFile);
|
|
}
|
|
|
|
return $configFile;
|
|
}
|
|
|
|
/**
|
|
* @param ContainerBuilder $container
|
|
* @throws \Exception
|
|
*/
|
|
public function process(ContainerBuilder $container)
|
|
{
|
|
$configFile = $this->getConfigFile($container);
|
|
$config = Yaml::parse(file_get_contents($configFile));
|
|
|
|
if (!isset($config['doctrine']['orm']['dql']) || empty($config['doctrine']['orm']['dql'])) {
|
|
throw new \Exception('could not load custom Doctrine functions from: ' . $configFile);
|
|
}
|
|
|
|
$sql = $config['doctrine']['orm']['dql'];
|
|
|
|
$ormConfig = $container->getDefinition('doctrine.orm.default_configuration');
|
|
|
|
foreach ($sql['string_functions'] as $name => $function) {
|
|
$ormConfig->addMethodCall('addCustomStringFunction', [$name, $function]);
|
|
}
|
|
foreach ($sql['numeric_functions'] as $name => $function) {
|
|
$ormConfig->addMethodCall('addCustomNumericFunction', [$name, $function]);
|
|
}
|
|
foreach ($sql['datetime_functions'] as $name => $function) {
|
|
$ormConfig->addMethodCall('addCustomDatetimeFunction', [$name, $function]);
|
|
}
|
|
}
|
|
}
|