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', 'editmywatchlist' );
    }

    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;
            $action = $title ? 'cleaning up' : 'deleting';
            wfDebug( "User {$user->getName()} has broken watchlist item ns($namespace):$dbKey, $action.\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;
        }

        $context = new DerivativeContext( $this->getContext() );
        $context->setTitle( $this->getTitle() ); // Remove subpage
        $form = new EditWatchlistNormalHTMLForm( $fields, $context );
        $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,
            ),
        );
        $context = new DerivativeContext( $this->getContext() );
        $context->setTitle( $this->getTitle( 'raw' ) ); // Reset subpage
        $form = new HTMLForm( $fields, $context );
        $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 );
    }
}