<?php
/**
 * 2007-2024 PrestaShop
 *
 * NOTICE OF LICENSE
 *
 * This source file is subject to the Academic Free License (AFL 3.0)
 * that is bundled with this package in the file LICENSE.txt.
 * It is also available through the world-wide-web at this URL:
 * http://opensource.org/licenses/afl-3.0.php
 * If you did not receive a copy of the license and are unable to
 * obtain it through the world-wide-web, please send an email
 * to license@prestashop.com so we can send you a copy immediately.
 *
 * DISCLAIMER
 *
 * Do not edit or add to this file if you wish to upgrade PrestaShop to newer
 * versions in the future. If you wish to customize PrestaShop for your
 * needs please refer to http://www.prestashop.com for more information.
 *
 *  @author    PrestaShop SA <contact@prestashop.com>
 *  @copyright 2007-2024 PrestaShop SA
 *  @license   http://opensource.org/licenses/afl-3.0.php  Academic Free License (AFL 3.0)
 *  International Registered Trademark & Property of PrestaShop SA
 */
namespace PrestaShop\Module\Chatgptcontentgenerator\Process;

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

use PrestaShop\Module\Chatgptcontentgenerator\Entity\GptContentPost;

class TranslatePost extends AbstractProcess
{
    private $post;
    private $isCron = false;

    const TRANSLATE_TYPE_TITLE = 'title';
    const TRANSLATE_TYPE_CONTENT = 'content';
    const TRANSLATE_TYPE_SHORT_CONTENT = 'short_content';

    public function __construct($id, $isCron = false)
    {
        parent::__construct();

        $this->post = new GptContentPost($id);
        $this->isCron = $isCron;
    }

    public function run($fromIso, $toIso = '')
    {
        if ($fromIso == $toIso) {
            return true;
        }

        if (!\Validate::isLoadedObject($this->post)) {
            throw new \Exception(sprintf('Post #%s is not found', $this->post->id));
        }

        if (!$fromLang = \Language::getLanguage(\Language::getIdByIso($fromIso))) {
            throw new \Exception('Initial language is not defined');
        }

        $toLanguages = [];

        if ($toIso) {
            $toLanguages[] = \Language::getLanguage(\Language::getIdByIso($toIso));
        } else {
            $toLanguages = \Language::getLanguages();
        }

        $res = [];
        foreach ($toLanguages as $toLang) {
            if ($toLang['id_lang'] == $fromLang['id_lang']) {
                continue;
            }

            $res[$toLang['iso_code']] = [
                self::TRANSLATE_TYPE_TITLE => $this->translateByType(self::TRANSLATE_TYPE_TITLE, $fromLang['iso_code'], $toLang['iso_code']),
                self::TRANSLATE_TYPE_CONTENT => $this->translateByType(self::TRANSLATE_TYPE_CONTENT, $fromLang['iso_code'], $toLang['iso_code']),
                self::TRANSLATE_TYPE_SHORT_CONTENT => $this->translateByType(self::TRANSLATE_TYPE_SHORT_CONTENT, $fromLang['iso_code'], $toLang['iso_code']),
            ];
        }

        return $res;
    }

    private function translateByType($type, $fromLangIso, $toLangIso, $text = '')
    {
        if (!$text) {
            $idLang = \Language::getIdByIso($fromLangIso);

            if (isset($this->post->{$type}[$idLang])) {
                $text = $this->post->{$type}[$idLang];
            }
        }

        if ($this->isCron && !$text) {
            return [];
        }

        switch ($type) {
            case self::TRANSLATE_TYPE_CONTENT:
                $text = $this->apiClient->translatePostContent($text, $fromLangIso, $toLangIso);
                break;

            default:
                $text = $this->apiClient->translateText($text, $fromLangIso, $toLangIso, 'blog');
                break;
        }

        return [
            'text' => $text['text'],
            'nbWords' => (!isset($text['nbWords']) ? str_word_count($text['text']) : $text['nbWords']),
            'inQueue' => $text['inQueue'],
            'requestId' => $text['requestId'],
        ];
    }
}