<?php

if (!defined('ABSPATH')) { exit; }

class AIEO_DMM_Trust_Badges extends AIEO_DMM_Module_Base {

    const OPTION_KEY = 'aieo_dmm_trust_badges_settings';

    public function get_id()       { return 'trust_badges'; }
    public function get_icon()     { return 'badge-check'; }
    public function get_label()    { return __('Trust badges', 'ai-eshop-optimizer'); }
    public function get_priority() { return 40; }
    public function get_group()    { return 'product_display'; }

    
    private function builtin_badges() {
        $base = plugins_url('assets/images/product-display/trust-badges/', AIEO_FILE);
        return array(
            'secure-checkout' => array(
                'label'       => __('Secure Checkout', 'ai-eshop-optimizer'),
                'description' => __('Your payment information is processed securely.', 'ai-eshop-optimizer'),
                'image_url'   => $base . 'secure-checkout.svg',
            ),
            'money-back' => array(
                'label'       => __('Money Back', 'ai-eshop-optimizer'),
                'description' => __('Full refund if you are not satisfied with your purchase.', 'ai-eshop-optimizer'),
                'image_url'   => $base . 'money-back.svg',
            ),
            'free-shipping' => array(
                'label'       => __('Free Shipping', 'ai-eshop-optimizer'),
                'description' => __('Free shipping on qualifying orders.', 'ai-eshop-optimizer'),
                'image_url'   => $base . 'free-shipping.svg',
            ),
            'quality' => array(
                'label'       => __('Quality', 'ai-eshop-optimizer'),
                'description' => __('High quality products guaranteed.', 'ai-eshop-optimizer'),
                'image_url'   => $base . 'quality.svg',
            ),
            'satisfaction' => array(
                'label'       => __('Satisfaction', 'ai-eshop-optimizer'),
                'description' => __('Your satisfaction is our priority.', 'ai-eshop-optimizer'),
                'image_url'   => $base . 'satisfaction.svg',
            ),
            'support' => array(
                'label'       => __('Support', 'ai-eshop-optimizer'),
                'description' => __('Customer support available when you need it.', 'ai-eshop-optimizer'),
                'image_url'   => $base . 'support.svg',
            ),
        );
    }

    public function register_hooks() {
        add_action('wp_ajax_aieo_dmm_trust_badges_save', array($this, 'ajax_save'));
        
        add_shortcode('aieo_trust_badges',   array($this, 'shortcode_handler'));
        add_shortcode('om_ppe_trust_badges', array($this, 'shortcode_handler'));

        $s = $this->get_settings();
        if (empty($s['enabled'])) { return; }

        add_action('wp_enqueue_scripts',                array($this, 'enqueue_styles'));
        if (empty($s['use_shortcode'])) {
            
            
            
            
            
            
            
            add_action('woocommerce_single_product_summary',  array($this, 'output'), 35);
        }
    }

    public function get_settings($raw = false) {
        $stored = get_option(self::OPTION_KEY, array());
        if (!is_array($stored)) { $stored = array(); }
        $legacy = get_option('om_ppe_trust-badges', array());
        if (!is_array($legacy)) { $legacy = array(); }
        $merged = array_merge(array(
            'enabled'           => false,
            'title'             => '', 
            'badges'            => isset($legacy['badges']) ? (string) $legacy['badges'] : '',
            'default_badges'    => isset($legacy['default_badges']) ? (array) $legacy['default_badges'] : array('secure-checkout','money-back','free-shipping'),
            'hidden_badges'     => isset($legacy['hidden_badges']) ? (array) $legacy['hidden_badges'] : array(),
            'badge_overrides'   => isset($legacy['default_badges_data']) && is_array($legacy['default_badges_data']) ? $legacy['default_badges_data'] : array(),
            'align'             => 'center',
            'font_size'         => 15,
            'text_color'        => '#212121',
            'border_color'      => '#e5e5e5',
            'margin_top'        => 20,
            'margin_bottom'     => 20,
            'image_max_width'   => 70,
            'image_max_height'  => 70,
            'mobile_columns'    => '2',
            'use_shortcode'     => false,
            'custom_class'      => '',
        ), $stored);
        return $raw ? $merged : self::fill_i18n_defaults($merged);
    }
    
