TranslatableBundleStatusStore.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageGroupProcessing;
 
use InvalidArgumentException;
use MediaWiki\Collation\Collation;
use MediaWiki\Extension\Translate\MessageBundleTranslation\MessageBundle;
use MediaWiki\Extension\Translate\PageTranslation\TranslatablePage;
use MediaWiki\Title\Title;
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\IMaintainableDatabase;
 
class TranslatableBundleStatusStore {
    private const TABLE_NAME = 'translate_translatable_bundles';
 
    private ?bool $tableExists = null;
 
    public function __construct(
        private readonly IDatabase $database,
        private readonly Collation $collation,
        private readonly IMaintainableDatabase $dbMaintenance,
    ) {
    }
 
    public function setStatus( Title $title, TranslatableBundleStatus $status, string $bundleType ): void {
        if ( !$this->doesTableExist() ) {
            return;
        }
 
        $sortKey = substr( $this->collation->getSortKey( $title->getPrefixedDBkey() ), 0, 255 );
        $bundleTypeId = $this->getBundleTypeId( $bundleType );
        $this->database->newReplaceQueryBuilder()
            ->replaceInto( self::TABLE_NAME )
            ->uniqueIndexFields( [ 'ttb_page_id' ] )
            ->row( [
                'ttb_page_id' => $title->getId(),
                'ttb_type' => $bundleTypeId,
                'ttb_status' => $status->getId(),
                'ttb_sortkey' => $sortKey
            ] )
            ->caller( __METHOD__ )
            ->execute();
    }
 
    public function getAllWithStatus(): array {
        if ( !$this->doesTableExist() ) {
            return [];
        }
 
        $resultSet = $this->database->newSelectQueryBuilder()
            ->select( [ 'ttb_page_id', 'ttb_status' ] )
            ->from( self::TABLE_NAME )
            ->caller( __METHOD__ )
            ->fetchResultSet();
 
        $result = [];
        foreach ( $resultSet as $row ) {
            $result[$row->ttb_page_id] = (int)$row->ttb_status;
        }
 
        return $result;
    }
 
    public function removeStatus( int ...$pageIds ): void {
        if ( !$this->doesTableExist() ) {
            return;
        }
 
        $this->database->newDeleteQueryBuilder()
            ->deleteFrom( self::TABLE_NAME )
            ->where( [ 'ttb_page_id' => $pageIds ] )
            ->caller( __METHOD__ )
            ->execute();
    }
 
    private function getBundleTypeId( string $bundle ): int {
        if ( $bundle === TranslatablePage::class ) {
            return 1;
        } elseif ( $bundle === MessageBundle::class ) {
            return 2;
        }
 
        throw new InvalidArgumentException( "Unknown translatable bundle type: $bundle" );
    }
 
    private function doesTableExist(): bool {
        if ( $this->tableExists === null ) {
            $this->tableExists = $this->dbMaintenance->tableExists( self::TABLE_NAME, __METHOD__ );
        }
 
        return $this->tableExists;
    }
}