<?php
if (!defined('ABSPATH')) {
    exit;
}

class EnspirePlugins_ACFW_Email_Display {
    
    private $custom_fields = array();
    
    public function __construct() {
        $this->load_settings();
        
        add_action('woocommerce_email_order_meta', array($this, 'add_custom_fields_to_email'), 10, 4);
        add_action('woocommerce_order_details_after_order_table', array($this, 'add_custom_fields_to_order_confirmation'), 10, 1);
    }
    
    private function load_settings() {
        $settings = get_option('enspireplugins_acfw_settings', array());
        $this->custom_fields = isset($settings['custom_fields']) ? $settings['custom_fields'] : array();
    }
    
    /**
     * Add custom fields to WooCommerce emails
     */
    public function add_custom_fields_to_email($order, $sent_to_admin, $plain_text, $email) {
        if (empty($this->custom_fields)) {
            return;
        }
        
        static $email_displayed_orders = array();
        $order_id = $order->get_id();
        $email_key = $email->id . '_' . $order_id;
        
        if (in_array($email_key, $email_displayed_orders)) {
            return;
        }
        $email_displayed_orders[] = $email_key;
        
        $custom_fields_data = $this->get_custom_fields_data($order, 'email');
        
        if (empty($custom_fields_data)) {
            return;
        }
        
        if (!$this->should_display_in_email($email->id)) {
            return;
        }
        
        if ($plain_text) {
            $this->display_plain_text_fields($custom_fields_data);
        } else {
            $this->display_fields($custom_fields_data);
        }
    }
    
    /**
     * Add custom fields to order confirmation page
     */
    public function add_custom_fields_to_order_confirmation($order) {
        if (defined('WOOCOMMERCE_CHECKOUT') || is_admin() || wp_doing_ajax()) {
            return;
        }
        
        if (!is_wc_endpoint_url('order-received')) {
            return;
        }
        
        static $displayed_orders = array();
        $order_id = $order->get_id();
        
        if (in_array($order_id, $displayed_orders)) {
            return;
        }
        $displayed_orders[] = $order_id;
        
        if (empty($this->custom_fields)) {
            return;
        }
        
        $custom_fields_data = $this->get_custom_fields_data($order, 'order_confirmation');
        
        if (empty($custom_fields_data)) {
            return;
        }
        
        $this->display_fields($custom_fields_data);
    }
    
    /**
     * Get custom fields data from order
     */
    private function get_custom_fields_data($order, $context = 'email') {
        $custom_fields_data = array();
        
        $active_fields = array_filter($this->custom_fields, function($field) {
            return isset($field['active']) && $field['active'] === 'yes';
        });
        
        foreach ($active_fields as $field) {
            if ($context === 'email') {
                $show_in_context = isset($field['show_in_emails']) && $field['show_in_emails'] === 'yes';
            } else {
                $show_in_context = isset($field['show_in_order_confirmation']) && $field['show_in_order_confirmation'] === 'yes';
            }
            
            if (!$show_in_context) {
                continue;
            }
            
            if ($field['type'] === 'heading') {
                $custom_fields_data[] = array(
                    'label' => $field['label'],
                    'value' => '',
                    'type' => 'heading'
                );
            } else {
                $field_id = 'enspireplugins_acfw_' . $field['id'];
                $field_value = $order->get_meta($field_id, true);

                $has_value = false;
                if ($field['type'] === 'checkbox') {
                    $has_value = true; 
                } else {
                    $has_value = !empty($field_value) || $field_value === '0' || $field_value === 0;
                }
                
                $custom_fields_data[] = array(
                    'label' => $field['label'],
                    'value' => $this->format_field_value($field_value, $field),
                    'type' => $field['type'],
                    'has_value' => $has_value
                );
            }
        }
        
        return $custom_fields_data;
    }
    
    /**
     * Check if we should display fields in this email type
     */
    private function should_display_in_email($email_id) {
        $allowed_emails = array(
            'new_order',
            'customer_processing_order',
            'customer_completed_order',
            'customer_invoice',
            'customer_note',
            'customer_on_hold_order',
            'customer_refunded_order',
            'cancelled_order',
            'failed_order'
        );
        
        return in_array($email_id, $allowed_emails);
    }
    
