Completed
Pull Request — dev/2.3.0 (#167)
by Maria Daniel Deepak
10:59 queued 05:38
created

TableManager::fetch_log_items()   B

Complexity

Conditions 11
Paths 96

Size

Total Lines 45
Code Lines 26

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 132

Importance

Changes 0
Metric Value
cc 11
eloc 26
nc 96
nop 3
dl 0
loc 45
ccs 0
cts 31
cp 0
crap 132
rs 7.3166
c 0
b 0
f 0

How to fix   Complexity   

Long Method

Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.

For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.

Commonly applied refactorings include:

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
	 * @param array $additional_args    {
162
	 *                                  Optional. Array of additional args.
163
	 *
164
	 * @type string $date_column_format MySQL date column format. Refer
165
	 *                                  @link  https://dev.mysql.com/doc/refman/5.5/en/date-and-time-functions.html#function_date-format
166
	 * }
167
	 *
168
	 * @return array Log item(s).
169
	 */
170
	public function fetch_log_items_by_id( $ids = array(), $additional_args = array() ) {
171
		global $wpdb;
172
		$table_name = $this->get_log_table_name();
173
174
		$query = "SELECT * FROM {$table_name}";
175
176
		// When `date_column_format` exists, should replace the `$query` var.
177
		$date_column_format_key = 'date_column_format';
178
		if ( array_key_exists( $date_column_format_key, $additional_args ) && ! empty( $additional_args[ $date_column_format_key ] ) ) {
179
			$query = "SELECT DATE_FORMAT(sent_date, \"{$additional_args[ $date_column_format_key ]}\") as sent_date_custom, el.* FROM {$table_name} as el";
180
		}
181
182
		if ( ! empty( $ids ) ) {
183
			$ids = array_map( 'absint', $ids );
184
185
			// Can't use wpdb->prepare for the below query. If used it results in this bug https://github.com/sudar/email-log/issues/13.
186
			$ids_list = esc_sql( implode( ',', $ids ) );
187
188
			$query .= " where id IN ( {$ids_list} )";
189
		}
190
191
		return $wpdb->get_results( $query, 'ARRAY_A' ); //@codingStandardsIgnoreLine
192
	}
193
194
	/**
195
	 * Fetch log items.
196
	 *
197
	 * @param array $request         Request object.
198
	 * @param int   $per_page        Entries per page.
199
	 * @param int   $current_page_no Current page no.
200
	 *
201
	 * @return array Log entries and total items count.
202
	 */
203
	public function fetch_log_items( $request, $per_page, $current_page_no ) {
204
		global $wpdb;
205
		$table_name = $this->get_log_table_name();
206
207
		$query       = 'SELECT * FROM ' . $table_name;
208
		$count_query = 'SELECT count(*) FROM ' . $table_name;
209
		$query_cond  = '';
210
211
		if ( isset( $request['s'] ) && $request['s'] !== '' ) {
212
			$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

212
			$search_term = trim( /** @scrutinizer ignore-type */ esc_sql( $request['s'] ) );
Loading history...
213
			$query_cond .= " WHERE ( to_email LIKE '%$search_term%' OR subject LIKE '%$search_term%' ) ";
214
		}
215
216
		if ( isset( $request['d'] ) && $request['d'] !== '' ) {
217
			$search_date = trim( esc_sql( $request['d'] ) );
218
			if ( '' === $query_cond ) {
219
				$query_cond .= " WHERE sent_date BETWEEN '$search_date 00:00:00' AND '$search_date 23:59:59' ";
220
			} else {
221
				$query_cond .= " AND sent_date BETWEEN '$search_date 00:00:00' AND '$search_date 23:59:59' ";
222
			}
223
		}
224
225
		// Ordering parameters.
226
		$orderby = ! empty( $request['orderby'] ) ? esc_sql( $request['orderby'] ) : 'sent_date';
227
		$order   = ! empty( $request['order'] ) ? esc_sql( $request['order'] ) : 'DESC';
228
229
		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...
230
			$query_cond .= ' ORDER BY ' . $orderby . ' ' . $order;
0 ignored issues
show
Bug introduced by
Are you sure $order of type array|string 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

230
			$query_cond .= ' ORDER BY ' . $orderby . ' ' . /** @scrutinizer ignore-type */ $order;
Loading history...
Bug introduced by
Are you sure $orderby of type array|string 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

230
			$query_cond .= ' ORDER BY ' . /** @scrutinizer ignore-type */ $orderby . ' ' . $order;
Loading history...
231
		}
232
233
		// Find total number of items.
234
		$count_query = $count_query . $query_cond;
235
		$total_items = $wpdb->get_var( $count_query );
236
237
		// Adjust the query to take pagination into account.
238
		if ( ! empty( $current_page_no ) && ! empty( $per_page ) ) {
239
			$offset = ( $current_page_no - 1 ) * $per_page;
240
			$query_cond .= ' LIMIT ' . (int) $offset . ',' . (int) $per_page;
241
		}
242
243
		// Fetch the items.
244
		$query = $query . $query_cond;
245
		$items = $wpdb->get_results( $query );
246
247
		return array( $items, $total_items );
248
	}
249
250
	/**
251
	 * Create email log table.
252
	 *
253
	 * @access private
254
	 *
255
	 * @global object $wpdb
256
	 */
257
	private function create_table_if_needed() {
258
		global $wpdb;
259
260
		$table_name = $this->get_log_table_name();
261
262
		if ( $wpdb->get_var( "show tables like '{$table_name}'" ) != $table_name ) {
263
264
			$sql = $this->get_create_table_query();
265
266
			require_once ABSPATH . 'wp-admin/includes/upgrade.php';
267
			dbDelta( $sql );
268
269
			add_option( self::DB_OPTION_NAME, self::DB_VERSION );
270
		}
271
	}
272
273
	/**
274
	 * Get the total number of email logs.
275
	 *
276
	 * @return int Total email log count
277
	 */
278
	public function get_logs_count() {
279
		global $wpdb;
280
281
		$query = 'SELECT count(*) FROM ' . $this->get_log_table_name();
282
283
		return $wpdb->get_var( $query );
284
	}
285
286
	/**
287
	 * Fetches the log item by the item data.
288
	 *
289
	 * Use this method to get the log item when the error instance only returns the log item data.
290
	 *
291
	 * @param array $data Array of Email information. {
292
	 *
293
	 * @type array|string to
294
	 * @type string       subject
295
	 * @type string       message
296
	 * @type array|string headers
297
	 * @type array|string attachments
298
	 *                    }
299
	 *
300
	 * @return int
301
	 */
302
	public function fetch_log_item_by_item_data( $data ) {
303
		if ( empty( $data ) || ! is_array( $data ) ) {
304
			return 0;
305
		}
306
307
		global $wpdb;
308
		$table_name = $this->get_log_table_name();
309
310
		$query      = "SELECT ID FROM {$table_name}";
311
		$query_cond = '';
312
		$where      = array();
313
314
		// Execute the following `if` conditions only when $data is array.
315
		if ( array_key_exists( 'to', $data ) ) {
316
			// Since the value is stored as CSV in DB, convert the values from error data to CSV to compare.
317
			$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

317
			$data['to'] = /** @scrutinizer ignore-call */ Util\join_array_elements_with_delimiter( $data['to'] );
Loading history...
318
319
			$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

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