forked from btclock/btclock_v3
144 lines
3.9 KiB
C++
144 lines
3.9 KiB
C++
#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();
|
|
}
|