first commit

wp-content/plugins/wpml-translation-management/classes/menu/dashboard/PostJobsRepository.php

@@ -0,0 +1,88 @@
+<?php
+
+namespace WPML\TM\Menu\Dashboard;
+
+class PostJobsRepository {
+
+	/**
+	 * @param int    $original_element_id
+	 * @param string $element_type
+	 *
+	 * @return array
+	 */
+	public function getJobsGroupedByLang( $original_element_id, $element_type ) {
+		return $this->getJobsFor( $original_element_id, $element_type )
+		            ->map( [ $this, 'mapJob' ] )
+		            ->keyBy( 'targetLanguage' )
+		            ->toArray();
+	}
+
+	/**
+	 * @param int    $original_element_id
+	 * @param string $element_type
+	 *
+	 * @return \WPML\Collect\Support\Collection
+	 */
+	private function getJobsFor( $original_element_id, $element_type ) {
+		return \wpml_collect(
+			wpml_tm_get_jobs_repository()->get( $this->buildSearchParams( $original_element_id, $element_type ) )
+		);
+	}
+
+	/**
+	 * @param int    $original_element_id
+	 * @param string $element_type
+	 *
+	 * @return \WPML_TM_Jobs_Search_Params
+	 */
+	private function buildSearchParams( $original_element_id, $element_type ) {
+		$params = new \WPML_TM_Jobs_Search_Params();
+		$params->set_original_element_id( $original_element_id );
+		$params->set_job_types( $element_type );
+
+		return $params;
+	}
+
+	/**
+	 * @param \WPML_TM_Post_Job_Entity $job
+	 *
+	 * @return array
+	 */
+	public function mapJob( \WPML_TM_Post_Job_Entity $job ) {
+		return [
+			'entity_id'      => $job->get_id(),
+			'job_id'         => $job->get_translate_job_id(),
+			'type'           => $job->get_type(),
+			'status'         => $this->getJobStatus( $job ),
+			'targetLanguage' => $job->get_target_language(),
+			'isLocal'        => 'local' === $job->get_translation_service(),
+		];
+	}
+
+	/**
+	 * @param \WPML_TM_Job_Entity $job
+	 *
+	 * @return int
+	 */
+	private function getJobStatus( \WPML_TM_Job_Entity $job ) {
+		if ( $job->does_need_update() ) {
+			return ICL_TM_NEEDS_UPDATE;
+		}
+
+		if ( $this->postHasTranslationButLatestJobCancelled( $job ) ) {
+			return ICL_TM_COMPLETE;
+		}
+
+		return $job->get_status();
+	}
+
+	/**
+	 * @param \WPML_TM_Job_Entity $job
+	 *
+	 * @return bool
+	 */
+	private function postHasTranslationButLatestJobCancelled( \WPML_TM_Job_Entity $job ) {
+		return $job->get_status() === ICL_TM_NOT_TRANSLATED && $job->has_completed_translation();
+	}
+
+}