<?php
/**
 * database check and restoration
 * This file is for temporary use and will be removed or changed in the future release.
 */
class DatabaseMaintenance {
	private function sanity_check() {
		global $wpdb;
		$results_table = array();
		$columns_to_check = array(
				$wpdb->prefix.'commentmeta' => array(
						'comment_id' => '\'0\'',
						'meta_key'   => 'NULL'
				),
				$wpdb->prefix.'comments' => array(
						'comment_post_ID'      => '\'0\'',
						'comment_author_email' => '\'\'',
						'comment_author_url'   => '\'\'',
						'comment_author_IP'    => '\'\'',
						'comment_date_gmt'     => '\'0000-00-00 00:00:00\'',
						'comment_date'         => '\'0000-00-00 00:00:00\'',
						'comment_karma'        => '\'0\'',
						'comment_approved'     => '\'1\'',
						'comment_agent'        => '\'\'',
						'comment_type'         => '\'\'',
						'comment_parent'       => '\'0\'',
						'user_id'              => '\'0\''
				),
				$wpdb->prefix.'links' => array(
						'link_url'         => '\'\'',
						'link_name'        => '\'\'',
						'link_image'       => '\'\'',
						'link_target'      => '\'\'',
						'link_description' => '\'\'',
						'link_visible'     => '\'Y\'',
						'link_owner'       => '\'1\'',
						'link_rating'      => '\'0\'',
						'link_updated'     => '\'0000-00-00 00:00:00\'',
						'link_rel'         => '\'\'',
						'link_rss'         => '\'\''
				),
				$wpdb->prefix.'options' => array(
						'option_name' => '\'\'',
						'autoload' => '\'yes\''
				),
				$wpdb->prefix.'postmeta' => array(
						'post_id' => '\'0\'',
						'meta_key' => 'NULL'
				),
				$wpdb->prefix.'posts' => array(
						'post_author' => '\'0\'',
						'post_date_gmt' => '\'0000-00-00 00:00:00\'',
						'post_date' => '\'0000-00-00 00:00:00\'',
						'post_status' => '\'publish\'',
						'comment_status' => '\'open\'',
						'ping_status' => '\'open\'',
						'post_password' => '\'\'',
						'post_name' => '\'\'',
						'post_modified_gmt' => '\'0000-00-00 00:00:00\'',
						'post_modified' => '\'0000-00-00 00:00:00\'',
						'post_parent' => '\'0\'',
						'guid' => '\'\'',
						'menu_order' => '\'0\'',
						'post_type' => '\'post\'',
						'post_mime_type' => '\'\'',
						'comment_count' => '\'0\''
				),
				$wpdb->prefix.'term_relationships' => array(
						'term_order' => '0'
				),
				$wpdb->prefix.'term_taxonomy' => array(
						'taxonomy' => '\'\'',
						'parent' => '0',
						'count' => '0'
				),
				$wpdb->prefix.'terms' => array(
						'name' => '\'\'',
						'slug' => '\'\'',
						'term_group' => '0'
				),
				$wpdb->prefix.'users' => array(
						'user_login' => '\'\'',
						'user_pass' => '\'\'',
						'user_nicename' => '\'\'',
						'user_email' => '\'\'',
						'user_url' => '\'\'',
						'user_registered' => '\'0000-00-00 00:00:00\'',
						'user_activation_key' => '\'\'',
						'user_status' => '\'0\'',
						'display_name' => '\'\'',
						// for network install
						'spam' => '\'0\'',
						'deleted' => '\'0\''
				),
				$wpdb->prefix.'usermeta' => array(
						'user_id' => '\'0\'',
						'meta_key' => 'NULL',
				),
				// for network install
				$wpdb->prefix.'blog_versions' => array(
						'blog_id' => '\'0\'',
						'db_version' => '\'\'',
						'last_updated' => '\'0000-00-00 00:00:00\''
				),
				$wpdb->prefix.'blogs' => array(
						'site_id' => '\'0\'',
						'domain' => '\'\'',
						'path' => '\'\'',
						'registered' => '\'0000-00-00 00:00:00\'',
						'last_updated' => '\'0000-00-00 00:00:00\'',
						'public' => '\'1\'',
						'mature' => '\'0\'',
						'spam' => '\'0\'',
						'deleted' => '\'0\'',
						'lang_id' => '\'0\''
				),
				$wpdb->prefix.'registration_log' => array(
						'email' => '\'\'',
						'IP' => '\'\'',
						'blog_id' => '\'0\'',
						'date_registered' => '\'0000-00-00 00:00:00\''
				),
				$wpdb->prefix.'signups' => array(
						'domain' => '\'\'',
						'path' => '\'\'',
						'user_login' => '\'\'',
						'user_email' => '\'\'',
						'registered' => '\'0000-00-00 00:00:00\'',
						'activated' => '\'0000-00-00 00:00:00\'',
						'active' => '\'0\'',
						'activation_key' => '\'\'',
				),
				$wpdb->prefix.'site' => array(
						'domain' => '\'\'',
						'path' => '\'\''
				),
				$wpdb->prefix.'sitemeta' => array(
						'site_id' => '\'0\'',
						'meta_key' => 'NULL',
				)
		);
		$tables = $wpdb->tables('all');
		foreach ($tables as $table) {
			$col_infos = $wpdb->get_results("SHOW COLUMNS FROM $table");
			foreach ($col_infos as $col) {
				if (array_key_exists($col->Field, $columns_to_check[$table]) && $col->Default != $columns_to_check[$table][$col->Field]) {
					$results_table[$table] = 'damaged';
					break;
				}
			}
		}
		if (empty($results_table)) {
			return true;
		} else {
			return $results_table;
		}
	}
	private function do_fix_database() {
		global $wpdb, $wp_version, $utils;
		$global_schema_to_change = array(
				$wpdb->prefix.'commentmeta' => array(
						"comment_id bigint(20) unsigned NOT NULL default '0'",
						"meta_key varchar(255) default NULL",
						"meta_value longtext"
				),
				$wpdb->prefix.'comments' => array(
						"comment_post_ID bigint(20) unsigned NOT NULL default '0'",
						"comment_author_email varchar(100) NOT NULL default ''",
						"comment_author_url varchar(200) NOT NULL default ''",
						"comment_author_IP varchar(100) NOT NULL default ''",
						"comment_date datetime NOT NULL default '0000-00-00 00:00:00'",
						"comment_date_gmt datetime NOT NULL default '0000-00-00 00:00:00'",
						"comment_karma int(11) NOT NULL default '0'",
						"comment_approved varchar(20) NOT NULL default '1'",
						"comment_agent varchar(255) NOT NULL default ''",
						"comment_type varchar(20) NOT NULL default ''",
						"comment_parent bigint(20) unsigned NOT NULL default '0'",
						"user_id bigint(20) unsigned NOT NULL default '0'"
				),
				$wpdb->prefix.'links' => array(
						"link_url varchar(255) NOT NULL default ''",
						"link_name varchar(255) NOT NULL default ''",
						"link_image varchar(255) NOT NULL default ''",
						"link_target varchar(25) NOT NULL default ''",
						"link_description varchar(255) NOT NULL default ''",
						"link_visible varchar(20) NOT NULL default 'Y'",
						"link_owner bigint(20) unsigned NOT NULL default '1'",
						"link_rating int(11) NOT NULL default '0'",
						"link_updated datetime NOT NULL default '0000-00-00 00:00:00'",
						"link_rel varchar(255) NOT NULL default ''",
						"link_notes mediumtext NOT NULL",
						"link_rss varchar(255) NOT NULL default ''"
				),
				$wpdb->prefix.'options' => array(
						"option_name varchar(64) NOT NULL default ''",
						"option_value longtext NOT NULL",
						"autoload varchar(20) NOT NULL default 'yes'"
				),
				$wpdb->prefix.'postmeta' => array(
						"post_id bigint(20) unsigned NOT NULL default '0'",
						"meta_key varchar(255) default NULL",
						"meta_value longtext"
				),
				$wpdb->prefix.'posts' => array(
						"post_author bigint(20) unsigned NOT NULL default '0'",
						"post_date datetime NOT NULL default '0000-00-00 00:00:00'",
						"post_date_gmt datetime NOT NULL default '0000-00-00 00:00:00'",
						"post_status varchar(20) NOT NULL default 'publish'",
						"comment_status varchar(20) NOT NULL default 'open'",
						"ping_status varchar(20) NOT NULL default 'open'",
						"post_password varchar(20) NOT NULL default ''",
						"post_name varchar(200) NOT NULL default ''",
						"post_modified datetime NOT NULL default '0000-00-00 00:00:00'",
						"post_modified_gmt datetime NOT NULL default '0000-00-00 00:00:00'",
						"post_content_filtered longtext NOT NULL",
						"post_parent bigint(20) unsigned NOT NULL default '0'",
						"guid varchar(255) NOT NULL default ''",
						"menu_order int(11) NOT NULL default '0'",
						"post_type varchar(20) NOT NULL default 'post'",
						"post_mime_type varchar(100) NOT NULL default ''",
						"comment_count bigint(20) NOT NULL default '0'"
				),
				$wpdb->prefix.'term_relationships' => array(
						"term_order int(11) NOT NULL default 0"
				),
				$wpdb->prefix.'term_taxonomy' => array(
						"taxonomy varchar(32) NOT NULL default ''",
						"description longtext NOT NULL",
						"parent bigint(20) unsigned NOT NULL default 0",
						"count bigint(20) NOT NULL default 0"
				),
				$wpdb->prefix.'terms' => array(
						"name varchar(200) NOT NULL default ''",
						"slug varchar(200) NOT NULL default ''",
						"term_group bigint(10) NOT NULL default 0"
				),
				$wpdb->prefix.'users' => array(
						"user_login varchar(60) NOT NULL default ''",
						"user_pass varchar(64) NOT NULL default ''",
						"user_nicename varchar(50) NOT NULL default ''",
						"user_email varchar(100) NOT NULL default ''",
						"user_url varchar(100) NOT NULL default ''",
						"user_registered datetime NOT NULL default '0000-00-00 00:00:00'",
						"user_activation_key varchar(60) NOT NULL default ''",
						"user_status int(11) NOT NULL default '0'",
						"display_name varchar(250) NOT NULL default ''"
				),
				$wpdb->prefix.'usermeta' => array(
						"user_id bigint(20) unsigned NOT NULL default '0'",
						"meta_key varchar(255) default NULL",
						"meta_value longtext"
				)
		);

		$network_schema_to_change = array(
				$wpdb->prefix.'blog_versions' => array(
						"blog_id bigint(20) NOT NULL default '0'",
						"db_version varchar(20) NOT NULL default ''",
						"last_updated datetime NOT NULL default '0000-00-00 00:00:00'"
				),
				$wpdb->prefix.'blogs' => array(
						"site_id bigint(20) NOT NULL default '0'",
						"domain varchar(200) NOT NULL default ''",
						"path varchar(100) NOT NULL default ''",
						"registered datetime NOT NULL default '0000-00-00 00:00:00'",
						"last_updated datetime NOT NULL default '0000-00-00 00:00:00'",
						"public tinyint(2) NOT NULL default '1'",
						"mature tinyint(2) NOT NULL default '0'",
						"spam tinyint(2) NOT NULL default '0'",
						"deleted tinyint(2) NOT NULL default '0'",
						"lang_id int(11) NOT NULL default '0'"
				),
				$wpdb->prefix.'registration_log' => array(
						"email varchar(255) NOT NULL default ''",
						"IP varchar(30) NOT NULL default ''",
						"blog_id bigint(20) NOT NULL default '0'",
						"date_registered datetime NOT NULL default '0000-00-00 00:00:00'"
				),
				$wpdb->prefix.'signups' => array(
						"domain varchar(200) NOT NULL default ''",
						"path varchar(100) NOT NULL default ''",
						"title longtext NOT NULL",
						"user_login varchar(60) NOT NULL default ''",
						"user_email varchar(100) NOT NULL default ''",
						"registered datetime NOT NULL default '0000-00-00 00:00:00'",
						"activated datetime NOT NULL default '0000-00-00 00:00:00'",
						"active tinyint(1) NOT NULL default '0'",
						"activation_key varchar(50) NOT NULL default ''",
						"meta longtext"
				),
				$wpdb->prefix.'site' => array(
						"domain varchar(200) NOT NULL default ''",
						"path varchar(100) NOT NULL default ''"
				),
				$wpdb->prefix.'sitemeta' => array(
						"site_id bigint(20) NOT NULL default '0'",
						"meta_key varchar(255) default NULL",
						"meta_value longtext"
				),
				$wpdb->prefix.'users' => array(
						"user_login varchar(60) NOT NULL default ''",
						"spam tinyint(2) NOT NULL default '0'",
						"deleted tinyint(2) NOT NULL default '0'"
				)
		);
		if (version_compare($wp_version, '3.6', '<')) return false;
		$results = $this->sanity_check();
		$return_val = array();
		$queries = array();
		if ($results !== true) {
			if (!$this->maintenance_backup()) {
				$message = __('Can\'t create backup file.', $domain);
				echo $message;
				return false;
			}
			$tables = array_keys($results);
			foreach ($tables as $table) {
				if (key_exists($table, $global_schema_to_change)) {
					$queries = $global_schema_to_change[$table];
				}
				if (key_exists($table, $network_schema_to_change)) {
					if (!empty($queries)) {
						$queries = array_merge($queries, $network_schema_to_change[$table]);
					} else {
						$queries = $network_schema_to_change[$table];
					}
				}
				foreach ($queries as $query) {
					$sql = 'ALTER TABLE' . ' ' . $table . ' ' . 'CHANGE COLUMN' . ' ' . $query;
					$res = $wpdb->query($sql);
					if ($res === false) {
						$return_val[] = __('Failed: ', $domain) . $query;
					}
				}
			}
		} else {
			return true;
		}
		if (empty($return_val)) {
			return true;
		} else {
			return $return_val;
		}
	}
	
