<?php

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

/**
 * Generate Draft Step
 * 
 * Creates initial content using OpenAI based on topic, length, and tone.
 * Can optionally use research context if available.
 */
class AIGH_Step_Generate_Draft extends AIGH_Pipeline_Step {
    
    protected $step_type = 'generate_draft';
    protected $step_name = 'Generate Draft';
    protected $description = 'Generate initial article content using AI';
    protected $category = 'processing';
    protected $estimated_duration = 20;
    
    protected $required_inputs = array('topic', 'length', 'tone');
    
    protected $output_keys = array('content', 'generated_at');
    
    /**
     * @var AIGH_OpenAI_Client
     */
    private $openai_client;
    
    /**
     * Constructor
     *
     * @param array $config
     * @param AIGH_OpenAI_Client|null $openai_client
     */
    public function __construct($config = array(), $openai_client = null) {
        parent::__construct($config);
        $this->openai_client = $openai_client ?: new AIGH_OpenAI_Client();
    }
    
    /**
     * Get default configuration
     *
     * @return array
     */
    protected function get_default_config() {
        return array(
            'use_research' => true,  // Use research_brief if available in context
            'temperature' => 0.7,
            'max_tokens' => 4000
        );
    }
    
    /**
     * Execute the step
     *
     * @param AIGH_Pipeline_Context $context
     * @return AIGH_Pipeline_Context|WP_Error
     */
    public function execute($context) {
        $this->log_start($context);
        
        // Get required inputs
        $topic = $this->get_from_context($context, 'topic');
        $length = $this->get_from_context($context, 'length', 'medium');
        $tone = $this->get_from_context($context, 'tone', 'professional');
        $length_guide = $this->get_from_context($context, 'length_guide', '800-1000 words');

        // Get persona if provided
        $persona = $this->get_from_context($context, 'persona', '');

        // Get custom style if provided (overrides tone)
        $custom_style = $this->get_from_context($context, 'custom_style', '');
        
        // Get SEO keywords - check both 'seo_keywords' and 'keywords' keys
        $seo_keywords = $this->get_from_context($context, 'seo_keywords', '');
        if (empty($seo_keywords)) {
            $seo_keywords = $this->get_from_context($context, 'keywords', '');
        }
        
        // Debug logging
        if (defined('WP_DEBUG') && WP_DEBUG) {
            // phpcs:disable WordPress.PHP.DevelopmentFunctions.error_log_error_log -- Debug logging wrapped in WP_DEBUG check
            error_log('[AIGH] generate_draft - seo_keywords from context: ' . ($seo_keywords ?: 'EMPTY'));
            error_log('[AIGH] generate_draft - custom_style: ' . (!empty($custom_style) ? 'PROVIDED' : 'EMPTY'));
            // phpcs:enable WordPress.PHP.DevelopmentFunctions.error_log_error_log
        }
        
        // Check for research brief
        $research_brief = $this->get_from_context($context, 'research_brief', '');
        $use_research = $this->get_config_value('use_research', true) && !empty($research_brief);
        
        // Generate content
        if ($use_research) {
            $content = $this->generate_with_research($topic, $length_guide, $tone, $research_brief, $seo_keywords, $custom_style, $persona);
        } else {
            $content = $this->generate_basic($topic, $length_guide, $tone, $seo_keywords, $custom_style, $persona);
        }
        
        if (is_wp_error($content)) {
            $this->log_end($context, false);
            return $content;
        }
        
        // Store results in context
        $this->set_in_context($context, 'content', $content);
        $this->set_in_context($context, 'generated_at', current_time('mysql'));
        $this->set_in_context($context, 'research_used', $use_research);
        
        $this->log_end($context, true);
        
        return $context;
    }
    
