1 files changed, 74 insertions, 0 deletions
diff --git a/AntiSpoof/maintenance/BatchAntiSpoofClass.php b/AntiSpoof/maintenance/BatchAntiSpoofClass.php
new file mode 100644
index 00000000..e05794cc
--- /dev/null
+++ b/AntiSpoof/maintenance/BatchAntiSpoofClass.php
@@ -0,0 +1,74 @@
+<?php
+
+$IP = getenv( 'MW_INSTALL_PATH' );
+if ( $IP === false ) {
+	$IP = __DIR__ . '/../../..';
+}
+require_once( "$IP/maintenance/Maintenance.php" );
+
+/**
+ * Go through all usernames and calculate and record spoof thingies
+ */
+class BatchAntiSpoof extends Maintenance {
+
+	/**
+	 * @param $items array
+	 */
+	protected function batchRecord( $items ) {
+		SpoofUser::batchRecord( $this->getDB( DB_MASTER ), $items );
+	}
+
+	/**
+	 * @return string
+	 */
+	protected function getTableName() {
+		return 'user';
+	}
+
+	/**
+	 * @return string
+	 */
+	protected function getUserColumn() {
+		return 'user_name';
+	}
+
+	/**
+	 * @param $name string
+	 * @return SpoofUser
+	 */
+	protected function makeSpoofUser( $name ) {
+		return new SpoofUser( $name );
+	}
+
+	/**
+	 * Do the actual work. All child classes will need to implement this
+	 */
+	public function execute() {
+		$dbw = $this->getDB( DB_MASTER );
+
+		$batchSize = 1000;
+
+		$this->output( "Creating username spoofs...\n" );
+		$userCol = $this->getUserColumn();
+		$result = $dbw->select( $this->getTableName(), $userCol, null, __FUNCTION__ );
+		$n = 0;
+		$items = array();
+		foreach( $result as $row ) {
+			if ( $n++ % $batchSize == 0 ) {
+				$this->output( "...$n\n" );
+			}
+
+			$items[] = $this->makeSpoofUser( $row->$userCol );
+
+			if ( $n % $batchSize == 0 ) {
+				$this->batchRecord( $items );
+				$items = array();
+				wfWaitForSlaves();
+			}
+		}
+
+		$this->batchRecord( $items );
+		$this->output( "$n user(s) done.\n" );
+	}
+}
+