1 files changed, 189 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/ChangePropagationEntityFinder.php b/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/ChangePropagationEntityFinder.php
new file mode 100644
index 00000000..c2e3994e
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/ChangePropagationEntityFinder.php
@@ -0,0 +1,189 @@
+<?php
+
+namespace SMW\SQLStore;
+
+use RuntimeException;
+use SMW\DIProperty;
+use SMW\DIWikiPage;
+use SMW\IteratorFactory;
+use SMW\Store;
+
+/**
+ * Find all entities related to a change propagation (only expected
+ * to be used by `ChangePropagationDispatchJob`).
+ *
+ * @license GNU GPL v2+
+ * @since 3.0
+ *
+ * @author mwjames
+ */
+class ChangePropagationEntityFinder {
+
+	/**
+	 * @var Store
+	 */
+	private $store;
+
+	/**
+	 * @var IteratorFactory
+	 */
+	private $iteratorFactory;
+
+	/**
+	 * @var boolean
+	 */
+	private $isTypePropagation = false;
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param Store $store
+	 * @param IteratorFactory $iteratorFactory
+	 */
+	public function __construct( Store $store, IteratorFactory $iteratorFactory ) {
+		$this->store = $store;
+		$this->iteratorFactory = $iteratorFactory;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param boolean $isTypePropagation
+	 */
+	public function isTypePropagation( $isTypePropagation ) {
+		$this->isTypePropagation = (bool)$isTypePropagation;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param DIProperty|DIWikiPage $entity
+	 *
+	 * @return Iterator
+	 * @throws RuntimeException
+	 */
+	public function findAll( $entity ) {
+
+		if ( $entity instanceof DIProperty ) {
+			return $this->findByProperty( $entity );
+		} elseif ( $entity instanceof DIWikiPage ) {
+			return $this->findByCategory( $entity );
+		}
+
+		throw new RuntimeException( 'Cannot match the entity type.' );
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param DIProperty $property
+	 *
+	 * @return Iterator
+	 */
+	public function findByProperty( DIProperty $property ) {
+
+		$dataItems = [];
+		$appendIterator = $this->iteratorFactory->newAppendIterator();
+
+		$res = $this->store->getAllPropertySubjects(
+			$property
+		);
+
+		$appendIterator->add(
+			$res
+		);
+
+		// Select any remaining references that are hidden or have been left out
+		// during an update
+		$appendIterator->add(
+			$this->fetchOtherReferencesOnTypePropagation( $property )
+		);
+
+		$dataItems = $this->store->getPropertySubjects(
+			new DIProperty( DIProperty::TYPE_ERROR ),
+			$property->getCanonicalDiWikiPage()
+		);
+
+		$appendIterator->add(
+			$dataItems
+		);
+
+		return $appendIterator;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param DIWikiPage $category
+	 *
+	 * @return Iterator
+	 */
+	public function findByCategory( DIWikiPage $category ) {
+
+		$appendIterator = $this->iteratorFactory->newAppendIterator();
+
+		$property = new DIProperty( '_INST' );
+
+		$appendIterator->add(
+			$this->store->getPropertySubjects( $property, $category )
+		);
+
+		// Only direct antecedents
+		$dataItems = $this->store->getPropertyValues(
+			$category,
+			new DIProperty( '_SUBC' )
+		);
+
+		foreach ( $dataItems as $dataItem ) {
+			$appendIterator->add(
+				$this->store->getPropertySubjects( $property, $dataItem )
+			);
+		}
+
+		return $appendIterator;
+	}
+
+	private function fetchOtherReferencesOnTypePropagation( $property ) {
+
+		// Find other references only on a type propagation (which causes a
+		// change of table/id assignments) for entity references
+		if ( $this->isTypePropagation === false ) {
+			return [];
+		}
+
+		$connection = $this->store->getConnection( 'mw.db' );
+		$pid = $this->store->getObjectIds()->getSMWPropertyID( $property );
+
+		$dataItemTables =  $this->store->getPropertyTableInfoFetcher()->getDefaultDataItemTables();
+		$idList = [];
+
+		// Matches may temporary create duplicates in regrads to
+		// Store::getAllPropertySubjects but it will be dealt with by the
+		// deduplication in the ChangePropagationUpdateJob
+		foreach ( $dataItemTables as $tableName ) {
+
+			// Select any references that are hidden or remained active
+			$rows = $connection->select(
+				$connection->tableName( $tableName ),
+				[
+					's_id'
+				],
+				[
+					'p_id' => $pid
+				],
+				__METHOD__
+			);
+
+			foreach ( $rows as $row ) {
+				$idList[] = $row->s_id;
+			}
+		}
+
+		if ( $idList === [] ) {
+			return $idList;
+		}
+
+		return $this->store->getObjectIds()->getDataItemPoolHashListFor( $idList );
+	}
+
+}