MLEB/Translate/src/PageTranslation/TranslationUnitStore.php


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

<?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';
	/** @var IDatabase */
	private $db;
	/** @var int */
	private $pageId;

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

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

		$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->selectFieldValues(
			self::TABLE,
			'trs_key',
			[ 'trs_page' => $this->pageId ],
			__METHOD__
		);
	}

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