wc-ebs-plugin.php

<?php

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

class WC_EBS {

    public function __construct() {

        // Get plugin options values
        $this->options = get_option('wc_ebs_options');

        if ( !is_admin() )
            add_action( 'wp_enqueue_scripts', array( $this, 'wc_ebs_enqueue_scripts' ));

        add_action( 'product_type_options', array( $this, 'wc_ebs_add_product_option_pricing' ));
        add_filter( 'woocommerce_process_product_meta', array( $this, 'wc_ebs_add_custom_price_fields_save' ));
        add_action( 'woocommerce_before_add_to_cart_button', array( $this, 'wc_ebs_before_add_to_cart_button' ));
        add_filter( 'woocommerce_get_price_html', array( $this, 'wc_ebs_add_price_html' ), 10, 2 );
        add_action( 'wp_ajax_add_new_price', array( $this, 'wc_ebs_get_new_price' ));
        add_action( 'wp_ajax_nopriv_add_new_price', array( $this, 'wc_ebs_get_new_price' ));
        add_action( 'wp_ajax_woocommerce_get_refreshed_fragments', array( $this, 'wc_ebs_new_price_fragment' ));
        add_action( 'wp_ajax_nopriv_woocommerce_get_refreshed_fragments',  array( $this, 'wc_ebs_new_price_fragment' ));
        add_filter( 'woocommerce_loop_add_to_cart_link', array($this, 'wc_ebs_custom_loop_add_to_cart' ), 10, 2 );
    }

    public function wc_ebs_enqueue_scripts() {
        global $woocommerce, $post;

        // Get page language in order to load Pickadate translation
        $site_language = get_bloginfo( 'language' );
        $lang = str_replace("-","_", $site_language);
        
        // Load scripts only on product page if "booking" option is checked
        $wc_ebs_options = get_post_meta($post->ID, '_booking_option', true);
        // Calculation mode (Days or Nights)
        $calc_mode = $this->options['wc_ebs_calc_mode'];

        if ( is_product() && $wc_ebs_options ) {

            // Concatenated and minified script including datepick.js, legacy.js, picker.js and picker.date.js
            wp_enqueue_script( 'datepicker', plugins_url( '/js/pickadate.min.js', __FILE__ ), array('jquery'), '1.0', true);

            // wp_enqueue_script( 'picker', plugins_url( '/js/picker.js', __FILE__ ), array('jquery'), '1.0', true);
            // wp_enqueue_script( 'picker.date', plugins_url( '/js/picker.date.js', __FILE__ ), array('jquery'), '1.0', true);
            // wp_enqueue_script( 'legacy', plugins_url( '/js/legacy.js', __FILE__ ), array('jquery'), '1.0', true);
            // wp_enqueue_script( 'datepick', plugins_url( '/js/datepick.js', __FILE__ ), array('jquery'), '1.0', true);

            wp_enqueue_script( 'datepicker.language', plugins_url( '/js/translations/' . $lang . '.js', __FILE__ ), array('jquery'), '1.0', true);

            wp_register_style( 'picker', plugins_url('/css/default.min.css', __FILE__), true);
            wp_enqueue_style( 'picker' );

            // in javascript, object properties are accessed as ajax_object.ajax_url, ajax_object.we_value
            wp_localize_script( 'datepicker', 'ajax_object',
                array( 
                    'ajax_url' => admin_url( 'admin-ajax.php' ),
                    'calc_mode' => $calc_mode
                )
            );
        }
    }

    // Add checkbox to the product admin page
    public function wc_ebs_add_product_option_pricing( $product_type_options ) {
        global $woocommerce, $post;

        $product_type_options['booking_option'] = array(
            'id'            => '_booking_option',
            'wrapper_class' => 'show_if_simple show_if_variable',
            'label'         => __( 'Bookable', 'wc_ebs' ),
            'description'   => __( 'Bookable products can be rent or booked on a daily schedule', 'wc_ebs' ),
            'default'       => 'no'
        );

        return $product_type_options;
    }

