ApiWatch.php

Go to the documentation of this file.

<?php
use MediaWiki\MainConfigNames;
use MediaWiki\Watchlist\WatchlistManager;
use Wikimedia\ParamValidator\ParamValidator;
use Wikimedia\ParamValidator\TypeDef\ExpiryDef;
 
class ApiWatch extends ApiBase {
    private $mPageSet = null;
 
    private $expiryEnabled;
 
    private $maxDuration;
 
    protected $watchlistManager;
 
    public function __construct( ApiMain $mainModule, $moduleName, WatchlistManager $watchlistManager ) {
        parent::__construct( $mainModule, $moduleName );
 
        $this->watchlistManager = $watchlistManager;
        $this->expiryEnabled = $this->getConfig()->get( MainConfigNames::WatchlistExpiry );
        $this->maxDuration = $this->getConfig()->get( MainConfigNames::WatchlistExpiryMaxDuration );
    }
 
    public function execute() {
        $user = $this->getUser();
        if ( !$user->isRegistered() ) {
            $this->dieWithError( 'watchlistanontext', 'notloggedin' );
        }
 
        $this->checkUserRightsAny( 'editmywatchlist' );
 
        $params = $this->extractRequestParams();
 
        $continuationManager = new ApiContinuationManager( $this, [], [] );
        $this->setContinuationManager( $continuationManager );
 
        $pageSet = $this->getPageSet();
        // by default we use pageset to extract the page to work on.
        // title is still supported for backward compatibility
        if ( !isset( $params['title'] ) ) {
            $pageSet->execute();
            $res = $pageSet->getInvalidTitlesAndRevisions( [
                'invalidTitles',
                'special',
                'missingIds',
                'missingRevIds',
                'interwikiTitles'
            ] );
 
            foreach ( $pageSet->getMissingTitles() as $title ) {
                $r = $this->watchTitle( $title, $user, $params );
                $r['missing'] = true;
                $res[] = $r;
            }
 
            foreach ( $pageSet->getGoodTitles() as $title ) {
                $r = $this->watchTitle( $title, $user, $params );
                $res[] = $r;
            }
            ApiResult::setIndexedTagName( $res, 'w' );
        } else {
            // dont allow use of old title parameter with new pageset parameters.
            $extraParams = array_keys( array_filter( $pageSet->extractRequestParams(), static function ( $x ) {
                return $x !== null && $x !== false;
            } ) );
 
            if ( $extraParams ) {
                $this->dieWithError(
                    [
                        'apierror-invalidparammix-cannotusewith',
                        $this->encodeParamName( 'title' ),
                        $pageSet->encodeParamName( $extraParams[0] )
                    ],
                    'invalidparammix'
                );
            }
 
            $title = Title::newFromText( $params['title'] );
            if ( !$title || !$this->watchlistManager->isWatchable( $title ) ) {
                $this->dieWithError( [ 'invalidtitle', $params['title'] ] );
            }
            $res = $this->watchTitle( $title, $user, $params, true );
        }
        $this->getResult()->addValue( null, $this->getModuleName(), $res );
 
        $this->setContinuationManager( null );
        $continuationManager->setContinuationIntoResult( $this->getResult() );
    }
 
    private function watchTitle( Title $title, User $user, array $params,
        $compatibilityMode = false
    ) {
        $res = [ 'title' => $title->getPrefixedText(), 'ns' => $title->getNamespace() ];
 
        if ( !$this->watchlistManager->isWatchable( $title ) ) {
            $res['watchable'] = 0;
            return $res;
        }
 
        if ( $params['unwatch'] ) {
            $status = $this->watchlistManager->removeWatch( $user, $title );
            $res['unwatched'] = $status->isOK();
        } else {
            $expiry = null;
 
            // NOTE: If an expiry parameter isn't given, any existing expiries remain unchanged.
            if ( $this->expiryEnabled && isset( $params['expiry'] ) ) {
                $expiry = $params['expiry'];
                $res['expiry'] = ApiResult::formatExpiry( $expiry );
            }
 
            $status = $this->watchlistManager->addWatch( $user, $title, $expiry );
            $res['watched'] = $status->isOK();
        }
 
        if ( !$status->isOK() ) {
            if ( $compatibilityMode ) {
                $this->dieStatus( $status );
            }
            $res['errors'] = $this->getErrorFormatter()->arrayFromStatus( $status, 'error' );
            $res['warnings'] = $this->getErrorFormatter()->arrayFromStatus( $status, 'warning' );
            if ( !$res['warnings'] ) {
                unset( $res['warnings'] );
            }
        }
 
        return $res;
    }
 
    private function getPageSet() {
        if ( $this->mPageSet === null ) {
            $this->mPageSet = new ApiPageSet( $this );
        }
 
        return $this->mPageSet;
    }
 
    public function mustBePosted() {
        return true;
    }
 
    public function isWriteMode() {
        return true;
    }
 
    public function needsToken() {
        return 'watch';
    }
 
    public function getAllowedParams( $flags = 0 ) {
        $result = [
            'title' => [
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_DEPRECATED => true,
            ],
            'expiry' => [
                ParamValidator::PARAM_TYPE => 'expiry',
                ExpiryDef::PARAM_MAX => $this->maxDuration,
                ExpiryDef::PARAM_USE_MAX => true,
            ],
            'unwatch' => false,
            'continue' => [
                ApiBase::PARAM_HELP_MSG => 'api-help-param-continue',
            ],
        ];
 
        // If expiry is not enabled, don't accept the parameter.
        if ( !$this->expiryEnabled ) {
            unset( $result['expiry'] );
        }
 
        if ( $flags ) {
            $result += $this->getPageSet()->getFinalParams( $flags );
        }
 
        return $result;
    }
 
    protected function getExamplesMessages() {
        $title = Title::newMainPage()->getPrefixedText();
        $mp = rawurlencode( $title );
 
        // Logically expiry example should go before unwatch examples.
        $examples = [
            "action=watch&titles={$mp}&token=123ABC"
                => 'apihelp-watch-example-watch',
        ];
        if ( $this->expiryEnabled ) {
            $examples["action=watch&titles={$mp}|Foo|Bar&expiry=1%20month&token=123ABC"]
                = 'apihelp-watch-example-watch-expiry';
        }
 
        return array_merge( $examples, [
            "action=watch&titles={$mp}&unwatch=&token=123ABC"
                => 'apihelp-watch-example-unwatch',
            'action=watch&generator=allpages&gapnamespace=0&token=123ABC'
                => 'apihelp-watch-example-generator',
        ] );
    }
 
    public function getHelpUrls() {
        return 'https://www.mediawiki.org/wiki/Special:MyLanguage/API:Watch';
    }
}