<?php
if (!defined('ABSPATH')) exit;

class AdPageX_Core
{
    public function __construct()
    {
        add_filter('the_content', [$this, 'insert_ad_into_content']);
        add_action('template_redirect', [$this, 'replace_body_on_search']);
        add_shortcode('adpagex', [$this, 'adpagex_shortcode']);
        add_action('wp_enqueue_scripts', [$this, 'insert_head_script']);
    }

    private static $instance = null;

    public static function get_instance() {
        if (null === self::$instance) {
            self::$instance = new self();
        }
        return self::$instance;
    }

    /**
     * Enqueues the main Google AdSense for Search script and the tracking snippet.
     * This function should be hooked to 'wp_enqueue_scripts' or 'wp_head'.
     */
    public function insert_head_script()
    {
        $active = $this->get_active_config();
        if ( ! $active || $active === '__disabled__' ) {
            return;
        }

        // Enqueue the main AdSense for Search script
        wp_enqueue_script('google-adsense-search', 'https://www.google.com/adsense/search/ads.js', [], null, false); // phpcs:ignore WordPress.WP.EnqueuedResourceParameters -- external Google CDN, version not applicable

        $inline_tracking_script = "(function(g,o){g[o]=g[o]||function(){(g[o]['q']=g[o]['q']||[]).push(arguments)},g[o]['t']=1*new Date})(window,'_googCsa');";
        wp_add_inline_script('google-adsense-search', $inline_tracking_script, 'after');

        // Placeholder footer handle for dynamic inline scripts added during the_content filter
        wp_register_script('adpagex-init', false, ['google-adsense-search'], false, true); // phpcs:ignore WordPress.WP.EnqueuedResourceParameters
        wp_enqueue_script('adpagex-init');
    }

    /**
     * This function is no longer explicitly echoing scripts.
     * Scripts intended for wp_body_open should be enqueued via wp_enqueue_script
     * with the 'true' parameter for footer loading.
     */
    public function insert_after_body()
    {
        // No direct echoing.
    }

    public function insert_ad_into_content($content)
    {
        static $adpagex_inserted = false;
        if (!is_main_query() || !in_the_loop()) {
            return $content;
        }

        $config = $this->get_active_config();
        if (!$config || $config === '__disabled__') {
            return $content;
        }

        if (isset($config['devices'])) {
            $current_device = $this->detect_device();
            if (!in_array($current_device, $config['devices'])) {
                return $content;
            }
        }

        if (in_array('shortcode', $config['insertion_type']) && has_shortcode($content, 'adpagex')) {
            return $content;
        }

        if (!$this->should_display_on_current_page($config)) {
            return $content;
        }

        // Prepare postQuery variable
        $post_query_var = '';
        if (is_single()) {
            $related_keys = get_post_meta(get_the_ID(), 'related_keys', true);
            if (empty($related_keys)) {
                $related_keys = get_the_title();
            }
            $related_keys = trim(preg_replace('/[^a-zA-Z0-9, ]/', '', $related_keys));
            $related_keys = esc_js($related_keys);
            $style_id = get_post_meta(get_the_ID(), 'styleid', true);
            $post_query_var = "var postQuery = '" . esc_js(trim($related_keys)) . "';";
        } else {
            $post_query_var = "var postQuery = '';";
        }
        $styleIdCustom = "var afsStyleId = '" . esc_js($style_id) . "';";

        if($config['status'] == 'disabled') {
            $post_query_var = "var postQuery = '';";
        }

        $rewarded_needs_readmore = isset($config['rewarded_enabled']) && $config['rewarded_enabled'] === '1'
            && (($config['rewarded_trigger_type'] ?? 'read_more') === 'read_more');

        if ((isset($config['show_readmore']) && $config['show_readmore'] === 'yes') || $rewarded_needs_readmore) {
            // Sử dụng readmore inserter để xử lý toàn bộ (bao gồm cả AFS script)
            $this->add_readmore($content, $config);
            return $content;
        } else {
            // Logic cũ cho trường hợp không dùng readmore
            $ad_html_containers = '
            <div id="adpagex_afscontainer"></div>
            <div id="adpagex_relatedsearches" style="margin-bottom:1rem;"></div>
            ';
            $alignment = isset($config['alignment']) ? $config['alignment'] : 'default';
            if ($alignment !== 'default') {
                $ad_html_containers = '<div style="text-align:' . esc_attr($alignment) . '">' . $ad_html_containers . '</div>';
            }
            $this->render_ad_script($config, $post_query_var, $styleIdCustom);
            return $this->insert_at_position($content, wp_kses_post($ad_html_containers), $config);
        }
    }

