first commit

wp-content/plugins/sitepress-multilingual-cms/classes/emails/report/class-wpml-tm-batch-report.php

@@ -0,0 +1,85 @@
+<?php
+
+/**
+ * Class WPML_TM_Batch_Report
+ */
+class WPML_TM_Batch_Report {
+	
+	const BATCH_REPORT_OPTION = '_wpml_batch_report';
+
+	/**
+	 * @var WPML_TM_Blog_Translators
+	 */
+	private $blog_translators;
+
+	/**
+	 * WPML_TM_Batch_Report constructor.
+	 *
+	 * @param WPML_TM_Blog_Translators $blog_translators
+	 */
+	public function __construct( WPML_TM_Blog_Translators $blog_translators) {
+		$this->blog_translators = $blog_translators;
+	}
+
+	/**
+	 * @param WPML_Translation_Job $job
+	 */
+	public function set_job( WPML_Translation_Job $job ) {
+		$batch_jobs = $batch_jobs_raw = $this->get_jobs();
+		$job_fields = $job->get_basic_data();
+
+		if ( WPML_User_Jobs_Notification_Settings::is_new_job_notification_enabled( $job_fields->translator_id ) ) {
+			$lang_pair = $job_fields->source_language_code . '|' . $job_fields->language_code;
+			$batch_jobs[ (int) $job_fields->translator_id ][$lang_pair][] = array(
+				'element_id' => isset( $job_fields->original_doc_id ) ? $job_fields->original_doc_id : null,
+				'type'       => strtolower( $job->get_type() ),
+				'job_id'     => $job->get_id(),
+			);
+		}
+
+		if ( $batch_jobs_raw !== $batch_jobs ) {
+			update_option( self::BATCH_REPORT_OPTION, $batch_jobs, 'no' );
+		}
+	}
+
+	/**
+	 * @return array
+	 */
+	public function get_unassigned_jobs() {
+		$batch_jobs = $this->get_jobs();
+		$unassigned_jobs = array();
+
+		if( array_key_exists( 0, $batch_jobs ) ) {
+			$unassigned_jobs = $batch_jobs[0];
+		}
+
+		return $unassigned_jobs;
+	}
+
+	/**
+	 * @return array
+	 */
+	public function get_unassigned_translators() {
+		$assigned_translators = array_keys( $this->get_jobs() );
+		$blog_translators = wp_list_pluck( $this->blog_translators->get_blog_translators() , 'ID');
+
+		return array_diff( $blog_translators, $assigned_translators );
+	}
+
+
+	/**
+	 * @return array
+	 */
+	public function get_jobs() {
+		return get_option( self::BATCH_REPORT_OPTION ) ? get_option( self::BATCH_REPORT_OPTION ) : array();
+	}
+
+	public function reset_batch_report( $translator_id ) {
+		$batch_jobs = $this->get_jobs();
+		if ( array_key_exists( $translator_id, $batch_jobs ) ) {
+			unset( $batch_jobs[$translator_id] );
+		}
+
+		update_option( self::BATCH_REPORT_OPTION, $batch_jobs, 'no' );
+	}
+}