/*
    Original copyright 2018 Benjamin Vedder benjamin@vedder.se and the VESC Tool project ( https://github.com/vedderb/vesc_tool )

    Forked to:
    Copyright 2018 Danny Bokma github@diebie.nl (https://github.com/DieBieEngineering/DieBieMS-Tool)

    Now forked to:
    Copyright 2019 - 2020 Kevin Dionne kevin.dionne@ennoid.me (https://github.com/EnnoidMe/ENNOID-BMS-Tool)

    This file is part of ENNOID-BMS Tool.

    ENNOID-BMS Tool is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    ENNOID-BMS Tool is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef BMSInterface_H
#define BMSInterface_H

#include <QObject>
#include <QTimer>
#include <QByteArray>
#include <QList>
#include <QTcpSocket>
#include <QSettings>
#include <QHash>

#ifdef HAS_SERIALPORT
#include <QSerialPort>
#endif

#include "datatypes.h"
#include "configparams.h"
#include "commands.h"
#include "packet.h"

#ifdef HAS_BLUETOOTH
#include "bleuart.h"
#endif

class BMSInterface : public QObject
{
    Q_OBJECT
public:
    explicit BMSInterface(QObject *parent = 0);
    ~BMSInterface();
    Q_INVOKABLE Commands *commands() const;
    Q_INVOKABLE ConfigParams *bmsConfig();
    Q_INVOKABLE ConfigParams *infoConfig();
    Q_INVOKABLE QStringList getSupportedFirmwares();
    QList<QPair<int, int> > getSupportedFirmwarePairs();
    Q_INVOKABLE QString getFirmwareNow();
    Q_INVOKABLE void emitStatusMessage(const QString &msg, bool isGood);
    Q_INVOKABLE void emitMessageDialog(const QString &title, const QString &msg, bool isGood, bool richText = false);
    Q_INVOKABLE bool fwRx();
    Q_INVOKABLE void storeSettings();
    //Q_INVOKABLE QVariantList getProfiles();
    //Q_INVOKABLE void addProfile(QVariant profile);
    //Q_INVOKABLE void clearProfiles();
    //Q_INVOKABLE void deleteProfile(int index);
    //Q_INVOKABLE void moveProfileUp(int index);
    //Q_INVOKABLE void moveProfileDown(int index);
    //Q_INVOKABLE MCCONF_TEMP getProfile(int index);
    //Q_INVOKABLE void updateProfile(int index, QVariant profile);
    //Q_INVOKABLE bool isProfileInUse(int index);
    //Q_INVOKABLE MCCONF_TEMP createMcconfTemp();
    //Q_INVOKABLE void updateMcconfFromProfile(MCCONF_TEMP profile);
//    Q_INVOKABLE QStringList getPairedUuids();
//    Q_INVOKABLE bool addPairedUuid(QString uuid);
//    Q_INVOKABLE bool deletePairedUuid(QString uuid);
//    Q_INVOKABLE void clearPairedUuids();
//    Q_INVOKABLE bool hasPairedUuid(QString uuid);
 //   Q_INVOKABLE QString getConnectedUuid();
 //   Q_INVOKABLE bool isIntroDone();
 //   Q_INVOKABLE void setIntroDone(bool done);
  //  Q_INVOKABLE QString getLastTcpServer() const;
  //  Q_INVOKABLE int getLastTcpPort() const;
//#ifdef HAS_SERIALPORT
//    Q_INVOKABLE QString getLastSerialPort() const;
//    Q_INVOKABLE int getLastSerialBaud() const;
//#endif
    bool swdEraseFlash();
    bool swdUploadFw(QByteArray newFirmware, uint32_t startAddr = 0);
    void swdCancel();
    bool swdReboot();

#ifdef HAS_BLUETOOTH
    Q_INVOKABLE BleUart* bleDevice();
    Q_INVOKABLE void storeBleName(QString address, QString name);
    Q_INVOKABLE QString getBleName(QString address);
    Q_INVOKABLE QString getLastBleAddr() const;
#endif

    // Connection
    Q_INVOKABLE bool isPortConnected();
    Q_INVOKABLE void disconnectPort();
    Q_INVOKABLE bool reconnectLastPort();
    Q_INVOKABLE bool autoconnect();
    Q_INVOKABLE QString getConnectedPortName();
    bool connectSerial(QString port, int baudrate = 115200);
    QList<VSerialInfo_t> listSerialPorts();
    Q_INVOKABLE void connectTcp(QString server, int port);
    Q_INVOKABLE void connectBle(QString address);
    Q_INVOKABLE bool isAutoconnectOngoing() const;
    Q_INVOKABLE double getAutoconnectProgress() const;
    Q_INVOKABLE QVector<int> scanCan();
    Q_INVOKABLE QVector<int> getCanDevsLast() const;
    Q_INVOKABLE void ignoreCanChange(bool ignore);

signals:
    void statusMessage(const QString &msg, bool isGood);
    void messageDialog(const QString &title, const QString &msg, bool isGood, bool richText);
    void fwUploadStatus(const QString &status, double progress, bool isOngoing);
    void serialPortNotWritable(const QString &port);
    void fwRxChanged(bool rx, bool limited);
    void portConnectedChanged();
    void autoConnectProgressUpdated(double progress, bool isOngoing);
    void autoConnectFinished();
    void pairingListUpdated();

public slots:

private slots:
#ifdef HAS_SERIALPORT
    void serialDataAvailable();
    void serialPortError(QSerialPort::SerialPortError error);
#endif

    void tcpInputConnected();
    void tcpInputDisconnected();
    void tcpInputDataAvailable();
    void tcpInputError(QAbstractSocket::SocketError socketError);

    void bleDataRx(QByteArray data);

    void timerSlot();
    void packetDataToSend(QByteArray &data);
    void packetReceived(QByteArray &data);
    void cmdDataToSend(QByteArray &data);
    void fwVersionReceived(int major, int minor, QString hw, QByteArray uuid);
    void bmsconfUpdated();
    void bmsconfStored();
    void ackReceived(QString ackType);

private:
    typedef enum {
        CONN_NONE = 0,
        CONN_SERIAL,
        CONN_TCP,
        CONN_BLE
    } conn_t;

    QSettings mSettings;
    QHash<QString, QString> mBleNames;
    QVariantList mProfiles;
    QStringList mPairedUuids;

    ConfigParams *mbmsConfig;
    ConfigParams *mInfoConfig;

    QTimer *mTimer;
    Packet *mPacket;
    Commands *mCommands;
    bool mFwVersionReceived;
    int mFwRetries;
    int mFwPollCnt;
    QString mFwTxt;
    QString mHwTxt;
    bool mIsUploadingFw;

    bool mCancelSwdUpload;

    // Connections
    conn_t mLastConnType;

#ifdef HAS_SERIALPORT
    QSerialPort *mSerialPort;
    QString mLastSerialPort;
    int mLastSerialBaud;
#endif

    QTcpSocket *mTcpSocket;
    bool mTcpConnected;
    QString mLastTcpServer;
    int mLastTcpPort;

#ifdef HAS_BLUETOOTH
    BleUart *mBleUart;
    QString mLastBleAddr;
#endif
    bool mSendCanBefore = false;
    int mCanIdBefore = 0;
    bool mWasConnected;
    bool mAutoconnectOngoing;
    double mAutoconnectProgress;
    bool mIgnoreCanChange;

    QVector<int> mCanDevsLast;

    void updateFwRx(bool fwRx);
    void setLastConnectionType(conn_t type);

};

#endif // BMSInterface_H