MessageGroupStatesUpdaterJob.php

Go to the documentation of this file.

<?php
use MediaWiki\Extension\Translate\Jobs\GenericTranslateJob;
use MediaWiki\Extension\Translate\MessageGroupProcessing\MessageGroups;
use MediaWiki\Extension\Translate\Services;
use MediaWiki\Extension\Translate\SystemUsers\FuzzyBot;
use MediaWiki\MediaWikiServices;
 
class MessageGroupStatesUpdaterJob extends GenericTranslateJob {
    public function __construct( $title, $params = [] ) {
        parent::__construct( __CLASS__, $title, $params );
        $this->removeDuplicates = true;
    }
 
    public static function onChange( MessageHandle $handle ) {
        $job = self::newJob( $handle->getTitle() );
        MediaWikiServices::getInstance()->getJobQueueGroup()->push( $job );
 
        return true;
    }
 
    public static function newJob( $title ) {
        $job = new self( $title );
 
        return $job;
    }
 
    public function run() {
        $lb = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
        if ( !$lb->waitForReplication() ) {
            $this->logWarning( 'Continuing despite replication lag' );
        }
 
        $title = $this->title;
        $handle = new MessageHandle( $title );
        $code = $handle->getCode();
 
        if ( !$code && !$handle->isValid() ) {
            return true;
        }
 
        $groups = self::getGroupsWithTransitions( $handle );
        $messageGroupReviewStore = Services::getInstance()->getMessageGroupReviewStore();
        foreach ( $groups as $id => $transitions ) {
            $group = MessageGroups::getGroup( $id );
            $stats = MessageGroupStats::forItem( $id, $code, MessageGroupStats::FLAG_IMMEDIATE_WRITES );
            $state = self::getNewState( $stats, $transitions );
            if ( $state ) {
                $messageGroupReviewStore->changeState( $group, $code, $state, FuzzyBot::getUser() );
            }
        }
 
        return true;
    }
 
    public static function getGroupsWithTransitions( MessageHandle $handle ) {
        $listeners = [];
        foreach ( $handle->getGroupIds() as $id ) {
            $group = MessageGroups::getGroup( $id );
 
            // No longer exists?
            if ( !$group ) {
                continue;
            }
 
            $conds = $group->getMessageGroupStates()->getConditions();
            if ( $conds ) {
                $listeners[$id] = $conds;
            }
        }
 
        return $listeners;
    }
 
    public static function getStatValue( $stats, $type ) {
        $total = $stats[MessageGroupStats::TOTAL];
        $translated = $stats[MessageGroupStats::TRANSLATED];
        $outdated = $stats[MessageGroupStats::FUZZY];
        $proofread = $stats[MessageGroupStats::PROOFREAD];
 
        switch ( $type ) {
            case 'UNTRANSLATED':
                return $total - $translated - $outdated;
            case 'OUTDATED':
                return $outdated;
            case 'TRANSLATED':
                return $translated;
            case 'PROOFREAD':
                return $proofread;
            default:
                throw new InvalidArgumentException( "Unknown condition $type" );
        }
    }
 
    public static function matchCondition( $value, $condition, $max ) {
        switch ( $condition ) {
            case 'ZERO':
                return $value === 0;
            case 'NONZERO':
                return $value > 0;
            case 'MAX':
                return $value === $max;
            default:
                throw new InvalidArgumentException( "Unknown condition value $condition" );
        }
    }
 
    public static function getNewState( $stats, $transitions ) {
        foreach ( $transitions as $transition ) {
            [ $newState, $conds ] = $transition;
            $match = true;
 
            foreach ( $conds as $type => $cond ) {
                $statValue = self::getStatValue( $stats, $type );
                $max = $stats[MessageGroupStats::TOTAL];
                $match = $match && self::matchCondition( $statValue, $cond, $max );
                // Conditions are AND, so no point trying more if no match
                if ( !$match ) {
                    break;
                }
            }
 
            if ( $match ) {
                return $newState;
            }
        }
 
        return false;
    }
}