// matth-x/ArduinoOcpp
// Copyright Matthias Akstaller 2019 - 2022
// MIT License

#ifndef OCPPSOCKET_H
#define OCPPSOCKET_H

#include <Arduino.h>
#include <functional>

namespace ArduinoOcpp {

using ReceiveTXTcallback = std::function<bool(const char*, size_t)>;

class OcppSocket {
public:
    OcppSocket() = default;
    virtual ~OcppSocket() = default;

    virtual void loop() = 0;

    virtual bool sendTXT(String &out) = 0;

    virtual void setReceiveTXTcallback(ReceiveTXTcallback &receiveTXT) = 0; //ReceiveTXTcallback is defined in OcppServer.h
};

} //end namespace ArduinoOcpp

#ifndef AO_CUSTOM_WS

#include <WebSocketsClient.h>
#include <WebSocketsServer.h>

namespace ArduinoOcpp {
namespace EspWiFi {

class OcppClientSocket : public OcppSocket {
private:
    //std::shared_ptr<WebSocketsClient> wsock;
    WebSocketsClient *wsock;
public:
    //OcppClientSocket(ReceiveTXTcallback &receiveTXT, std::shared_ptr<WebSocketsClient> wsock);
    OcppClientSocket(WebSocketsClient *wsock);

    void loop();

    bool sendTXT(String &out);

    void setReceiveTXTcallback(ReceiveTXTcallback &receiveTXT);
};

class OcppServerSocket : public OcppSocket {
private:
    IPAddress ip_addr;
public:
    OcppServerSocket(IPAddress &ip_addr);
    ~OcppServerSocket();

    void loop();

    bool sendTXT(String &out);

    void setReceiveTXTcallback(ReceiveTXTcallback &receiveTXT);
};

} //end namespace EspWiFi
} //end namespace ArduinoOcpp
#endif //ndef AO_CUSTOM_WS
#endif