#pragma once #include "common/Channel.hpp" #include "providers/irc/IrcConnection2.hpp" #include #include #include #include namespace chatterino { class AbstractIrcServer { public: virtual ~AbstractIrcServer() = default; // connection IrcConnection *getReadConnection() const; IrcConnection *getWriteConnection() const; void connect(); void disconnect(); void sendMessage(const QString &channelName, const QString &message); // channels std::shared_ptr getOrAddChannel(const QString &dirtyChannelName); std::shared_ptr getChannelOrEmpty(const QString &dirtyChannelName); // signals pajlada::Signals::NoArgSignal connected; pajlada::Signals::NoArgSignal disconnected; // pajlada::Signals::Signal onPrivateMessage; void addFakeMessage(const QString &data); // iteration void forEachChannel(std::function func); protected: AbstractIrcServer(); virtual void initializeConnection(IrcConnection *connection, bool isRead, bool isWrite) = 0; virtual std::shared_ptr createChannel(const QString &channelName) = 0; virtual void privateMessageReceived(Communi::IrcPrivateMessage *message); virtual void messageReceived(Communi::IrcMessage *message); virtual void writeConnectionMessageReceived(Communi::IrcMessage *message); virtual void onConnected(); virtual void onDisconnected(); virtual std::shared_ptr getCustomChannel(const QString &channelName); virtual QString cleanChannelName(const QString &dirtyChannelName); QMap> channels; std::mutex channelMutex; private: void initConnection(); std::unique_ptr writeConnection_ = nullptr; std::unique_ptr readConnection_ = nullptr; std::mutex connectionMutex_; // bool autoReconnect_ = false; }; } // namespace chatterino