<?php
if (!defined('ABSPATH')) exit;

add_action('wp_ajax_ai_analyze','ai_analyze');

function ai_analyze(){

check_ajax_referer('ai_ajax_nonce','nonce');

$keyword = sanitize_text_field($_POST['keyword']);

$urls = ai_get_top3_results($keyword);

if(empty($urls)){
echo "❌ Không lấy được dữ liệu SERP";
wp_die();
}

$h1 = ai_extract_headers($urls[0] ?? '');
$h2 = ai_extract_headers($urls[1] ?? '');
$h3 = ai_extract_headers($urls[2] ?? '');

$max = max(count($h1),count($h2),count($h3));

echo '<div class="ai-serp-table">';
echo '<table class="widefat">';

echo '<tr>
<th>Top 1</th>
<th>Top 2</th>
<th>Top 3</th>
<th>Custom Outline</th>
</tr>';

for($i=0;$i<$max;$i++){

echo '<tr>';

foreach([$h1,$h2,$h3] as $col){

$item = $col[$i] ?? null;

if($item){

echo '<td>
<span class="ai-tag">'.$item['tag'].'</span>
'.esc_html($item['text']).'
</td>';

}else{

echo '<td></td>';

}

}

if($i == 0){

echo '<td rowspan="'.$max.'">
<textarea id="ai-outline" name="outline"
style="width:100%;height:400px"
placeholder="Tự viết outline tại đây...

Ví dụ:

H1: '.$keyword.'

H2: Giới thiệu
H2: Cách thực hiện
H3: Bước 1
H3: Bước 2
"></textarea>
</td>';

}

echo '</tr>';

}

echo '</table>';
echo '</div>';

echo '<button id="ai-next" class="button button-primary">Tiếp tục</button>';

wp_die();
}


/* ===== SERP ===== */

function ai_get_top3_results($keyword){

$key = get_option('ai_seo_serp_key');

if(empty($key)) return [];

$url = 'https://serpapi.com/search.json?q='.urlencode($keyword).'&num=10&api_key='.$key;

$response = wp_remote_get($url);

if(is_wp_error($response)) return [];

$body = json_decode(wp_remote_retrieve_body($response),true);

$results = [];

if(!empty($body['organic_results'])){

foreach($body['organic_results'] as $item){

$link = $item['link'];

if(strpos($link,'shopee') !== false) continue;
if(strpos($link,'lazada') !== false) continue;

$results[] = $link;

if(count($results) == 3) break;

}

}

return $results;

}


/* ===== HEADER EXTRACT ===== */

function ai_extract_headers($url){

if(empty($url)) return [];

$response = wp_remote_get($url);

if(is_wp_error($response)) return [];

$html = wp_remote_retrieve_body($response);

libxml_use_internal_errors(true);

$dom = new DOMDocument();

if(!$dom->loadHTML($html)) return [];

$xpath = new DOMXPath($dom);

$headers = [];

foreach(['h1','h2','h3'] as $tag){

$nodes = $xpath->query("//$tag");

foreach($nodes as $node){

$text = trim($node->nodeValue);

if(!empty($text)){

$headers[] = [
'tag'=>strtoupper($tag),
'text'=>$text
];

}

}

}

return $headers;

}﻿<?php
if (!defined('ABSPATH')) exit;

add_action('wp_ajax_ai_generate','ai_generate');

function ai_generate(){

check_ajax_referer('ai_ajax_nonce','nonce');

$data = $_POST['data'];
parse_str($data,$form);

$keyword = sanitize_text_field($form['keyword']);
$type = sanitize_text_field($form['type']);
$tone = sanitize_text_field($form['tone']);
$length = sanitize_text_field($form['length']);
$post_type_style = sanitize_text_field($form['post_type']);
$audience = sanitize_text_field($form['audience']);
$highlight = sanitize_text_field($form['highlight']);
$outline = sanitize_textarea_field($form['outline']);

if(empty($outline)){

echo "❌ Bạn chưa nhập Custom Outline";
wp_die();

}

$prompt = "
Viết bài SEO hoàn chỉnh.

Keyword: $keyword
Phong cách: $tone
Độ dài: $length
Loại bài: $post_type_style
Audience: $audience
Điểm nổi bật: $highlight

Outline:
$outline

Yêu cầu:

- Có Title
- Có Meta description
- Heading rõ ràng H1 H2 H3
- Văn phong tự nhiên
- Chuẩn SEO
- Có CTA cuối bài
";

$content = ai_call_gemini($prompt);

if(strpos($content,'❌') !== false){

echo $content;
wp_die();

}

$post_type = ($type == 'product') ? 'product' : 'post';

$post_id = wp_insert_post([

'post_title'=>$keyword,
'post_content'=>$content,
'post_status'=>'draft',
'post_type'=>$post_type,
'post_author'=>get_current_user_id()

]);

echo "<h3>✅ Draft đã tạo</h3>";

echo "<a class='button button-primary'
href='".admin_url("post.php?post=$post_id&action=edit")."'
target='_blank'>Mở bài viết</a>";

echo "<textarea style='width:100%;height:400px'>".esc_textarea($content)."</textarea>";

wp_die();

}


/* ===== GEMINI API ===== */

function ai_call_gemini($prompt){

$key = get_option('ai_seo_gemini_key');

if(empty($key)) return "❌ Chưa cấu hình Gemini API key";

$url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=".$key;

$body = [

"contents"=>[
[
"parts"=>[
["text"=>$prompt]
]
]
]

];

$response = wp_remote_post($url,[

'headers'=>[
'Content-Type'=>'application/json'
],

'body'=>json_encode($body),
'timeout'=>60

]);

if(is_wp_error($response)){

return "❌ Lỗi kết nối Gemini";

}

$data = json_decode(wp_remote_retrieve_body($response),true);

if(empty($data['candidates'][0]['content']['parts'][0]['text'])){

return "❌ Gemini không trả dữ liệu";

}

return $data['candidates'][0]['content']['parts'][0]['text'];

}