#include "sen0321.h"
#include "esphome/core/log.h"
#include "esphome/core/hal.h"

namespace esphome {
namespace sen0321_sensor {

static const char *const TAG = "sen0321_sensor.sensor";

void Sen0321Sensor::setup() {
  if (!this->write_byte(SENSOR_MODE_REGISTER, SENSOR_MODE_AUTO)) {
    ESP_LOGW(TAG, "Error setting measurement mode.");
    this->mark_failed();
  };
}

void Sen0321Sensor::update() { this->read_data_(); }

void Sen0321Sensor::dump_config() {
  ESP_LOGCONFIG(TAG, "DF Robot Ozone Sensor sen0321:");
  LOG_I2C_DEVICE(this);
  if (this->is_failed()) {
    ESP_LOGE(TAG, ESP_LOG_MSG_COMM_FAIL);
  }
  LOG_UPDATE_INTERVAL(this);
}

void Sen0321Sensor::read_data_() {
  uint8_t result[2];
  this->read_bytes(SENSOR_AUTO_READ_REG, result, (uint8_t) 2);
  this->publish_state(((uint16_t) (result[0] << 8) + result[1]));
}

}  // namespace sen0321_sensor
}  // namespace esphome
