#include "v2_notify.hpp"

WebSocketsClient webSocket;
TaskHandle_t v2NotifyTaskHandle;

void setupV2Notify()
{
    String hostname = "ws.btclock.dev";
    if ( preferences.getBool("stagingSource", DEFAULT_STAGING_SOURCE)) {
        Serial.println(F("Connecting to V2 staging source"));
        hostname = "ws-staging.btclock.dev";
    } else {
        Serial.println(F("Connecting to V2 source"));
    }

    webSocket.beginSSL(hostname, 443, "/api/v2/ws");
    webSocket.onEvent(onWebsocketV2Event);
    webSocket.setReconnectInterval(5000);
    webSocket.enableHeartbeat(15000, 3000, 2);
    
    setupV2NotifyTask();
}

void onWebsocketV2Event(WStype_t type, uint8_t * payload, size_t length) {
    switch(type) {
		case WStype_DISCONNECTED:
			Serial.printf("[WSc] Disconnected!\n");
			break;
		case WStype_CONNECTED:
        {
			Serial.printf("[WSc] Connected to url: %s\n", payload);

            JsonDocument response;

            response["type"] = "subscribe";
            response["eventType"] = "blockfee";
            size_t responseLength = measureMsgPack(response);
            uint8_t* buffer = new uint8_t[responseLength];
            serializeMsgPack(response, buffer, responseLength);
            webSocket.sendBIN(buffer, responseLength);
            delete[] buffer;

            buffer = new uint8_t[responseLength];

            response["type"] = "subscribe";
            response["eventType"] = "blockheight";
            responseLength = measureMsgPack(response);
            buffer = new uint8_t[responseLength];
            serializeMsgPack(response, buffer, responseLength);
            webSocket.sendBIN(buffer, responseLength);

            delete[] buffer;

            buffer = new uint8_t[responseLength];

            response["type"] = "subscribe";
            response["eventType"] = "price";

            JsonArray currenciesArray = response["currencies"].to<JsonArray>();

            for (const auto &str : getActiveCurrencies())
            {
                currenciesArray.add(str);
            }
  
//            response["currencies"] = currenciesArray;
            responseLength = measureMsgPack(response);
            buffer = new uint8_t[responseLength];
            serializeMsgPack(response, buffer, responseLength);
            webSocket.sendBIN(buffer, responseLength);
			break;
        }
		case WStype_TEXT:
			Serial.printf("[WSc] get text: %s\n", payload);

			// send message to server
			// webSocket.sendTXT("message here");
			break;
		case WStype_BIN:
        {
            JsonDocument doc;
            DeserializationError error = deserializeMsgPack(doc, payload, length);

            handleV2Message(doc);
			break;
        }
		case WStype_ERROR:			
		case WStype_FRAGMENT_TEXT_START:
		case WStype_FRAGMENT_BIN_START:
		case WStype_FRAGMENT:
        case WStype_PING:
        case WStype_PONG:
		case WStype_FRAGMENT_FIN:
			break;
	}
}

void handleV2Message(JsonDocument doc) {
  if (doc.containsKey("blockheight"))
  {
    uint newBlockHeight = doc["blockheight"].as<uint>();

    if (newBlockHeight == getBlockHeight()) {
      return;
    }

    processNewBlock(newBlockHeight);
  }
  else if (doc.containsKey("blockfee"))
  {
    uint medianFee = doc["blockfee"].as<uint>();

    processNewBlockFee(medianFee);
  } else if (doc.containsKey("price"))
  {

    // Iterate through the key-value pairs of the "price" object
    for (JsonPair kv : doc["price"].as<JsonObject>()) {
        const char* currency = kv.key().c_str();
        uint newPrice = kv.value().as<uint>();
        
        processNewPrice(newPrice, getCurrencyChar(currency));

    }
  } 
}

void taskV2Notify(void *pvParameters) {
    for(;;) {
        webSocket.loop();
        vTaskDelay(10 / portTICK_PERIOD_MS);
    }
}

void setupV2NotifyTask() {
  xTaskCreate(taskV2Notify, "v2Notify", (6 * 1024), NULL, tskIDLE_PRIORITY,
              &v2NotifyTaskHandle);

}

bool isV2NotifyConnected()
{
    return webSocket.isConnected();
}