ndkimpl/data/uds.cpp

/*
 * Copyright (c) 2024 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.
 */
#define LOG_TAG "Uds"

#include "uds.h"
#include "logger.h"
#include "utd_common.h"
#include "unified_meta.h"
#include "unified_record.h"
#include "udmf_capi_common.h"
#include "udmf_meta.h"
#include "udmf_err_code.h"
#include "pixel_map.h"
#include "pixelmap_native_impl.h"

using namespace OHOS::UDMF;

static const char* GetUdsStrValue(UdsObject* pThis, NdkStructId ndkStructId, const char* pramName)
{
    if (IsInvalidUdsObjectPtr(pThis, ndkStructId)) {
        return nullptr;
    }
    auto value = pThis->GetUdsValue<std::string>(pramName);
    return value == nullptr ? nullptr : value->c_str();
}

bool IsInvalidUdsObjectPtr(const UdsObject* pThis, int cid)
{
    return pThis == nullptr || pThis->cid != cid || pThis->obj == nullptr;
}

bool IsInvalidUdsObjectByType(const UdsObject* pThis, const UDType& type)
{
    switch (type) {
        case PLAIN_TEXT:
            return IsInvalidUdsObjectPtr(pThis, UDS_PLAIN_TEXT_STRUCT_ID);
        case HYPERLINK:
            return IsInvalidUdsObjectPtr(pThis, UDS_HYPERLINK_STRUCT_ID);
        case HTML:
            return IsInvalidUdsObjectPtr(pThis, UDS_HTML_STRUCT_ID);
        case SYSTEM_DEFINED_APP_ITEM:
            return IsInvalidUdsObjectPtr(pThis, UDS_APP_ITEM_STRUCT_ID);
        case FILE_URI:
            return IsInvalidUdsObjectPtr(pThis, UDS_FILE_URI_STRUCT_ID);
        case SYSTEM_DEFINED_PIXEL_MAP:
            return IsInvalidUdsObjectPtr(pThis, UDS_PIXEL_MAP_STRUCT_ID);
        default:
            return false;
    }
}

UdsObject::UdsObject(const int cid) : cid(cid) {}

OH_UdsPlainText::OH_UdsPlainText() : UdsObject(NdkStructId::UDS_PLAIN_TEXT_STRUCT_ID) {}

OH_UdsHyperlink::OH_UdsHyperlink() : UdsObject(NdkStructId::UDS_HYPERLINK_STRUCT_ID) {}

OH_UdsHtml::OH_UdsHtml() : UdsObject(NdkStructId::UDS_HTML_STRUCT_ID) {}

OH_UdsAppItem::OH_UdsAppItem() : UdsObject(NdkStructId::UDS_APP_ITEM_STRUCT_ID) {}

OH_UdsFileUri::OH_UdsFileUri() : UdsObject(NdkStructId::UDS_FILE_URI_STRUCT_ID) {}

OH_UdsPixelMap::OH_UdsPixelMap() : UdsObject(NdkStructId::UDS_PIXEL_MAP_STRUCT_ID) {}

OH_UdsArrayBuffer::OH_UdsArrayBuffer() : UdsObject(NdkStructId::UDS_ARRAY_BUFFER_STRUCT_ID) {}

template<typename T>
bool UdsObject::HasObjectKey(const char* paramName)
{
    auto it = obj->value_.find(paramName);
    if (it == obj->value_.end() || !std::holds_alternative<T>(it->second)) {
        LOG_ERROR(UDMF_CAPI, "Don't have property %{public}s.", paramName);
        return false;
    }
    return true;
}

template<typename T>
T* UdsObject::GetUdsValue(const char* paramName)
{
    if (!HasObjectKey<T>(paramName)) {
        return nullptr;
    }
    return std::get_if<T>(&(obj->value_[paramName]));
}

template<typename T>
int UdsObject::SetUdsValue(const char* paramName, const T pramValue)
{
    if (!HasObjectKey<T>(paramName)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    std::lock_guard<std::mutex> lock(mutex);
    obj->value_[paramName] = pramValue;
    return Udmf_ErrCode::UDMF_E_OK;
}

OH_UdsPlainText* OH_UdsPlainText_Create()
{
    OH_UdsPlainText* plainText = new (std::nothrow) OH_UdsPlainText();
    if (plainText == nullptr) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory.");
        return nullptr;
    }
    plainText->obj = std::make_shared<Object>();
    plainText->obj->value_[UNIFORM_DATA_TYPE] = UDMF_META_PLAIN_TEXT;
    plainText->obj->value_[CONTENT] = "";
    plainText->obj->value_[ABSTRACT] = "";
    return plainText;
}

