ksWSServer.h

/*
 *  Copyright (c) 2021-2025, Krzysztof Strehlau
 *
 *  This file is a part of the ksIotFramework library.
 *  All licensing information can be found inside LICENSE.md file.
 *
 *  https://github.com/cziter15/ksIotFrameworkLib/blob/master/LICENSE
 */
 
#pragma once
 
#include <memory>
#include <inttypes.h>
#include <functional>
#include <string_view>
#include <WebSocketsServer.h>
 
namespace ksf::misc
{
    typedef std::function<void(uint8_t client, std::string_view message)> ksWsServerMessageFunc_t;
 
    class ksWSServer : public WebSocketsServerCore 
    {
        protected:
            std::unique_ptr<WEBSOCKETS_NETWORK_SERVER_CLASS> wsListener;        
            uint64_t requriedAuthToken{0};                                      
            ksWsServerMessageFunc_t onWebsocketTextMessage;                     
 
            void handleNonWebsocketConnection(WSclient_t* client) override;
 
        public:
            ksWSServer(uint16_t port);
 
            virtual ~ksWSServer();
 
            uint64_t getRequiredAuthToken() const;
 
            void setRequiredAuthToken(uint64_t authToken);
 
            void begin();
 
            void loop();
 
            void setMessageHandler(ksWsServerMessageFunc_t func);
    };
}