/*
    ModbusRTU Library for ESP8266/ESP32
    Copyright (C) 2019-2020 Alexander Emelianov (a.m.emelianov@gmail.com)
	https://github.com/emelianov/modbus-esp8266
	This code is licensed under the BSD New License. See LICENSE.txt for more info.
*/
#pragma once

#include "mbed.h"
#include <Modbus.h>

#if defined(ESP8266)
#include <SoftwareSerial.h>
#endif

//#define MODBUSRTU_DEBUG
#define MODBUSRTU_BROADCAST 0
#define MB_RESERVE 248
#define MB_SERIAL_BUFFER 128
#define MB_MAX_TIME 10
//#define MODBUSRTU_TIMEOUT 1s
#define MODBUSRTU_ADD_REG
//#define MB_STATIC_FRAME 1

class ModbusRTU : public Modbus {
protected:
    CircularBuffer<char, 200> RX_RingBuff;
    UnbufferedSerial *_port;
    DigitalOut *_txPin;

    void SerialRXHandler();

    void SerialTXHandler();

    uint8_t isDataAvailable();

    uint8_t Read();

    //    unsigned int
    Kernel::Clock::duration MODBUSRTU_TIMEOUT = 1s;
    Kernel::Clock::duration _t; // inter-frame delay in mS
    Kernel::Clock::time_point t; // time sience last data byte arrived
    bool isMaster = false;
    uint8_t _slaveId;
    Kernel::Clock::time_point _timestamp;
    cbTransaction _cb = nullptr;
    void *_data = nullptr;
    uint8_t *_sentFrame = nullptr;
    TAddress _sentReg = COIL(0);
    uint16_t maxRegs = 0x007D;
#ifdef ESP32
    portMUX_TYPE mux = portMUX_INITIALIZER_UNLOCKED;
#endif

    bool send(uint8_t slaveId, TAddress startreg, cbTransaction cb, void *data = nullptr, bool waitResponse = true);

    // Prepare and send ModbusRTU frame. _frame buffer and _len should be filled with Modbus data
    // slaveId - slave id
    // startreg - first local register to save returned data to (miningless for write to slave operations)
    // cb - transaction callback function
    // data - if not null use buffer to save returned data instead of local registers
    bool rawSend(uint8_t slaveId, uint8_t *frame, uint8_t len);

    bool cleanup(); // Free clients if not connected and remove timedout transactions and transaction with forced events
    static uint16_t crc16(uint8_t address, uint8_t *frame, uint8_t pdulen);

public:
    void setBaudrate(uint32_t baud = -1);

#if defined(ESP8266)
    bool begin(BufferedSerial* port, int16_t txPin = -1);
#endif

    bool begin(UnbufferedSerial *port, PinName txPin = NC);

    bool begin(UnbufferedSerial *port);

    void task();

    void master() { isMaster = true; };

    void slave(uint8_t slaveId) { _slaveId = slaveId; };

    uint8_t slave() const { return _slaveId; }

    uint32_t eventSource() override { return _slaveId; }

    uint16_t writeHreg(uint8_t slaveId, uint16_t offset, uint16_t value, cbTransaction cb = nullptr);

    uint16_t writeCoil(uint8_t slaveId, uint16_t offset, bool value, cbTransaction cb = nullptr);

    uint16_t readCoil(uint8_t slaveId, uint16_t offset, bool *value, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t writeCoil(uint8_t slaveId, uint16_t offset, bool *value, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t
    writeHreg(uint8_t slaveId, uint16_t offset, uint16_t *value, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t readIsts(uint8_t slaveId, uint16_t offset, bool *value, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t
    readHreg(uint8_t slaveId, uint16_t offset, uint16_t *value, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t
    readIreg(uint8_t slaveId, uint16_t offset, uint16_t *value, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t pushCoil(uint8_t slaveId, uint16_t to, uint16_t from, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t pullCoil(uint8_t slaveId, uint16_t from, uint16_t to, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t pullIsts(uint8_t slaveId, uint16_t from, uint16_t to, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t pushHreg(uint8_t slaveId, uint16_t to, uint16_t from, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t pullHreg(uint8_t slaveId, uint16_t from, uint16_t to, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t pullIreg(uint8_t slaveId, uint16_t from, uint16_t to, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t pullHregToIreg(uint8_t slaveId, uint16_t offset, uint16_t startreg, uint16_t numregs = 1,
                            cbTransaction cb = nullptr);

    uint16_t pullCoilToIsts(uint8_t slaveId, uint16_t offset, uint16_t startreg, uint16_t numregs = 1,
                            cbTransaction cb = nullptr);

    uint16_t
    pushIstsToCoil(uint8_t slaveId, uint16_t to, uint16_t from, uint16_t numregs = 1, cbTransaction cb = nullptr);

    uint16_t
    pushIregToHreg(uint8_t slaveId, uint16_t to, uint16_t from, uint16_t numregs = 1, cbTransaction cb = nullptr);
};