api_connection_buffer.h

Go to the documentation of this file.

#pragma once
 
#include "esphome/core/defines.h"
#ifdef USE_API
 
// Inline APIConnection methods that need APIServer complete. Include this
// instead of api_connection.h when calling encode_to_buffer or get_batch_delay_ms_.
 
#include "api_connection.h"
#include "api_server.h"
 
namespace esphome::api {
 
inline uint16_t ESPHOME_ALWAYS_INLINE APIConnection::encode_to_buffer(uint32_t calculated_size,
                                                                      MessageEncodeFn encode_fn, const void *msg,
                                                                      APIConnection *conn, uint32_t remaining_size) {
#ifdef HAS_PROTO_MESSAGE_DUMP
  if (conn->flags_.log_only_mode) {
    auto *proto_msg = static_cast<const ProtoMessage *>(msg);
    DumpBuffer dump_buf;
    conn->log_send_message_(proto_msg->message_name(), proto_msg->dump_to(dump_buf));
    return 1;
  }
#endif
  const uint8_t footer_size = conn->helper_->frame_footer_size();
 
  // First message uses max padding (already in buffer), subsequent use exact header size
  size_t to_add;
  if (conn->flags_.batch_first_message) {
    conn->flags_.batch_first_message = false;
    conn->batch_header_size_ = conn->helper_->frame_header_padding();
    to_add = calculated_size;
  } else {
    conn->batch_header_size_ = conn->helper_->frame_header_size(calculated_size, conn->batch_message_type_);
    to_add = calculated_size + conn->batch_header_size_ + footer_size;
  }
 
  // Check if it fits (using actual header size, not max padding)
  uint16_t total_calculated_size = calculated_size + conn->batch_header_size_ + footer_size;
  if (total_calculated_size > remaining_size)
    return 0;
 
  auto &shared_buf = conn->parent_->get_shared_buffer_ref();
  shared_buf.resize(shared_buf.size() + to_add);
  ProtoWriteBuffer buffer{&shared_buf, shared_buf.size() - calculated_size};
  encode_fn(msg, buffer PROTO_ENCODE_DEBUG_INIT(&shared_buf));
 
  return total_calculated_size;
}
 
inline uint32_t APIConnection::get_batch_delay_ms_() const { return this->parent_->get_batch_delay(); }
 
}  // namespace esphome::api
#endif