manager.php

Go to the documentation of this file.

<?php
class XoopsDatabaseManager
{
    private $s_tables = array();

    private $f_tables = array();

    public $db;

    public $successStrings = array();

    public $failureStrings = array();

    public function __construct()
    {
        $xoops = Xoops::getInstance();
        $xoops->db();
        global $xoopsDB;
        $this->db = $xoopsDB;
        $this->db->setPrefix(\XoopsBaseConfig::get('db-prefix'));
        $this->successStrings = array(
            'create' => XoopsLocale::SF_TABLE_CREATED,
            'insert' => XoopsLocale::SF_ENTRIES_INSERTED_TO_TABLE,
            'alter'  => XoopsLocale::SF_TABLE_UPDATED,
            'drop'   => XoopsLocale::SF_TABLE_DROPPED,
        );
        $this->failureStrings = array(
            'create' => XoopsLocale::EF_TABLE_NOT_CREATED,
            'insert' => XoopsLocale::EF_ENTRIES_NOT_INSERTED_TO_TABLE,
            'alter'  => XoopsLocale::EF_TABLE_NOT_UPDATED,
            'drop'   => XoopsLocale::EF_TABLE_NOT_DROPPED,
        );
    }

    public function isConnectable()
    {
        return ($this->db->connect(false) != false) ? true : false;
    }

    public function dbExists()
    {
        return ($this->db->connect() != false) ? true : false;
    }

    public function createDB()
    {
        $this->db->connect(false);

        $result = $this->db->query("CREATE DATABASE " . \XoopsBaseConfig::get('db-name'));

        return ($result != false) ? true : false;
    }

