<?php
/**
 * 2.44 — AEO drafts: the platform writes a short answer (TL;DR) and up to
 * three FAQ from the page's own text; the site owner reads them in the SEO
 * box and accepts or discards them. Nothing is published without a click.
 *
 * The draft lives in post meta until it is used or discarded. The
 * platform drops any sentence whose numbers are not on the page and never
 * proposes a question the page already answers; the box says how many
 * drafts were dropped so the owner knows the filter is working.
 */

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

define('CITAMI_META_AEO_DRAFT', '_citami_aeo_draft');
define('CITAMI_AEO_DRAFT_TEXT_MIN', 120);
define('CITAMI_AEO_DRAFT_TEXT_MAX', 12000);
define('CITAMI_AEO_DRAFT_NOTICE', 'citami_aeo_draft_notice_');

/* ---------------------------------------------------------------- */
/*  Storage                                                           */
/* ---------------------------------------------------------------- */

/** The pending draft for a post, or null. */
function citami_aeo_draft_get($post_id) {
    $d = get_post_meta((int) $post_id, CITAMI_META_AEO_DRAFT, true);
    if (!is_array($d)) {
        return null;
    }
    $tldr = isset($d['tldr']) ? trim((string) $d['tldr']) : '';
    $faq = array();
    foreach ((array) (isset($d['faq']) ? $d['faq'] : array()) as $it) {
        if (is_array($it) && !empty($it['q']) && !empty($it['a'])) {
            $faq[] = array('q' => (string) $it['q'], 'a' => (string) $it['a']);
        }
    }
    if ($tldr === '' && empty($faq)) {
        return null;
    }
    return array('tldr' => $tldr, 'faq' => $faq, 'dropped' => isset($d['dropped']) ? (int) $d['dropped'] : 0, 'ts' => isset($d['ts']) ? (int) $d['ts'] : 0);
}

function citami_aeo_draft_discard($post_id) {
    delete_post_meta((int) $post_id, CITAMI_META_AEO_DRAFT);
}

/** Title + plain text the platform will work on (the saved content, never the editor state). */
function citami_aeo_draft_page_text($post_id) {
    $post = get_post((int) $post_id);
    if (!$post) {
        return array('title' => '', 'text' => '');
    }
    $html = (string) $post->post_content;
    if (function_exists('do_blocks') && strpos($html, '<!-- wp:') !== false) {
        $html = do_blocks($html);
    }
    $text = function_exists('citami_aeo_plain') ? citami_aeo_plain($html) : trim(preg_replace('/\s+/u', ' ', wp_strip_all_tags($html)));
    // The TL;DR and FAQ already on the page are not "page text": the model must not paraphrase its own past output.
    return array('title' => trim((string) get_the_title($post)), 'text' => mb_substr($text, 0, CITAMI_AEO_DRAFT_TEXT_MAX));
}

/* ---------------------------------------------------------------- */
/*  Platform call                                                     */
/* ---------------------------------------------------------------- */

/**
 * Asks the platform for a draft and stores it. Returns ok/message; the
 * message is meant for the owner (quota, too little text, network).
 */
