Queryr\TermStore\TermStoreWriter

Complexity

Total Complexity

11

Size/Duplication

Total Lines	108
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Test Coverage

Coverage

93.75%

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	storeEntityFingerprint()	0	15	2
A	storeFingerprintParts()	0	15	3
A	dropTermsForId()	0	16	2
A	storeLabel()	0	12	1
A	storeAliases()	0	14	2

<?php

namespace Queryr\TermStore;

use Doctrine\DBAL\Connection;
use Doctrine\DBAL\DBALException;
use Wikibase\DataModel\Entity\EntityId;
use Wikibase\DataModel\Term\AliasGroup;
use Wikibase\DataModel\Term\Fingerprint;
use Wikibase\DataModel\Term\Term;

/**
 * Package public
 * @since 0.2
 *
 * @licence GNU GPL v2+
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class TermStoreWriter {

	private $connection;
	private $config;

	public function __construct( Connection $connection, TermStoreConfig $config ) {
		$this->connection = $connection;
		$this->config = $config;
	}

	/**
	 * @param EntityId $id
	 * @param Fingerprint $fingerprint
	 *
	 * @throws TermStoreException
	 */
	public function storeEntityFingerprint( EntityId $id, Fingerprint $fingerprint ): void {
		$this->connection->beginTransaction();

		$this->dropTermsForId( $id );

		try {
			$this->storeFingerprintParts( $id, $fingerprint );
		}
		catch ( DBALException $ex ) {
			$this->connection->rollBack();
			throw new TermStoreException( $ex->getMessage(), $ex );
		}

		$this->connection->commit();
	}

	/**
	 * @throws DBALException
	 */
	private function storeFingerprintParts( EntityId $id, Fingerprint $fingerprint ) {
		/**
		 * @var Term $label
		 */
		foreach ( $fingerprint->getLabels() as $label ) {
			$this->storeLabel( $label->getLanguageCode(), $label->getText(), $id );
		}

		/**
		 * @var AliasGroup $aliasGroup
		 */
		foreach ( $fingerprint->getAliasGroups() as $aliasGroup ) {
			$this->storeAliases( $aliasGroup, $id );
		}
	}

	/**
	 * @param EntityId $id
	 *
	 * @throws TermStoreException
	 */
	public function dropTermsForId( EntityId $id ): void {
		try {
			$this->connection->delete(
				$this->config->getLabelTableName(),
				[ 'entity_id' => $id->getSerialization() ]
			);

			$this->connection->delete(
				$this->config->getAliasesTableName(),
				[ 'entity_id' => $id->getSerialization() ]
			);
		}
		catch ( DBALException $ex ) {
			throw new TermStoreException( $ex->getMessage(), $ex );
		}
	}

	/**
	 * @throws DBALException
	 */
	private function storeLabel( string $languageCode, string $text, EntityId $id ) {
		$this->connection->insert(
			$this->config->getLabelTableName(),
			[
				'text' => $text,
				'text_lowercase' => strtolower( $text ),
				'language' => $languageCode,
				'entity_id' => $id->getSerialization(),
				'entity_type' => $id->getEntityType()
			]
		);
	}

	/**
	 * @throws DBALException
	 */
	private function storeAliases( AliasGroup $aliasGroup, EntityId $id ) {
		foreach ( $aliasGroup->getAliases() as $alias ) {
			$this->connection->insert(
				$this->config->getAliasesTableName(),
				[
					'text' => $alias,
					'text_lowercase' => strtolower( $alias ),
					'language' => $aliasGroup->getLanguageCode(),
					'entity_id' => $id->getSerialization(),
					'entity_type' => $id->getEntityType()
				]
			);
		}
	}

}