SpecialEditWatchlist.php

Go to the documentation of this file.

<?php
class SpecialEditWatchlist extends UnlistedSpecialPage {

        const EDIT_CLEAR = 1;
        const EDIT_RAW = 2;
        const EDIT_NORMAL = 3;

        protected $successMessage;

        protected $toc;

        private $badItems = array();

        public function __construct() {
                parent::__construct( 'EditWatchlist' );
        }

        public function execute( $mode ) {
                $this->setHeaders();

                $out = $this->getOutput();

                # Anons don't get a watchlist
                if( $this->getUser()->isAnon() ) {
                        $out->setPageTitle( $this->msg( 'watchnologin' ) );
                        $llink = Linker::linkKnown(
                                SpecialPage::getTitleFor( 'Userlogin' ),
                                $this->msg( 'loginreqlink' )->escaped(),
                                array(),
                                array( 'returnto' => $this->getTitle()->getPrefixedText() )
                        );
                        $out->addHTML( $this->msg( 'watchlistanontext' )->rawParams( $llink )->parse() );
                        return;
                }

                $this->checkPermissions();
                $this->checkReadOnly();

                $this->outputHeader();

                $out->addSubtitle( $this->msg( 'watchlistfor2', $this->getUser()->getName()
                        )->rawParams( SpecialEditWatchlist::buildTools( null ) ) );

                # B/C: $mode used to be waaay down the parameter list, and the first parameter
                # was $wgUser
                if( $mode instanceof User ) {
                        $args = func_get_args();
                        if( count( $args >= 4 ) ) {
                                $mode = $args[3];
                        }
                }
                $mode = self::getMode( $this->getRequest(), $mode );

                switch( $mode ) {
                        case self::EDIT_CLEAR:
                                // The "Clear" link scared people too much.
                                // Pass on to the raw editor, from which it's very easy to clear.

                        case self::EDIT_RAW:
                                $out->setPageTitle( $this->msg( 'watchlistedit-raw-title' ) );
                                $form = $this->getRawForm();
                                if( $form->show() ) {
                                        $out->addHTML( $this->successMessage );
                                        $out->addReturnTo( SpecialPage::getTitleFor( 'Watchlist' ) );
                                }
                                break;

                        case self::EDIT_NORMAL:
                        default:
                                $out->setPageTitle( $this->msg( 'watchlistedit-normal-title' ) );
                                $form = $this->getNormalForm();
                                if( $form->show() ) {
                                        $out->addHTML( $this->successMessage );
                                        $out->addReturnTo( SpecialPage::getTitleFor( 'Watchlist' ) );
                                } elseif ( $this->toc !== false ) {
                                        $out->prependHTML( $this->toc );
                                }
                                break;
                }
        }

        private function extractTitles( $list ) {
                $list = explode( "\n", trim( $list ) );
                if( !is_array( $list ) ) {
                        return array();
                }
                $titles = array();
                foreach( $list as $text ) {
                        $text = trim( $text );
                        if( strlen( $text ) > 0 ) {
                                $title = Title::newFromText( $text );
                                if( $title instanceof Title && $title->isWatchable() ) {
                                        $titles[] = $title;
                                }
                        }
                }

                GenderCache::singleton()->doTitlesArray( $titles );

                $list = array();
                foreach( $titles as $title ) {
                        $list[] = $title->getPrefixedText();
                }
                return array_unique( $list );
        }

        public function submitRaw( $data ) {
                $wanted = $this->extractTitles( $data['Titles'] );
                $current = $this->getWatchlist();

                if( count( $wanted ) > 0 ) {
                        $toWatch = array_diff( $wanted, $current );
                        $toUnwatch = array_diff( $current, $wanted );
                        $this->watchTitles( $toWatch );
                        $this->unwatchTitles( $toUnwatch );
                        $this->getUser()->invalidateCache();

                        if( count( $toWatch ) > 0 || count( $toUnwatch ) > 0 ) {
                                $this->successMessage = $this->msg( 'watchlistedit-raw-done' )->parse();
                        } else {
                                return false;
                        }

                        if( count( $toWatch ) > 0 ) {
                                $this->successMessage .= ' ' . $this->msg( 'watchlistedit-raw-added'
                                        )->numParams( count( $toWatch ) )->parse();
                                $this->showTitles( $toWatch, $this->successMessage );
                        }

                        if( count( $toUnwatch ) > 0 ) {
                                $this->successMessage .= ' ' . $this->msg( 'watchlistedit-raw-removed'
                                        )->numParams( count( $toUnwatch ) )->parse();
                                $this->showTitles( $toUnwatch, $this->successMessage );
                        }
                } else {
                        $this->clearWatchlist();
                        $this->getUser()->invalidateCache();

                        if( count( $current ) > 0 ) {
                                $this->successMessage = $this->msg( 'watchlistedit-raw-done' )->parse();
                        } else {
                                return false;
                        }

                        $this->successMessage .= ' ' . $this->msg( 'watchlistedit-raw-removed'
                                )->numParams( count( $current ) )->parse();
                        $this->showTitles( $current, $this->successMessage );
                }
                return true;
        }

