<?php
/**
 * Elementor integration handler for AILinkMaster.
 */

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

class AILM_Elementor_Handler {

    /**
     * Check if a post is built with Elementor.
     *
     * @param int $post_id
     * @return bool
     */
    public static function is_elementor_post( $post_id ) {
        return get_post_meta( $post_id, '_elementor_edit_mode', true ) === 'builder';
    }

    /**
     * Process Elementor data using a callback that manipulates HTML.
     * The callback receives raw HTML, uses AILM_DOM_Helper, and returns new HTML.
     * 
     * @param int $post_id
     * @param callable $callback function( $html ) { return $new_html; }
     * @return bool True if modifications were made and saved.
     */
    public static function process_content( $post_id, $callback ) {
        if ( ! self::is_elementor_post( $post_id ) ) {
            return false;
        }

        // Elementor uses the \Elementor\Plugin instance to save data properly
        if ( ! class_exists( '\Elementor\Plugin' ) ) {
            return false;
        }

        $document = \Elementor\Plugin::$instance->documents->get( $post_id );
        if ( ! $document ) {
            return false;
        }

        $elements_data = $document->get_elements_data();
        if ( empty( $elements_data ) ) {
            return false;
        }

        $modified = self::traverse_and_modify( $elements_data, $callback );

        if ( $modified ) {
            // Save back through Elementor API. This updates both _elementor_data and post_content
            $document->save( [
                'elements' => $elements_data,
            ] );
            return true;
        }

        return false;
    }

    /**
     * Recursively traverse Elementor elements and apply the callback to text widgets.
     *
     * @param array $elements
     * @param callable $callback
     * @return bool
     */
    private static function traverse_and_modify( &$elements, $callback ) {
        $modified = false;

        foreach ( $elements as &$element ) {
            // Check if this is a widget that contains text we want to modify
            if ( isset( $element['elType'] ) && $element['elType'] === 'widget' ) {
                $widget_type = $element['widgetType'];
                
                if ( $widget_type === 'text-editor' && isset( $element['settings']['editor'] ) ) {
                    $old_html = $element['settings']['editor'];
                    $new_html = call_user_func( $callback, $old_html );
                    if ( $new_html !== $old_html ) {
                        $element['settings']['editor'] = $new_html;
                        $modified = true;
                    }
                } elseif ( $widget_type === 'heading' && isset( $element['settings']['title'] ) ) {
                    $old_html = $element['settings']['title'];
                    $new_html = call_user_func( $callback, $old_html );
                    if ( $new_html !== $old_html ) {
                        $element['settings']['title'] = $new_html;
                        $modified = true;
                    }
                }
            }

            // Recurse into children
            if ( ! empty( $element['elements'] ) ) {
                if ( self::traverse_and_modify( $element['elements'], $callback ) ) {
                    $modified = true;
                }
            }
        }

        return $modified;
    }
}