wikimedia/mediawiki-extensions-Translate

<?php
declare( strict_types = 1 );

namespace MediaWiki\Extension\Translate\PageTranslation;

use Wikimedia\Rdbms\IDatabase;

/**
 * @author Niklas Laxström
 * @license GPL-2.0-or-later
 * @since 2021.05
 */
class TranslationUnitStore implements TranslationUnitReader {
    private const TABLE = 'translate_sections';
    private IDatabase $db;
    private int $pageId;

    public function __construct( IDatabase $db, int $pageId ) {
        $this->db = $db;
        $this->pageId = $pageId;
    }

    public function getUnits(): array {
        $res = $this->db->newSelectQueryBuilder()
            ->select( [ 'trs_key', 'trs_text' ] )
            ->from( self::TABLE )
            ->where( [ 'trs_page' => $this->pageId ] )
            ->caller( __METHOD__ )
            ->fetchResultSet();

        $units = [];
        foreach ( $res as $row ) {
            $units[$row->trs_key] = new TranslationUnit( $row->trs_text, $row->trs_key );
        }

        return $units;
    }

    /** @return string[] */
    public function getNames(): array {
        return $this->db->newSelectQueryBuilder()
            ->select( 'trs_key' )
            ->from( self::TABLE )
            ->where( [ 'trs_page' => $this->pageId ] )
            ->caller( __METHOD__ )
            ->fetchFieldValues();
    }

    public function delete(): void {
        $this->db->newDeleteQueryBuilder()
            ->deleteFrom( self::TABLE )
            ->where( [ 'trs_page' => $this->pageId ] )
            ->caller( __METHOD__ )
            ->execute();
    }
}