first commit

wp-content/plugins/wpml-string-translation/classes/MO/Generate/Process/SingleSiteProcess.php

@@ -0,0 +1,108 @@
+<?php
+
+namespace WPML\ST\MO\Generate\Process;
+
+
+use WPML\ST\MO\File\Manager;
+use WPML\ST\MO\Generate\DomainsAndLanguagesRepository;
+use WPML\Utils\Pager;
+
+class SingleSiteProcess implements Process {
+
+	CONST TIMEOUT = 5;
+
+	/** @var DomainsAndLanguagesRepository */
+	private $domainsAndLanguagesRepository;
+
+	/** @var Manager */
+	private $manager;
+
+	/** @var Status */
+	private $status;
+
+	/** @var Pager */
+	private $pager;
+
+	/** @var callable */
+	private $migrateAdminTexts;
+
+	/**
+	 * @param DomainsAndLanguagesRepository $domainsAndLanguagesRepository
+	 * @param Manager                       $manager
+	 * @param Status                        $status
+	 * @param Pager                         $pager
+	 * @param callable                      $migrateAdminTexts
+	 */
+	public function __construct(
+		DomainsAndLanguagesRepository $domainsAndLanguagesRepository,
+		Manager $manager,
+		Status $status,
+		Pager $pager,
+		callable $migrateAdminTexts
+	) {
+		$this->domainsAndLanguagesRepository = $domainsAndLanguagesRepository;
+		$this->manager                       = $manager;
+		$this->status                        = $status;
+		$this->pager                         = $pager;
+		$this->migrateAdminTexts             = $migrateAdminTexts;
+	}
+
+
+	public function runAll() {
+		call_user_func( $this->migrateAdminTexts );
+		$this->getDomainsAndLanguages()->each( function ( $row ) {
+			$this->manager->add( $row->domain, $row->locale );
+		} );
+
+		$this->status->markComplete();
+	}
+
+	/**
+	 * @return int Remaining
+	 */
+	public function runPage() {
+		if ( $this->pager->getProcessedCount() === 0 ) {
+			call_user_func( $this->migrateAdminTexts );
+		}
+
+		$domains   = $this->getDomainsAndLanguages();;
+		$remaining = $this->pager->iterate( $domains, function ( $row ) {
+			$this->manager->add( $row->domain, $row->locale );
+
+			return true;
+		}, self::TIMEOUT );
+
+		if ( $remaining === 0 ) {
+			$this->status->markComplete();
+		}
+
+		return $remaining;
+	}
+
+	public function getPagesCount() {
+		if ( $this->status->isComplete() ) {
+			return 0;
+		}
+
+		$domains = $this->getDomainsAndLanguages();
+
+		if ( $domains->count() === 0 ) {
+			$this->status->markComplete();
+		}
+
+		return $domains->count();
+	}
+
+	private function getDomainsAndLanguages() {
+		return DomainsAndLanguagesRepository::hasTranslationFilesTable()
+			? $this->domainsAndLanguagesRepository->get()
+			: wpml_collect();
+	}
+
+	/**
+	 * @return bool
+	 */
+	public function isCompleted() {
+		return $this->getPagesCount() === 0;
+	}
+}