Qpdb\QueryBuilder\Traits\TableValidation

Complexity

Total Complexity

8

Size/Duplication

Total Lines	55
Duplicated Lines	0 %

Importance

Changes	3
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	validateTableName()	0	8	2
A	validateTableSubQuery()	0	9	3
A	validateTable()	0	16	3

<?php
/**
 * Created by PhpStorm.
 * User: Adrian Dumitru
 * Date: 8/20/2017
 * Time: 12:36 AM
 */

namespace Qpdb\QueryBuilder\Traits;

use Qpdb\QueryBuilder\Dependencies\QueryException;
use Qpdb\QueryBuilder\Dependencies\QueryHelper;
use Qpdb\QueryBuilder\Dependencies\QueryStructure;
use Qpdb\QueryBuilder\Statements\QuerySelect;
use Qpdb\QueryBuilder\Statements\QueryStatement;

trait TableValidation
{

	/**
	 * @param $table
	 * @return mixed|string
	 * @throws QueryException
	 */
	private function validateTable( $table )
	{
		switch ( gettype( $table ) ) {

			case QueryStructure::ELEMENT_TYPE_STRING:
				$table = $this->validateTableName( $table );
				break;
			case QueryStructure::ELEMENT_TYPE_OBJECT:
				$table = $this->validateTableSubQuery( $table );
				break;
			default:
				throw new QueryException( 'Invalid table type parameter: ' . gettype( $table ), QueryException::QUERY_ERROR_INVALID_TABLE_STATEMENT );

		}

		return $table;
	}

	/**
	 * @param $table
	 * @return mixed|string
	 * @throws QueryException
	 */
	private function validateTableName( $table )
	{
		$table = trim( $table );

		if ( '' === $table )
			throw new QueryException( 'Table name is empty string!', QueryException::QUERY_ERROR_INVALID_TABLE_STATEMENT );

		return QueryHelper::addBacktick($table);
	}

	/**
	 * @param $table
	 * @return mixed
	 * @throws QueryException
	 */
	private function validateTableSubQuery( $table )
	{
		if ( $this->statement !== QueryStatement::QUERY_STATEMENT_SELECT )
			throw new QueryException( 'Invalid subQuery statement!', QueryException::QUERY_ERROR_INVALID_TABLE_STATEMENT );

		if ( !is_a( $table, QuerySelect::class ) )
			throw new QueryException( 'Invalid subQuery statement!', QueryException::QUERY_ERROR_INVALID_TABLE_STATEMENT );

		return $table;
	}


}