SpecialImport.php

Go to the documentation of this file.

<?php
class SpecialImport extends SpecialPage {
    private $interwiki = false;
    private $namespace;
    private $rootpage = '';
    private $frompage = '';
    private $logcomment = false;
    private $history = true;
    private $includeTemplates = false;
    private $pageLinkDepth;

    public function __construct() {
        parent::__construct( 'Import', 'import' );
        global $wgImportTargetNamespace;
        $this->namespace = $wgImportTargetNamespace;
    }

    function execute( $par ) {
        $this->setHeaders();
        $this->outputHeader();

        $user = $this->getUser();
        if ( !$user->isAllowedAny( 'import', 'importupload' ) ) {
            throw new PermissionsError( 'import' );
        }

        # @todo Allow Title::getUserPermissionsErrors() to take an array
        # @todo FIXME: Title::checkSpecialsAndNSPermissions() has a very wierd expectation of what
        # getUserPermissionsErrors() might actually be used for, hence the 'ns-specialprotected'
        $errors = wfMergeErrorArrays(
            $this->getTitle()->getUserPermissionsErrors(
                'import', $user, true,
                array( 'ns-specialprotected', 'badaccess-group0', 'badaccess-groups' )
            ),
            $this->getTitle()->getUserPermissionsErrors(
                'importupload', $user, true,
                array( 'ns-specialprotected', 'badaccess-group0', 'badaccess-groups' )
            )
        );

        if ( $errors ) {
            throw new PermissionsError( 'import', $errors );
        }

        $this->checkReadOnly();

        $request = $this->getRequest();
        if ( $request->wasPosted() && $request->getVal( 'action' ) == 'submit' ) {
            $this->doImport();
        }
        $this->showForm();
    }

    private function doImport() {
        global $wgImportSources, $wgExportMaxLinkDepth;

        $isUpload = false;
        $request = $this->getRequest();
        $this->namespace = $request->getIntOrNull( 'namespace' );
        $sourceName = $request->getVal( "source" );

        $this->logcomment = $request->getText( 'log-comment' );
        $this->pageLinkDepth = $wgExportMaxLinkDepth == 0 ? 0 : $request->getIntOrNull( 'pagelink-depth' );
        $this->rootpage = $request->getText( 'rootpage' );

        $user = $this->getUser();
        if ( !$user->matchEditToken( $request->getVal( 'editToken' ) ) ) {
            $source = Status::newFatal( 'import-token-mismatch' );
        } elseif ( $sourceName == 'upload' ) {
            $isUpload = true;
            if ( $user->isAllowed( 'importupload' ) ) {
                $source = ImportStreamSource::newFromUpload( "xmlimport" );
            } else {
                throw new PermissionsError( 'importupload' );
            }
        } elseif ( $sourceName == "interwiki" ) {
            if ( !$user->isAllowed( 'import' ) ) {
                throw new PermissionsError( 'import' );
            }
            $this->interwiki = $request->getVal( 'interwiki' );
            if ( !in_array( $this->interwiki, $wgImportSources ) ) {
                $source = Status::newFatal( "import-invalid-interwiki" );
            } else {
                $this->history = $request->getCheck( 'interwikiHistory' );
                $this->frompage = $request->getText( "frompage" );
                $this->includeTemplates = $request->getCheck( 'interwikiTemplates' );
                $source = ImportStreamSource::newFromInterwiki(
                    $this->interwiki,
                    $this->frompage,
                    $this->history,
                    $this->includeTemplates,
                    $this->pageLinkDepth );
            }
        } else {
            $source = Status::newFatal( "importunknownsource" );
        }

        $out = $this->getOutput();
        if ( !$source->isGood() ) {
            $out->wrapWikiMsg(
                "<p class=\"error\">\n$1\n</p>",
                array( 'importfailed', $source->getWikiText() )
            );
        } else {
            $importer = new WikiImporter( $source->value );
            if ( !is_null( $this->namespace ) ) {
                $importer->setTargetNamespace( $this->namespace );
            }
            if ( !is_null( $this->rootpage ) ) {
                $statusRootPage = $importer->setTargetRootPage( $this->rootpage );
                if ( !$statusRootPage->isGood() ) {
                    $out->wrapWikiMsg(
                        "<p class=\"error\">\n$1\n</p>",
                        array(
                            'import-options-wrong',
                            $statusRootPage->getWikiText(),
                            count( $statusRootPage->getErrorsArray() )
                        )
                    );

                    return;
                }
            }

            $out->addWikiMsg( "importstart" );

            $reporter = new ImportReporter(
                $importer,
                $isUpload,
                $this->interwiki,
                $this->logcomment
            );
            $reporter->setContext( $this->getContext() );
            $exception = false;

            $reporter->open();
            try {
                $importer->doImport();
            } catch ( MWException $e ) {
                $exception = $e;
            }
            $result = $reporter->close();

            if ( $exception ) {
                # No source or XML parse error
                $out->wrapWikiMsg(
                    "<p class=\"error\">\n$1\n</p>",
                    array( 'importfailed', $exception->getMessage() )
                );
            } elseif ( !$result->isGood() ) {
                # Zero revisions
                $out->wrapWikiMsg(
                    "<p class=\"error\">\n$1\n</p>",
                    array( 'importfailed', $result->getWikiText() )
                );
            } else {
                # Success!
                $out->addWikiMsg( 'importsuccess' );
            }
            $out->addHTML( '<hr />' );
        }
    }

