Completed
Push — v2 ( a4c7ad...18d8b1 )
by Berend
14:41 queued 34s
created

AbstractActiveRecord::getSearchQueryWhere()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 7
Code Lines 3

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 4
CRAP Score 2

Importance

Changes 0
Metric Value
eloc 3
dl 0
loc 7
c 0
b 0
f 0
ccs 4
cts 4
cp 1
rs 10
cc 2
nc 2
nop 2
crap 2
1
<?php
2
3
/**
4
 * This file is part of the miBadger package.
5
 *
6
 * @author Michael Webbers <[email protected]>
7
 * @license http://opensource.org/licenses/Apache-2.0 Apache v2 License
8
 */
9
10
namespace miBadger\ActiveRecord;
11
12
use miBadger\Query\Query;
13
14
/**
15
 * The abstract active record class.
16
 *
17
 * @since 1.0.0
18
 */
19
abstract class AbstractActiveRecord implements ActiveRecordInterface
20
{
21
	const COLUMN_NAME_ID = 'id';
22
	const COLUMN_TYPE_ID = 'INT UNSIGNED';
23
24
	/** @var \PDO The PDO object. */
25
	protected $pdo;
26
27
	/** @var null|int The ID. */
28
	private $id;
29
30
	/** @var array A map of column name to functions that hook the insert function */
31
	protected $registeredCreateHooks;
32
33
	/** @var array A map of column name to functions that hook the read function */
34
	protected $registeredReadHooks;
35
36
	/** @var array A map of column name to functions that hook the update function */
37
	protected $registeredUpdateHooks;
38
39
	/** @var array A map of column name to functions that hook the update function */
40
	protected $registeredDeleteHooks;	
41
42
	/** @var array A map of column name to functions that hook the search function */
43
	protected $registeredSearchHooks;
44
45
	/** @var array A list of table column definitions */
46
	protected $tableDefinition;
47
48
	/**
49
	 * Construct an abstract active record with the given PDO.
50
	 *
51
	 * @param \PDO $pdo
52
	 */
53 82
	public function __construct(\PDO $pdo)
54
	{
55 82
		$pdo->setAttribute(\PDO::ATTR_DEFAULT_FETCH_MODE, \PDO::FETCH_ASSOC);
56 82
		$pdo->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
57
58 82
		$this->setPdo($pdo);
59 82
		$this->tableDefinition = $this->getActiveRecordTableDefinition();
60 82
		$this->registeredCreateHooks = [];
61 82
		$this->registeredReadHooks = [];
62 82
		$this->registeredUpdateHooks = [];
63 82
		$this->registeredDeleteHooks = [];
64 82
		$this->registeredSearchHooks = [];
65
66
		// Extend table definition with default ID field, throw exception if field already exists
67 82
		if (array_key_exists('id', $this->tableDefinition)) {
68
			$message = "Table definition in record contains a field with name \"id\"";
69
			$message .= ", which is a reserved name by ActiveRecord";
70
			throw new ActiveRecordException($message, 0);
71
		}
72
73 82
		$this->tableDefinition[self::COLUMN_NAME_ID] =
74
		[
75 82
			'value' => &$this->id,
76
			'validate' => null,
77 82
			'type' => self::COLUMN_TYPE_ID,
78 82
			'properties' => ColumnProperty::NOT_NULL | ColumnProperty::IMMUTABLE | ColumnProperty::AUTO_INCREMENT | ColumnProperty::PRIMARY_KEY
79
		];
80 82
	}
81
82 34
	private function checkHookConstraints($columnName, $hookMap)
83
	{
84
		// Check whether column exists
85 34
		if (!array_key_exists($columnName, $this->tableDefinition)) 
86
		{
87 5
			throw new ActiveRecordException("Hook is trying to register on non-existing column \"$columnName\"", 0);
88
		}
89
90
		// Enforcing 1 hook per table column
91 29
		if (array_key_exists($columnName, $hookMap)) {
92 5
			$message = "Hook is trying to register on an already registered column \"$columnName\", ";
93 5
			$message .= "do you have conflicting traits?";
94 5
			throw new ActiveRecordException($message, 0);
95
		}
96 29
	}
97
98
	/**
99
	 * Register a new hook for a specific column that gets called before execution of the create() method
100
	 * Only one hook per column can be registered at a time
101
	 * @param string $columnName The name of the column that is registered.
102
	 * @param string|callable $fn Either a callable, or the name of a method on the inheriting object.
103
	 */
104 7
	public function registerCreateHook($columnName, $fn)
105
	{
106 7
		$this->checkHookConstraints($columnName, $this->registeredCreateHooks);
107
108 6
		if (is_string($fn) && is_callable([$this, $fn])) {
109 3
			$this->registeredCreateHooks[$columnName] = [$this, $fn];
110 3
		} else if (is_callable($fn)) {
111 2
			$this->registeredCreateHooks[$columnName] = $fn;
112
		} else {
113 1
			throw new ActiveRecordException("Provided hook on column \"$columnName\" is not callable", 0);
114
		}
115 5
	}
116
117
	/**
118
	 * Register a new hook for a specific column that gets called before execution of the read() method
119
	 * Only one hook per column can be registered at a time
120
	 * @param string $columnName The name of the column that is registered.
121
	 * @param string|callable $fn Either a callable, or the name of a method on the inheriting object.
122
	 */
123 4
	public function registerReadHook($columnName, $fn)
124
	{
125 4
		$this->checkHookConstraints($columnName, $this->registeredReadHooks);
126
127 3
		if (is_string($fn) && is_callable([$this, $fn])) {
128
			$this->registeredReadHooks[$columnName] = [$this, $fn];
129 3
		} else if (is_callable($fn)) {
130 2
			$this->registeredReadHooks[$columnName] = $fn;
131
		} else {
132 1
			throw new ActiveRecordException("Provided hook on column \"$columnName\" is not callable", 0);
133
		}
134 2
	}
135
136
	/**
137
	 * Register a new hook for a specific column that gets called before execution of the update() method
138
	 * Only one hook per column can be registered at a time
139
	 * @param string $columnName The name of the column that is registered.
140
	 * @param string|callable $fn Either a callable, or the name of a method on the inheriting object.
141
	 */
142 7
	public function registerUpdateHook($columnName, $fn)
143
	{
144 7
		$this->checkHookConstraints($columnName, $this->registeredUpdateHooks);
145
146 6
		if (is_string($fn) && is_callable([$this, $fn])) {
147 3
			$this->registeredUpdateHooks[$columnName] = [$this, $fn];
148 3
		} else if (is_callable($fn)) {
149 2
			$this->registeredUpdateHooks[$columnName] = $fn;
150
		} else {
151 1
			throw new ActiveRecordException("Provided hook on column \"$columnName\" is not callable", 0);
152
		}
153 5
	}
154
155
	/**
156
	 * Register a new hook for a specific column that gets called before execution of the delete() method
157
	 * Only one hook per column can be registered at a time
158
	 * @param string $columnName The name of the column that is registered.
159
	 * @param string|callable $fn Either a callable, or the name of a method on the inheriting object.
160
	 */
161 4
	public function registerDeleteHook($columnName, $fn)
162
	{
163 4
		$this->checkHookConstraints($columnName, $this->registeredDeleteHooks);
164
165 3
		if (is_string($fn) && is_callable([$this, $fn])) {
166
			$this->registeredDeleteHooks[$columnName] = [$this, $fn];
167 3
		} else if (is_callable($fn)) {
168 2
			$this->registeredDeleteHooks[$columnName] = $fn;
169
		} else {
170 1
			throw new ActiveRecordException("Provided hook on column \"$columnName\" is not callable", 0);
171
		}
172 2
	}
173
174
	/**
175
	 * Register a new hook for a specific column that gets called before execution of the search() method
176
	 * Only one hook per column can be registered at a time
177
	 * @param string $columnName The name of the column that is registered.
178
	 * @param string|callable $fn Either a callable, or the name of a method on the inheriting object. The callable is required to take one argument: an instance of miBadger\Query\Query; 
179
	 */
180 15
	public function registerSearchHook($columnName, $fn)
181
	{
182 15
		$this->checkHookConstraints($columnName, $this->registeredSearchHooks);
183
184 14
		if (is_string($fn) && is_callable([$this, $fn])) {
185 11
			$this->registeredSearchHooks[$columnName] = [$this, $fn];
186 3
		} else if (is_callable($fn)) {
187 2
			$this->registeredSearchHooks[$columnName] = $fn;
188
		} else {
189 1
			throw new ActiveRecordException("Provided hook on column \"$columnName\" is not callable", 0);
190
		}
191 13
	}
192
193
	/**
194
	 * Adds a new column definition to the table.
195
	 * @param string $columnName The name of the column that is registered.
196
	 * @param Array $definition The definition of that column.
197
	 */
198 49
	public function extendTableDefinition($columnName, $definition)
199
	{
200 49
		if ($this->tableDefinition === null) {
201 1
			throw new ActiveRecordException("tableDefinition is null, most likely due to parent class not having been initialized in constructor");
202
		}
203
204
		// Enforcing table can only be extended with new columns
205 48
		if (array_key_exists($columnName, $this->tableDefinition)) {
206 1
			$message = "Table is being extended with a column that already exists, ";
207 1
			$message .= "\"$columnName\" conflicts with your table definition";
208 1
			throw new ActiveRecordException($message, 0);
209
		}
210
211 48
		$this->tableDefinition[$columnName] = $definition;
212 48
	}
213
214
	/**
215
	 * Returns the type string as it should appear in the mysql create table statement for the given column
216
	 * @return string The type string
217
	 */
218 27
	private function getDatabaseTypeString($colName, $type, $length)
219
	{
220 27
		switch (strtoupper($type)) {
221 27
			case '':
222 1
				throw new ActiveRecordException(sprintf("Column %s has invalid type \"NULL\"", $colName));
223
				
224 26
			case 'DATETIME':
225 26
			case 'DATE':
226 26
			case 'TIME':
227 26
			case 'TEXT':
228 26
			case 'INT UNSIGNED':
229 26
				return $type;
230
231 25
			case 'VARCHAR':
232 25
				if ($length === null) {
233
					throw new ActiveRecordException(sprintf("field type %s requires specified column field \"LENGTH\"", $colName));
234
				} else {
235 25
					return sprintf('%s(%d)', $type, $length);	
236
				}
237
238 9
			case 'INT':
239
			case 'TINYINT':
240
			case 'BIGINT':
241
			default: 	
242
				// @TODO(Default): throw exception, or implicitly assume that type is correct? (For when using SQL databases with different types)
243 9
				if ($length === null) {
244
					return $type;
245
				} else {
246 9
					return sprintf('%s(%d)', $type, $length);	
247
				}
248
		}
249
	}
250
251
	/**
252
	 * Builds the part of a MySQL create table statement that corresponds to the supplied column
253
	 * @param string $colName 	Name of the database column
254
	 * @param string $type 		The type of the string
255
	 * @param int $properties 	The set of Column properties that apply to this column (See ColumnProperty for options)
256
	 * @return string
257
	 */
258 27
	private function buildCreateTableColumnEntry($colName, $type, $length, $properties, $default)
259
	{
260 27
		$stmnt = sprintf('`%s` %s ', $colName, $this->getDatabaseTypeString($colName, $type, $length));
261 26
		if ($properties & ColumnProperty::NOT_NULL) {
262 26
			$stmnt .= 'NOT NULL ';
263
		} else {
264 26
			$stmnt .= 'NULL ';
265
		}
266
267 26
		if ($default !== NULL) {
268 17
			$stmnt .= ' DEFAULT ' . $default . ' ';
269
		}
270
271 26
		if ($properties & ColumnProperty::AUTO_INCREMENT) {
272 26
			$stmnt .= 'AUTO_INCREMENT ';
273
		}
274
275 26
		if ($properties & ColumnProperty::UNIQUE) {
276 8
			$stmnt .= 'UNIQUE ';
277
		}
278
279 26
		if ($properties & ColumnProperty::PRIMARY_KEY) {
280 26
			$stmnt .= 'PRIMARY KEY ';
281
		}
282
283 26
		return $stmnt;
284
	}
285
286
	/**
287
	 * Sorts the column statement components in the order such that the id appears first, 
288
	 * 		followed by all other columns in alphabetical ascending order
289
	 * @param   Array $colStatements Array of column statements
290
	 * @return  Array
291
	 */
292 26
	private function sortColumnStatements($colStatements)
293
	{
294
		// Find ID statement and put it first
295 26
		$sortedStatements = [];
296
297 26
		$sortedStatements[] = $colStatements[self::COLUMN_NAME_ID];
298 26
		unset($colStatements[self::COLUMN_NAME_ID]);
299
300
		// Sort remaining columns in alphabetical order
301 26
		$columns = array_keys($colStatements);
302 26
		sort($columns);
303 26
		foreach ($columns as $colName) {
304 26
			$sortedStatements[] = $colStatements[$colName];
305
		}
306
307 26
		return $sortedStatements;
308
	}
309
310
	/**
311
	 * Builds the MySQL Create Table statement for the internal table definition
312
	 * @return string
313
	 */
314 27
	public function buildCreateTableSQL()
315
	{
316 27
		$columnStatements = [];
317 27
		foreach ($this->tableDefinition as $colName => $definition) {
318
			// Destructure column definition
319 27
			$type    = $definition['type'] ?? null;
320 27
			$default = $definition['default'] ?? null;
321 27
			$length  = $definition['length'] ?? null;
322 27
			$properties = $definition['properties'] ?? null;
323
324 27
			if (isset($definition['relation']) && $type !== null) {
325
				$msg = "Column \"$colName\": ";
326
				$msg .= "Relationship columns have an automatically inferred type, so type should be omitted";
327
				throw new ActiveRecordException($msg);
328 27
			} else if (isset($definition['relation'])) {
329 2
				$type = self::COLUMN_TYPE_ID;
330
			}
331
332 27
			$columnStatements[$colName] = $this->buildCreateTableColumnEntry($colName, $type, $length, $properties, $default);
333
		}
334
335
		// Sort table (first column is id, the remaining are alphabetically sorted)
336 26
		$columnStatements = $this->sortColumnStatements($columnStatements);
337
338 26
		$sql = 'CREATE TABLE ' . $this->getActiveRecordTable() . ' ';
339 26
		$sql .= "(\n";
340 26
		$sql .= implode(",\n", $columnStatements);
341 26
		$sql .= "\n);";
342
343 26
		return $sql;
344
	}
345
346
	/**
347
	 * Creates the entity as a table in the database
348
	 */
349 27
	public function createTable()
350
	{
351 27
		$this->pdo->query($this->buildCreateTableSQL());
352 26
	}
353
354
	/**
355
	 * builds a MySQL constraint statement for the given parameters
356
	 * @param string $parentTable
357
	 * @param string $parentColumn
358
	 * @param string $childTable
359
	 * @param string $childColumn
360
	 * @return string The MySQL table constraint string
361
	 */
362 4
	protected function buildConstraint($parentTable, $parentColumn, $childTable, $childColumn)
363
	{
364
		$template = <<<SQL
365 4
ALTER TABLE `%s`
366
ADD CONSTRAINT
367
FOREIGN KEY (`%s`)
368
REFERENCES `%s`(`%s`)
369
ON DELETE CASCADE;
370
SQL;
371 4
		return sprintf($template, $childTable, $childColumn, $parentTable, $parentColumn);
372
	}
373
374
	/**
375
	 * Iterates over the specified constraints in the table definition, 
376
	 * 		and applies these to the database.
377
	 */
378 1
	public function createTableConstraints()
379
	{
380
		// Iterate over columns, check whether "relation" field exists, if so create constraint
381 1
		foreach ($this->tableDefinition as $colName => $definition) {
382 1
			if (isset($definition['relation']) && $definition['relation'] instanceof AbstractActiveRecord) {
383
				// Forge new relation
384 1
				$target = $definition['relation'];
385 1
				$constraintSql = $this->buildConstraint($target->getActiveRecordTable(), 'id', $this->getActiveRecordTable(), $colName);
386
387 1
				$this->pdo->query($constraintSql);
388
			}
389
		}
390 1
	}
391
392
	/**
393
	 * Returns the name -> variable mapping for the table definition.
394
	 * @return Array The mapping
395
	 */
396 46
	protected function getActiveRecordColumns()
397
	{
398 46
		$bindings = [];
399 46
		foreach ($this->tableDefinition as $colName => $definition) {
400
401
			// Ignore the id column (key) when inserting or updating
402 46
			if ($colName == self::COLUMN_NAME_ID) {
403 46
				continue;
404
			}
405
406 46
			$bindings[$colName] = &$definition['value'];
407
		}
408 46
		return $bindings;
409
	}
410
411
	/**
412
	 * {@inheritdoc}
413
	 */
414 22
	public function create()
415
	{
416 22
		foreach ($this->registeredCreateHooks as $colName => $fn) {
417
			// @TODO: Would it be better to pass the Query to the function?
418 3
			$fn();
419
		}
420
421
		try {
422 22
			(new Query($this->getPdo(), $this->getActiveRecordTable()))
423 22
				->insert($this->getActiveRecordColumns())
424 22
				->execute();
425
426 20
			$this->setId(intval($this->getPdo()->lastInsertId()));
427 2
		} catch (\PDOException $e) {
428 2
			throw new ActiveRecordException($e->getMessage(), 0, $e);
429
		}
430
431 20
		return $this;
432
	}
433
434
	/**
435
	 * {@inheritdoc}
436
	 */
437 17
	public function read($id)
438
	{
439 17
		foreach ($this->registeredReadHooks as $colName => $fn) {
440
			// @TODO: Would it be better to pass the Query to the function?
441 1
			$fn();
442
		}
443
444
		try {
445 17
			$row = (new Query($this->getPdo(), $this->getActiveRecordTable()))
446 17
				->select()
447 17
				->where(Query::Equal('id', $id))
448 17
				->execute()
449 16
				->fetch();
450
451 16
			if ($row === false) {
452 3
				throw new ActiveRecordException(sprintf('Can not read the non-existent active record entry %d from the `%s` table.', $id, $this->getActiveRecordTable()));
453
			}
454
455 13
			$this->fill($row)->setId($id);
456 4
		} catch (\PDOException $e) {
457 1
			throw new ActiveRecordException($e->getMessage(), 0, $e);
458
		}
459
460 13
		return $this;
461
	}
462
463
	/**
464
	 * {@inheritdoc}
465
	 */
466 8
	public function update()
467
	{
468 8
		foreach ($this->registeredUpdateHooks as $colName => $fn) {
469
			// @TODO: Would it be better to pass the Query to the function?
470 2
			$fn();
471
		}
472
473
		try {
474 8
			(new Query($this->getPdo(), $this->getActiveRecordTable()))
475 8
				->update($this->getActiveRecordColumns())
476 8
				->where(Query::Equal('id', $this->getId()))
0 ignored issues
show
Bug introduced by
It seems like $this->getId() can also be of type integer; however, parameter $right of miBadger\Query\Query::Equal() does only seem to accept miBadger\Query\any, 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

476
				->where(Query::Equal('id', /** @scrutinizer ignore-type */ $this->getId()))
Loading history...
477 8
				->execute();
478 2
		} catch (\PDOException $e) {
479 2
			throw new ActiveRecordException($e->getMessage(), 0, $e);
480
		}
481
482 6
		return $this;
483
	}
484
485
	/**
486
	 * {@inheritdoc}
487
	 */
488 6
	public function delete()
489
	{
490 6
		foreach ($this->registeredDeleteHooks as $colName => $fn) {
491
			// @TODO: Would it be better to pass the Query to the function?
492 1
			$fn();
493
		}
494
495
		try {
496 6
			(new Query($this->getPdo(), $this->getActiveRecordTable()))
497 6
				->delete()
498 6
				->where(Query::Equal('id', $this->getId()))
0 ignored issues
show
Bug introduced by
It seems like $this->getId() can also be of type integer; however, parameter $right of miBadger\Query\Query::Equal() does only seem to accept miBadger\Query\any, 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

498
				->where(Query::Equal('id', /** @scrutinizer ignore-type */ $this->getId()))
Loading history...
499 6
				->execute();
500
501 5
			$this->setId(null);
502 1
		} catch (\PDOException $e) {
503 1
			throw new ActiveRecordException($e->getMessage(), 0, $e);
504
		}
505
506 5
		return $this;
507
	}
508
509
	/**
510
	 * {@inheritdoc}
511
	 */
512 2
	public function sync()
513
	{
514 2
		if (!$this->exists()) {
515 1
			return $this->create();
516
		}
517
518 1
		return $this->update();
519
	}
520
521
	/**
522
	 * {@inheritdoc}
523
	 */
524 3
	public function exists()
525
	{
526 3
		return $this->getId() !== null;
527
	}
528
529
	/**
530
	 * {@inheritdoc}
531
	 */
532 26
	public function fill(array $attributes)
533
	{
534 26
		$columns = $this->getActiveRecordColumns();
535 26
		$columns['id'] = &$this->id;
536
537 26
		foreach ($attributes as $key => $value) {
538 26
			if (array_key_exists($key, $columns)) {
539 26
				$columns[$key] = $value;
540
			}
541
		}
542
543 26
		return $this;
544
	}
545
546
	/**
547
	 * {@inheritdoc}
548
	 */
549 17
	public function search(array $ignoredTraits = [])
550
	{
551 17
		$clauses = [];
552 17
		foreach ($this->registeredSearchHooks as $column => $fn) {
553 2
			if (!in_array($column, $ignoredTraits)) {
554 2
				$clauses[] = $fn();
555
			}
556
		}
557
558 17
		return new ActiveRecordQuery($this, $this->getActiveRecordTable(), $clauses);
559
	}
560
561
	/**
562
	 * Returns the PDO.
563
	 *
564
	 * @return \PDO the PDO.
565
	 */
566 53
	public function getPdo()
567
	{
568 53
		return $this->pdo;
569
	}
570
571
	/**
572
	 * Set the PDO.
573
	 *
574
	 * @param \PDO $pdo
575
	 * @return $this
576
	 */
577 82
	protected function setPdo($pdo)
578
	{
579 82
		$this->pdo = $pdo;
580
581 82
		return $this;
582
	}
583
584
	/**
585
	 * Returns the ID.
586
	 *
587
	 * @return null|int The ID.
588
	 */
589 22
	public function getId()
590
	{
591 22
		return $this->id;
592
	}
593
594
	/**
595
	 * Set the ID.
596
	 *
597
	 * @param int $id
598
	 * @return $this
599
	 */
600 30
	protected function setId($id)
601
	{
602 30
		$this->id = $id;
603
604 30
		return $this;
605
	}
606
607
	/**
608
	 * Returns the active record table.
609
	 *
610
	 * @return string the active record table name.
611
	 */
612
	abstract protected function getActiveRecordTable();
613
614
	/**
615
	 * Returns the active record columns.
616
	 *
617
	 * @return array the active record columns.
618
	 */
619
	abstract protected function getActiveRecordTableDefinition();
620
}
621