1 files changed, 113 insertions, 0 deletions
diff --git a/Swift/Controllers/Highlighting/Highlighter.cpp b/Swift/Controllers/Highlighting/Highlighter.cpp
new file mode 100644
index 0000000..b05de2d
--- /dev/null
+++ b/Swift/Controllers/Highlighting/Highlighter.cpp
@@ -0,0 +1,113 @@
+/*
+ * Copyright (c) 2012 Maciej Niedzielski
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+/*
+ * Copyright (c) 2014-2017 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
+ */
+
+#include <Swift/Controllers/Highlighting/Highlighter.h>
+
+#include <set>
+#include <string>
+
+#include <Swiften/Base/String.h>
+#include <Swiften/Base/format.h>
+#include <Swiften/Client/NickResolver.h>
+
+#include <Swift/Controllers/Highlighting/HighlightManager.h>
+#include <Swift/Controllers/Intl.h>
+#include <Swift/Controllers/XMPPEvents/MessageEvent.h>
+
+namespace Swift {
+
+Highlighter::Highlighter(HighlightManager* manager, NickResolver* nickResolver)
+    : manager_(manager), nickResolver_(nickResolver) {
+}
+
+void Highlighter::handleSystemNotifications(const ChatWindow::ChatMessage& message, std::shared_ptr<MessageEvent> event) {
+    if (std::shared_ptr<MessageEvent> messageEvent = std::dynamic_pointer_cast<MessageEvent>(event)) {
+        JID jid = messageEvent->getStanza()->getFrom();
+        std::string nickname = nickResolver_->jidToNick(jid);
+
+        std::string messageText = messageEvent->getStanza()->getBody().get_value_or("");
+        if (boost::starts_with(messageText, "/me ")) {
+            messageText = "*" + String::getSplittedAtFirst(messageText, ' ').second + "*";
+        }
+
+        if (message.getHighlightActionDirectMessage().isSystemNotificationEnabled()) {
+            // title: Romeo says
+            // message: message
+            std::string title = str(format(QT_TRANSLATE_NOOP("", "%1% says")) % nickname);
+            event->addNotification(title, messageText);
+        }
+        if (message.getHighlightActionGroupMessage().isSystemNotificationEnabled()) {
+            // title: Romeo in $roomJID says
+            // message: message
+            std::string roomName = jid.getNode();
+            std::string title = str(format(QT_TRANSLATE_NOOP("", "%1% in %2% says")) % nickname % roomName);
+            event->addNotification(title, messageText);
+        }
+        if (message.getHighlightActionOwnMention().isSystemNotificationEnabled()) {
+            // title: Romeo mentioned you in $roomJID
+            // message: message
+            std::string roomName = jid.getNode();
+            std::string title = str(format(QT_TRANSLATE_NOOP("", "%1% mentioned you in %2%")) % nickname % roomName);
+            event->addNotification(title, messageText);
+        }
+        if (message.getHighlightActionSender().isSystemNotificationEnabled()) {
+            // title: Romeo says
+            // message: message
+            auto title = str(format(QT_TRANSLATE_NOOP("", "%1% says")) % nickname);
+            event->addNotification(title, messageText);
+        }
+        for (auto&& part : message.getParts()) {
+            auto highlightPart = std::dynamic_pointer_cast<ChatWindow::ChatHighlightingMessagePart>(part);
+            if (highlightPart && highlightPart->action.isSystemNotificationEnabled()) {
+                // title: Romeo mentioned '$keyword'
+                // message: message
+                auto title = str(format(QT_TRANSLATE_NOOP("", "%1% mentioned '%2%'")) % nickname % highlightPart->text);
+                event->addNotification(title, messageText);
+            }
+        }
+    }
+}
+
+void Highlighter::handleSoundNotifications(const ChatWindow::ChatMessage& chatMessage) {
+    std::set<std::string> playedSoundPaths;
+    std::vector<HighlightAction> actionsToPlay;
+
+    // collect unique sounds to play
+    auto checkSoundActionAndQueueUnique = [&](const HighlightAction& action) {
+        if (action.getSoundFilePath()) {
+            auto soundFilePath = action.getSoundFilePath().get_value_or("");
+            if (playedSoundPaths.find(soundFilePath) == playedSoundPaths.end()) {
+                playedSoundPaths.insert(soundFilePath);
+                actionsToPlay.push_back(action);
+            }
+        }
+    };
+
+    for (auto&& part : chatMessage.getParts()) {
+        auto highlightMessage = std::dynamic_pointer_cast<ChatWindow::ChatHighlightingMessagePart>(part);
+        if (highlightMessage) {
+            checkSoundActionAndQueueUnique(highlightMessage->action);
+        }
+    }
+
+    checkSoundActionAndQueueUnique(chatMessage.getHighlightActionSender());
+    checkSoundActionAndQueueUnique(chatMessage.getHighlightActionOwnMention());
+    checkSoundActionAndQueueUnique(chatMessage.getHighlightActionDirectMessage());
+    checkSoundActionAndQueueUnique(chatMessage.getHighlightActionGroupMessage());
+
+    // play sounds
+    for (const auto& action : actionsToPlay) {
+        manager_->onHighlight(action);
+    }
+}
+
+}