<?php
/**
 * Media Handler
 * No images are stored in WordPress — everything is delivered from the backend.
 * Featured images use post meta 'affirm_featured_image_id' instead of WP attachments.
 * Adds "aFFirmFirst Vault" tab to media modals for selecting featured images.
 */
class Affirm_Media_Handler {

    public function __construct() {
        // Media modal vault tab
        add_action('admin_enqueue_scripts', array($this, 'enqueue_media_vault_scripts'));

        // AJAX: set/get featured image from vault
        add_action('wp_ajax_affirm_set_featured_image', array($this, 'ajax_set_featured_image'));
        add_action('wp_ajax_affirm_remove_featured_image', array($this, 'ajax_remove_featured_image'));
        add_action('wp_ajax_affirm_get_featured_image', array($this, 'ajax_get_featured_image'));

        // AJAX: WooCommerce product gallery
        add_action('wp_ajax_affirm_save_product_gallery', array($this, 'ajax_save_product_gallery'));
        add_action('wp_ajax_affirm_get_product_gallery', array($this, 'ajax_get_product_gallery'));

        // AJAX: Toggle image delivery
        add_action('wp_ajax_affirm_toggle_image_delivery', array($this, 'ajax_toggle_image_delivery'));

        // Featured image metabox override
        add_filter('admin_post_thumbnail_html', array($this, 'override_featured_image_metabox'), 10, 3);
    }

    /**
     * Enqueue vault browser scripts on screens that use the media modal
     */
    public function enqueue_media_vault_scripts($hook) {
        // Load on ALL post editing screens (post, page, product, custom post types)
        // Also load on media, widgets, and customizer screens
        $allowed_hooks = array(
            'post.php', 'post-new.php',          // Posts, pages, CPTs, products
            'upload.php', 'media-new.php',        // Media library
            'widgets.php',                         // Widgets
            'customize.php',                       // Customizer
            'term.php', 'edit-tags.php',          // Taxonomy editors
            'site-editor.php',                     // Full site editor
        );
        if (!in_array($hook, $allowed_hooks)) return;
        if (!get_option('affirm_api_key')) return;

        wp_enqueue_media();
        wp_enqueue_style('affirm-media-vault', AFFIRM_PLUGIN_URL . 'assets/css/admin.css', array(), AFFIRM_VERSION);
        wp_enqueue_script('affirm-media-vault', AFFIRM_PLUGIN_URL . 'assets/js/media-vault.js', array('jquery', 'media-views'), AFFIRM_VERSION, true);

        global $post;
        $post_id = ($post && isset($post->ID)) ? $post->ID : 0;
        $affirm_featured = $post_id ? get_post_meta($post_id, 'affirm_featured_image_id', true) : '';
        $post_type = $post_id ? get_post_type($post_id) : '';

        wp_localize_script('affirm-media-vault', 'affirmVault', array(
            'ajaxUrl'    => admin_url('admin-ajax.php'),
            'nonce'      => wp_create_nonce('affirm_admin_nonce'),
            'postId'     => $post_id,
            'postType'   => $post_type,
            'backendUrl' => affirm_get_backend_url(),
            'featuredImageId' => $affirm_featured,
        ));
    }

    /**
     * AJAX: Set a vault image as the featured image for a post
     */
    public function ajax_set_featured_image() {
        check_ajax_referer('affirm_admin_nonce', 'nonce');
        if (!current_user_can('edit_posts')) {
            wp_send_json_error(array('message' => 'Unauthorized'));
        }

        $post_id  = isset($_POST['post_id']) ? absint($_POST['post_id']) : 0;
        $image_id = isset($_POST['image_id']) ? sanitize_text_field(wp_unslash($_POST['image_id'])) : '';
        $thumb_url = isset($_POST['thumbnail_url']) ? esc_url_raw(wp_unslash($_POST['thumbnail_url'])) : '';
        $filename = isset($_POST['filename']) ? sanitize_text_field(wp_unslash($_POST['filename'])) : '';

        if (!$post_id || !$image_id) {
            wp_send_json_error(array('message' => 'Post ID and Image ID required'));
        }

        // Store vault image ID in post meta (no WP attachment created)
        update_post_meta($post_id, 'affirm_featured_image_id', $image_id);
        update_post_meta($post_id, 'affirm_featured_thumbnail_url', $thumb_url);
        update_post_meta($post_id, 'affirm_featured_filename', $filename);

        // Remove any standard WP featured image
        delete_post_thumbnail($post_id);

        wp_send_json_success(array(
            'imageId'      => $image_id,
            'thumbnailUrl' => $thumb_url,
            'filename'     => $filename,
        ));
    }

    /**
     * AJAX: Remove the vault featured image from a post
     */
    public function ajax_remove_featured_image() {
        check_ajax_referer('affirm_admin_nonce', 'nonce');
        if (!current_user_can('edit_posts')) {
            wp_send_json_error(array('message' => 'Unauthorized'));
        }

        $post_id = isset($_POST['post_id']) ? absint($_POST['post_id']) : 0;
        if (!$post_id) {
            wp_send_json_error(array('message' => 'Post ID required'));
        }

        delete_post_meta($post_id, 'affirm_featured_image_id');
        delete_post_meta($post_id, 'affirm_featured_thumbnail_url');
        delete_post_meta($post_id, 'affirm_featured_filename');

        wp_send_json_success();
    }