	private function show_columns() {
		global $wpdb, $utils;
		$domain = $utils->text_domain;
		$tables = $wpdb->tables('all');
		if (!isset($_POST['table'])) {
			$message = __('Table name is not selected.', $domain);
			return $message;
		} elseif (!in_array($_POST['table'], $tables)) {
			$message = __('There\'s no such table', $domain);
			return $message;
		}	else {
			$table_name = $_POST['table'];
			$results = $wpdb->get_results("SHOW COLUMNS FROM $table_name");
			return $results;
		}
	}

	private function maintenance_backup() {
		$result = array();
		$database_file = FQDB;
		$db_name = basename(FQDB);
		if (!file_exists($database_file)) {
			return false;
		}
		$today = date("Ymd-His");
		if (!extension_loaded('zip')) {
			$backup_file = $database_file . '.' . $today . '.maintenance-backup';
			if (copy($database_file, $backup_file)) {
				$result = true;
			} else {
				$result = false;
			}
		} else {
			$backup_file = $database_file . '.' . $today . '.maintenance-backup.zip';
			$zip = new ZipArchive();
			$res = $zip->open($backup_file, ZipArchive::CREATE | ZipArchive::OVERWRITE);
			if ($res === true) {
				$zip->addFile($database_file, $db_name);
				$result = true;
			} else {
				$result = false;
			}
			$zip->close();
		}
		return $result;
	}

