Passed
Pull Request — dev/2.3.0 (#163)
by Maria Daniel Deepak
02:42
created

TableManager::get_create_table_query()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 20
Code Lines 6

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 2

Importance

Changes 0
Metric Value
cc 1
eloc 6
nc 1
nop 0
dl 0
loc 20
ccs 0
cts 7
cp 0
crap 2
rs 10
c 0
b 0
f 0
1
<?php namespace EmailLog\Core\DB;
2
3
/**
4
 * Handle installation and db table creation.
5
 */
6
use EmailLog\Core\Loadie;
7
use EmailLog\Util;
8
9
defined( 'ABSPATH' ) || exit; // Exit if accessed directly.
10
11
/**
12
 * Helper class to create table.
13
 *
14
 * @since 2.0.0
15
 */
16
class TableManager implements Loadie {
17
18
	/* Database table name */
19
	const LOG_TABLE_NAME = 'email_log';
20
21
	/* Database option name */
22
	const DB_OPTION_NAME = 'email-log-db';
23
24
	/* Database version */
25
	const DB_VERSION = '0.2';
26
27
	/**
28
	 * Setup hooks.
29
	 */
30
	public function load() {
31
		add_action( 'wpmu_new_blog', array( $this, 'create_table_for_new_blog' ) );
32
33
		add_filter( 'wpmu_drop_tables', array( $this, 'delete_table_from_deleted_blog' ) );
34
35
		// Do any DB upgrades.
36
		$this->update_table_if_needed();
37
	}
38
39
	/**
40
	 * On plugin activation, create table if needed.
41
	 *
42
	 * @param bool $network_wide True if the plugin was network activated.
43
	 */
44
	public function on_activate( $network_wide ) {
45
		if ( is_multisite() && $network_wide ) {
46
			// Note: if there are more than 10,000 blogs or
47
			// if `wp_is_large_network` filter is set, then this may fail.
48
			$sites = get_sites();
49
50
			foreach ( $sites as $site ) {
51
				switch_to_blog( $site['blog_id'] );
52
				$this->create_table_if_needed();
53
				restore_current_blog();
54
			}
55
		} else {
56
			$this->create_table_if_needed();
57
		}
58
	}
59
60
	/**
61
	 * Create email log table when a new blog is created.
62
	 *
63
	 * @param int $blog_id Blog Id.
64
	 */
65
	public function create_table_for_new_blog( $blog_id ) {
66
		if ( is_plugin_active_for_network( 'email-log/email-log.php' ) ) {
67
			switch_to_blog( $blog_id );
68
			$this->create_table_if_needed();
69
			restore_current_blog();
70
		}
71
	}
72
73
	/**
74
	 * Add email log table to the list of tables deleted when a blog is deleted.
75
	 *
76
	 * @param array $tables List of tables to be deleted.
77
	 *
78
	 * @return string[] $tables Modified list of tables to be deleted.
79
	 */
80 1
	public function delete_table_from_deleted_blog( $tables ) {
81 1
		$tables[] = $this->get_log_table_name();
82
83 1
		return $tables;
84
	}
85
86
	/**
87
	 * Get email log table name.
88
	 *
89
	 * @return string Email Log Table name.
90
	 */
91 2
	public function get_log_table_name() {
92 2
		global $wpdb;
93
94 2
		return $wpdb->prefix . self::LOG_TABLE_NAME;
95
	}
96
97
	/**
98
	 * Insert log data into DB.
99
	 *
100
	 * @param array $data Data to be inserted.
101
	 */
102
	public function insert_log( $data ) {
103
		global $wpdb;
104
105
		$table_name = $this->get_log_table_name();
106
		$wpdb->insert( $table_name, $data );
107
	}
108
109
	/**
110
	 * Delete log entries by ids.
111
	 *
112
	 * @param string $ids Comma separated list of log ids.
113
	 *
114
	 * @return false|int Number of log entries that got deleted. False on failure.
115
	 */
116
	public function delete_logs( $ids ) {
117
		global $wpdb;
118
119
		$table_name = $this->get_log_table_name();
120
121
		// Can't use wpdb->prepare for the below query. If used it results in this bug // https://github.com/sudar/email-log/issues/13.
122
		$ids = esc_sql( $ids );
123
124
		return $wpdb->query( "DELETE FROM {$table_name} where id IN ( {$ids} )" ); //@codingStandardsIgnoreLine
125
	}
126
127
	/**
128
	 * Delete all log entries.
129
	 *
130
	 * @return false|int Number of log entries that got deleted. False on failure.
131
	 */
132
	public function delete_all_logs() {
133
		global $wpdb;
134
135
		$table_name = $this->get_log_table_name();
136
137
		return $wpdb->query( "DELETE FROM {$table_name}" ); //@codingStandardsIgnoreLine
138
	}
139
140
	/**
141
	 * Deletes Email Logs older than the specified interval.
142
	 *
143
	 * @param int $interval_in_days No. of days beyond which logs are to be deleted.
144
	 *
145
	 * @return int $deleted_rows_count  Count of rows deleted.
146
	 */
147
	public function delete_logs_older_than( $interval_in_days ) {
148
		global $wpdb;
149
		$table_name = $this->get_log_table_name();
150
151
		$query              = $wpdb->prepare( "DELETE FROM {$table_name} WHERE sent_date < DATE_SUB( CURDATE(), INTERVAL %d DAY )", $interval_in_days );
152
		$deleted_rows_count = $wpdb->query( $query );
153
154
		return $deleted_rows_count;
155
	}
156
157
	/**
158
	 * Fetch log item by ID.
159
	 *
160
	 * @param array $ids Optional. Array of IDs of the log items to be retrieved.
161
	 *
162
	 * @return array Log item(s).
163
	 */
164
	public function fetch_log_items_by_id( $ids = array() ) {
165
		global $wpdb;
166
		$table_name = $this->get_log_table_name();
167
168
		$query = "SELECT * FROM {$table_name}";
169
170
		if ( ! empty( $ids ) ) {
171
			$ids = array_map( 'absint', $ids );
172
173
			// Can't use wpdb->prepare for the below query. If used it results in this bug https://github.com/sudar/email-log/issues/13.
174
			$ids_list = esc_sql( implode( ',', $ids ) );
175
176
			$query .= " where id IN ( {$ids_list} )";
177
		}
178
179
		return $wpdb->get_results( $query, 'ARRAY_A' ); //@codingStandardsIgnoreLine
180
	}
181
182
	/**
183
	 * Fetch log items.
184
	 *
185
	 * @param array $request         Request object.
186
	 * @param int   $per_page        Entries per page.
187
	 * @param int   $current_page_no Current page no.
188
	 *
189
	 * @return array Log entries and total items count.
190
	 */
191
	public function fetch_log_items( $request, $per_page, $current_page_no ) {
192
		global $wpdb;
193
		$table_name = $this->get_log_table_name();
194
195
		$query       = 'SELECT * FROM ' . $table_name;
196
		$count_query = 'SELECT count(*) FROM ' . $table_name;
197
		$query_cond  = '';
198
199
		if ( isset( $request['s'] ) && $request['s'] !== '' ) {
200
			$search_term = trim( esc_sql( $request['s'] ) );
0 ignored issues
show
Bug introduced by
It seems like esc_sql($request['s']) can also be of type array; however, parameter $str of trim() does only seem to accept string, maybe add an additional type check? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

200
			$search_term = trim( /** @scrutinizer ignore-type */ esc_sql( $request['s'] ) );
Loading history...
201
			$query_cond .= " WHERE ( to_email LIKE '%$search_term%' OR subject LIKE '%$search_term%' ) ";
202
		}
203
204
		if ( isset( $request['d'] ) && $request['d'] !== '' ) {
205
			$search_date = trim( esc_sql( $request['d'] ) );
206
			if ( '' === $query_cond ) {
207
				$query_cond .= " WHERE sent_date BETWEEN '$search_date 00:00:00' AND '$search_date 23:59:59' ";
208
			} else {
209
				$query_cond .= " AND sent_date BETWEEN '$search_date 00:00:00' AND '$search_date 23:59:59' ";
210
			}
211
		}
212
213
		// Ordering parameters.
214
		$orderby = ! empty( $request['orderby'] ) ? esc_sql( $request['orderby'] ) : 'sent_date';
215
		$order   = ! empty( $request['order'] ) ? esc_sql( $request['order'] ) : 'DESC';
216
217
		if ( ! empty( $orderby ) & ! empty( $order ) ) {
0 ignored issues
show
Bug introduced by
Are you sure you want to use the bitwise & or did you mean &&?
Loading history...
218
			$query_cond .= ' ORDER BY ' . $orderby . ' ' . $order;
0 ignored issues
show
Bug introduced by
Are you sure $orderby of type string|array can be used in concatenation? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

218
			$query_cond .= ' ORDER BY ' . /** @scrutinizer ignore-type */ $orderby . ' ' . $order;
Loading history...
Bug introduced by
Are you sure $order of type string|array can be used in concatenation? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

218
			$query_cond .= ' ORDER BY ' . $orderby . ' ' . /** @scrutinizer ignore-type */ $order;
Loading history...
219
		}
220
221
		// Find total number of items.
222
		$count_query = $count_query . $query_cond;
223
		$total_items = $wpdb->get_var( $count_query );
224
225
		// Adjust the query to take pagination into account.
226
		if ( ! empty( $current_page_no ) && ! empty( $per_page ) ) {
227
			$offset = ( $current_page_no - 1 ) * $per_page;
228
			$query_cond .= ' LIMIT ' . (int) $offset . ',' . (int) $per_page;
229
		}
230
231
		// Fetch the items.
232
		$query = $query . $query_cond;
233
		$items = $wpdb->get_results( $query );
234
235
		return array( $items, $total_items );
236
	}
237
238
	/**
239
	 * Create email log table.
240
	 *
241
	 * @access private
242
	 *
243
	 * @global object $wpdb
244
	 */
245
	private function create_table_if_needed() {
246
		global $wpdb;
247
248
		$table_name = $this->get_log_table_name();
249
250
		if ( $wpdb->get_var( "show tables like '{$table_name}'" ) != $table_name ) {
251
252
			$sql = $this->get_create_table_query();
253
254
			require_once ABSPATH . 'wp-admin/includes/upgrade.php';
255
			dbDelta( $sql );
256
257
			add_option( self::DB_OPTION_NAME, self::DB_VERSION );
258
		}
259
	}
260
261
	/**
262
	 * Get the total number of email logs.
263
	 *
264
	 * @return int Total email log count
265
	 */
266
	public function get_logs_count() {
267
		global $wpdb;
268
269
		$query = 'SELECT count(*) FROM ' . $this->get_log_table_name();
270
271
		return $wpdb->get_var( $query );
272
	}
273
274
	/**
275
	 * Fetches the log item by the item data.
276
	 *
277
	 * Use this method to get the log item when the error instance only returns the log item data.
278
	 *
279
	 * @param array $data Array of Email information. {
280
	 *
281
	 * @type array|string to
282
	 * @type string       subject
283
	 * @type string       message
284
	 * @type array|string headers
285
	 * @type array|string attachments
286
	 *                    }
287
	 *
288
	 * @return int
289
	 */
290
	public function fetch_log_item_by_item_data( $data ) {
291
		if ( empty( $data ) || ! is_array( $data ) ) {
292
			return 0;
293
		}
294
295
		global $wpdb;
296
		$table_name = $this->get_log_table_name();
297
298
		$query      = "SELECT ID FROM {$table_name}";
299
		$query_cond = '';
300
		$where      = array();
301
302
		// Execute the following `if` conditions only when $data is array.
303
		if ( array_key_exists( 'to', $data ) ) {
304
			// Since the value is stored as CSV in DB, convert the values from error data to CSV to compare.
305
			$data['to'] = Util\join_array_elements_with_delimiter( $data['to'] );
0 ignored issues
show
Bug introduced by
The function join_array_elements_with_delimiter was not found. Maybe you did not declare it correctly or list all dependencies? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-call  annotation

305
			$data['to'] = /** @scrutinizer ignore-call */ Util\join_array_elements_with_delimiter( $data['to'] );
Loading history...
306
307
			$to_email = trim( esc_sql( $data['to'] ) );
0 ignored issues
show
Bug introduced by
It seems like esc_sql($data['to']) can also be of type array; however, parameter $str of trim() does only seem to accept string, maybe add an additional type check? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

307
			$to_email = trim( /** @scrutinizer ignore-type */ esc_sql( $data['to'] ) );
Loading history...
308
			$where[]  = "to_email = '$to_email'";
309
		}
310
311
		if ( array_key_exists( 'subject', $data ) ) {
312
			$subject = trim( esc_sql( $data['subject'] ) );
313
			$where[] = "subject = '$subject'";
314
		}
315
316
		if ( array_key_exists( 'attachments', $data ) ) {
317
			if ( is_array( $data['attachments'] ) ) {
318
				$attachments = count( $data['attachments'] ) > 0 ? 'true' : 'false';
319
			} else {
320
				$attachments = empty( $data['attachments'] ) ? 'false' : 'true';
321
			}
322
			$attachments = trim( esc_sql( $attachments ) );
323
			$where[]     = "attachments = '$attachments'";
324
		}
325
326
		foreach ( $where as $index => $value ) {
327
			$query_cond .= 0 === $index ? ' WHERE ' : ' AND ';
328
			$query_cond .= $value;
329
		}
330
331
		// Get only the latest logged item when multiple rows match.
332
		$query_cond .= ' ORDER BY id DESC LIMIT 1';
333
334
		$query = $query . $query_cond;
335
336
		return absint( $wpdb->get_var( $query ) );
337
	}
338
339
	/**
340
	 * Sets email sent status as failed for the given log item.
341
	 *
342
	 * @since 2.3.0
343
	 *
344
	 * @param int $log_item_id ID of the log item whose email sent status should be set to failed.
345
	 */
346
	public function set_log_item_fail_status_by_id( $log_item_id ) {
347
		global $wpdb;
348
		$table_name = $this->get_log_table_name();
349
350
		$wpdb->update(
351
			$table_name,
352
			array( 'result' => '0', ),
353
			array( 'ID' => $log_item_id ),
354
			array( '%d' ),
355
			array( '%d' )
356
		);
357
	}
358
359
	/**
360
	 * Updates the DB schema.
361
	 *
362
	 * Adds new columns to the Database as of v0.2.
363
	 *
364
	 * @since 2.3.0
365
	 */
366
	private function update_table_if_needed() {
367
		$existing_db_version = get_option( self::DB_OPTION_NAME, false );
368
		$updated_db_version  = self::DB_VERSION;
369
370
		// Bail out when the DB version is `0.1` or equals to self::DB_VERSION
371
		if ( ! $existing_db_version || $existing_db_version !== '0.1' || $existing_db_version === $updated_db_version ) {
372
			return;
373
		}
374
375
		$sql = $this->get_create_table_query();
376
377
		require_once ABSPATH . 'wp-admin/includes/upgrade.php';
378
		dbDelta( $sql );
379
380
		update_option( self::DB_OPTION_NAME, self::DB_VERSION );
381
	}
382
383
	/**
384
	 * Gets the Create Table query.
385
	 *
386
	 * @since 2.3.0
387
	 *
388
	 * @return string
389
	 */
390
	private function get_create_table_query() {
391
		global $wpdb;
392
		$table_name      = $this->get_log_table_name();
393
		$charset_collate = $wpdb->get_charset_collate();
394
395
		$sql = 'CREATE TABLE ' . $table_name . ' (
396
				id mediumint(9) NOT NULL AUTO_INCREMENT,
397
				to_email VARCHAR(250) NOT NULL,
398
				subject VARCHAR(250) NOT NULL,
399
				message TEXT NOT NULL,
400
				headers TEXT NOT NULL,
401
				attachments TEXT NOT NULL,
402
				sent_date timestamp NOT NULL,
403
				attachment_name VARCHAR(1000),
404
				ip_address VARCHAR(15),
405
				result TINYINT(1)
406
				PRIMARY KEY  (id)
407
			) ' . $charset_collate . ' ;';
408
409
		return $sql;
410
	}
411
}
412