1 files changed, 152 insertions, 0 deletions
diff --git a/Swift/Controllers/HighlightRule.cpp b/Swift/Controllers/HighlightRule.cpp
new file mode 100644
index 0000000..3251067
--- /dev/null
+++ b/Swift/Controllers/HighlightRule.cpp
@@ -0,0 +1,152 @@
+/*
+ * Copyright (c) 2012 Maciej Niedzielski
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+/*
+ * Copyright (c) 2014 Kevin Smith and Remko Tronçon
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <algorithm>
+#include <boost/algorithm/string.hpp>
+#include <boost/lambda/lambda.hpp>
+
+#include <Swiften/Base/foreach.h>
+#include <Swiften/Base/Regex.h>
+#include <Swift/Controllers/HighlightRule.h>
+
+namespace Swift {
+
+HighlightRule::HighlightRule()
+	: nickIsKeyword_(false)
+	, matchCase_(false)
+	, matchWholeWords_(false)
+	, matchChat_(false)
+	, matchMUC_(false)
+{
+}
+
+boost::regex HighlightRule::regexFromString(const std::string & s) const
+{
+	std::string escaped = Regex::escape(s);
+	std::string word = matchWholeWords_ ? "\\b" : "";
+	boost::regex::flag_type flags = boost::regex::normal;
+	if (!matchCase_) {
+		flags |= boost::regex::icase;
+	}
+	return boost::regex(word + escaped + word, flags);
+}
+
+void HighlightRule::updateRegex() const
+{
+	keywordRegex_.clear();
+	foreach (const std::string & k, keywords_) {
+		keywordRegex_.push_back(regexFromString(k));
+	}
+	senderRegex_.clear();
+	foreach (const std::string & s, senders_) {
+		senderRegex_.push_back(regexFromString(s));
+	}
+}
+
+std::string HighlightRule::boolToString(bool b)
+{
+	return b ? "1" : "0";
+}
+
+bool HighlightRule::boolFromString(const std::string& s)
+{
+	return s == "1";
+}
+
+bool HighlightRule::isMatch(const std::string& body, const std::string& sender, const std::string& nick, MessageType messageType) const
+{
+	if ((messageType == HighlightRule::ChatMessage && matchChat_) || (messageType == HighlightRule::MUCMessage && matchMUC_)) {
+
+		bool matchesKeyword = keywords_.empty() && (nick.empty() || !nickIsKeyword_);
+		bool matchesSender = senders_.empty();
+
+		if (!matchesKeyword && nickIsKeyword_ && !nick.empty()) {
+			if (boost::regex_search(body, regexFromString(nick))) {
+				matchesKeyword = true;
+			}
+		}
+
+		foreach (const boost::regex & rx, senderRegex_) {
+			if (boost::regex_search(sender, rx)) {
+				matchesSender = true;
+				break;
+			}
+		}
+
+		if (matchesKeyword && matchesSender) {
+			return true;
+		}
+	}
+
+	return false;
+}
+
+void HighlightRule::setSenders(const std::vector<std::string>& senders)
+{
+	senders_ = senders;
+	updateRegex();
+}
+
+void HighlightRule::setKeywords(const std::vector<std::string>& keywords)
+{
+	keywords_ = keywords;
+	updateRegex();
+}
+
+std::vector<boost::regex> HighlightRule::getKeywordRegex(const std::string& nick) const {
+	if (nickIsKeyword_) {
+		std::vector<boost::regex> regex;
+		if (!nick.empty()) {
+			regex.push_back(regexFromString(nick));
+		}
+		return regex;
+	} else {
+		return keywordRegex_;
+	}
+}
+
+void HighlightRule::setNickIsKeyword(bool nickIsKeyword)
+{
+	nickIsKeyword_ = nickIsKeyword;
+	updateRegex();
+}
+
+void HighlightRule::setMatchCase(bool matchCase)
+{
+	matchCase_ = matchCase;
+	updateRegex();
+}
+
+void HighlightRule::setMatchWholeWords(bool matchWholeWords)
+{
+	matchWholeWords_ = matchWholeWords;
+	updateRegex();
+}
+
+void HighlightRule::setMatchChat(bool matchChat)
+{
+	matchChat_ = matchChat;
+	updateRegex();
+}
+
+void HighlightRule::setMatchMUC(bool matchMUC)
+{
+	matchMUC_ = matchMUC;
+	updateRegex();
+}
+
+bool HighlightRule::isEmpty() const
+{
+	return senders_.empty() && keywords_.empty() && !nickIsKeyword_ && !matchChat_ && !matchMUC_ && action_.isEmpty();
+}
+
+}