qdr.app.woo.plugins/studiou-wc-free-photo-product/includes/class-studiou-wc-fpp-single-product.php

<?php
if (!defined('WPINC')) {
    die;
}

class Studiou_WC_FPP_Single_Product {

    public function __construct() {
        // Render upload area below the variation table on single product page
        add_action('woocommerce_single_product_summary', array($this, 'render_upload_area'), 35);

        // Server-side validation: block add-to-cart without photo (priority 1 = very early)
        add_filter('woocommerce_add_to_cart_validation', array($this, 'validate_add_to_cart'), 1, 6);

        // Safety net: if item was added without photo, remove it immediately
        add_action('woocommerce_add_to_cart', array($this, 'check_after_add_to_cart'), 10, 6);

        // AJAX: store uploaded file reference in WC session
        add_action('wp_ajax_studiou_wcfpp_set_upload_session', array($this, 'ajax_set_upload_session'));
        add_action('wp_ajax_nopriv_studiou_wcfpp_set_upload_session', array($this, 'ajax_set_upload_session'));
        add_action('wp_ajax_studiou_wcfpp_clear_upload_session', array($this, 'ajax_clear_upload_session'));
        add_action('wp_ajax_nopriv_studiou_wcfpp_clear_upload_session', array($this, 'ajax_clear_upload_session'));
    }

    public function render_upload_area() {
        global $product;
        if (!$product || !Studiou_WC_FPP_Product::is_fpp_product($product->get_id())) {
            return;
        }

        wp_enqueue_style('studiou-wcfpp-frontend');
        wp_enqueue_script('studiou-wcfpp-frontend');

        $product_id = $product->get_id();
        $max_file_size = Studiou_WC_FPP_Product::get_product_max_file_size($product_id);

        // Pass existing session state to JS for restore on page load
        $session_data = array();
        if (WC()->session) {
            $att_id = WC()->session->get('studiou_fpp_attachment_id');
            if ($att_id) {
                $thumb = wp_get_attachment_image_src($att_id, 'thumbnail');
                $preview = wp_get_attachment_image_src($att_id, 'woocommerce_single');
                $session_data = array(
                    'attachment_id'  => $att_id,
                    'file_record_id' => WC()->session->get('studiou_fpp_file_record_id'),
                    'file_name'      => WC()->session->get('studiou_fpp_file_name'),
                    'thumbnail_url'  => $thumb ? $thumb[0] : '',
                    'preview_url'    => $preview ? $preview[0] : ($thumb ? $thumb[0] : ''),
                );
            }
        }

        wp_localize_script('studiou-wcfpp-frontend', 'studiouWcfppSession', $session_data);

        include STUDIOU_WCFPP_PLUGIN_DIR . 'views/single-product-upload.php';
    }

    /**
     * Resolve whether a product_id or its parent is FPP-enabled.
     */
    private function resolve_fpp_product_id($product_id, $variation_id = 0) {
        // Check direct product
        if (Studiou_WC_FPP_Product::is_fpp_product($product_id)) {
            return $product_id;
        }

        // Check parent of the product (in case product_id is variation)
        $product = wc_get_product($product_id);
        if ($product && $product->get_parent_id() && Studiou_WC_FPP_Product::is_fpp_product($product->get_parent_id())) {
            return $product->get_parent_id();
        }

        // Check variation_id's parent
        if ($variation_id && $variation_id !== $product_id) {
            $variation = wc_get_product($variation_id);
            if ($variation && $variation->get_parent_id() && Studiou_WC_FPP_Product::is_fpp_product($variation->get_parent_id())) {
                return $variation->get_parent_id();
            }
        }

        return false;
    }

    /**
     * Validate before add-to-cart: block if FPP product and no photo in session.
     */
    public function validate_add_to_cart($passed, $product_id, $quantity, $variation_id = 0, $variations = array(), $cart_item_data = array()) {
        $fpp_id = $this->resolve_fpp_product_id($product_id, $variation_id);
        if (!$fpp_id) {
            return $passed;
        }

        // Check WC session for uploaded file
        if (WC()->session) {
            $attachment_id = WC()->session->get('studiou_fpp_attachment_id');
            if ($attachment_id) {
                return $passed;
            }
        }

        wc_add_notice(__('Please upload a photo before adding to cart.', 'studiou-wc-free-photo-product'), 'error');
        return false;
    }

    /**
     * Safety net: if an FPP item was added without photo data, remove it.
     * This catches cases where validation was somehow bypassed.
     */
    public function check_after_add_to_cart($cart_item_key, $product_id, $quantity, $variation_id, $variation, $cart_item_data) {
        $fpp_id = $this->resolve_fpp_product_id($product_id, $variation_id);
        if (!$fpp_id) {
            return;
        }

        // If the cart item has no FPP attachment, remove it
        if (empty($cart_item_data['studiou_fpp_attachment_id'])) {
            WC()->cart->remove_cart_item($cart_item_key);
            wc_add_notice(__('Please upload a photo before adding to cart.', 'studiou-wc-free-photo-product'), 'error');
        }
    }

    public function ajax_set_upload_session() {
        check_ajax_referer('studiou-wcfpp-front-nonce', 'nonce');

        $attachment_id = isset($_POST['attachment_id']) ? absint($_POST['attachment_id']) : 0;
        $file_record_id = isset($_POST['file_record_id']) ? absint($_POST['file_record_id']) : 0;
        $file_name = isset($_POST['file_name']) ? sanitize_text_field($_POST['file_name']) : '';

        if (!$attachment_id || !$file_record_id) {
            wp_send_json_error(array('message' => __('Invalid file data.', 'studiou-wc-free-photo-product')));
            return;
        }

        if (WC()->session) {
            WC()->session->set('studiou_fpp_attachment_id', $attachment_id);
            WC()->session->set('studiou_fpp_file_record_id', $file_record_id);
            WC()->session->set('studiou_fpp_file_name', $file_name);

            // Store thumbnail URL directly so it can be used in cart without wp_get_attachment lookup
            $thumb_url = '';
            $thumb = wp_get_attachment_image_src($attachment_id, 'thumbnail');
            if ($thumb) {
                $thumb_url = $thumb[0];
            } else {
                $thumb_url = wp_get_attachment_url($attachment_id);
            }
            WC()->session->set('studiou_fpp_thumb_url', $thumb_url ?: '');
        }

        wp_send_json_success();
    }

    public function ajax_clear_upload_session() {
        check_ajax_referer('studiou-wcfpp-front-nonce', 'nonce');

        if (WC()->session) {
            WC()->session->set('studiou_fpp_attachment_id', null);
            WC()->session->set('studiou_fpp_file_record_id', null);
            WC()->session->set('studiou_fpp_file_name', null);
        }

        wp_send_json_success();
    }
}