Load product types

includes/api/class-wc-rest-report-products-totals-controller.php

@@ -0,0 +1,127 @@
+<?php
+/**
+ * REST API Reports Products Totals controller
+ *
+ * Handles requests to the /reports/products/count endpoint.
+ *
+ * @package WooCommerce/API
+ * @since   3.5.0
+ */
+
+defined( 'ABSPATH' ) || exit;
+
+/**
+ * REST API Reports Products Totals controller class.
+ *
+ * @package WooCommerce/API
+ * @extends WC_REST_Reports_Controller
+ */
+class WC_REST_Report_Products_Totals_Controller extends WC_REST_Reports_Controller {
+
+	/**
+	 * Endpoint namespace.
+	 *
+	 * @var string
+	 */
+	protected $namespace = 'wc/v3';
+
+	/**
+	 * Route base.
+	 *
+	 * @var string
+	 */
+	protected $rest_base = 'reports/products/totals';
+
+	/**
+	 * Get reports list.
+	 *
+	 * @since 3.5.0
+	 * @return array
+	 */
+	protected function get_reports() {
+		$types = wc_get_product_types();
+		$data  = array();
+
+		foreach ( get_terms( 'product_type' ) as $product_type ) {
+			if ( ! isset( $types[ $product_type->name ] ) ) {
+				continue;
+			}
+
+			$data[] = array(
+				'slug'  => $product_type->name,
+				'name'  => $types[ $product_type->name ],
+				'total' => (int) $product_type->count,
+			);
+		}
+
+		return $data;
+	}
+
+	/**
+	 * Prepare a report object for serialization.
+	 *
+	 * @param  stdClass        $report Report data.
+	 * @param  WP_REST_Request $request Request object.
+	 * @return WP_REST_Response $response Response data.
+	 */
+	public function prepare_item_for_response( $report, $request ) {
+		$data = array(
+			'slug'  => $report->slug,
+			'name'  => $report->name,
+			'total' => $report->total,
+		);
+
+		$context = ! empty( $request['context'] ) ? $request['context'] : 'view';
+		$data    = $this->add_additional_fields_to_object( $data, $request );
+		$data    = $this->filter_response_by_context( $data, $context );
+
+		// Wrap the data in a response object.
+		$response = rest_ensure_response( $data );
+
+		/**
+		 * Filter a report returned from the API.
+		 *
+		 * Allows modification of the report data right before it is returned.
+		 *
+		 * @param WP_REST_Response $response The response object.
+		 * @param object           $report   The original report object.
+		 * @param WP_REST_Request  $request  Request used to generate the response.
+		 */
+		return apply_filters( 'woocommerce_rest_prepare_report_products_count', $response, $report, $request );
+	}
+
+	/**
+	 * Get the Report's schema, conforming to JSON Schema.
+	 *
+	 * @return array
+	 */
+	public function get_item_schema() {
+		$schema = array(
+			'$schema'    => 'http://json-schema.org/draft-04/schema#',
+			'title'      => 'report_product_total',
+			'type'       => 'object',
+			'properties' => array(
+				'slug'  => array(
+					'description' => __( 'An alphanumeric identifier for the resource.', 'woocommerce' ),
+					'type'        => 'string',
+					'context'     => array( 'view' ),
+					'readonly'    => true,
+				),
+				'name'  => array(
+					'description' => __( 'Product type name.', 'woocommerce' ),
+					'type'        => 'string',
+					'context'     => array( 'view' ),
+					'readonly'    => true,
+				),
+				'total' => array(
+					'description' => __( 'Amount of products.', 'woocommerce' ),
+					'type'        => 'string',
+					'context'     => array( 'view' ),
+					'readonly'    => true,
+				),
+			),
+		);
+
+		return $this->add_additional_fields_schema( $schema );
+	}
+}