    public function queryFromFile($sql_file_path, $force = false)
    {
        if (!XoopsLoad::fileExists($sql_file_path)) {
            return false;
        }
        $queryFunc = (bool)$force ? "queryF" : "query";
        $sql_query = trim(fread(fopen($sql_file_path, 'r'), filesize($sql_file_path)));
        SqlUtility::splitMySqlFile($pieces, $sql_query);
        $this->db->connect();
        foreach ($pieces as $piece) {
            $piece = trim($piece);
            // [0] contains the prefixed query
            // [4] contains unprefixed table name
            $prefixed_query = SqlUtility::prefixQuery($piece, $this->db->prefix());
            if ($prefixed_query != false) {
                $table = $this->db->prefix($prefixed_query[4]);
                if ($prefixed_query[1] == 'CREATE TABLE') {
                    if ($this->db->$queryFunc($prefixed_query[0]) != false) {
                        if (!isset($this->s_tables['create'][$table])) {
                            $this->s_tables['create'][$table] = 1;
                        }
                    } else {
                        if (!isset($this->f_tables['create'][$table])) {
                            $this->f_tables['create'][$table] = 1;
                        }
                    }
                } else {
                    if ($prefixed_query[1] == 'INSERT INTO') {
                        if ($this->db->$queryFunc($prefixed_query[0]) != false) {
                            if (!isset($this->s_tables['insert'][$table])) {
                                $this->s_tables['insert'][$table] = 1;
                            } else {
                                $this->s_tables['insert'][$table]++;
                            }
                        } else {
                            if (!isset($this->f_tables['insert'][$table])) {
                                $this->f_tables['insert'][$table] = 1;
                            } else {
                                $this->f_tables['insert'][$table]++;
                            }
                        }
                    } else {
                        if ($prefixed_query[1] == 'ALTER TABLE') {
                            if ($this->db->$queryFunc($prefixed_query[0]) != false) {
                                if (!isset($this->s_tables['alter'][$table])) {
                                    $this->s_tables['alter'][$table] = 1;
                                }
                            } else {
                                if (!isset($this->s_tables['alter'][$table])) {
                                    $this->f_tables['alter'][$table] = 1;
                                }
                            }
                        } else {
                            if ($prefixed_query[1] == 'DROP TABLE') {
                                if ($this->db->$queryFunc('DROP TABLE ' . $table) != false) {
                                    if (!isset($this->s_tables['drop'][$table])) {
                                        $this->s_tables['drop'][$table] = 1;
                                    }
                                } else {
                                    if (!isset($this->s_tables['drop'][$table])) {
                                        $this->f_tables['drop'][$table] = 1;
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
        return true;
    }

    public function report()
    {
        $commands = array('create', 'insert', 'alter', 'drop');
        $content = '<ul class="log">';
        foreach ($commands as $cmd) {
            if (!@empty($this->s_tables[$cmd])) {
                foreach ($this->s_tables[$cmd] as $key => $val) {
                    $content .= '<li class="success">';
                    $content .= ($cmd != 'insert')
                        ? sprintf($this->successStrings[$cmd], $key)
                        : sprintf($this->successStrings[$cmd], $val, $key);
                    $content .= "</li>\n";
                }
            }
        }
        foreach ($commands as $cmd) {
            if (!@empty($this->f_tables[$cmd])) {
                foreach ($this->f_tables[$cmd] as $key => $val) {
                    $content .= '<li class="failure">';
                    $content .= ($cmd != 'insert')
                        ? sprintf($this->failureStrings[$cmd], $key)
                        : sprintf($this->failureStrings[$cmd], $val, $key);
                    $content .= "</li>\n";
                }
            }
        }
        $content .= '</ul>';
        return $content;
    }

    public function query($sql)
    {
        $this->db->connect();
        return $this->db->query($sql);
    }

    public function prefix($table)
    {
        $this->db->connect();
        return $this->db->prefix($table);
    }

    public function fetchArray($ret)
    {
        $this->db->connect();
        return $this->db->fetchArray($ret);
    }

    public function insert($table, $query)
    {
        $this->db->connect();
        $table = $this->db->prefix($table);
        $query = 'INSERT INTO ' . $table . ' ' . $query;
        if (!$this->db->queryF($query)) {
            if (!isset($this->f_tables['insert'][$table])) {
                $this->f_tables['insert'][$table] = 1;
            } else {
                $this->f_tables['insert'][$table]++;
            }
            return false;
        } else {
            if (!isset($this->s_tables['insert'][$table])) {
                $this->s_tables['insert'][$table] = 1;
            } else {
                $this->s_tables['insert'][$table]++;
            }
            return $this->db->getInsertId();
        }
    }

    public function isError()
    {
        return (isset($this->f_tables)) ? true : false;
    }

    public function deleteTables($tables)
    {
        $deleted = array();
        $this->db->connect();
        foreach ($tables as $key => $val) {
            //was: if (!$this->db->query("DROP TABLE " . $this->db->prefix($key))) {
            if (!$this->db->query("DROP TABLE " . $this->db->prefix($val))) {
                $deleted[] = $val;
            }
        }
        return $deleted;
    }

    public function tableExists($table)
    {
        $table = trim($table);
        $ret = false;
        if ($table != '') {
            $this->db->connect();
            $sql = 'SELECT COUNT(*) FROM ' . $this->db->prefix($table);
            $ret = (false != $this->db->query($sql)) ? true : false;
        }
        return $ret;
    }

    public function copyFields(
        $fieldsMap,
        $oTableName,
        $nTableName,
        $dropTable = false
    ) {
        $sql = "SHOW COLUMNS FROM " . $this->db->prefix($oTableName);
        $result = $this->db->queryF($sql);
        if (($rows = $this->db->getRowsNum($result)) == count($fieldsMap)) {
            $sql = "SELECT * FROM " . $this->db->prefix($oTableName);
            $result = $this->db->queryF($sql);
            while ($myrow = $this->db->fetchArray($result)) {
                ksort($fieldsMap);
                ksort($myrow);
                $sql = "INSERT INTO `" . $this->db->prefix($nTableName)
                    . "` " . "(`" . implode("`,`", $fieldsMap) . "`)" .
                    " VALUES ('" . implode("','", $myrow) . "')";

                $this->db->queryF($sql);
            }
            if ($dropTable) {
                $sql = "DROP TABLE " . $this->db->prefix($oTableName);
                $this->db->queryF($sql);
            }
        }
    }
}