<?php
/**
 * Booster for WooCommerce - Module - Shipping by User Role
 *
 * @version 7.1.6
 * @since  1.0.0
 * @author  Pluggabl LLC.
 * @package Booster_Plus_For_WooCommerce/includes
 */

if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly.
}

if ( ! class_exists( 'WCJ_Shipping_By_User_Role' ) ) :
		/**
		 * WCJ_Shipping_By_User_Role.
		 *
		 * @version 7.1.6
		 * @since   1.0.0
		 */
	class WCJ_Shipping_By_User_Role extends WCJ_Module_Shipping_By_Condition {

		/**
		 * The module user_id
		 *
		 * @var varchar $user_id Module user_id.
		 */
		public $user_id;

		/**
		 * The module customer_roles
		 *
		 * @var varchar $customer_roles Module customer_roles.
		 */
		public $customer_roles;

		/**
		 * Constructor.
		 *
		 * @version 6.0.0
		 * @since  1.0.0
		 */
		public function __construct() {

			$this->id         = 'shipping_by_user_role';
			$this->short_desc = __( 'Shipping Methods by Users', 'woocommerce-jetpack' );
			$this->desc       = __( 'Set user roles, users or membership plans to include/exclude for shipping methods to show up (Free shipping available in Plus).', 'woocommerce-jetpack' );
			$this->desc_pro   = __( 'Set user roles, users or membership plans to include/exclude for shipping methods to show up.', 'woocommerce-jetpack' );
			$this->link_slug  = 'woocommerce-shipping-methods-by-users';

			$this->condition_options = array(
				'user_roles'      => array(
					'title' => __( 'User Roles', 'woocommerce-jetpack' ),
					'desc'  => sprintf(
						/* translators: %s: translators Added */
						__( 'Custom roles can be added via "Add/Manage Custom Roles" tool in Booster\'s <a href="%s">General</a> module.', 'woocommerce-jetpack' ),
						admin_url( wcj_admin_tab_url() . '&wcj-cat=emails_and_misc&section=general' )
					),
				),
				'user_id'         => array(
					'title' => __( 'Users', 'woocommerce-jetpack' ),
					'desc'  => '',
				),
				'user_membership' => array(
					'title' => __( 'User Membership Plans', 'woocommerce-jetpack' ),
					'desc'  => sprintf(
						/* translators: %s: translators Added */
						__( 'This section requires <a target="_blank" href="%s">WooCommerce Memberships</a> plugin.', 'woocommerce-jetpack' ),
						'https://woocommerce.com/products/woocommerce-memberships/'
					),
				),
			);

			parent::__construct();

		}

		/**
		 * Check_multiple_roles.
		 *
		 * @version 6.0.0
		 * @since   1.0.0
		 *
		 * @return bool
		 */
		public function add_multiple_roles_option() {
			return true;
		}

		/**
		 * Check.
		 *
		 * @version 7.0.0
		 * @since  1.0.0
		 * @todo    use `$package` (and in this case update `wcj_get_left_to_free_shipping()`)
		 * @param string         $options_id defines the options_id.
		 * @param string         $user_roles_or_ids_or_membership_plans defines the user_roles_or_ids_or_membership_plans.
		 * @param string         $include_or_exclude defines the include_or_exclude.
		 * @param string | array $package defines the package.
		 */
		public function check( $options_id, $user_roles_or_ids_or_membership_plans, $include_or_exclude, $package ) {
			switch ( $options_id ) {
				case 'user_roles':
					if ( empty( $this->customer_roles ) ) {
						$multi_role_check     = wcj_get_option( 'wcj_' . $this->id . '_check_multiple_roles', 'no' );
						$this->customer_roles = 'no' === ( $multi_role_check ) ? array( wcj_get_current_user_first_role() ) : wcj_get_current_user_all_roles();
					}
					return count( array_intersect( $this->customer_roles, $user_roles_or_ids_or_membership_plans ) ) > 0;
				case 'user_id':
					if ( ! isset( $this->user_id ) ) {
						$this->user_id = get_current_user_id();
					}
					return in_array( (string) $this->user_id, $user_roles_or_ids_or_membership_plans, true );
				case 'user_membership':
					if ( ! isset( $this->user_id ) ) {
						$this->user_id = get_current_user_id();
					}
					if ( ! function_exists( 'wc_memberships_is_user_active_member' ) ) {
						return false;
					}
					foreach ( $user_roles_or_ids_or_membership_plans as $membership_plan ) {
						if ( wc_memberships_is_user_active_member( $this->user_id, $membership_plan ) ) {
							return true;
						}
					}
					return false;
			}
		}

		/**
		 * Get_condition_options.
		 *
		 * @version 6.0.0
		 * @since   1.0.0
		 * @param string $options_id defines the options_id.
		 */
		public function get_condition_options( $options_id ) {
			switch ( $options_id ) {
				case 'user_roles':
					return wcj_get_user_roles_options();
				case 'user_membership':
					$membership_plans = array();
					$block_size       = 512;
					$offset           = 0;
					while ( true ) {
						$args = array(
							'post_type'      => 'wc_membership_plan',
							'post_status'    => 'any',
							'posts_per_page' => $block_size,
							'offset'         => $offset,
							'orderby'        => 'title',
							'order'          => 'ASC',
							'fields'         => 'ids',
						);
						$loop = new WP_Query( $args );
						if ( ! $loop->have_posts() ) {
							break;
						}
						foreach ( $loop->posts as $post_id ) {
							$membership_plans[ $post_id ] = get_the_title( $post_id );
						}
						$offset += $block_size;
					}
					return $membership_plans;
			}
		}

	}

endif;

return new WCJ_Shipping_By_User_Role();