function citami_aeo_draft_request($post_id, $kind = 'both') {
    $post_id = (int) $post_id;
    if (!get_post($post_id)) {
        return array('ok' => false, 'message' => citami_t('Pagina non trovata.', 'Page not found.'));
    }
    if (citami_get_api_key() === '') {
        return array('ok' => false, 'message' => citami_t('Collega prima il sito alla piattaforma (Impostazioni → API key).', 'Connect the site to the platform first (Settings → API key).'));
    }
    $page = citami_aeo_draft_page_text($post_id);
    if (mb_strlen($page['text']) < CITAMI_AEO_DRAFT_TEXT_MIN) {
        return array('ok' => false, 'message' => citami_t('La pagina ha troppo poco testo salvato per scriverci sopra una bozza: aggiungi contenuto e salva.', 'The page has too little saved text to draft from: add content and save.'));
    }
    // 2.63 — the questions already on the page (ours and a Yoast / Rank Math / text FAQ) are not written again.
    $existing_items = function_exists('citami_content_faq') ? citami_content_faq($post_id) : array();
    if (function_exists('citami_content_existing_faq') && ($p = get_post($post_id))) {
        $existing_items = array_merge($existing_items, citami_content_existing_faq((string) $p->post_content)['items']);
    }
    $existing = array_values(array_unique(array_map(function ($it) { return $it['q']; }, $existing_items)));
    $kind = in_array($kind, array('tldr', 'faq', 'both'), true) ? $kind : 'both';
    $res = wp_remote_post(citami_get_endpoint() . '/api/plugin/aeo-draft', array(
        'timeout' => 60,
        'headers' => array(
            'Authorization' => 'Bearer ' . citami_get_api_key(),
            'Content-Type'  => 'application/json',
            'User-Agent'    => 'CitamiConnector/' . CITAMI_PLUGIN_VERSION . '; ' . home_url('/'),
        ),
        'body' => wp_json_encode(array(
            'post_id'            => $post_id,
            'title'              => $page['title'] !== '' ? $page['title'] : '(senza titolo)',
            'text'               => $page['text'],
            'kind'               => $kind,
            'locale'             => citami_lang(),
            'existing_questions' => array_slice($existing, 0, 30),
        ), JSON_UNESCAPED_SLASHES | JSON_UNESCAPED_UNICODE | JSON_HEX_TAG),
    ));
    if (is_wp_error($res)) {
        return array('ok' => false, 'message' => sprintf(citami_t('Piattaforma non raggiungibile (%s).', 'Platform unreachable (%s).'), $res->get_error_message()));
    }
    $code = (int) wp_remote_retrieve_response_code($res);
    $data = json_decode((string) wp_remote_retrieve_body($res), true);
    if ($code === 429) {
        return array('ok' => false, 'message' => citami_t('Troppe richieste in un minuto: riprova tra poco.', 'Too many requests in a minute: try again shortly.'));
    }
    if ($code < 200 || $code >= 300 || !is_array($data)) {
        return array('ok' => false, 'message' => sprintf(citami_t('La piattaforma ha risposto con un errore (HTTP %d).', 'The platform answered with an error (HTTP %d).'), $code));
    }
    if (isset($data['reason']) && $data['reason'] === 'quota') {
        return array('ok' => false, 'upgrade' => 'aeo_draft', 'message' => sprintf(citami_t('Bozze AI finite per questo mese (%1$d su %2$d): riprendono il mese prossimo.', 'AI drafts used up for this month (%1$d of %2$d): back next month.'), (int) $data['used'], (int) $data['quota']));
    }
    $tldr = mb_substr(trim(wp_strip_all_tags((string) (isset($data['tldr']) ? $data['tldr'] : ''))), 0, 600);
    $faq = array();
    foreach ((array) (isset($data['faq']) ? $data['faq'] : array()) as $it) {
        if (!is_array($it)) {
            continue;
        }
        $q = mb_substr(trim(wp_strip_all_tags((string) (isset($it['q']) ? $it['q'] : ''))), 0, 200);
        $a = mb_substr(trim(wp_strip_all_tags((string) (isset($it['a']) ? $it['a'] : ''))), 0, 2000);
        if ($q !== '' && $a !== '') {
            $faq[] = array('q' => $q, 'a' => $a);
        }
    }
    $dropped = isset($data['dropped']) ? (int) $data['dropped'] : 0;
    if ($tldr === '' && empty($faq)) {
        citami_aeo_draft_discard($post_id);
        return array('ok' => false, 'message' => $dropped > 0
            ? sprintf(citami_t('Nessuna bozza tenuta: %d scartate perché contenevano numeri o domande non presenti nella pagina.', 'No draft kept: %d dropped because they carried numbers or questions not on the page.'), $dropped)
            : citami_t('La piattaforma non ha trovato nel testo una risposta da proporre.', 'The platform found no answer in the text worth proposing.'));
    }
    $draft = array('tldr' => $tldr, 'faq' => $faq, 'dropped' => $dropped, 'ts' => time(), 'used' => isset($data['used']) ? (int) $data['used'] : null, 'quota' => isset($data['quota']) ? (int) $data['quota'] : null);
    update_post_meta($post_id, CITAMI_META_AEO_DRAFT, $draft);
    $parts = array();
    if ($tldr !== '') $parts[] = citami_t('una risposta breve', 'a short answer');
    if (!empty($faq)) $parts[] = sprintf('%d FAQ', count($faq));
    return array('ok' => true, 'draft' => $draft, 'message' => sprintf(citami_t('Bozza pronta: %s. Leggila qui sotto e scegli se usarla.', 'Draft ready: %s. Read it below and decide whether to use it.'), implode(' + ', $parts)));
}

