Completed
Push — 177-fix/php-fatal-error-while-... ( c47480 )
by Maria Daniel Deepak
10:40 queued 04:47
created

TableManager::get_logs_count()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 6
Code Lines 3

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 2

Importance

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

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

231
			$query_cond .= ' ORDER BY ' . /** @scrutinizer ignore-type */ $orderby . ' ' . $order;
Loading history...
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

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

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

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