<?php

namespace ContentEgg\application\components\ai;

use ContentEgg\application\admin\GeneralConfig;
use ContentEgg\application\helpers\TextHelper;

defined('\ABSPATH') || exit;

/**
 * AiProcessor class file
 *
 * @author keywordrush.com <support@keywordrush.com>
 * @link https://www.keywordrush.com
 * @copyright Copyright &copy; 2025 keywordrush.com
 */

class AiProcessor
{

    public static function applayAiItems(array $items, $title_method = '', $description_method = '')
    {
        $is_selected = self::isSelected($items);
        foreach ($items as $i => $item)
        {
            if ($is_selected && empty($item['_selected']))
                continue;

            $items[$i] = self::applayAiItem($item, $title_method, $description_method);
        }

        return $items;
    }

    public static function isSelected(array $items)
    {
        foreach ($items as $item)
        {
            if (isset($item['_selected']) && $item['_selected'])
                return true;
        }

        return false;
    }

    public static function applayAiItem(array $item, $title_method = '', $description_method = '')
    {
        if (!GeneralConfig::getInstance()->option('ai_key'))
            return $item;

        if (!$title_method && !$description_method)
            return $item;

        $prompt = self::createProductPrompt();

        $prompt->setProduct($item);
        $prompt->setProductNew($item);

        if (\ContentEgg\application\Plugin::isDevEnvironment())
            mt_srand(12345678);

        $title_methods = array(
            'rephrase' => 'rephraseProductTitle',
            'translate' => 'translateProductTitle',
            'shorten' => 'shortenProductTitle',
            'subtitle_perfect_for' => 'generateSubtitlePerfectFor',
            'prompt1' => 'customPromptTitle2',
            'prompt2' => 'customPromptTitle2',
            'prompt3' => 'customPromptTitle3',
            'prompt4' => 'customPromptTitle4',
        );

        if ($title_method && isset($title_methods[$title_method]))
        {
            $method = $title_methods[$title_method];
            if (method_exists($prompt, $method))
            {
                try
                {
                    $title = $prompt->$method();
                    if ($title)
                    {
                        if (strstr($title_method, 'subtitle'))
                            $item['subtitle'] = $title;
                        else
                            $item['title'] = $title;
                    }
                }
                catch (\Exception $e)
                {
                    throw new \Exception(
                        'AI: Title generation error: ' . esc_html(wp_strip_all_tags($e->getMessage()))
                    );
                }
            }
        }

        $description_methods = array(
            'rewrite' => 'rewriteProductDescription',
            'paraphrase' => 'paraphraseProductDescription',
            'translate' => 'translateProductDescription',
            'summarize' => 'summarizeProductDescription',
            'bullet_points' => 'bulletPointsProductDescription',
            'bullet_points_compact' => 'bulletPointsCompactProductDescription',
            'turn_into_advertising' => 'turnIntoAdvertisingProductDescription',
            'cta_text' => 'ctaTextProductDescription',
            'write_paragraphs' => 'writeParagraphsProductDescription',
            'craft_description' => 'craftProductDescription',
            'write_article' => 'writeArticleProductDescription',
            'write_buyers_guide' => 'writeBuyersGuideProductDescription',
            'write_review' => 'writeReviewProductDescription',
            'write_how_to_use' => 'writeHowToUseProductDescription',
            'prompt1' => 'customPromptDescription1',
            'prompt2' => 'customPromptDescription2',
            'prompt3' => 'customPromptDescription3',
            'prompt4' => 'customPromptDescription4',
        );

        if ($description_method && isset($description_methods[$description_method]))
        {
            $method = $description_methods[$description_method];
            if (method_exists($prompt, $method))
            {
                try
                {
                    $item['description'] = trim($prompt->$method());
                }
                catch (\Exception $e)
                {
                    throw new \Exception(
                        'AI: Description generation error: ' . esc_html(wp_strip_all_tags($e->getMessage()))
                    );
                }
            }
        }

        return $item;
    }

    public static function createProductPrompt(): ProductPrompt
    {
        $config    = GeneralConfig::getInstance();
        $apiKeys   = explode(',', $config->option('ai_key'));
        $api_key   = trim($apiKeys[array_rand($apiKeys)]);
        $model     = $config->option('ai_model');
        $lang      = $config->option('ai_language');
        $temp      = $config->option('ai_temperature');
        $extraOpts = [];

        if ($model === 'openrouter/auto')
        {
            $openList   = $config->option('openrouter_models');
            $extraOpts  = TextHelper::getArrayFromCommaList($openList);
        }

        // reproducible in dev
        if (\ContentEgg\application\Plugin::isDevEnvironment())
        {
            mt_srand(12345678);
        }

        $prompt = new ProductPrompt($api_key, $model, $extraOpts);
        $prompt->setLang($lang);
        $prompt->setTemperature($temp);

        return $prompt;
    }

    public static function applaySmartGroups(array $data, $method)
    {
        if (!$api_key = GeneralConfig::getInstance()->option('ai_key'))
            return $data;

        $model = GeneralConfig::getInstance()->option('ai_model');

        $api_key = explode(',', $api_key);
        $api_key = trim($api_key[array_rand($api_key)]);

        if ($model == 'openrouter/auto')
        {
            $openrouter_models_value = GeneralConfig::getInstance()->option('openrouter_models');
            $openrouter_models = TextHelper::getArrayFromCommaList($openrouter_models_value);
        }
        else
        {
            $openrouter_models = array();
        }

        $prompt = new SmartGroupsPrompt($api_key, $model, $openrouter_models);
        $prompt->setData($data);

        if (\ContentEgg\application\Plugin::isDevEnvironment())
            mt_srand(12345678);

        $methods = array(
            'price_comparison' => 'categorizePriceComparison',
            'auto' => 'categorizeAuto',
            'product_category' => 'categorizeProductCategory',
            'features' => 'categorizeFeatures',
            'brand' => 'categorizeBrand',
            'price_range' => 'categorizePriceRange',
            'by_usage' => 'categorizeUsage',
            'age_group' => 'categorizeAgeGroup',
            'material_ingredients' => 'categorizeMaterialIngredients',
            'size_volume' => 'categorizeSizeVolume',
        );

        if (!isset($methods[$method]))
            return $data;

        $call_method = $methods[$method];
        if (method_exists($prompt, $call_method))
        {
            try
            {
                $data = $prompt->$call_method();
            }
            catch (\Exception $e)
            {
                throw new \Exception(
                    'AI: Smart Groups error: ' . esc_html(wp_strip_all_tags($e->getMessage()))
                );
            }
        }

        return $data;
    }
}