improv_base.cpp

Go to the documentation of this file.

#include "improv_base.h"
 
#include "esphome/components/network/util.h"
#include "esphome/core/application.h"
#include "esphome/core/defines.h"
 
namespace esphome {
namespace improv_base {
 
#if defined(USE_ESP32_IMPROV_NEXT_URL) || defined(USE_IMPROV_SERIAL_NEXT_URL)
static constexpr const char DEVICE_NAME_PLACEHOLDER[] = "{{device_name}}";
static constexpr size_t DEVICE_NAME_PLACEHOLDER_LEN = sizeof(DEVICE_NAME_PLACEHOLDER) - 1;
static constexpr const char IP_ADDRESS_PLACEHOLDER[] = "{{ip_address}}";
static constexpr size_t IP_ADDRESS_PLACEHOLDER_LEN = sizeof(IP_ADDRESS_PLACEHOLDER) - 1;
 
static void replace_all_in_place(std::string &str, const char *placeholder, size_t placeholder_len,
                                 const std::string &replacement) {
  size_t pos = 0;
  const size_t replacement_len = replacement.length();
  while ((pos = str.find(placeholder, pos)) != std::string::npos) {
    str.replace(pos, placeholder_len, replacement);
    pos += replacement_len;
  }
}
 
std::string ImprovBase::get_formatted_next_url_() {
  if (this->next_url_.empty()) {
    return "";
  }
 
  std::string formatted_url = this->next_url_;
 
  // Replace all occurrences of {{device_name}}
  replace_all_in_place(formatted_url, DEVICE_NAME_PLACEHOLDER, DEVICE_NAME_PLACEHOLDER_LEN, App.get_name());
 
  // Replace all occurrences of {{ip_address}}
  for (auto &ip : network::get_ip_addresses()) {
    if (ip.is_ip4()) {
      replace_all_in_place(formatted_url, IP_ADDRESS_PLACEHOLDER, IP_ADDRESS_PLACEHOLDER_LEN, ip.str());
      break;
    }
  }
 
  // Note: {{esphome_version}} is replaced at code generation time in Python
 
  return formatted_url;
}
#endif
 
}  // namespace improv_base
}  // namespace esphome