/*
 * Copyright (C) 2025 Huawei Device Co., Ltd.
 * 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.
 */

#ifndef AVBUFFER_H
#define AVBUFFER_H
 
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <memory>
#include <string>
#include "buffer/avallocator.h"
#include "common/status.h"

namespace OHOS {
namespace Media {
class AVMemory;
class AVBuffer {
public:
    AVBuffer() = default;
    virtual ~AVBuffer() = default;
    MOCK_METHOD(const AVBufferConfig&, GetConfig, (), ());
    MOCK_METHOD(uint64_t, GetUniqueId, (), ());
    MOCK_METHOD(bool, WriteToMessageParcel, (MessageParcel& parcel), ());
    MOCK_METHOD(bool, ReadFromMessageParcel, (MessageParcel& parcel, bool isSurfaceBuffer), ());
    static std::shared_ptr<AVBuffer> CreateAVBuffer(const AVBufferConfig &config)
    {
        (void)config;
        return std::make_shared<AVBuffer>();
    }
    static std::shared_ptr<AVBuffer> CreateAVBuffer(std::shared_ptr<AVAllocator> allocator, int32_t capacity = 0,
                                                    int32_t align = 0)
    {
        (void)allocator;
        (void)capacity;
        (void)align;
        return std::make_shared<AVBuffer>();
    }
    static std::shared_ptr<AVBuffer> CreateAVBuffer(uint8_t *ptr, int32_t capacity, int32_t size = 0)
    {
        (void)ptr;
        (void)capacity;
        (void)size;
        return std::make_shared<AVBuffer>();
    }
    static std::shared_ptr<AVBuffer> CreateAVBuffer(sptr<SurfaceBuffer> surfaceBuffer)
    {
        (void)surfaceBuffer;
        return std::make_shared<AVBuffer>();
    }
protected:
    using MetaData = std::vector<uint8_t>;
    int64_t pts_;
    int64_t dts_;
    int64_t duration_;
    uint32_t flag_;
    std::shared_ptr<Meta> meta_;
    std::shared_ptr<AVMemory> memory_;
    AVBufferConfig config_;
    MOCK_METHOD(Status, Init, (std::shared_ptr<AVAllocator> allocator, int32_t capacity, int32_t align), ());
    MOCK_METHOD(Status, Init, (uint8_t* ptr, int32_t capacity, int32_t size), ());
    MOCK_METHOD(Status, Init, (sptr<SurfaceBuffer> surfaceBuffer), ());
};

class AVMemory {
public:
    AVMemory() = default;
    virtual ~AVMemory() = default;
    MOCK_METHOD(MemoryType, GetMemoryType, (), ());
    MOCK_METHOD(MemoryFlag, GetMemoryFlag, (), ());
    MOCK_METHOD(int32_t, GetCapacity, (), ());
    MOCK_METHOD(int32_t, GetSize, (), ());
    MOCK_METHOD(Status, SetSize, (int32_t size), ());
    MOCK_METHOD(int32_t, GetOffset, (), ());
    MOCK_METHOD(Status, SetOffset, (int32_t offset), ());
    MOCK_METHOD(int32_t, GetFileDescriptor, (), ());
    MOCK_METHOD(uint8_t*, GetAddr, (), ());
    MOCK_METHOD(int32_t, Write, (const uint8_t* in, int32_t writeSize, int32_t position), ());
    MOCK_METHOD(int32_t, Read, (uint8_t* out, int32_t readSize, int32_t position), ());
    MOCK_METHOD(void, Reset, (), ());
    MOCK_METHOD(sptr<SurfaceBuffer>, GetSurfaceBuffer, (), ());

protected:
    MOCK_METHOD(Status, Init, (), ());
    MOCK_METHOD(Status, Init, (MessageParcel& parcel), ());
    MOCK_METHOD(Status, InitSurfaceBuffer, (MessageParcel& parcel), ());
    MOCK_METHOD(Status, InitSurfaceBuffer, (sptr<SurfaceBuffer> surfaceBuffer), ());
    MOCK_METHOD(bool, WriteToMessageParcel, (MessageParcel& parcel), ());
    MOCK_METHOD(bool, ReadFromMessageParcel, (MessageParcel& parcel), ());
    MOCK_METHOD(bool, ReadCommonFromMessageParcel, (MessageParcel& parcel), ());
    MOCK_METHOD(bool, SkipCommonFromMessageParcel, (MessageParcel& parcel), ());
    MOCK_METHOD(bool, WriteCommonToMessageParcel, (MessageParcel& parcel), ());
    int32_t capacity_ = 0;
    int32_t align_;
    int32_t offset_;
    int32_t size_;
    uint8_t *base_;
    uint64_t uid_;
    std::shared_ptr<AVAllocator> allocator_;
    static std::shared_ptr<AVMemory> CreateAVMemory(std::shared_ptr<AVAllocator> allocator,
                                                    int32_t capacity = 0, int32_t align = 0)
    {
        return std::make_shared<AVMemory>();
    }
    static std::shared_ptr<AVMemory> CreateAVMemory(uint8_t *ptr, int32_t capacity, int32_t size)
    {
        return std::make_shared<AVMemory>();
    }
    static std::shared_ptr<AVMemory> CreateAVMemory(MessageParcel &parcel, bool isSurfaceBuffer = false)
    {
        return std::make_shared<AVMemory>();
    }
    static std::shared_ptr<AVMemory> CreateAVMemory(sptr<SurfaceBuffer> surfaceBuffer)
    {
        return std::make_shared<AVMemory>();
    }
};
}
}
#endif // MOCK_AVBUFFER_H