#pragma once

#include "esphome/core/component.h"
#include "esphome/components/tuya/tuya.h"
#include "esphome/components/climate/climate.h"

namespace esphome::tuya {

class TuyaClimate : public climate::Climate, public Component {
 public:
  void setup() override;
  void loop() override;
  void dump_config() override;
  void set_supports_heat(bool supports_heat) { this->supports_heat_ = supports_heat; }
  void set_supports_cool(bool supports_cool) { this->supports_cool_ = supports_cool; }
  void set_switch_id(uint8_t switch_id) { this->switch_id_ = switch_id; }
  void set_active_state_id(uint8_t state_id) { this->active_state_id_ = state_id; }
  void set_active_state_heating_value(uint8_t value) { this->active_state_heating_value_ = value; }
  void set_active_state_cooling_value(uint8_t value) { this->active_state_cooling_value_ = value; }
  void set_active_state_drying_value(uint8_t value) { this->active_state_drying_value_ = value; }
  void set_active_state_fanonly_value(uint8_t value) { this->active_state_fanonly_value_ = value; }
  void set_heating_state_pin(GPIOPin *pin) { this->heating_state_pin_ = pin; }
  void set_cooling_state_pin(GPIOPin *pin) { this->cooling_state_pin_ = pin; }
  void set_swing_vertical_id(uint8_t swing_vertical_id) { this->swing_vertical_id_ = swing_vertical_id; }
  void set_swing_horizontal_id(uint8_t swing_horizontal_id) { this->swing_horizontal_id_ = swing_horizontal_id; }
  void set_fan_speed_id(uint8_t fan_speed_id) { this->fan_speed_id_ = fan_speed_id; }
  void set_fan_speed_low_value(uint8_t fan_speed_low_value) { this->fan_speed_low_value_ = fan_speed_low_value; }
  void set_fan_speed_medium_value(uint8_t fan_speed_medium_value) {
    this->fan_speed_medium_value_ = fan_speed_medium_value;
  }
  void set_fan_speed_middle_value(uint8_t fan_speed_middle_value) {
    this->fan_speed_middle_value_ = fan_speed_middle_value;
  }
  void set_fan_speed_high_value(uint8_t fan_speed_high_value) { this->fan_speed_high_value_ = fan_speed_high_value; }
  void set_fan_speed_auto_value(uint8_t fan_speed_auto_value) { this->fan_speed_auto_value_ = fan_speed_auto_value; }
  void set_target_temperature_id(uint8_t target_temperature_id) {
    this->target_temperature_id_ = target_temperature_id;
  }
  void set_current_temperature_id(uint8_t current_temperature_id) {
    this->current_temperature_id_ = current_temperature_id;
  }
  void set_current_temperature_multiplier(float temperature_multiplier) {
    this->current_temperature_multiplier_ = temperature_multiplier;
  }
  void set_target_temperature_multiplier(float temperature_multiplier) {
    this->target_temperature_multiplier_ = temperature_multiplier;
  }
  void set_eco_id(uint8_t eco_id) { this->eco_id_ = eco_id; }
  void set_eco_temperature(float eco_temperature) { this->eco_temperature_ = eco_temperature; }
  void set_sleep_id(uint8_t sleep_id) { this->sleep_id_ = sleep_id; }

  void set_reports_fahrenheit() { this->reports_fahrenheit_ = true; }

  void set_tuya_parent(Tuya *parent) { this->parent_ = parent; }

 protected:
  /// Override control to change settings of the climate device.
  void control(const climate::ClimateCall &call) override;

  /// Override control to change settings of swing mode.
  void control_swing_mode_(const climate::ClimateCall &call);

  /// Override control to change settings of fan mode.
  void control_fan_mode_(const climate::ClimateCall &call);

  /// Return the traits of this controller.
  climate::ClimateTraits traits() override;

  /// Re-compute the active preset of this climate controller.
  void compute_preset_();

  /// Re-compute the target temperature of this climate controller.
  void compute_target_temperature_();

  /// Re-compute the state of this climate controller.
  void compute_state_();

  /// Re-Compute the swing mode of this climate controller.
  void compute_swingmode_();

  /// Re-Compute the fan mode of this climate controller.
  void compute_fanmode_();

  /// Switch the climate device to the given climate mode.
  void switch_to_action_(climate::ClimateAction action);

  Tuya *parent_;
  bool supports_heat_;
  bool supports_cool_;
  optional<uint8_t> switch_id_{};
  optional<uint8_t> active_state_id_{};
  optional<uint8_t> active_state_heating_value_{};
  optional<uint8_t> active_state_cooling_value_{};
  optional<uint8_t> active_state_drying_value_{};
  optional<uint8_t> active_state_fanonly_value_{};
  GPIOPin *heating_state_pin_{nullptr};
  GPIOPin *cooling_state_pin_{nullptr};
  optional<uint8_t> target_temperature_id_{};
  optional<uint8_t> current_temperature_id_{};
  float current_temperature_multiplier_{1.0f};
  float target_temperature_multiplier_{1.0f};
  float hysteresis_{1.0f};
  optional<uint8_t> eco_id_{};
  optional<uint8_t> sleep_id_{};
  optional<float> eco_temperature_{};
  TuyaDatapointType eco_type_{};
  uint8_t active_state_{0};
  uint8_t fan_state_{0};
  optional<uint8_t> swing_vertical_id_{};
  optional<uint8_t> swing_horizontal_id_{};
  optional<uint8_t> fan_speed_id_{};
  optional<uint8_t> fan_speed_low_value_{};
  optional<uint8_t> fan_speed_medium_value_{};
  optional<uint8_t> fan_speed_middle_value_{};
  optional<uint8_t> fan_speed_high_value_{};
  optional<uint8_t> fan_speed_auto_value_{};
  bool swing_vertical_{false};
  bool swing_horizontal_{false};
  bool heating_state_{false};
  bool cooling_state_{false};
  float manual_temperature_{NAN};
  bool eco_{false};
  bool sleep_{false};
  bool reports_fahrenheit_{false};
};

}  // namespace esphome::tuya
