ImportReporter.php

Go to the documentation of this file.

<?php
use MediaWiki\HookContainer\ProtectedHookAccessorTrait;
use MediaWiki\MediaWikiServices;
use MediaWiki\Page\PageIdentity;
 
class ImportReporter extends ContextSource {
    use ProtectedHookAccessorTrait;
 
    private $reason;
    private $logTags = [];
    private $mOriginalLogCallback;
    private $mOriginalPageOutCallback;
    private $mLogItemCount = 0;
    private $mPageCount;
    private $mIsUpload;
    private $mInterwiki;
 
    public function __construct( $importer, $upload, $interwiki, $reason = false ) {
        $this->mOriginalPageOutCallback =
            $importer->setPageOutCallback( [ $this, 'reportPage' ] );
        $this->mOriginalLogCallback =
            $importer->setLogItemCallback( [ $this, 'reportLogItem' ] );
        $importer->setNoticeCallback( [ $this, 'reportNotice' ] );
        $this->mPageCount = 0;
        $this->mIsUpload = $upload;
        $this->mInterwiki = $interwiki;
        $this->reason = $reason;
    }
 
    public function setChangeTags( array $tags ) {
        $this->logTags = $tags;
    }
 
    public function open() {
        $this->getOutput()->addHTML( "<ul>\n" );
    }
 
    private function reportNotice( $msg, array $params ) {
        $this->getOutput()->addHTML(
            Html::element( 'li', [], $this->msg( $msg, $params )->text() )
        );
    }
 
    private function reportLogItem( ...$args ) {
        $this->mLogItemCount++;
        if ( is_callable( $this->mOriginalLogCallback ) ) {
            call_user_func_array( $this->mOriginalLogCallback, $args );
        }
    }
 
    public function reportPage( ?PageIdentity $pageIdentity, $foreignTitle, $revisionCount,
            $successCount, $pageInfo ) {
        call_user_func_array( $this->mOriginalPageOutCallback, func_get_args() );
 
        if ( $pageIdentity === null ) {
            # Invalid or non-importable title; a notice is already displayed
            return;
        }
 
        $this->mPageCount++;
        $services = MediaWikiServices::getInstance();
        $linkRenderer = $services->getLinkRenderer();
        if ( $successCount > 0 ) {
            // <bdi> prevents jumbling of the versions count
            // in RTL wikis in case the page title is LTR
            $this->getOutput()->addHTML(
                "<li>" . $linkRenderer->makeLink( $pageIdentity ) . " " .
                    "<bdi>" .
                    $this->msg( 'import-revision-count' )->numParams( $successCount )->escaped() .
                    "</bdi>" .
                    "</li>\n"
            );
 
            $logParams = [ '4:number:count' => $successCount ];
            if ( $this->mIsUpload ) {
                $detail = $this->msg( 'import-logentry-upload-detail' )->numParams(
                    $successCount )->inContentLanguage()->text();
                $action = 'upload';
            } else {
                $pageTitle = $foreignTitle->getFullText();
                $fullInterwikiPrefix = $this->mInterwiki;
                $this->getHookRunner()->onImportLogInterwikiLink(
                    $fullInterwikiPrefix, $pageTitle );
 
                $interwikiTitleStr = $fullInterwikiPrefix . ':' . $pageTitle;
                $interwiki = '[[:' . $interwikiTitleStr . ']]';
                $detail = $this->msg( 'import-logentry-interwiki-detail' )->numParams(
                    $successCount )->params( $interwiki )->inContentLanguage()->text();
                $action = 'interwiki';
                $logParams['5:title-link:interwiki'] = $interwikiTitleStr;
            }
            if ( $this->reason ) {
                $detail .= $this->msg( 'colon-separator' )->inContentLanguage()->text()
                    . $this->reason;
            }
 
            $comment = CommentStoreComment::newUnsavedComment( $detail );
            $dbw = wfGetDB( DB_PRIMARY );
            $revStore = $services->getRevisionStore();
            $nullRevRecord = $revStore->newNullRevision(
                $dbw,
                $pageIdentity,
                $comment,
                true,
                $this->getUser()
            );
 
            $nullRevId = null;
            if ( $nullRevRecord !== null ) {
                $inserted = $revStore->insertRevisionOn( $nullRevRecord, $dbw );
                $nullRevId = $inserted->getId();
                $parentRevId = $inserted->getParentId();
                $page = $services->getWikiPageFactory()->newFromTitle( $pageIdentity );
 
                // Update page record
                $page->updateRevisionOn( $dbw, $inserted );
 
                $fakeTags = [];
                $this->getHookRunner()->onRevisionFromEditComplete(
                    $page, $inserted, $parentRevId, $this->getUser(), $fakeTags
                );
            }
 
            // Create the import log entry
            $logEntry = new ManualLogEntry( 'import', $action );
            $logEntry->setTarget( TitleValue::castPageToLinkTarget( $pageIdentity ) );
            $logEntry->setComment( $this->reason );
            $logEntry->setPerformer( $this->getUser() );
            $logEntry->setParameters( $logParams );
            // Make sure the null revision will be tagged as well
            $logEntry->setAssociatedRevId( $nullRevId );
            if ( count( $this->logTags ) ) {
                $logEntry->addTags( $this->logTags );
            }
            $logid = $logEntry->insert();
            $logEntry->publish( $logid );
        } else {
            $this->getOutput()->addHTML( "<li>" . $linkRenderer->makeKnownLink( $pageIdentity ) . " " .
                $this->msg( 'import-nonewrevisions' )->escaped() . "</li>\n" );
        }
    }
 
    public function close() {
        $out = $this->getOutput();
        if ( $this->mLogItemCount > 0 ) {
            $msg = $this->msg( 'imported-log-entries' )->numParams( $this->mLogItemCount )->parse();
            $out->addHTML( Xml::tags( 'li', null, $msg ) );
        } elseif ( $this->mPageCount == 0 && $this->mLogItemCount == 0 ) {
            $out->addHTML( "</ul>\n" );
 
            return Status::newFatal( 'importnopages' );
        }
        $out->addHTML( "</ul>\n" );
 
        return Status::newGood( $this->mPageCount );
    }
}