        private function showTitles( $titles, &$output ) {
                $talk = $this->msg( 'talkpagelinktext' )->escaped();
                // Do a batch existence check
                $batch = new LinkBatch();
                foreach( $titles as $title ) {
                        if( !$title instanceof Title ) {
                                $title = Title::newFromText( $title );
                        }
                        if( $title instanceof Title ) {
                                $batch->addObj( $title );
                                $batch->addObj( $title->getTalkPage() );
                        }
                }
                $batch->execute();
                // Print out the list
                $output .= "<ul>\n";
                foreach( $titles as $title ) {
                        if( !$title instanceof Title ) {
                                $title = Title::newFromText( $title );
                        }
                        if( $title instanceof Title ) {
                                $output .= "<li>"
                                        . Linker::link( $title )
                                        . ' (' . Linker::link( $title->getTalkPage(), $talk )
                                        . ")</li>\n";
                        }
                }
                $output .= "</ul>\n";
        }

        private function getWatchlist() {
                $list = array();
                $dbr = wfGetDB( DB_MASTER );
                $res = $dbr->select(
                        'watchlist',
                        array(
                                'wl_namespace', 'wl_title'
                        ), array(
                                'wl_user' => $this->getUser()->getId(),
                        ),
                        __METHOD__
                );
                if( $res->numRows() > 0 ) {
                        $titles = array();
                        foreach ( $res as $row ) {
                                $title = Title::makeTitleSafe( $row->wl_namespace, $row->wl_title );
                                if ( $this->checkTitle( $title, $row->wl_namespace, $row->wl_title )
                                        && !$title->isTalkPage()
                                ) {
                                        $titles[] = $title;
                                }
                        }
                        $res->free();

                        GenderCache::singleton()->doTitlesArray( $titles );

                        foreach( $titles as $title ) {
                                $list[] = $title->getPrefixedText();
                        }
                }
                $this->cleanupWatchlist();
                return $list;
        }

        private function getWatchlistInfo() {
                $titles = array();
                $dbr = wfGetDB( DB_MASTER );

                $res = $dbr->select(
                        array( 'watchlist' ),
                        array( 'wl_namespace', 'wl_title' ),
                        array( 'wl_user' => $this->getUser()->getId() ),
                        __METHOD__,
                        array( 'ORDER BY' => array( 'wl_namespace', 'wl_title' ) )
                );

                $lb = new LinkBatch();
                foreach ( $res as $row ) {
                        $lb->add( $row->wl_namespace, $row->wl_title );
                        if ( !MWNamespace::isTalk( $row->wl_namespace ) ) {
                                $titles[$row->wl_namespace][$row->wl_title] = 1;
                        }
                }

                $lb->execute();
                return $titles;
        }

        private function checkTitle( $title, $namespace, $dbKey ) {
                if ( $title
                        && ( $title->isExternal()
                                || $title->getNamespace() < 0
                        )
                ) {
                        $title = false; // unrecoverable
                }
                if ( !$title
                        || $title->getNamespace() != $namespace
                        || $title->getDBkey() != $dbKey
                ) {
                        $this->badItems[] = array( $title, $namespace, $dbKey );
                }
                return (bool)$title;
        }

        private function cleanupWatchlist() {
                if( !count( $this->badItems ) ) {
                        return; //nothing to do
                }
                $dbw = wfGetDB( DB_MASTER );
                $user = $this->getUser();
                foreach ( $this->badItems as $row ) {
                        list( $title, $namespace, $dbKey ) = $row;
                        wfDebug( "User {$user->getName()} has broken watchlist item ns($namespace):$dbKey, "
                                . ( $title ? 'cleaning up' : 'deleting' ) . ".\n"
                        );

                        $dbw->delete( 'watchlist',
                                array(
                                        'wl_user' => $user->getId(),
                                        'wl_namespace' => $namespace,
                                        'wl_title' => $dbKey,
                                ),
                                __METHOD__
                        );

                        // Can't just do an UPDATE instead of DELETE/INSERT due to unique index
                        if ( $title ) {
                                $user->addWatch( $title );
                        }
                }
        }

        private function clearWatchlist() {
                $dbw = wfGetDB( DB_MASTER );
                $dbw->delete(
                        'watchlist',
                        array( 'wl_user' => $this->getUser()->getId() ),
                        __METHOD__
                );
        }

