Support Grouped Products for Gravity Forms Product Addons

ES_GravityForms_Grouped_Product_Add_To_Cart_Handler.php

class ES_GravityForms_Grouped_Product_Add_To_Cart_Handler {

	private static $instance;

	public static function register() {
		if ( self::$instance == null ) {
			self::$instance = new ES_GravityForms_Grouped_Product_Add_To_Cart_Handler();
		}
	}

	protected $adding_to_cart;
	protected $is_grouped_product;

	public function __construct() {
		add_filter( 'woocommerce_add_to_cart_product_id', [ $this, 'on_woocommerce_add_to_cart_product_id' ] );
	}

	public function on_woocommerce_add_to_cart_product_id( $product_id ) {
		$adding_to_cart = wc_get_product( $product_id );
		if ( ! $adding_to_cart ) {
			return $adding_to_cart;
		}

		$add_to_cart_handler = apply_filters( 'woocommerce_add_to_cart_handler', $adding_to_cart->get_type(), $adding_to_cart );
		if ( 'grouped' === $add_to_cart_handler ) {
			$this->adding_to_cart = $product_id;
			add_filter( 'woocommerce_gforms_get_product_form_data', [ $this, 'get_product_form_data' ], 10, 3 );
		}

		return $adding_to_cart;
	}

	public function get_product_form_data( $data, $product_id, $context ) {

		if ( empty( $this->adding_to_cart ) || $product_id == $this->adding_to_cart ) {
			return $data;
		}

		return WC_GFPA_Main::instance()->get_gravity_form_data( $this->adding_to_cart );
	}
}


ES_GravityForms_Grouped_Product_Add_To_Cart_Handler::register();