2018-02-05 15:11:50 +01:00
|
|
|
#pragma once
|
|
|
|
|
2018-04-01 16:43:30 +02:00
|
|
|
#include "channel.hpp"
|
|
|
|
|
2018-04-03 02:55:32 +02:00
|
|
|
#include <IrcConnection>
|
2018-02-05 15:11:50 +01:00
|
|
|
#include <IrcMessage>
|
2018-04-03 02:55:32 +02:00
|
|
|
#include <pajlada/signals/signal.hpp>
|
|
|
|
|
2018-02-05 21:20:38 +01:00
|
|
|
#include <functional>
|
2018-02-05 15:11:50 +01:00
|
|
|
#include <mutex>
|
|
|
|
|
|
|
|
namespace chatterino {
|
|
|
|
namespace providers {
|
|
|
|
namespace irc {
|
2018-04-01 16:43:30 +02:00
|
|
|
|
2018-02-05 15:11:50 +01:00
|
|
|
class AbstractIrcServer
|
|
|
|
{
|
|
|
|
public:
|
2018-04-03 02:55:32 +02:00
|
|
|
virtual ~AbstractIrcServer() = default;
|
|
|
|
|
2018-02-05 15:11:50 +01:00
|
|
|
// connection
|
|
|
|
Communi::IrcConnection *getReadConnection() const;
|
|
|
|
|
|
|
|
void connect();
|
|
|
|
void disconnect();
|
|
|
|
|
|
|
|
void sendMessage(const QString &channelName, const QString &message);
|
|
|
|
|
|
|
|
// channels
|
2018-04-21 00:40:17 +02:00
|
|
|
std::shared_ptr<Channel> getOrAddChannel(const QString &dirtyChannelName);
|
|
|
|
std::shared_ptr<Channel> getChannelOrEmpty(const QString &dirtyChannelName);
|
2018-02-05 15:11:50 +01:00
|
|
|
|
|
|
|
// signals
|
|
|
|
pajlada::Signals::NoArgSignal connected;
|
|
|
|
pajlada::Signals::NoArgSignal disconnected;
|
|
|
|
pajlada::Signals::Signal<Communi::IrcPrivateMessage *> onPrivateMessage;
|
|
|
|
|
|
|
|
void addFakeMessage(const QString &data);
|
|
|
|
|
2018-02-05 21:20:38 +01:00
|
|
|
// iteration
|
|
|
|
void forEachChannel(std::function<void(ChannelPtr)> func);
|
|
|
|
|
2018-02-05 15:11:50 +01:00
|
|
|
protected:
|
|
|
|
AbstractIrcServer();
|
|
|
|
|
|
|
|
virtual void initializeConnection(Communi::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();
|
|
|
|
virtual void onDisconnected();
|
|
|
|
|
|
|
|
virtual std::shared_ptr<Channel> getCustomChannel(const QString &channelName);
|
|
|
|
|
2018-04-21 00:40:17 +02:00
|
|
|
virtual QString cleanChannelName(const QString &dirtyChannelName);
|
2018-04-01 15:10:15 +02:00
|
|
|
|
2018-02-05 21:20:38 +01:00
|
|
|
QMap<QString, std::weak_ptr<Channel>> channels;
|
|
|
|
std::mutex channelMutex;
|
|
|
|
|
2018-02-05 15:11:50 +01:00
|
|
|
private:
|
|
|
|
void initConnection();
|
|
|
|
|
|
|
|
std::unique_ptr<Communi::IrcConnection> writeConnection = nullptr;
|
|
|
|
std::unique_ptr<Communi::IrcConnection> readConnection = nullptr;
|
|
|
|
|
|
|
|
std::mutex connectionMutex;
|
|
|
|
};
|
2018-04-01 16:43:30 +02:00
|
|
|
|
2018-02-05 15:11:50 +01:00
|
|
|
} // namespace irc
|
|
|
|
} // namespace providers
|
|
|
|
} // namespace chatterino
|