From 124a4291b0c455553d979d5099512abcbeb99da0 Mon Sep 17 00:00:00 2001
From: Claudio Sanches <contato@claudiosmweb.com>
Date: Tue, 29 Mar 2016 22:06:05 -0300
Subject: [PATCH] Created new method to save product taxonomy terms

---
 .../api/class-wc-rest-products-controller.php | 29 ++++++++++++++-----
 1 file changed, 22 insertions(+), 7 deletions(-)

diff --git a/includes/api/class-wc-rest-products-controller.php b/includes/api/class-wc-rest-products-controller.php
index bdb8d723d8e..e5835b9f053 100644
--- a/includes/api/class-wc-rest-products-controller.php
+++ b/includes/api/class-wc-rest-products-controller.php
@@ -746,6 +746,23 @@ class WC_REST_Products_Controller extends WC_REST_Posts_Controller {
 		update_post_meta( $id, '_downloadable_files', $files );
 	}
 
+	/**
+	 * Save taxonomy terms.
+	 *
+	 * @param WC_Product $product
+	 * @param array $terms
+	 * @param string $taxonomy
+	 * @return array
+	 */
+	protected function save_taxonomy_terms( $product, $terms, $taxonomy = 'cat' ) {
+		$term_ids = wp_list_pluck( $terms, 'id' );
+		$term_ids = array_unique( array_map( 'intval', $term_ids ) );
+
+		wp_set_object_terms( $product->id, $term_ids, 'product_' . $taxonomy );
+
+		return $terms;
+	}
+
 	/**
 	 * Save product meta.
 	 *
@@ -1117,14 +1134,12 @@ class WC_REST_Products_Controller extends WC_REST_Posts_Controller {
 
 		// Product categories.
 		if ( isset( $request['categories'] ) && is_array( $request['categories'] ) ) {
-			$term_ids = array_unique( array_map( 'intval', $request['categories'] ) );
-			wp_set_object_terms( $product->id, $term_ids, 'product_cat' );
+			$this->save_taxonomy_terms( $product, $request['categories'] );
 		}
 
 		// Product tags.
 		if ( isset( $request['tags'] ) && is_array( $request['tags'] ) ) {
-			$term_ids = array_unique( array_map( 'intval', $request['tags'] ) );
-			wp_set_object_terms( $product->id, $term_ids, 'product_tag' );
+			$this->save_taxonomy_terms( $product, $request['tags'], 'tag' );
 		}
 
 		// Downloadable.
@@ -1181,7 +1196,7 @@ class WC_REST_Products_Controller extends WC_REST_Posts_Controller {
 	 * @return bool
 	 * @throws WC_REST_Exception
 	 */
-	protected function save_variations( $product, $request ) {
+	protected function save_variations_data( $product, $request ) {
 		global $wpdb;
 
 		$variations = $request['variations'];
@@ -1526,7 +1541,7 @@ class WC_REST_Products_Controller extends WC_REST_Posts_Controller {
 
 			// Save variations.
 			if ( isset( $request['type'] ) && 'variable' == $request['type'] && isset( $request['variations'] ) && is_array( $request['variations'] ) ) {
-				$this->save_variations( $product, $request );
+				$this->save_variations_data( $product, $request );
 			}
 
 			return true;
@@ -1557,7 +1572,7 @@ class WC_REST_Products_Controller extends WC_REST_Posts_Controller {
 			// Save variations.
 			if ( $product->is_type( 'variable' ) ) {
 				if ( isset( $request['variations'] ) && is_array( $request['variations'] ) ) {
-					$this->save_variations( $product, $request );
+					$this->save_variations_data( $product, $request );
 				} else {
 					// Just sync variations.
 					WC_Product_Variable::sync( $product->id );