transport.cpp

Go to the documentation of this file.

#include <proto/main.pb.h>
#include "Udp.h"
#include "utils/logging.h"
 
#include <Arduino.h>
#include <Client.h>
#include <pb_common.h>
#include <pb_decode.h>
#include <pb_encode.h>
#include <string_view>
#include <cassert>
 
namespace transport {
 
bool client_ostream_callback(pb_ostream_t *pb_stream, const pb_byte_t *buf, size_t count) {
  auto stream = static_cast<Stream *>(pb_stream->state);
  while (true) {
    auto write_count = stream->write(buf, count);
    if (write_count == count)
      break;
 
    auto client = dynamic_cast<Client *>(stream);
    if (client && !client->connected())
      return false;
 
    buf += write_count;
    count -= write_count;
  }
  return true;
}
 
pb_MessageV1& init_v1_message(pb_Envelope& envelope, pb_size_t which_kind) {
  envelope = pb_Envelope_init_default;
  envelope.which_kind = pb_Envelope_message_v1_tag;
  auto& message = envelope.kind.message_v1 = pb_MessageV1_init_default;
  message.which_kind = which_kind;
  return message;
}
 
const pb_MessageV1& as_v1_message(const pb_Envelope& envelope, pb_size_t which_kind) {
  auto& message = envelope.kind.message_v1;
  assert(envelope.which_kind == pb_Envelope_message_v1_tag);
  assert(message.which_kind == which_kind);
  return message;
}
 
void create_error_message(pb_Envelope& envelope, std::string_view error_description, pb_ErrorCode error_code = pb_ErrorCode_None) {
  auto& msg_out = init_v1_message(envelope, pb_MessageV1_error_message_tag);
  msg_out.has_id = false;
  auto& err_msg = msg_out.kind.error_message;
  auto it = error_description.copy(err_msg.description, std::size(err_msg.description) - 1);
  err_msg.description[it] = 0;
  err_msg.code = error_code;
}
 
void create_success_message(pb_Envelope& envelope) {
  init_v1_message(envelope, pb_MessageV1_success_message_tag);
}
 
bool client_istream_callback(pb_istream_t *stream, pb_byte_t *buf, size_t count){
  auto client = static_cast<Client *>(stream->state);
  while (true) {
    auto read_count = client->readBytes(buf, count);
 
    if (count == read_count)
      return true;
 
    if (!client->connected())
      return false;
 
    buf += read_count;
    count -= read_count;
  }
}
 
 
} // namespace utils