TranslatableBundleStatusStore.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageGroupProcessing;
 
use Collation;
use InvalidArgumentException;
use MediaWiki\Extension\Translate\MessageBundleTranslation\MessageBundle;
use MediaWiki\Extension\Translate\PageTranslation\TranslatablePage;
use Title;
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\IMaintainableDatabase;
 
class TranslatableBundleStatusStore {
    private const TABLE_NAME = 'translate_translatable_bundles';
    private IDatabase $database;
    private Collation $collation;
    private IMaintainableDatabase $dbMaintenance;
    private ?bool $tableExists = null;
 
    public function __construct(
        IDatabase $database,
        Collation $collation,
        IMaintainableDatabase $dbMaintenance
    ) {
        $this->database = $database;
        $this->collation = $collation;
        $this->dbMaintenance = $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->replace(
            self::TABLE_NAME,
            [ 'ttb_page_id' ],
            [
                'ttb_page_id' => $title->getArticleID(),
                'ttb_type' => $bundleTypeId,
                'ttb_status' => $status->getId(),
                'ttb_sortkey' => $sortKey
            ],
            __METHOD__
        );
    }
 
    public function getAllWithStatus(): array {
        if ( !$this->doesTableExist() ) {
            return [];
        }
 
        $resultSet = $this->database->newSelectQueryBuilder()
            ->select( [ 'ttb_page_id' , 'ttb_status' ] )
            ->from( self::TABLE_NAME )
            ->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->delete(
            self::TABLE_NAME,
            [ 'ttb_page_id' => $pageIds ],
            __METHOD__
        );
    }
 
    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;
    }
}