summaryrefslogtreecommitdiff
blob: 764af9e5fa10a0f8238a15877ec539344d4695fa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
<?php
declare( strict_types = 1 );

namespace MediaWiki\Extension\Translate\Statistics;

use Maintenance;
use MediaWiki\MediaWikiServices;
use MessageGroups;
use RawMessage;
use TranslateUtils;
use const DB_MASTER;

/**
 * @since 2021.03
 * @license GPL-2.0-or-later
 * @author Niklas Laxström
 */
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()->getConnectionRef( DB_MASTER );

		$dbGroupIds = $db->selectFieldValues(
			'translate_groupstats',
			'DISTINCT(tgs_group)',
			'*',
			__METHOD__
		);
		$knownGroupIds = array_map(
			'MessageGroupStats::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->selectFieldValues(
			'translate_groupstats',
			'DISTINCT(tgs_lang)',
			'*',
			__METHOD__
		);
		$knownLanguages = array_keys( TranslateUtils::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" );
	}
}