    public function add_readmore(&$content, $config)
    {
        // Kiểm tra class readmore inserter có tồn tại không
        if (!class_exists('AdPageX_Inserter')) {
            // Fallback: return content without readmore processing
            return;
        }
        
        // Sử dụng readmore inserter class
        $inserter = AdPageX_Inserter::get_instance();
        
        // Chuẩn bị ad HTML containers
        $ad_html_containers = '
            <div id="adpagex_afscontainer"></div>
            <div id="adpagex_relatedsearches" style="margin-bottom:1rem;"></div>
        ';
        
        $alignment = isset($config['alignment']) ? $config['alignment'] : 'default';
        if ($alignment !== 'default') {
            $ad_html_containers = '<div style="text-align:' . esc_attr($alignment) . '">' . $ad_html_containers . '</div>';
        }
        
        // Prepare postQuery và styleId variables
        $post_query_var = '';
        $style_id = '';
        if (is_single()) {
            $related_keys = get_post_meta(get_the_ID(), 'related_keys', true);
            if (empty($related_keys)) {
                $related_keys = get_the_title();
            }
            $related_keys = trim(preg_replace('/[^a-zA-Z0-9, ]/', '', $related_keys));
            $style_id = get_post_meta(get_the_ID(), 'styleid', true);
            $post_query_var = "var postQuery = '" . esc_js(trim($related_keys)) . "';";
        } else {
            $post_query_var = "var postQuery = '';";
        }
        $styleIdCustom = "var afsStyleId = '" . esc_js($style_id) . "';";

        if($config['status'] == 'disabled') {
            $post_query_var = "var postQuery = '';";
        }
        
        // Render ad script (registers inline JS via wp_add_inline_script, no direct echo)
        $this->render_ad_script($config, $post_query_var, $styleIdCustom);
        
        // Kết hợp ad HTML và script
        $full_ad_html = wp_kses_post($ad_html_containers);

        // Force readmore mode so inserter always uses the with_readmore path
        $config['show_readmore'] = 'yes';

        // Process content với readmore và full ad HTML
        $content = $inserter->process_content($content, $full_ad_html, $config);
    }


    private function get_active_config()
    {
        $active  = get_option('adpagex_active_config');
        $configs = get_option('adpagex_configurations', []);
        if ($active === '__disabled__' || empty($active)) {
            return !empty($configs) ? $configs[0] : '__disabled__';
        }
        foreach ($configs as $cfg) {
            if ($cfg['id'] === $active) {
                return $cfg;
            }
        }
        return null;
    }

    private function should_display_on_current_page($config)
    {
        $enabled_on = isset($config['enabled_on']) ? $config['enabled_on'] : ['posts'];

        if (is_single() && in_array('posts', $enabled_on)) {
            return true;
        }
        if (is_front_page() && in_array('home', $enabled_on)) {
            return true;
        }
        if (is_category() && in_array('category', $enabled_on)) {
            return true;
        }
        if (is_page() && in_array('pages', $enabled_on)) {
            return true;
        }
        if (is_search() && in_array('search', $enabled_on)) {
            return true;
        }
        if ((is_tag() || is_archive()) && in_array('archive', $enabled_on)) {
            return true;
        }

        return false;
    }

    /**
     * Extracts only the raw JavaScript content (without <script> tags)
     * from the configuration.
     *
     * @param array $config The ad placement configuration.
     * @return string The raw JavaScript content.
     */
    private function generate_afs_script($config)
    {
        $full_script_html = isset($config['script']) ? $config['script'] : '';
        
        // Use regex to extract only the content inside the <script> tags
        // This is crucial to avoid WordPress's filtering of full <script> tags in content.
        if (preg_match('/<script[^>]*>(.*?)<\/script>/s', $full_script_html, $matches)) {
            // Remove the <div id="adpagex_afscontainer"></div> and <div id="adpagex_relatedsearches"></div>
            // from the extracted script content if they were part of the 'script' field
            $js_content = $matches[1];
            // Remove initial HTML divs if they were mistakenly included in the script field
            $js_content = preg_replace('/^\s*<div id="adpagex_afscontainer"><\/div>\s*<div id="adpagex_relatedsearches" style="margin-bottom:1rem;"><\/div>\s*/', '', $js_content);
            return trim($js_content);
        } else {
            // If no script tags found, assume the entire content is raw JS (less common but handles cases)
            // Still attempt to remove HTML divs just in case
            $full_script_html = preg_replace('/^\s*<div id="adpagex_afscontainer"><\/div>\s*<div id="adpagex_relatedsearches" style="margin-bottom:1rem;"><\/div>\s*/', '', $full_script_html);
            return trim($full_script_html);
        }
    }