/* ---------------------------------------------------------------- */
/*  Apply                                                             */
/* ---------------------------------------------------------------- */

function citami_aeo_draft_norm_question($q) {
    return trim(preg_replace('/\s+/u', ' ', preg_replace('/[?¿!.,;:\'"“”‘’]/u', '', mb_strtolower((string) $q))));
}

/** Writes the draft into the page: TL;DR replaced, FAQ appended (duplicates skipped). */
function citami_aeo_draft_apply($post_id) {
    $post_id = (int) $post_id;
    $draft = citami_aeo_draft_get($post_id);
    if ($draft === null || !function_exists('citami_content_save')) {
        return array('ok' => false, 'message' => citami_t('Nessuna bozza da usare.', 'No draft to use.'));
    }
    $tldr = $draft['tldr'] !== '' ? $draft['tldr'] : citami_content_tldr($post_id);
    $faq = citami_content_faq($post_id);
    $seen = array();
    foreach ($faq as $it) {
        $seen[citami_aeo_draft_norm_question($it['q'])] = true;
    }
    $added = 0;
    foreach ($draft['faq'] as $it) {
        $k = citami_aeo_draft_norm_question($it['q']);
        if ($k === '' || isset($seen[$k])) {
            continue;
        }
        $seen[$k] = true;
        $faq[] = $it;
        $added++;
    }
    citami_content_save($post_id, $tldr, $faq);
    citami_aeo_draft_discard($post_id);
    if (function_exists('citami_aeo_store_on_save')) {
        citami_aeo_store_on_save($post_id);
    }
    if (function_exists('citami_indexnow_enqueue_post')) {
        citami_indexnow_enqueue_post($post_id, true);
    }
    $what = array();
    if ($draft['tldr'] !== '') $what[] = citami_t('risposta breve', 'short answer');
    if ($added > 0) $what[] = sprintf('%d FAQ', $added);
    $summary = !empty($what) ? implode(' + ', $what) : citami_t('niente di nuovo (FAQ già presenti)', 'nothing new (FAQ already present)');
    if (function_exists('citami_activity_log')) {
        citami_activity_log('aeo', sprintf(citami_t('Bozza AI usata in "%1$s": %2$s', 'AI draft used in "%1$s": %2$s'), get_the_title($post_id), $summary), 1, array('post' => $post_id));
    }
    if (function_exists('citami_pages_on_aeo_applied')) {
        citami_pages_on_aeo_applied($post_id);
    }
    return array('ok' => true, 'added' => $added, 'tldr' => $draft['tldr'] !== '', 'message' => sprintf(citami_t('Scritto nella pagina: %s. Il punteggio AEO è stato ricalcolato; IndexNow avvisato.', 'Written into the page: %s. AEO score recomputed; IndexNow notified.'), $summary));
}

/* ---------------------------------------------------------------- */
/*  Handler (nonce-protected links: the box sits inside the post form) */
/* ---------------------------------------------------------------- */

function citami_aeo_draft_url($post_id, $do, $back = '') {
    $url = admin_url('admin-post.php?action=citami_aeo_draft&do=' . rawurlencode($do) . '&post_id=' . (int) $post_id . ($back !== '' ? '&back=' . rawurlencode($back) : ''));
    return wp_nonce_url($url, 'citami_aeo_draft_' . (int) $post_id);
}

function citami_aeo_draft_edit_url($post_id) {
    $edit = get_edit_post_link((int) $post_id, 'raw');
    return ($edit ? $edit : admin_url('post.php?post=' . (int) $post_id . '&action=edit')) . '#citami-aeo-draft';
}