void OH_UdsPlainText_Destroy(OH_UdsPlainText* pThis)
{
    if (pThis != nullptr && pThis->cid == NdkStructId::UDS_PLAIN_TEXT_STRUCT_ID) {
        delete pThis;
    }
}

const char* OH_UdsPlainText_GetType(OH_UdsPlainText* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_PLAIN_TEXT_STRUCT_ID, UNIFORM_DATA_TYPE);
}

const char* OH_UdsPlainText_GetContent(OH_UdsPlainText* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_PLAIN_TEXT_STRUCT_ID, CONTENT);
}

const char* OH_UdsPlainText_GetAbstract(OH_UdsPlainText* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_PLAIN_TEXT_STRUCT_ID, ABSTRACT);
}

int OH_UdsPlainText_SetContent(OH_UdsPlainText* pThis, const char* content)
{
    if (content == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_PLAIN_TEXT_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(CONTENT, content);
}

int OH_UdsPlainText_SetAbstract(OH_UdsPlainText* pThis, const char* abstract)
{
    if (abstract == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_PLAIN_TEXT_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(ABSTRACT, abstract);
}

OH_UdsHyperlink* OH_UdsHyperlink_Create()
{
    OH_UdsHyperlink* hyperlink = new (std::nothrow) OH_UdsHyperlink();
    if (hyperlink == nullptr) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory.");
        return nullptr;
    }
    hyperlink->obj = std::make_shared<Object>();
    hyperlink->obj->value_[UNIFORM_DATA_TYPE] = UDMF_META_HYPERLINK;
    hyperlink->obj->value_[URL] = "";
    hyperlink->obj->value_[DESCRIPTION] = "";
    return hyperlink;
}

void OH_UdsHyperlink_Destroy(OH_UdsHyperlink* pThis)
{
    if (pThis != nullptr && pThis->cid == NdkStructId::UDS_HYPERLINK_STRUCT_ID) {
        delete pThis;
    }
}

const char* OH_UdsHyperlink_GetType(OH_UdsHyperlink* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_HYPERLINK_STRUCT_ID, UNIFORM_DATA_TYPE);
}

const char* OH_UdsHyperlink_GetUrl(OH_UdsHyperlink* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_HYPERLINK_STRUCT_ID, URL);
}

const char* OH_UdsHyperlink_GetDescription(OH_UdsHyperlink* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_HYPERLINK_STRUCT_ID, DESCRIPTION);
}

int OH_UdsHyperlink_SetUrl(OH_UdsHyperlink* pThis, const char* url)
{
    if (url == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_HYPERLINK_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(URL, url);
}

int OH_UdsHyperlink_SetDescription(OH_UdsHyperlink* pThis, const char* description)
{
    if (description == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_HYPERLINK_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(DESCRIPTION, description);
}

OH_UdsHtml* OH_UdsHtml_Create()
{
    OH_UdsHtml* html = new (std::nothrow) OH_UdsHtml();
    if (html == nullptr) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory.");
        return nullptr;
    }
    html->obj = std::make_shared<Object>();
    html->obj->value_[UNIFORM_DATA_TYPE] = UDMF_META_HTML;
    html->obj->value_[HTML_CONTENT] = "";
    html->obj->value_[PLAIN_CONTENT] = "";
    return html;
}

void OH_UdsHtml_Destroy(OH_UdsHtml* pThis)
{
    if (pThis != nullptr && pThis->cid == NdkStructId::UDS_HTML_STRUCT_ID) {
        delete pThis;
    }
}

const char* OH_UdsHtml_GetType(OH_UdsHtml* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_HTML_STRUCT_ID, UNIFORM_DATA_TYPE);
}

const char* OH_UdsHtml_GetContent(OH_UdsHtml* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_HTML_STRUCT_ID, HTML_CONTENT);
}

const char* OH_UdsHtml_GetPlainContent(OH_UdsHtml* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_HTML_STRUCT_ID, PLAIN_CONTENT);
}

int OH_UdsHtml_SetContent(OH_UdsHtml* pThis, const char* content)
{
    if (content == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_HTML_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(HTML_CONTENT, content);
}

int OH_UdsHtml_SetPlainContent(OH_UdsHtml* pThis, const char* plainContent)
{
    if (plainContent == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_HTML_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(PLAIN_CONTENT, plainContent);
}

OH_UdsAppItem* OH_UdsAppItem_Create()
{
    OH_UdsAppItem* appItem = new (std::nothrow) OH_UdsAppItem();
    if (appItem == nullptr) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory.");
        return nullptr;
    }
    appItem->obj = std::make_shared<Object>();
    appItem->obj->value_[UNIFORM_DATA_TYPE] = UDMF_META_OPENHARMONY_APP_ITEM;
    appItem->obj->value_[APP_ID] = "";
    appItem->obj->value_[APP_NAME] = "";
    appItem->obj->value_[APP_ICON_ID] = "";
    appItem->obj->value_[APP_LABEL_ID] = "";
    appItem->obj->value_[BUNDLE_NAME] = "";
    appItem->obj->value_[ABILITY_NAME] = "";
    return appItem;
}

void OH_UdsAppItem_Destroy(OH_UdsAppItem* pThis)
{
    if (pThis != nullptr && pThis->cid == NdkStructId::UDS_APP_ITEM_STRUCT_ID) {
        delete pThis;
    }
}

const char* OH_UdsAppItem_GetType(OH_UdsAppItem* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID, UNIFORM_DATA_TYPE);
}