    /**
     * Generate basic content without research
     *
     * @param string $topic
     * @param string $length_guide
     * @param string $tone
     * @param string $seo_keywords
     * @param string $custom_style
     * @return string|WP_Error
     */
    private function generate_basic($topic, $length_guide, $tone, $seo_keywords = '', $custom_style = '', $persona = '') {
        // Build style instruction
        $style_instruction = '';
        if (!empty($custom_style)) {
            $style_instruction = "Follow this writing style guide:\n{$custom_style}";
        } else {
            $style_instruction = "Use a {$tone} tone.";
        }

        // Build persona instruction
        $persona_instruction = '';
        if (!empty($persona)) {
            $persona_instruction = "You are writing as: {$persona}. Bring this expertise and perspective to the content.";
        }

        $system_prompt = "You are a professional content writer and SEO specialist. Create high-quality, engaging, SEO-optimized content. {$style_instruction}";
        if (!empty($persona_instruction)) {
            $system_prompt .= "\n\n{$persona_instruction}";
        }
        
        $user_prompt = "Write a blog article about: {$topic}\n\n";
        
        // Add SEO keywords instruction if provided
        if (!empty($seo_keywords)) {
            // Parse keywords - first one is primary, rest are secondary
            $keywords_array = array_map('trim', explode(',', $seo_keywords));
            $primary_keyword = $keywords_array[0] ?? '';
            $secondary_keywords = array_slice($keywords_array, 1);
            
            $user_prompt .= "=== SEO KEYWORDS (MUST USE) ===\n";
            $user_prompt .= "PRIMARY KEYWORD: {$primary_keyword}\n";
            if (!empty($secondary_keywords)) {
                $user_prompt .= "SECONDARY KEYWORDS: " . implode(', ', $secondary_keywords) . "\n";
            }
            $user_prompt .= "\nCRITICAL REQUIREMENTS:\n";
            $user_prompt .= "- Use the PRIMARY keyword \"{$primary_keyword}\" in:\n";
            $user_prompt .= "  * The first paragraph (within first 100 words)\n";
            $user_prompt .= "  * At least 2 subheadings (<h2> or <h3> tags)\n";
            $user_prompt .= "  * The conclusion\n";
            $user_prompt .= "  * Naturally 3-5 times throughout the body\n";
            if (!empty($secondary_keywords)) {
                $user_prompt .= "- Use EACH secondary keyword at least 1-2 times naturally in the content:\n";
                foreach ($secondary_keywords as $sec_kw) {
                    $user_prompt .= "  * \"{$sec_kw}\"\n";
                }
            }
            $user_prompt .= "=== END SEO KEYWORDS ===\n\n";
        }
        
        $user_prompt .= "Requirements:\n";
        $user_prompt .= "- Length: {$length_guide}\n";
        if (!empty($custom_style)) {
            $user_prompt .= "- Writing style: Follow the custom style guide provided in the system prompt\n";
        } else {
            $user_prompt .= "- Tone: {$tone}\n";
        }
        $user_prompt .= "- Use proper HTML formatting with <h2>, <h3>, <p> tags\n";
        $user_prompt .= "- Include an engaging introduction and a strong closing section\n";
        $user_prompt .= "- NEVER use 'Conclusion' as a heading - use a creative, topic-relevant heading instead\n";
        $user_prompt .= "- Use subheadings to break up content\n";
        if (!empty($seo_keywords)) {
            $user_prompt .= "- Optimize heavily for the specified SEO keywords\n";
        }
        $user_prompt .= "\nWrite the complete article content:";
        
        return $this->openai_client->generate_text(
            $system_prompt,
            $user_prompt,
            $this->get_config_value('temperature', 0.7),
            $this->get_config_value('max_tokens', 4000)
        );
    }
    
