first commit

wp-content/plugins/wpml-translation-management/classes/ATE/REST/Download.php

@@ -0,0 +1,74 @@
+<?php
+/**
+ * @author OnTheGo Systems
+ */
+
+namespace WPML\TM\ATE\REST;
+
+use WP_REST_Request;
+use WPML\Collect\Support\Collection;
+use function WPML\Container\make;
+use WPML\TM\ATE\Download\Job;
+use WPML\TM\ATE\Download\Process;
+use WPML\TM\Jobs\Utils\ElementLinkFactory;
+use WPML\TM\REST\Base;
+use WPML_TM_ATE_AMS_Endpoints;
+
+class Download extends Base {
+
+	const PROCESS_QUANTITY = 5;
+
+	/**
+	 * @return array
+	 */
+	public function get_routes() {
+		return [
+			[
+				'route' => WPML_TM_ATE_AMS_Endpoints::DOWNLOAD_JOBS,
+				'args'  => [
+					'methods'  => 'POST',
+					'callback' => [ $this, 'download' ],
+				],
+			],
+		];
+	}
+
+	/**
+	 * @param WP_REST_Request $request
+	 *
+	 * @return array
+	 */
+	public function get_allowed_capabilities( WP_REST_Request $request ) {
+		return [
+			'manage_options',
+			'manage_translations',
+			'translate',
+		];
+	}
+
+	public function download() {
+		$result = make( Process::class )->run( self::PROCESS_QUANTITY );
+
+		return [
+			'jobs'              => $this->getJobs( $result->processedJobs ),
+			'downloadQueueSize' => $result->downloadQueueSize,
+		];
+	}
+
+	private function getJobs( Collection $processedJobs ) {
+		$jobIds    = $processedJobs->pluck( 'wpmlJobId' );
+		$viewLinks = $jobIds->map( [ wpml_tm_load_job_factory(), 'get_translation_job' ] )
+							->map( [ ElementLinkFactory::create(), 'getTranslation' ] );
+
+		return $jobIds->zip( $viewLinks )
+					->map(
+						function ( $pair ) {
+							return [
+								'jobId'    => (int) $pair[0],
+								'viewLink' => $pair[1],
+							];
+						}
+					)
+					  ->toArray();
+	}
+}