<?php

// Note: not localised as the messages aren't important enough
class Red_Database_231 extends Red_Database_Upgrader {
	/**
	 * @return array<string, string>
	 */
	public function get_stages() {
		return [
			'remove_404_module_231' => 'Remove 404 module',
			'create_404_table_231' => 'Create 404 table',
		];
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	protected function remove_404_module_231( $wpdb ) {
		return $this->do_query( $wpdb, "UPDATE {$wpdb->prefix}redirection_groups SET module_id=1 WHERE module_id=3" );
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	protected function create_404_table_231( $wpdb ) {
		$this->do_query( $wpdb, $this->get_404_table( $wpdb ) );
		return true;
	}

	/**
	 * @param \wpdb $wpdb
	 * @return string
	 */
	private function get_404_table( $wpdb ) {
		$charset_collate = $this->get_charset();

		return "CREATE TABLE `{$wpdb->prefix}redirection_404` (
			`id` int(11) unsigned NOT NULL AUTO_INCREMENT,
			`created` datetime NOT NULL,
			`url` varchar(255) NOT NULL DEFAULT '',
			`agent` varchar(255) DEFAULT NULL,
			`referrer` varchar(255) DEFAULT NULL,
			`ip` int(10) unsigned NOT NULL,
			PRIMARY KEY (`id`),
			KEY `created` (`created`),
			KEY `url` (`url`(191)),
			KEY `ip` (`ip`),
			KEY `referrer` (`referrer`(191))
		  ) $charset_collate";
	}
}