    // Chữ ký hàm insert_at_position đã được sửa để nhận $config
    private function insert_at_position($content, $html_to_insert, $config)
    {
        // Kiểm tra class readmore inserter có tồn tại không
        if (!class_exists('AdPageX_Inserter')) {
            // Fallback: return content without readmore processing
            return;
        }
        
        // Sử dụng readmore inserter class
        $inserter = AdPageX_Inserter::get_instance();
        return $inserter->process_content($content, $html_to_insert, $config);
    }


    public function replace_body_on_search()
    {
        if (
            is_search()
            && isset($_GET['query'], $_GET['rsToken'], $_GET['styleId']) // phpcs:ignore WordPress.Security.NonceVerification.Recommended
        ) {
            $configs = get_option('adpagex_configurations', []);
            $config = null;

            // Tìm config theo styleId
            foreach ($configs as $cfg) {
                if (
                    isset($cfg['styleid']) &&
                    $cfg['styleid'] == sanitize_text_field(wp_unslash($_GET['styleId'])) // phpcs:ignore WordPress.Security.NonceVerification.Recommended
                ) {
                    $config = $cfg;
                    break;
                }
            }
            // Nếu không tìm thấy, lấy config active
            if (!$config) {
                $config = $this->get_active_config();
            }

            $ad_html_containers = '
                <div id="adpagex_afscontainer"></div>
                <div id="adpagex_relatedsearches" style="margin-bottom:1rem;"></div>
            ';

            if (ob_get_length()) ob_clean();

            echo '<!DOCTYPE html>
            <html ' . esc_attr(get_language_attributes()) . '>
            <head>
                <meta charset="' . esc_attr(get_bloginfo('charset')) . '">
                <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1">
                <title>' . esc_html(get_bloginfo('name')) . ' - Search</title>';

                $this->insert_head_script();
                wp_head();
            echo '</head>
            <body class="adpagex-search-override">';

            // Chuẩn bị biến postQuery cho JS
            $post_query_var = "var postQuery = '" . esc_js(sanitize_text_field(wp_unslash($_GET['query'] ?? ''))) . "';"; // phpcs:ignore WordPress.Security.NonceVerification.Recommended

            // Render script (ưu tiên script_file, fallback về script cũ)
            $this->render_ad_script($config, $post_query_var);

            echo '<div class="wp-site-blocks"><div class="entry-content alignfull wp-block-post-content has-global-padding is-layout-constrained wp-block-post-content-is-layout-constrained"><h2 style="padding-bottom:var(--wp--preset--spacing--60);padding-top:var(--wp--preset--spacing--60);" class="wp-block-post-title">Search Results</h2>';
            echo wp_kses_post($ad_html_containers);

            echo '</div></div>';
            wp_footer();
            echo '</body></html>';
            exit;
        }
    }

    public function adpagex_shortcode($atts)
    {
        $atts = shortcode_atts([
            'id' => '',
        ], $atts);
        
        if (empty($atts['id'])) {
            return '';
        }
        
        return $this->get_ad_by_id($atts['id'], 'shortcode');
    }