function citami_handle_aeo_draft_action() {
    $post_id = isset($_REQUEST['post_id']) ? (int) $_REQUEST['post_id'] : 0;
    if (!current_user_can('edit_post', $post_id) && !current_user_can('manage_options')) {
        wp_die('Permesso negato.', '', array('response' => 403));
    }
    if (!wp_verify_nonce(isset($_REQUEST['_wpnonce']) ? sanitize_text_field(wp_unslash($_REQUEST['_wpnonce'])) : '', 'citami_aeo_draft_' . $post_id)) {
        wp_die('Nonce non valido.', '', array('response' => 403));
    }
    $do = isset($_REQUEST['do']) ? sanitize_key((string) $_REQUEST['do']) : 'generate';
    $back = isset($_REQUEST['back']) ? sanitize_key((string) $_REQUEST['back']) : '';
    if ($do === 'apply') {
        $r = citami_aeo_draft_apply($post_id);
    } elseif ($do === 'discard') {
        citami_aeo_draft_discard($post_id);
        $r = array('ok' => true, 'message' => citami_t('Bozza scartata.', 'Draft discarded.'));
    } else {
        $r = citami_aeo_draft_request($post_id, isset($_REQUEST['kind']) ? sanitize_key((string) $_REQUEST['kind']) : 'both');
    }
    $notice = array('type' => !empty($r['ok']) ? 'success' : 'error', 'message' => $r['message'], 'upgrade' => isset($r['upgrade']) ? $r['upgrade'] : '');
    if ($back === 'overview' && (empty($r['ok']) || $do !== 'generate')) {
        set_transient('citami_action_notice', $notice, 30);
        wp_safe_redirect(function_exists('citami_admin_url') ? citami_admin_url('overview') : admin_url('admin.php?page=citami-connector'));
        exit;
    }
    set_transient(CITAMI_AEO_DRAFT_NOTICE . $post_id, $notice, 120);
    wp_safe_redirect(citami_aeo_draft_edit_url($post_id));
    exit;
}
add_action('admin_post_citami_aeo_draft', 'citami_handle_aeo_draft_action');

/* ---------------------------------------------------------------- */
/*  SEO box section                                                   */
/* ---------------------------------------------------------------- */

