summaryrefslogtreecommitdiffstats
blob: 88cb95c2d3af132560aa2c5a6ce633ac8763284a (plain)
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
/*
 * Copyright (c) 2010-2017 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <map>
#include <memory>
#include <string>
#include <vector>

#include <boost/date_time/posix_time/posix_time.hpp>
#include <boost/filesystem/path.hpp>
#include <boost/optional.hpp>
#include <boost/signals2.hpp>

#include <Swiften/Base/IDGenerator.h>
#include <Swiften/Elements/DiscoInfo.h>
#include <Swiften/Elements/ErrorPayload.h>
#include <Swiften/Elements/SecurityLabelsCatalog.h>
#include <Swiften/Elements/Stanza.h>
#include <Swiften/JID/JID.h>
#include <Swiften/MUC/MUCRegistry.h>
#include <Swiften/Presence/PresenceOracle.h>
#include <Swiften/Queries/IQRouter.h>

#include <Swift/Controllers/Highlighting/HighlightManager.h>
#include <Swift/Controllers/HistoryController.h>
#include <Swift/Controllers/UIInterfaces/ChatWindow.h>
#include <Swift/Controllers/XMPPEvents/MUCInviteEvent.h>
#include <Swift/Controllers/XMPPEvents/MessageEvent.h>

namespace Swift {
    class AutoAcceptMUCInviteDecider;
    class AvatarManager;
    class ChatMessageParser;
    class ChatWindowFactory;
    class EntityCapsProvider;
    class EventController;
    class HighlightManager;
    class Highlighter;
    class IQRouter;
    class NickResolver;
    class StanzaChannel;
    class UIEventStream;

    class ChatControllerBase : public boost::signals2::trackable {
        public:
            class StreamWindowMessageIDPair {
                public:
                    std::string idInStream;
                    std::string idInWindow;
            };

        public:
            virtual ~ChatControllerBase();
            void showChatWindow();
            void activateChatWindow();
            bool hasOpenWindow() const;
            virtual void setAvailableServerFeatures(std::shared_ptr<DiscoInfo> info);
            virtual void handleIncomingOwnMessage(std::shared_ptr<Message> /*message*/) {}
            void handleIncomingMessage(std::shared_ptr<MessageEvent> message);
            std::string addMessage(const ChatWindow::ChatMessage& chatMessage, const std::string& senderName, bool senderIsSelf, std::shared_ptr<SecurityLabel> label, const boost::filesystem::path& avatarPath, const boost::posix_time::ptime& time);
            void replaceMessage(const ChatWindow::ChatMessage& chatMessage, const std::string& id, const boost::posix_time::ptime& time);
            virtual void setOnline(bool online);
            void setEnabled(bool enabled);
            virtual void setToJID(const JID& jid) {toJID_ = jid;}
            /** Used for determining when something is recent.*/
            boost::signals2::signal<void (const std::string& /*activity*/)> onActivity;
            boost::signals2::signal<void ()> onUnreadCountChanged;
            boost::signals2::signal<void ()> onWindowClosed;
            int getUnreadCount();
            const JID& getToJID() {return toJID_;}
            void handleCapsChanged(const JID& jid);
            void setCanStartImpromptuChats(bool supportsImpromptu);
            virtual ChatWindow* detachChatWindow();
            boost::signals2::signal<void(ChatWindow* /*window to reuse*/, const std::vector<JID>& /*invite people*/, const std::string& /*reason*/)> onConvertToMUC;

        protected:
            ChatControllerBase(const JID& self, StanzaChannel* stanzaChannel, IQRouter* iqRouter, ChatWindowFactory* chatWindowFactory, const JID &toJID, NickResolver* nickResolver, PresenceOracle* presenceOracle, AvatarManager* avatarManager, bool useDelayForLatency, UIEventStream* eventStream, EventController* eventController, EntityCapsProvider* entityCapsProvider, HistoryController* historyController, MUCRegistry* mucRegistry, HighlightManager* highlightManager, std::shared_ptr<ChatMessageParser> chatMessageParser, AutoAcceptMUCInviteDecider* autoAcceptMUCInviteDecider);

            /**
             * Pass the Message appended, and the stanza used to send it.
             */
            virtual void postSendMessage(const std::string&, std::shared_ptr<Stanza>) {}
            virtual std::string senderDisplayNameFromMessage(const JID& from) = 0;
            virtual std::string senderHighlightNameFromMessage(const JID& from) = 0;
            virtual bool isIncomingMessageFromMe(std::shared_ptr<Message>) = 0;
            virtual void preHandleIncomingMessage(std::shared_ptr<MessageEvent>) {}
            virtual void addMessageHandleIncomingMessage(const JID& from, const ChatWindow::ChatMessage& message, const std::string& messageID, bool senderIsSelf, std::shared_ptr<SecurityLabel> label, const boost::posix_time::ptime& time) = 0;
            virtual void handleIncomingReplaceMessage(const JID& from, const ChatWindow::ChatMessage& chatMessage, const std::string& messageID, const std::string& idToReplace, bool senderIsSelf, std::shared_ptr<SecurityLabel> label, const boost::posix_time::ptime& timeStamp) = 0;
            virtual void postHandleIncomingMessage(std::shared_ptr<MessageEvent>, const ChatWindow::ChatMessage&) {}
            virtual void preSendMessageRequest(std::shared_ptr<Message>) {}
            virtual bool isFromContact(const JID& from);
            virtual boost::optional<boost::posix_time::ptime> getMessageTimestamp(std::shared_ptr<Message>) const = 0;
            virtual void dayTicked() {}
            virtual void handleBareJIDCapsChanged(const JID& jid) = 0;
            std::string getErrorMessage(std::shared_ptr<ErrorPayload>);
            virtual void setContactIsReceivingPresence(bool /* isReceivingPresence */) {}
            virtual void cancelReplaces() = 0;
            /** JID any iq for account should go to - bare except for PMs */
            virtual JID getBaseJID();
            virtual void logMessage(const std::string& message, const JID& fromJID, const JID& toJID, const boost::posix_time::ptime& timeStamp, bool isIncoming) = 0;
            ChatWindow::ChatMessage buildChatWindowChatMessage(const std::string& message, const std::string& senderName, bool senderIsSelf);
            void updateMessageCount();
            virtual bool shouldIgnoreMessage(std::shared_ptr<Message> /* message */) {
                return false;
            }

            /**
             * What JID should be used for last message correction (XEP-0308) tracking.
             */
            virtual JID messageCorrectionJID(const JID& fromJID) = 0;

        private:
            IDGenerator idGenerator_;
            std::string lastSentMessageStanzaID_;

            void handleSendMessageRequest(const std::string &body, bool isCorrectionMessage);
            void handleAllMessagesRead();
            void handleSecurityLabelsCatalogResponse(std::shared_ptr<SecurityLabelsCatalog>, ErrorPayload::ref error);
            void handleMUCInvitation(Message::ref message);
            void handleMediatedMUCInvitation(Message::ref message);
            void handleGeneralMUCInvitation(MUCInviteEvent::ref event);
            void handleContinuationsBroken();

        protected:
            JID selfJID_;
            std::vector<std::shared_ptr<StanzaEvent> > unreadMessages_;
            std::vector<std::shared_ptr<StanzaEvent> > targetedUnreadMessages_;
            StanzaChannel* stanzaChannel_;
            IQRouter* iqRouter_;
            ChatWindowFactory* chatWindowFactory_;
            ChatWindow* chatWindow_;
            JID toJID_;
            bool labelsEnabled_;
            std::map<JID, StreamWindowMessageIDPair> lastMessagesIDs_;
            PresenceOracle* presenceOracle_;
            AvatarManager* avatarManager_;
            bool useDelayForLatency_;
            EventController* eventController_;
            EntityCapsProvider* entityCapsProvider_;
            SecurityLabelsCatalog::Item lastLabel_;
            HistoryController* historyController_;
            MUCRegistry* mucRegistry_;
            Highlighter* highlighter_;
            std::shared_ptr<ChatMessageParser> chatMessageParser_;
            AutoAcceptMUCInviteDecider* autoAcceptMUCInviteDecider_;
            UIEventStream* eventStream_;
            bool lastWasPresence_ = false;
    };
}