wikimedia/mediawiki-extensions-Translate

<?php
declare( strict_types = 1 );

namespace MediaWiki\Extension\Translate\TranslatorSandbox;

use MediaWiki\Title\Title;
use MediaWiki\User\User;
use Wikimedia\Rdbms\IDatabase;

/**
 * Storage class for stashed translations. This one uses sql tables as storage.
 *
 * @author Niklas Laxström
 * @license GPL-2.0-or-later
 * @since 2013.06 (namespaced in 2020.11)
 */
class TranslationStashStorage implements TranslationStashReader, TranslationStashWriter {
    /** @var IDatabase */
    protected $db;
    /** @var string */
    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();
    }
}