/*
 * 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 INTERFACES_INNERKITS_INCLUDE_KV_METADATA_H
#define INTERFACES_INNERKITS_INCLUDE_KV_METADATA_H

#include <map>
#include <mutex>
#include "image_type.h"
#include "metadata.h"

namespace OHOS {
namespace Media {
class ImageKvMetadata : public ImageMetadata {
public:
    ImageKvMetadata();
    ImageKvMetadata(MetadataType metadataType);
    ImageKvMetadata(const ImageKvMetadata &kvMetadata);
    ImageKvMetadata& operator=(const ImageKvMetadata &kvMetadata);
    virtual ~ImageKvMetadata();
    virtual int GetValue(const std::string &key, std::string &value) const override;
    virtual bool SetValue(const std::string &key, const std::string &value) override;
    virtual bool RemoveEntry(const std::string &key) override;
    virtual const ImageMetadata::PropertyMapPtr GetAllProperties() override;
    virtual std::shared_ptr<ImageMetadata> CloneMetadata() override;
    bool Marshalling(Parcel &parcel) const override;
    static ImageKvMetadata *Unmarshalling(Parcel &parcel);
    static ImageKvMetadata *Unmarshalling(Parcel &parcel, PICTURE_ERR &error);
    static bool IsKvMetadata(MetadataType metadataType);
    MetadataType GetType() const override
    {
        return metadataType_;
    }
    bool RemoveExifThumbnail() override
    {
        return false;
    }
    static bool IsFragmentMetadataKey(const std::string& key);
    static bool IsGifMetadataKey(const std::string& key);
    static bool IsWebPMetadataKey(const std::string& key);
    static bool IsAvisMetadataKey(const std::string& key);
    static bool IsJfifMetadataKey(const std::string& key);
    static bool IsHeifsMetadataKey(const std::string& key);
    static bool IsPngMetadataKey(const std::string& key);
    static std::set<std::string> GetFragmentMetadataKeys();
    static std::set<std::string> GetGifMetadataKeys();
    static std::set<std::string> GetWebPMetadataKeys();
    static std::set<std::string> GetAvisMetadataKeys();
    static const std::set<std::string>& GetJfifMetadataKeys();
    static const std::set<std::string>& GetHeifsMetadataKeys();
    static const std::set<std::string>& GetPngMetadataKeys();
    virtual uint32_t GetBlob(uint32_t bufferSize, uint8_t *dst) override;
    virtual uint32_t SetBlob(const uint8_t *source, const uint32_t bufferSize) override;
    virtual uint32_t GetBlobSize() override;
protected:
    MetadataType metadataType_ = MetadataType::UNKNOWN;
    ImageMetadata::PropertyMapPtr properties_ = std::make_shared<ImageMetadata::PropertyMap>();
    mutable std::mutex mutex_;
};
} // namespace Media
} // namespace OHOS

#endif // INTERFACES_INNERKITS_INCLUDE_KV_METADATA_H