    private function showForm() {
        global $wgImportSources, $wgExportMaxLinkDepth;

        $action = $this->getTitle()->getLocalURL( array( 'action' => 'submit' ) );
        $user = $this->getUser();
        $out = $this->getOutput();

        if ( $user->isAllowed( 'importupload' ) ) {
            $out->addHTML(
                Xml::fieldset( $this->msg( 'import-upload' )->text() ) .
                    Xml::openElement(
                        'form',
                        array(
                            'enctype' => 'multipart/form-data',
                            'method' => 'post',
                            'action' => $action,
                            'id' => 'mw-import-upload-form'
                        )
                    ) .
                    $this->msg( 'importtext' )->parseAsBlock() .
                    Html::hidden( 'action', 'submit' ) .
                    Html::hidden( 'source', 'upload' ) .
                    Xml::openElement( 'table', array( 'id' => 'mw-import-table-upload' ) ) .
                    "<tr>
                    <td class='mw-label'>" .
                    Xml::label( $this->msg( 'import-upload-filename' )->text(), 'xmlimport' ) .
                    "</td>
                    <td class='mw-input'>" .
                    Html::input( 'xmlimport', '', 'file', array( 'id' => 'xmlimport' ) ) . ' ' .
                    "</td>
                </tr>
                <tr>
                    <td class='mw-label'>" .
                    Xml::label( $this->msg( 'import-comment' )->text(), 'mw-import-comment' ) .
                    "</td>
                    <td class='mw-input'>" .
                    Xml::input( 'log-comment', 50, '',
                        array( 'id' => 'mw-import-comment', 'type' => 'text' ) ) . ' ' .
                    "</td>
                </tr>
                <tr>
                    <td class='mw-label'>" .
                    Xml::label( $this->msg( 'import-interwiki-rootpage' )->text(), 'mw-interwiki-rootpage-upload' ) .
                    "</td>
                    <td class='mw-input'>" .
                    Xml::input( 'rootpage', 50, $this->rootpage,
                        array( 'id' => 'mw-interwiki-rootpage-upload', 'type' => 'text' ) ) . ' ' .
                    "</td>
                </tr>
                <tr>
                    <td></td>
                    <td class='mw-submit'>" .
                    Xml::submitButton( $this->msg( 'uploadbtn' )->text() ) .
                    "</td>
                </tr>" .
                    Xml::closeElement( 'table' ) .
                    Html::hidden( 'editToken', $user->getEditToken() ) .
                    Xml::closeElement( 'form' ) .
                    Xml::closeElement( 'fieldset' )
            );
        } else {
            if ( empty( $wgImportSources ) ) {
                $out->addWikiMsg( 'importnosources' );
            }
        }

        if ( $user->isAllowed( 'import' ) && !empty( $wgImportSources ) ) {
            # Show input field for import depth only if $wgExportMaxLinkDepth > 0
            $importDepth = '';
            if ( $wgExportMaxLinkDepth > 0 ) {
                $importDepth = "<tr>
                            <td class='mw-label'>" .
                    $this->msg( 'export-pagelinks' )->parse() .
                    "</td>
                            <td class='mw-input'>" .
                    Xml::input( 'pagelink-depth', 3, 0 ) .
                    "</td>
                </tr>";
            }

            $out->addHTML(
                Xml::fieldset( $this->msg( 'importinterwiki' )->text() ) .
                    Xml::openElement(
                        'form',
                        array(
                            'method' => 'post',
                            'action' => $action,
                            'id' => 'mw-import-interwiki-form'
                        )
                    ) .
                    $this->msg( 'import-interwiki-text' )->parseAsBlock() .
                    Html::hidden( 'action', 'submit' ) .
                    Html::hidden( 'source', 'interwiki' ) .
                    Html::hidden( 'editToken', $user->getEditToken() ) .
                    Xml::openElement( 'table', array( 'id' => 'mw-import-table-interwiki' ) ) .
                    "<tr>
                    <td class='mw-label'>" .
                    Xml::label( $this->msg( 'import-interwiki-source' )->text(), 'interwiki' ) .
                    "</td>
                    <td class='mw-input'>" .
                    Xml::openElement(
                        'select',
                        array( 'name' => 'interwiki', 'id' => 'interwiki' )
                    )
            );

            foreach ( $wgImportSources as $prefix ) {
                $selected = ( $this->interwiki === $prefix ) ? ' selected="selected"' : '';
                $out->addHTML( Xml::option( $prefix, $prefix, $selected ) );
            }

            $out->addHTML(
                Xml::closeElement( 'select' ) .
                    Xml::input( 'frompage', 50, $this->frompage, array( 'id' => 'frompage' ) ) .
                    "</td>
                </tr>
                <tr>
                    <td>
                    </td>
                    <td class='mw-input'>" .
                    Xml::checkLabel(
                        $this->msg( 'import-interwiki-history' )->text(),
                        'interwikiHistory',
                        'interwikiHistory',
                        $this->history
                    ) .
                    "</td>
                </tr>
                <tr>
                    <td>
                    </td>
                    <td class='mw-input'>" .
                    Xml::checkLabel(
                        $this->msg( 'import-interwiki-templates' )->text(),
                        'interwikiTemplates',
                        'interwikiTemplates',
                        $this->includeTemplates
                    ) .
                    "</td>
                </tr>
                $importDepth
                <tr>
                    <td class='mw-label'>" .
                    Xml::label( $this->msg( 'import-interwiki-namespace' )->text(), 'namespace' ) .
                    "</td>
                    <td class='mw-input'>" .
                    Html::namespaceSelector(
                        array(
                            'selected' => $this->namespace,
                            'all' => '',
                        ), array(
                            'name' => 'namespace',
                            'id' => 'namespace',
                            'class' => 'namespaceselector',
                        )
                    ) .
                    "</td>
                </tr>
                <tr>
                    <td class='mw-label'>" .
                    Xml::label( $this->msg( 'import-comment' )->text(), 'mw-interwiki-comment' ) .
                    "</td>
                    <td class='mw-input'>" .
                    Xml::input( 'log-comment', 50, '',
                        array( 'id' => 'mw-interwiki-comment', 'type' => 'text' ) ) . ' ' .
                    "</td>
                </tr>
                <tr>
                    <td class='mw-label'>" .
                    Xml::label(
                        $this->msg( 'import-interwiki-rootpage' )->text(),
                        'mw-interwiki-rootpage-interwiki'
                    ) .
                    "</td>
                    <td class='mw-input'>" .
                    Xml::input( 'rootpage', 50, $this->rootpage,
                        array( 'id' => 'mw-interwiki-rootpage-interwiki', 'type' => 'text' ) ) . ' ' .
                    "</td>
                </tr>
                <tr>
                    <td>
                    </td>
                    <td class='mw-submit'>" .
                    Xml::submitButton(
                        $this->msg( 'import-interwiki-submit' )->text(),
                        Linker::tooltipAndAccesskeyAttribs( 'import' )
                    ) .
                    "</td>
                </tr>" .
                    Xml::closeElement( 'table' ) .
                    Xml::closeElement( 'form' ) .
                    Xml::closeElement( 'fieldset' )
            );
        }
    }

