class-wpml-tp-apply-translation.php
3.5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
<?php
class WPML_TP_Apply_Translations {
/** @var WPML_TM_Jobs_Repository */
private $jobs_repository;
/** @var WPML_TP_Apply_Single_Job */
private $apply_single_job;
/** @var WPML_TP_Sync_Jobs */
private $tp_sync;
/**
* @param WPML_TM_Jobs_Repository $jobs_repository
* @param WPML_TP_Apply_Single_Job $apply_single_job
* @param WPML_TP_Sync_Jobs $tp_sync
*/
public function __construct(
WPML_TM_Jobs_Repository $jobs_repository,
WPML_TP_Apply_Single_Job $apply_single_job,
WPML_TP_Sync_Jobs $tp_sync
) {
$this->jobs_repository = $jobs_repository;
$this->apply_single_job = $apply_single_job;
$this->tp_sync = $tp_sync;
}
/**
* @param array $params
*
* @return WPML_TM_Jobs_Collection
* @throws WPML_TP_API_Exception
*/
public function apply( array $params ) {
$jobs = $this->get_jobs( $params );
if ( $this->has_in_progress_jobs( $jobs ) ) {
$jobs = $this->sync_jobs( $jobs );
}
$cancelled_jobs = $jobs->filter_by_status( ICL_TM_NOT_TRANSLATED );
$downloaded_jobs = new WPML_TM_Jobs_Collection(
$jobs->filter_by_status( [ ICL_TM_TRANSLATION_READY_TO_DOWNLOAD, ICL_TM_COMPLETE ] )
->map( array( $this->apply_single_job, 'apply' ) )
);
return $downloaded_jobs->append( $cancelled_jobs );
}
/**
* @param WPML_TM_Jobs_Collection $jobs
*
* @return bool
*/
private function has_in_progress_jobs( WPML_TM_Jobs_Collection $jobs ) {
return count( $jobs->filter_by_status( ICL_TM_IN_PROGRESS ) ) > 0;
}
/**
* @param array $params
*
* @return WPML_TM_Jobs_Collection
*/
private function get_jobs( array $params ) {
if ( $params ) {
if ( isset( $params['original_element_id'], $params['element_type'] ) ) {
$jobs = $this->get_jobs_by_original_element( $params['original_element_id'], $params['element_type'] );
} else {
$jobs = $this->get_jobs_by_ids( $params );
}
} else {
$jobs = $this->get_all_ready_jobs();
}
return $jobs;
}
/**
* @param int $original_element_id
* @param string $element_type
*
* @return WPML_TM_Jobs_Collection
*/
private function get_jobs_by_original_element( $original_element_id, $element_type ) {
$params = new WPML_TM_Jobs_Search_Params();
$params->set_scope( WPML_TM_Jobs_Search_Params::SCOPE_REMOTE );
$params->set_original_element_id( $original_element_id );
$params->set_job_types( $element_type );
return $this->jobs_repository->get_collection( $params );
}
/**
* @param array $params
*
* @return WPML_TM_Jobs_Collection
*/
private function get_jobs_by_ids( array $params ) {
$jobs = array();
foreach ( $params as $param ) {
$jobs[] = $this->jobs_repository->get_job( $param['id'], $param['type'] );
}
return new WPML_TM_Jobs_Collection( $jobs );
}
/**
* @return WPML_TM_Jobs_Collection
*/
private function get_all_ready_jobs() {
return $this->jobs_repository->get_collection(
new WPML_TM_Jobs_Search_Params(
array(
'status' => array( ICL_TM_TRANSLATION_READY_TO_DOWNLOAD ),
'scope' => WPML_TM_Jobs_Search_Params::SCOPE_REMOTE,
)
)
);
}
/**
* @param WPML_TM_Jobs_Collection $jobs
*
* @return WPML_TM_Jobs_Collection
* @throws WPML_TP_API_Exception
*/
private function sync_jobs( WPML_TM_Jobs_Collection $jobs ) {
$synced_jobs = $this->tp_sync->sync();
/** @var WPML_TM_Job_Entity $job */
foreach ( $jobs as $job ) {
foreach ( $synced_jobs as $synced_job ) {
if ( $job->is_equal( $synced_job ) ) {
$job->set_status( $synced_job->get_status() );
break;
}
}
}
return $jobs;
}
}