ApiWatchlistTrait.php

Go to the documentation of this file.

<?php
 
use MediaWiki\MediaWikiServices;
use MediaWiki\User\UserIdentity;
use MediaWiki\User\UserOptionsLookup;
use MediaWiki\Watchlist\WatchlistManager;
use Wikimedia\ParamValidator\ParamValidator;
use Wikimedia\ParamValidator\TypeDef\ExpiryDef;
 
trait ApiWatchlistTrait {
 
    private $watchlistExpiryEnabled;
 
    private $watchlistMaxDuration;
 
    private $watchlistManager;
 
    private $userOptionsLookup;
 
    private function initServices() {
        if ( $this->watchlistManager !== null && $this->userOptionsLookup !== null ) {
            return;
        }
        // This trait is used outside of core and therefor fallback to global state - T263904
        $services = MediaWikiServices::getInstance();
        if ( $this->watchlistManager === null ) {
            $this->watchlistManager = $services->getWatchlistManager();
        }
        if ( $this->userOptionsLookup === null ) {
            $this->userOptionsLookup = $services->getUserOptionsLookup();
        }
    }
 
    protected function getWatchlistParams( array $watchOptions = [] ): array {
        if ( !$watchOptions ) {
            $watchOptions = [
                'watch',
                'unwatch',
                'preferences',
                'nochange',
            ];
        }
 
        $result = [
            'watchlist' => [
                ParamValidator::PARAM_DEFAULT => 'preferences',
                ParamValidator::PARAM_TYPE => $watchOptions,
            ],
        ];
 
        if ( $this->watchlistExpiryEnabled ) {
            $result['watchlistexpiry'] = [
                ParamValidator::PARAM_TYPE => 'expiry',
                ExpiryDef::PARAM_MAX => $this->watchlistMaxDuration,
                ExpiryDef::PARAM_USE_MAX => true,
            ];
        }
 
        return $result;
    }
 
    protected function setWatch(
        string $watch,
        Title $title,
        User $user,
        ?string $userOption = null,
        ?string $expiry = null
    ): void {
        $value = $this->getWatchlistValue( $watch, $title, $user, $userOption );
        $this->watchlistManager->setWatch( $value, $user, $title, $expiry );
    }
 
    protected function getWatchlistValue(
        string $watchlist,
        Title $title,
        User $user,
        ?string $userOption = null
    ): bool {
        $this->initServices();
        $userWatching = $this->watchlistManager->isWatchedIgnoringRights( $user, $title );
 
        switch ( $watchlist ) {
            case 'watch':
                return true;
 
            case 'unwatch':
                return false;
 
            case 'preferences':
                // If the user is already watching, don't bother checking
                if ( $userWatching ) {
                    return true;
                }
                // If the user is a bot, act as 'nochange' to avoid big watchlists on single users
                if ( $user->isBot() ) {
                    return $userWatching;
                }
                // If no user option was passed, use watchdefault and watchcreations
                if ( $userOption === null ) {
                    return $this->userOptionsLookup->getBoolOption( $user, 'watchdefault' ) ||
                        $this->userOptionsLookup->getBoolOption( $user, 'watchcreations' ) &&
                        !$title->exists();
                }
 
                // Watch the article based on the user preference
                return $this->userOptionsLookup->getBoolOption( $user, $userOption );
 
            case 'nochange':
                return $userWatching;
 
            default:
                return $userWatching;
        }
    }
 
    protected function getExpiryFromParams( array $params ): ?string {
        $watchlistExpiry = null;
        if ( $this->watchlistExpiryEnabled && isset( $params['watchlistexpiry'] ) ) {
            $watchlistExpiry = ApiResult::formatExpiry( $params['watchlistexpiry'] );
        }
 
        return $watchlistExpiry;
    }
 
    protected function getWatchlistExpiry(
        WatchedItemStoreInterface $store,
        Title $title,
        UserIdentity $user
    ): ?string {
        $watchedItem = $store->getWatchedItem( $user, $title );
 
        if ( $watchedItem ) {
            $expiry = $watchedItem->getExpiry();
 
            if ( $expiry !== null ) {
                return ApiResult::formatExpiry( $expiry );
            }
        }
 
        return null;
    }
}