1 files changed, 82 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/Search/SearchResult.php b/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/Search/SearchResult.php
new file mode 100644
index 00000000..28f650d7
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/Search/SearchResult.php
@@ -0,0 +1,82 @@
+<?php
+
+namespace SMW\MediaWiki\Search;
+
+use SMW\DataValueFactory;
+use SMW\DIWikiPage;
+
+/**
+ * @ingroup SMW
+ *
+ * @license GNU GPL v2+
+ * @since 3.0
+ *
+ * @author mwjames
+ */
+class SearchResult extends \SearchResult {
+
+	/**
+	 * @var boolean
+	 */
+	private $hasHighlight = false;
+
+	/**
+	 * @see SearchResult::getTextSnippet
+	 */
+	function getTextSnippet( $terms ) {
+
+		if ( $this->hasHighlight ) {
+			return str_replace( [ '<em>', '</em>' ], [ "<span class='searchmatch'>", '</span>' ], $this->mText );
+		}
+
+		return parent::getTextSnippet( $terms );
+	}
+
+	/**
+	 * @see SearchResult::getSectionTitle
+	 */
+	function getSectionTitle() {
+
+		if ( !isset( $this->mTitle ) || $this->mTitle->getFragment() === '' ) {
+			return null;
+		}
+
+		return $this->mTitle;
+	}
+
+	/**
+	 * Set a text excerpt retrieved from a different back-end.
+	 *
+	 * @param string $text|null
+	 * @param boolean $hasHighlight
+	 */
+	public function setExcerpt( $text = null, $hasHighlight = false ) {
+		$this->mText = $text;
+		$this->hasHighlight = $hasHighlight;
+	}
+
+	/**
+	 * @return string|null
+	 */
+	public function getExcerpt() {
+		return $this->mText;
+	}
+
+	/**
+	 * @see SearchResult::getTitleSnippet
+	 */
+	public function getTitleSnippet() {
+
+		if ( !isset( $this->mTitle ) ) {
+			return '';
+		}
+
+		$dataValue = DataValueFactory::getInstance()->newDataValueByItem(
+			DIWikiPage::newFromTitle( $this->mTitle )
+		);
+
+		// Will return the DISPLAYTITLE, if available
+		return $dataValue->getPreferredCaption();
+	}
+
+}