127 lines
3.9 KiB
PHP
127 lines
3.9 KiB
PHP
<?php
|
|
define('MSZ_DATABASE_OBJECTS_STORE', '_msz_database_connects');
|
|
define('MSZ_DATABASE_DEFAULT_NAME_STORE', '_msz_database_default');
|
|
define('MSZ_DATABASE_OPTIONS_STORE', '_msz_database_options');
|
|
|
|
define('MSZ_DATABASE_SUPPORTED', [
|
|
'mysql',
|
|
'sqlite',
|
|
]);
|
|
define('MSZ_DATABASE_MYSQL_DEFAULTS', [
|
|
'host' => '127.0.0.1',
|
|
'port' => 3306,
|
|
]);
|
|
|
|
function db_setup(array $databases, ?string $default = null): void
|
|
{
|
|
$GLOBALS[MSZ_DATABASE_OPTIONS_STORE] = $databases;
|
|
$GLOBALS[MSZ_DATABASE_DEFAULT_NAME_STORE] = $default ?? key($databases);
|
|
}
|
|
|
|
function db_connection(?string $name = null): ?PDO
|
|
{
|
|
$name = $name ?? $GLOBALS[MSZ_DATABASE_DEFAULT_NAME_STORE] ?? '';
|
|
|
|
if (empty($GLOBALS[MSZ_DATABASE_OBJECTS_STORE][$name])
|
|
&& !empty($GLOBALS[MSZ_DATABASE_OPTIONS_STORE][$name])) {
|
|
return db_connect($name, $GLOBALS[MSZ_DATABASE_OPTIONS_STORE][$name]);
|
|
}
|
|
|
|
return $GLOBALS[MSZ_DATABASE_OBJECTS_STORE][$name] ?? null;
|
|
}
|
|
|
|
function db_prepare(string $statement, ?string $connection = null, $options = []): PDOStatement
|
|
{
|
|
return db_connection($connection)->prepare($statement, $options);
|
|
}
|
|
|
|
function db_query(string $statement, ?string $connection = null): PDOStatement
|
|
{
|
|
return db_connection($connection)->query($statement);
|
|
}
|
|
|
|
function db_exec(string $statement, ?string $connection = null)
|
|
{
|
|
return db_connection($connection)->exec($statement);
|
|
}
|
|
|
|
function db_last_insert_id(?string $name = null, ?string $connection = null): string
|
|
{
|
|
return db_connection($connection)->lastInsertId($name);
|
|
}
|
|
|
|
function db_query_count(?string $connection = null): int
|
|
{
|
|
return (int)db_query('SHOW SESSION STATUS LIKE "Questions"', $connection)->fetch()['Value'];
|
|
}
|
|
|
|
// starting at 2
|
|
define('MSZ_DATABASE_CONNECT_UNSUPPORTED', 2);
|
|
define('MSZ_DATABASE_CONNECT_NO_DATABASE', 3);
|
|
|
|
function db_connect(string $name, ?array $options = null)
|
|
{
|
|
if (!empty($GLOBALS[MSZ_DATABASE_OBJECTS_STORE][$name])) {
|
|
return $GLOBALS[MSZ_DATABASE_OBJECTS_STORE][$name];
|
|
}
|
|
|
|
if ($options === null) {
|
|
$options = $GLOBALS[MSZ_DATABASE_OPTIONS_STORE][$name] ?? [];
|
|
}
|
|
|
|
if (!in_array($options['driver'], MSZ_DATABASE_SUPPORTED)) {
|
|
return MSZ_DATABASE_CONNECT_UNSUPPORTED;
|
|
}
|
|
|
|
$dsn = $options['driver'] . ':';
|
|
$pdoOptions = [
|
|
PDO::ATTR_CASE => PDO::CASE_NATURAL,
|
|
PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
|
|
PDO::ATTR_ORACLE_NULLS => PDO::NULL_NATURAL,
|
|
PDO::ATTR_STRINGIFY_FETCHES => false,
|
|
PDO::ATTR_EMULATE_PREPARES => false,
|
|
];
|
|
|
|
switch ($options['driver']) {
|
|
case 'sqlite':
|
|
if ($options['memory']) {
|
|
$dsn .= ':memory:';
|
|
} else {
|
|
$databasePath = realpath($options['database'] ?? MSZ_ROOT . '/store/misuzu.db');
|
|
|
|
if ($databasePath === false) {
|
|
return MSZ_DATABASE_CONNECT_NO_DATABASE;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case 'mysql':
|
|
$options = array_merge(MSZ_DATABASE_MYSQL_DEFAULTS, $options);
|
|
|
|
if ($options['unix_socket'] ?? false) {
|
|
$dsn .= 'unix_socket=' . $options['unix_socket'] . ';';
|
|
} else {
|
|
$dsn .= 'host=' . $options['host'] . ';';
|
|
$dsn .= 'port=' . intval($options['port']) . ';';
|
|
}
|
|
|
|
$dsn .= 'charset=' . ($options['charset'] ?? 'utf8mb4') . ';';
|
|
$dsn .= 'dbname=' . ($options['database'] ?? 'misuzu') . ';';
|
|
|
|
$pdoOptions[PDO::MYSQL_ATTR_INIT_COMMAND] = "
|
|
SET SESSION
|
|
sql_mode = 'STRICT_TRANS_TABLES,NO_ZERO_IN_DATE,NO_ZERO_DATE,ERROR_FOR_DIVISION_BY_ZERO,NO_ENGINE_SUBSTITUTION',
|
|
time_zone = '+00:00';
|
|
";
|
|
break;
|
|
}
|
|
|
|
$connection = new PDO(
|
|
$dsn,
|
|
($options['username'] ?? null),
|
|
($options['password'] ?? null),
|
|
$pdoOptions
|
|
);
|
|
|
|
return $GLOBALS[MSZ_DATABASE_OBJECTS_STORE][$name] = $connection;
|
|
}
|