misuzu/src/db.php

<?php
define('MSZ_DATABASE_SUPPORTED', [
    'mysql',
    'sqlite',
]);
define('MSZ_DATABASE_MYSQL_DEFAULTS', [
    'host' => '127.0.0.1',
    'port' => 3306,
]);

// Output of PDOException::getCode() is string for god knows what reason
define('MSZ_DATABASE_DUPLICATE_KEY', '23000');

function db_settings(?array $databases = [], ?string $default = null): array
{
    static $settings = [];

    if(!empty($databases)) {
        $settings['databases'] = array_merge_recursive($settings['databases'] ?? [], $databases);
        $settings['default'] = $default ?? $settings['default'] ?? array_key_first($settings['databases']);
    }

    return $settings;
}

function db_connection(?string $name = null, bool $skipConnect = false, ?PDO $object = null): ?PDO
{
    static $connections = [];

    $settings = db_settings();
    $name = $name ?? $settings['default'] ?? '';

    if (!$skipConnect && empty($connections[$name])) {
        if(!empty($object)) {
            $connections[$name] = $object;
        } elseif(!empty($settings['databases'][$name])) {
            return db_connect($name, $settings['databases'][$name]);
        }
    }

    return $connections[$name] ?? null;
}

function db_prepare(string $statement, ?string $connection = null, $options = []): PDOStatement
{
    static $stmts = [];
    $encodedOptions = serialize($options);

    if (!empty($stmts[$connection][$statement][$encodedOptions])) {
        return $stmts[$connection][$statement][$encodedOptions];
    }

    return $stmts[$connection][$statement][$encodedOptions] = db_prepare_direct($statement, $connection, $options);
}

function db_prepare_direct(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)->fetchColumn(1);
}

function db_fetch(PDOStatement $stmt, $default = [])
{
    $out = $stmt->execute() ? $stmt->fetch(PDO::FETCH_ASSOC) : false;
    return $out ? $out : $default;
}

function db_fetch_all(PDOStatement $stmt, $default = [])
{
    $out = $stmt->execute() ? $stmt->fetchAll(PDO::FETCH_ASSOC) : false;
    return $out ? $out : $default;
}

// starting at 2
define('MSZ_DATABASE_CONNECT_UNSUPPORTED', 2);
define('MSZ_DATABASE_CONNECT_NO_DATABASE', 3);

function db_connect(string $name, ?array $options = null)
{
    $existing = db_connection($name, true);

    if (!empty($existing)) {
        return $existing;
    }

    if ($options === null) {
        $options = db_settings()['databases'][$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);

            $dsn .= empty($options['unix_socket'])
                ? sprintf('host=%s;port=%d;', $options['host'], $options['port'])
                : sprintf('unix_socket=%s;', $options['unix_socket']);

            $dsn .= sprintf(
                'charset=%s;dbname=%s;',
                $options['charset'] ?? 'utf8mb4',
                $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 db_connection($name, false, $connection);
}