    /**
     * Generate content with research context
     *
     * @param string $topic
     * @param string $length_guide
     * @param string $tone
     * @param string $research_brief
     * @param string $seo_keywords
     * @param string $custom_style
     * @return string|WP_Error
     */
    private function generate_with_research($topic, $length_guide, $tone, $research_brief, $seo_keywords = '', $custom_style = '', $persona = '') {
        // Build style instruction
        $style_instruction = '';
        if (!empty($custom_style)) {
            $style_instruction = "Follow this writing style guide:\n{$custom_style}";
        } else {
            $style_instruction = "Use a {$tone} tone.";
        }

        // Build persona instruction
        $persona_instruction = '';
        if (!empty($persona)) {
            $persona_instruction = "You are writing as: {$persona}. Bring this expertise and perspective to the content.";
        }

        $system_prompt = "You are a professional content writer and SEO specialist. Create high-quality, well-researched, SEO-optimized content based on verified sources. {$style_instruction} You MUST incorporate the specific information provided in the research - do not write generic content.";
        if (!empty($persona_instruction)) {
            $system_prompt .= "\n\n{$persona_instruction}";
        }
        
        $user_prompt = "Write a blog article about: {$topic}\n\n";
        
        // Add SEO keywords instruction if provided
        if (!empty($seo_keywords)) {
            // Parse keywords - first one is primary, rest are secondary
            $keywords_array = array_map('trim', explode(',', $seo_keywords));
            $primary_keyword = $keywords_array[0] ?? '';
            $secondary_keywords = array_slice($keywords_array, 1);
            
            $user_prompt .= "=== SEO KEYWORDS (MUST USE) ===\n";
            $user_prompt .= "PRIMARY KEYWORD: {$primary_keyword}\n";
            if (!empty($secondary_keywords)) {
                $user_prompt .= "SECONDARY KEYWORDS: " . implode(', ', $secondary_keywords) . "\n";
            }
            $user_prompt .= "\nCRITICAL REQUIREMENTS:\n";
            $user_prompt .= "- Use the PRIMARY keyword \"{$primary_keyword}\" in:\n";
            $user_prompt .= "  * The first paragraph (within first 100 words)\n";
            $user_prompt .= "  * At least 2 subheadings (<h2> or <h3> tags)\n";
            $user_prompt .= "  * The conclusion\n";
            $user_prompt .= "  * Naturally 3-5 times throughout the body\n";
            if (!empty($secondary_keywords)) {
                $user_prompt .= "- Use EACH secondary keyword at least 1-2 times naturally in the content:\n";
                foreach ($secondary_keywords as $sec_kw) {
                    $user_prompt .= "  * \"{$sec_kw}\"\n";
                }
            }
            $user_prompt .= "=== END SEO KEYWORDS ===\n\n";
        }
        
        $user_prompt .= "=== RESEARCH TO USE ===\n";
        $user_prompt .= "{$research_brief}\n";
        $user_prompt .= "=== END RESEARCH ===\n\n";
        
        // Detect what types of research are present
        $has_statistics = stripos($research_brief, 'STATISTICS') !== false;
        $has_examples = stripos($research_brief, 'EXAMPLES') !== false || stripos($research_brief, 'CASE STUD') !== false;
        $has_quotes = stripos($research_brief, 'EXPERT') !== false || stripos($research_brief, 'QUOTES') !== false;
        
        $user_prompt .= "CRITICAL REQUIREMENTS - YOU MUST FOLLOW THESE:\n";
        
        if (!empty($seo_keywords)) {
            $user_prompt .= "- INCORPORATE THE SEO KEYWORDS as specified above\n";
        }
        
        if ($has_statistics) {
            $user_prompt .= "- Include SPECIFIC statistics and numbers from the research above (percentages, dollar amounts, dates)\n";
        }
        
        if ($has_examples) {
            $user_prompt .= "- Reference SPECIFIC case studies and real-world examples from the research\n";
            $user_prompt .= "- Name actual companies, products, or people mentioned in the examples\n";
        }
        
        if ($has_quotes) {
            $user_prompt .= "- Include DIRECT QUOTES from experts mentioned in the research\n";
            $user_prompt .= "- Attribute quotes properly (e.g., 'According to [Name], [Title]...')\n";
            $user_prompt .= "- Use at least 2-3 expert quotes throughout the article\n";
        }
        
        $user_prompt .= "- Use real names, companies, and specific details from the research\n";
        $user_prompt .= "- DO NOT write generic content - every claim should reference the research\n";
        $user_prompt .= "- When linking to sources, use descriptive anchor text (e.g., 'according to Harvard Business Review' not 'source' or 'click here')\n\n";

        $user_prompt .= "Format Requirements:\n";
        $user_prompt .= "- Length: {$length_guide}\n";
        if (!empty($custom_style)) {
            $user_prompt .= "- Writing style: Follow the custom style guide provided in the system prompt\n";
        } else {
            $user_prompt .= "- Tone: {$tone}\n";
        }
        $user_prompt .= "- HTML formatting with <h2>, <h3>, <p> tags\n";
        $user_prompt .= "- Engaging introduction with strong hook based on research\n";
        $user_prompt .= "- Descriptive subheadings";
        if (!empty($seo_keywords)) {
            $user_prompt .= " (include SEO keywords in subheadings)";
        }
        $user_prompt .= "\n- Strong closing section - NEVER use 'Conclusion' as a heading, use a creative topic-relevant heading instead\n\n";

        $user_prompt .= "Write the complete article using the research above (HTML formatted):";
        
        return $this->openai_client->generate_text(
            $system_prompt,
            $user_prompt,
            $this->get_config_value('temperature', 0.7),
            $this->get_config_value('max_tokens', 4000)
        );
    }
    
    /**
     * Set the OpenAI client (for dependency injection/testing)
     *
     * @param AIGH_OpenAI_Client $client
     */
    public function set_openai_client($client) {
        $this->openai_client = $client;
    }
    
    /**
     * Get configuration schema for UI rendering
     *
     * @return array
     */
    public function get_config_schema() {
        return array(
            'use_research' => array(
                'type' => 'checkbox',
                'label' => 'Use Research',
                'description' => 'Use research brief if available in the pipeline',
                'default' => true
            ),
            'temperature' => array(
                'type' => 'range',
                'label' => 'Creativity',
                'description' => 'Higher values = more creative, lower = more focused',
                'min' => 0.1,
                'max' => 1.0,
                'step' => 0.1,
                'default' => 0.7
            )
        );
    }
}