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

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

class Studiou_WC_FPP_Cart {

    /** @var Studiou_WC_FPP_DB */
    private $db;

    public function __construct($db) {
        $this->db = $db;

        // Capture file data from WC session on add-to-cart
        add_filter('woocommerce_add_cart_item_data', array($this, 'add_cart_item_data'), 10, 3);

        // Display uploaded file info in cart
        add_filter('woocommerce_get_item_data', array($this, 'display_cart_item_data'), 10, 2);

        // Replace cart item thumbnail with uploaded image
        add_filter('woocommerce_cart_item_thumbnail', array($this, 'cart_item_thumbnail'), 10, 3);

        // Save file reference to order item
        add_action('woocommerce_checkout_create_order_line_item', array($this, 'save_order_item_meta'), 10, 4);

        // Link file records to order after checkout
        add_action('woocommerce_checkout_order_processed', array($this, 'link_files_to_order'), 10, 3);

        // Replace order item thumbnail with uploaded image (admin)
        add_filter('woocommerce_admin_order_item_thumbnail', array($this, 'admin_order_item_thumbnail'), 10, 3);
    }

    public function add_cart_item_data($cart_item_data, $product_id, $variation_id) {
        // Check if this product (or its parent) is FPP enabled
        $fpp_product_id = $product_id;
        if (!Studiou_WC_FPP_Product::is_fpp_product($product_id)) {
            $product = wc_get_product($product_id);
            if ($product && $product->get_parent_id() && Studiou_WC_FPP_Product::is_fpp_product($product->get_parent_id())) {
                $fpp_product_id = $product->get_parent_id();
            } else {
                return $cart_item_data;
            }
        }

        // Read file data from WC session
        if (!WC()->session) {
            return $cart_item_data;
        }

        $attachment_id = WC()->session->get('studiou_fpp_attachment_id');
        $file_record_id = WC()->session->get('studiou_fpp_file_record_id');
        $file_name = WC()->session->get('studiou_fpp_file_name');

        if (!$attachment_id || !$file_record_id) {
            return $cart_item_data;
        }

        // Verify the file record exists
        $record = $this->db->get_file_record($file_record_id);
        if (!$record || (int) $record->attachment_id !== (int) $attachment_id) {
            return $cart_item_data;
        }

        $cart_item_data['studiou_fpp_attachment_id'] = $attachment_id;
        $cart_item_data['studiou_fpp_file_record_id'] = $file_record_id;
        $cart_item_data['studiou_fpp_file_name'] = $file_name ?: $record->file_name;

        // Update file record with variation
        $actual_variation_id = $variation_id ?: $product_id;
        $this->db->update_file_record($file_record_id, array(
            'variation_id' => $actual_variation_id,
        ));

        // Clear session data after reading (so next add-to-cart is clean)
        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);

        return $cart_item_data;
    }

    public function cart_item_thumbnail($thumbnail, $cart_item, $cart_item_key) {
        if (!empty($cart_item['studiou_fpp_attachment_id'])) {
            $image = wp_get_attachment_image($cart_item['studiou_fpp_attachment_id'], 'woocommerce_thumbnail');
            if ($image) {
                return $image;
            }
        }
        return $thumbnail;
    }

    public function display_cart_item_data($item_data, $cart_item) {
        if (isset($cart_item['studiou_fpp_file_name'])) {
            $item_data[] = array(
                'name'    => __('Uploaded Photo', 'studiou-wc-free-photo-product'),
                'value'   => esc_html($cart_item['studiou_fpp_file_name']),
                'display' => '',
            );
        }
        return $item_data;
    }

    public function save_order_item_meta($item, $cart_item_key, $values, $order) {
        if (isset($values['studiou_fpp_attachment_id'])) {
            $item->add_meta_data('_studiou_fpp_attachment_id', $values['studiou_fpp_attachment_id']);
        }
        if (isset($values['studiou_fpp_file_record_id'])) {
            $item->add_meta_data('_studiou_fpp_file_record_id', $values['studiou_fpp_file_record_id']);
        }
        if (isset($values['studiou_fpp_file_name'])) {
            $item->add_meta_data('_studiou_fpp_file_name', $values['studiou_fpp_file_name']);
            $item->add_meta_data(__('Uploaded Photo', 'studiou-wc-free-photo-product'), $values['studiou_fpp_file_name']);
        }
    }

    public function link_files_to_order($order_id, $posted_data, $order) {
        foreach ($order->get_items() as $item_id => $item) {
            $file_record_id = $item->get_meta('_studiou_fpp_file_record_id');
            if ($file_record_id) {
                $this->db->link_file_to_order((int) $file_record_id, $order_id, $item_id);
            }
        }
    }

    public function admin_order_item_thumbnail($thumbnail, $item_id, $item) {
        $attachment_id = $item->get_meta('_studiou_fpp_attachment_id');
        if ($attachment_id) {
            $image = wp_get_attachment_image($attachment_id, 'thumbnail');
            if ($image) {
                return $image;
            }
        }
        return $thumbnail;
    }
}