    private static function i18n_default() {
        return __('Product Quality Guaranteed!', 'ai-eshop-optimizer');
    }

    
    private static function fill_i18n_defaults(array $s) {
        if (!did_action('init')) { return $s; }
        if (!isset($s['title']) || '' === trim((string) $s['title'])) {
            $s['title'] = self::i18n_default();
        }
        return $s;
    }

    public function ajax_save() {
        AIEO_DMM::verify_ajax_request();
        $valid_align = array('flex-start','center','flex-end');
        $align = isset($_POST['align']) ? sanitize_key(wp_unslash($_POST['align'])) : 'center';
        $defaults = isset($_POST['default_badges']) && is_array($_POST['default_badges'])
            ? array_values(array_filter(array_map('sanitize_key', wp_unslash($_POST['default_badges']))))
            : array();
        $hidden   = isset($_POST['hidden_badges']) && is_array($_POST['hidden_badges'])
            ? array_values(array_filter(array_map('sanitize_key', wp_unslash($_POST['hidden_badges']))))
            : array();
        
        $overrides = array();
        if (isset($_POST['badge_overrides']) && is_array($_POST['badge_overrides'])) {
            foreach (wp_unslash($_POST['badge_overrides']) as $key => $v) {
                $key = sanitize_key($key);
                if ($key === '') { continue; }
                $label = isset($v['label']) ? sanitize_text_field($v['label']) : '';
                $desc  = isset($v['description']) ? sanitize_text_field($v['description']) : '';
                if ($label !== '' || $desc !== '') {
                    $overrides[] = array('badge_key' => $key, 'label' => $label, 'description' => $desc);
                }
            }
        }
        update_option(self::OPTION_KEY, array(
            'enabled'           => !empty($_POST['enabled']),
            'title'             => isset($_POST['title']) ? sanitize_text_field(wp_unslash($_POST['title'])) : '',
            'badges'            => isset($_POST['badges']) ? sanitize_text_field(wp_unslash($_POST['badges'])) : '',
            'default_badges'    => $defaults,
            'hidden_badges'     => $hidden,
            'badge_overrides'   => $overrides,
            'align'             => in_array($align, $valid_align, true) ? $align : 'center',
            'font_size'         => max(10, min(30, isset($_POST['font_size']) ? (int) $_POST['font_size'] : 15)),
            'text_color'        => isset($_POST['text_color']) ? (sanitize_hex_color(wp_unslash($_POST['text_color'])) ?: '#212121') : '#212121',
            'border_color'      => isset($_POST['border_color']) ? (sanitize_hex_color(wp_unslash($_POST['border_color'])) ?: '#e5e5e5') : '#e5e5e5',
            'margin_top'        => max(0, min(100, isset($_POST['margin_top']) ? (int) $_POST['margin_top'] : 20)),
            'margin_bottom'     => max(0, min(100, isset($_POST['margin_bottom']) ? (int) $_POST['margin_bottom'] : 20)),
            'image_max_width'   => max(20, min(200, isset($_POST['image_max_width']) ? (int) $_POST['image_max_width'] : 70)),
            'image_max_height'  => max(20, min(200, isset($_POST['image_max_height']) ? (int) $_POST['image_max_height'] : 70)),
            'mobile_columns'    => isset($_POST['mobile_columns']) ? (string) (int) $_POST['mobile_columns'] : '2',
            'use_shortcode'     => !empty($_POST['use_shortcode']),
            'custom_class'      => isset($_POST['custom_class']) ? sanitize_html_class(wp_unslash($_POST['custom_class'])) : '',
        ), false);
        wp_send_json_success(array('message' => __('Trust Badges settings saved.', 'ai-eshop-optimizer')));
    }

    
    public function enqueue_styles() {
        if (!function_exists('is_product') || !is_product()) { return; }
        $version = defined('AIEO_VERSION') ? AIEO_VERSION : '1.0';
        wp_enqueue_style(
            'aieo-trust-badges-strip',
            plugins_url('assets/css/product-display/trust-badges.css', AIEO_FILE),
            array(), $version
        );
        $s = $this->get_settings();
        $css = sprintf(
            '.aieo-trust-badges{' .
                '--aieo-tb-font-size:%dpx;--aieo-tb-text-color:%s;--aieo-tb-border-color:%s;' .
                '--aieo-tb-margin-top:%dpx;--aieo-tb-margin-bottom:%dpx;--aieo-tb-align:%s;' .
                '--aieo-tb-image-max-width:%dpx;--aieo-tb-image-max-height:%dpx;' .
            '}',
            (int) $s['font_size'], esc_attr($s['text_color']), esc_attr($s['border_color']),
            (int) $s['margin_top'], (int) $s['margin_bottom'], esc_attr($s['align']),
            (int) $s['image_max_width'], (int) $s['image_max_height']
        );
        wp_add_inline_style('aieo-trust-badges-strip', $css);
    }

    
    public function is_active() {
        $s = $this->get_settings();
        return !empty($s['enabled']);
    }

