settings.php

<?php
 
namespace Bitrix\Socialnetwork\Integration\AI;
 
use Bitrix\AI\Context;
use Bitrix\AI\Engine;
use Bitrix\Main\Loader;
use Bitrix\Main\Localization\Loc;
use Bitrix\AI\Tuning;
use Bitrix\Main\Entity;
 
 
final class Settings
{
    private const TEXT_CATEGORY = 'text';
    private const IMAGE_CATEGORY = 'image';
    private const TUNING_CODE_IMAGE = 'socialnetwork_allow_image_generate';
    private const TUNING_CODE_TEXT = 'socialnetwork_allow_text_generate';
    private const TUNING_CODE_IMAGE_COMMENT = 'socialnetwork_comment_allow_image_generate';
    private const TUNING_CODE_TEXT_COMMENT = 'socialnetwork_comment_allow_text_generate';
 
    public static function isTextAvailable(): bool
    {
        if (!self::checkEngineAvailable(self::TEXT_CATEGORY))
        {
            return false;
        }
 
        $item = (new Tuning\Manager())->getItem(self::TUNING_CODE_TEXT);
 
        return $item ? $item->getValue() : true;
    }
 
    public static function isTextCommentAvailable(): bool
    {
        if (!self::checkEngineAvailable(self::TEXT_CATEGORY))
        {
            return false;
        }
 
        $item = (new Tuning\Manager())->getItem(self::TUNING_CODE_TEXT_COMMENT);
 
        return $item ? $item->getValue() : true;
    }
 
    public static function isImageAvailable(): bool
    {
        if (!self::checkEngineAvailable(self::IMAGE_CATEGORY))
        {
            return false;
        }
 
        $item = (new Tuning\Manager())->getItem(self::TUNING_CODE_IMAGE);
 
        return $item ? $item->getValue() : true;
    }
 
    public static function isImageCommentAvailable(): bool
    {
        if (!self::checkEngineAvailable(self::IMAGE_CATEGORY))
        {
            return false;
        }
 
        $item = (new Tuning\Manager())->getItem(self::TUNING_CODE_IMAGE_COMMENT);
 
        return $item ? $item->getValue() : true;
    }
 
    private static function checkEngineAvailable(string $type): bool
    {
        if (!Loader::includeModule('ai'))
        {
            return false;
        }
 
        $engine = Engine::getByCategory($type, Context::getFake());
        if (!$engine)
        {
            return false;
        }
 
        return true;
    }
 
    public static function onTuningLoad(): Entity\EventResult
    {
        $result = new Entity\EventResult();
 
        $items = [];
        $groups = [];
 
        if (Engine::getByCategory(self::TEXT_CATEGORY, Context::getFake()))
        {
            $items[self::TUNING_CODE_TEXT] = [
                'group' => Tuning\Defaults::GROUP_TEXT,
                'header' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_ALLOW_TEXT_COPILOT_DESC'),
                'title' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_COPILOT_TITLE'),
                'type' => Tuning\Type::BOOLEAN,
                'default' => true,
                'sort' => 100,
            ];
 
            $items[self::TUNING_CODE_TEXT_COMMENT] = [
                'group' => Tuning\Defaults::GROUP_TEXT,
                'header' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_ALLOW_TEXT_COMMENT_COPILOT_DESC'),
                'title' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_COPILOT_COMMENT_TITLE'),
                'type' => Tuning\Type::BOOLEAN,
                'default' => true,
                'sort' => 110,
            ];
        }
 
        if (Engine::getByCategory(self::IMAGE_CATEGORY, Context::getFake()))
        {
            $items[self::TUNING_CODE_IMAGE] = [
                'group' => Tuning\Defaults::GROUP_IMAGE,
                'header' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_ALLOW_IMAGE_COPILOT_DESC'),
                'title' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_COPILOT_TITLE'),
                'type' => Tuning\Type::BOOLEAN,
                'default' => true,
                'sort' => 100,
            ];
 
            $items[self::TUNING_CODE_IMAGE_COMMENT] = [
                'group' => Tuning\Defaults::GROUP_IMAGE,
                'header' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_ALLOW_IMAGE_COMMENT_COPILOT_DESC'),
                'title' => Loc::getMessage('SOCIALNETWORK_AI_SETTINGS_COPILOT_COMMENT_TITLE'),
                'type' => Tuning\Type::BOOLEAN,
                'default' => true,
                'sort' => 110,
            ];
        }
 
        $result->modifyFields([
            'items' => $items,
            'groups' => $groups,
        ]);
 
        return $result;
    }
}