ProductsCount.php
50 lines
| 1 | <?php |
| 2 | |
| 3 | namespace WCML\Rest\Wrapper\Reports; |
| 4 | |
| 5 | use WCML\Rest\Exceptions\InvalidLanguage; |
| 6 | use WCML\Rest\Wrapper\Handler; |
| 7 | use WPML\FP\Obj; |
| 8 | |
| 9 | class ProductsCount extends Handler { |
| 10 | |
| 11 | private $sitepress; |
| 12 | private $wpdb; |
| 13 | |
| 14 | public function __construct( \SitePress $sitepress, \wpdb $wpdb ) { |
| 15 | $this->sitepress = $sitepress; |
| 16 | $this->wpdb = $wpdb; |
| 17 | } |
| 18 | |
| 19 | public function prepare( $response, $object, $request ) { |
| 20 | |
| 21 | $language = Obj::prop( 'lang', $request->get_params() ); |
| 22 | |
| 23 | if ( $language ) { |
| 24 | |
| 25 | if ( ! $this->sitepress->is_active_language( $language ) ) { |
| 26 | throw new InvalidLanguage( $language ); |
| 27 | } |
| 28 | |
| 29 | $term = get_term_by( 'slug', $object->slug, 'product_type' ); |
| 30 | |
| 31 | $count = $this->wpdb->get_var( |
| 32 | $this->wpdb->prepare( |
| 33 | "SELECT count( p.ID ) FROM {$this->wpdb->posts} as p |
| 34 | LEFT JOIN {$this->wpdb->term_relationships} as tr ON p.ID = tr.object_id |
| 35 | LEFT JOIN {$this->wpdb->prefix}icl_translations as icl ON p.ID = icl.element_id |
| 36 | WHERE tr.term_taxonomy_id = %d AND icl.language_code = %s AND icl.element_type = 'post_product'", |
| 37 | $term->term_taxonomy_id, $language |
| 38 | ) |
| 39 | ); |
| 40 | |
| 41 | $data = \WPML\FP\Obj::assoc( 'total', $count, $response->get_data() ); |
| 42 | $data = \WPML\FP\Obj::assoc( 'lang', $language, $data ); |
| 43 | |
| 44 | $response->set_data( $data ); |
| 45 | } |
| 46 | |
| 47 | return $response; |
| 48 | } |
| 49 | |
| 50 | } |