#pragma once

#include <chrono>
#include "haier_base.h"

namespace esphome {
namespace haier {

class Smartair2Climate : public HaierClimateBase {
 public:
  Smartair2Climate();
  Smartair2Climate(const Smartair2Climate &) = delete;
  Smartair2Climate &operator=(const Smartair2Climate &) = delete;
  ~Smartair2Climate();
  void dump_config() override;

 protected:
  void set_answers_handlers() override;
  void process_phase(std::chrono::steady_clock::time_point now) override;
  haier_protocol::HaierMessage get_control_message() override;
  bool is_message_invalid(uint8_t message_type) override;
  // Answers handlers
  haier_protocol::HandlerError status_handler_(uint8_t request_type, uint8_t message_type, const uint8_t *data,
                                               size_t data_size);
  // Helper functions
  haier_protocol::HandlerError process_status_message_(const uint8_t *packet, uint8_t size);
  std::unique_ptr<uint8_t[]> last_status_message_;
};

}  // namespace haier
}  // namespace esphome