topic.php

<?php
namespace Bitrix\Forum\Copy\Implement;
 
use Bitrix\Main\Copy\Container;
use Bitrix\Main\Copy\ContainerCollection;
use Bitrix\Main\Copy\CopyImplementer;
use Bitrix\Main\Copy\EntityCopier;
use Bitrix\Main\Error;
use Bitrix\Main\Result;
 
class Topic extends CopyImplementer
{
    const TOPIC_COPY_ERROR = "TOPIC_COPY_ERROR";
 
    private $commentCopier;
 
    public function setCommentCopier(EntityCopier $commentCopier): void
    {
        $this->commentCopier = $commentCopier;
    }
 
    public function add(Container $container, array $fields)
    {
        $topicId = \CForumTopic::add($fields);
 
        if (!$topicId)
        {
            $this->result->addError(new Error("Error creating a new topic", self::TOPIC_COPY_ERROR));
        }
 
        return $topicId;
    }
 
    public function getFields(Container $container, $entityId)
    {
        $topic = \CForumTopic::GetByIDEx($entityId);
 
        return ($topic ? $topic : []);
    }
 
    public function prepareFieldsToCopy(Container $container, array $fields)
    {
        $fields = $this->cleanDataToCopy($fields);
 
        $dictionary = $container->getDictionary();
 
        if (!empty($dictionary["XML_ID"]))
        {
            $fields["XML_ID"] = $dictionary["XML_ID"];
        }
 
        return $fields;
    }
 
    public function copyChildren(Container $topicContainer, $topicId, $copiedTopicId)
    {
        if (!$this->commentCopier)
        {
            return new Result();
        }
 
        $containerCollection = new ContainerCollection();
 
        $topicDictionary = $topicContainer->getDictionary();
 
        $queryObject = \CForumMessage::getList([], ["TOPIC_ID" => $topicId]);
        while ($forumMessage = $queryObject->Fetch())
        {
            $container = new Container($forumMessage["ID"]);
            $container->setParentId($copiedTopicId);
            $container->setDictionary($topicDictionary);
            $containerCollection[] = $container;
        }
 
        $results = [];
 
        if (!$containerCollection->isEmpty())
        {
            $results[] = $this->commentCopier->copy($containerCollection);
        }
 
        return $this->getResult($results);
    }
 
    private function cleanDataToCopy(array $fields)
    {
        unset($fields["ID"]);
        unset($fields["POSTS"]);
        unset($fields["START_DATE"]);
        unset($fields["LAST_POST_DATE"]);
        unset($fields["ABS_LAST_POST_DATE"]);
 
        return $fields;
    }
}