<?php

$IP = getenv( 'MW_INSTALL_PATH' );
if ( $IP === false ) {
	$IP = __DIR__ . '/../../..';
}
require_once "$IP/maintenance/Maintenance.php";

class BackfillReadBundles extends Maintenance {
	public function __construct() {
		parent::__construct();

		$this->mDescription = "Backfill echo_notification.notification_read_timestamp for bundles";

		$this->setBatchSize( 300 );

		$this->requireExtension( 'Echo' );
	}

	public function execute() {
		$dbFactory = MWEchoDbFactory::newFromDefault();
		$dbw = $dbFactory->getEchoDb( DB_MASTER );
		$dbr = $dbFactory->getEchoDb( DB_SLAVE );
		$iterator = new BatchRowIterator(
			$dbr,
			'echo_notification',
			[ 'notification_user', 'notification_event' ],
			$this->mBatchSize
		);
		$iterator->setFetchColumns( [ 'notification_bundle_display_hash', 'notification_read_timestamp' ] );

		$unreadNonBase = $dbr->selectSQLText(
			'echo_notification',
			'notification_bundle_display_hash',
			[
				'notification_bundle_base' => 0,
				'notification_read_timestamp IS NULL',
				"notification_bundle_display_hash <> ''",
			]
		);

		$iterator->addConditions( [
			'notification_bundle_base' => 1,
			'notification_read_timestamp IS NOT NULL',
			"notification_bundle_display_hash IN ( $unreadNonBase )",
		] );

		$processed = 0;
		foreach ( $iterator as $batch ) {
			foreach ( $batch as $row ) {
				$userId = $row->notification_user;
				$displayHash = $row->notification_bundle_display_hash;
				$readTimestamp = $row->notification_read_timestamp;

				$result = $dbw->update(
					'echo_notification',
					[ 'notification_read_timestamp' => $readTimestamp ],
					[
						'notification_user' => $userId,
						'notification_bundle_display_hash' => $displayHash,
						'notification_bundle_base' => 0,
						'notification_read_timestamp IS NULL',
					]
				);

				if ( !$result ) {
					$this->output( "Failed to set read_timestamp on notifications with bundle_display_hash: $displayHash\n" );
				}

				$processed += $dbw->affectedRows();
			}

			$this->output( "Updated $processed notifications.\n" );
			$dbFactory->waitForSlaves();
		}
	}
}

$maintClass = "BackfillReadBundles";
require_once DO_MAINTENANCE;