    public function output() {
        
        if (!$this->is_active()) { return; }
        
        
        
        
        static $rendered = false;
        if ($rendered) { return; }
        $rendered = true;
        echo $this->get_html(); 
    }

    public function shortcode_handler($atts = array()) {
        $s = $this->get_settings();
        if (empty($s['enabled']) || empty($s['use_shortcode'])) { return ''; }
        if (!is_singular('product')) { return ''; }
        return $this->get_html();
    }

    private function effective_badge($key) {
        $s = $this->get_settings();
        $catalog = $this->builtin_badges();
        $b = isset($catalog[$key]) ? $catalog[$key] : array();
        $custom = array();
        foreach ((array) $s['badge_overrides'] as $o) {
            if (isset($o['badge_key']) && $o['badge_key'] === $key) { $custom = $o; break; }
        }
        return array(
            'label'       => !empty($custom['label']) ? (string) $custom['label'] : (isset($b['label']) ? (string) $b['label'] : ucwords(str_replace('-', ' ', $key))),
            'description' => !empty($custom['description']) ? (string) $custom['description'] : (isset($b['description']) ? (string) $b['description'] : ''),
            'image_url'   => isset($b['image_url']) ? (string) $b['image_url'] : '',
        );
    }

    private function get_html() {
        $s = $this->get_settings();
        $defaults = (array) $s['default_badges'];
        $hidden   = (array) $s['hidden_badges'];
        $visible  = array_values(array_diff($defaults, $hidden));
        $customs  = array_filter(array_map('absint', explode(',', (string) $s['badges'])));

        if (empty($visible) && empty($customs)) { return ''; }

        
        
        
        
        
        
        $cls = 'aieo-trust-badges om-ppe-trust-badges' . (!empty($s['custom_class']) ? ' ' . esc_attr((string) $s['custom_class']) : '');
        $cols = (string) (int) $s['mobile_columns'];

        
        
        
        $visible_count_for_grid = 0;
        foreach ($visible as $key) {
            $b = $this->effective_badge($key);
            if (!empty($b['image_url'])) { $visible_count_for_grid++; }
        }
        $visible_count_for_grid += count($customs);
        $odd_attr = ($visible_count_for_grid % 2 === 1) ? ' data-total-odd="true"' : '';

        ob_start();
        ?>
        <fieldset class="<?php echo esc_attr($cls); ?>" data-mobile-cols="<?php echo esc_attr($cols); ?>"<?php echo $odd_attr; ?>>
            <?php if (!empty($s['title'])): ?>
                <legend class="aieo-trust-badges-title"><?php echo esc_html((string) $s['title']); ?></legend>
            <?php endif; ?>
            <div class="aieo-trust-badges-images">
                <?php foreach ($visible as $key):
                    $b = $this->effective_badge($key);
                    if ($b['image_url'] === '') { continue; }
                    $title_attr = $b['description'] !== '' ? $b['description'] : $b['label'];
                ?>
                    <div class="aieo-trust-badges-item">
                        <img src="<?php echo esc_url($b['image_url']); ?>" class="aieo-trust-badges-default" alt="<?php echo esc_attr($b['label']); ?>" title="<?php echo esc_attr($title_attr); ?>" width="70" height="70" loading="lazy">
                        <span class="aieo-trust-badges-text"><?php echo esc_html($b['label']); ?></span>
                    </div>
                <?php endforeach; ?>
                <?php foreach ($customs as $img_id):
                    $alt = (string) get_post_meta($img_id, '_wp_attachment_image_alt', true);
                    if ($alt === '') {
                        $att = get_post($img_id);
                        $alt = $att ? (string) $att->post_title : '';
                    }
                    $img = wp_get_attachment_image($img_id, 'thumbnail', false, array('class' => 'aieo-trust-badges-custom', 'alt' => $alt, 'loading' => 'lazy'));
                    if ($img === '') { continue; }
                ?>
                    <div class="aieo-trust-badges-item"><?php echo $img; ?></div>
                <?php endforeach; ?>
            </div>
        </fieldset>
        <?php
        return (string) ob_get_clean();
    }