    protected function getGroupName() {
        return 'pagetools';
    }
}

class ImportReporter extends ContextSource {
    private $reason = false;
    private $mOriginalLogCallback = null;
    private $mOriginalPageOutCallback = null;
    private $mLogItemCount = 0;


    function __construct( $importer, $upload, $interwiki, $reason = false ) {
        $this->mOriginalPageOutCallback =
            $importer->setPageOutCallback( array( $this, 'reportPage' ) );
        $this->mOriginalLogCallback =
            $importer->setLogItemCallback( array( $this, 'reportLogItem' ) );
        $importer->setNoticeCallback( array( $this, 'reportNotice' ) );
        $this->mPageCount = 0;
        $this->mIsUpload = $upload;
        $this->mInterwiki = $interwiki;
        $this->reason = $reason;
    }

    function open() {
        $this->getOutput()->addHTML( "<ul>\n" );
    }

    function reportNotice( $msg, array $params ) {
        $this->getOutput()->addHTML( Html::element( 'li', array(), $this->msg( $msg, $params )->text() ) );
    }

    function reportLogItem( /* ... */ ) {
        $this->mLogItemCount++;
        if ( is_callable( $this->mOriginalLogCallback ) ) {
            call_user_func_array( $this->mOriginalLogCallback, func_get_args() );
        }
    }

