<?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;

		/**
		 * MessageHandler constructor.
		 *
		 */
		public function __construct()
		{
			$this->fileHandler = new \Bitrix\Forum\Replica\ForumMessageAttachmentHandler();
			if (\Bitrix\Main\Loader::includeModule('tasks'))
			{
				$this->tasksForumId = (int)\CTasksTools::GetForumIdForIntranet();
			}
		}

		/**
		 * Registers event handlers for database operations like add new, update or delete.
		 *
		 * @return void
		 */
		public function initDataManagerEvents()
		{
			parent::initDataManagerEvents();
			$this->fileHandler->initDataManagerEvents();
		}

		/**
		 * Forum event onBeforeMessageAdd handler.
		 *
		 * @param $data
		 * @param $uploadDir
		 *
		 * @see \CForumMessage::Add()
		 * @see \Bitrix\Forum\Message::add()
		 */
		public function onBeforeMessageAdd($data, $uploadDir): void
		{
			$this->messageData = $data;
		}

		/**
		 * Forum event onAfterMessageAdd handler.
		 *
		 * @param integer &$id Forum message identifier.
		 * @param array $message Message record.
		 * @param array $topicInfo Forum topic fields.
		 * @param array $forumInfo Forum fields.
		 * @param array $fields Message record before add.
		 *
		 * @return void
		 * @see \CForumMessage::Add()
		 * @see \Bitrix\Forum\Message::add()
		 */
		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);
			}
		}

		/**
		 * Forum event onBeforeMessageUpdate handler.
		 *
		 * @param integer &$id Forum message identifier.
		 * @param array &$fields Message record.
		 * @param string &$uploadDir Forum topic fields.
		 *
		 * @return void
		 * @see \CForumMessage::Update()
		 */
		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);
			}
		}

		/**
		 * Forum event onAfterMessageUpdate handler.
		 *
		 * @param integer &$id Forum message identifier.
		 * @param array &$newMessage Message record.
		 * @param array $oldMessage Message record before update.
		 *
		 * @return void
		 * @see \CForumMessage::Update()
		 */
		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);
			}
		}

		/**
		 * Forum event onBeforeMessageDelete handler.
		 *
		 * @param integer $id Forum message identifier.
		 *
		 * @return void
		 * @see \CForumMessage::Delete()
		 */
		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);
			}
		}

		/**
		 * Forum event onAfterMessageDelete handler.
		 *
		 * @param integer $id Forum message identifier.
		 * @param array $message Message record before delete.
		 *
		 * @return void
		 * @see \CForumMessage::Delete()
		 */
		public function onAfterMessageDelete($id, $message)
		{
			$this->fileHandler->onAfterDelete($id);
			$op = \Bitrix\Replica\Db\Operation::writeDelete($this->tableName, array('ID'), array('ID' => $id));
		}

		/**
		 * Method will be invoked after writing an missed record.
		 *
		 * @param array $record All fields of the record.
		 *
		 * @return void
		 */
		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);
			}
		}

		/**
		 * Method will be invoked before new database record inserted.
		 * When an array returned the insert will be cancelled and map for
		 * returned record will be added.
		 *
		 * @param array &$newRecord All fields of inserted record.
		 *
		 * @return null|array
		 */
		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 = preg_replace("/\\[USER=[0-9]+\\](.*?)\\[\\/USER\\]/", "\\1", $newRecord["POST_MESSAGE"]);
				if ($fixed != null)
				{
					$newRecord["POST_MESSAGE"] = $fixed;
				}

				$fixed = preg_replace("/\\[USER=[0-9]+\\](.*?)\\[\\/USER\\]/", "\\1", $newRecord["POST_MESSAGE_HTML"]);
				if ($fixed != null)
				{
					$newRecord["POST_MESSAGE_HTML"] = $fixed;
				}

				$this->messageData = $newRecord;
			}
			return null;
		}

		/**
		 * Method will be invoked after new database record inserted.
		 *
		 * @param array $newRecord All fields of inserted record.
		 *
		 * @return void
		 */
		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"] ?? ''),
						)
					);
				}
			}
		}

		/**
		 * Method will be invoked after an database record updated.
		 *
		 * @param array $oldRecord All fields before update.
		 * @param array $newRecord All fields after update.
		 *
		 * @return void
		 */
		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,
						)
					);
				}
			}
		}

		/**
		 * Method will be invoked after an database record deleted.
		 *
		 * @param array $oldRecord All fields before delete.
		 *
		 * @return void
		 */
		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,
						)
					);
				}
			}
		}

		/**
		 * Method will be invoked before an database record updated.
		 *
		 * @param array $oldRecord All fields before update.
		 * @param array &$newRecord All fields after update.
		 *
		 * @return void
		 */
		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 = preg_replace("/\\[USER=[0-9]+\\](.*?)\\[\\/USER\\]/", "\\1", $newRecord["POST_MESSAGE"]);
				if ($fixed != null)
				{
					$newRecord["POST_MESSAGE"] = $fixed;
				}

				$fixed = preg_replace("/\\[USER=[0-9]+\\](.*?)\\[\\/USER\\]/", "\\1", $newRecord["POST_MESSAGE_HTML"]);
				if ($fixed != null)
				{
					$newRecord["POST_MESSAGE_HTML"] = $fixed;
				}

				$this->fileHandler->replaceGuidsWithFiles($newRecord);
			}
		}

		/**
		 * Called before insert operation log write. You may return false and not log write will take place.
		 *
		 * @param array $record Database record.
		 *
		 * @return boolean
		 */
		public function beforeLogInsert(array $record)
		{
			if ($this->tasksForumId > 0 && $record['FORUM_ID'] == $this->tasksForumId)
			{
				return true;
			}
			else
			{
				return false;
			}
		}

		/**
		 * Called before record transformed for log writing.
		 *
		 * @param array &$record Database record.
		 *
		 * @return void
		 */
		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];
					}
				}
			}
		}
	}
}