125 lines
2.8 KiB
PHP
125 lines
2.8 KiB
PHP
<?php namespace System\DB;
|
|
|
|
use System\Config;
|
|
|
|
class Connector {
|
|
|
|
/**
|
|
* The PDO connection options.
|
|
*
|
|
* @var array
|
|
*/
|
|
public static $options = array(
|
|
\PDO::ATTR_CASE => \PDO::CASE_LOWER,
|
|
\PDO::ATTR_ERRMODE => \PDO::ERRMODE_EXCEPTION,
|
|
\PDO::ATTR_ORACLE_NULLS => \PDO::NULL_NATURAL,
|
|
\PDO::ATTR_STRINGIFY_FETCHES => false,
|
|
\PDO::ATTR_EMULATE_PREPARES => false,
|
|
);
|
|
|
|
/**
|
|
* Establish a PDO database connection.
|
|
*
|
|
* @param string $connection
|
|
* @return PDO
|
|
*/
|
|
public static function connect($connection)
|
|
{
|
|
$config = static::configuration($connection);
|
|
|
|
switch ($config->driver)
|
|
{
|
|
case 'sqlite':
|
|
return static::connect_to_sqlite($config);
|
|
|
|
case 'mysql':
|
|
case 'pgsql':
|
|
return static::connect_to_server($config);
|
|
|
|
default:
|
|
return static::connect_to_generic($config);
|
|
}
|
|
|
|
throw new \Exception('Database driver '.$config->driver.' is not supported.');
|
|
}
|
|
|
|
/**
|
|
* Establish a PDO connection to a SQLite database.
|
|
*
|
|
* SQLite database paths can be specified either relative to the application/db
|
|
* directory, or as an absolute path to any location on the file system.
|
|
*
|
|
* @param object $config
|
|
* @return PDO
|
|
*/
|
|
private static function connect_to_sqlite($config)
|
|
{
|
|
if (file_exists($path = DATABASE_PATH.$config->database.'.sqlite'))
|
|
{
|
|
return new \PDO('sqlite:'.$path, null, null, static::$options);
|
|
}
|
|
elseif (file_exists($config->database))
|
|
{
|
|
return new \PDO('sqlite:'.$config->database, null, null, static::$options);
|
|
}
|
|
else
|
|
{
|
|
throw new \Exception("SQLite database [".$config->database."] could not be found.");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Connect to a MySQL or PostgreSQL database server.
|
|
*
|
|
* @param object $config
|
|
* @return PDO
|
|
*/
|
|
private static function connect_to_server($config)
|
|
{
|
|
$dsn = $config->driver.':host='.$config->host.';dbname='.$config->database;
|
|
|
|
if (isset($config->port))
|
|
{
|
|
$dsn .= ';port='.$config->port;
|
|
}
|
|
|
|
$connection = new \PDO($dsn, $config->username, $config->password, static::$options);
|
|
|
|
if (isset($config->charset))
|
|
{
|
|
$connection->prepare("SET NAMES '".$config->charset."'")->execute();
|
|
}
|
|
|
|
return $connection;
|
|
}
|
|
|
|
/**
|
|
* Connect to a generic data source.
|
|
*
|
|
* @param object $config
|
|
* @return PDO
|
|
*/
|
|
private static function connect_to_generic($config)
|
|
{
|
|
return new \PDO($config->driver.':'.$config->dsn, $config->username, $config->password, static::$options);
|
|
}
|
|
|
|
/**
|
|
* Get the configuration options for a database connection.
|
|
*
|
|
* @param string $connection
|
|
* @return object
|
|
*/
|
|
private static function configuration($connection)
|
|
{
|
|
$config = Config::get('db.connections');
|
|
|
|
if ( ! array_key_exists($connection, $config))
|
|
{
|
|
throw new \Exception("Database connection [$connection] is not defined.");
|
|
}
|
|
|
|
return (object) $config[$connection];
|
|
}
|
|
|
|
} |