#ifdef ESP_TARGET

#include "I2c.h"
#include "Log.h"
#include "Mutex.h"

#include <esp_check.h>

namespace tt::hal::i2c {

typedef struct Data {
    Mutex mutex;
    bool isConfigured = false;
    bool isStarted = false;
    Configuration configuration;
} Data;

static Data dataArray[I2C_NUM_MAX];

#define TAG "i2c"

void printInfo(const Data& data) {
    TT_LOG_V(TAG, "I2C info for port %d", data.configuration.port);
    TT_LOG_V(TAG, "  isStarted: %d", data.isStarted);
    TT_LOG_V(TAG, "  isConfigured: %d", data.isConfigured);
    TT_LOG_V(TAG, "  initMode: %d", data.configuration.initMode);
    TT_LOG_V(TAG, "  canReinit: %d", data.configuration.canReinit);
    TT_LOG_V(TAG, "  hasMutableConfiguration: %d", data.configuration.hasMutableConfiguration);
    TT_LOG_V(TAG, "  SDA pin: %d", data.configuration.config.sda_io_num);
    TT_LOG_V(TAG, "  SCL pin: %d", data.configuration.config.scl_io_num);
}

bool init(const std::vector<i2c::Configuration>& configurations) {
   TT_LOG_I(TAG, "Init");
   for (const auto& configuration: configurations) {
       if (configuration.config.mode != I2C_MODE_MASTER) {
           TT_LOG_E(TAG, "Currently only master mode is supported");
           return false;
       }
       Data& data = dataArray[configuration.port];
       data.configuration = configuration;
       data.isConfigured = true;
   }

   for (const auto& config: configurations) {
       printInfo(dataArray[config.port]);
       if (config.initMode == InitByTactility) {
           if (!start(config.port)) {
               return false;
           }
       } else if (config.initMode == InitByExternal) {
           dataArray[config.port].isStarted = true;
       }
   }

   return true;
}

static bool configure_locked(i2c_port_t port, const i2c_config_t& configuration) {
    Data& data = dataArray[port];
    if (data.isStarted) {
        TT_LOG_E(TAG, "(%d) Cannot reconfigure while interface is started", port);
        return ESP_ERR_INVALID_STATE;
    } else if (!data.configuration.hasMutableConfiguration) {
        TT_LOG_E(TAG, "(%d) Mutation not allowed by original configuration", port);
        return ESP_ERR_NOT_ALLOWED;
    } else {
        data.configuration.config = configuration;
        return ESP_OK;
    }
}

bool configure(i2c_port_t port, const i2c_config_t& configuration) {
    lock(port);
    bool result = configure_locked(port, configuration);
    unlock(port);
    return result;
}

static bool startLocked(i2c_port_t port) {
    Data& data = dataArray[port];
    printInfo(data);
    Configuration& config = data.configuration;

    if (data.isStarted) {
        TT_LOG_E(TAG, "(%d) Starting: Already started", port);
        return false;
    }

    if (!data.isConfigured) {
        TT_LOG_E(TAG, "(%d) Starting: Not configured", port);
        return false;
    }

    esp_err_t result = i2c_param_config(port, &config.config);
    if (result != ESP_OK) {
        TT_LOG_E(TAG, "(%d) Starting: Failed to configure: %s", port, esp_err_to_name(result));
        return false;
    }

    result = i2c_driver_install(port, config.config.mode, 0, 0, 0);
    if (result != ESP_OK) {
        TT_LOG_E(TAG, "(%d) Starting: Failed to install driver: %s", port, esp_err_to_name(result));
        return false;
    } else {
        data.isStarted = true;
    }

    TT_LOG_I(TAG, "(%d) Started", port);
    return true;
}

bool start(i2c_port_t port) {
    lock(port);
    bool result = startLocked(port);
    unlock(port);
    return result;
}

static bool stopLocked(i2c_port_t port) {
    Data& data = dataArray[port];
    Configuration& config = data.configuration;

    if (!config.canReinit) {
        TT_LOG_E(TAG, "(%d) Stopping: Not allowed to re-init", port);
        return false;
    }

    if (!data.isStarted) {
        TT_LOG_E(TAG, "(%d) Stopping: Not started", port);
        return false;
    }

    esp_err_t result = i2c_driver_delete(port);
    if (result != ESP_OK) {
        TT_LOG_E(TAG, "(%d) Stopping: Failed to delete driver: %s", port, esp_err_to_name(result));
        return false;
    } else {
        data.isStarted = false;
    }

    TT_LOG_I(TAG, "(%d) Stopped", port);
    return true;
}

bool stop(i2c_port_t port) {
    lock(port);
    bool result = stopLocked(port);
    unlock(port);
    return result;
}

bool isStarted(i2c_port_t port) {
    lock(port);
    bool started = dataArray[port].isStarted;
    unlock(port);
    return started;
}

bool masterRead(i2c_port_t port, uint8_t address, uint8_t* data, size_t dataSize) {
    lock(port);
    esp_err_t result = i2c_master_read_from_device(port, address, data, dataSize, dataArray[port].configuration.timeout);
    unlock(port);
    return result == ESP_OK;
}

bool masterWrite(i2c_port_t port, uint16_t address, const uint8_t* data, uint16_t dataSize) {
    lock(port);
    esp_err_t result = i2c_master_write_to_device(port, address, data, dataSize, dataArray[port].configuration.timeout);
    unlock(port);
    return result == ESP_OK;
}

bool masterWriteRead(i2c_port_t port, uint8_t address, const uint8_t* writeData, size_t writeDataSize, uint8_t* readData, size_t readDataSize) {
    lock(port);
    esp_err_t result = i2c_master_write_read_device(port, address, writeData, writeDataSize, readData, readDataSize, dataArray[port].configuration.timeout);
    unlock(port);
    return result == ESP_OK;
}

TtStatus lock(i2c_port_t port, uint32_t timeout) {
    return dataArray[port].mutex.acquire(timeout);
}

TtStatus unlock(i2c_port_t port) {
    return dataArray[port].mutex.release();
}

} // namespace

#endif