File "Connection.php"

Full Path: /home/warrior1/public_html/wp-content/plugins/file-manager/vendor/bitapps/wp-database/src/Connection.php
File size: 3.46 KB
MIME-type: text/x-php
Charset: utf-8

<?php

// phpcs:disable Squiz.NamingConventions.ValidVariableName.MemberNotCamelCaps
/**
 * Class For Database.
 */

namespace BitApps\WPDatabase;

use Exception;

/**
 * Wrapper for wpdb.
 *
 * @static $prefix $wpdb $prefix
 *
 * @see wpdb File: wp-includes/wp-db.php
 */
final class Connection
{
    public static $instance;

    private static $_previousErrorState;

    private static $_previousError;

    private static $_pluginPrefix;

    private static $_isLogEnabled = false;

    private static $_queries = [];

    private static $_errors = [];

    public function __get($var)
    {
        global $wpdb;
        if (property_exists($wpdb, $var)) {
            return $wpdb->{$var};
        }

        $instance = self::instance();

        if (property_exists($instance, $var)) {
            return $instance->{$var};
        }
    }

    public function __call($name, $arguments)
    {
        global $wpdb;

        if (method_exists($wpdb, $name)) {
            return $this->_forwadCall($wpdb, $name, $arguments);
        }
    }

    public static function __callStatic($name, $arguments)
    {
        global $wpdb;

        if (method_exists($wpdb, $name)) {
            return self::_forwadCall($wpdb, $name, $arguments);
        }

        $instance = self::instance();

        if (method_exists($instance, $name)) {
            return \call_user_func_array([$instance, $name], $arguments);
        }

        throw new Exception("Method {$name} not found");
    }

    public static function suppressError()
    {
        global $wpdb;
        self::$_previousError      = $wpdb->last_error;
        self::$_previousErrorState = $wpdb->suppress_errors;
        $wpdb->last_error          = '';
        $wpdb->suppress_errors     = true;
    }

    public static function restoreErrorState()
    {
        global $wpdb;
        $wpdb->last_error      = self::$_previousError;
        $wpdb->suppress_errors = self::$_previousErrorState;
    }

    public static function setPluginPrefix($prefix)
    {
        self::$_pluginPrefix = $prefix;
    }

    public static function getPrefix()
    {
        return self::wpPrefix() . self::$_pluginPrefix;
    }

    public static function wpPrefix()
    {
        return self::prop('prefix');
    }

    public static function enableQuery()
    {
        self::$_isLogEnabled = true;
    }

    public static function queries()
    {
        return self::$_queries;
    }

    public static function errors()
    {
        return self::$_errors;
    }

    public static function prop($var)
    {
        global $wpdb;
        if (property_exists($wpdb, $var)) {
            return $wpdb->{$var};
        }

        $instance = self::instance();

        if (property_exists($instance, $var)) {
            return $instance->{$var};
        }
    }

    private static function _forwadCall($instance, $method, $args)
    {
        if (self::$_isLogEnabled && $method === 'query') {
            Connection::suppressError();
            $returnedData     = \call_user_func_array([$instance, $method], $args);
            self::$_queries[] = self::prop('last_query');
            self::$_errors[]  = self::prop('last_error');
            Connection::restoreErrorState();

            return $returnedData;
        }

        return \call_user_func_array([$instance, $method], $args);
    }

    private static function instance()
    {
        if (\is_null(self::$instance)) {
            self::$instance = new static();
        }

        return self::$instance;
    }
}