    // Save checkbox value to the product admin page
    public function wc_ebs_add_custom_price_fields_save( $post_id ) {

        $woocommerce_checkbox = isset( $_POST['_booking_option'] ) ? 'yes' : '';
        update_post_meta( $post_id, '_booking_option', $woocommerce_checkbox );

    }

    // Add custom form to the product page.
    public function wc_ebs_before_add_to_cart_button() {
        global $woocommerce, $post, $product;

        // Is product bookable ?
        $wc_ebs_options = get_post_meta($post->ID, '_booking_option', true);
        $info_text = $this->options['wc_ebs_info_text'];
        $start_date_text = $this->options['wc_ebs_start_date_text'];
        $end_date_text = $this->options['wc_ebs_end_date_text'];

        // Product is bookable
        if ( isset($wc_ebs_options) && $wc_ebs_options == 'yes' ) {

            // Display info text
            if ( isset( $info_text ) && ! empty ( $info_text ) ) {
                echo apply_filters( 'wc_ebs_before_picker_form',
                    '<p class="woocommerce-info">' . esc_html__( $info_text ) . '</p>', $info_text );
            }

            echo '<div class="wc_ebs_errors">' . wc_print_notices() . '</div>';

            // Please do not remove inputs' attributes (classes, ids, etc.)
            echo apply_filters( 'wc_ebs_picker_form',
                '<p>
                    <label for="start_date">' . esc_html__( $start_date_text ) . ' : </label>
                    <input type="hidden" id="variation_id" name="variation_id" data-product_id="' . $product->id . '" value="">
                    <input type="text" id="start_date" class="datepicker datepicker_start" data-value="">
                </p>
                <p>
                    <label for="end_date">' . esc_html__( $end_date_text ) . ' : </label>
                    <input type="text" id="end_date" class="datepicker datepicker_end" data-value="">
                </p>', $start_date_text, $end_date_text );

            // If product is not variable, add a new price field before add to cart button
            if ( ! $product->is_type( 'variable' ) )
                echo '<p class="booking_price"><span class="price"></span></p>';

        }
    }

    // Display base price or new price
    public function wc_ebs_add_price_html($content) {

        global $woocommerce, $post;
        
        $product_id = isset($_POST['product_id']) && intval($_POST['product_id']) ? $_POST['product_id'] : $post->ID; // Product ID

        $wc_ebs_options = get_post_meta($product_id, '_booking_option', true); // Is it bookable ?

        // If bookable, return a price / day. If not, return normal price
        if ( isset( $wc_ebs_options ) && $wc_ebs_options == 'yes' ) {
            return $content . __(' / day', 'wc_ebs');
        } else {
            return $content;
        }

    }

    // Calculate new price, update product meta and refresh fragments
    public function wc_ebs_get_new_price() {
        global $woocommerce, $post;

        $product_id = isset($_POST['product_id']) && intval($_POST['product_id'] ) ? $_POST['product_id'] : ''; // Product ID
        $variation_id = isset($_POST['variation_id']) && intval($_POST['variation_id'] ) ? $_POST['variation_id'] : ''; // Variation ID

        $days = isset($_POST['days']) && intval($_POST['days']) ? $_POST['days'] : 1; // Booking duration
        $calc_mode = $this->options['wc_ebs_calc_mode']; // Calculation mode (Days or Nights)

        $start_date = isset($_POST['start']) ? sanitize_text_field($_POST['start']) : ''; // Booking start date
        $end_date = isset($_POST['end']) ? sanitize_text_field($_POST['end']) : ''; // Booking end date

        $start = isset($_POST['start_format']) && intval( $_POST['start_format'] ) ? $_POST['start_format'] : 1;
        $end = isset($_POST['end_format']) && intval( $_POST['end_format'] ) ? $_POST['end_format'] : 1;

        // If calculation mode is set to "Days", add one day
        if ( $calc_mode == "days" && ( $start != $end ) ) {
            $duration = $days + 1;
        } elseif ( ( $calc_mode == "days" ) && ( $start == $end ) ) {
            $duration = $days;
        } else {
            $duration = $days;
        }

        $product = get_product( $product_id ); // Product object
        
        // If product is variable, get variation price
        if ( $product->is_type( 'variable' ) ) {

            if ( empty( $variation_id ) ) // If no variation was selected
                $error_code = 3;

            $variable_product = new WC_Product_Variation( $variation_id );
            $variation_price = $variable_product->price;
            $new_price = $variation_price * $duration;

        } else {

            $price = get_post_meta($product_id,'_price', true); // Product price (Regular or sale)
            $new_price = $price * $duration;

        }

        $array = array(
            'new_price' => $new_price,
            'duration' => $duration,
            'start_date' => $start_date,
            'end_date' => $end_date,
            'start' => $start,
            'end' => $end
        );

        $booking_data[$product_id] = $array;

        // If number of days is inferior to 0
        if ( $duration <= 0 )
            $error_code = 1;

        // If one date is empty
        if ( $start_date == '' || $end_date == '' )
            $error_code = 2;

        // Show error message
        if ( isset( $error_code ) ) {

            $error_message = $this->wc_ebs_get_date_error( $error_code );
            wc_add_notice( $error_message, 'error' );

            $this->wc_ebs_error_fragment($error_message);

        } else {

            // Update session data
            WC()->session->set_customer_session_cookie(true);
            WC()->session->set( 'booking', $booking_data );

            // Return fragments
            $this->wc_ebs_new_price_fragment();

        }

        die();

    }

    // Get error messages
    public function wc_ebs_get_date_error( $error_code ) {

        switch ( $error_code ) {
            case 1:
                $err = __( 'Please choose valid dates', 'wc_ebs' );
            break;
            case 2:
                $err = __( 'Please choose two dates', 'wc_ebs' );
            break;
            case 3:
                $err = __( 'Please select product option', 'wc_ebs' );
            break;
            default:
                $err = '';
            break;
        }

        return $err;
    }

    // Update error messages with Ajax
    public function wc_ebs_error_fragment( $messages ) {

        global $woocommerce;

        header( 'Content-Type: application/json; charset=utf-8' );

        ob_start();
        wc_print_notices();
        $messages = ob_get_clean();

            $data = array(
                'errors' => array(
                    'div.wc_ebs_errors' => '<div class="wc_ebs_errors">' . $messages . '</div>'
                )
            );

        wp_send_json( $data );

        die();

    }

    // Update price fragment
    public function wc_ebs_new_price_fragment() {
        global $woocommerce, $post, $product;

        header( 'Content-Type: application/json; charset=utf-8' );
        
        $product_id = isset($_POST['product_id']) && intval($_POST['product_id']) ? $_POST['product_id'] : '';
        $booking_session = WC()->session->get( 'booking' );
        $new_price = $booking_session[$product_id]['new_price']; // New booking price
        $currency = get_woocommerce_currency_symbol(); // Currency

        if ( $booking_session[$product_id]['duration'] ) {

            ob_start();
            $fragments = ob_get_clean();

                $data = array(
                    'fragments' => apply_filters( 'wc_ebs_fragments', array(
                        'span.price' => '<span class="price">' . sprintf( get_woocommerce_price_format(), $currency, $new_price ) . '</span>'
                        )
                    )
                );

            wp_send_json( $data );
            die();
        }

    }

    // Add custom text link on product archive
    public function wc_ebs_custom_loop_add_to_cart($content, $product) {

        global $woocommerce, $post, $product;
        $wc_ebs_options = get_post_meta($post->ID, '_booking_option', true);

        // If product is bookable
        if ( isset($wc_ebs_options) && $wc_ebs_options == "yes" ) {

            $link = get_permalink( $product->id );
            $label = __( 'Select dates', 'wc_ebs' );

            return '<a href="' . esc_url( $link ) . '" rel="nofollow" class="product_type_variable button">' . esc_html( $label  ) . '</a>';
        } else {
            return $content;
        }
    }

}

$wcebs = new WC_EBS;