CleanupTranslationProgressStatsMaintenanceScript.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\Statistics;
 
use Maintenance;
use MediaWiki\Extension\Translate\MessageGroupProcessing\MessageGroups;
use MediaWiki\Extension\Translate\Utilities\Utilities;
use MediaWiki\Language\RawMessage;
use MediaWiki\MediaWikiServices;
use const DB_PRIMARY;
 
class CleanupTranslationProgressStatsMaintenanceScript extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Remove obsolete entries from translate_groupstats table' );
        $this->requireExtension( 'Translate' );
    }
 
    public function execute() {
        $services = MediaWikiServices::getInstance();
        $db = $services->getDBLoadBalancer()->getConnection( DB_PRIMARY );
 
        $dbGroupIds = $db->newSelectQueryBuilder()
            ->select( 'tgs_group' )
            ->distinct()
            ->from( 'translate_groupstats' )
            ->caller( __METHOD__ )
            ->fetchFieldValues();
        $knownGroupIds = array_map(
            [ MessageGroupStats::class, 'getDatabaseIdForGroupId' ],
            array_keys( MessageGroups::singleton()->getGroups() )
        );
        $unknownGroupIds = array_diff( $dbGroupIds, $knownGroupIds );
 
        if ( $unknownGroupIds !== [] ) {
            $msg = ( new RawMessage( "Removing rows for $1 unknown group{{PLURAL:$1||s}}:\n" ) )
                ->numParams( count( $unknownGroupIds ) )
                ->inLanguage( 'en' )
                ->text();
            $this->output( $msg );
            foreach ( $unknownGroupIds as $id ) {
                $this->output( "* $id\n" );
            }
            $db->delete(
                'translate_groupstats',
                [ 'tgs_group' => $unknownGroupIds ],
                __METHOD__
            );
        }
 
        $dbLanguages = $db->newSelectQueryBuilder()
            ->select( 'tgs_lang' )
            ->distinct()
            ->from( 'translate_groupstats' )
            ->caller( __METHOD__ )
            ->fetchFieldValues();
        $knownLanguages = array_keys( Utilities::getLanguageNames( 'en' ) );
        $unknownLanguages = array_diff( $dbLanguages, $knownLanguages );
 
        if ( $unknownLanguages !== [] ) {
            $msg = ( new RawMessage( "Removing rows for $1 unknown language{{PLURAL:$1||s}}:\n" ) )
                ->numParams( count( $unknownLanguages ) )
                ->inLanguage( 'en' )
                ->text();
            $this->output( $msg );
            foreach ( $unknownLanguages as $languageCode ) {
                $this->output( "* $languageCode\n" );
            }
            $db->delete(
                'translate_groupstats',
                [ 'tgs_lang' => $unknownLanguages ],
                __METHOD__
            );
        }
 
        $this->output( "Done.\n" );
    }
}