#pragma once

#include "esphome/components/i2c/i2c.h"  // Ensure this header defines i2c::I2CDevice
#include "esphome/core/component.h"     // Ensure this header defines Component
#include "esphome/components/sensor/sensor.h"

#include <vector>
#include <esphome/core/gpio.h>
#include <freertos/FreeRTOS.h>
#include <freertos/semphr.h>

namespace esphome {
namespace ads1115_int {

enum ADS1115Multiplexer {
  ADS1115_MULTIPLEXER_P0_N1 = 0b000,
  ADS1115_MULTIPLEXER_P0_N3 = 0b001,
  ADS1115_MULTIPLEXER_P1_N3 = 0b010,
  ADS1115_MULTIPLEXER_P2_N3 = 0b011,
  ADS1115_MULTIPLEXER_P0_NG = 0b100,
  ADS1115_MULTIPLEXER_P1_NG = 0b101,
  ADS1115_MULTIPLEXER_P2_NG = 0b110,
  ADS1115_MULTIPLEXER_P3_NG = 0b111,
  ADS1115_MULTIPLEXER_INVALID = 0xFF
};

enum ADS1115Gain {
  ADS1115_GAIN_6P144 = 0b000,
  ADS1115_GAIN_4P096 = 0b001,
  ADS1115_GAIN_2P048 = 0b010,
  ADS1115_GAIN_1P024 = 0b011,
  ADS1115_GAIN_0P512 = 0b100,
  ADS1115_GAIN_0P256 = 0b101,
};

enum ADS1115Resolution {
  ADS1115_16_BITS = 16,
  ADS1015_12_BITS = 12,
};

enum ADS1115Samplerate {
  ADS1115_8SPS = 0b000,
  ADS1115_16SPS = 0b001,
  ADS1115_32SPS = 0b010,
  ADS1115_64SPS = 0b011,
  ADS1115_128SPS = 0b100,
  ADS1115_250SPS = 0b101,
  ADS1115_475SPS = 0b110,
  ADS1115_860SPS = 0b111
};

// we will force single-shot mode for interrupt operation
class ADS1115Component : public Component, public i2c::I2CDevice {
  private:
    struct mux_data_item {
      ADS1115Multiplexer multiplexer;
      ADS1115Gain gain;
      ADS1115Resolution resolution;
      ADS1115Samplerate samplerate;
      sensor::Sensor *sensor_ptr;
      mux_data_item() { multiplexer = ADS1115_MULTIPLEXER_INVALID; }
    }; 
    int mux_data_index_{0};
    mux_data_item muxdata_[4];
    int next_mux_data_index(int start_index);
    void read_request_next();    
    int get_mux_index(ADS1115Multiplexer multiplexer);

  public:
    // Semaphore handle
    SemaphoreHandle_t data_ready_sem = NULL;  
    void setup() override;
    void loop() override;
    void dump_config() override;
    double read_data(int16_t& raw_value, int mux_index);
    bool start_single_shot_conversion(int mux_index);
    void set_alert_pin(InternalGPIOPin *pin) { alert_pin_ = pin; }
    bool set_data_ready_mode();
    int set_mux_data(ADS1115Multiplexer multiplexer, ADS1115Gain gain, ADS1115Resolution resolution, ADS1115Samplerate samplerate, sensor::Sensor *sensor_ptr);

 protected:
    uint16_t prev_config_{0};
    InternalGPIOPin *alert_pin_;
    static void IRAM_ATTR isr_handler(ADS1115Component *arg);
};

}  // namespace ads1115_int
}  // namespace esphome