tetrees/Plugins/Wwise/ThirdParty/include/AK/WwiseAuthoringAPI/AkAutobahn/WebSocketClient.h

/*******************************************************************************
The content of this file includes portions of the AUDIOKINETIC Wwise Technology
released in source code form as part of the SDK installer package.

Commercial License Usage

Licensees holding valid commercial licenses to the AUDIOKINETIC Wwise Technology
may use this file in accordance with the end user license agreement provided
with the software or, alternatively, in accordance with the terms contained in a
written agreement between you and Audiokinetic Inc.

  Copyright (c) 2023 Audiokinetic Inc.
*******************************************************************************/

#pragma once

#include <AK/Tools/Common/AkAssert.h>

#include <string>

extern "C" {
    struct mg_connection;
}

namespace AK
{
	namespace WwiseAuthoringAPI
	{
		class IWebSocketClientHandler;

		class WebSocketClient final
		{
		public:

			WebSocketClient(const WebSocketClient&) = delete;
			WebSocketClient(WebSocketClient&&) = delete;

			WebSocketClient(IWebSocketClientHandler* handler)
				: m_handler(handler)
			{
				AKASSERT(handler != nullptr);
			}

			~WebSocketClient()
			{
				// Must wait for the connection close to be handled before being able to delete the object, otherwise
				// we could have a dangling callback in the lib that would call us from another thread
				// on an already deleted object.
			}

			bool Connect(const char* host, const int port);
			void Close();

			bool SendUTF8(const std::string& message, std::string& out_errorMessage);

			bool IsConnected()
			{
				return (m_connection != nullptr);
			}

		private:

			static bool m_isNetworkInitialized;

			static inline bool EnsureNetworkInit()
			{
				if (!m_isNetworkInitialized)
				{
					m_isNetworkInitialized = InitializeNetwork();
				}

				return m_isNetworkInitialized;
			}

			static bool InitializeNetwork();

			static int OnMessage(
				mg_connection *conn,
				int flags,
				char* data,
				size_t data_len,
				void* user_data);

			static void OnClose(
				const mg_connection *conn,
				void *user_data);

			mg_connection* m_connection = nullptr;
			IWebSocketClientHandler* m_handler;
		};
	}
}