messagehandler.php

<?php
namespace Bitrix\Forum\Replica;
 
use Bitrix\Main\Loader;
 
if (Loader::includeModule('replica'))
{
    class MessageHandler extends \Bitrix\Replica\Client\BaseHandler
    {
        protected $tasksForumId = 0;
        protected $messageData = [];
 
        protected $tableName = "b_forum_message";
        protected $moduleId = "forum";
 
        protected $primary = array(
            "ID" => "auto_increment",
        );
        protected $predicates = array(
            "TOPIC_ID" => "b_forum_topic.ID",
        );
        protected $translation = array(
            "ID" => "b_forum_message.ID",
            "TOPIC_ID" => "b_forum_topic.ID",
            "AUTHOR_ID" => "b_user.ID",
            //TODO GUEST_ID
            "EDITOR_ID" => "b_user.ID",
        );
        protected $children = array(
        );
 
        protected $fileHandler = null;
 
        public function __construct()
        {
            $this->fileHandler = new \Bitrix\Forum\Replica\ForumMessageAttachmentHandler();
            if (\Bitrix\Main\Loader::includeModule('tasks'))
            {
                $this->tasksForumId = (int)\CTasksTools::GetForumIdForIntranet();
            }
        }
 
        public function initDataManagerEvents()
        {
            parent::initDataManagerEvents();
            $this->fileHandler->initDataManagerEvents();
        }
 
        public function onBeforeMessageAdd($data, $uploadDir): void
        {
            $this->messageData = $data;
        }
 
        public function onAfterMessageAdd(&$id, $message, $topicInfo, $forumInfo, $fields)
        {
            $op = \Bitrix\Replica\Db\Operation::writeInsert($this->tableName, array('ID'), array('ID' => $id));
            $nodes = $op->getTableRecord()->getNodes(true, false);
            if ($nodes)
            {
                $this->fileHandler->onAfterAdd($id, $message);
            }
        }
 
        public function onBeforeMessageUpdate(&$id, &$fields, &$uploadDir)
        {
            $mapper = \Bitrix\Replica\Mapper::getInstance();
            $map = $mapper->getByPrimaryValue($this->tableName, array('ID'), array('ID' => $id));
            if ($map)
            {
                $this->fileHandler->onBeforeUpdate($id);
            }
        }
 
        public function onAfterMessageUpdate(&$id, &$newMessage, $oldMessage)
        {
            $op = \Bitrix\Replica\Db\Operation::writeUpdate($this->tableName, array('ID'), array('ID' => $id));
            $mapper = \Bitrix\Replica\Mapper::getInstance();
            $map = $mapper->getByPrimaryValue($this->tableName, array('ID'), array('ID' => $id));
            if ($map)
            {
                $nodes = current($map);
                $this->fileHandler->onAfterUpdate($id, $newMessage, $nodes);
            }
        }
 
        public function onBeforeMessageDelete($id)
        {
            $mapper = \Bitrix\Replica\Mapper::getInstance();
            $map = $mapper->getByPrimaryValue($this->tableName, array('ID'), array('ID' => $id));
            if ($map)
            {
                $this->fileHandler->onBeforeDelete($id);
            }
        }
 
        public function onAfterMessageDelete($id, $message)
        {
            $this->fileHandler->onAfterDelete($id);
            $op = \Bitrix\Replica\Db\Operation::writeDelete($this->tableName, array('ID'), array('ID' => $id));
        }
 
        public function afterWriteMissing(array $record)
        {
            //AddMessage2Log($record);
            $mapper = \Bitrix\Replica\Mapper::getInstance();
            $map = $mapper->getByPrimaryValue($this->tableName, array('ID'), array('ID' => $record["ID"]));
            if ($map)
            {
                $this->fileHandler->onAfterAdd($record["ID"], $record);
            }
        }
 
        public function beforeInsertTrigger(array &$newRecord)
        {
            if ($this->tasksForumId > 0 && $newRecord['FORUM_ID'] === 'tasks_forum')
            {
                $mapper = \Bitrix\Replica\Mapper::getInstance();
                $newRecord['FORUM_ID'] = $this->tasksForumId;
 
                if (preg_match("/^TASK_(.+)\$/", $newRecord["XML_ID"], $match))
                {
                    $taskId = $mapper->resolveLogGuid(false, 'b_tasks.ID', $match[1]);
                    if ($taskId)
                    {
                        $newRecord['XML_ID'] = 'TASK_'.$taskId;
                        $newRecord['PARAM2'] = $taskId;
                    }
                }
 
                if ($newRecord["PARAM1"] === "TK" && $newRecord["PARAM2"])
                {
                    $taskId = $mapper->resolveLogGuid(false, 'b_tasks.ID', $newRecord["PARAM2"]);
                    if ($taskId)
                    {
                        $newRecord['PARAM2'] = $taskId;
                    }
                }
 
                $this->fileHandler->replaceGuidsWithFiles($newRecord);
 
                $fixed = $this->clearUserBbCodes($newRecord['POST_MESSAGE']);
                if ($fixed != null)
                {
                    $newRecord["POST_MESSAGE"] = $fixed;
                }
 
                $fixed = $this->clearUserBbCodes($newRecord['POST_MESSAGE_HTML']);
                if ($fixed != null)
                {
                    $newRecord["POST_MESSAGE_HTML"] = $fixed;
                }
 
                $this->messageData = $newRecord;
            }
            return null;
        }
 
        public function afterInsertTrigger(array $newRecord)
        {
            if (
                $this->tasksForumId > 0
                && $newRecord['FORUM_ID'] == $this->tasksForumId
                && $newRecord['PARAM1'] !== 'TK'
                && $newRecord['PARAM2'] > 0
            )
            {
                $taskId = $newRecord['PARAM2'];
                if ($taskId > 0)
                {
                    $fields = ['AUX', 'AUX_DATA'];
                    foreach ($fields as $key)
                    {
                        if (array_key_exists($key, $this->messageData))
                        {
                            $newRecord[$key] = $this->messageData[$key];
                        }
                    }
 
                    \Bitrix\Tasks\Integration\Forum\Task\Comment::onAfterAdd(
                        'TK',
                        $taskId,
                        array(
                            "replica" => true, //will suppress ::fireEvent('Add',
                            "TOPIC_ID" => $newRecord["TOPIC_ID"],
                            "MESSAGE_ID" => $newRecord["ID"],
                            "PARAMS" => array(
                                "POST_MESSAGE" => $newRecord["POST_MESSAGE"],
                                "AUTHOR_ID" => $newRecord["AUTHOR_ID"],
                                "AUTHOR_NAME" => $newRecord["AUTHOR_NAME"],
                                "AUTHOR_EMAIL" => $newRecord["AUTHOR_EMAIL"],
                                "USE_SMILES" => $newRecord["USE_SMILES"],
                                //TODO "FILES" => $params["FILES"]
                                "AUTHOR_IP" => $newRecord["AUTHOR_IP"],
                                "AUTHOR_REAL_IP" => $newRecord["AUTHOR_REAL_IP"],
                                "GUEST_ID" => $newRecord["GUEST_ID"],
                                "AUX" => ($newRecord["AUX"] ?? 'N'),
                            ),
                            "MESSAGE" => $newRecord,
                            "AUX_DATA" => ($newRecord["AUX_DATA"] ?? ''),
                        )
                    );
                }
            }
        }
 
        public function afterUpdateTrigger(array $oldRecord, array $newRecord)
        {
            if (
                $this->tasksForumId > 0
                && $newRecord['FORUM_ID'] == 'tasks_forum'
                && $newRecord['PARAM1'] !== 'TK'
            )
            {
                $taskId = 0;
                if (preg_match("/^TASK_(.+)\$/", $newRecord["XML_ID"], $match))
                {
                    $mapper = \Bitrix\Replica\Mapper::getInstance();
                    $taskId = $mapper->resolveLogGuid(false, 'b_tasks.ID', $match[1]);
                }
 
                if ($taskId > 0)
                {
                    \Bitrix\Tasks\Integration\Forum\Task\Comment::onAfterUpdate(
                        'TK',
                        $taskId,
                        array(
                            "ACTION" => "EDIT",
                            "replica" => true, //will suppress ::fireEvent('Add',
                            "TOPIC_ID" => $newRecord["TOPIC_ID"],
                            "MESSAGE_ID" => $newRecord["ID"],
                            "PARAMS" => array(
                                "POST_MESSAGE" => $newRecord["POST_MESSAGE"],
                                "AUTHOR_ID" => $newRecord["AUTHOR_ID"],
                                "AUTHOR_NAME" => $newRecord["AUTHOR_NAME"],
                                "AUTHOR_EMAIL" => $newRecord["AUTHOR_EMAIL"],
                                "USE_SMILES" => $newRecord["USE_SMILES"],
                                //TODO "FILES" => $params["FILES"]
                                "AUTHOR_IP" => $newRecord["AUTHOR_IP"],
                                "AUTHOR_REAL_IP" => $newRecord["AUTHOR_REAL_IP"],
                                "GUEST_ID" => $newRecord["GUEST_ID"],
                            ),
                            "MESSAGE" => $newRecord,
                        )
                    );
                }
            }
        }
 
        public function afterDeleteTrigger(array $oldRecord)
        {
            if (
                $this->tasksForumId > 0
                && $oldRecord['FORUM_ID'] == $this->tasksForumId
            )
            {
                $taskId = 0;
                if (preg_match("/^TASK_(.+)\$/", $oldRecord["XML_ID"], $match))
                {
                    $taskId = intval($match[1]);
                }
 
                if ($taskId > 0)
                {
                    \Bitrix\Tasks\Integration\Forum\Task\Comment::onAfterUpdate(
                        'TK',
                        $taskId,
                        array(
                            "ACTION" => "DEL",
                            "replica" => true, //will suppress ::fireEvent('Add',
                            "TOPIC_ID" => $oldRecord["TOPIC_ID"],
                            "MESSAGE_ID" => $oldRecord["ID"],
                            "MESSAGE" => $oldRecord,
                        )
                    );
                }
            }
        }
 
        public function beforeUpdateTrigger(array $oldRecord, array &$newRecord)
        {
            if ($this->tasksForumId > 0 && $newRecord['FORUM_ID'] === 'tasks_forum')
            {
                $newRecord['FORUM_ID'] = $this->tasksForumId;
                if (preg_match("/^TASK_(.+)\$/", $newRecord["XML_ID"], $match))
                {
                    $mapper = \Bitrix\Replica\Mapper::getInstance();
                    $taskId = $mapper->resolveLogGuid(false, 'b_tasks.ID', $match[1]);
                    if ($taskId)
                    {
                        $newRecord['XML_ID'] = 'TASK_'.$taskId;
                        $newRecord['PARAM2'] = $taskId;
                    }
                }
 
                $fixed = $this->clearUserBbCodes($newRecord['POST_MESSAGE']);
                if ($fixed != null)
                {
                    $newRecord["POST_MESSAGE"] = $fixed;
                }
 
                $fixed = $this->clearUserBbCodes($newRecord['POST_MESSAGE_HTML']);
                if ($fixed != null)
                {
                    $newRecord["POST_MESSAGE_HTML"] = $fixed;
                }
 
                $this->fileHandler->replaceGuidsWithFiles($newRecord);
            }
        }
 
        public function beforeLogInsert(array $record)
        {
            if ($this->tasksForumId > 0 && $record['FORUM_ID'] == $this->tasksForumId)
            {
                return true;
            }
            else
            {
                return false;
            }
        }
 
        public function beforeLogFormat(array &$record)
        {
            if ($this->tasksForumId > 0 && $record['FORUM_ID'] == $this->tasksForumId)
            {
                $mapper = \Bitrix\Replica\Mapper::getInstance();
                $record['FORUM_ID'] = 'tasks_forum';
 
                if (preg_match("/^TASK_([0-9]+)\$/", $record["XML_ID"], $match))
                {
                    $guid = $mapper->getLogGuid("b_tasks.ID", $match[1]);
                    if ($guid)
                        $record['XML_ID'] = 'TASK_'.$guid;
                }
 
                if ($record["PARAM1"] == "TK" && $record["PARAM2"] > 0)
                {
                    $guid = $mapper->getLogGuid("b_tasks.ID", $record["PARAM2"]);
                    if ($guid)
                        $record["PARAM2"] = $guid;
                }
 
                $this->fileHandler->replaceFilesWithGuids($record["ID"], $record);
 
                $fields = ['AUX', 'AUX_DATA'];
                foreach ($fields as $key)
                {
                    if (array_key_exists($key, $this->messageData))
                    {
                        $record[$key] = $this->messageData[$key];
                    }
                }
            }
        }
 
        private function clearUserBbCodes($string)
        {
            // (\\/|\\\\\\/)
            // \\/ if for usual messages with bb-code [USER=*]name[/USER]
            // \\\\\\/ is for service messages with bb-code [USER=*]name[\/USER]
            return preg_replace("/\\[USER=[0-9]+\\](.*?)\\[(\\/|\\\\\\/)USER\\]/", "\\1", $string);
        }
    }
}