#pragma once

#include "esphome/core/component.h"
#include "esphome/core/log.h"
#include "esphome/components/i2c/i2c.h"

namespace esphome::mcp4461 {

struct WiperState {
  bool enabled = true;
  uint16_t state = 0;
  optional<float> initial_value;
  bool terminal_a = true;
  bool terminal_b = true;
  bool terminal_w = true;
  bool terminal_hw = true;
  bool wiper_lock_active = false;
  bool update_level = false;
  bool update_terminal = false;
};

// default wiper state is 128 / 0x80h
enum class Mcp4461Commands : uint8_t { WRITE = 0x00, INCREMENT = 0x04, DECREMENT = 0x08, READ = 0x0C };

enum class Mcp4461Addresses : uint8_t {
  MCP4461_VW0 = 0x00,
  MCP4461_VW1 = 0x10,
  MCP4461_VW2 = 0x60,
  MCP4461_VW3 = 0x70,
  MCP4461_STATUS = 0x50,
  MCP4461_TCON0 = 0x40,
  MCP4461_TCON1 = 0xA0,
  MCP4461_EEPROM_1 = 0xB0
};

enum class Mcp4461WiperIdx : uint8_t {
  MCP4461_WIPER_0 = 0,
  MCP4461_WIPER_1 = 1,
  MCP4461_WIPER_2 = 2,
  MCP4461_WIPER_3 = 3,
  MCP4461_WIPER_4 = 4,
  MCP4461_WIPER_5 = 5,
  MCP4461_WIPER_6 = 6,
  MCP4461_WIPER_7 = 7
};

enum class Mcp4461EepromLocation : uint8_t {
  MCP4461_EEPROM_0 = 0,
  MCP4461_EEPROM_1 = 1,
  MCP4461_EEPROM_2 = 2,
  MCP4461_EEPROM_3 = 3,
  MCP4461_EEPROM_4 = 4
};

enum class Mcp4461TerminalIdx : uint8_t { MCP4461_TERMINAL_0 = 0, MCP4461_TERMINAL_1 = 1 };

class Mcp4461Wiper;

// Mcp4461Component
class Mcp4461Component : public Component, public i2c::I2CDevice {
 public:
  Mcp4461Component(bool disable_wiper_0, bool disable_wiper_1, bool disable_wiper_2, bool disable_wiper_3)
      : wiper_0_disabled_(disable_wiper_0),
        wiper_1_disabled_(disable_wiper_1),
        wiper_2_disabled_(disable_wiper_2),
        wiper_3_disabled_(disable_wiper_3) {
    this->reg_[0].enabled = !wiper_0_disabled_;
    this->reg_[1].enabled = !wiper_1_disabled_;
    this->reg_[2].enabled = !wiper_2_disabled_;
    this->reg_[3].enabled = !wiper_3_disabled_;
  }

  void setup() override;
  void dump_config() override;
  float get_setup_priority() const override { return setup_priority::HARDWARE; }
  void loop() override;
  /// @brief get eeprom value from location
  /// @param[in] location - eeprom location 0-4
  /// @return eeprom value of specified location (9 bits max)
  uint16_t get_eeprom_value(Mcp4461EepromLocation location);
  /// @brief set eeprom value at specified location
  /// @param[in] location - eeprom location 0-4
  /// @param[in] value - 9 bits value to store
  bool set_eeprom_value(Mcp4461EepromLocation location, uint16_t value);
  /// @brief public function used to set initial value
  /// @param[in] wiper - the wiper to set the value for
  /// @param[in] initial_value - the initial value in range 0-1.0 as float
  void set_initial_value(Mcp4461WiperIdx wiper, float initial_value);
  /// @brief public function used to set disable terminal config
  /// @param[in] wiper - the wiper to set the value for
  /// @param[in] terminal - the terminal to disable, one of ['a','b','w','h']
  void initialize_terminal_disabled(Mcp4461WiperIdx wiper, char terminal);
  /// @brief read status register to log
  void read_status_register_to_log();

