1 files changed, 23 insertions, 16 deletions
diff --git a/Swift/Controllers/Highlighter.h b/Swift/Controllers/Highlighter.h
index d5d846b..9ad3339 100644
--- a/Swift/Controllers/Highlighter.h
+++ b/Swift/Controllers/Highlighter.h
@@ -4,35 +4,42 @@
  * See Documentation/Licenses/BSD-simplified.txt for more information.
  */
 
+/*
+ * Copyright (c) 2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
+ */
+
 #pragma once
 
 #include <string>
+#include <vector>
 
 #include <Swift/Controllers/HighlightRule.h>
 
 namespace Swift {
 
-	class HighlightManager;
+    class HighlightManager;
 
-	class Highlighter {
-		public:
-			Highlighter(HighlightManager* manager);
+    class Highlighter {
+        public:
+            Highlighter(HighlightManager* manager);
 
-			enum Mode { ChatMode, MUCMode };
-			void setMode(Mode mode);
+            enum Mode { ChatMode, MUCMode };
+            void setMode(Mode mode);
 
-			void setNick(const std::string& nick) { nick_ = nick; }
-			std::string getNick() const { return nick_; }
+            void setNick(const std::string& nick) { nick_ = nick; }
+            std::string getNick() const { return nick_; }
 
-			HighlightAction findAction(const std::string& body, const std::string& sender) const;
+            HighlightAction findFirstFullMessageMatchAction(const std::string& body, const std::string& sender) const;
 
-			void handleHighlightAction(const HighlightAction& action);
+            void handleHighlightAction(const HighlightAction& action);
 
-		private:
-			HighlightManager* manager_;
-			Mode mode_;
-			HighlightRule::MessageType messageType_;
-			std::string nick_;
-	};
+        private:
+            HighlightManager* manager_;
+            Mode mode_;
+            HighlightRule::MessageType messageType_;
+            std::string nick_;
+    };
 
 }