controller.php

<?php
 
namespace Bitrix\Socialnetwork\Integration\AI;
 
use Bitrix\AI\Context;
use Bitrix\AI\Engine;
use Bitrix\Forum\MessageTable;
use Bitrix\Main\Config\Option;
use Bitrix\Main\Event;
use Bitrix\Main\Loader;
use Bitrix\Socialnetwork\Integration\AI\User\Author;
 
final class Controller
{
    private static array $listMessages = [];
    private static int $blogAuthorId = 0;
    private const LIMIT = 20;
 
    public static function onContextGetMessages(Event $event): array
    {
        $moduleId = $event->getParameter('module');
        $contextId = $event->getParameter('id');
        $contextParameters = $event->getParameter('params');
        $nextStep = $event->getParameter('next_step');
 
        $isCommentContext = (
            $moduleId === 'socialnetwork'
            && str_starts_with($contextId, 'sonet_comment_')
        );
        if ($isCommentContext)
        {
            $messages = [];
 
            $xmlId = is_string($contextParameters['xmlId'] ?? null) ? $contextParameters['xmlId'] : null;
            if (!$xmlId)
            {
                return ['messages' => []];
            }
 
            if (isset(self::$listMessages[$xmlId]))
            {
                return ['messages' => self::$listMessages[$xmlId]];
            }
 
            if (str_starts_with($xmlId, 'BLOG_') && Loader::includeModule('blog'))
            {
                $postId = (int) mb_substr($xmlId, 5);
                $postMessages = self::getPostContext($postId);
                foreach ($postMessages as $postMessage)
                {
                    $messages[] = ['content' => $postMessage];
                }
            }
            else if (
                str_starts_with($xmlId, 'TASK_')
                && Loader::includeModule('tasks')
                && Loader::includeModule('forum')
            )
            {
                $postMessages = self::getTaskContext($xmlId);
                foreach ($postMessages as $postMessage)
                {
                    $messages[] = ['content' => $postMessage];
                }
            }
 
            $messages[0] = self::modifyOriginalMessage($messages[0] ?? []);
 
            if ($messages)
            {
                self::$listMessages[$xmlId] = $messages;
            }
 
            return ['messages' => $messages];
        }
 
        return ['messages' => []];
    }
 
    private static function getPostContext(int $postId): array
    {
        $messages = [];
 
        $textParser = new \CTextParser();
 
        $post = \CBlogPost::getByID($postId);
        if ($post)
        {
            self::setBlogAuthorId((int)$post['AUTHOR_ID']);
 
            $messages[] = $textParser->clearAllTags($post['DETAIL_TEXT']);
 
            $comments = self::getLastComments($postId);
 
            $messages = array_merge($messages, $comments);
        }
 
        return $messages;
    }
 
    private static function getTaskContext(string $xmlId): array
    {
        $taskId = (int) mb_substr($xmlId, 5);
 
        $textParser = new \CTextParser();
 
        $messages = [];
 
        $task = \Bitrix\Tasks\Internals\Registry\TaskRegistry::getInstance()->getObject($taskId);
        self::setBlogAuthorId($task->getCreatedBy());
        $messages[] = $textParser->clearAllTags($task->getDescription());
 
        $liveFeedEntity = \Bitrix\Socialnetwork\Livefeed\Provider::init([
            'ENTITY_TYPE' => \Bitrix\Socialnetwork\Livefeed\Provider::DATA_ENTITY_TYPE_TASKS_TASK,
            'ENTITY_ID' => $taskId,
        ]);
        if ($liveFeedEntity)
        {
            $logId = (int) $liveFeedEntity->getLogId();
            if ($logId)
            {
                $comments = self::getForumComments($xmlId);
 
                $messages = array_merge($messages, array_reverse($comments));
            }
        }
 
        return $messages;
    }
 
    private static function getForumComments(string $xmlId): array
    {
        $textParser = new \CTextParser();
 
        $comments = [];
 
        $query = MessageTable::query();
        $query
            ->setSelect(['ID', 'POST_MESSAGE'])
            ->where('XML_ID', $xmlId)
            ->whereNull('SERVICE_TYPE')
            ->whereNull('PARAM1')
            ->setOrder(['POST_DATE' => 'desc'])
            ->setLimit(self::LIMIT);
 
        $postMessages = $query->exec()->fetchCollection();
        foreach ($postMessages as $postMessage)
        {
            $comments[] = $textParser->clearAllTags($postMessage->getPostMessage());
        }
 
        return $comments;
    }
 
    private static function getLastComments(int $postId): array
    {
        $textParser = new \CTextParser();
 
        $comments = [];
 
        $queryCommentObject = \CBlogComment::getList(
            ['ID' => 'DESC'],
            [
                'PUBLISH_STATUS' => BLOG_PUBLISH_STATUS_PUBLISH,
                'POST_ID' => $postId,
                '!=POST_TEXT' => \Bitrix\Socialnetwork\CommentAux\TaskInfo::POST_TEXT,
            ],
            false,
            [
                'nTopCount' => self::LIMIT
            ],
            ['POST_TEXT']
        );
        while ($commentData = $queryCommentObject->fetch())
        {
            $comments[] = $textParser->clearAllTags($commentData['POST_TEXT']);
        }
 
        return $comments;
    }
 
    private static function modifyOriginalMessage(array $message): array
    {
        $message['is_original_message'] = true;
 
        if (self::$blogAuthorId)
        {
            $author = new Author(self::$blogAuthorId);
            $message['meta'] = $author->toMeta();
        }
 
        return $message;
    }
 
    private static function setBlogAuthorId(int $userId): void
    {
        self::$blogAuthorId = $userId;
    }
}