WebSocketOutput.h

Go to the documentation of this file.

#pragma once
#include "AudioTools/CoreAudio/AudioOutput.h"
#include "WebSocketsClient.h"  // https://github.com/Links2004/arduinoWebSockets
#include "WebSocketsServer.h"  // https://github.com/Links2004/arduinoWebSockets
 
namespace audio_tools {
 
class WebSocketOutput : public AudioOutput {
 public:
  WebSocketOutput() = default;
  WebSocketOutput(WebSocketsClient &ws) { setWebSocket(ws); }
  WebSocketOutput(WebSocketsServer &ws) { setWebSocket(ws); }
 
  void setWebSocket(WebSocketsClient &ws) { p_ws = &ws; };
  void setWebSocket(WebSocketsServer &ws) { p_ws_server = &ws; };
 
  size_t write(const uint8_t *data, size_t len) override {
    bool rc = false;
    if (p_ws != nullptr) rc = p_ws->sendBIN(data, len);
    if (p_ws_server != nullptr) {
      if (clientNo >= 0) {
        rc = p_ws_server->sendBIN(clientNo, data, len);
      } else {
        // broadcast to all clients
        rc = p_ws_server->broadcastBIN(data, len);
      }
      rc = p_ws_server->broadcastBIN(data, len);
    }
    return rc ? len : 0;
  }
 
  void setTargetNo(int clientNo) { this->clientNo = clientNo; }
 
 protected:
  WebSocketsClient *p_ws = nullptr;
  WebSocketsServer *p_ws_server = nullptr;
  int clientNo = -1;
};
 
}  // namespace audio_tools