dieIfEchoNotInstalled(); $user = $this->getUser(); $this->dieOnBadUser( $user ); $params = $this->extractRequestParams(); $revision = $this->getRevisionFromParams( $params ); if ( $this->userAlreadySentThanksForRevision( $user, $revision ) ) { $this->markResultSuccess( $revision ); } else { $recipient = $this->getUserFromRevision( $revision ); $this->dieOnBadRecipient( $user, $recipient ); $this->sendThanks( $user, $revision, $recipient, $this->getSourceFromParams( $params ) ); } } private function userAlreadySentThanksForRevision( User $user, Revision $revision ) { return $user->getRequest()->getSessionData( "thanks-thanked-{$revision->getId()}" ); } private function dieIfEchoNotInstalled() { if ( !class_exists( 'EchoNotifier' ) ) { $this->dieUsage( 'Echo is not installed on this wiki', 'echonotinstalled' ); } } private function dieOnBadUser( User $user ) { if ( $user->isAnon() ) { $this->dieUsage( 'Anonymous users cannot send thanks', 'notloggedin' ); } elseif ( $user->pingLimiter( 'thanks-notification' ) ) { $this->dieUsageMsg( array( 'actionthrottledtext' ) ); } elseif ( $user->isBlocked() ) { $this->dieUsageMsg( array( 'blockedtext' ) ); } } private function getRevisionFromParams( $params ) { $revision = Revision::newFromId( $params['rev'] ); if ( !$revision ) { $this->dieUsage( 'Revision ID is not valid', 'invalidrevision' ); } elseif ( $revision->isDeleted( Revision::DELETED_TEXT ) ) { $this->dieUsage( 'Revision has been deleted', 'revdeleted' ); } return $revision; } private function getTitleFromRevision( Revision $revision ) { $title = Title::newFromID( $revision->getPage() ); if ( !$title instanceof Title ) { $this->dieUsage( 'Page title could not be retrieved', 'notitle' ); } return $title; } /** * Set the source of the thanks, e.g. 'diff' or 'history' */ private function getSourceFromParams( $params ) { if ( $params['source'] ) { return trim( $params['source'] ); } else { return 'undefined'; } } private function getUserFromRevision( Revision $revision ) { $recipient = $revision->getUser(); if ( !$recipient ) { $this->dieUsage( 'No valid recipient found', 'invalidrecipient' ); } return User::newFromId( $recipient ); } private function markResultSuccess( Revision $revision ) { $this->getResult()->addValue( null, 'result', array( 'success' => 1, 'recipient' => $revision->getUserText( Revision::FOR_PUBLIC ) ) ); } private function dieOnBadRecipient( User $agent, User $recipient ) { global $wgThanksSendToBots; if ( $agent->getId() === $recipient->getId() ) { $this->dieUsage( 'You cannot thank yourself', 'invalidrecipient' ); } elseif ( !$wgThanksSendToBots && in_array( 'bot', $recipient->getGroups() ) ) { $this->dieUsage( 'Bots cannot be thanked', 'invalidrecipient' ); } } private function sendThanks( User $user, Revision $revision, User $recipient, $source ) { global $wgThanksLogging; $title = $this->getTitleFromRevision( $revision ); // Create the notification via Echo extension EchoEvent::create( array( 'type' => 'edit-thank', 'title' => $title, 'extra' => array( 'revid' => $revision->getId(), 'thanked-user-id' => $recipient->getId(), 'source' => $source, ), 'agent' => $user, ) ); // Mark the thank in session to prevent duplicates (Bug 46690) $user->getRequest()->setSessionData( "thanks-thanked-{$revision->getId()}", true ); // Set success message $this->markResultSuccess( $revision ); // Log it if we're supposed to log it if ( $wgThanksLogging ) { $logEntry = new ManualLogEntry( 'thanks', 'thank' ); $logEntry->setPerformer( $user ); $target = $recipient->getUserPage(); $logEntry->setTarget( $target ); $logId = $logEntry->insert(); $logEntry->publish( $logId, 'udp' ); } } public function getDescription() { return array( 'This API is for sending thank you notifications from one editor to another.', ); } public function getParamDescription() { return array( 'rev' => 'A revision ID for an edit that you want to thank someone for', 'token' => 'An edit token (to prevent CSRF abuse)', 'source' => "A short string describing the source of the request, for example, 'diff' or 'history'", ); } public function getAllowedParams() { return array( 'rev' => array( ApiBase::PARAM_TYPE => 'integer', ApiBase::PARAM_MIN => 1, ApiBase::PARAM_REQUIRED => true, ), 'token' => array( ApiBase::PARAM_TYPE => 'string', ApiBase::PARAM_REQUIRED => true, ), 'source' => array( ApiBase::PARAM_TYPE => 'string', ApiBase::PARAM_REQUIRED => false, ) ); } public function needsToken() { return true; } // Writes to the Echo database and sometimes log tables. public function isWriteMode() { return true; } public function getTokenSalt() { return ''; } public function getHelpUrls() { return array( 'https://www.mediawiki.org/wiki/Extension:Thanks#API_Documentation', ); } public function getVersion() { return __CLASS__ . '-1.0'; } }