/*
 *   Copyright (c) 2022 Project CHIP Authors
 *   All rights reserved.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 */

#include "Decoder.h"

#include "../logging/Log.h"

#include <lib/core/CHIPSafeCasts.h>
#include <protocols/bdx/BdxMessages.h>

namespace {
constexpr char kProtocolName[] = "Bulk Data Exchange";

constexpr char kUnknown[]            = "Unknown";
constexpr char kSendInit[]           = "Send Init";
constexpr char kSendAccept[]         = "Send Accept";
constexpr char kReceiveInit[]        = "Receive Init";
constexpr char kReceiveAccept[]      = "Receive Accept";
constexpr char kBlockQuery[]         = "Block Query";
constexpr char kBlock[]              = "Block";
constexpr char kBlockEOF[]           = "Block End Of File";
constexpr char kBlockAck[]           = "Block Ack";
constexpr char kBlockAckEOF[]        = "Block Ack End Of File";
constexpr char kBlockQueryWithSkip[] = "Block Query With Skip";

constexpr char kDataHeader[] = "Data";
} // namespace

using MessageType          = chip::bdx::MessageType;
using RangeControlFlags    = chip::bdx::RangeControlFlags;
using TransferControlFlags = chip::bdx::TransferControlFlags;

namespace chip {
namespace trace {
namespace bdx {

using namespace logging;

CHIP_ERROR DecodeTransferInit(System::PacketBufferHandle msgData);
CHIP_ERROR DecodeSendAccept(System::PacketBufferHandle msgData);
CHIP_ERROR DecodeReceiveAccept(System::PacketBufferHandle msgData);
CHIP_ERROR DecodeBlockCounter(System::PacketBufferHandle msgData);
CHIP_ERROR DecodeDataBlock(System::PacketBufferHandle msgData);
CHIP_ERROR DecodeBlockQueryWithSkip(System::PacketBufferHandle msgData);
void DecodeAndPrintTransferControl(const char * header, BitFlags<TransferControlFlags> & flags);
void DecodeAndPrintRangeControl(const char * header, BitFlags<RangeControlFlags> & flags);
void DecodeAndPrintMetadata(const ByteSpan & data);

const char * ToProtocolName()
{
    return kProtocolName;
}

const char * ToProtocolMessageTypeName(uint8_t protocolCode)
{
    switch (protocolCode)
    {
    case to_underlying(MessageType::SendInit):
        return kSendInit;
    case to_underlying(MessageType::SendAccept):
        return kSendAccept;
    case to_underlying(MessageType::ReceiveInit):
        return kReceiveInit;
    case to_underlying(MessageType::ReceiveAccept):
        return kReceiveAccept;
    case to_underlying(MessageType::BlockQuery):
        return kBlockQuery;
    case to_underlying(MessageType::Block):
        return kBlock;
    case to_underlying(MessageType::BlockEOF):
        return kBlockEOF;
    case to_underlying(MessageType::BlockAck):
        return kBlockAck;
    case to_underlying(MessageType::BlockAckEOF):
        return kBlockAckEOF;
    case to_underlying(MessageType::BlockQueryWithSkip):
        return kBlockQueryWithSkip;
    }

    return kUnknown;
}

CHIP_ERROR LogAsProtocolMessage(uint8_t protocolCode, const uint8_t * data, size_t len)
{
    auto msgData = System::PacketBufferHandle::NewWithData(data, len);

    switch (protocolCode)
    {
    case to_underlying(MessageType::SendInit):
        return DecodeTransferInit(std::move(msgData));
    case to_underlying(MessageType::SendAccept):
        return DecodeSendAccept(std::move(msgData));
    case to_underlying(MessageType::ReceiveInit):
        return DecodeTransferInit(std::move(msgData));
    case to_underlying(MessageType::ReceiveAccept):
        return DecodeReceiveAccept(std::move(msgData));
    case to_underlying(MessageType::BlockQuery):
        return DecodeBlockCounter(std::move(msgData));
    case to_underlying(MessageType::Block):
        return DecodeDataBlock(std::move(msgData));
    case to_underlying(MessageType::BlockEOF):
        return DecodeDataBlock(std::move(msgData));
    case to_underlying(MessageType::BlockAck):
        return DecodeBlockCounter(std::move(msgData));
    case to_underlying(MessageType::BlockAckEOF):
        return DecodeBlockCounter(std::move(msgData));
    case to_underlying(MessageType::BlockQueryWithSkip):
        return DecodeBlockQueryWithSkip(std::move(msgData));
    }

    return CHIP_ERROR_NOT_IMPLEMENTED;
}

CHIP_ERROR DecodeTransferInit(System::PacketBufferHandle msgData)
{
    auto scopedIndent = ScopedLogIndentWithSize(kDataHeader, msgData->DataLength());

    chip::bdx::TransferInit msg;
    ReturnErrorOnFailure(msg.Parse(msgData.Retain()));

    auto ptc      = msg.TransferCtlOptions;
    auto rc       = msg.mRangeCtlFlags;
    auto pmbs     = msg.MaxBlockSize;
    auto startofs = msg.StartOffset;
    auto len      = msg.MaxLength;
    auto fdl      = msg.FileDesLength;
    auto fd       = CharSpan(Uint8::to_const_char(msg.FileDesignator), msg.FileDesLength);
    auto mdata    = ByteSpan(msg.Metadata, msg.MetadataLength);

    DecodeAndPrintTransferControl("Proposed Transfer Control", ptc);
    DecodeAndPrintRangeControl("Range Control", rc);
    Log("Proposed Max Block Size", pmbs);

    if (rc.Has(RangeControlFlags::kStartOffset))
    {
        LogAsHex("Start Offset", startofs);
    }

    if (rc.Has(RangeControlFlags::kDefLen))
    {
        LogAsHex("Proposed Max Length", len);
    }

    Log("File Designator Length", fdl);
    Log("File Designator", fd);
    DecodeAndPrintMetadata(mdata);

    return CHIP_NO_ERROR;
}

CHIP_ERROR DecodeSendAccept(System::PacketBufferHandle msgData)
{
    auto scopedIndent = ScopedLogIndentWithSize(kDataHeader, msgData->DataLength());

    chip::bdx::SendAccept msg;
    ReturnErrorOnFailure(msg.Parse(msgData.Retain()));

    auto tc    = msg.TransferCtlFlags;
    auto mbs   = msg.MaxBlockSize;
    auto mdata = ByteSpan(msg.Metadata, msg.MetadataLength);

    DecodeAndPrintTransferControl("Transfer Control", tc);
    Log("Max Block Size", mbs);
    DecodeAndPrintMetadata(mdata);

    return CHIP_NO_ERROR;
}

CHIP_ERROR DecodeReceiveAccept(System::PacketBufferHandle msgData)
{
    auto scopedIndent = ScopedLogIndentWithSize(kDataHeader, msgData->DataLength());

    chip::bdx::ReceiveAccept msg;
    ReturnErrorOnFailure(msg.Parse(msgData.Retain()));

    auto tc    = msg.TransferCtlFlags;
    auto rc    = msg.mRangeCtlFlags;
    auto mbs   = msg.MaxBlockSize;
    auto len   = msg.Length;
    auto mdata = ByteSpan(msg.Metadata, msg.MetadataLength);

    DecodeAndPrintTransferControl("Transfer Control", tc);
    DecodeAndPrintRangeControl("Range Control", rc);
    Log("Max Block Size", mbs);

    if (rc.Has(RangeControlFlags::kDefLen))
    {
        LogAsHex("Length", len);
    }

    DecodeAndPrintMetadata(mdata);

    return CHIP_NO_ERROR;
}

CHIP_ERROR DecodeBlockCounter(System::PacketBufferHandle msgData)
{
    auto scopedIndent = ScopedLogIndent(kDataHeader);

    chip::bdx::CounterMessage msg;
    ReturnErrorOnFailure(msg.Parse(msgData.Retain()));

    Log("BlockCounter", msg.BlockCounter);

    return CHIP_NO_ERROR;
}

CHIP_ERROR DecodeDataBlock(System::PacketBufferHandle msgData)
{
    auto scopedIndent = ScopedLogIndentWithSize(kDataHeader, msgData->DataLength());

    chip::bdx::DataBlock msg;
    ReturnErrorOnFailure(msg.Parse(msgData.Retain()));

    Log("BlockCounter", msg.BlockCounter);

    auto data = ByteSpan(msg.Data, msg.DataLength);
    Log("Data", data);

    return CHIP_NO_ERROR;
}

CHIP_ERROR DecodeBlockQueryWithSkip(System::PacketBufferHandle msgData)
{
    auto scopedIndent = ScopedLogIndent(kDataHeader);

    chip::bdx::BlockQueryWithSkip msg;
    ReturnErrorOnFailure(msg.Parse(msgData.Retain()));

    Log("BlockCounter", msg.BlockCounter);
    LogAsHex("BytesToSkip", msg.BytesToSkip);

    return CHIP_NO_ERROR;
}

void DecodeAndPrintTransferControl(const char * header, BitFlags<TransferControlFlags> & flags)
{
    auto scopedIndent = ScopedLogIndentWithFlags(header, flags.Raw());

    if (flags.Has(TransferControlFlags::kSenderDrive))
    {
        Log("SenderDrive");
    }

    if (flags.Has(TransferControlFlags::kReceiverDrive))
    {
        Log("ReceivedDrive");
    }

    if (flags.Has(TransferControlFlags::kAsync))
    {
        Log("Async");
    }
}

void DecodeAndPrintRangeControl(const char * header, BitFlags<RangeControlFlags> & flags)
{
    auto scopedIndent = ScopedLogIndentWithFlags(header, flags.Raw());

    if (flags.Has(chip::bdx::RangeControlFlags::kDefLen))
    {
        Log("DefLen");
    }

    if (flags.Has(RangeControlFlags::kStartOffset))
    {
        Log("StartOffset");
    }

    if (flags.Has(RangeControlFlags::kWiderange))
    {
        Log("Widerange");
    }
}

void DecodeAndPrintMetadata(const ByteSpan & data)
{
    if (data.size())
    {
        // TODO Make metadata decoding prettier.
        Log("HasMetadata");
    }
}

} // namespace bdx
} // namespace trace
} // namespace chip