TaskCreateSite.php

См. документацию.

<?php
declare(strict_types=1);
 
namespace Bitrix\Landing\Copilot\Generation\Step;
 
use Bitrix\Landing\Copilot\Connector;
use Bitrix\Landing\Copilot\Generation\GenerationException;
use Bitrix\Landing\Copilot\Generation\Type\GenerationErrors;
use Bitrix\Landing\Landing;
use Bitrix\Landing\Manager;
use Bitrix\Landing\Rights;
use Bitrix\Landing\Site;
use CUtil;
 
class TaskCreateSite extends TaskStep
{
    private const DEFAULT_PAGE_CODE = 'create-by-copilot';

    public function execute(): bool
    {
        parent::execute();
 
        $this->createSite();
 
        return true;
    }

    private function createSite(): void
    {
        $siteId = $this->addSite();
        if ($siteId > 0)
        {
            $this->siteData->setSiteId($siteId);
            $landingId = $this->addLanding($siteId);
            if ($landingId > 0)
            {
                $this->siteData->setLandingId($landingId);
                $this->addAdditionalFieldsToLanding();
            }
 
            $previewSrc = $this->siteData->getPreviewImageSrc();
            if ($previewSrc === null || $previewSrc === '')
            {
                $defaultPreview = Manager::getUrlFromFile('/bitrix/images/landing/copilot-preview.jpg');
                $this->siteData->setPreviewImageSrc($defaultPreview);
                $previewSrc = $defaultPreview;
            }
 
            $additionalFields = [
                'METAOG_IMAGE' => $previewSrc,
            ];
            Rights::setGlobalOff();
            Landing::saveAdditionalFields($this->siteData->getLandingId(), $additionalFields);
            Rights::setGlobalOn();
 
            if ($this->generation->getChatId())
            {
                (new Connector\Chat\Chat())
                    ->setUserId($this->generation->getAuthorId())
                    ->setChatForSite(
                        $siteId,
                        $this->generation->getChatId()
                    )
                ;
            }
        }
    }

    protected function addSite(): int
    {
        $colors = $this->siteData->getColors();
        $additionalFields = [];
 
        $this->processColor($colors->theme, 'THEME_COLOR', $additionalFields);
        $this->processColor($colors->background, 'BACKGROUND_COLOR', $additionalFields);
        if (isset($additionalFields['BACKGROUND_COLOR']))
        {
            $additionalFields['BACKGROUND_USE'] = 'Y';
        }
        $this->processColor($colors->headersBg, 'THEMEFONTS_COLOR_H', $additionalFields);
        $this->processColor($colors->textsBg, 'THEMEFONTS_COLOR', $additionalFields);
 
        $fonts = $this->siteData->getFonts();
        $additionalFields['THEMEFONTS_CODE_H'] = $fonts->headers;
        $additionalFields['THEMEFONTS_CODE'] = $fonts->texts;
 
        $siteTitle = $this->siteData->getSiteTitle();
        $siteFields = [
            'TITLE' => $siteTitle,
            'CODE' => self::getTranslitedCode($siteTitle),
            'TYPE' => 'PAGE',
            'ADDITIONAL_FIELDS' => $additionalFields,
        ];
        Rights::setGlobalOff();
        $siteAddResult = Site::add($siteFields);
        Rights::setGlobalOn();
 
        return $siteAddResult->isSuccess() ? $siteAddResult->getId() : 0;
    }

    protected function addLanding(int $siteId): int
    {
        $pageTitle = $this->siteData->getPageTitle();
        $landingFields = [
            'SITE_ID' => $siteId,
            'TITLE' => $pageTitle,
            'CODE' => self::getTranslitedCode($pageTitle),
        ];
 
        Rights::setGlobalOff();
        $landingAddResult = Landing::add($landingFields);
        Rights::setGlobalOn();
 
        if (!$landingAddResult->isSuccess())
        {
            return 0;
        }
 
        return $landingAddResult->getId();
    }
 
    protected function addAdditionalFieldsToLanding(): void
    {
        $pageTitle = $this->siteData->getPageTitle();
        $pageDescription = $this->siteData->getPageDescription();
        $additionalFields = [
            'METAOG_DESCRIPTION' => $pageDescription,
            'METAMAIN_USE' => 'Y',
            'METAMAIN_TITLE' => $pageTitle,
            'METAMAIN_DESCRIPTION' => $pageDescription,
            'METAMAIN_KEYWORDS' => $this->siteData->getKeywords() ?? '',
        ];
 
        $landingId = $this->siteData->getLandingId();
 
        Rights::setGlobalOff();
        Landing::saveAdditionalFields($landingId, $additionalFields);
        Rights::setGlobalOn();
    }

    private function processColor(string $color, string $fieldKey, array &$additionalFields): void
    {
        $code = $this->convertToLowercase($color);
        if ($this->isValidHexCode($code))
        {
            $additionalFields[$fieldKey] = $code;
        }
    }

    private function convertToLowercase(string $string): string
    {
        return strtr($string, 'ABCDEF', 'abcdef');
    }

    private function isValidHexCode(string $string): bool
    {
        return preg_match('/^#[0-9a-f]{6}$/', $string) === 1;
    }

    protected static function getTranslitedCode(string $code): ?string
    {
        $params = [
            "max_len" => 30,
            "change_case" => "L",
            "replace_space" => "-",
            "replace_other" => "",
            "delete_repeat_replace" => true,
        ];
        $translitedCode = CUtil::translit(
            $code,
            LANGUAGE_ID,
            $params
        );
        $pattern = '/[^-]/';
        if (LANGUAGE_ID !== 'en' && !preg_match($pattern, $translitedCode))
        {
            $translitedCode = CUtil::translit(
                $translitedCode,
                'en',
                $params
            );
        }
        if (!preg_match($pattern, $translitedCode))
        {
            $translitedCode = self::DEFAULT_PAGE_CODE;
        }
 
        return $translitedCode;
    }

    public function isFinished(): bool
    {
        $siteId = $this->siteData->getSiteId();
        $landingId = $this->siteData->getLandingId();
 
        if (is_int($siteId) && is_int($landingId) && $siteId > 0 && $landingId > 0)
        {
            return true;
        }
 
        if (is_int($siteId) && is_int($landingId))
        {
            $exceptionMessage = "Landing id not correct.";
        }
        else
        {
            $exceptionMessage = "Site id not correct.";
        }
 
        throw new GenerationException(
            GenerationErrors::dataValidation,
            $exceptionMessage,
        );
    }
}