#pragma once
#include <FreeRTOS.h>
#include <semphr.h>
#include <drivers/include/nrfx_twi.h> // NRF_TWIM_Type
#include <cstdint>

namespace Pinetime {
  namespace Drivers {
    class TwiMaster {
      public:
        enum class Modules { TWIM1 };
        enum class Frequencies {Khz100, Khz250, Khz400};
        enum class ErrorCodes {NoError, TransactionFailed};
        struct Parameters {
          uint32_t frequency;
          uint8_t pinSda;
          uint8_t pinScl;
        };

        TwiMaster(const Modules module, const Parameters& params);

        void Init();
        ErrorCodes Read(uint8_t deviceAddress, uint8_t registerAddress, uint8_t* buffer, size_t size);
        ErrorCodes Write(uint8_t deviceAddress, uint8_t registerAddress, const uint8_t* data, size_t size);

        void Sleep();
        void Wakeup();

      private:

        ErrorCodes Read(uint8_t deviceAddress, uint8_t* buffer, size_t size, bool stop);
        ErrorCodes Write(uint8_t deviceAddress, const uint8_t* data, size_t size, bool stop);
        void FixHwFreezed();
        NRF_TWIM_Type* twiBaseAddress;
        SemaphoreHandle_t mutex;
        const Modules module;
        const Parameters params;
        static constexpr uint8_t maxDataSize{16};
        static constexpr uint8_t registerSize{1};
        uint8_t internalBuffer[maxDataSize + registerSize];
        uint32_t txStartedCycleCount = 0;
        static constexpr uint32_t HwFreezedDelay{161000};
    };
  }
}