/*
 *
 *    Copyright (c) 2024 Project CHIP Authors
 *
 *    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 <lib/core/TLVVectorWriter.h>

#include <cstdint>
#include <vector>

#include <lib/core/CHIPError.h>
#include <lib/core/TLVCommon.h>

namespace chip {
namespace TLV {

namespace {

constexpr uint32_t kIpv6MtuSize = 1280;

} // namespace

TlvVectorWriter::TlvVectorWriter(std::vector<uint8_t> & buffer) : mVectorBuffer(buffer)
{
    Init(mVectorBuffer);
}

TlvVectorWriter::~TlvVectorWriter() = default;

TlvVectorWriter::TlvVectorBuffer::TlvVectorBuffer(std::vector<uint8_t> & buffer) : mFinalBuffer(buffer) {}

TlvVectorWriter::TlvVectorBuffer::~TlvVectorBuffer() {}

CHIP_ERROR TlvVectorWriter::TlvVectorBuffer::OnInit(TLVWriter & /*writer*/, uint8_t *& bufStart, uint32_t & bufLen)
{
    VerifyOrReturnError(mFinalBuffer.empty(), CHIP_ERROR_INCORRECT_STATE);

    ResizeWriteBuffer(bufStart, bufLen);

    return CHIP_NO_ERROR;
}

CHIP_ERROR TlvVectorWriter::TlvVectorBuffer::GetNewBuffer(TLVWriter & /*writer*/, uint8_t *& bufStart, uint32_t & bufLen)
{
    VerifyOrReturnError(!mFinalBuffer.empty(), CHIP_ERROR_INCORRECT_STATE);
    VerifyOrReturnError(mWritingBuffer.data() == bufStart, CHIP_ERROR_INCORRECT_STATE);

    ResizeWriteBuffer(bufStart, bufLen);

    return CHIP_NO_ERROR;
}

CHIP_ERROR TlvVectorWriter::TlvVectorBuffer::FinalizeBuffer(TLVWriter & /*writer*/, uint8_t * bufStart, uint32_t bufLen)
{
    VerifyOrReturnError(mWritingBuffer.data() == bufStart, CHIP_ERROR_INCORRECT_STATE);
    VerifyOrReturnError(bufLen <= mWritingBuffer.size(), CHIP_ERROR_BUFFER_TOO_SMALL);

    mWritingBuffer.resize(bufLen);

    mFinalBuffer.insert(mFinalBuffer.end(), mWritingBuffer.begin(), mWritingBuffer.end());
    mWritingBuffer.resize(0);

    mFinalBuffer.shrink_to_fit();

    return CHIP_NO_ERROR;
}

void TlvVectorWriter::TlvVectorBuffer::ResizeWriteBuffer(uint8_t *& bufStart, uint32_t & bufLen)
{
    VerifyOrReturn(mWritingBuffer.empty());

    mWritingBuffer.resize(kIpv6MtuSize);
    bufStart = mWritingBuffer.data();

    auto size = mWritingBuffer.size();
    VerifyOrReturn(size <= std::numeric_limits<uint32_t>::max());
    bufLen = static_cast<uint32_t>(size);
}

} // namespace TLV
} // namespace chip