Merge "Add maint script to opt out active users from the new topic tool"

extension.json

@@ -336,7 +336,8 @@
 		]
 	},
 	"AutoloadNamespaces": {
-		"MediaWiki\\Extension\\DiscussionTools\\": "includes/"
+		"MediaWiki\\Extension\\DiscussionTools\\": "includes/",
+		"MediaWiki\\Extension\\DiscussionTools\\Maintenance\\": "maintenance/"
 	},
 	"TestAutoloadNamespaces": {
 		"MediaWiki\\Extension\\DiscussionTools\\Tests\\Unit\\": "tests/phpunit/unit/",

maintenance/NewTopicOptOutActiveUsers.php

@@ -0,0 +1,119 @@
+<?php
+
+namespace MediaWiki\Extension\DiscussionTools\Maintenance;
+
+use IDatabase;
+use Maintenance;
+use MediaWiki\Extension\DiscussionTools\Hooks\HookUtils;
+
+class NewTopicOptOutActiveUsers extends Maintenance {
+
+	private IDatabase $dbw;
+
+	public function __construct() {
+		parent::__construct();
+		$this->requireExtension( 'DiscussionTools' );
+		$this->addDescription( 'Opt out active users from the new topic tool' );
+		$this->addOption( 'dry-run', 'Output information, do not save changes' );
+		$this->addOption( 'save', 'Save the changes to the database' );
+		$this->setBatchSize( 100 );
+	}
+
+	public function execute() {
+		if ( $this->hasOption( 'dry-run' ) ) {
+			$save = false;
+			$this->output( "Dry run:\n" );
+		} elseif ( $this->hasOption( 'save' ) ) {
+			$save = true;
+			$this->output( "CHANGING PREFERENCES!\n" );
+			$this->countDown( 5 );
+		} else {
+			$this->error( "Please provide '--dry-run' or '--save' option" );
+			return;
+		}
+
+		$this->dbw = $this->getDB( DB_PRIMARY );
+
+		$userRows = $this->dbw->newSelectQueryBuilder()
+			->caller( __METHOD__ )
+			->table( 'querycachetwo' )
+			->where( [
+				'qcc_type' => 'activeusers',
+				'qcc_namespace' => NS_USER,
+			] )
+			->join( 'user', null, 'qcc_title=user_name' )
+			->where( [ 'user_editcount >= 100' ] )
+			->fields( [ 'user_id', 'user_name' ] )
+			->fetchResultSet();
+
+		$count = count( $userRows );
+		$countUpdated = 0;
+		$this->output( "Found $count active users with enough edits\n" );
+
+		foreach ( $userRows as $i => $row ) {
+			$skipReason = $this->skipReason( $row->user_id );
+			if ( $skipReason ) {
+				$this->output( "Won't update '$row->user_name' because: $skipReason\n" );
+			} else {
+				$this->output( "Will update '$row->user_name'\n" );
+				$countUpdated++;
+				if ( $save ) {
+					$this->updatePrefs( $row->user_id );
+					if ( $countUpdated % $this->getBatchSize() === 0 ) {
+						$this->waitForReplication();
+					}
+				}
+			}
+		}
+
+		if ( $save ) {
+			$this->output( "Updated $countUpdated out of $count users\n" );
+		} else {
+			$this->output( "Would update $countUpdated out of $count users\n" );
+		}
+	}
+
+	private function skipReason( int $userId ): ?string {
+		// We can't use UserOptionsLookup here, because we're not interested in the default options,
+		// but only in the options actually stored in the database.
+
+		// We're not looking at global preferences, because if the user has set them, then they will
+		// override our local preferences anyway.
+
+		// Check that the user has not already set their preference for new topic tool to any value
+		$foundRow = $this->dbw->newSelectQueryBuilder()
+			->caller( __METHOD__ )
+			->table( 'user_properties' )
+			->where( [ 'up_user' => $userId, 'up_property' => 'discussiontools-' . HookUtils::NEWTOPICTOOL ] )
+			->field( '1' )
+			->fetchField();
+		if ( $foundRow ) {
+			return HookUtils::NEWTOPICTOOL;
+		}
+
+		// Check that the user has not already opted into the beta feature
+		$foundRow = $this->dbw->newSelectQueryBuilder()
+			->caller( __METHOD__ )
+			->table( 'user_properties' )
+			->where( [ 'up_user' => $userId, 'up_property' => 'discussiontools-betaenable', 'up_value' => 1 ] )
+			->field( '1' )
+			->fetchField();
+		if ( $foundRow ) {
+			return 'betaenable';
+		}
+
+		return null;
+	}
+
+	private function updatePrefs( int $userId ): void {
+		// We can't use UserOptionsManager here, because we want to store the preference
+		// in the database even if it's identical to the current default
+		// (this script is only used when we're about to change the default).
+		$this->dbw->insert(
+			'user_properties',
+			[ 'up_user' => $userId, 'up_property' => 'discussiontools-' . HookUtils::NEWTOPICTOOL, 'up_value' => 0 ],
+			__METHOD__
+		);
+	}
+
+}