TranslationStashStorage.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\TranslatorSandbox;
 
use MediaWiki\Title\Title;
use MediaWiki\User\User;
use Wikimedia\Rdbms\IDatabase;
 
class TranslationStashStorage implements TranslationStashReader, TranslationStashWriter {
    protected $db;
    protected $dbTable;
 
    public function __construct( IDatabase $db, string $table = 'translate_stash' ) {
        $this->db = $db;
        $this->dbTable = $table;
    }
 
    public function getTranslations( User $user ): array {
        $res = $this->db->newSelectQueryBuilder()
            ->select( [ 'ts_namespace', 'ts_title', 'ts_value', 'ts_metadata' ] )
            ->from( $this->dbTable )
            ->where( [ 'ts_user' => $user->getId() ] )
            ->caller( __METHOD__ )
            ->fetchResultSet();
 
        $objects = [];
        foreach ( $res as $row ) {
            $objects[] = new StashedTranslation(
                $user,
                Title::makeTitle( (int)$row->ts_namespace, $row->ts_title ),
                $row->ts_value,
                unserialize( $row->ts_metadata )
            );
        }
 
        return $objects;
    }
 
    public function addTranslation( StashedTranslation $item ): void {
        $row = [
            'ts_user' => $item->getUser()->getId(),
            'ts_title' => $item->getTitle()->getDBkey(),
            'ts_namespace' => $item->getTitle()->getNamespace(),
            'ts_value' => $item->getValue(),
            'ts_metadata' => serialize( $item->getMetadata() ),
        ];
 
        $this->db->newReplaceQueryBuilder()
            ->replaceInto( $this->dbTable )
            ->uniqueIndexFields( [ 'ts_user', 'ts_namespace', 'ts_title' ] )
            ->row( $row )
            ->caller( __METHOD__ )
            ->execute();
    }
 
    public function deleteTranslations( User $user ): void {
        $this->db->newDeleteQueryBuilder()
            ->deleteFrom( $this->dbTable )
            ->where( [ 'ts_user' => $user->getId() ] )
            ->caller( __METHOD__ )
            ->execute();
    }
}