<?php
namespace AI_Blogger\Admin;

defined('ABSPATH') || exit;

class Settings {
    public function __construct() {
        add_action('admin_menu', array($this, 'add_settings_page'));
        add_action('admin_init', array($this, 'register_settings'));
    }

    private function sanitize_api_key($input) {
        return sanitize_text_field(trim($input));
    }

    /**
     * Sanitize all plugin options
     * 
     * @param array $options The options array to sanitize
     * @return array Sanitized options
     */
    public function sanitize_options($options) {
        if (!isset($options['ai_blogger_api_key'])) {
            return $options;
        }

        // Verify nonce
        $nonce = isset($_POST['_wpnonce']) ? sanitize_text_field(wp_unslash($_POST['_wpnonce'])) : '';
        if (!wp_verify_nonce($nonce, 'ai_blogger_options-options')) {
            return $options;
        }
        
        $clean_options = array();
        $clean_options['ai_blogger_api_key'] = $this->sanitize_api_key($options['ai_blogger_api_key']);
        $clean_options['ai_blogger_model'] = $this->sanitize_model($options['ai_blogger_model']);
        
        return $clean_options;
    }

    public function sanitize_model($input) {
        return self::sanitize_model_static($input);
    }

    public function add_settings_page() {
        add_options_page(
            __('AI Blogger Settings', 'ai-blogger'),
            __('AI Blogger', 'ai-blogger'),
            'manage_options',
            'ai-blogger-settings',
            array($this, 'render_settings_page')
        );
    }

    /**
     * Static sanitization callback for API key
     * 
     * @param string $input The input to sanitize
     * @return string Sanitized input
     */
    public static function sanitize_api_key_static($input) {
        return sanitize_text_field(trim($input));
    }

    /**
     * Static sanitization callback for model selection
     * 
     * @param string $input The input to sanitize
     * @return string Sanitized input
     */
    public static function sanitize_model_static($input) {
        $api_key = get_option('ai_blogger_api_key');
        $valid_models = [];

        if (!empty($api_key)) {
            if (!class_exists('AI_Blogger\\API_Handler')) {
                require_once AI_BLOGGER_PLUGIN_DIR . 'includes/class-api-handler.php';
            }
            $api_handler = new \AI_Blogger\API_Handler();
            $valid_models = $api_handler->get_models($api_key);
        }

        // Fallback to a default list if API call fails or no API key
        if (empty($valid_models)) {
            $valid_models = [
                'llama-3.3-70b-versatile',
                'mixtral-8x7b-32768',
                'deepseek-r1-distill-llama-70b',
                'gemma2-9b-it',
                'gemma-7b-it',
                'llama3-70b-8192',
                'llama3-8b-8192',
            ];
        }

        $default_model = !empty($valid_models) ? $valid_models[0] : '';

        return in_array($input, $valid_models) ? $input : $default_model;
    }

    public function register_settings() {
        // Register API key setting with static sanitization callback
        register_setting(
            'ai_blogger_options',
            'ai_blogger_api_key',
            'AI_Blogger\\Admin\\Settings::sanitize_api_key_static'
        );

        // Register model setting with static sanitization callback
        register_setting(
            'ai_blogger_options', 
            'ai_blogger_model',
            'AI_Blogger\\Admin\\Settings::sanitize_model_static'
        );
        
        // Register Pexels API key setting
        register_setting(
            'ai_blogger_options',
            'ai_blogger_pexels_key',
            'AI_Blogger\\Admin\\Settings::sanitize_api_key_static'
        );

        add_settings_section(
            'ai_blogger_main',
            __('API Configuration', 'ai-blogger'),
            null,
            'ai-blogger-settings'
        );

        add_settings_field(
            'ai_api_key',
            __('Groq API Key', 'ai-blogger'),
            array($this, 'render_api_key_field'),
            'ai-blogger-settings',
            'ai_blogger_main'
        );

        add_settings_field(
            'ai_model',
            __('AI Model', 'ai-blogger'),
            array($this, 'render_model_field'),
            'ai-blogger-settings',
            'ai_blogger_main'
        );
        
        add_settings_field(
            'ai_pexels_key',
            __('Pexels API Key', 'ai-blogger'),
            array($this, 'render_pexels_key_field'),
            'ai-blogger-settings',
            'ai_blogger_main'
        );
    }

    public function render_settings_page() {
        ?>
        <div class="wrap">
            <h1><?php esc_html_e('AI Blogger Settings', 'ai-blogger'); ?></h1>
            <form action="options.php" method="post">
                <?php
                settings_fields('ai_blogger_options');
            do_settings_sections('ai-blogger-settings');
                submit_button();
                ?>
            </form>
        </div>
        <?php
    }

    public function render_api_key_field() {
        $api_key = get_option('ai_blogger_api_key');
        echo '<input type="password" name="ai_blogger_api_key" value="' . esc_attr($api_key) . '" class="regular-text">';
            echo '<p class="description">' . esc_html__('Your Groq API key can be found in your ', 'ai-blogger') 
            . ' <a href="https://console.groq.com/keys" target="_blank" rel="noopener noreferrer">'
            . esc_html__('Groq Cloud Console', 'ai-blogger') 
            . '</a></p>';
    }

    public function render_model_field() {
        $selected_model = get_option('ai_blogger_model', 'llama-3.3-70b-versatile');
        $api_key = get_option('ai_blogger_api_key');
        $models = [];

        if (!empty($api_key)) {
            if (!class_exists('AI_Blogger\\API_Handler')) {
                require_once AI_BLOGGER_PLUGIN_DIR . 'includes/class-api-handler.php';
            }
            $api_handler = new \AI_Blogger\API_Handler();
            $models = $api_handler->get_models($api_key);
        }

        // Fallback to a default list if API call fails or no API key
        if (empty($models)) {
            $models = [
                'llama-3.3-70b-versatile',
                'mixtral-8x7b-32768',
                'deepseek-r1-distill-llama-70b',
                'gemma2-9b-it',
                'gemma-7b-it',
                'llama3-70b-8192',
                'llama3-8b-8192',
            ];
        }
        
        echo '<select name="ai_blogger_model" class="regular-text">';
        foreach ($models as $model) {
            echo '<option value="' . esc_attr($model) . '" ' . selected($selected_model, $model, false) . '>'
                . esc_html($model) . '</option>';
        }
        echo '</select>';

        if (empty($api_key)) {
            echo '<p class="description">' . esc_html__('Enter a valid API key to fetch the latest models from Groq.', 'ai-blogger') . '</p>';
        }
    }
    
    public function render_pexels_key_field() {
        $api_key = get_option('ai_blogger_pexels_key');
        echo '<input type="password" name="ai_blogger_pexels_key" value="' . esc_attr($api_key) . '" class="regular-text">';
        echo '<p class="description">' . esc_html__('Your Pexels API key can be obtained from ', 'ai-blogger') 
            . ' <a href="https://www.pexels.com/api/new/" target="_blank" rel="noopener noreferrer">'
            . esc_html__('Pexels API', 'ai-blogger') 
            . '</a></p>';
    }
}