    /**
     * Get ad by configuration ID
     *
     * @param string $id Configuration ID
     * @param string $type Insertion type ('shortcode' or 'php')
     * @return string Generated ad HTML (placeholder div)
     */
    public function get_ad_by_id($id, $type = 'shortcode')
    {
        static $adpagex_inserted = false;
        if ($adpagex_inserted && $type !== 'php') {
            return '';
        }

        $configs = get_option('adpagex_configurations', []);
        $config = null;

        foreach ($configs as $cfg) {
            if ($cfg['id'] === $id && in_array($type, $cfg['insertion_type'])) {
                $config = $cfg;
                break;
            }
        }

        if (!$config) {
            return '';
        }

        if (isset($config['devices'])) {
            $current_device = $this->detect_device();
            if (!in_array($current_device, $config['devices'])) {
                return '';
            }
        }

        $post_query_var = '';
        if (is_single()) {
            $related_keys = get_post_meta(get_the_ID(), 'related_keys', true);
            if (empty($related_keys)) {
                $related_keys = get_the_title();
            }
            $style_id = get_post_meta(get_the_ID(), 'styleid', true);
            $post_query_var = "var postQuery = '" . esc_js(trim($related_keys)) . "';";
        } else {
            $post_query_var = "var postQuery = '';";
        }
        $styleIdCustom = "var afsStyleId = '" . esc_js($style_id) . "';";
        

        $ad_html_containers = '
            <div id="adpagex_afscontainer"></div>
            <div id="adpagex_relatedsearches" style="margin-bottom:1rem;"></div>
        ';
        $alignment = isset($config['alignment']) ? $config['alignment'] : 'default';
        if ($alignment !== 'default') {
            $ad_html_containers = '<div style="text-align:' . esc_attr($alignment) . '">' . $ad_html_containers . '</div>';
        }

        $this->render_ad_script($config, $post_query_var, $styleIdCustom);

        if ($type !== 'php') {
            $adpagex_inserted = true;
        }

        return wp_kses_post($ad_html_containers);
    }

    private function detect_device() {
        $user_agent = '';
        if (isset($_SERVER['HTTP_USER_AGENT'])) {
            $user_agent = sanitize_text_field(wp_unslash($_SERVER['HTTP_USER_AGENT']));
        }

        if (preg_match('/(android|iphone|ipod|iemobile)/i', $user_agent)) {
            if (preg_match('/(tablet|ipad|playbook|silk)/i', $user_agent)) {
                return 'tablet';
            }
            return 'mobile';
        }
        
        return 'desktop';
    }

    private function get_script_file_url($config) {
        if (!empty($config['script_file'])) {
            $version = !empty($config['script_file_version']) ? $config['script_file_version'] : time();
            return $config['script_file'] . '?v=' . $version;
        }
        return '';
    }

    private function render_ad_script($config, $post_query_var = '', $styleIdCustom = '') {
        $init_handle = 'adpagex-init';

        // Ensure handle is registered regardless of when this method is called.
        // wp_add_inline_script() silently drops scripts for unregistered handles.
        if ( ! wp_script_is( $init_handle, 'registered' ) ) {
            wp_register_script( $init_handle, false, [], false, true ); // phpcs:ignore WordPress.WP.EnqueuedResourceParameters
        }
        if ( ! wp_script_is( $init_handle, 'enqueued' ) ) {
            wp_enqueue_script( $init_handle );
        }

        if (!empty($config['script_content'])) {
            $js_content   = $post_query_var . $styleIdCustom . base64_decode( $config['script_content'] ); // phpcs:ignore WordPress.PHP.DiscouragedPHPFunctions.obfuscation_base64_decode
            $js_sanitized = str_replace( '</script>', '<\/script>', $js_content );
            wp_add_inline_script( $init_handle, $js_sanitized );

        } elseif (!empty($config['script_file'])) {
            $ext_handle = 'adpagex-ext-' . sanitize_key($config['id']);
            wp_enqueue_script($ext_handle, esc_url_raw($config['script_file']), [], null, true); // phpcs:ignore WordPress.WP.EnqueuedResourceParameters
            if (!empty($post_query_var) || !empty($styleIdCustom)) {
                wp_add_inline_script($ext_handle, $post_query_var . $styleIdCustom, 'before');
            }

        } elseif (!empty($config['script'])) {
            $raw_ads_js = str_replace( '</script>', '<\/script>', $this->generate_afs_script( $config ) );
            wp_add_inline_script( $init_handle, $post_query_var . $styleIdCustom . "\n" . $raw_ads_js );
        }
    }
}

/**
 * PHP helper function to display ad by configuration ID
 *
 * @param string $id Configuration ID
 * @return string Generated ad HTML (placeholder div)
 */
function adpagex($id) {
    $core = AdPageX_Core::get_instance();
    // Ensure 'php' type allows multiple insertions if needed, as per the updated get_ad_by_id logic
    return $core->get_ad_by_id($id, 'php');
}