Tiny File Manager

Back Back Up Advanced Editor

<?php

class Red_Database_420 extends Red_Database_Upgrader
{
    public function get_stages()
    {
        return ['add_extra_logging' => 'Add extra logging support', 'remove_module_id' => 'Remove module ID from logs', 'remove_group_id' => 'Remove group ID from logs', 'add_extra_404' => 'Add extra 404 logging support'];
    }
    protected function remove_module_id($wpdb)
    {
        if (!$this->has_module_id($wpdb)) {
            return true;
        }
        return $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` DROP `module_id`");
    }
    protected function remove_group_id($wpdb)
    {
        if (!$this->has_group_id($wpdb)) {
            return true;
        }
        return $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` DROP `group_id`");
    }
    private function has_module_id($wpdb)
    {
        $existing = $wpdb->get_row("SHOW CREATE TABLE `{$wpdb->prefix}redirection_logs`", ARRAY_N);
        if (isset($existing[1]) && strpos(strtolower($existing[1]), 'module_id') !== false) {
            return true;
        }
        return false;
    }
    private function has_group_id($wpdb)
    {
        $existing = $wpdb->get_row("SHOW CREATE TABLE `{$wpdb->prefix}redirection_logs`", ARRAY_N);
        if (isset($existing[1]) && strpos(strtolower($existing[1]), 'group_id') !== false) {
            return true;
        }
        return false;
    }
    private function has_log_domain($wpdb)
    {
        $existing = $wpdb->get_row("SHOW CREATE TABLE `{$wpdb->prefix}redirection_logs`", ARRAY_N);
        if (isset($existing[1]) && strpos(strtolower($existing[1]), 'domain` varchar') !== false) {
            return true;
        }
        return false;
    }
    private function has_404_domain($wpdb)
    {
        $existing = $wpdb->get_row("SHOW CREATE TABLE `{$wpdb->prefix}redirection_404`", ARRAY_N);
        if (isset($existing[1]) && strpos(strtolower($existing[1]), 'domain` varchar') !== false) {
            return true;
        }
        return false;
    }
    protected function add_extra_logging($wpdb)
    {
        if ($this->has_log_domain($wpdb)) {
            return true;
        }
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `domain` VARCHAR(255) NULL DEFAULT NULL AFTER `url`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `http_code` INT(11) unsigned NOT NULL DEFAULT 0 AFTER `referrer`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `request_method` VARCHAR(10) NULL DEFAULT NULL AFTER `http_code`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `redirect_by` VARCHAR(50) NULL DEFAULT NULL AFTER `request_method`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `request_data` MEDIUMTEXT NULL DEFAULT NULL AFTER `request_method`");
        return $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` CHANGE COLUMN `agent` `agent` MEDIUMTEXT NULL");
    }
    protected function add_extra_404($wpdb)
    {
        if ($this->has_404_domain($wpdb)) {
            return true;
        }
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `domain` VARCHAR(255) NULL DEFAULT NULL AFTER `url`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `http_code` INT(11) unsigned NOT NULL DEFAULT 0 AFTER `referrer`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `request_method` VARCHAR(10) NULL DEFAULT NULL AFTER `http_code`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `request_data` MEDIUMTEXT NULL DEFAULT NULL AFTER `request_method`");
        $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` DROP INDEX `url`");
        return $this->do_query($wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` CHANGE COLUMN `url` `url` MEDIUMTEXT NOT NULL");
    }
}