<?php
if (!defined('ABSPATH')) exit;

add_action('wp_ajax_aicont_generate_and_save_seo_title', 'aicont_generate_and_save_seo_title_handler');

function aicont_generate_and_save_seo_title_handler() {
    check_ajax_referer('aicont_nonce', 'nonce');

    $post_id = absint($_POST['post_id'] ?? 0);
    $keyword = sanitize_text_field($_POST['keyword'] ?? '');
    $custom_prompt = sanitize_textarea_field($_POST['custom_prompt'] ?? '');

    if (!$post_id || !get_post($post_id)) {
        wp_send_json_error(['code' => 'err_post_id_invalid']);
    }
    if (empty($keyword)) {
        wp_send_json_error(['code' => 'err_keyword_empty']);
    }

    $post_title = get_the_title($post_id);
    if (empty($post_title)) {
        wp_send_json_error(['code' => 'err_post_title_missing']);
    }

    if (empty($custom_prompt)) {
        $custom_prompt = get_post_meta($post_id, 'aicont_singlepost_seo_title_prompt_custom', true);
    }
    if (empty($custom_prompt)) {
        $custom_prompt = get_option('aicont_plugin_seo_title_prompt_custom', '');
    }

    $active_license = get_option('aicont_active_license', '');

    $result = aicont_call_webtinus_seo_title([
        'keyword'     => $keyword,
        'post_id'     => $post_id,
        'post_title'  => $post_title,
        'prompt'      => $custom_prompt,
        'api_key'     => $active_license
    ]);

    if (is_wp_error($result)) {
        wp_send_json_error(['code' => 'err_connection_failed']);
    }

    if (!empty($result['error_code'])) {
        wp_send_json_error(['code' => $result['error_code']]);
    }

    if (empty($result['seo_title'])) {
        wp_send_json_error(['code' => 'err_content_empty']);
    }

    $seo_title = wp_strip_all_tags(sanitize_text_field($result['seo_title']));

    if (empty($seo_title)) {
        wp_send_json_error(['code' => 'err_content_empty']);
    }

    if (class_exists('WPSEO_Meta')) {
        aicont_save_yoast_seo_title($post_id, $seo_title);
    }
    elseif (defined('RANK_MATH_VERSION') || class_exists('RankMath\Core')) {
        aicont_save_rankmath_seo_title($post_id, $seo_title);
    }
    else {
        wp_send_json_error(['code' => 'err_no_seo_plugin']);
    }

    wp_update_post([
        'ID' => $post_id,
        'post_modified'     => current_time('mysql'),
        'post_modified_gmt' => current_time('mysql', 1)
    ]);

    wp_send_json_success([
        'message' => 'success_seo_title_generated',
        'seo_title' => $seo_title
    ]);
}

function aicont_save_yoast_seo_title($post_id, $seo_title) {
    update_post_meta($post_id, '_yoast_wpseo_title', $seo_title);
    
    $saved = get_post_meta($post_id, '_yoast_wpseo_title', true);
    if ($saved !== $seo_title) {
        wp_die('Failed to save Yoast SEO title');
    }
}

function aicont_save_rankmath_seo_title($post_id, $seo_title) {
    if (!function_exists('rank_math')) {
        wp_die('Rank Math is not active');
    }

    if (method_exists('\RankMath\Helper', 'update_post_meta')) {
        \RankMath\Helper::update_post_meta('rank_math_title', $seo_title, $post_id);
    } 
    else {
        update_post_meta($post_id, 'rank_math_title', $seo_title);
    }

    $saved = get_post_meta($post_id, 'rank_math_title', true);
    if ($saved !== $seo_title) {
        do_action('rank_math/before_save_post', $post_id);
        update_post_meta($post_id, 'rank_math_title', $seo_title);
        do_action('rank_math/after_save_post', $post_id);
    }
}

function aicont_call_webtinus_seo_title($args) {
    $client_site_url = home_url('', 'https');
    $server_url = 'https://webtinus.com/wp-json/aicont/v1/generate_seo_title1';

    $body = wp_json_encode([
        'keyword'         => $args['keyword'],
        'post_id'         => $args['post_id'],
        'post_title'      => $args['post_title'],
        'prompt'          => $args['prompt'],
        'api_key'         => $args['api_key'] ?? '',
        'client_site_url' => $client_site_url,
        'site_title'      => get_bloginfo('name'),
        'site_language'   => get_locale()
    ], JSON_UNESCAPED_UNICODE | JSON_UNESCAPED_SLASHES);

    $response = wp_remote_post($server_url, [
        'headers' => [
            'Content-Type' => 'application/json',
            'X-Client-Site' => $client_site_url
        ],
        'body'    => $body,
        'timeout' => 90
    ]);

    if (is_wp_error($response)) {
        return ['error_code' => 'err_connection_failed'];
    }

    $http_code = wp_remote_retrieve_response_code($response);
    
    if ($http_code !== 200) {
        return ['error_code' => 'err_server_error'];
    }

    $body_response = wp_remote_retrieve_body($response);
    
    if (empty($body_response)) {
        return ['error_code' => 'err_content_empty'];
    }

    $data = json_decode($body_response, true);

    if (!is_array($data) || json_last_error() !== JSON_ERROR_NONE) {
        return ['error_code' => 'err_server_error'];
    }

    if (empty($data['seo_title'])) {
        return ['error_code' => $data['error_code'] ?? 'err_content_empty'];
    }

    return ['seo_title' => trim($data['seo_title'])];
}