/*
 * Copyright (c) 2021 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.
 */

#include "notification_basic_content.h"
#include "ans_log_wrapper.h"
#include "ans_image_util.h"
#include "ans_const_define.h"

namespace OHOS {
namespace Notification {
NotificationBasicContent::~NotificationBasicContent()
{}

void NotificationBasicContent::SetAdditionalText(const std::string &additionalText)
{
    additionalText_ = additionalText;
}

std::string NotificationBasicContent::GetAdditionalText() const
{
    return additionalText_;
}

void NotificationBasicContent::SetText(const std::string &text)
{
    text_ = text;
}

std::string NotificationBasicContent::GetText() const
{
    return text_;
}

void NotificationBasicContent::SetTitle(const std::string &title)
{
    title_ = title;
}

std::string NotificationBasicContent::GetTitle() const
{
    return title_;
}

void NotificationBasicContent::SetLockScreenPicture(const std::shared_ptr<Media::PixelMap> &lockScreenPicture)
{
    lockScreenPicture_ = lockScreenPicture;
}

std::shared_ptr<Media::PixelMap> NotificationBasicContent::GetLockScreenPicture() const
{
    return lockScreenPicture_;
}

void NotificationBasicContent::SetStructuredText(
    const std::vector<std::pair<std::string, std::string>> &structuredText)
{
    structuredText_ = structuredText;
}

std::vector<std::pair<std::string, std::string>> NotificationBasicContent::GetStructuredText() const
{
    return structuredText_;
}

std::string NotificationBasicContent::Dump()
{
    return "title = " + title_ + ", text = " + text_ + ", additionalText = " + additionalText_ +
    ", lockScreenPicture = " + (lockScreenPicture_ ? "not null" : "null") +
    ", structuredText = " + (structuredText_.empty() ? "null" : "not null");
}

bool NotificationBasicContent::ToJson(nlohmann::json &jsonObject) const
{
    jsonObject["text"]           = text_;
    jsonObject["title"]          = title_;
    jsonObject["additionalText"] = additionalText_;
    jsonObject["lockscreenPicture"] = AnsImageUtil::PackImage(lockScreenPicture_);

    nlohmann::json structuredTextJson;
    for (const auto& [key, value] : structuredText_) {
        structuredTextJson[key] = value;
    }
    jsonObject["structuredText"] = structuredTextJson;

    return true;
}

void NotificationBasicContent::ReadFromJson(const nlohmann::json &jsonObject)
{
    if (jsonObject.is_null() or !jsonObject.is_object()) {
        ANS_LOGE("Invalid JSON object");
        return;
    }

    const auto &jsonEnd = jsonObject.cend();
    if (jsonObject.find("text") != jsonEnd && jsonObject.at("text").is_string()) {
        text_ = jsonObject.at("text").get<std::string>();
    }

    if (jsonObject.find("title") != jsonEnd && jsonObject.at("title").is_string()) {
        title_ = jsonObject.at("title").get<std::string>();
    }

    if (jsonObject.find("additionalText") != jsonEnd && jsonObject.at("additionalText").is_string()) {
        additionalText_ = jsonObject.at("additionalText").get<std::string>();
    }

    if (jsonObject.find("lockscreenPicture") != jsonEnd && jsonObject.at("lockscreenPicture").is_string()) {
        auto lockScreenPictureStr = jsonObject.at("lockscreenPicture").get<std::string>();
        lockScreenPicture_ = AnsImageUtil::UnPackImage(lockScreenPictureStr);
    }

    if (jsonObject.find("structuredText") != jsonEnd) {
        auto structuredTextJson = jsonObject.at("structuredText");
        if (!structuredTextJson.empty()) {
            for (const auto& iter : structuredTextJson.items()) {
                if (iter.value().is_string()) {
                    structuredText_.emplace_back(iter.key(), iter.value().get<std::string>());
                }
            }
        }
    }
    return;
}

bool NotificationBasicContent::Marshalling(Parcel &parcel) const
{
    if (!parcel.WriteString(text_)) {
        ANS_LOGE("Failed to write text");
        return false;
    }

    if (!parcel.WriteString(title_)) {
        ANS_LOGE("Failed to write title");
        return false;
    }

    if (!parcel.WriteString(additionalText_)) {
        ANS_LOGE("Failed to write additional text");
        return false;
    }

    auto valid = lockScreenPicture_ ? true : false;
    if (!parcel.WriteBool(valid)) {
        ANS_LOGE("Failed to write the flag which indicate whether lockScreenPicture is null");
        return false;
    }

    if (valid) {
        if (!parcel.WriteParcelable(lockScreenPicture_.get())) {
            ANS_LOGE("Failed to write lockScreenPicture");
            return false;
        }
    }

    int32_t size = static_cast<int32_t>(structuredText_.size());
    if (!parcel.WriteInt32(size)) {
        ANS_LOGE("Failed to write structuredText size.");
        return false;
    }
    for (const auto& [key, value] : structuredText_) {
        if (!parcel.WriteString(key)) {
            ANS_LOGE("Failed to write structuredText key.");
            return false;
        }
        if (!parcel.WriteString(value)) {
            ANS_LOGE("Failed to write structuredText value.");
            return false;
        }
    }

    return true;
}

bool NotificationBasicContent::ReadFromParcel(Parcel &parcel)
{
    if (!parcel.ReadString(text_)) {
        ANS_LOGE("Failed to read text");
        return false;
    }

    if (!parcel.ReadString(title_)) {
        ANS_LOGE("Failed to read title");
        return false;
    }

    if (!parcel.ReadString(additionalText_)) {
        ANS_LOGE("Failed to read additional text");
        return false;
    }

    auto valid = parcel.ReadBool();
    if (valid) {
        lockScreenPicture_ = std::shared_ptr<Media::PixelMap>(parcel.ReadParcelable<Media::PixelMap>());
        if (!lockScreenPicture_) {
            ANS_LOGE("Failed to read lockScreenPicture");
            return false;
        }
    }

    int32_t size = -1;
    if (!parcel.ReadInt32(size)) {
        ANS_LOGE("Failed to read structuredText size.");
        return false;
    }
    if (size < 0 || size > MAX_PARCELABLE_VECTOR_NUM) {
        ANS_LOGE("Invalid structuredText size: %{public}d", size);
        return false;
    }
    for (int32_t i = 0; i < size; ++i) {
        std::string key;
        std::string value;
        if (!parcel.ReadString(key)) {
            ANS_LOGE("Failed to read structuredText key.");
            return false;
        }
        if (!parcel.ReadString(value)) {
            ANS_LOGE("Failed to read structuredText value.");
            return false;
        }
        structuredText_.emplace_back(key, value);
    }

    return true;
}
}  // namespace Notification
}  // namespace OHOS