1 files changed, 169 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/Query/ProfileAnnotatorFactory.php b/www/wiki/extensions/SemanticMediaWiki/src/Query/ProfileAnnotatorFactory.php
new file mode 100644
index 00000000..52f1e618
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/Query/ProfileAnnotatorFactory.php
@@ -0,0 +1,169 @@
+<?php
+
+namespace SMW\Query;
+
+use SMW\DIWikiPage;
+use SMW\Query\ProfileAnnotators\DescriptionProfileAnnotator;
+use SMW\Query\ProfileAnnotators\DurationProfileAnnotator;
+use SMW\Query\ProfileAnnotators\FormatProfileAnnotator;
+use SMW\Query\ProfileAnnotators\NullProfileAnnotator;
+use SMW\Query\ProfileAnnotators\ParametersProfileAnnotator;
+use SMW\Query\ProfileAnnotators\SourceProfileAnnotator;
+use SMW\Query\ProfileAnnotators\StatusCodeProfileAnnotator;
+use SMW\Query\ProfileAnnotators\SchemaLinkProfileAnnotator;
+use SMWContainerSemanticData as ContainerSemanticData;
+use SMWDIContainer as DIContainer;
+use SMWQuery as Query;
+
+/**
+ * @license GNU GPL v2+
+ * @since 2.1
+ *
+ * @author mwjames
+ */
+class ProfileAnnotatorFactory {
+
+	/**
+	 * @since 2.1
+	 *
+	 * @param Query $query
+	 * @param string $format
+	 *
+	 * @return ProfileAnnotator
+	 */
+	public function newProfileAnnotator( Query $query, $format ) {
+
+		$profileAnnotator = $this->newDescriptionProfileAnnotator(
+			$query
+		);
+
+		$profileAnnotator = $this->newFormatProfileAnnotator(
+			$profileAnnotator,
+			$format
+		);
+
+		$profileAnnotator = $this->newParametersProfileAnnotator(
+			$profileAnnotator,
+			$query
+		);
+
+		$profileAnnotator = $this->newDurationProfileAnnotator(
+			$profileAnnotator,
+			$query->getOption( Query::PROC_QUERY_TIME )
+		);
+
+		$profileAnnotator = $this->newSourceProfileAnnotator(
+			$profileAnnotator,
+			$query->getQuerySource()
+		);
+
+		$profileAnnotator = $this->newStatusCodeProfileAnnotator(
+			$profileAnnotator,
+			$query->getOption( Query::PROC_STATUS_CODE )
+		);
+
+		$profileAnnotator = $this->newSchemaLinkProfileAnnotator(
+			$profileAnnotator,
+			$query->getOption( 'schema_link' )
+		);
+
+		return $profileAnnotator;
+	}
+
+	/**
+	 * @since 2.5
+	 *
+	 * @param Query $query
+	 *
+	 * @return DescriptionProfileAnnotator
+	 */
+	public function newDescriptionProfileAnnotator( Query $query ) {
+
+		$profileAnnotator = new NullProfileAnnotator(
+			$this->newDIContainer( $query )
+		);
+
+		$profileAnnotator = new DescriptionProfileAnnotator(
+			$profileAnnotator,
+			$query->getDescription()
+		);
+
+		return $profileAnnotator;
+	}
+
+	private function newFormatProfileAnnotator( $profileAnnotator, $format ) {
+		return new FormatProfileAnnotator( $profileAnnotator, $format );
+	}
+
+	private function newParametersProfileAnnotator( $profileAnnotator, $query ) {
+
+		if ( $query->getOption( Query::OPT_PARAMETERS ) === false ) {
+			return $profileAnnotator;
+		}
+
+		return new ParametersProfileAnnotator( $profileAnnotator, $query );
+	}
+
+	private function newDurationProfileAnnotator( $profileAnnotator, $duration ) {
+
+		if ( $duration == 0 ) {
+			return $profileAnnotator;
+		}
+
+		return new DurationProfileAnnotator( $profileAnnotator, $duration );
+	}
+
+	private function newSourceProfileAnnotator( $profileAnnotator, $querySource ) {
+
+		if ( $querySource === '' || $querySource === null ) {
+			return $profileAnnotator;
+		}
+
+		return new SourceProfileAnnotator( $profileAnnotator, $querySource );
+	}
+
+	private function newStatusCodeProfileAnnotator( $profileAnnotator, $statusCodes ) {
+
+		if ( $statusCodes === false || $statusCodes === null || $statusCodes === [] ) {
+			return $profileAnnotator;
+		}
+
+		return new StatusCodeProfileAnnotator( $profileAnnotator, $statusCodes );
+	}
+
+	private function newSchemaLinkProfileAnnotator( $profileAnnotator, $schemaLink ) {
+
+		if ( $schemaLink === false || $schemaLink === null ) {
+			return $profileAnnotator;
+		}
+
+		return new SchemaLinkProfileAnnotator( $profileAnnotator, $schemaLink );
+	}
+
+	/**
+	 * #1416 create container manually to avoid any issues that may arise from
+	 * a failed Title::makeTitleSafe.
+	 */
+	private function newDIContainer( Query $query ) {
+
+		$subject = $query->getContextPage();
+
+		if ( $subject === null ) {
+			$containerSemanticData = ContainerSemanticData::makeAnonymousContainer();
+		} else {
+			$subject = new DIWikiPage(
+				$subject->getDBkey(),
+				$subject->getNamespace(),
+				$subject->getInterwiki(),
+				$query->getQueryId()
+			);
+
+			$containerSemanticData = new ContainerSemanticData( $subject );
+		}
+
+		return new DIContainer(
+			$containerSemanticData
+		);
+	}
+
+}