<?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();
	}

}