Swift/Controllers/HighlightRule.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

/*
 * Copyright (c) 2012 Maciej Niedzielski
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

/*
 * Copyright (c) 2014-2016 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <string>
#include <vector>

#include <boost/archive/text_iarchive.hpp>
#include <boost/archive/text_oarchive.hpp>
#include <boost/regex.hpp>

#include <Swift/Controllers/HighlightAction.h>

namespace Swift {

    class HighlightRule {
        public:
            HighlightRule();

            enum MessageType { ChatMessage, MUCMessage };

            bool isMatch(const std::string& body, const std::string& sender, const std::string& nick, MessageType) const;

            const HighlightAction& getAction() const { return action_; }
            HighlightAction& getAction() { return action_; }

            const std::vector<std::string>& getSenders() const { return senders_; }
            void setSenders(const std::vector<std::string>&);
            const std::vector<boost::regex>& getSenderRegex() const { return senderRegex_; }

            const std::vector<std::string>& getKeywords() const { return keywords_; }
            void setKeywords(const std::vector<std::string>&);
            std::vector<boost::regex> getKeywordRegex(const std::string& nick) const;

            bool getNickIsKeyword() const { return nickIsKeyword_; }
            void setNickIsKeyword(bool);

            bool getMatchCase() const { return matchCase_; }
            void setMatchCase(bool);

            bool getMatchWholeWords() const { return matchWholeWords_; }
            void setMatchWholeWords(bool);

            bool getMatchChat() const { return matchChat_; }
            void setMatchChat(bool);

            bool getMatchMUC() const { return matchMUC_; }
            void setMatchMUC(bool);

            bool isEmpty() const;

        private:
            friend class boost::serialization::access;
            template<class Archive> void serialize(Archive & ar, const unsigned int version);

            static std::string boolToString(bool);
            static bool boolFromString(const std::string&);

            std::vector<std::string> senders_;
            std::vector<std::string> keywords_;
            bool nickIsKeyword_;

            mutable std::vector<boost::regex> senderRegex_;
            mutable std::vector<boost::regex> keywordRegex_;
            void updateRegex() const;
            boost::regex regexFromString(const std::string&) const;

            bool matchCase_;
            bool matchWholeWords_;

            bool matchChat_;
            bool matchMUC_;

            HighlightAction action_;
    };

    bool operator ==(HighlightRule const& a, HighlightRule const& b);

    template<class Archive>
    void HighlightRule::serialize(Archive& ar, const unsigned int /*version*/)
    {
        ar & senders_;
        ar & keywords_;
        ar & nickIsKeyword_;
        ar & matchChat_;
        ar & matchMUC_;
        ar & matchCase_;
        ar & matchWholeWords_;
        ar & action_;
        updateRegex();
    }

}