<?php

class Red_Database_420 extends Red_Database_Upgrader {
	/**
	 * @return array<string, string>
	 */
	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',
		];
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	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`" );
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	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`" );
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	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;
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	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;
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	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;
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	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;
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	protected function add_extra_logging( $wpdb ) {
		if ( $this->has_log_domain( $wpdb ) ) {
			return true;
		}

		// Update any URL with a double slash at the end
		$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" );
	}

	/**
	 * @param \wpdb $wpdb
	 * @return bool
	 */
	protected function add_extra_404( $wpdb ) {
		if ( $this->has_404_domain( $wpdb ) ) {
			return true;
		}

		// Update any URL with a double slash at the end
		$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`" );

		// Same as log table
		$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" );
	}
}