 protected:
  friend class Mcp4461Wiper;
  bool read_16_(uint8_t address, uint16_t *buf);
  void update_write_protection_status_();
  uint8_t get_wiper_address_(uint8_t wiper);
  uint16_t read_wiper_level_(uint8_t wiper);
  uint8_t get_status_register_();
  uint16_t get_wiper_level_(Mcp4461WiperIdx wiper);
  bool set_wiper_level_(Mcp4461WiperIdx wiper, uint16_t value);
  bool update_wiper_level_(Mcp4461WiperIdx wiper);
  void enable_wiper_(Mcp4461WiperIdx wiper);
  void disable_wiper_(Mcp4461WiperIdx wiper);
  bool increase_wiper_(Mcp4461WiperIdx wiper);
  bool decrease_wiper_(Mcp4461WiperIdx wiper);
  void enable_terminal_(Mcp4461WiperIdx wiper, char terminal);
  void disable_terminal_(Mcp4461WiperIdx, char terminal);
  bool is_writing_();
  bool is_eeprom_ready_for_writing_(bool wait_if_not_ready);
  void write_wiper_level_(uint8_t wiper, uint16_t value);
  bool mcp4461_write_(uint8_t addr, uint16_t data, bool nonvolatile = false);
  uint8_t calc_terminal_connector_byte_(Mcp4461TerminalIdx terminal_connector);
  void update_terminal_register_(Mcp4461TerminalIdx terminal_connector);
  uint8_t get_terminal_register_(Mcp4461TerminalIdx terminal_connector);
  bool set_terminal_register_(Mcp4461TerminalIdx terminal_connector, uint8_t data);

  // Converts a status to a human readable string
  static const LogString *get_message_string(int status) {
    switch (status) {
      case MCP4461_STATUS_I2C_ERROR:
        return LOG_STR("I2C error - communication with MCP4461 failed!");
      case MCP4461_STATUS_REGISTER_ERROR:
        return LOG_STR("Status register could not be read");
      case MCP4461_STATUS_REGISTER_INVALID:
        return LOG_STR("Invalid status register value - bits 1,7 or 8 are 0");
      case MCP4461_VALUE_INVALID:
        return LOG_STR("Invalid value for wiper given");
      case MCP4461_WRITE_PROTECTED:
        return LOG_STR("MCP4461 is write protected. Setting nonvolatile wipers/eeprom values is prohibited.");
      case MCP4461_WIPER_ENABLED:
        return LOG_STR("MCP4461 Wiper is already enabled, ignoring cmd to enable.");
      case MCP4461_WIPER_DISABLED:
        return LOG_STR("MCP4461 Wiper is disabled. All actions on this wiper are prohibited.");
      case MCP4461_WIPER_LOCKED:
        return LOG_STR("MCP4461 Wiper is locked using WiperLock-technology. All actions on this wiper are prohibited.");
      case MCP4461_STATUS_OK:
        return LOG_STR("Status OK");
      default:
        return LOG_STR("Unknown");
    }
  }

  enum ErrorCode {
    MCP4461_STATUS_OK = 0,               // CMD completed successfully
    MCP4461_FAILED,                      // component failed
    MCP4461_STATUS_I2C_ERROR,            // Unable to communicate with device
    MCP4461_STATUS_REGISTER_INVALID,     // Status register value was invalid
    MCP4461_STATUS_REGISTER_ERROR,       // Error fetching status register
    MCP4461_PROHIBITED_FOR_NONVOLATILE,  //
    MCP4461_VALUE_INVALID,               // Invalid value given for wiper / eeprom
    MCP4461_WRITE_PROTECTED,  // The value was read, but the CRC over the payload (valid and data) does not match
    MCP4461_WIPER_ENABLED,    // The wiper is enabled, discard additional enabling actions
    MCP4461_WIPER_DISABLED,   // The wiper is disabled - all actions for this wiper will be aborted/discarded
    MCP4461_WIPER_LOCKED,     // The wiper is locked using WiperLock-technology - all actions for this wiper will be
                              // aborted/discarded
  } error_code_{MCP4461_STATUS_OK};

  WiperState reg_[8];
  bool last_eeprom_write_timed_out_{false};
  bool write_protected_{false};
  bool wiper_0_disabled_{false};
  bool wiper_1_disabled_{false};
  bool wiper_2_disabled_{false};
  bool wiper_3_disabled_{false};
};
}  // namespace esphome::mcp4461
