mirror of
https://github.com/Chatterino/chatterino2.git
synced 2024-11-21 22:24:07 +01:00
82 lines
2.3 KiB
C++
82 lines
2.3 KiB
C++
#pragma once
|
|
|
|
#include "providers/irc/IrcConnection2.hpp"
|
|
|
|
#include <IrcMessage>
|
|
#include <pajlada/signals/signal.hpp>
|
|
|
|
#include <functional>
|
|
#include <mutex>
|
|
|
|
namespace chatterino {
|
|
|
|
class Channel;
|
|
using ChannelPtr = std::shared_ptr<Channel>;
|
|
|
|
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<Channel> getOrAddChannel(const QString &dirtyChannelName);
|
|
std::shared_ptr<Channel> getChannelOrEmpty(const QString &dirtyChannelName);
|
|
|
|
// signals
|
|
pajlada::Signals::NoArgSignal connected;
|
|
pajlada::Signals::NoArgSignal disconnected;
|
|
// pajlada::Signals::Signal<Communi::IrcPrivateMessage *>
|
|
// onPrivateMessage;
|
|
|
|
void addFakeMessage(const QString &data);
|
|
|
|
// iteration
|
|
void forEachChannel(std::function<void(ChannelPtr)> func);
|
|
|
|
protected:
|
|
AbstractIrcServer();
|
|
|
|
virtual void initializeConnection(IrcConnection *connection, bool isRead,
|
|
bool isWrite) = 0;
|
|
virtual std::shared_ptr<Channel> 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<Channel> getCustomChannel(
|
|
const QString &channelName);
|
|
|
|
virtual bool hasSeparateWriteConnection() const = 0;
|
|
virtual QString cleanChannelName(const QString &dirtyChannelName);
|
|
|
|
QMap<QString, std::weak_ptr<Channel>> channels;
|
|
std::mutex channelMutex;
|
|
|
|
private:
|
|
void initConnection();
|
|
|
|
std::unique_ptr<IrcConnection> writeConnection_ = nullptr;
|
|
std::unique_ptr<IrcConnection> readConnection_ = nullptr;
|
|
|
|
QTimer reconnectTimer_;
|
|
int falloffCounter_ = 1;
|
|
|
|
std::mutex connectionMutex_;
|
|
|
|
// bool autoReconnect_ = false;
|
|
};
|
|
|
|
} // namespace chatterino
|