1 files changed, 267 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/PropertyTableIdReferenceFinder.php b/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/PropertyTableIdReferenceFinder.php
new file mode 100644
index 00000000..bd36d587
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/SQLStore/PropertyTableIdReferenceFinder.php
@@ -0,0 +1,267 @@
+<?php
+
+namespace SMW\SQLStore;
+
+use SMW\ApplicationFactory;
+use SMW\DIProperty;
+use SMWDataItem as DataItem;
+
+/**
+ * @license GNU GPL v2+
+ * @since 2.4
+ *
+ * @author mwjames
+ */
+class PropertyTableIdReferenceFinder {
+
+	/**
+	 * @var SQLStore
+	 */
+	private $store;
+
+	/**
+	 * @var Database
+	 */
+	private $connection;
+
+	/**
+	 * @var NamespaceExaminer
+	 */
+	private $namespaceExaminer;
+
+	/**
+	 * @var boolean
+	 */
+	private $isCapitalLinks = true;
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param SQLStore $store
+	 */
+	public function __construct( SQLStore $store ) {
+		$this->store = $store;
+		$this->connection = $this->store->getConnection( 'mw.db' );
+		$this->namespaceExaminer = ApplicationFactory::getInstance()->getNamespaceExaminer();
+	}
+
+	/**
+	 * @note If $wgCapitalLinks is set false then it will avoid forcing the first
+	 * letter of page titles (including included pages, images and categories)
+	 * to capitals
+	 *
+	 * @since 2.4
+	 *
+	 * @param booelan $isCapitalLinks
+	 */
+	public function isCapitalLinks( $isCapitalLinks ) {
+		$this->isCapitalLinks = $isCapitalLinks;
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param DIProperty $property
+	 *
+	 * @return DataItem|false
+	 */
+	public function tryToFindAtLeastOneReferenceForProperty( DIProperty $property ) {
+
+		$dataItem = $property->getDiWikiPage();
+
+		$sid = $this->store->getObjectIds()->getSMWPageID(
+			$dataItem->getDBkey(),
+			$dataItem->getNamespace(),
+			$dataItem->getInterwiki(),
+			''
+		);
+
+		// Lets see if we have some lower/upper case matching for
+		// when wgCapitalLinks setting was involved
+		if ( !$this->isCapitalLinks && $sid == 0 ) {
+			$sid = $this->store->getObjectIds()->getSMWPageID(
+				lcfirst( $dataItem->getDBkey() ),
+				$dataItem->getNamespace(),
+				$dataItem->getInterwiki(),
+				''
+			);
+		}
+
+		return $this->findAtLeastOneActiveReferenceById( $sid );
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param integer $id
+	 *
+	 * @return boolean
+	 */
+	public function hasResidualPropertyTableReference( $id ) {
+
+		if ( $id == SQLStore::FIXED_PROPERTY_ID_UPPERBOUND ) {
+			return true;
+		}
+
+		return (bool)$this->findAtLeastOneActiveReferenceById( $id, false );
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param integer $id
+	 *
+	 * @return boolean
+	 */
+	public function hasResidualReferenceForId( $id ) {
+
+		if ( $id == SQLStore::FIXED_PROPERTY_ID_UPPERBOUND ) {
+			return true;
+		}
+
+		return (bool)$this->findAtLeastOneActiveReferenceById( $id );
+	}
+
+	/**
+	 * @since 2.5
+	 *
+	 * @param integer $id
+	 *
+	 * @return array
+	 */
+	public function searchAllTablesToFindAtLeastOneReferenceById( $id ) {
+
+		$references = [];
+
+		foreach ( $this->store->getPropertyTables() as $proptable ) {
+			$reference = false;
+
+			if ( ( $reference = $this->findReferenceByPropertyTable( $proptable, $id ) ) !== false ) {
+				$references[$proptable->getName()] = $reference;
+			}
+		}
+
+		if ( ( $reference = $this->findQueryLinksTableReferenceById( $id ) ) !== false ) {
+			$references[SQLStore::QUERY_LINKS_TABLE] = $reference;
+		}
+
+		return $references;
+	}
+
+	/**
+	 * @since 2.4
+	 *
+	 * @param integer $id
+	 * @param boolean $secondary_ref
+	 *
+	 * @return DataItem|false
+	 */
+	public function findAtLeastOneActiveReferenceById( $id, $secondary_ref = true ) {
+
+		$reference = false;
+
+		foreach ( $this->store->getPropertyTables() as $proptable ) {
+
+			if ( ( $reference = $this->findReferenceByPropertyTable( $proptable, $id ) ) !== false ) {
+
+				// If null is returned it means that a reference was found but no DI could
+				// be matched therefore is categorized as false positive
+				if ( isset( $reference->s_id ) ) {
+					$reference = $this->store->getObjectIds()->getDataItemById( $reference->s_id );
+
+					// If the reference is for some reason not part of a  supported namespace,
+					// it is assumed to be invalid
+					if ( $reference !== null && !$this->namespaceExaminer->isSemanticEnabled( $reference->getNamespace() ) ) {
+						$reference = false;
+					}
+				}
+			}
+
+			if ( $reference instanceof DataItem ) {
+				return $reference;
+			}
+		}
+
+		if ( $secondary_ref && !isset( $reference->s_id ) ) {
+			$reference = $this->findQueryLinksTableReferenceById( $id );
+		}
+
+		if ( isset( $reference->s_id ) ) {
+			$reference = $this->store->getObjectIds()->getDataItemById( $reference->s_id );
+		}
+
+		if ( $reference === false || $reference === null ) {
+			return false;
+		}
+
+		return $reference;
+	}
+
+	private function findReferenceByPropertyTable( $proptable, $id ) {
+
+		$row = false;
+
+		if ( $proptable->usesIdSubject() ) {
+			$row = $this->connection->selectRow(
+				$proptable->getName(),
+				[ 's_id' ],
+				[ 's_id' => $id ],
+				__METHOD__
+			);
+		}
+
+		if ( $row !== false ) {
+			return $row;
+		}
+
+		$fields = $proptable->getFields( $this->store );
+
+		// Check whether an object reference exists or not
+		if ( isset( $fields['o_id'] ) ) {
+
+			// This next time someone ... I'm going to Alaska
+			$field = strpos( $proptable->getName(), 'redi' ) ? [ 's_title', 's_namespace' ] : [ 's_id' ];
+
+			$row = $this->connection->selectRow(
+				$proptable->getName(),
+				$field,
+				[ 'o_id' => $id ],
+				__METHOD__
+			);
+
+			if ( $row !== false && strpos( $proptable->getName(), 'redi' ) ) {
+				$row->s_id = $this->store->getObjectIds()->findRedirect( $row->s_title, $row->s_namespace );
+			}
+		}
+
+		// If the property table is not a fixed table (== assigns a whole
+		// table to a specific property with the p_id column being suppressed)
+		// then check for the p_id field
+		if ( $row === false && !$proptable->isFixedPropertyTable() ) {
+			$row = $this->connection->selectRow(
+				$proptable->getName(),
+				[ 's_id' ],
+				[ 'p_id' => $id ],
+				__METHOD__
+			);
+		}
+
+		return $row;
+	}
+
+	private function findQueryLinksTableReferenceById( $id ) {
+
+		// If the query table contains a reference then we keep the object (could
+		// be a subject, property, or printrequest) where in case the query is
+		// removed the object will also loose its reference
+		$row = $this->connection->selectRow(
+			SQLStore::QUERY_LINKS_TABLE,
+			[ 's_id' ],
+			[ 'o_id' => $id ],
+			__METHOD__
+		);
+
+		return $row;
+	}
+
+}