function citami_aeo_draft_render_box($post) {
    $post_id = (int) $post->ID;
    $draft = citami_aeo_draft_get($post_id);
    $notice = get_transient(CITAMI_AEO_DRAFT_NOTICE . $post_id);
    if (is_array($notice)) {
        delete_transient(CITAMI_AEO_DRAFT_NOTICE . $post_id);
    }
    $connected = citami_get_api_key() !== '';
    $has_tldr = function_exists('citami_content_tldr') && citami_content_tldr($post_id) !== '';
    // 2.49 — with the to-do list on, the "write it for me" buttons live next to the thing to do; this box only shows a draft or its outcome.
    if ($draft === null && !is_array($notice) && function_exists('citami_aeo_settings') && citami_aeo_settings()['enabled'] === 'on') {
        return;
    }
    ?>
    <div class="citami-aeo-draft" id="citami-aeo-draft" data-citami="aeo-draft">
        <p class="citami-aeo-draft-head"><strong><?php echo esc_html(citami_t('Bozza scritta dalla piattaforma', 'Draft written by the platform')); ?></strong>
            <span class="description"><?php echo esc_html(citami_t('Solo dal testo salvato di questa pagina: niente numeri, nomi o promesse che non ci sono già. Tu decidi se usarla.', 'From this page\'s saved text only: no numbers, names or promises that are not already there. You decide whether to use it.')); ?></span></p>
        <?php if (is_array($notice) && !empty($notice['message'])): ?>
            <p class="citami-aeo-draft-notice citami-aeo-draft-notice--<?php echo esc_attr($notice['type']); ?>"><?php echo wp_kses(citami_notice_text_html($notice), citami_kses_allowed()); ?></p>
        <?php endif; ?>
        <?php if ($draft !== null): ?>
            <?php if ($draft['tldr'] !== ''): $cur_tldr = $has_tldr ? citami_content_tldr($post_id) : ''; ?>
                <div class="citami-aeo-compare">
                    <div class="is-old"><span class="citami-aeo-draft-label"><?php echo esc_html(citami_t('Ora nella pagina', 'Now on the page')); ?></span><p><?php echo $cur_tldr !== '' ? esc_html($cur_tldr) : '<em>' . esc_html(citami_t('nessuna risposta breve', 'no short answer')) . '</em>'; ?></p></div>
                    <div class="is-new"><span class="citami-aeo-draft-label"><?php echo esc_html(citami_t('Proposta', 'Proposed')); ?></span><p class="citami-aeo-draft-tldr"><?php echo esc_html($draft['tldr']); ?></p></div>
                </div>
            <?php endif; ?>
            <?php if (!empty($draft['faq'])): ?>
                <span class="citami-aeo-draft-label"><?php echo esc_html(sprintf(citami_t('%d FAQ da aggiungere in fondo alla pagina', '%d FAQ to add at the bottom of the page'), count($draft['faq']))); ?></span>
                <dl class="citami-aeo-draft-faq">
                    <?php foreach ($draft['faq'] as $it): ?>
                        <dt><?php echo esc_html($it['q']); ?></dt>
                        <dd><?php echo esc_html($it['a']); ?></dd>
                    <?php endforeach; ?>
                </dl>
            <?php endif; ?>
            <?php if ($draft['dropped'] > 0): ?>
                <p class="description"><?php echo esc_html(sprintf(citami_t('Proposte scartate dalla piattaforma perché contenevano numeri o domande non presenti nella pagina: %d.', 'Proposals dropped by the platform because they carried numbers or questions not on the page: %d.'), $draft['dropped'])); ?></p>
            <?php endif; ?>
            <p class="citami-aeo-draft-actions">
                <a class="button button-primary" href="<?php echo esc_url(citami_aeo_draft_url($post_id, 'apply')); ?>"><?php echo esc_html(citami_t('Usa la bozza', 'Use the draft')); ?></a>
                <a class="button" href="<?php echo esc_url(citami_aeo_draft_url($post_id, 'generate')); ?>"><?php echo esc_html(citami_t('Riscrivi', 'Rewrite')); ?></a>
                <a class="button-link" href="<?php echo esc_url(citami_aeo_draft_url($post_id, 'discard')); ?>"><?php echo esc_html(citami_t('Scarta', 'Discard')); ?></a>
            </p>
            <p class="description"><?php echo esc_html(citami_t('"Usa la bozza" scrive risposta breve e FAQ nei campi qui sopra e salva subito; le FAQ già presenti restano.', '"Use the draft" writes the short answer and FAQ into the fields above and saves right away; existing FAQ stay.')); ?></p>
        <?php elseif ($connected): $u = citami_usage(); $q = $u ? $u['aeo_draft'] : null; ?>
            <?php if ($q && $q['level'] === 'exhausted'): ?>
                <p class="description citami-aeo-draft-quota" data-level="exhausted"><?php echo esc_html(sprintf(citami_t('Bozze AI finite per questo mese (%1$d su %2$d): riprendono il %3$s.', 'AI drafts used up for this month (%1$d of %2$d): back on %3$s.'), (int) $q['used'], (int) $q['quota'], wp_date('j F', (int) $u['resets_at']))); ?> <?php echo wp_kses(citami_notice_text_html(array('message' => '', 'upgrade' => 'aeo_draft')), citami_kses_allowed()); ?></p>
            <?php else: ?>
            <p class="citami-aeo-draft-actions">
                <a class="button" href="<?php echo esc_url(citami_aeo_draft_url($post_id, 'generate')); ?>"><?php echo esc_html(citami_t('Scrivi risposta breve e 3 FAQ', 'Write a short answer and 3 FAQ')); ?></a>
                <span class="description"><?php echo esc_html(citami_t('Usa il testo salvato: salva la pagina prima, se l\'hai modificata.', 'Uses the saved text: save the page first if you changed it.')); ?><?php if ($q): ?> <span class="citami-aeo-draft-quota" data-level="<?php echo esc_attr($q['level']); ?>"><?php echo esc_html(sprintf(citami_t('%1$d di %2$d bozze usate questo mese.', '%1$d of %2$d drafts used this month.'), (int) $q['used'], (int) $q['quota'])); ?></span><?php endif; ?></span>
            </p>
            <?php endif; ?>
        <?php else: ?>
            <p class="description"><?php echo esc_html(citami_t('Collega il sito alla piattaforma per farti scrivere una bozza.', 'Connect the site to the platform to get a draft written.')); ?></p>
        <?php endif; ?>
    </div>
    <?php
}
add_action('citami_seo_meta_box_essentials', 'citami_aeo_draft_render_box', 20);

