Wikibase\DataModel\Services\Lookup\DisabledEntityTypesEntityLookup

Complexity

Total Complexity

4

Size/Duplication

Total Lines	55
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	getEntity()	0	10	2
A	hasEntity()	0	3	1

<?php

namespace Wikibase\DataModel\Services\Lookup;

use Wikibase\DataModel\Entity\EntityDocument;
use Wikibase\DataModel\Entity\EntityId;
use Wikimedia\Assert\Assert;

/**
 * EntityLookup that checks entities have been loaded through it and throws
 * an exception if the accessing to that entity type is disabled.
 *
 * @since 3.9
 *
 * @license GPL-2.0+
 * @author Amir Sarabadani
 */
class DisabledEntityTypesEntityLookup implements EntityLookup {

	/**
	 * @var EntityLookup
	 */
	private $entityLookup;

	/**
	 * @var string[]
	 */
	private $disabledEntityTypes = [];

	/**
	 * @param EntityLookup $entityLookup
	 * @param string[] $disabledEntityTypes
	 */
	public function __construct( EntityLookup $entityLookup, array $disabledEntityTypes ) {
		Assert::parameterElementType( 'string', $disabledEntityTypes, '$disabledEntityTypes' );

		$this->entityLookup = $entityLookup;
		$this->disabledEntityTypes = $disabledEntityTypes;
	}

	/**
	 * @see EntityLookup::getEntity
	 *
	 * @param EntityId $entityId
	 *
	 * @throws EntityLookupException
	 * @return EntityDocument
	 */
	public function getEntity( EntityId $entityId ) {
		if ( in_array( $entityId->getEntityType(), $this->disabledEntityTypes ) ) {
			throw new EntityLookupException(
				$entityId,
				'Entity access for this type of entity is disabled: ' . $entityId->getEntityType()
			);
		}

		return $this->entityLookup->getEntity( $entityId );
	}

	/**
	 * @see EntityLookup::hasEntity
	 *
	 * @param EntityId $entityId
	 *
	 * @return bool
	 * @throws EntityLookupException
	 */
	public function hasEntity( EntityId $entityId ) {
		return $this->entityLookup->hasEntity( $entityId );
	}

}