WatchAction.php

Go to the documentation of this file.

<?php
use MediaWiki\Context\IContextSource;
use MediaWiki\HTMLForm\HTMLForm;
use MediaWiki\MainConfigNames;
use MediaWiki\Status\Status;
use MediaWiki\User\User;
use MediaWiki\Watchlist\WatchedItem;
use MediaWiki\Watchlist\WatchedItemStore;
use MediaWiki\Watchlist\WatchlistManager;
use MediaWiki\Xml\XmlSelect;
use Wikimedia\ParamValidator\TypeDef\ExpiryDef;
 
class WatchAction extends FormAction {
 
    protected $watchlistExpiry;
 
    protected $expiryFormFieldName = 'expiry';
 
    protected $watchedItem = false;
 
    private WatchlistManager $watchlistManager;
 
    public function __construct(
        Article $article,
        IContextSource $context,
        WatchlistManager $watchlistManager,
        WatchedItemStore $watchedItemStore
    ) {
        parent::__construct( $article, $context );
        $this->watchlistExpiry = $this->getContext()->getConfig()->get( MainConfigNames::WatchlistExpiry );
        if ( $this->watchlistExpiry ) {
            // The watchedItem is only used in this action's form if $wgWatchlistExpiry is enabled.
            $this->watchedItem = $watchedItemStore->getWatchedItem(
                $this->getUser(),
                $this->getTitle()
            );
        }
        $this->watchlistManager = $watchlistManager;
    }
 
    public function getName() {
        return 'watch';
    }
 
    public function requiresUnblock() {
        return false;
    }
 
    protected function getDescription() {
        return '';
    }
 
    public function onSubmit( $data ) {
        // Even though we're never unwatching here, use WatchlistManager::setWatch()
        // because it also checks for changed expiry.
        $result = $this->watchlistManager->setWatch(
            true,
            $this->getAuthority(),
            $this->getTitle(),
            $this->getRequest()->getVal( 'wp' . $this->expiryFormFieldName )
        );
 
        return Status::wrap( $result );
    }
 
    protected function checkCanExecute( User $user ) {
        if ( !$user->isRegistered()
            || ( $user->isTemp() && !$user->isAllowed( 'editmywatchlist' ) )
        ) {
            throw new UserNotLoggedIn( 'watchlistanontext', 'watchnologin' );
        }
 
        parent::checkCanExecute( $user );
    }
 
    public function getRestriction() {
        return 'editmywatchlist';
    }
 
    protected function usesOOUI() {
        return true;
    }
 
    protected function getFormFields() {
        // If watchlist expiry is not enabled, return a simple confirmation message.
        if ( !$this->watchlistExpiry ) {
            return [
                'intro' => [
                    'type' => 'info',
                    'raw' => true,
                    'default' => $this->msg( 'confirm-watch-top' )->parse(),
                ],
            ];
        }
 
        // Otherwise, use a select-list of expiries.
        $expiryOptions = static::getExpiryOptions( $this->getContext(), $this->watchedItem );
        return [
            $this->expiryFormFieldName => [
                'type' => 'select',
                'label-message' => 'confirm-watch-label',
                'options' => $expiryOptions['options'],
                'default' => $expiryOptions['default'],
            ]
        ];
    }
 
    public static function getExpiryOptions( MessageLocalizer $msgLocalizer, $watchedItem ) {
        $expiryOptions = self::getExpiryOptionsFromMessage( $msgLocalizer );
        $default = in_array( 'infinite', $expiryOptions )
            ? 'infinite'
            : current( $expiryOptions );
        if ( $watchedItem instanceof WatchedItem && $watchedItem->getExpiry() ) {
            // If it's already being temporarily watched,
            // add the existing expiry as the default option in the dropdown.
            $default = $watchedItem->getExpiry( TS_ISO_8601 );
            $daysLeft = $watchedItem->getExpiryInDaysText( $msgLocalizer, true );
            $expiryOptions = array_merge( [ $daysLeft => $default ], $expiryOptions );
        }
        return [
            'options' => $expiryOptions,
            'default' => $default,
        ];
    }
 
    private static function getExpiryOptionsFromMessage(
        MessageLocalizer $msgLocalizer, ?string $lang = null
    ): array {
        $expiryOptionsMsg = $msgLocalizer->msg( 'watchlist-expiry-options' );
        $optionsText = !$lang ? $expiryOptionsMsg->text() : $expiryOptionsMsg->inLanguage( $lang )->text();
        $options = XmlSelect::parseOptionsMessage(
            $optionsText
        );
 
        $expiryOptions = [];
        foreach ( $options as $label => $value ) {
            if ( strtotime( $value ) || wfIsInfinity( $value ) ) {
                $expiryOptions[$label] = $value;
            }
        }
 
        // If message options is invalid try to recover by returning
        // english options (T267611)
        if ( !$expiryOptions && $expiryOptionsMsg->getLanguage()->getCode() !== 'en' ) {
            return self::getExpiryOptionsFromMessage( $msgLocalizer, 'en' );
        }
 
        return $expiryOptions;
    }
 
    protected function alterForm( HTMLForm $form ) {
        $msg = $this->watchlistExpiry && $this->watchedItem ? 'updatewatchlist' : 'addwatch';
        $form->setWrapperLegendMsg( $msg );
        $submitMsg = $this->watchlistExpiry ? 'confirm-watch-button-expiry' : 'confirm-watch-button';
        $form->setSubmitTextMsg( $submitMsg );
        $form->setTokenSalt( 'watch' );
    }
 
    public function onSuccess() {
        $msgKey = $this->getTitle()->isTalkPage() ? 'addedwatchtext-talk' : 'addedwatchtext';
        $expiryLabel = null;
        $submittedExpiry = $this->getContext()->getRequest()->getText( 'wp' . $this->expiryFormFieldName );
        if ( $submittedExpiry ) {
            // We can't use $this->watcheditem to get the expiry because it's not been saved at this
            // point in the request and so its values are those from before saving.
            $expiry = ExpiryDef::normalizeExpiry( $submittedExpiry, TS_ISO_8601 );
 
            // If the expiry label isn't one of the predefined ones in the dropdown, calculate 'x days'.
            $expiryDays = WatchedItem::calculateExpiryInDays( $expiry );
            $defaultLabels = static::getExpiryOptions( $this->getContext(), false )['options'];
            $localizedExpiry = array_search( $submittedExpiry, $defaultLabels );
            $expiryLabel = $expiryDays && $localizedExpiry === false
                ? $this->getContext()->msg( 'days', $expiryDays )->text()
                : $localizedExpiry;
 
            // Determine which message to use, depending on whether this is a talk page or not
            // and whether an expiry was selected.
            $isTalk = $this->getTitle()->isTalkPage();
            if ( wfIsInfinity( $expiry ) ) {
                $msgKey = $isTalk ? 'addedwatchindefinitelytext-talk' : 'addedwatchindefinitelytext';
            } elseif ( $expiryDays > 0 ) {
                $msgKey = $isTalk ? 'addedwatchexpirytext-talk' : 'addedwatchexpirytext';
            } elseif ( $expiryDays < 1 ) {
                $msgKey = $isTalk ? 'addedwatchexpiryhours-talk' : 'addedwatchexpiryhours';
            }
        }
        $this->getOutput()->addWikiMsg( $msgKey, $this->getTitle()->getPrefixedText(), $expiryLabel );
    }
 
    public function doesWrites() {
        return true;
    }
}