    /**
     * AJAX: Get the current vault featured image for a post
     */
    public function ajax_get_featured_image() {
        check_ajax_referer('affirm_admin_nonce', 'nonce');

        $post_id = isset($_POST['post_id']) ? absint($_POST['post_id']) : 0;
        if (!$post_id) {
            wp_send_json_error(array('message' => 'Post ID required'));
        }

        $image_id = get_post_meta($post_id, 'affirm_featured_image_id', true);
        if (!$image_id) {
            wp_send_json_error(array('message' => 'No vault featured image'));
        }

        wp_send_json_success(array(
            'imageId'      => $image_id,
            'thumbnailUrl' => get_post_meta($post_id, 'affirm_featured_thumbnail_url', true),
            'filename'     => get_post_meta($post_id, 'affirm_featured_filename', true),
        ));
    }

    /**
     * AJAX: Save vault image IDs for WooCommerce product gallery
     */
    public function ajax_save_product_gallery() {
        check_ajax_referer('affirm_admin_nonce', 'nonce');
        if (!current_user_can('edit_posts')) {
            wp_send_json_error(array('message' => 'Unauthorized'));
        }

        $post_id   = isset($_POST['post_id']) ? absint($_POST['post_id']) : 0;
        $image_ids = isset($_POST['image_ids']) ? sanitize_text_field(wp_unslash($_POST['image_ids'])) : '[]';

        if (!$post_id) {
            wp_send_json_error(array('message' => 'Post ID required'));
        }

        $ids = json_decode(stripslashes($image_ids), true);
        if (!is_array($ids)) $ids = array();

        // Sanitize each ID
        $ids = array_map('sanitize_text_field', $ids);

        update_post_meta($post_id, '_affirm_product_gallery', $ids);

        wp_send_json_success(array('saved' => count($ids)));
    }

    /**
     * AJAX: Get saved vault gallery image IDs for a product
     */
    public function ajax_get_product_gallery() {
        check_ajax_referer('affirm_admin_nonce', 'nonce');

        $post_id = isset($_POST['post_id']) ? absint($_POST['post_id']) : 0;
        if (!$post_id) {
            wp_send_json_error(array('message' => 'Post ID required'));
        }

        $ids = get_post_meta($post_id, '_affirm_product_gallery', true);
        if (!is_array($ids)) $ids = array();

        wp_send_json_success(array('image_ids' => $ids));
    }

    /**
     * AJAX: Toggle image delivery on/off
     */
    public function ajax_toggle_image_delivery() {
        check_ajax_referer('affirm_admin_nonce', 'nonce');
        if (!current_user_can('edit_posts')) {
            wp_send_json_error(array('message' => 'Unauthorized'));
        }

        $image_id = isset($_POST['image_id']) ? sanitize_text_field(wp_unslash($_POST['image_id'])) : '';
        $enabled  = isset($_POST['enabled']) ? filter_var(wp_unslash($_POST['enabled']), FILTER_VALIDATE_BOOLEAN) : false;

        if (!$image_id) {
            wp_send_json_error(array('message' => 'Image ID required'));
        }

        $result = Affirm_Api::toggle_image_delivery($image_id, $enabled);

        if (is_wp_error($result)) {
            wp_send_json_error(array('message' => $result->get_error_message()));
        }

        wp_send_json_success(array(
            'message' => $enabled ? 'Delivery enabled' : 'Delivery disabled',
            'enabled' => $enabled
        ));
    }

    /**
     * Override the Featured Image metabox to show vault image if set
     */
    public function override_featured_image_metabox($content, $post_id, $thumbnail_id) {
        $affirm_id = get_post_meta($post_id, 'affirm_featured_image_id', true);
        if (!$affirm_id) return $content;

        $thumb_url = get_post_meta($post_id, 'affirm_featured_thumbnail_url', true);
        $filename  = get_post_meta($post_id, 'affirm_featured_filename', true);

        $html = '<div id="affirm-featured-image-preview" style="margin-bottom: 12px;">';
        if ($thumb_url) {
            $html .= '<img src="' . esc_url($thumb_url) . '" alt="' . esc_attr($filename) . '" style="max-width: 100%; height: auto; border-radius: 6px; border: 2px solid #10b981;" />';
        }
        $html .= '<p style="margin: 8px 0 4px; font-size: 12px; color: #10b981; font-weight: 600;">';
        $html .= '<span class="dashicons dashicons-shield" style="font-size: 14px; vertical-align: middle;"></span> aFFirmFirst Protected';
        $html .= '</p>';
        $html .= '<p style="margin: 0; font-size: 11px; color: #64748b;">' . esc_html($filename) . '</p>';
        $html .= '</div>';
        $html .= '<p><a href="#" id="affirm-remove-featured" style="color: #ef4444;">Remove featured image</a></p>';
        $html .= '<p><a href="#" class="affirm-set-featured-link">Set aFFirmFirst image</a></p>';

        return $html;
    }
}