Queryr\Resources\Builders\SimpleStatementsBuilder

Complexity

Total Complexity

11

Size/Duplication

Total Lines	76
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	8

Test Coverage

Coverage

100%

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	buildFromStatements()	0	20	3
A	getEntityName()	0	5	2
A	getStatementValuesWithPropertyId()	0	16	3
A	handle()	0	9	2

<?php

namespace Queryr\Resources\Builders;

use DataValues\DataValue;
use DataValues\StringValue;
use Queryr\Resources\SimpleStatement;
use Wikibase\DataModel\Entity\EntityId;
use Wikibase\DataModel\Entity\EntityIdValue;
use Wikibase\DataModel\Entity\PropertyId;
use Wikibase\DataModel\Snak\PropertyValueSnak;
use Wikibase\DataModel\Statement\Statement;
use Wikibase\DataModel\Statement\StatementList;

/**
 * @licence GNU GPL v2+
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class SimpleStatementsBuilder {

	private $languageCode;
	private $labelLookup;

	public function __construct( $languageCode, ResourceLabelLookup $labelLookup ) {
		$this->languageCode = $languageCode;
		$this->labelLookup = $labelLookup;
	}

	/**
	 * @param StatementList $statements
	 *
	 * @return array
	 */
	public function buildFromStatements( StatementList $statements ) {
		$simpleStatements = [];

		foreach ( $statements->getPropertyIds() as $propertyId ) {
			$simpleStatement = new SimpleStatement();

			$statementValues = $this->getStatementValuesWithPropertyId( $statements, $propertyId );

			if ( !empty( $statementValues ) ) {
				$simpleStatement->values = $statementValues;
				$simpleStatement->valueType = $statementValues[0]->getType();
				$simpleStatement->propertyName = $this->getEntityName( $propertyId );
				$simpleStatement->propertyId = $propertyId;

				$simpleStatements[] = $simpleStatement;
			}
		}

		return $simpleStatements;
	}

	private function getEntityName( EntityId $id ) {
		$label = $this->labelLookup->getLabelByIdAndLanguage( $id, $this->languageCode );

		return $label === null ? $id->getSerialization() : $label;
	}

	/**
	 * @param StatementList $statements
	 * @param PropertyId $propertyId
	 *
	 * @return DataValue[]
	 */
	private function getStatementValuesWithPropertyId( StatementList $statements, PropertyId $propertyId ) {
		$statementValues = [];

		/**
		 * @var Statement $statement
		 */
		foreach ( $statements->getByPropertyId( $propertyId )->getBestStatements() as $statement ) {
			$snak = $statement->getMainSnak();

			if ( $snak instanceof PropertyValueSnak ) {
				$statementValues[] = $this->handle( $snak );
			}
		}

		return $statementValues;
	}

	private function handle( PropertyValueSnak $snak ) {
		$value = $snak->getDataValue();

		if ( $value instanceof EntityIdValue ) {
			return new StringValue( $this->getEntityName( $value->getEntityId() ) );
		}

		return $value;
	}

}