RollbackAction.php

Go to the documentation of this file.

<?php
class RollbackAction extends FormAction {

    public function getName() {
        return 'rollback';
    }

    public function getRestriction() {
        return 'rollback';
    }

    protected function usesOOUI() {
        return true;
    }

    protected function getDescription() {
        return '';
    }

    public function doesWrites() {
        return true;
    }

    public function onSuccess() {
        return false;
    }

    public function onSubmit( $data ) {
        return false;
    }

    protected function alterForm( HTMLForm $form ) {
        $form->setWrapperLegendMsg( 'confirm-rollback-top' );
        $form->setSubmitTextMsg( 'confirm-rollback-button' );
        $form->setTokenSalt( 'rollback' );

        $from = $this->getRequest()->getVal( 'from' );
        if ( $from === null ) {
            throw new BadRequestError( 'rollbackfailed', 'rollback-missingparam' );
        }
        foreach ( [ 'from', 'bot', 'hidediff', 'summary', 'token' ] as $param ) {
            $val = $this->getRequest()->getVal( $param );
            if ( $val !== null ) {
                $form->addHiddenField( $param, $val );
            }
        }
    }

    public function show() {
        if ( $this->getUser()->getOption( 'showrollbackconfirmation' ) == false ||
             $this->getRequest()->wasPosted() ) {
            $this->handleRollbackRequest();
        } else {
            $this->showRollbackConfirmationForm();
        }
    }

    public function handleRollbackRequest() {
        $this->enableTransactionalTimelimit();

        $request = $this->getRequest();
        $user = $this->getUser();
        $from = $request->getVal( 'from' );
        $rev = $this->page->getRevision();
        if ( $from === null ) {
            throw new ErrorPageError( 'rollbackfailed', 'rollback-missingparam' );
        }
        if ( !$rev ) {
            throw new ErrorPageError( 'rollbackfailed', 'rollback-missingrevision' );
        }
        if ( $from !== $rev->getUserText() ) {
            throw new ErrorPageError( 'rollbackfailed', 'alreadyrolled', [
                $this->getTitle()->getPrefixedText(),
                $from,
                $rev->getUserText()
            ] );
        }

        $data = null;
        $errors = $this->page->doRollback(
            $from,
            $request->getText( 'summary' ),
            $request->getVal( 'token' ),
            $request->getBool( 'bot' ),
            $data,
            $this->getUser()
        );

        if ( in_array( [ 'actionthrottledtext' ], $errors ) ) {
            throw new ThrottledError;
        }

        if ( $this->hasRollbackRelatedErrors( $errors ) ) {
            $this->getOutput()->setPageTitle( $this->msg( 'rollbackfailed' ) );
            $errArray = $errors[0];
            $errMsg = array_shift( $errArray );
            $this->getOutput()->addWikiMsgArray( $errMsg, $errArray );

            if ( isset( $data['current'] ) ) {
                $current = $data['current'];

                if ( $current->getComment() != '' ) {
                    $this->getOutput()->addWikiMsg(
                        'editcomment',
                        Message::rawParam(
                            Linker::formatComment( $current->getComment() )
                        )
                    );
                }
            }

            return;
        }

        # NOTE: Permission errors already handled by Action::checkExecute.
        if ( $errors == [ [ 'readonlytext' ] ] ) {
            throw new ReadOnlyError;
        }

        # XXX: Would be nice if ErrorPageError could take multiple errors, and/or a status object.
        #      Right now, we only show the first error
        foreach ( $errors as $error ) {
            throw new ErrorPageError( 'rollbackfailed', $error[0], array_slice( $error, 1 ) );
        }

        $current = $data['current'];
        $target = $data['target'];
        $newId = $data['newid'];
        $this->getOutput()->setPageTitle( $this->msg( 'actioncomplete' ) );
        $this->getOutput()->setRobotPolicy( 'noindex,nofollow' );

        $old = Linker::revUserTools( $current );
        $new = Linker::revUserTools( $target );
        $this->getOutput()->addHTML(
            $this->msg( 'rollback-success' )
                ->rawParams( $old, $new )
                ->params( $current->getUserText( Revision::FOR_THIS_USER, $user ) )
                ->params( $target->getUserText( Revision::FOR_THIS_USER, $user ) )
                ->parseAsBlock()
        );

        if ( $user->getBoolOption( 'watchrollback' ) ) {
            $user->addWatch( $this->page->getTitle(), User::IGNORE_USER_RIGHTS );
        }

        $this->getOutput()->returnToMain( false, $this->getTitle() );

        if ( !$request->getBool( 'hidediff', false ) &&
            !$this->getUser()->getBoolOption( 'norollbackdiff' )
        ) {
            $contentHandler = $current->getContentHandler();
            $de = $contentHandler->createDifferenceEngine(
                $this->getContext(),
                $current->getId(),
                $newId,
                false,
                true
            );
            $de->showDiff( '', '' );
        }
    }

    private function enableTransactionalTimelimit() {
        // If Rollbacks are made POST-only, use $this->useTransactionalTimeLimit()
        wfTransactionalTimeLimit();
        if ( !$this->getRequest()->wasPosted() ) {
            $fname = __METHOD__;
            $trxLimits = $this->context->getConfig()->get( 'TrxProfilerLimits' );
            $trxProfiler = Profiler::instance()->getTransactionProfiler();
            $trxProfiler->redefineExpectations( $trxLimits['POST'], $fname );
            DeferredUpdates::addCallableUpdate( function () use ( $trxProfiler, $trxLimits, $fname
            ) {
                $trxProfiler->redefineExpectations( $trxLimits['PostSend-POST'], $fname );
            } );
        }
    }

    private function showRollbackConfirmationForm() {
        $form = $this->getForm();
        if ( $form->show() ) {
            $this->onSuccess();
        }
    }

    protected function getFormFields() {
        return [
            'intro' => [
                'type' => 'info',
                'vertical-label' => true,
                'raw' => true,
                'default' => $this->msg( 'confirm-rollback-bottom' )->parse()
            ]
        ];
    }

    private function hasRollbackRelatedErrors( array $errors ) {
        return isset( $errors[0][0] ) &&
            ( $errors[0][0] == 'alreadyrolled' ||
                $errors[0][0] == 'cantrollback'
            );
    }
}