1 files changed, 128 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/Hooks/FileUpload.php b/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/Hooks/FileUpload.php
new file mode 100644
index 00000000..97d3bd0b
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/Hooks/FileUpload.php
@@ -0,0 +1,128 @@
+<?php
+
+namespace SMW\MediaWiki\Hooks;
+
+use File;
+use Hooks;
+use ParserOptions;
+use SMW\ApplicationFactory;
+use SMW\Localizer;
+use SMW\NamespaceExaminer;
+use Title;
+use User;
+
+/**
+ * Fires when a local file upload occurs
+ *
+ * @see https://www.mediawiki.org/wiki/Manual:Hooks/FileUpload
+ *
+ * @license GNU GPL v2+
+ * @since 1.9
+ *
+ * @author mwjames
+ */
+class FileUpload extends HookHandler {
+
+	/**
+	 * @var NamespaceExaminer
+	 */
+	private $namespaceExaminer;
+
+	/**
+	 * @since 1.9
+	 *
+	 * @param NamespaceExaminer $namespaceExaminer
+	 */
+	public function __construct( NamespaceExaminer $namespaceExaminer ) {
+		$this->namespaceExaminer = $namespaceExaminer;
+	}
+
+	/**
+	 * @since 3.0
+	 *
+	 * @param File $file
+	 * @param boolean $reUploadStatus
+	 *
+	 * @return true
+	 */
+	public function process( File $file, $reUploadStatus = false ) {
+
+		 if ( $this->canProcess( $file->getTitle() ) ) {
+			$this->doProcess( $file, $reUploadStatus );
+		}
+
+		return true;
+	}
+
+	private function canProcess( $title ) {
+		return $title !== null && $this->namespaceExaminer->isSemanticEnabled( $title->getNamespace() );
+	}
+
+	private function doProcess( $file, $reUploadStatus = false ) {
+
+		$applicationFactory = ApplicationFactory::getInstance();
+		$filePage = $this->makeFilePage( $file, $reUploadStatus );
+
+		// Avoid WikiPage.php: The supplied ParserOptions are not safe to cache.
+		// Fix the options or set $forceParse = true.
+		$forceParse = true;
+
+		$parserData = $applicationFactory->newParserData(
+			$file->getTitle(),
+			$filePage->getParserOutput( $this->makeCanonicalParserOptions(), null, $forceParse )
+		);
+
+		$pageInfoProvider = $applicationFactory->newMwCollaboratorFactory()->newPageInfoProvider(
+			$filePage
+		);
+
+		$propertyAnnotatorFactory = $applicationFactory->singleton( 'PropertyAnnotatorFactory' );
+
+		$semanticData = $parserData->getSemanticData();
+		$semanticData->setOption( 'is.fileupload', true );
+
+		$propertyAnnotator = $propertyAnnotatorFactory->newNullPropertyAnnotator(
+			$semanticData
+		);
+
+		$propertyAnnotator = $propertyAnnotatorFactory->newPredefinedPropertyAnnotator(
+			$propertyAnnotator,
+			$pageInfoProvider
+		);
+
+		$propertyAnnotator->addAnnotation();
+
+		// 2.4+
+		Hooks::run( 'SMW::FileUpload::BeforeUpdate', [ $filePage, $semanticData ] );
+
+		$parserData->setOrigin( 'FileUpload' );
+
+		$parserData->pushSemanticDataToParserOutput();
+		$parserData->updateStore( true );
+
+		return true;
+	}
+
+	private function makeFilePage( $file, $reUploadStatus ) {
+
+		$filePage = ApplicationFactory::getInstance()->newPageCreator()->createFilePage(
+			$file->getTitle()
+		);
+
+		$filePage->setFile( $file );
+		$filePage->smwFileReUploadStatus = $reUploadStatus;
+
+		return $filePage;
+	}
+
+	/**
+	 * Anonymous user with default preferences and content language
+	 */
+	private function makeCanonicalParserOptions() {
+		return ParserOptions::newFromUserAndLang(
+			new User(),
+			Localizer::getInstance()->getContentLanguage()
+		);
+	}
+
+}