Your IP : 216.73.216.86


Current Path : /var/www/homesaver/www/bitrix/modules/translate/lib/controller/export/
Upload File :
Current File : /var/www/homesaver/www/bitrix/modules/translate/lib/controller/export/exportfilesearch.php

<?php
namespace Bitrix\Translate\Controller\Export;

use Bitrix\Main;
use Bitrix\Main\Localization\Loc;
use Bitrix\Translate;
use Bitrix\Translate\Index;

/**
 * Harvester of phrases using search result with index.
 */
class ExportFileSearch
	extends ExportAction
	implements Translate\Controller\ITimeLimit, Translate\Controller\IProcessParameters
{
	use Translate\Controller\Stepper;
	use Translate\Controller\ProcessParams;

	private int $seekPathId = 0;
	private string $seekPhraseCode = '';

	/**
	 * \Bitrix\Main\Engine\Action constructor.
	 *
	 * @param string $name Action name.
	 * @param Main\Engine\Controller $controller Parent controller object.
	 * @param array $config Additional configuration.
	 */
	public function __construct($name, Main\Engine\Controller $controller, array $config = [])
	{
		$this->keepField(['seekPathId', 'seekPhraseCode']);

		Loc::loadLanguageFile(__DIR__ . '/exportaction.php');

		parent::__construct($name, $controller, $config);
	}


	/**
	 * Runs controller action.
	 *
	 * @param string $path Path to export.
	 * @param boolean $runBefore Flag to run onBeforeRun event handler.
	 *
	 * @return array
	 */
	public function run(string $path = '', bool $runBefore = false): array
	{
		if (empty($path))
		{
			$path = Translate\Config::getDefaultPath();
		}

		if ($runBefore)
		{
			$this->onBeforeRun();
		}

		if ($this->isNewProcess)
		{
			$this->totalItems = (int)Index\FileIndexSearch::getCount($this->processFilter($path));
			$this->processedItems = 0;

			if ($this->totalItems > 0)
			{
				$this->exportFileName = $this->generateExportFileName($path, $this->languages);
				$csvFile = $this->createExportTempFile($this->exportFileName);
				$this->exportFilePath = $csvFile->getPhysicalPath();
				$this->exportFileSize = $csvFile->getSize();
			}
			if ($this->appendSamples)
			{
				$this->samplesFileName = $this->generateExportFileName($path.'-samples', $this->languages);
				$sampleFile = $this->createExportTempFile($this->samplesFileName);
				$this->samplesFilePath = $sampleFile->getPhysicalPath();
				$this->samplesFileSize = $sampleFile->getSize();
			}

			$this->saveProgressParameters();

			return [
				'STATUS' => ($this->totalItems > 0 ? Translate\Controller\STATUS_PROGRESS : Translate\Controller\STATUS_COMPLETED),
				'PROCESSED_ITEMS' => 0,
				'TOTAL_ITEMS' => $this->totalItems,
				'TOTAL_PHRASES' => $this->exportedPhraseCount,
				'TOTAL_SAMPLES' => $this->exportedSamplesCount,
			];
		}

		return $this->performStep('runExporting', ['path' => $path]);
	}


	/**
	 * Collects lang files paths.
	 *
	 * @param array $params Path to indexing.
	 *
	 * @return array
	 */
	private function runExporting(array $params): array
	{
		$path = \rtrim($params['path'], '/');

		$csvFile = new Translate\IO\CsvFile($this->exportFilePath);
		$this->configureExportCsvFile($csvFile);
		$csvFile->openWrite( Main\IO\FileStreamOpenMode::APPEND);

		if ($this->appendSamples)
		{
			$samplesFile = new Translate\IO\CsvFile($this->samplesFilePath);
			$this->configureExportCsvFile($samplesFile);
			$samplesFile->openWrite( Main\IO\FileStreamOpenMode::APPEND);
		}

		$pathFilter = $this->processFilter($path);
		if (!empty($this->seekPathId))
		{
			$pathFilter['>PATH_ID'] = $this->seekPathId;
		};

		$select = ['PATH_ID', 'PATH'];

		foreach ($this->languages as $langId)
		{
			$select[] = \mb_strtoupper($langId)."_LANG";
		}
		if (!\in_array($this->filter['LANGUAGE_ID'], $this->languages))
		{
			$select[] = \mb_strtoupper($this->filter['LANGUAGE_ID'])."_LANG";
		}

		$currentLangId = Loc::getCurrentLang();

		/** @var Main\ORM\Query\Result $cachePathRes */
		$pathInxRes = Index\FileIndexSearch::getList([
			'filter' => $pathFilter,
			'order' => ['PATH_ID' => 'ASC'],
			'select' => $select,
		]);

		$processedItemCount = 0;
		while ($pathInx = $pathInxRes->fetch())
		{
			$fullPaths = $this->getFullPath($pathInx['PATH_ID']);

			$rows = $this->mergeLangFiles($pathInx['PATH'], $fullPaths, $this->collectUntranslated);
			foreach ($rows as $code => $row)
			{
				if (!empty($this->seekPhraseCode))
				{
					if ($code == $this->seekPhraseCode)
					{
						$this->seekPhraseCode = '';
					}
					continue;
				}

				$csvFile->put(array_values($row));

				if (
					$this->appendSamples
					&& !empty($row[$currentLangId])
					&& mb_strlen($row[$currentLangId]) < $this->maxSampleSourceLength
				)
				{
					$samples = $this->findSamples(
						$row[$currentLangId],
						$currentLangId,
						(int)$pathInx['PATH_ID'],
						$this->samplesCount,
						$this->samplesRestriction
					);
					foreach ($samples as $sample)
					{
						$samplesFile->put(\array_values($sample));
						$this->exportedSamplesCount ++;
					}
				}

				$this->exportedPhraseCount ++;

				if ($this->instanceTimer()->hasTimeLimitReached())
				{
					$this->seekPhraseCode = $code;
					break;
				}
				else
				{
					$this->seekPhraseCode = '';
				}
			}

			$processedItemCount ++;

			if ($this->instanceTimer()->hasTimeLimitReached())
			{
				$this->seekPathId = (int)$pathInx['PATH_ID'];
				break;
			}
		}

		$this->exportFileSize = $csvFile->getSize();
		$csvFile->close();

		if ($this->appendSamples)
		{
			$this->samplesFileSize = $samplesFile->getSize();
			$samplesFile->close();
		}

		$this->processedItems += $processedItemCount;

		if ($this->instanceTimer()->hasTimeLimitReached() !== true)
		{
			$this->declareAccomplishment();
			$this->clearProgressParameters();
		}

		$result = [
			'PROCESSED_ITEMS' => $this->processedItems,
			'TOTAL_ITEMS' => $this->totalItems,
			'TOTAL_PHRASES' => $this->exportedPhraseCount,
			'TOTAL_SAMPLES' => $this->exportedSamplesCount,
		];

		if ($csvFile->hasErrors())
		{
			$errors = $csvFile->getErrors();
			foreach ($errors as $err)
			{
				if ($err->getCode() == Translate\IO\CsvFile::ERROR_32K_FIELD_LENGTH)
				{
					$result['WARNING'] = Loc::getMessage('TR_EXPORT_ERROR_32K_LENGTH');
				}
				else
				{
					$this->addError($err);
				}
			}
		}

		return $result;
	}

	/**
	 * Process incoming filter from client.
	 *
	 * @param string $path Path to export.
	 *
	 * @return array
	 */
	private function processFilter(string $path): array
	{
		$filterOut = [];

		if ($this->filter->count() > 0)
		{
			foreach ($this->filter as $key => $value)
			{
				if (!\in_array($key, ['FILTER_ID', 'PRESET_ID', 'FILTER_APPLIED', 'FIND', 'tabId']))
				{
					$filterOut[$key] = $value;
				}
			}
		}

		$filterOut['PATH'] = Translate\IO\Path::replaceLangId($path, '#LANG_ID#');

		return $filterOut;
	}
}