mediawiki-extensions-AbuseF.../includes/EditRevUpdater.php
Matěj Suchánek 8f6a428f02 Replace deprecated database object access methods
Use the very new getPrimaryDatabase and getReplicaDatabase.
We skip FilterLookup and CentralDBManager in this patch.

Change-Id: I22c6f8fa60be90599ee177a4ac4a97e1547f79be
2023-03-08 16:50:56 +01:00

139 lines
3.6 KiB
PHP

<?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->update( 'abuse_filter_log',
[ 'afl_rev_id' => $revisionRecord->getId() ],
[ 'afl_id' => $logs['local'] ],
__METHOD__
);
$ret = true;
}
if ( $logs[ 'global' ] ) {
$fdb = $this->centralDBManager->getConnection( DB_PRIMARY );
$fdb->update( 'abuse_filter_log',
[ 'afl_rev_id' => $revisionRecord->getId() ],
[ 'afl_id' => $logs['global'], 'afl_wiki' => $this->wikiID ],
__METHOD__
);
$ret = true;
}
return $ret;
}
/**
* @param LinkTarget $target
* @return string
*/
private function getCacheKey( LinkTarget $target ): string {
return $target->getNamespace() . '|' . $target->getText();
}
}