Arsse/tests/lib/DatabaseDrivers/PostgreSQL.php

<?php
/** @license MIT
 * Copyright 2017 J. King, Dustin Wilson et al.
 * See LICENSE and AUTHORS files for details */

declare(strict_types=1);
namespace JKingWeb\Arsse\Test\DatabaseDrivers;

use JKingWeb\Arsse\Arsse;
use JKingWeb\Arsse\Db\Driver;

trait PostgreSQL {
    protected static $implementation = "PostgreSQL";
    protected static $backend = "PostgreSQL";
    protected static $dbResultClass = \JKingWeb\Arsse\Db\PostgreSQL\Result::class;
    protected static $dbStatementClass = \JKingWeb\Arsse\Db\PostgreSQL\Statement::class;
    protected static $dbDriverClass = \JKingWeb\Arsse\Db\PostgreSQL\Driver::class;
    protected static $stringOutput = true;
    
    public static function dbInterface() {
        $connString = \JKingWeb\Arsse\Db\PostgreSQL\Driver::makeConnectionString(false, Arsse::$conf->dbPostgreSQLUser, Arsse::$conf->dbPostgreSQLPass, Arsse::$conf->dbPostgreSQLDb, Arsse::$conf->dbPostgreSQLHost, Arsse::$conf->dbPostgreSQLPort, "");
        if ($d = @pg_connect($connString, \PGSQL_CONNECT_FORCE_NEW)) {
            foreach (\JKingWeb\Arsse\Db\PostgreSQL\Driver::makeSetupQueries(Arsse::$conf->dbPostgreSQLSchema) as $q) {
                pg_query($d, $q);
            }
            return $d;
        } else {
            return;
        }
    }

    public static function dbExec($db, $q) {
        if ($db instanceof Driver) {
            $db->exec($q);
        } elseif ($db instanceof \PDO) {
            $db->exec($q);
        } else {
            pg_query($db, $q);
        }
    }
    
    public static function dbTableList($db): array {
        $listObjects = "SELECT table_name as name, 'TABLE' as type from information_schema.tables where table_schema = current_schema() and table_name like 'arsse_%' union SELECT collation_name as name, 'COLLATION' as type from information_schema.collations where collation_schema = current_schema()";
        if ($db instanceof Driver) {
            return $db->query($listObjects)->getAll();
        } elseif ($db instanceof \PDO) {
            return $db->query($listObjects)->fetchAll(\PDO::FETCH_ASSOC);
        } else {
            $r = @pg_query($db, $listObjects);
            $out = $r ? pg_fetch_all($r) : false;
            return $out ? $out : [];
        }
    }

    public static function dbTruncate($db, array $afterStatements = []) {
        // rollback any pending transaction
        try {
            @self::dbExec($db, "ROLLBACK");
        } catch (\Throwable $e) {
        }
        foreach (self::dbTableList($db) as $obj) {
            if ($obj['type'] !== "TABLE") {
                continue;
            } elseif ($obj['name'] === "arsse_meta") {
                self::dbExec($db, "DELETE FROM {$obj['name']} where key <> 'schema_version'");
            } else {
                self::dbExec($db, "TRUNCATE TABLE {$obj['name']} restart identity cascade");
            }
        }
        foreach ($afterStatements as $st) {
            self::dbExec($db, $st);
        }
    }

    public static function dbRaze($db, array $afterStatements = []) {
        // rollback any pending transaction
        try {
            @self::dbExec($db, "ROLLBACK");
        } catch (\Throwable $e) {
        }
        foreach (self::dbTableList($db) as $obj) {
            self::dbExec($db, "DROP {$obj['type']} IF EXISTS {$obj['name']} cascade");
        }
        foreach ($afterStatements as $st) {
            self::dbExec($db, $st);
        }
    }
}