        private function watchTitles( $titles ) {
                $dbw = wfGetDB( DB_MASTER );
                $rows = array();
                foreach( $titles as $title ) {
                        if( !$title instanceof Title ) {
                                $title = Title::newFromText( $title );
                        }
                        if( $title instanceof Title ) {
                                $rows[] = array(
                                        'wl_user' => $this->getUser()->getId(),
                                        'wl_namespace' => MWNamespace::getSubject( $title->getNamespace() ),
                                        'wl_title' => $title->getDBkey(),
                                        'wl_notificationtimestamp' => null,
                                );
                                $rows[] = array(
                                        'wl_user' => $this->getUser()->getId(),
                                        'wl_namespace' => MWNamespace::getTalk( $title->getNamespace() ),
                                        'wl_title' => $title->getDBkey(),
                                        'wl_notificationtimestamp' => null,
                                );
                        }
                }
                $dbw->insert( 'watchlist', $rows, __METHOD__, 'IGNORE' );
        }

        private function unwatchTitles( $titles ) {
                $dbw = wfGetDB( DB_MASTER );
                foreach( $titles as $title ) {
                        if( !$title instanceof Title ) {
                                $title = Title::newFromText( $title );
                        }
                        if( $title instanceof Title ) {
                                $dbw->delete(
                                        'watchlist',
                                        array(
                                                'wl_user' => $this->getUser()->getId(),
                                                'wl_namespace' => MWNamespace::getSubject( $title->getNamespace() ),
                                                'wl_title' => $title->getDBkey(),
                                        ),
                                        __METHOD__
                                );
                                $dbw->delete(
                                        'watchlist',
                                        array(
                                                'wl_user' => $this->getUser()->getId(),
                                                'wl_namespace' => MWNamespace::getTalk( $title->getNamespace() ),
                                                'wl_title' => $title->getDBkey(),
                                        ),
                                        __METHOD__
                                );
                                $page = WikiPage::factory( $title );
                                wfRunHooks( 'UnwatchArticleComplete', array( $this->getUser(), &$page ) );
                        }
                }
        }

        public function submitNormal( $data ) {
                $removed = array();

                foreach( $data as $titles ) {
                        $this->unwatchTitles( $titles );
                        $removed = array_merge( $removed, $titles );
                }

                if( count( $removed ) > 0 ) {
                        $this->successMessage = $this->msg( 'watchlistedit-normal-done'
                                )->numParams( count( $removed ) )->parse();
                        $this->showTitles( $removed, $this->successMessage );
                        return true;
                } else {
                        return false;
                }
        }

        protected function getNormalForm() {
                global $wgContLang;

                $fields = array();
                $count = 0;

                foreach( $this->getWatchlistInfo() as $namespace => $pages ) {
                        if ( $namespace >= 0 ) {
                                $fields['TitlesNs' . $namespace] = array(
                                        'class' => 'EditWatchlistCheckboxSeriesField',
                                        'options' => array(),
                                        'section' => "ns$namespace",
                                );
                        }

                        foreach( array_keys( $pages ) as $dbkey ) {
                                $title = Title::makeTitleSafe( $namespace, $dbkey );
                                if ( $this->checkTitle( $title, $namespace, $dbkey ) ) {
                                        $text = $this->buildRemoveLine( $title );
                                        $fields['TitlesNs' . $namespace]['options'][$text] = $title->getPrefixedText();
                                        $count++;
                                }
                        }
                }
                $this->cleanupWatchlist();

                if ( count( $fields ) > 1 && $count > 30 ) {
                        $this->toc = Linker::tocIndent();
                        $tocLength = 0;
                        foreach( $fields as $data ) {

                                # strip out the 'ns' prefix from the section name:
                                $ns = substr( $data['section'], 2 );

                                $nsText = ($ns == NS_MAIN)
                                        ? $this->msg( 'blanknamespace' )->escaped()
                                        : htmlspecialchars( $wgContLang->getFormattedNsText( $ns ) );
                                $this->toc .= Linker::tocLine( "editwatchlist-{$data['section']}", $nsText,
                                        $this->getLanguage()->formatNum( ++$tocLength ), 1 ) . Linker::tocLineEnd();
                        }
                        $this->toc = Linker::tocList( $this->toc );
                } else {
                        $this->toc = false;
                }

                $form = new EditWatchlistNormalHTMLForm( $fields, $this->getContext() );
                $form->setTitle( $this->getTitle() );
                $form->setSubmitTextMsg( 'watchlistedit-normal-submit' );
                # Used message keys: 'accesskey-watchlistedit-normal-submit', 'tooltip-watchlistedit-normal-submit'
                $form->setSubmitTooltip( 'watchlistedit-normal-submit' );
                $form->setWrapperLegendMsg( 'watchlistedit-normal-legend' );
                $form->addHeaderText( $this->msg( 'watchlistedit-normal-explain' )->parse() );
                $form->setSubmitCallback( array( $this, 'submitNormal' ) );
                return $form;
        }

