redirection / database / schema / 420.php

420.php in Redirection 5.0, at database/schema/420.php

100 lines 3.8 KB

Raw Download Zip

1	<?php
2
3	class Red_Database_420 extends Red_Database_Upgrader {
4	public function get_stages() {
5	return [
6	'add_extra_logging' => 'Add extra logging support',
7	'remove_module_id' => 'Remove module ID from logs',
8	'remove_group_id' => 'Remove group ID from logs',
9	'add_extra_404' => 'Add extra 404 logging support',
10	];
11	}
12
13	protected function remove_module_id( $wpdb ) {
14	if ( ! $this->has_module_id( $wpdb ) ) {
15	return true;
16	}
17
18	return $this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` DROP `module_id`" );
19	}
20
21	protected function remove_group_id( $wpdb ) {
22	if ( ! $this->has_group_id( $wpdb ) ) {
23	return true;
24	}
25
26	return $this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` DROP `group_id`" );
27	}
28
29	private function has_module_id( $wpdb ) {
30	$existing = $wpdb->get_row( "SHOW CREATE TABLE `{$wpdb->prefix}redirection_logs`", ARRAY_N );
31
32	if ( isset( $existing[1] ) && strpos( strtolower( $existing[1] ), 'module_id' ) !== false ) {
33	return true;
34	}
35
36	return false;
37	}
38
39	private function has_group_id( $wpdb ) {
40	$existing = $wpdb->get_row( "SHOW CREATE TABLE `{$wpdb->prefix}redirection_logs`", ARRAY_N );
41
42	if ( isset( $existing[1] ) && strpos( strtolower( $existing[1] ), 'group_id' ) !== false ) {
43	return true;
44	}
45
46	return false;
47	}
48
49	private function has_log_domain( $wpdb ) {
50	$existing = $wpdb->get_row( "SHOW CREATE TABLE `{$wpdb->prefix}redirection_logs`", ARRAY_N );
51
52	if ( isset( $existing[1] ) && strpos( strtolower( $existing[1] ), 'domain` varchar' ) !== false ) {
53	return true;
54	}
55
56	return false;
57	}
58
59	private function has_404_domain( $wpdb ) {
60	$existing = $wpdb->get_row( "SHOW CREATE TABLE `{$wpdb->prefix}redirection_404`", ARRAY_N );
61
62	if ( isset( $existing[1] ) && strpos( strtolower( $existing[1] ), 'domain` varchar' ) !== false ) {
63	return true;
64	}
65
66	return false;
67	}
68
69	protected function add_extra_logging( $wpdb ) {
70	if ( $this->has_log_domain( $wpdb ) ) {
71	return true;
72	}
73
74	// Update any URL with a double slash at the end
75	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `domain` VARCHAR(255) NULL DEFAULT NULL AFTER `url`" );
76	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `http_code` INT(11) unsigned NOT NULL DEFAULT 0 AFTER `referrer`" );
77	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `request_method` VARCHAR(10) NULL DEFAULT NULL AFTER `http_code`" );
78	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `redirect_by` VARCHAR(50) NULL DEFAULT NULL AFTER `request_method`" );
79	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` ADD `request_data` MEDIUMTEXT NULL DEFAULT NULL AFTER `request_method`" );
80
81	return $this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_logs` CHANGE COLUMN `agent` `agent` MEDIUMTEXT NULL" );
82	}
83
84	protected function add_extra_404( $wpdb ) {
85	if ( $this->has_404_domain( $wpdb ) ) {
86	return true;
87	}
88
89	// Update any URL with a double slash at the end
90	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `domain` VARCHAR(255) NULL DEFAULT NULL AFTER `url`" );
91	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `http_code` INT(11) unsigned NOT NULL DEFAULT 0 AFTER `referrer`" );
92	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `request_method` VARCHAR(10) NULL DEFAULT NULL AFTER `http_code`" );
93	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` ADD `request_data` MEDIUMTEXT NULL DEFAULT NULL AFTER `request_method`" );
94
95	// Same as log table
96	$this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` DROP INDEX `url`" );
97	return $this->do_query( $wpdb, "ALTER TABLE `{$wpdb->prefix}redirection_404` CHANGE COLUMN `url` `url` MEDIUMTEXT NOT NULL" );
98	}
99	}
100