question.php

<?php
namespace Bitrix\Vote\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\Result;
use Bitrix\Vote\AnswerTable;
use Bitrix\Vote\QuestionTable;
 
class Question extends CopyImplementer
{
    private $resetVotingResult = true;
 
    public function setResetVotingResult(bool $bool): void
    {
        $this->resetVotingResult = $bool;
    }
 
    private $answerCopier = null;
 
    public function setAnswerCopier(EntityCopier $answerCopier): void
    {
        $this->answerCopier = $answerCopier;
    }
 
    public function add(Container $container, array $fields)
    {
        $result = QuestionTable::add($fields);
        if ($result->isSuccess())
        {
            return $result->getId();
        }
        else
        {
            $this->result->addErrors($result->getErrors());
            return false;
        }
    }
 
    public function getFields(Container $container, $entityId)
    {
        $queryObject = QuestionTable::getById($entityId);
        return (($fields = $queryObject->fetch()) ? $fields : []);
    }
 
    public function prepareFieldsToCopy(Container $container, array $fields)
    {
        unset($fields["ID"]);
 
        $fields["FIELD_TYPE"] = (int) $fields["FIELD_TYPE"];
 
        if ($container->getParentId())
        {
            $fields["VOTE_ID"] = $container->getParentId();
        }
 
        if ($this->resetVotingResult)
        {
            unset($fields["COUNTER"]);
        }
 
        return $fields;
    }
 
    public function copyChildren(Container $container, $entityId, $copiedEntityId)
    {
        $results = [];
 
        $results[] = $this->copyAnswer($entityId, $copiedEntityId);
 
        return $this->getResult($results);
    }
 
    private function copyAnswer(int $questionId, int $copiedQuestionId)
    {
        if (!$this->answerCopier)
        {
            return new Result();
        }
 
        $containerCollection = new ContainerCollection();
        $queryObject = AnswerTable::getList(["order" => [], "filter" => ["QUESTION_ID" => $questionId]]);
        while ($question = $queryObject->fetch())
        {
            $container = new Container($question["ID"]);
            $container->setParentId($copiedQuestionId);
            $containerCollection[] = $container;
        }
 
        if (!$containerCollection->isEmpty())
        {
            return $this->answerCopier->copy($containerCollection);
        }
 
        return new Result();
    }
}