Bug 32673: Keep the username in the input field if not existing
[lhc/web/wiklou.git] / includes / specials / SpecialEditWatchlist.php
index 47576e3..cde2ca3 100644 (file)
@@ -18,6 +18,8 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
 
        protected $successMessage;
 
+       protected $toc;
+
        public function __construct(){
                parent::__construct( 'EditWatchlist' );
        }
@@ -28,32 +30,29 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
         * @param $mode int
         */
        public function execute( $mode ) {
-               global $wgUser, $wgOut, $wgRequest;
-               if( wfReadOnly() ) {
-                       $wgOut->readOnlyPage();
-                       return;
-               }
+               $this->setHeaders();
+
+               $out = $this->getOutput();
 
                # Anons don't get a watchlist
-               if( $wgUser->isAnon() ) {
-                       $wgOut->setPageTitle( wfMsg( 'watchnologin' ) );
-                       $llink = $wgUser->getSkin()->linkKnown(
+               if( $this->getUser()->isAnon() ) {
+                       $out->setPageTitle( $this->msg( 'watchnologin' ) );
+                       $llink = Linker::linkKnown(
                                SpecialPage::getTitleFor( 'Userlogin' ),
                                wfMsgHtml( 'loginreqlink' ),
                                array(),
                                array( 'returnto' => $this->getTitle()->getPrefixedText() )
                        );
-                       $wgOut->addWikiMsgArray( 'watchlistanontext', array( $llink ), array( 'replaceafter' ) );
+                       $out->addHTML( wfMessage( 'watchlistanontext' )->rawParams( $llink )->parse() );
                        return;
                }
 
-               $sub  = wfMsgExt(
-                       'watchlistfor2',
-                       array( 'parseinline', 'replaceafter' ),
-                       $wgUser->getName(),
-                       SpecialEditWatchlist::buildTools( $wgUser->getSkin() )
-               );
-               $wgOut->setSubtitle( $sub );
+               $this->checkPermissions();
+
+               $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
@@ -63,7 +62,7 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                                $mode = $args[3];
                        }
                }
-               $mode = self::getMode( $wgRequest, $mode );
+               $mode = self::getMode( $this->getRequest(), $mode );
 
                switch( $mode ) {
                        case self::EDIT_CLEAR:
@@ -71,21 +70,23 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                                // Pass on to the raw editor, from which it's very easy to clear.
 
                        case self::EDIT_RAW:
-                               $wgOut->setPageTitle( wfMsg( 'watchlistedit-raw-title' ) );
-                               $form = $this->getRawForm( $wgUser );
+                               $out->setPageTitle( $this->msg( 'watchlistedit-raw-title' ) );
+                               $form = $this->getRawForm();
                                if( $form->show() ){
-                                       $wgOut->addHTML( $this->successMessage );
-                                       $wgOut->returnToMain();
+                                       $out->addHTML( $this->successMessage );
+                                       $out->returnToMain();
                                }
                                break;
 
                        case self::EDIT_NORMAL:
                        default:
-                               $wgOut->setPageTitle( wfMsg( 'watchlistedit-normal-title' ) );
-                               $form = $this->getNormalForm( $wgUser );
+                               $out->setPageTitle( $this->msg( 'watchlistedit-normal-title' ) );
+                               $form = $this->getNormalForm();
                                if( $form->show() ){
-                                       $wgOut->addHTML( $this->successMessage );
-                                       $wgOut->returnToMain();
+                                       $out->addHTML( $this->successMessage );
+                                       $out->returnToMain();
+                               } elseif ( $this->toc !== false ) {
+                                       $out->prependHTML( $this->toc );
                                }
                                break;
                }
@@ -117,16 +118,15 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
        }
 
        public function submitRaw( $data ){
-               global $wgUser, $wgLang;
                $wanted = $this->extractTitles( $data['Titles'] );
-               $current = $this->getWatchlist( $wgUser );
+               $current = $this->getWatchlist();
 
                if( count( $wanted ) > 0 ) {
                        $toWatch = array_diff( $wanted, $current );
                        $toUnwatch = array_diff( $current, $wanted );
-                       $this->watchTitles( $toWatch, $wgUser );
-                       $this->unwatchTitles( $toUnwatch, $wgUser );
-                       $wgUser->invalidateCache();
+                       $this->watchTitles( $toWatch );
+                       $this->unwatchTitles( $toUnwatch );
+                       $this->getUser()->invalidateCache();
 
                        if( count( $toWatch ) > 0 || count( $toUnwatch ) > 0 ){
                                $this->successMessage = wfMessage( 'watchlistedit-raw-done' )->parse();
@@ -135,28 +135,35 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                        }
 
                        if( count( $toWatch ) > 0 ) {
-                               $this->successMessage .= wfMessage(
+                               $this->successMessage .= ' ' . wfMessage(
                                        'watchlistedit-raw-added',
-                                       $wgLang->formatNum( count( $toWatch ) )
+                                       $this->getLanguage()->formatNum( count( $toWatch ) )
                                );
-                               $this->showTitles( $toWatch, $this->successMessage, $wgUser->getSkin() );
+                               $this->showTitles( $toWatch, $this->successMessage );
                        }
 
                        if( count( $toUnwatch ) > 0 ) {
-                               $this->successMessage .= wfMessage(
+                               $this->successMessage .= ' ' . wfMessage(
                                        'watchlistedit-raw-removed',
-                                       $wgLang->formatNum( count( $toUnwatch ) )
+                                       $this->getLanguage()->formatNum( count( $toUnwatch ) )
                                );
-                               $this->showTitles( $toUnwatch, $this->successMessage, $wgUser->getSkin() );
+                               $this->showTitles( $toUnwatch, $this->successMessage );
                        }
                } else {
-                       $this->clearWatchlist( $wgUser );
-                       $wgUser->invalidateCache();
-                       $this->successMessage .= wfMessage(
+                       $this->clearWatchlist();
+                       $this->getUser()->invalidateCache();
+
+                       if( count( $current ) > 0 ){
+                               $this->successMessage = wfMessage( 'watchlistedit-raw-done' )->parse();
+                       } else {
+                               return false;
+                       }
+
+                       $this->successMessage .= ' ' . wfMessage(
                                'watchlistedit-raw-removed',
-                               $wgLang->formatNum( count( $current ) )
+                               $this->getLanguage()->formatNum( count( $current ) )
                        );
-                       $this->showTitles( $current, $this->successMessage, $wgUser->getSkin() );
+                       $this->showTitles( $current, $this->successMessage );
                }
                return true;
        }
@@ -169,9 +176,8 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
         *
         * @param $titles array of strings, or Title objects
         * @param $output String
-        * @param $skin Skin
         */
-       private function showTitles( $titles, &$output, $skin ) {
+       private function showTitles( $titles, &$output ) {
                $talk = wfMsgHtml( 'talkpagelinktext' );
                // Do a batch existence check
                $batch = new LinkBatch();
@@ -193,42 +199,28 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                        }
                        if( $title instanceof Title ) {
                                $output .= "<li>"
-                                       . $skin->link( $title )
-                                       . ' (' . $skin->link( $title->getTalkPage(), $talk )
+                                       . Linker::link( $title )
+                                       . ' (' . Linker::link( $title->getTalkPage(), $talk )
                                        . ")</li>\n";
                        }
                }
                $output .= "</ul>\n";
        }
 
-       /**
-        * Count the number of titles on a user's watchlist, excluding talk pages
-        *
-        * @param $user User
-        * @return int
-        */
-       private function countWatchlist( $user ) {
-               $dbr = wfGetDB( DB_MASTER );
-               $res = $dbr->select( 'watchlist', 'COUNT(*) AS count', array( 'wl_user' => $user->getId() ), __METHOD__ );
-               $row = $dbr->fetchObject( $res );
-               return ceil( $row->count / 2 ); // Paranoia
-       }
-
        /**
         * Prepare a list of titles on a user's watchlist (excluding talk pages)
         * and return an array of (prefixed) strings
         *
-        * @param $user User
         * @return array
         */
-       private function getWatchlist( $user ) {
+       private function getWatchlist() {
                $list = array();
                $dbr = wfGetDB( DB_MASTER );
                $res = $dbr->select(
                        'watchlist',
                        '*',
                        array(
-                               'wl_user' => $user->getId(),
+                               'wl_user' => $this->getUser()->getId(),
                        ),
                        __METHOD__
                );
@@ -245,84 +237,42 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
 
        /**
         * Get a list of titles on a user's watchlist, excluding talk pages,
-        * and return as a two-dimensional array with namespace, title and
-        * redirect status
+        * and return as a two-dimensional array with namespace and title.
         *
-        * @param $user User
         * @return array
         */
-       private function getWatchlistInfo( $user ) {
+       private function getWatchlistInfo() {
                $titles = array();
                $dbr = wfGetDB( DB_MASTER );
 
                $res = $dbr->select(
-                       array( 'watchlist', 'page' ),
-                       array(
-                               'wl_namespace',
-                               'wl_title',
-                               'page_id',
-                               'page_len',
-                               'page_is_redirect',
-                               'page_latest'
-                       ),
-                       array( 'wl_user' => $user->getId() ),
+                       array( 'watchlist' ),
+                       array( 'wl_namespace',  'wl_title' ),
+                       array( 'wl_user' => $this->getUser()->getId() ),
                        __METHOD__,
-                       array( 'ORDER BY' => 'wl_namespace, wl_title' ),
-                       array( 'page' => array(
-                               'LEFT JOIN',
-                               'wl_namespace = page_namespace AND wl_title = page_title'
-                       ) )
+                       array( 'ORDER BY' => 'wl_namespace, wl_title' )
                );
 
-               if( $res && $dbr->numRows( $res ) > 0 ) {
-                       $cache = LinkCache::singleton();
-                       foreach ( $res as $row ) {
-                               $title = Title::makeTitleSafe( $row->wl_namespace, $row->wl_title );
-                               if( $title instanceof Title ) {
-                                       // Update the link cache while we're at it
-                                       if( $row->page_id ) {
-                                               $cache->addGoodLinkObj( $row->page_id, $title, $row->page_len, $row->page_is_redirect, $row->page_latest );
-                                       } else {
-                                               $cache->addBadLinkObj( $title );
-                                       }
-                                       // Ignore non-talk
-                                       if( !$title->isTalkPage() ) {
-                                               $titles[$row->wl_namespace][$row->wl_title] = $row->page_is_redirect;
-                                       }
-                               }
+               $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;
                        }
                }
-               return $titles;
-       }
 
-       /**
-        * Show a message indicating the number of items on the user's watchlist,
-        * and return this count for additional checking
-        *
-        * @param $output OutputPage
-        * @param $user User
-        * @return int
-        */
-       private function showItemCount( $output, $user ) {
-               if( ( $count = $this->countWatchlist( $user ) ) > 0 ) {
-                       $output->addHTML( wfMsgExt( 'watchlistedit-numitems', 'parse',
-                               $GLOBALS['wgLang']->formatNum( $count ) ) );
-               } else {
-                       $output->addHTML( wfMsgExt( 'watchlistedit-noitems', 'parse' ) );
-               }
-               return $count;
+               $lb->execute();
+               return $titles;
        }
 
        /**
         * Remove all titles from a user's watchlist
-        *
-        * @param $user User
         */
-       private function clearWatchlist( $user ) {
+       private function clearWatchlist() {
                $dbw = wfGetDB( DB_MASTER );
                $dbw->delete(
                        'watchlist',
-                       array( 'wl_user' => $user->getId() ),
+                       array( 'wl_user' => $this->getUser()->getId() ),
                        __METHOD__
                );
        }
@@ -334,9 +284,8 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
         * is preferred, since Titles are very memory-heavy
         *
         * @param $titles Array of strings, or Title objects
-        * @param $user User
         */
-       private function watchTitles( $titles, $user ) {
+       private function watchTitles( $titles ) {
                $dbw = wfGetDB( DB_MASTER );
                $rows = array();
                foreach( $titles as $title ) {
@@ -345,13 +294,13 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                        }
                        if( $title instanceof Title ) {
                                $rows[] = array(
-                                       'wl_user' => $user->getId(),
+                                       'wl_user' => $this->getUser()->getId(),
                                        'wl_namespace' => ( $title->getNamespace() & ~1 ),
                                        'wl_title' => $title->getDBkey(),
                                        'wl_notificationtimestamp' => null,
                                );
                                $rows[] = array(
-                                       'wl_user' => $user->getId(),
+                                       'wl_user' => $this->getUser()->getId(),
                                        'wl_namespace' => ( $title->getNamespace() | 1 ),
                                        'wl_title' => $title->getDBkey(),
                                        'wl_notificationtimestamp' => null,
@@ -368,9 +317,8 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
         * is preferred, since Titles are very memory-heavy
         *
         * @param $titles Array of strings, or Title objects
-        * @param $user User
         */
-       private function unwatchTitles( $titles, $user ) {
+       private function unwatchTitles( $titles ) {
                $dbw = wfGetDB( DB_MASTER );
                foreach( $titles as $title ) {
                        if( !$title instanceof Title ) {
@@ -380,7 +328,7 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                                $dbw->delete(
                                        'watchlist',
                                        array(
-                                               'wl_user' => $user->getId(),
+                                               'wl_user' => $this->getUser()->getId(),
                                                'wl_namespace' => ( $title->getNamespace() & ~1 ),
                                                'wl_title' => $title->getDBkey(),
                                        ),
@@ -389,34 +337,32 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                                $dbw->delete(
                                        'watchlist',
                                        array(
-                                               'wl_user' => $user->getId(),
+                                               'wl_user' => $this->getUser()->getId(),
                                                'wl_namespace' => ( $title->getNamespace() | 1 ),
                                                'wl_title' => $title->getDBkey(),
                                        ),
                                        __METHOD__
                                );
-                               $article = new Article($title);
-                               wfRunHooks('UnwatchArticleComplete',array(&$user,&$article));
+                               $article = new Article( $title, 0 );
+                               wfRunHooks( 'UnwatchArticleComplete', array( $this->getUser(), &$article ) );
                        }
                }
        }
 
        public function submitNormal( $data ) {
-               global $wgUser;
                $removed = array();
 
                foreach( $data as $titles ) {
-                       $this->unwatchTitles( $titles, $wgUser );
+                       $this->unwatchTitles( $titles );
                        $removed += $titles;
                }
 
                if( count( $removed ) > 0 ) {
-                       global $wgLang;
                        $this->successMessage = wfMessage(
                                'watchlistedit-normal-done',
-                               $wgLang->formatNum( count( $removed ) )
+                               $this->getLanguage()->formatNum( count( $removed ) )
                        );
-                       $this->showTitles( $removed, $this->successMessage, $wgUser->getSkin() );
+                       $this->showTitles( $removed, $this->successMessage );
                        return true;
                } else {
                        return false;
@@ -426,34 +372,59 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
        /**
         * Get the standard watchlist editing form
         *
-        * @param $user User
         * @return HTMLForm
         */
-       protected function getNormalForm( $user ){
+       protected function getNormalForm(){
                global $wgContLang;
-               $skin = $user->getSkin();
-               $fields = array();
 
-               foreach( $this->getWatchlistInfo( $user ) as $namespace => $pages ){
+               $fields = array();
+               $count = 0;
 
-                       $namespace == NS_MAIN
-                               ? wfMsgHtml( 'blanknamespace' )
-                               : htmlspecialchars( $wgContLang->getFormattedNsText( $namespace ) );
+               $haveInvalidNamespaces = false;
+               foreach( $this->getWatchlistInfo() as $namespace => $pages ){
+                       if ( $namespace < 0 ) {
+                               $haveInvalidNamespaces = true;
+                               continue;
+                       }
 
                        $fields['TitlesNs'.$namespace] = array(
-                               'type' => 'multiselect',
+                               'class' => 'EditWatchlistCheckboxSeriesField',
                                'options' => array(),
                                'section' => "ns$namespace",
                        );
 
-                       foreach( $pages as $dbkey => $redirect ){
+                       foreach( array_keys( $pages ) as $dbkey ){
                                $title = Title::makeTitleSafe( $namespace, $dbkey );
-                               $text = $this->buildRemoveLine( $title, $redirect, $skin );
+                               $text = $this->buildRemoveLine( $title );
                                $fields['TitlesNs'.$namespace]['options'][$text] = $title->getEscapedText();
+                               $count++;
                        }
                }
+               if ( $haveInvalidNamespaces ) {
+                       wfDebug( "User {$this->getContext()->getUser()->getId()} has invalid watchlist entries, cleaning up...\n" );
+                       $this->getContext()->getUser()->cleanupWatchlist();
+               }
+
+               if ( count( $fields ) > 1 && $count > 30 ) {
+                       $this->toc = Linker::tocIndent();
+                       $tocLength = 0;
+                       foreach( $fields as $key => $data ) {
+
+                               # strip out the 'ns' prefix from the section name:
+                               $ns = substr( $data['section'], 2 );
 
-               $form = new EditWatchlistNormalHTMLForm( $fields );
+                               $nsText = ($ns == NS_MAIN)
+                                       ? wfMsgHtml( 'blanknamespace' )
+                                       : 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->setSubmitText( wfMessage( 'watchlistedit-normal-submit' )->text() );
                $form->setWrapperLegend( wfMessage( 'watchlistedit-normal-legend' )->text() );
@@ -466,50 +437,42 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
         * Build the label for a checkbox, with a link to the title, and various additional bits
         *
         * @param $title Title
-        * @param $redirect bool
-        * @param $skin Skin
         * @return string
         */
-       private function buildRemoveLine( $title, $redirect, $skin ) {
-               global $wgLang;
-
-               $link = $skin->link( $title );
-               if( $redirect ) {
+       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[] = $skin->link( $title->getTalkPage(), wfMsgHtml( 'talkpagelinktext' ) );
+               $tools[] = Linker::link( $title->getTalkPage(), wfMsgHtml( 'talkpagelinktext' ) );
                if( $title->exists() ) {
-                       $tools[] = $skin->link(
+                       $tools[] = Linker::linkKnown(
                                $title,
                                wfMsgHtml( 'history_short' ),
                                array(),
-                               array( 'action' => 'history' ),
-                               array( 'known', 'noclasses' )
+                               array( 'action' => 'history' )
                        );
                }
                if( $title->getNamespace() == NS_USER && !$title->isSubpage() ) {
-                       $tools[] = $skin->link(
+                       $tools[] = Linker::linkKnown(
                                SpecialPage::getTitleFor( 'Contributions', $title->getText() ),
-                               wfMsgHtml( 'contributions' ),
-                               array(),
-                               array(),
-                               array( 'known', 'noclasses' )
+                               wfMsgHtml( 'contributions' )
                        );
                }
 
-               wfRunHooks( 'WatchlistEditorBuildRemoveLine', array( &$tools, $title, $redirect, $skin ) );
+               wfRunHooks( 'WatchlistEditorBuildRemoveLine', array( &$tools, $title, $title->isRedirect(), $this->getSkin() ) );
 
-               return $link . " (" . $wgLang->pipeList( $tools ) . ")";
+               return $link . " (" . $this->getLanguage()->pipeList( $tools ) . ")";
        }
 
        /**
         * Get a form for editing the watchlist in "raw" mode
         *
-        * @param $user User
         * @return HTMLForm
         */
-       protected function getRawForm( $user ){
-               $titles = implode( array_map( 'htmlspecialchars', $this->getWatchlist( $user ) ), "\n" );
+       protected function getRawForm(){
+               $titles = implode( $this->getWatchlist(), "\n" );
                $fields = array(
                        'Titles' => array(
                                'type' => 'textarea',
@@ -517,7 +480,7 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                                'default' => $titles,
                        ),
                );
-               $form = new HTMLForm( $fields );
+               $form = new HTMLForm( $fields, $this->getContext() );
                $form->setTitle( $this->getTitle( 'raw' ) );
                $form->setSubmitText( wfMessage( 'watchlistedit-raw-submit' )->text() );
                $form->setWrapperLegend( wfMessage( 'watchlistedit-raw-legend' )->text() );
@@ -558,10 +521,10 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
         * Build a set of links for convenient navigation
         * between watchlist viewing and editing modes
         *
-        * @param $skin Skin to use
+        * @param $unused Unused
         * @return string
         */
-       public static function buildTools( $skin ) {
+       public static function buildTools( $unused ) {
                global $wgLang;
 
                $tools = array();
@@ -572,7 +535,7 @@ class SpecialEditWatchlist extends UnlistedSpecialPage {
                );
                foreach( $modes as $mode => $arr ) {
                        // can use messages 'watchlisttools-view', 'watchlisttools-edit', 'watchlisttools-raw'
-                       $tools[] = $skin->linkKnown(
+                       $tools[] = Linker::linkKnown(
                                SpecialPage::getTitleFor( $arr[0], $arr[1] ),
                                wfMsgHtml( "watchlisttools-{$mode}" )
                        );
@@ -591,10 +554,30 @@ class WatchlistEditor extends SpecialEditWatchlist {}
  */
 class EditWatchlistNormalHTMLForm extends HTMLForm {
        public function getLegend( $namespace ){
-               global $wgLang;
                $namespace = substr( $namespace, 2 );
                return $namespace == NS_MAIN
                        ? wfMsgHtml( 'blanknamespace' )
-                       : htmlspecialchars( $wgLang->getFormattedNsText( $namespace ) );
+                       : htmlspecialchars( $this->getContext()->getLanguage()->getFormattedNsText( $namespace ) );
+       }
+       public function getBody() {
+               return $this->displaySection( $this->mFieldTree, '', 'editwatchlist-' );
        }
-}
\ No newline at end of file
+}
+
+class EditWatchlistCheckboxSeriesField extends HTMLMultiSelectField {
+       /**
+        * HTMLMultiSelectField throws validation errors if we get input data
+        * that doesn't match the data set in the form setup. This causes
+        * problems if something gets removed from the watchlist while the
+        * form is open (bug 32126), but we know that invalid items will
+        * be harmless so we can override it here.
+        *
+        * @param $value String the value the field was submitted with
+        * @param $alldata Array the data collected from the form
+        * @return Mixed Bool true on success, or String error to display.
+        */
+       function validate( $value, $alldata ) {
+               // Need to call into grandparent to be a good citizen. :)
+               return HTMLFormField::validate( $value, $alldata );
+       }
+}