	function show_maintenance_page() {
		global $utils;
		$domain = $utils->text_domain;
		if (is_multisite() && !current_user_can('manage_network_options')) {
			die(__('You are not allowed to access this page!', $domain));
		} elseif (!current_user_can('manage_options')) {
			die(__('You are not allowed to access this page!', $domain));
		}


    if (isset($_GET['page']) && $_GET['page'] == 'maintenance') : ?>
    <div class="navigation">
      <ul class="navi-menu">
        <li class="menu-item"><a href="<?php echo $utils->show_parent();?>?page=doc"><?php _e('Documentation', $domain);?></a></li>
        <li class="menu-item"><a href="<?php echo $utils->show_parent();?>?page=sys-info"><?php _e('System Info', $domain);?></a></li>
        <li class="menu-item"><a href="<?php echo $utils->show_parent();?>?page=setting-file"><?php _e('Miscellaneous', $domain);?></a></li>
        <li class="menu-item"><a href="<?php echo $utils->show_parent();?>?page=patch"><?php _e('Patch Utility', $domain);?></a></li>
        <li class="menu-selected"><?php _e('Maintenance', $domain);?></li>
      </ul>
    </div>
    <div class="wrap" id="sqlite-admin-wrap">
	    <h2><?php _e('Database Maintenace', $domain);?></h2>
	    <h3><?php _e('Important Notice', $domain);?></h3>
	    <p>
	    	<span style="color: red;"><?php _e('When you installed WordPress 3.5.x with SQLite Integration and upgraded to 3.6, your database might not function as expected.', $domain);?></span>
	    	<?php _e('This page provide you the database sanity check utility and the restore utility.', $domain);?>
	    </p>
	    <p>
	    	<?php _e('Click "Sanity Check" button first, and see if you need to fix database or not. If needed, click "Fix Database" button. Afterward you may go to Miscellaneous page and optimize database (this is not required).', $domain);?>
	    </p>
	    <p>
	    	<?php _e('Fix Database procedure will create a database backup file each time the button clicked. The backup file is named with "maintenance-backup", so you can remove it if you don\'t need it. Please go to Miscellaneous page and check if there is one.', $domain);?>
	    </p>
	    <p>
	    	<?php _e('If you installed WordPress 3.6 (not upgraded), you don\'t have to restore the database.', $domain);?>
	    </p>
	    
	    <form action="" method="post">
        <?php
        	if (function_exists('wp_nonce_field')) {
            wp_nonce_field('sqliteintegration-database-manip-stats');
          }
        ?>
	    	<input type="submit" name="sanity-check" class="button-primary" value="<?php _e('Sanity Check', $domain);?>" onclick="return confirm('<?php _e('Are you sure to check the database? This will take some time.\n\nClick [Cancel] to stop, [OK] to continue.', $domain);?>')" />
	    	<input type="submit" name="do-fix-database" class="button-primary" value="<?php _e('Fix database', $domain);?>" onclick="return confirm('<?php _e('Are you sure to do fix the database? This will take some time.\n\nClick [Cancel] to stop, [OK] to continue.', $domain);?>')" />
	    </form>
	   	<h3><?php _e('Columns Information', $domain);?></h3>
	   	<p>
	   		<?php _e('Select a table name and click "Display Columns" button, and you\'ll see the column property of that table. This information is for debug use.', $domain);?>
	   	</p>
	   	<form action="" method="post">
        <?php
        	if (function_exists('wp_nonce_field')) {
            wp_nonce_field('sqliteintegration-database-manip-stats');
          }
        ?>
	   		<label for="table"/>Table Name: </label>
	    	<select name="table" id="table">
	    		<?php foreach ($wp_tables as $table) :?>
	    		<option value="<?php echo $table;?>"><?php echo $table;?></option>
	    		<?php endforeach;?>
	    	</select>
	    	<input type="submit" name="show-columns" class="button-secondary" value="<?php _e('Display Columns', $domain);?>" onclick="return confirm('<?php _e('Display columns in the selected table.\n\nClick [Cancel] to stop, [OK] to continue.', $domain);?>')" />
	    </form>
	    </div>
    <?php endif;
    
    if (isset($_POST['do-fix-database'])) {
    	check_admin_referer('sqliteintegration-database-manip-stats');
    	if (is_multisite() && !current_user_can('manage_network_options')) {
    		die(__('You are not allowed to do this operation!', $domain));
    	} elseif (!current_user_can('manage_options')) {
    		die(__('You are not allowed to do this operation!', $domain));
    	}
    	$fix_results = $this->do_fix_database();
    	if ($fix_results !== true) {
				$title = '<h3>'. __('Results', $domain) . '</h3>';
				echo '<div class="wrap" id="sqlite-admin-side-wrap">';
				echo $title;
				echo '<ul>';
    		foreach ($fix_results as $result) {
    			echo '<li>' . $result . '</li>';
    		}
    		echo '</ul>';
    		echo '</div>';
    	} else {
				$title = '<h3>'. __('Results', $domain) . '</h3>';
				$message = __('Your database restoration is successfully finished!', $domain);
				echo '<div class="wrap" id="sqlite-admin-side-wrap">';
				echo $title;
				echo '<p>'.$message.'</p>';
				echo '</div>';
			}
    }
    if (isset($_POST['sanity-check'])) {
    	check_admin_referer('sqliteintegration-database-manip-stats');
    	if (is_multisite() && !current_user_can('manage_network_options')) {
    		die(__('You are not allowed to do this operation!', $domain));
    	} elseif (!current_user_can('manage_options')) {
    		die(__('You are not allowed to do this operation!', $domain));
    	}
    	$check_results = $this->sanity_check();
    	if ($check_results !== true) {
				$title = '<h3>'. __('Checked Results', $domain) . '</h3>';
				echo '<div class="wrap" id="sqlite-admin-side-wrap">';
				echo $title;
				echo '<ul>';
    		foreach ($check_results as $table => $damaged) {
    			$message = __(' needs restoring.', $domain);
    			echo '<li><span class="em">' . $table . '</span>' . $message . '</li>';
    		}
    		echo '</ul>';
    		echo '</div>';
    	} else {
				$title = '<h3>'. __('Checked Results', $domain) . '</h3>';
    		$message = __('Your database is OK. You don\'t have to restore it.', $domain);
				echo '<div class="wrap" id="sqlite-admin-side-wrap">';
    		echo $title;
				echo '<p>'.$message.'</p>';
    		echo '</div>';
    	}
    }
    if (isset($_POST['show-columns'])) {
			check_admin_referer('sqliteintegration-database-manip-stats');
    	if (is_multisite() && !current_user_can('manage_network_options')) {
    		die(__('You are not allowed to do this operation!', $domain));
    	} elseif (!current_user_can('manage_options')) {
    		die(__('You are not allowed to do this operation!', $domain));
    	}
    	$results = $this->show_columns();
    	if (is_array($results)) {
				$title = '<h3>'. sprintf(__('Columns In %s', $domain), $_POST['table']) . '</h3>';
				echo '<div class="wrap" id="sqlite-admin-side-wrap">';
				echo $title;
				echo '<table><thead><tr><th>Column</th><th>Type</th><th>Null</th><th>Default</th></tr></thead>';
				echo '<tbody>';
				foreach ($results as $column) {
					echo '<tr>';
					echo '<td>' . $column->Field . '</td>';
					echo '<td>' . $column->Type . '</td>';
					echo '<td>' . $column->Null . '</td>';
					echo '<td>' . $column->Default . '</td>';
					echo '</tr>';
				}
				echo '</tbody></table></div>';
			} else {
				$title = '<h3>'. __('Columns Info', $domain) . '</h3>';
				echo '<div class="wrap" id="sqlite-admin-side-wrap">';
				echo $title;
				echo '<ul>';
				echo $results;
				echo '</ul></div>';
			}
		}
	}
}
?>