1 files changed, 185 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/PropertyAnnotators/CategoryPropertyAnnotator.php b/www/wiki/extensions/SemanticMediaWiki/src/PropertyAnnotators/CategoryPropertyAnnotator.php
new file mode 100644
index 00000000..2b2201f3
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/PropertyAnnotators/CategoryPropertyAnnotator.php
@@ -0,0 +1,185 @@
+<?php
+
+namespace SMW\PropertyAnnotators;
+
+use SMW\ApplicationFactory;
+use SMW\DIProperty;
+use SMW\DIWikiPage;
+use SMW\ProcessingErrorMsgHandler;
+use SMW\PropertyAnnotator;
+
+/**
+ * Handling category annotation
+ *
+ * @ingroup SMW
+ *
+ * @license GNU GPL v2+
+ * @since 1.9
+ *
+ * @author mwjames
+ */
+class CategoryPropertyAnnotator extends PropertyAnnotatorDecorator {
+
+	/**
+	 * @var array
+	 */
+	private $categories;
+
+	/**
+	 * @var array|null
+	 */
+	private $hiddenCategories = null;
+
+	/**
+	 * @var boolean
+	 */
+	private $showHiddenCategories = true;
+
+	/**
+	 * @var boolean
+	 */
+	private $useCategoryInstance = true;
+
+	/**
+	 * @var boolean
+	 */
+	private $useCategoryHierarchy = true;
+
+	/**
+	 * @var boolean
+	 */
+	private $useCategoryRedirect = true;
+
+	/**
+	 * @since 1.9
+	 *
+	 * @param PropertyAnnotator $propertyAnnotator
+	 * @param array $categories
+	 */
+	public function __construct( PropertyAnnotator $propertyAnnotator, array $categories ) {
+		parent::__construct( $propertyAnnotator );
+		$this->categories = $categories;
+	}
+
+	/**
+	 * @since 2.3
+	 *
+	 * @param boolean $showHiddenCategories
+	 */
+	public function showHiddenCategories( $showHiddenCategories ) {
+		$this->showHiddenCategories = (bool)$showHiddenCategories;
+	}
+
+	/**
+	 * @since 2.3
+	 *
+	 * @param boolean $useCategoryInstance
+	 */
+	public function useCategoryInstance( $useCategoryInstance ) {
+		$this->useCategoryInstance = (bool)$useCategoryInstance;
+	}
+
+	/**
+	 * @since 2.3
+	 *
+	 * @param boolean $useCategoryHierarchy
+	 */
+	public function useCategoryHierarchy( $useCategoryHierarchy ) {
+		$this->useCategoryHierarchy = (bool)$useCategoryHierarchy;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param boolean $useCategoryRedirect
+	 */
+	public function useCategoryRedirect( $useCategoryRedirect ) {
+		$this->useCategoryRedirect = (bool)$useCategoryRedirect;
+	}
+
+	/**
+	 * @see PropertyAnnotatorDecorator::addPropertyValues
+	 */
+	protected function addPropertyValues() {
+
+		$namespace = $this->getSemanticData()->getSubject()->getNamespace();
+		$property = null;
+
+		$this->processingErrorMsgHandler = new ProcessingErrorMsgHandler(
+			$this->getSemanticData()->getSubject()
+		);
+
+		if ( $this->useCategoryInstance && ( $namespace !== NS_CATEGORY ) ) {
+			$property = new DIProperty( DIProperty::TYPE_CATEGORY );
+		}
+
+		if ( $this->useCategoryHierarchy && ( $namespace === NS_CATEGORY ) ) {
+			$property = new DIProperty( DIProperty::TYPE_SUBCATEGORY );
+		}
+
+		foreach ( $this->categories as $catname ) {
+
+			if ( ( !$this->showHiddenCategories && $this->isHiddenCategory( $catname ) ) || $property === null ) {
+				continue;
+			}
+
+			$this->modifySemanticData( $property, $catname );
+		}
+	}
+
+	private function modifySemanticData( $property, $catname ) {
+
+		$cat = new DIWikiPage( $catname, NS_CATEGORY );
+
+		if ( ( $cat = $this->getRedirectTarget( $cat ) ) && $cat->getNamespace() === NS_CATEGORY ) {
+			return $this->getSemanticData()->addPropertyObjectValue(
+				$property,
+				$cat
+			);
+		}
+
+		$container = $this->processingErrorMsgHandler->newErrorContainerFromMsg(
+			[
+				'smw-category-invalid-redirect-target',
+				str_replace( '_', ' ', $catname )
+			]
+		);
+
+		$this->processingErrorMsgHandler->addToSemanticData(
+			$this->getSemanticData(),
+			$container
+		);
+	}
+
+	private function isHiddenCategory( $catName ) {
+
+		if ( $this->hiddenCategories === null ) {
+
+			$wikipage = ApplicationFactory::getInstance()->newPageCreator()->createPage(
+				$this->getSemanticData()->getSubject()->getTitle()
+			);
+
+			$this->hiddenCategories = $wikipage->getHiddenCategories();
+		}
+
+		foreach ( $this->hiddenCategories as $hiddenCategory ) {
+
+			if ( $hiddenCategory->getText() === $catName ) {
+				return true;
+			};
+
+		}
+
+		return false;
+	}
+
+	private function getRedirectTarget( $subject ) {
+
+		if ( $this->useCategoryRedirect ) {
+			return ApplicationFactory::getInstance()->getStore()->getRedirectTarget( $subject );
+		}
+
+		return $subject;
+	}
+
+}