<?php

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

/**
 * Improve SEO Refinement Step
 * 
 * Improves SEO optimization of the content.
 */
class AIGH_Step_Refine_Improve_SEO extends AIGH_Step_Refinement_Base {
    
    protected $step_type = 'refine_improve_seo';
    protected $step_name = 'Improve SEO';
    protected $description = 'Improve SEO optimization with better keywords and headings while maintaining readability';
    
    /**
     * Get the refinement instruction
     *
     * @return string
     */
    protected function get_refinement_instruction() {
        return "Improve the SEO optimization. Add relevant keywords naturally, improve headings for better search visibility. Maintain readability.";
    }
    
    /**
     * Build the user prompt
     *
     * @param string $original_content
     * @param string $research_context
     * @param AIGH_Pipeline_Context $context
     * @return string
     */
    protected function build_user_prompt($original_content, $research_context, $context) {
        $prompt = "Here is the original article:\n\n{$original_content}\n\n";
        
        // Check for keywords in context
        $keywords = $this->get_from_context($context, 'keywords', '');
        $topic = $this->get_from_context($context, 'topic', '');
        
        if (!empty($research_context)) {
            $prompt .= "\n{$research_context}\n\n";
            $prompt .= "IMPORTANT: Use the specific data, statistics, and examples from the research above to improve the article.\n\n";
        }
        
        $prompt .= "Refinement Task: " . $this->get_refinement_instruction();
        
        if (!empty($keywords)) {
            $prompt .= " Target keywords: {$keywords}.";
        }
        
        if (!empty($topic)) {
            $prompt .= " Main topic: {$topic}.";
        }
        
        if (!empty($research_context)) {
            $prompt .= " Use trending topics from the research.";
        }
        
        $prompt .= "\n\n";
        
        $prompt .= "SEO Guidelines:\n";
        $prompt .= "- Include target keywords in headings where natural\n";
        $prompt .= "- Use descriptive H2 and H3 headings\n";
        $prompt .= "- Ensure good keyword density without stuffing\n";
        $prompt .= "- Add internal linking opportunities (suggest with placeholder links)\n";
        $prompt .= "- Optimize meta-friendly opening paragraph\n\n";
        
        $prompt .= $this->get_standard_requirements(!empty($research_context));
        
        return $prompt;
    }
}