ApiSetNotificationTimestamp.php

Go to the documentation of this file.

<?php
 
use MediaWiki\Revision\RevisionStore;
use MediaWiki\Title\Title;
use Wikimedia\ParamValidator\ParamValidator;
use Wikimedia\Rdbms\ILoadBalancer;
 
class ApiSetNotificationTimestamp extends ApiBase {
 
    private $mPageSet = null;
 
    private $revisionStore;
 
    private $loadBalancer;
 
    private $watchedItemStore;
 
    public function __construct(
        ApiMain $main,
        $action,
        ILoadBalancer $loadBalancer,
        RevisionStore $revisionStore,
        WatchedItemStoreInterface $watchedItemStore
    ) {
        parent::__construct( $main, $action );
 
        $this->loadBalancer = $loadBalancer;
        $this->revisionStore = $revisionStore;
        $this->watchedItemStore = $watchedItemStore;
    }
 
    public function execute() {
        $user = $this->getUser();
 
        if ( !$user->isRegistered() ) {
            $this->dieWithError( 'watchlistanontext', 'notloggedin' );
        }
        $this->checkUserRightsAny( 'editmywatchlist' );
 
        $params = $this->extractRequestParams();
        $this->requireMaxOneParameter( $params, 'timestamp', 'torevid', 'newerthanrevid' );
 
        $continuationManager = new ApiContinuationManager( $this, [], [] );
        $this->setContinuationManager( $continuationManager );
 
        $pageSet = $this->getPageSet();
        if ( $params['entirewatchlist'] && $pageSet->getDataSource() !== null ) {
            $this->dieWithError(
                [
                    'apierror-invalidparammix-cannotusewith',
                    $this->encodeParamName( 'entirewatchlist' ),
                    $pageSet->encodeParamName( $pageSet->getDataSource() )
                ],
                'multisource'
            );
        }
 
        $dbw = $this->loadBalancer->getConnectionRef( DB_PRIMARY );
 
        $timestamp = null;
        if ( isset( $params['timestamp'] ) ) {
            $timestamp = $dbw->timestamp( $params['timestamp'] );
        }
 
        if ( !$params['entirewatchlist'] ) {
            $pageSet->execute();
        }
 
        if ( isset( $params['torevid'] ) ) {
            if ( $params['entirewatchlist'] || $pageSet->getGoodTitleCount() > 1 ) {
                $this->dieWithError( [ 'apierror-multpages', $this->encodeParamName( 'torevid' ) ] );
            }
            $titles = $pageSet->getGoodTitles();
            $title = reset( $titles );
            if ( $title ) {
                // XXX $title isn't actually used, can we just get rid of the previous six lines?
                $timestamp = $this->revisionStore->getTimestampFromId(
                    $params['torevid'],
                    IDBAccessObject::READ_LATEST
                );
                if ( $timestamp ) {
                    $timestamp = $dbw->timestamp( $timestamp );
                } else {
                    $timestamp = null;
                }
            }
        } elseif ( isset( $params['newerthanrevid'] ) ) {
            if ( $params['entirewatchlist'] || $pageSet->getGoodTitleCount() > 1 ) {
                $this->dieWithError( [ 'apierror-multpages', $this->encodeParamName( 'newerthanrevid' ) ] );
            }
            $titles = $pageSet->getGoodTitles();
            $title = reset( $titles );
            if ( $title ) {
                $timestamp = null;
                $currRev = $this->revisionStore->getRevisionById(
                    $params['newerthanrevid'],
                    Title::READ_LATEST
                );
                if ( $currRev ) {
                    $nextRev = $this->revisionStore->getNextRevision(
                        $currRev,
                        Title::READ_LATEST
                    );
                    if ( $nextRev ) {
                        $timestamp = $dbw->timestamp( $nextRev->getTimestamp() );
                    }
                }
            }
        }
 
        $apiResult = $this->getResult();
        $result = [];
        if ( $params['entirewatchlist'] ) {
            // Entire watchlist mode: Just update the thing and return a success indicator
            $this->watchedItemStore->resetAllNotificationTimestampsForUser( $user, $timestamp );
 
            $result['notificationtimestamp'] = $timestamp === null
                ? ''
                : wfTimestamp( TS_ISO_8601, $timestamp );
        } else {
            // First, log the invalid titles
            foreach ( $pageSet->getInvalidTitlesAndReasons() as $r ) {
                $r['invalid'] = true;
                $result[] = $r;
            }
            foreach ( $pageSet->getMissingPageIDs() as $p ) {
                $page = [];
                $page['pageid'] = $p;
                $page['missing'] = true;
                $page['notwatched'] = true;
                $result[] = $page;
            }
            foreach ( $pageSet->getMissingRevisionIDs() as $r ) {
                $rev = [];
                $rev['revid'] = $r;
                $rev['missing'] = true;
                $rev['notwatched'] = true;
                $result[] = $rev;
            }
 
            if ( $pageSet->getPages() ) {
                // Now process the valid titles
                $this->watchedItemStore->setNotificationTimestampsForUser(
                    $user,
                    $timestamp,
                    $pageSet->getPages()
                );
 
                // Query the results of our update
                $timestamps = $this->watchedItemStore->getNotificationTimestampsBatch(
                    $user,
                    $pageSet->getPages()
                );
 
                // Now, put the valid titles into the result
                foreach ( $pageSet->getTitles() as $title ) {
                    $ns = $title->getNamespace();
                    $dbkey = $title->getDBkey();
                    $r = [
                        'ns' => $ns,
                        'title' => $title->getPrefixedText(),
                    ];
                    if ( !$title->exists() ) {
                        $r['missing'] = true;
                        if ( $title->isKnown() ) {
                            $r['known'] = true;
                        }
                    }
                    if ( isset( $timestamps[$ns] ) && array_key_exists( $dbkey, $timestamps[$ns] )
                        && $timestamps[$ns][$dbkey] !== false
                    ) {
                        $r['notificationtimestamp'] = '';
                        if ( $timestamps[$ns][$dbkey] !== null ) {
                            $r['notificationtimestamp'] = wfTimestamp( TS_ISO_8601, $timestamps[$ns][$dbkey] );
                        }
                    } else {
                        $r['notwatched'] = true;
                    }
                    $result[] = $r;
                }
            }
 
            ApiResult::setIndexedTagName( $result, 'page' );
        }
        $apiResult->addValue( null, $this->getModuleName(), $result );
 
        $this->setContinuationManager( null );
        $continuationManager->setContinuationIntoResult( $apiResult );
    }
 
