mirror of
https://github.com/WarmUpTill/SceneSwitcher.git
synced 2026-03-21 17:34:57 -05:00
6ba01dca39
If the connection was aborted by the local machine, the active subsciptions were not cleared causing the new connection not attempt to register any new subsciptions. An event sub connection without any active subsciptions will be dropped by Twitch after a certain amount of time. Additionally the reconnection logic was triggering to frequently causing unecessary load.
128 lines
3.1 KiB
C++
128 lines
3.1 KiB
C++
#pragma once
|
|
#include "message-dispatcher.hpp"
|
|
|
|
#include <obs.hpp>
|
|
#include <websocketpp/client.hpp>
|
|
#include <QObject>
|
|
#include <mutex>
|
|
#include <condition_variable>
|
|
#include <set>
|
|
|
|
#ifdef USE_TWITCH_CLI_MOCK
|
|
#include <websocketpp/config/asio_no_tls_client.hpp>
|
|
#include <websocketpp/config/asio_no_tls.hpp>
|
|
#else
|
|
#include <websocketpp/config/asio_client.hpp>
|
|
#include <websocketpp/config/asio.hpp>
|
|
#endif
|
|
|
|
namespace advss {
|
|
|
|
#ifdef USE_TWITCH_CLI_MOCK
|
|
using EventSubWSClient = websocketpp::client<websocketpp::config::asio_client>;
|
|
#else
|
|
using EventSubWSClient =
|
|
websocketpp::client<websocketpp::config::asio_tls_client>;
|
|
#endif
|
|
|
|
struct Event;
|
|
class TwitchToken;
|
|
|
|
using websocketpp::connection_hdl;
|
|
using EventSubMessageBuffer = std::shared_ptr<MessageBuffer<Event>>;
|
|
using EventSubMessageDispatcher = MessageDispatcher<Event>;
|
|
|
|
struct Event {
|
|
std::string id;
|
|
std::string type;
|
|
OBSData data;
|
|
|
|
std::string ToString() const;
|
|
};
|
|
|
|
struct Subscription {
|
|
OBSData data;
|
|
std::string id;
|
|
bool operator<(const Subscription &) const;
|
|
};
|
|
|
|
class EventSub : public QObject {
|
|
public:
|
|
explicit EventSub();
|
|
virtual ~EventSub();
|
|
|
|
void Connect();
|
|
void Disconnect();
|
|
[[nodiscard]] EventSubMessageBuffer RegisterForEvents();
|
|
bool SubscriptionIsActive(const std::string &id);
|
|
static std::string AddEventSubscription(std::shared_ptr<TwitchToken>,
|
|
Subscription);
|
|
void ClearActiveSubscriptions();
|
|
void EnableTimestampValidation(bool enable);
|
|
|
|
private:
|
|
static void SetupClient(EventSubWSClient &);
|
|
|
|
void OnOpen(connection_hdl hdl);
|
|
void OnMessage(connection_hdl hdl,
|
|
EventSubWSClient::message_ptr message);
|
|
void OnClose(connection_hdl hdl);
|
|
void OnFail(connection_hdl hdl);
|
|
|
|
void ConnectThread();
|
|
void WaitAndReconnect();
|
|
|
|
bool IsValidMessageID(const std::string &);
|
|
bool IsValidID(const std::string &);
|
|
|
|
void HandleWelcome(obs_data_t *);
|
|
void HandleKeepAlive() const;
|
|
void HandleNotification(obs_data_t *);
|
|
void HandleServerMigration(obs_data_t *);
|
|
void HandleRevocation(obs_data_t *);
|
|
|
|
void RegisterInstance();
|
|
void UnregisterInstance();
|
|
|
|
void StartServerMigrationClient(const std::string &url);
|
|
void OnServerMigrationWelcome(connection_hdl,
|
|
std::unique_ptr<EventSubWSClient> &);
|
|
|
|
struct ParsedMessage {
|
|
std::string type;
|
|
OBSDataAutoRelease payload;
|
|
};
|
|
|
|
std::optional<const ParsedMessage>
|
|
ParseWebSocketMessage(const EventSubWSClient::message_ptr &);
|
|
|
|
std::unique_ptr<EventSubWSClient> _client;
|
|
connection_hdl _connection;
|
|
|
|
std::unique_ptr<EventSubWSClient> _migrationClient;
|
|
connection_hdl _migrationConnection;
|
|
std::atomic_bool _migrating{false};
|
|
std::string _migrationSessionID;
|
|
|
|
std::thread _thread;
|
|
std::mutex _waitMtx;
|
|
std::mutex _connectMtx;
|
|
std::condition_variable _cv;
|
|
std::atomic_bool _connected{false};
|
|
std::atomic_bool _disconnect{false};
|
|
std::atomic_bool _reconnecting{false};
|
|
|
|
std::string _url;
|
|
std::string _sessionID;
|
|
bool _validateTimestamps = true;
|
|
|
|
std::deque<std::string> _messageIDs;
|
|
std::mutex _subscriptionMtx;
|
|
std::set<Subscription> _activeSubscriptions;
|
|
static std::mutex _instancesMtx;
|
|
static std::vector<EventSub *> _instances;
|
|
EventSubMessageDispatcher _dispatcher;
|
|
};
|
|
|
|
} // namespace advss
|