mirror-chatterino2/src/ircmanager.hpp

#pragma once

#define TWITCH_MAX_MESSAGELENGTH 500

#include "messages/message.hpp"
#include "twitch/twitchuser.hpp"

#include <IrcMessage>
#include <QMap>
#include <QMutex>
#include <QNetworkAccessManager>
#include <QString>
#include <pajlada/signals/signal.hpp>

#include <memory>
#include <mutex>

namespace chatterino {

class ChannelManager;
class Resources;
class EmoteManager;
class WindowManager;

class IrcManager : public QObject
{
    Q_OBJECT

public:
    IrcManager(ChannelManager &_channelManager, Resources &_resources, EmoteManager &_emoteManager,
               WindowManager &_windowManager);

    void connect();
    void disconnect();

    bool isTwitchBlockedUser(QString const &username);
    bool tryAddIgnoredUser(QString const &username, QString &errorMessage);
    void addIgnoredUser(QString const &username);
    bool tryRemoveIgnoredUser(QString const &username, QString &errorMessage);
    void removeIgnoredUser(QString const &username);

    void sendMessage(const QString &channelName, const QString &message);

    void joinChannel(const QString &channelName);
    void partChannel(const QString &channelName);

    const twitch::TwitchUser &getUser() const;
    void setUser(const twitch::TwitchUser &account);

    pajlada::Signals::Signal<Communi::IrcPrivateMessage *> onPrivateMessage;

private:
    ChannelManager &channelManager;
    Resources &resources;
    EmoteManager &emoteManager;
    WindowManager &windowManager;

    // variables
    twitch::TwitchUser _account;

    pajlada::Settings::Setting<std::string> currentUser;

    std::shared_ptr<Communi::IrcConnection> writeConnection = nullptr;
    std::shared_ptr<Communi::IrcConnection> readConnection = nullptr;

    std::mutex connectionMutex;
    uint32_t connectionGeneration = 0;

    QMap<QString, bool> _twitchBlockedUsers;
    QMutex _twitchBlockedUsersMutex;

    QNetworkAccessManager networkAccessManager;

    // methods
    Communi::IrcConnection *createConnection(bool doRead);

    void refreshIgnoredUsers(const QString &username, const QString &oauthClient,
                             const QString &oauthToken);

    void beginConnecting();

    void privateMessageReceived(Communi::IrcPrivateMessage *message);
    void messageReceived(Communi::IrcMessage *message);

    void handleRoomStateMessage(Communi::IrcMessage *message);
    void handleClearChatMessage(Communi::IrcMessage *message);
    void handleUserStateMessage(Communi::IrcMessage *message);
    void handleWhisperMessage(Communi::IrcMessage *message);
    void handleUserNoticeMessage(Communi::IrcMessage *message);
};

}  // namespace chatterino