    private function getPageSet() {
        $this->mPageSet ??= new ApiPageSet( $this );
 
        return $this->mPageSet;
    }
 
    public function mustBePosted() {
        return true;
    }
 
    public function isWriteMode() {
        return true;
    }
 
    public function needsToken() {
        return 'csrf';
    }
 
    public function getAllowedParams( $flags = 0 ) {
        $result = [
            'entirewatchlist' => [
                ParamValidator::PARAM_TYPE => 'boolean'
            ],
            'timestamp' => [
                ParamValidator::PARAM_TYPE => 'timestamp'
            ],
            'torevid' => [
                ParamValidator::PARAM_TYPE => 'integer'
            ],
            'newerthanrevid' => [
                ParamValidator::PARAM_TYPE => 'integer'
            ],
            'continue' => [
                ApiBase::PARAM_HELP_MSG => 'api-help-param-continue',
            ],
        ];
        if ( $flags ) {
            $result += $this->getPageSet()->getFinalParams( $flags );
        }
 
        return $result;
    }
 
    protected function getExamplesMessages() {
        $title = Title::newMainPage()->getPrefixedText();
        $mp = rawurlencode( $title );
 
        return [
            'action=setnotificationtimestamp&entirewatchlist=&token=123ABC'
                => 'apihelp-setnotificationtimestamp-example-all',
            "action=setnotificationtimestamp&titles={$mp}&token=123ABC"
                => 'apihelp-setnotificationtimestamp-example-page',
            "action=setnotificationtimestamp&titles={$mp}&" .
                'timestamp=2012-01-01T00:00:00Z&token=123ABC'
                => 'apihelp-setnotificationtimestamp-example-pagetimestamp',
            'action=setnotificationtimestamp&generator=allpages&gapnamespace=2&token=123ABC'
                => 'apihelp-setnotificationtimestamp-example-allpages',
        ];
    }
 
    public function getHelpUrls() {
        return 'https://www.mediawiki.org/wiki/Special:MyLanguage/API:SetNotificationTimestamp';
    }
}