    public function render_subtab() {
        $s = $this->get_settings();
        $catalog = $this->builtin_badges();
        $defaults = (array) $s['default_badges'];
        $hidden   = (array) $s['hidden_badges'];
        $overrides_by_key = array();
        foreach ((array) $s['badge_overrides'] as $o) {
            if (!empty($o['badge_key'])) { $overrides_by_key[$o['badge_key']] = $o; }
        }

        
        
        
        
        
        
        
        
        $id_of  = static function ($name) {
            return 'aieo-tb-' . str_replace('_', '-', $name);
        };
        $row    = static function ($name, $label, $control, $help = '') use ($id_of) {
            $field = array('id' => $id_of($name), 'label' => $label, 'control' => $control);
            if ('' !== $help) { $field['help'] = $help; }
            return \Rooster\Admin\Ui::field_row($field);
        };
        $switch = static function ($name, $text, $hidden_label = false) use ($s, $id_of) {
            return \Rooster\Admin\Ui::toggle($name, !empty($s[$name]), array('id' => $id_of($name), 'hidden' => false, 'label' => $text, 'label_hidden' => $hidden_label));
        };
        $px     = static function ($name, $min, $max) use ($s, $id_of) {
            return \Rooster\Admin\Ui::input($name, (string) (int) $s[$name], array('id' => $id_of($name), 'type' => 'number', 'xs' => true, 'unit' => 'px', 'min' => $min, 'max' => $max));
        };
        $card   = static function ($anchor, $icon, $title, $col, $fields, $extra = array()) {
            $form_id = 'aieo-card-' . $anchor . '-form';
            list($open, $close) = AIEO_DMM_UI::card(array(
                'anchor' => $anchor,
                'icon'   => $icon,
                'title'  => $title,
                'col'    => $col,
                'footer' => class_exists('\Rooster\Admin\States') ? \Rooster\Admin\States::save_button($form_id) : '',
            ) + $extra);
            echo $open; 
            echo AIEO_DMM_UI::card_form('aieo_dmm_trust_badges_save', 'trust_badges', $form_id, '<div class="rooster-admin-fields">' . $fields . '</div>'); 
            echo $close; 
        };

        rooster_admin_band_open(array('cols' => 1));

        
        $fields  = $row('enabled', __('Enable', 'ai-eshop-optimizer'), $switch('enabled', __('On', 'ai-eshop-optimizer'), true));
        $fields .= $row('title', __('Title', 'ai-eshop-optimizer'), \Rooster\Admin\Ui::input('title', (string) $s['title'], array('id' => $id_of('title'), 'medium' => true, 'placeholder' => self::i18n_default())));
        $fields .= $row('badges', __('Custom badge attachment IDs', 'ai-eshop-optimizer'), \Rooster\Admin\Ui::input('badges', (string) $s['badges'], array('id' => $id_of('badges'), 'medium' => true, 'placeholder' => '123,456,789')), __('Comma-separated Media Library attachment IDs. Appended after the built-in badges.', 'ai-eshop-optimizer'));
        $card(
            'trust-badges-overview',
            'shield-check',
            __('Settings', 'ai-eshop-optimizer'),
            1,
            $fields,
            array('description' => __('Reassurance badges (Free Shipping, Secure Checkout, Money Back, etc.) on single product pages. Pick which built-ins to show, override their labels, optionally append custom badges by Media Library attachment IDs.', 'ai-eshop-optimizer'))
        );

        
        $fields = '';
        foreach ($catalog as $key => $info) {
            $o   = isset($overrides_by_key[$key]) ? $overrides_by_key[$key] : array();
            $lab = isset($o['label']) ? (string) $o['label'] : '';
            $des = isset($o['description']) ? (string) $o['description'] : '';
            $k   = sanitize_html_class((string) $key);
            
            
            
            $switches = '<div class="rooster-admin-checks is-grid">'
                . \Rooster\Admin\Ui::toggle('default_badges[]', in_array($key, $defaults, true), array('id' => 'aieo-tb-show-' . $k, 'value' => (string) $key, 'hidden' => false, 'label' => __('Show', 'ai-eshop-optimizer'), 'attrs' => array('aria-label' => $info['label'] . ', ' . __('Show', 'ai-eshop-optimizer'))))
                . \Rooster\Admin\Ui::toggle('hidden_badges[]', in_array($key, $hidden, true), array('id' => 'aieo-tb-hide-' . $k, 'value' => (string) $key, 'hidden' => false, 'label' => __('Temporarily hide', 'ai-eshop-optimizer'), 'attrs' => array('aria-label' => $info['label'] . ', ' . __('Temporarily hide', 'ai-eshop-optimizer'))))
                . '</div>';
            $texts = \Rooster\Admin\Fields::compound(
                array(
                    array('label' => __('Label', 'ai-eshop-optimizer'), 'id' => 'aieo-tb-label-' . $k, 'control' => \Rooster\Admin\Ui::input('badge_overrides[' . $key . '][label]', $lab, array('id' => 'aieo-tb-label-' . $k, 'medium' => true, 'placeholder' => (string) $info['label']))),
                    array('label' => __('Description', 'ai-eshop-optimizer'), 'id' => 'aieo-tb-desc-' . $k, 'control' => \Rooster\Admin\Ui::input('badge_overrides[' . $key . '][description]', $des, array('id' => 'aieo-tb-desc-' . $k, 'medium' => true, 'placeholder' => (string) $info['description']))),
                ),
                array('label' => (string) $info['label'])
            );
            $fields .= \Rooster\Admin\Ui::field_row(array(
                'id'      => 'aieo-tb-badge-' . $k,
                'label'   => (string) $info['label'],
                'group'   => true,
                'control' => $switches . $texts,
            ));
        }
        $card('trust-badges-built-in-badges', 'badge-check', __('Built-in badges', 'ai-eshop-optimizer'), 1, $fields, array('description' => __('Choose which ready-made reassurance badges show on product pages and change their wording.', 'ai-eshop-optimizer')));

        
        $fields  = $row('align', __('Alignment', 'ai-eshop-optimizer'), \Rooster\Admin\Ui::select('align', array('flex-start' => __('Left', 'ai-eshop-optimizer'), 'center' => __('Center', 'ai-eshop-optimizer'), 'flex-end' => __('Right', 'ai-eshop-optimizer')), (string) $s['align'], array('id' => $id_of('align'), 'short' => true)));
        $fields .= $row('font_size', __('Font size', 'ai-eshop-optimizer'), $px('font_size', 10, 30));
        $fields .= $row('text_color', __('Text colour', 'ai-eshop-optimizer'), \Rooster\Admin\Fields::color('text_color', (string) $s['text_color'], array('id' => $id_of('text_color'), 'default' => '#212121')));
        $fields .= $row('border_color', __('Border colour', 'ai-eshop-optimizer'), \Rooster\Admin\Fields::color('border_color', (string) $s['border_color'], array('id' => $id_of('border_color'), 'default' => '#e5e5e5')));
        $fields .= $row('margin_top', __('Margin top', 'ai-eshop-optimizer'), $px('margin_top', 0, 100));
        $fields .= $row('margin_bottom', __('Margin bottom', 'ai-eshop-optimizer'), $px('margin_bottom', 0, 100));
        $fields .= $row('image_max_width', __('Image max width', 'ai-eshop-optimizer'), $px('image_max_width', 20, 200));
        $fields .= $row('image_max_height', __('Image max height', 'ai-eshop-optimizer'), $px('image_max_height', 20, 200));
        $fields .= $row('mobile_columns', __('Mobile columns', 'ai-eshop-optimizer'), \Rooster\Admin\Ui::select('mobile_columns', array('1' => '1', '2' => '2', '3' => '3'), (string) $s['mobile_columns'], array('id' => $id_of('mobile_columns'), 'xs' => true)));
        $fields .= $row('custom_class', __('Custom CSS class', 'ai-eshop-optimizer'), \Rooster\Admin\Ui::input('custom_class', (string) $s['custom_class'], array('id' => $id_of('custom_class'), 'medium' => true, 'code' => true)));
        $fields .= $row('use_shortcode', __('Use shortcode (manual)', 'ai-eshop-optimizer'), $switch('use_shortcode', '[aieo_trust_badges]'), __('(legacy [om_ppe_trust_badges] alias still works)', 'ai-eshop-optimizer'));
        $card('trust-badges-style', 'palette', __('Style', 'ai-eshop-optimizer'), 1, $fields, array('description' => __('How the badges look on the product page: alignment, text size, colours, spacing and image size.', 'ai-eshop-optimizer')));

        rooster_admin_band_close();
    }
}