#pragma once #include "providers/irc/IrcConnection2.hpp" #include #include #include #include namespace chatterino { class Channel; using ChannelPtr = std::shared_ptr; class AbstractIrcServer { public: virtual ~AbstractIrcServer() = default; // connection void connect(); void disconnect(); void sendMessage(const QString &channelName, const QString &message); void sendRawMessage(const QString &rawMessage); // 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(IrcConnection *connection); virtual void onDisconnected(); virtual void onSocketError(); virtual std::shared_ptr getCustomChannel( const QString &channelName); virtual bool hasSeparateWriteConnection() const = 0; virtual QString cleanChannelName(const QString &dirtyChannelName); QMap> channels; std::mutex channelMutex; private: void initConnection(); std::unique_ptr writeConnection_ = nullptr; std::unique_ptr readConnection_ = nullptr; QTimer reconnectTimer_; int falloffCounter_ = 1; std::mutex connectionMutex_; // bool autoReconnect_ = false; }; } // namespace chatterino