        private function buildRemoveLine( $title ) {
                $link = Linker::link( $title );
                if( $title->isRedirect() ) {
                        // Linker already makes class mw-redirect, so this is redundant
                        $link = '<span class="watchlistredir">' . $link . '</span>';
                }
                $tools[] = Linker::link( $title->getTalkPage(), $this->msg( 'talkpagelinktext' )->escaped() );
                if( $title->exists() ) {
                        $tools[] = Linker::linkKnown(
                                $title,
                                $this->msg( 'history_short' )->escaped(),
                                array(),
                                array( 'action' => 'history' )
                        );
                }
                if( $title->getNamespace() == NS_USER && !$title->isSubpage() ) {
                        $tools[] = Linker::linkKnown(
                                SpecialPage::getTitleFor( 'Contributions', $title->getText() ),
                                $this->msg( 'contributions' )->escaped()
                        );
                }

                wfRunHooks( 'WatchlistEditorBuildRemoveLine', array( &$tools, $title, $title->isRedirect(), $this->getSkin() ) );

                return $link . " (" . $this->getLanguage()->pipeList( $tools ) . ")";
        }

        protected function getRawForm() {
                $titles = implode( $this->getWatchlist(), "\n" );
                $fields = array(
                        'Titles' => array(
                                'type' => 'textarea',
                                'label-message' => 'watchlistedit-raw-titles',
                                'default' => $titles,
                        ),
                );
                $form = new HTMLForm( $fields, $this->getContext() );
                $form->setTitle( $this->getTitle( 'raw' ) );
                $form->setSubmitTextMsg( 'watchlistedit-raw-submit' );
                # Used message keys: 'accesskey-watchlistedit-raw-submit', 'tooltip-watchlistedit-raw-submit'
                $form->setSubmitTooltip( 'watchlistedit-raw-submit' );
                $form->setWrapperLegendMsg( 'watchlistedit-raw-legend' );
                $form->addHeaderText( $this->msg( 'watchlistedit-raw-explain' )->parse() );
                $form->setSubmitCallback( array( $this, 'submitRaw' ) );
                return $form;
        }

        public static function getMode( $request, $par ) {
                $mode = strtolower( $request->getVal( 'action', $par ) );
                switch( $mode ) {
                        case 'clear':
                        case self::EDIT_CLEAR:
                                return self::EDIT_CLEAR;

                        case 'raw':
                        case self::EDIT_RAW:
                                return self::EDIT_RAW;

                        case 'edit':
                        case self::EDIT_NORMAL:
                                return self::EDIT_NORMAL;

                        default:
                                return false;
                }
        }

        public static function buildTools( $unused ) {
                global $wgLang;

                $tools = array();
                $modes = array(
                        'view' => array( 'Watchlist', false ),
                        'edit' => array( 'EditWatchlist', false ),
                        'raw' => array( 'EditWatchlist', 'raw' ),
                );
                foreach( $modes as $mode => $arr ) {
                        // can use messages 'watchlisttools-view', 'watchlisttools-edit', 'watchlisttools-raw'
                        $tools[] = Linker::linkKnown(
                                SpecialPage::getTitleFor( $arr[0], $arr[1] ),
                                wfMessage( "watchlisttools-{$mode}" )->escaped()
                        );
                }
                return Html::rawElement( 'span',
                                        array( 'class' => 'mw-watchlist-toollinks' ),
                                        wfMessage( 'parentheses', $wgLang->pipeList( $tools ) )->text() );
        }
}

# B/C since 1.18
class WatchlistEditor extends SpecialEditWatchlist {}

class EditWatchlistNormalHTMLForm extends HTMLForm {
        public function getLegend( $namespace ) {
                $namespace = substr( $namespace, 2 );
                return $namespace == NS_MAIN
                        ? $this->msg( 'blanknamespace' )->escaped()
                        : htmlspecialchars( $this->getContext()->getLanguage()->getFormattedNsText( $namespace ) );
        }
        public function getBody() {
                return $this->displaySection( $this->mFieldTree, '', 'editwatchlist-' );
        }
}

class EditWatchlistCheckboxSeriesField extends HTMLMultiSelectField {
        function validate( $value, $alldata ) {
                // Need to call into grandparent to be a good citizen. :)
                return HTMLFormField::validate( $value, $alldata );
        }
}