Page MenuHomeWickedGov Phorge
Files F1428359

EditRevUpdater.php
No OneTemporary
Actions

Size
3 KB
Referenced Files
None
Subscribers
None

EditRevUpdater.php
View Options

<?php
namespace MediaWiki\Extension\AbuseFilter;
use InvalidArgumentException;
use MediaWiki\Linker\LinkTarget;
use MediaWiki\Revision\RevisionLookup;
use MediaWiki\Revision\RevisionRecord;
use Wikimedia\Rdbms\LBFactory;
use WikiPage;
/**
* This service allows "linking" the edit filter hook and the page save hook
*/
class EditRevUpdater {
public const SERVICE_NAME = 'AbuseFilterEditRevUpdater';
/** @var CentralDBManager */
private $centralDBManager;
/** @var RevisionLookup */
private $revisionLookup;
/** @var LBFactory */
private $lbFactory;
/** @var string */
private $wikiID;
/** @var WikiPage|null */
private $wikiPage;
/**
* @var int[][][] IDs of logged filters like [ page title => [ 'local' => [ids], 'global' => [ids] ] ].
* @phan-var array<string,array{local:int[],global:int[]}>
*/
private $logIds = [];
/**
* @param CentralDBManager $centralDBManager
* @param RevisionLookup $revisionLookup
* @param LBFactory $lbFactory
* @param string $wikiID
*/
public function __construct(
CentralDBManager $centralDBManager,
RevisionLookup $revisionLookup,
LBFactory $lbFactory,
string $wikiID
) {
$this->centralDBManager = $centralDBManager;
$this->revisionLookup = $revisionLookup;
$this->lbFactory = $lbFactory;
$this->wikiID = $wikiID;
}
/**
* Set the WikiPage object used for the ongoing edit
*
* @param WikiPage $page
*/
public function setLastEditPage( WikiPage $page ): void {
$this->wikiPage = $page;
}
/**
* Clear the WikiPage object used for the ongoing edit
*/
public function clearLastEditPage(): void {
$this->wikiPage = null;
}
/**
* @param LinkTarget $target
* @param int[][] $logIds
* @phan-param array{local:int[],global:int[]} $logIds
*/
public function setLogIdsForTarget( LinkTarget $target, array $logIds ): void {
if ( count( $logIds ) !== 2 || array_diff( array_keys( $logIds ), [ 'local', 'global' ] ) ) {
throw new InvalidArgumentException( 'Wrong keys; got: ' . implode( ', ', array_keys( $logIds ) ) );
}
$key = $this->getCacheKey( $target );
$this->logIds[$key] = $logIds;
}
/**
* @param WikiPage $wikiPage
* @param RevisionRecord $revisionRecord
* @return bool Whether the DB was updated
*/
public function updateRev( WikiPage $wikiPage, RevisionRecord $revisionRecord ): bool {
$key = $this->getCacheKey( $wikiPage->getTitle() );
if ( !isset( $this->logIds[ $key ] ) || $wikiPage !== $this->wikiPage ) {
// This isn't the edit $this->logIds was set for
$this->logIds = [];
return false;
}
// Ignore null edit.
$parentRevId = $revisionRecord->getParentId();
if ( $parentRevId !== null ) {
$parentRev = $this->revisionLookup->getRevisionById( $parentRevId );
if ( $parentRev && $revisionRecord->hasSameContent( $parentRev ) ) {
$this->logIds = [];
return false;
}
}
$this->clearLastEditPage();
$ret = false;
$logs = $this->logIds[ $key ];
if ( $logs[ 'local' ] ) {
$dbw = $this->lbFactory->getPrimaryDatabase();
$dbw->newUpdateQueryBuilder()
->update( 'abuse_filter_log' )
->set( [ 'afl_rev_id' => $revisionRecord->getId() ] )
->where( [ 'afl_id' => $logs['local'] ] )
->caller( __METHOD__ )
->execute();
$ret = true;
}
if ( $logs[ 'global' ] ) {
$fdb = $this->centralDBManager->getConnection( DB_PRIMARY );
$fdb->newUpdateQueryBuilder()
->update( 'abuse_filter_log' )
->set( [ 'afl_rev_id' => $revisionRecord->getId() ] )
->where( [ 'afl_id' => $logs['global'], 'afl_wiki' => $this->wikiID ] )
->caller( __METHOD__ )
->execute();
$ret = true;
}
return $ret;
}
/**
* @param LinkTarget $target
* @return string
*/
private function getCacheKey( LinkTarget $target ): string {
return $target->getNamespace() . '|' . $target->getText();
}
}

File Metadata

Mime Type
text/x-php
Expires
Sat, May 16, 16:07 (13 h, 43 m)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
b5/85/54672823d23003e152edffc34ecd
Default Alt Text
EditRevUpdater.php (3 KB)

Event Timeline