/** Overview: the small action next to a weak page. */
function citami_aeo_draft_overview_action($post_id) {
    $post_id = (int) $post_id;
    // 2.59 — the draft is read and approved on the Overview itself (section "AI drafts to approve").
    if (citami_aeo_draft_get($post_id) !== null) {
        return '<a class="citami-link" href="#citami-draft-' . $post_id . '">' . esc_html(citami_t('Bozza pronta: leggila', 'Draft ready: read it')) . '</a>';
    }
    if (citami_get_api_key() === '') {
        return '';
    }
    $href = function_exists('citami_pages_action_url') ? citami_pages_action_url('aeo', $post_id, 'overview') : citami_aeo_draft_url($post_id, 'generate', 'overview');
    return '<a class="citami-link" data-citami-busy="' . esc_attr(citami_t('Scrivo… 10-20 s', 'Writing… 10-20 s')) . '" href="' . esc_url($href) . '">' . esc_html(citami_t('Scrivi bozza AI', 'Write AI draft')) . '</a>';
}

/* ---------------------------------------------------------------- */
/*  2.45 — REST for the block editor sidebar                          */
/* ---------------------------------------------------------------- */

/** Draft as the sidebar wants it (null when none). */
function citami_aeo_draft_public($post_id) {
    $d = citami_aeo_draft_get($post_id);
    if ($d === null) {
        return null;
    }
    return array('tldr' => $d['tldr'], 'faq' => $d['faq'], 'dropped' => $d['dropped']);
}

/**
 * POST citami/v1/aeo-draft {post_id, do: get|generate|apply|discard}
 * Same functions as the links in the SEO box; the sidebar shows the
 * result without leaving the editor. `apply` also returns what is now
 * on the page so the sidebar can refresh the SEO box fields.
 */
function citami_rest_aeo_draft(WP_REST_Request $request) {
    $body = $request->get_json_params();
    if (!is_array($body)) {
        $body = array();
    }
    $post_id = isset($body['post_id']) ? (int) $body['post_id'] : 0;
    if ($post_id <= 0 || !get_post($post_id)) {
        return new WP_Error('citami_aeo_draft_post', 'post_id required', array('status' => 400));
    }
    if (!current_user_can('edit_post', $post_id)) {
        return new WP_Error('citami_aeo_draft_forbidden', 'forbidden', array('status' => 403));
    }
    $do = isset($body['do']) ? sanitize_key((string) $body['do']) : 'get';
    $u = citami_usage();
    $out = array('ok' => true, 'message' => '', 'connected' => citami_get_api_key() !== '', 'usage' => $u ? $u['aeo_draft'] + array('resets_at' => (int) $u['resets_at']) : null);
    if ($do === 'generate') {
        $r = citami_aeo_draft_request($post_id, isset($body['kind']) ? sanitize_key((string) $body['kind']) : 'both');
        $out['ok'] = !empty($r['ok']);
        $out['message'] = $r['message'];
    } elseif ($do === 'apply') {
        $r = citami_aeo_draft_apply($post_id);
        $out['ok'] = !empty($r['ok']);
        $out['message'] = $r['message'];
        if ($out['ok']) {
            $out['applied'] = array('tldr' => citami_content_tldr($post_id), 'faq' => citami_content_faq($post_id));
        }
    } elseif ($do === 'discard') {
        citami_aeo_draft_discard($post_id);
        $out['message'] = citami_t('Bozza scartata.', 'Draft discarded.');
    }
    $out['draft'] = citami_aeo_draft_public($post_id);
    return rest_ensure_response($out);
}

add_action('rest_api_init', function () {
    register_rest_route('citami/v1', '/aeo-draft', array(
        'methods'             => 'POST',
        'callback'            => 'citami_rest_aeo_draft',
        'permission_callback' => function () { return current_user_can('edit_posts'); },
    ));
});