Allow options to be provided to Doctrine

Signed-off-by: Andrew Summers <[email protected]>

config/config.sample.php

@@ -1824,6 +1824,34 @@
 	PDO::MYSQL_ATTR_INIT_COMMAND => 'SET wait_timeout = 28800'
 ],
 
+/**
+ * Allows defining custom DB types and options as well as the ability to override
+ * the default connection parameters as defined by
+ * OC\DB\ConnectionFactory::defaultConnectionParams.
+ */
+'dbconnectionparams' => [
+	'dbtype' => [
+		'adapter' => AdapterMySQL::class,
+		'charset' => 'UTF8',
+		'driver' => 'pdo_dbtype',
+		'wrapperClass' => Doctrine\DBAL\Connection::class,
+	],
+],
+
+/**
+ * Allows additional configuration options to be provided for the database
+ * connection as defined by Doctrine\DBAL\Configuration::class. This is useful
+ * for specifying a custom logger, middlewares, etc.
+ */
+'dbconfigurationparams' => [
+	"sqllogger" => 'Doctrine\DBAL\Logging\SQLLogger',
+	"resultcache" => 'Psr\Cache\CacheItemPoolInterface',
+	"schemaassetsfilter" => 'callable',
+	"autocommit" => true,
+	"middlewares" => array(
+		'Doctrine\DBAL\Driver\Middleware',
+	),
+],
 /**
  * sqlite3 journal mode can be specified using this configuration parameter -
  * can be 'WAL' or 'DELETE' see for more details https://www.sqlite.org/wal.html

lib/private/AppFramework/App.php

@@ -34,16 +34,15 @@
 use OC\AppFramework\DependencyInjection\DIContainer;
 use OC\AppFramework\Http\Dispatcher;
 use OC\AppFramework\Http\Request;
-use OCP\App\IAppManager;
-use OCP\Profiler\IProfiler;
 use OC\Profiler\RoutingDataCollector;
-use OCP\AppFramework\QueryException;
 use OCP\AppFramework\Http;
 use OCP\AppFramework\Http\ICallbackResponse;
 use OCP\AppFramework\Http\IOutput;
+use OCP\AppFramework\QueryException;
 use OCP\Diagnostics\IEventLogger;
 use OCP\HintException;
 use OCP\IRequest;
+use OCP\Profiler\IProfiler;
 
 /**
  * Entry point for every request in your app. You can consider this as your
@@ -69,7 +68,7 @@ public static function buildAppNamespace(string $appId, string $topNamespace = '
 			return $topNamespace . self::$nameSpaceCache[$appId];
 		}
 
-		$appInfo = \OCP\Server::get(IAppManager::class)->getAppInfo($appId);
+		$appInfo = self::getAppInfo($appId);
 		if (isset($appInfo['namespace'])) {
 			self::$nameSpaceCache[$appId] = trim($appInfo['namespace']);
 		} else {
@@ -91,11 +90,42 @@ public static function buildAppNamespace(string $appId, string $topNamespace = '
 		return $topNamespace . self::$nameSpaceCache[$appId];
 	}
 
+	public static function getAppInfo(string $appId, bool $path = false, $lang = null) {
+		if ($path) {
+			$file = $appId;
+		} else {
+			try {
+				if ($appPath = \OC_App::findAppInDirectories($appId, true)) {
+					$appPath = $appPath['path'] . '/' . $appId;
+				}
+			} catch (\OCP\App\AppPathNotFoundException $e) {
+				return null;
+			}
+			$file = $appPath . '/appinfo/info.xml';
+		}
+
+		$parser = new \OC\App\InfoParser();
+		$data = $parser->parse($file);
+
+		if (is_array($data)) {
+			$data = \OC_App::parseAppInfo($data, $lang);
+		}
+
+		return $data;
+	}
+
 	public static function getAppIdForClass(string $className, string $topNamespace = 'OCA\\'): ?string {
 		if (!str_starts_with($className, $topNamespace)) {
 			return null;
 		}
 
+		if (str_starts_with($className, $topNamespace)) {
+			$classNoTopNamespace = substr($className, strlen($topNamespace));
+			$appNameParts = explode('\\', $classNoTopNamespace, 2);
+			$appName = reset($appNameParts);
+			return strtolower($appName);
+		}
+
 		foreach (self::$nameSpaceCache as $appId => $namespace) {
 			if (str_starts_with($className, $topNamespace . $namespace . '\\')) {
 				return $appId;
@@ -105,6 +135,16 @@ public static function getAppIdForClass(string $className, string $topNamespace
 		return null;
 	}
 
+	public static function registerAppClass(string $className): void {
+		$classParts = explode('\\', $className, 2);
+		$topNamespace = reset($classParts) . '\\';
+		$appId = self::getAppIdForClass($className, $topNamespace);
+
+		if ($appPath = \OC_App::findAppInDirectories($appId, true)) {
+			$appPath = $appPath['path'] . '/' . $appId;
+			\OC_App::registerAutoloading($appId, $appPath);
+		}
+	}
 
 	/**
 	 * Shortcut for calling a controller method and printing the result

lib/private/DB/ConnectionFactory.php

@@ -33,6 +33,7 @@
 use Doctrine\DBAL\DriverManager;
 use Doctrine\DBAL\Event\Listeners\OracleSessionInit;
 use Doctrine\DBAL\Event\Listeners\SQLSessionInit;
+use OC\AppFramework\App;
 use OC\SystemConfig;
 
 /**
@@ -93,6 +94,21 @@ public function __construct(SystemConfig $systemConfig) {
 		if ($collationOverride) {
 			$this->defaultConnectionParams['mysql']['collation'] = $collationOverride;
 		}
+
+		$connectionParams = $this->config->getValue('dbconnectionparams', array());
+
+		foreach ($connectionParams as $type) {
+			foreach ($type as $key => $param) {
+				switch ($key) {
+					case 'adapter':
+					case 'wrapperClass':
+						\OC::$server->get(App::class)::registerAppClass($param);
+						break;
+				}
+			}
+		}
+
+		$this->defaultConnectionParams = array_replace_recursive($this->defaultConnectionParams, $connectionParams);
 	}
 
 	/**
@@ -158,9 +174,40 @@ public function getConnection($type, $additionalConnectionParams) {
 				break;
 		}
 		/** @var Connection $connection */
+		$configuration = new Configuration();
+
+		foreach ($this->config->getValue('dbconfigurationparams', array()) as $param => $value) {
+			switch ($param) {
+				case "sqllogger":
+					\OC::$server->get(App::class)::registerAppClass($value);
+					$configuration->setSQLLogger(new $value());
+					break;
+				case "resultcache":
+					\OC::$server->get(App::class)::registerAppClass($value);
+					$configuration->setResultCache(new $value());
+					break;
+				case "schemaassetsfilter":
+					$configuration->setSchemaAssetsFilter($value);
+					break;
+				case "autocommit":
+					$configuration->setAutoCommit($value);
+					break;
+				case "middlewares":
+					$middlewares = array();
+
+					foreach ($value as $middleware) {
+						\OC::$server->get(App::class)::registerAppClass($middleware);
+						array_push($middlewares, new $middleware());
+					}
+
+					$configuration->setMiddlewares($value);
+					break;
+			}
+		}
+
 		$connection = DriverManager::getConnection(
 			array_merge($this->getDefaultConnectionParams($type), $additionalConnectionParams),
-			new Configuration(),
+			$configuration,
 			$eventManager
 		);
 		return $connection;