// SPDX-License-Identifier: LGPL-3.0-or-later

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
*
* nymea-zigbee
* Zigbee integration module for nymea
*
* Copyright (C) 2013 - 2024, nymea GmbH
* Copyright (C) 2024 - 2025, chargebyte austria GmbH
*
* This file is part of nymea-zigbee.
*
* nymea-zigbee is free software: you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* nymea-zigbee 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with nymea-zigbee. If not, see <https://www.gnu.org/licenses/>.
*
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#ifndef ZIGBEECLUSTEROTA_H
#define ZIGBEECLUSTEROTA_H

#include <QObject>
#include <QDateTime>

#include "zcl/zigbeecluster.h"
#include "zcl/zigbeeclusterreply.h"

class ZigbeeNode;
class ZigbeeNetwork;
class ZigbeeNodeEndpoint;
class ZigbeeNetworkReply;

class ZigbeeClusterOta : public ZigbeeCluster
{
    Q_OBJECT

    friend class ZigbeeNode;
    friend class ZigbeeNetwork;

public:
    enum Attribute {
        AttributeUpgradeServerId = 0x0000,
        AttributeFileOffset = 0x0001,
        AttributeCurrentFileVersion = 0x0002,
        AttributeCurrentZigbeeStackVersion = 0x0003,
        AttributeDownloadedFileVersion = 0x0004,
        AttributeDownloadedZigbeeStackVersion = 0x0005,
        AttributeImageUpgradeStatus = 0x0006,
        AttributeManufacturerId = 0x0007,
        AttributeImageTypeId = 0x0008,
        AttributeMinimumBockPeriod = 0x0009,
        AttributeImageStamp = 0x000a
    };
    Q_ENUM(Attribute)

    enum Command {
        CommandImageNotify = 0x00,
        CommandQueryNextImageRequest = 0x01,
        CommandQueryNextImageResponse = 0x02,
        CommandImageBlockRequest = 0x03,
        CommandImagePageRequest = 0x04,
        CommandImageBlockResponse = 0x05,
        CommandUpgradeEndRequest = 0x06,
        CommandUpgradeEndResponse = 0x07,
        CommandQueryDeviceSpecificFileRequest = 0x08,
        CommandQueryDeviceSpecificFileResponse = 0x09
    };
    Q_ENUM(Command)

    enum StatusCode {
        StatusCodeSuccess = 0x00,
        StatusCodeAbort = 0x95,
        StatusCodeNotAuthorized = 0x7E,
        StatusCodeInvalidImage = 0x96,
        StatusCodeWaitForData = 0x97,
        StatusCodeNoImageAvailable = 0x98,
        StatusCodeMalformedCommand = 0x80,
        StatusCodeUnsupportedClusterCommand = 0x81,
        StatusCodeRequireMoreImage = 0x99
    };
    Q_ENUM(StatusCode)

    enum PayloadType {
        PayloadTypeQueryJitter = 0x00,
        PayloadTypeQueryJitterAndManufacturerCode = 0x01,
        PayloadTypeQueryJitterAndManufacturerCodeAndImageType = 0x02,
        PayloadTypeQueryJitterAndManufacturerCodeAndImageTypeAndNewFileVersion = 0x03
    };
    Q_ENUM(PayloadType)

    typedef struct FileVersion {
        quint8 applicationRelease;
        quint8 applicationBuild;
        quint8 stackRelease;
        quint8 stackBuild;
    } FileVersion;


    explicit ZigbeeClusterOta(ZigbeeNetwork *network, ZigbeeNode *node, ZigbeeNodeEndpoint *endpoint, Direction direction, QObject *parent = nullptr);

    ZigbeeClusterReply *sendImageNotify(PayloadType payloadType = PayloadTypeQueryJitter, quint8 queryJitter = 100, quint16 manufacturerCode = 0, quint16 imageType = 0, quint32 newFileVersion = 0);
    ZigbeeClusterReply *sendQueryNextImageResponse(quint8 transactionSequenceNumber, StatusCode statusCode = StatusCodeNoImageAvailable, quint16 manufacturerCode = 0, quint16 imageType = 0, quint32 fileVersion = 0, quint32 imageSize = 0);

    ZigbeeClusterReply *sendImageBlockResponse(quint8 transactionSequenceNumber, quint16 manufacturerCode, quint16 imageType, quint32 fileVersion, quint32 fileOffset, const QByteArray &imageData);
    ZigbeeClusterReply *sendAbortImageBlockResponse(quint8 transactionSequenceNumber);
    ZigbeeClusterReply *sendDelayImageBlockResponse(quint8 transactionSequenceNumber, const QDateTime &requestTime, quint16 minimumBlockPeriod);

    ZigbeeClusterReply *sendUpgradeEndResponse(quint8 transactionSequenceNumber, quint16 manufacturerCode, quint16 imageType, quint32 fileVersion, quint32 serverTime = 0, quint32 requestTime = 1);
    ZigbeeClusterReply *sendAbortUpgradeEndResponse(quint8 transactionSequenceNumber);

    static FileVersion parseFileVersion(quint32 fileVersionValue);

signals:
    void queryNextImageRequestReceived(quint8 transactionSequenceNumber, quint16 manufactuerCode, quint16 imageType, quint32 fileVersion, quint16 hardwareVersion);
    void imageBlockRequestReceived(quint8 transactionSequenceNumber, quint16 manufacturerCode, quint16 imageType, quint32 fileVersion, quint32 fileOffset, quint8 maximumDataSize, const ZigbeeAddress &requestNodeAddress, quint16 minimumBlockPeriod);
    void upgradeEndRequestReceived(quint8 transactionSequenceNumber, StatusCode statusCode, quint16 manufacturerCode, quint16 imageType, quint32 fileVersion);
protected:
    void processDataIndication(ZigbeeClusterLibrary::Frame frame) override;

};

QDebug operator<<(QDebug debug, const ZigbeeClusterOta::FileVersion &fileVersion);


#endif // ZIGBEECLUSTEROTA_H