add images to product variation

BeAPI · Nov 20, 2020 · 28599bf · 28599bf
1 parent 438c2c8
commit 28599bf
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/Feed/Generator.php b/src/Feed/Generator.php
@@ -152,7 +152,8 @@ function (
 					}
 				}
 
-				if ( ! empty( $sf_product->get_images() ) ) {
+				$images = $sf_product->get_images();
+				if ( ! empty( $images ) ) {
 					$product->setAdditionalImages( $sf_product->get_images() );
 				}
 
@@ -201,6 +202,10 @@ function (
 						$variation
 							->setMainImage( $sf_product_variation['image_main'] );
 					}
+					$variation_images = $sf_product->get_variation_images();
+					if ( ! empty( $variation_images ) ) {
+						$variation->setAdditionalImages( $variation_images );
+					}
 				}
 			}
 		);

diff --git a/src/Products/Product.php b/src/Products/Product.php
@@ -278,6 +278,13 @@ function ( $img ) {
 		);
 	}
 
+	/**
+	 * @return array
+	 */
+	public function get_variation_images() {
+		return apply_filters( 'shopping_feed_variation_images', [], $this->product );
+	}
+
 	public function has_variations() {
 		return ! empty( $this->get_variations() );
 	}
@@ -358,6 +365,6 @@ public function get_variation_attributes( $variation ) {
 	 * Field : ['name'=>'', 'value'=>'']
 	 */
 	public function get_extra_fields() {
-		return apply_filters( 'shopping_feed_extra_variation_attributes', [], $this->product );
+		return apply_filters( 'shopping_feed_extra_fields', [], $this->product );
 	}
 }