SuspendWpmlFilters.php
50 lines
| 1 | <?php |
| 2 | |
| 3 | namespace WCML\Terms; |
| 4 | |
| 5 | use WCML\Utilities\Suspend\Filters as SuspendFilters; |
| 6 | use WCML\Utilities\Suspend\Suspend; |
| 7 | use WPML\Collect\Support\Collection; |
| 8 | use function WPML\FP\spreadArgs; |
| 9 | |
| 10 | class SuspendWpmlFilters implements Suspend { |
| 11 | |
| 12 | private $suspendFilters; |
| 13 | |
| 14 | private $filterArgs; |
| 15 | |
| 16 | public function __construct( SuspendFilters $suspendFilters ) { |
| 17 | $this->suspendFilters = $suspendFilters; |
| 18 | $this->getTaxonomyChildrenOptionFiltersArgs()->each( spreadArgs( 'add_filter' ) ); |
| 19 | } |
| 20 | |
| 21 | private function getTaxonomyChildrenOptionFiltersArgs() { |
| 22 | $this->filterArgs = $this->filterArgs ?: wpml_collect( get_taxonomies() ) |
| 23 | ->filter( 'is_taxonomy_translated' ) |
| 24 | ->map( function( $taxonomy ) { |
| 25 | return [ "pre_option_{$taxonomy}_children", self::getTaxonomyChildrenInAllLanguages( $taxonomy ), 20 ]; |
| 26 | } ); |
| 27 | |
| 28 | return $this->filterArgs; |
| 29 | } |
| 30 | |
| 31 | private static function getTaxonomyChildrenInAllLanguages( $taxonomy ) { |
| 32 | return function() use ( $taxonomy ) { |
| 33 | return get_option( "{$taxonomy}_children_all", false ); |
| 34 | }; |
| 35 | } |
| 36 | |
| 37 | public function resume() { |
| 38 | $this->suspendFilters->resume(); |
| 39 | $this->getTaxonomyChildrenOptionFiltersArgs()->each( spreadArgs( 'remove_filter' ) ); |
| 40 | } |
| 41 | |
| 42 | public function runAndResume( callable $function ) { |
| 43 | $result = $function(); |
| 44 | |
| 45 | $this->resume(); |
| 46 | |
| 47 | return $result; |
| 48 | } |
| 49 | } |
| 50 |