const char* OH_UdsAppItem_GetId(OH_UdsAppItem* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID, APP_ID);
}

const char* OH_UdsAppItem_GetName(OH_UdsAppItem* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID, APP_NAME);
}

const char* OH_UdsAppItem_GetIconId(OH_UdsAppItem* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID, APP_ICON_ID);
}

const char* OH_UdsAppItem_GetLabelId(OH_UdsAppItem* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID, APP_LABEL_ID);
}

const char* OH_UdsAppItem_GetBundleName(OH_UdsAppItem* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID, BUNDLE_NAME);
}

const char* OH_UdsAppItem_GetAbilityName(OH_UdsAppItem* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID, ABILITY_NAME);
}

int OH_UdsAppItem_SetId(OH_UdsAppItem* pThis, const char* appId)
{
    if (appId == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(APP_ID, appId);
}

int OH_UdsAppItem_SetName(OH_UdsAppItem* pThis, const char* appName)
{
    if (appName == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(APP_NAME, appName);
}

int OH_UdsAppItem_SetIconId(OH_UdsAppItem* pThis, const char* appIconId)
{
    if (appIconId == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(APP_ICON_ID, appIconId);
}

int OH_UdsAppItem_SetLabelId(OH_UdsAppItem* pThis, const char* appLabelId)
{
    if (appLabelId == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(APP_LABEL_ID, appLabelId);
}

int OH_UdsAppItem_SetBundleName(OH_UdsAppItem* pThis, const char* bundleName)
{
    if (bundleName == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(BUNDLE_NAME, bundleName);
}

int OH_UdsAppItem_SetAbilityName(OH_UdsAppItem* pThis, const char* abilityName)
{
    if (abilityName == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_APP_ITEM_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(ABILITY_NAME, abilityName);
}

OH_UdsFileUri* OH_UdsFileUri_Create()
{
    OH_UdsFileUri* fileUri = new (std::nothrow) OH_UdsFileUri();
    if (fileUri == nullptr) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory.");
        return nullptr;
    }
    fileUri->obj = std::make_shared<Object>();
    fileUri->obj->value_[UNIFORM_DATA_TYPE] = UDMF_META_GENERAL_FILE_URI;
    fileUri->obj->value_[FILE_URI_PARAM] = "";
    fileUri->obj->value_[FILE_TYPE] = "";
    return fileUri;
}

void OH_UdsFileUri_Destroy(OH_UdsFileUri* pThis)
{
    if (pThis != nullptr && pThis->cid == NdkStructId::UDS_FILE_URI_STRUCT_ID) {
        delete pThis;
    }
}

const char* OH_UdsFileUri_GetType(OH_UdsFileUri* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_FILE_URI_STRUCT_ID, UNIFORM_DATA_TYPE);
}

const char* OH_UdsFileUri_GetFileUri(OH_UdsFileUri* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_FILE_URI_STRUCT_ID, FILE_URI_PARAM);
}

const char* OH_UdsFileUri_GetFileType(OH_UdsFileUri* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_FILE_URI_STRUCT_ID, FILE_TYPE);
}

int OH_UdsFileUri_SetFileUri(OH_UdsFileUri* pThis, const char* fileUri)
{
    if (fileUri == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_FILE_URI_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(FILE_URI_PARAM, fileUri);
}

int OH_UdsFileUri_SetFileType(OH_UdsFileUri* pThis, const char* fileType)
{
    if (fileType == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_FILE_URI_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::string>(FILE_TYPE, fileType);
}

OH_UdsPixelMap* OH_UdsPixelMap_Create()
{
    OH_UdsPixelMap* pixelMap = new (std::nothrow) OH_UdsPixelMap();
    if (pixelMap == nullptr) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory.");
        return nullptr;
    }
    pixelMap->obj = std::make_shared<Object>();
    pixelMap->obj->value_[UNIFORM_DATA_TYPE] = UDMF_META_OPENHARMONY_PIXEL_MAP;
    pixelMap->obj->value_[PIXEL_MAP] = std::make_shared<OHOS::Media::PixelMap>();
    return pixelMap;
}

void OH_UdsPixelMap_Destroy(OH_UdsPixelMap* pThis)
{
    if (pThis != nullptr && pThis->cid == NdkStructId::UDS_PIXEL_MAP_STRUCT_ID) {
        delete pThis;
    }
}

const char* OH_UdsPixelMap_GetType(OH_UdsPixelMap* pThis)
{
    return GetUdsStrValue(pThis, NdkStructId::UDS_PIXEL_MAP_STRUCT_ID, UNIFORM_DATA_TYPE);
}

void OH_UdsPixelMap_GetPixelMap(OH_UdsPixelMap* pThis, OH_PixelmapNative* pixelmapNative)
{
    if (IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_PIXEL_MAP_STRUCT_ID)) {
        return;
    }
    auto pixelMap = pThis->GetUdsValue<std::shared_ptr<OHOS::Media::PixelMap>>(PIXEL_MAP);
    if (pixelMap != nullptr) {
        *pixelmapNative = OH_PixelmapNative(*pixelMap);
    }
}

int OH_UdsPixelMap_SetPixelMap(OH_UdsPixelMap* pThis, OH_PixelmapNative* pixelmapNative)
{
    if (pixelmapNative == nullptr || IsInvalidUdsObjectPtr(pThis, NdkStructId::UDS_PIXEL_MAP_STRUCT_ID)) {
        return Udmf_ErrCode::UDMF_E_INVALID_PARAM;
    }
    return pThis->SetUdsValue<std::shared_ptr<OHOS::Media::PixelMap>>(PIXEL_MAP, pixelmapNative->GetInnerPixelmap());
}

OH_UdsArrayBuffer* OH_UdsArrayBuffer_Create()
{
    auto *buffer = new (std::nothrow) OH_UdsArrayBuffer();
    if (buffer == nullptr) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory.");
        return nullptr;
    }
    buffer->obj = std::make_shared<Object>();
    buffer->obj->value_[UNIFORM_DATA_TYPE] = "";
    buffer->obj->value_[ARRAY_BUFFER] = std::vector<uint8_t>();
    buffer->obj->value_[ARRAY_BUFFER_LENGTH] = 0;
    return buffer;
}

int OH_UdsArrayBuffer_Destroy(OH_UdsArrayBuffer* buffer)
{
    if (IsInvalidUdsObjectPtr(buffer, NdkStructId::UDS_ARRAY_BUFFER_STRUCT_ID)) {
        LOG_ERROR(UDMF_CAPI, "Param is invalid.");
        return UDMF_E_INVALID_PARAM;
    }
    delete buffer;
    return UDMF_E_OK;
}

int OH_UdsArrayBuffer_SetData(OH_UdsArrayBuffer* buffer, unsigned char* data, unsigned int len)
{
    if (data == nullptr || len == 0 || IsInvalidUdsObjectPtr(buffer, NdkStructId::UDS_ARRAY_BUFFER_STRUCT_ID) ||
        len > MAX_GENERAL_ENTRY_SIZE) {
        LOG_ERROR(UDMF_CAPI, "Param is invalid.");
        return UDMF_E_INVALID_PARAM;
    }
    std::vector<uint8_t> arrayBuffer(data, data + len);
    int ret = buffer->SetUdsValue<std::vector<uint8_t>>(ARRAY_BUFFER, arrayBuffer);
    if (ret != UDMF_E_OK) {
        LOG_ERROR(UDMF_CAPI, "Failed to apply for memory. ret: %{public}d", ret);
        return ret;
    }
    ret = buffer->SetUdsValue<int>(ARRAY_BUFFER_LENGTH, static_cast<int>(len));
    return ret;
}

int OH_UdsArrayBuffer_GetData(OH_UdsArrayBuffer* buffer, unsigned char** data, unsigned int* len)
{
    if (buffer == nullptr || IsInvalidUdsObjectPtr(buffer, NdkStructId::UDS_ARRAY_BUFFER_STRUCT_ID)) {
        LOG_ERROR(UDMF_CAPI, "Param is invalid.");
        return UDMF_E_INVALID_PARAM;
    }
    const auto arrayBuffer = buffer->GetUdsValue<std::vector<uint8_t>>(ARRAY_BUFFER);
    if (arrayBuffer == nullptr || arrayBuffer->empty()) {
        return UDMF_ERR;
    }
    const auto length = buffer->GetUdsValue<int>(ARRAY_BUFFER_LENGTH);
    if (length == nullptr || *length <= 0 || *length > MAX_GENERAL_ENTRY_SIZE) {
        return UDMF_ERR;
    }

    *data = arrayBuffer->data();
    *len = *length;
    return UDMF_E_OK;
}