// matth-x/ESP8266-OCPP
// Copyright Matthias Akstaller 2019 - 2021
// MIT License

#include <ArduinoOcpp/Core/OcppSocket.h>
#include <Variants.h>

using namespace ArduinoOcpp;
using namespace ArduinoOcpp::EspWiFi;

OcppSocket::OcppSocket() {
    
}

OcppClientSocket::OcppClientSocket(WebSocketsClient *wsock) : wsock(wsock) {

}

bool OcppClientSocket::sendTXT(String &out) {
    return wsock->sendTXT(out);
}

void OcppClientSocket::setReceiveTXTcallback(ReceiveTXTcallback &callback) {
    wsock->onEvent([callback](WStype_t type, uint8_t * payload, size_t length) {
        switch (type) {
            case WStype_DISCONNECTED:
                Serial.print(F("[OcppClientSocket] Disconnected!\n"));
                break;
            case WStype_CONNECTED:
                Serial.printf("[OcppClientSocket] Connected to url: %s\n", payload);
                break;
            case WStype_TEXT:
                if (DEBUG_OUT || TRAFFIC_OUT) Serial.printf("[OcppClientSocket] get text: %s\n", payload);

                if (!callback((const char *) payload, length)) { //forward message to OcppEngine
                    Serial.print(F("[OcppClientSocket] Processing WebSocket input event failed!\n"));
                }
                break;
            case WStype_BIN:
                Serial.print(F("[OcppClientSocket] Incoming binary data stream not supported"));
                break;
            case WStype_PING:
                // pong will be send automatically
                Serial.print(F("[OcppClientSocket] get ping\n"));
                break;
            case WStype_PONG:
                // answer to a ping we send
                Serial.print(F("[OcppClientSocket] get pong\n"));
                break;
            case WStype_FRAGMENT_TEXT_START: //fragments are not supported
            default:
                Serial.print(F("[OcppClientSocket] Unsupported WebSocket event type\n"));
                break;
        }
    });
}

OcppServerSocket::OcppServerSocket(IPAddress &ip_addr) : ip_addr(ip_addr) {
    
}

OcppServerSocket::~OcppServerSocket() {
    OcppServer::getInstance()->removeReceiveTXTcallback(this->ip_addr);
}

bool OcppServerSocket::sendTXT(String &out) {
    return OcppServer::getInstance()->sendTXT(ip_addr, out);
}

void OcppServerSocket::setReceiveTXTcallback(ReceiveTXTcallback &callback) {
    OcppServer::getInstance()->setReceiveTXTcallback(ip_addr, callback);
}