ksWSServer.h

/*
 *  Copyright (c) 2021-2023, 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 <WebSocketsServer.h>
#include <functional>
#include <string_view>
 
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);

            uint64_t getRequiredAuthToken() const;

            void setRequiredAuthToken(uint64_t authToken);

            void begin();

            void loop();

            void setMessageHandler(ksWsServerMessageFunc_t func);

            virtual ~ksWSServer();
    };
}