    /**
     * Display fields
     */
    private function display_fields($custom_fields_data) {
        ?>
        <div style="margin: 20px 0;">
            <h2 style="color: #333; font-size: clamp(15.747px, 0.984rem + ((1vw - 3.2px) * 0.809), 24px); margin-bottom: 15px; border-bottom: 2px solid #0073aa; padding-bottom: 8px;">
                <?php echo esc_html__('Additional Checkout Fields for Woo', 'additional-checkout-fields-for-woo'); ?>
            </h2>
            
            <table style="width: 100%; border-collapse: collapse; border: 1px solid #ddd;">
                <tbody>
                    <?php foreach ($custom_fields_data as $field_data) : ?>
                        <?php if ($field_data['type'] === 'heading') : ?>
                            <tr>
                                <td colspan="2" style="padding: 12px 15px; background: #f5f5f5; color: #333; font-weight: bold; border: 1px solid #ddd;">
                                    <?php echo esc_html($field_data['label']); ?>
                                </td>
                            </tr>
                        <?php else : ?>
                            <tr>
                                <td style="padding: 10px 15px; font-weight: bold; color: #555; background: #fff; width: 35%; vertical-align: top; border: 1px solid #ddd;">
                                    <?php echo esc_html($field_data['label']); ?>:
                                </td>
                                <td style="padding: 10px 15px; color: #333; background: #fff; vertical-align: top; border: 1px solid #ddd;">
                                    <?php if ($field_data['has_value']) : ?>
                                        <?php if ($field_data['type'] === 'url') : ?>
                                            <a href="<?php echo esc_url($field_data['value']); ?>" target="_blank" style="color: #0073aa; text-decoration: none;">
                                                <?php echo esc_html($field_data['value']); ?>
                                            </a>
                                        <?php elseif ($field_data['type'] === 'email') : ?>
                                            <a href="mailto:<?php echo esc_attr($field_data['value']); ?>" style="color: #0073aa; text-decoration: none;">
                                                <?php echo esc_html($field_data['value']); ?>
                                            </a>
                                        <?php elseif ($field_data['type'] === 'textarea') : ?>
                                            <div style="white-space: pre-line;">
                                                <?php echo nl2br(esc_html($field_data['value'])); ?>
                                            </div>
                                        <?php else : ?>
                                            <?php echo esc_html($field_data['value']); ?>
                                        <?php endif; ?>
                                    <?php else : ?>
                                        <em style="color: #999;">
                                            <?php echo esc_html__('-', 'additional-checkout-fields-for-woo'); ?>
                                        </em>
                                    <?php endif; ?>
                                </td>
                            </tr>
                        <?php endif; ?>
                    <?php endforeach; ?>
                </tbody>
            </table>
        </div>
        <?php
    }
    
    /**
     * Display fields in plain text format
     */
    private function display_plain_text_fields($custom_fields_data) {
        echo "\n\n" . esc_html__('Additional Checkout Fields for Woo', 'additional-checkout-fields-for-woo') . "\n";
        echo esc_html(str_repeat('=', 40)) . "\n\n";
        
        foreach ($custom_fields_data as $field_data) {
            if ($field_data['type'] === 'heading') {
                echo "\n" . esc_html(strtoupper($field_data['label'])) . "\n";
                echo esc_html(str_repeat('-', strlen($field_data['label']))) . "\n";
            } else {
                $value = $field_data['has_value'] ? $field_data['value'] : esc_html__('-', 'additional-checkout-fields-for-woo');
                echo esc_html($field_data['label']) . ': ' . esc_html($value) . "\n";
            }
        }
        
        echo "\n";
    }
    
    /**
     * Format field value for display
     */
    private function format_field_value($value, $field) {
        $type = $field['type'];
        
        switch ($type) {
            case 'checkbox':
                if ($value === 'yes' || $value === '1' || $value === 1) {
                    return esc_html__('Yes', 'additional-checkout-fields-for-woo');
                } else {
                    return esc_html__('No', 'additional-checkout-fields-for-woo');
                }
            
            case 'textarea':
                return wp_strip_all_tags($value);
            
            case 'url':
                return esc_url($value);
            
            case 'email':
                return esc_html($value);
            
            case 'number':
                return is_numeric($value) ? esc_html($value) : esc_html($value);
            
            case 'select':
            case 'radio':
                if (isset($field['options']) && is_array($field['options'])) {
                    foreach ($field['options'] as $option) {
                        if (isset($option['value']) && $option['value'] === $value) {
                            return esc_html($option['label']);
                        }
                    }
                }
                return esc_html($value);
            
            default:
                return esc_html($value);
        }
    }
    
    /**
     * Helper function to get an instance of the class
     */
    public static function get_instance() {
        static $instance = null;
        
        if (null === $instance) {
            $instance = new self();
        }
        
        return $instance;
    }
}

EnspirePlugins_ACFW_Email_Display::get_instance();