    function reportPage( $title, $origTitle, $revisionCount, $successCount, $pageInfo ) {
        $args = func_get_args();
        call_user_func_array( $this->mOriginalPageOutCallback, $args );

        if ( $title === null ) {
            # Invalid or non-importable title; a notice is already displayed
            return;
        }

        $this->mPageCount++;

        if ( $successCount > 0 ) {
            $this->getOutput()->addHTML(
                "<li>" . Linker::linkKnown( $title ) . " " .
                    $this->msg( 'import-revision-count' )->numParams( $successCount )->escaped() .
                    "</li>\n"
            );

            $log = new LogPage( 'import' );
            if ( $this->mIsUpload ) {
                $detail = $this->msg( 'import-logentry-upload-detail' )->numParams(
                    $successCount )->inContentLanguage()->text();
                if ( $this->reason ) {
                    $detail .= $this->msg( 'colon-separator' )->inContentLanguage()->text() . $this->reason;
                }
                $log->addEntry( 'upload', $title, $detail, array(), $this->getUser() );
            } else {
                $interwiki = '[[:' . $this->mInterwiki . ':' .
                    $origTitle->getPrefixedText() . ']]';
                $detail = $this->msg( 'import-logentry-interwiki-detail' )->numParams(
                    $successCount )->params( $interwiki )->inContentLanguage()->text();
                if ( $this->reason ) {
                    $detail .= $this->msg( 'colon-separator' )->inContentLanguage()->text() . $this->reason;
                }
                $log->addEntry( 'interwiki', $title, $detail, array(), $this->getUser() );
            }

            $comment = $detail; // quick
            $dbw = wfGetDB( DB_MASTER );
            $latest = $title->getLatestRevID();
            $nullRevision = Revision::newNullRevision( $dbw, $title->getArticleID(), $comment, true );
            if ( !is_null( $nullRevision ) ) {
                $nullRevision->insertOn( $dbw );
                $page = WikiPage::factory( $title );
                # Update page record
                $page->updateRevisionOn( $dbw, $nullRevision );
                wfRunHooks( 'NewRevisionFromEditComplete', array( $page, $nullRevision, $latest, $this->getUser() ) );
            }
        } else {
            $this->getOutput()->addHTML( "<li>" . Linker::linkKnown( $title ) . " " .
                $this->msg( 'import-nonewrevisions' )->escaped() . "</li>\n" );
        }
    }

    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 );
    }
}