#include "i2c.hpp"

#include <unistd.h>

extern "C"
{
#include <i2c/smbus.h>
#include <linux/i2c-dev.h>
#include <linux/i2c.h>
}

namespace phosphor::i2c
{

int I2C::open()
{
    int ret = 0;
    fd = ::open(busStr.c_str(), O_RDWR);
    if (fd < 0)
    {
        return fd;
    }

    ret = ioctl(fd, I2C_SLAVE_FORCE, deviceNode);
    if (ret < 0)
    {
        close();
        return ret;
    }

    return 0;
}

sdbusplus::async::task<bool> I2C::sendReceive(
    uint8_t* writeData, uint8_t writeSize, uint8_t* readData,
    uint8_t readSize) const
{
    bool result = true;

    if (fd <= 0)
    {
        result = false;
    }
    else
    {
        struct i2c_msg msg[2];
        struct i2c_rdwr_ioctl_data readWriteData;
        int msgIndex = 0;

        if (writeSize)
        {
            msg[msgIndex].addr = deviceNode;
            msg[msgIndex].flags = 0;
            msg[msgIndex].len = writeSize;
            msg[msgIndex].buf = writeData;
            msgIndex++;
        }

        if (readSize)
        {
            msg[msgIndex].addr = deviceNode;
            msg[msgIndex].flags = I2C_M_RD;
            msg[msgIndex].len = readSize;
            msg[msgIndex].buf = readData;
            msgIndex++;
        }

        readWriteData.msgs = msg;
        readWriteData.nmsgs = msgIndex;

        if (ioctl(fd, I2C_RDWR, &readWriteData) < 0)
        {
            result = false;
        }
    }
    co_return result;
}

bool I2C::sendReceive(const std::vector<uint8_t>& writeData,
                      std::vector<uint8_t>& readData) const
{
    bool result = true;

    if (fd <= 0)
    {
        return false;
    }
    else
    {
        struct i2c_msg msg[2];
        struct i2c_rdwr_ioctl_data readWriteData;
        int msgIndex = 0;

        if (!writeData.empty())
        {
            msg[msgIndex].addr = deviceNode;
            msg[msgIndex].flags = 0;
            msg[msgIndex].len = writeData.size();
            msg[msgIndex].buf = const_cast<uint8_t*>(writeData.data());
            msgIndex++;
        }

        if (!readData.empty())
        {
            msg[msgIndex].addr = deviceNode;
            msg[msgIndex].flags = I2C_M_RD;
            msg[msgIndex].len = readData.size();
            msg[msgIndex].buf = readData.data();
            msgIndex++;
        }

        readWriteData.msgs = msg;
        readWriteData.nmsgs = msgIndex;

        if (ioctl(fd, I2C_RDWR, &readWriteData) < 0)
        {
            result = false;
        }
    }

    return result;
}

int I2C::close() const
{
    return ::close(fd);
}

} // namespace phosphor::i2c