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

#include "asset_napi_add.h"

#include "asset_log.h"
#include "asset_system_api.h"
#include "asset_system_type.h"

#include "asset_napi_check.h"
#include "asset_napi_common.h"

namespace OHOS {
namespace Security {
namespace Asset {
namespace {

const uint32_t ADD_ARG_COUNT = 1;
const uint32_t ADD_ARG_COUNT_AS_USER = 2;

const std::vector<uint32_t> REQUIRED_TAGS = {
    SEC_ASSET_TAG_SECRET,
    SEC_ASSET_TAG_ALIAS
};

const std::vector<uint32_t> OPTIONAL_TAGS = {
    SEC_ASSET_TAG_SECRET,
    SEC_ASSET_TAG_CONFLICT_RESOLUTION
};

napi_status CheckAddArgsCommon(const napi_env env, const std::vector<AssetAttr> &attrs,
    const std::vector<uint32_t> &validTags)
{
    IF_ERROR_THROW_RETURN(env, CheckAssetRequiredTag(env, attrs, REQUIRED_TAGS, SEC_ASSET_INVALID_ARGUMENT));
    IF_ERROR_THROW_RETURN(env, CheckAssetTagValidity(env, attrs, validTags, SEC_ASSET_INVALID_ARGUMENT));
    IF_ERROR_THROW_RETURN(env, CheckAssetValueValidity(env, attrs, SEC_ASSET_INVALID_ARGUMENT));
    return napi_ok;
}

napi_status CheckAddArgs(const napi_env env, const std::vector<AssetAttr> &attrs)
{
    std::vector<uint32_t> validTags;
    validTags.insert(validTags.end(), CRITICAL_LABEL_TAGS.begin(), CRITICAL_LABEL_TAGS.end());
    validTags.insert(validTags.end(), NORMAL_LABEL_TAGS.begin(), NORMAL_LABEL_TAGS.end());
    validTags.insert(validTags.end(), NORMAL_LOCAL_LABEL_TAGS.begin(), NORMAL_LOCAL_LABEL_TAGS.end());
    validTags.insert(validTags.end(), ACCESS_CONTROL_TAGS.begin(), ACCESS_CONTROL_TAGS.end());
    validTags.insert(validTags.end(), ASSET_SYNC_TAGS.begin(), ASSET_SYNC_TAGS.end());
    validTags.insert(validTags.end(), OPTIONAL_TAGS.begin(), OPTIONAL_TAGS.end());
    return CheckAddArgsCommon(env, attrs, validTags);
}

napi_status CheckBatchAddArgs(const napi_env env, const std::vector<AssetAttr> &attrs)
{
    std::vector<uint32_t> validTags;
    validTags.insert(validTags.end(), CRITICAL_LABEL_TAGS.begin(), CRITICAL_LABEL_TAGS.end());
    validTags.insert(validTags.end(), NORMAL_LABEL_TAGS.begin(), NORMAL_LABEL_TAGS.end());
    validTags.insert(validTags.end(), NORMAL_LOCAL_LABEL_TAGS.begin(), NORMAL_LOCAL_LABEL_TAGS.end());
    validTags.insert(validTags.end(), ACCESS_CONTROL_TAGS_FOR_BATCH.begin(), ACCESS_CONTROL_TAGS_FOR_BATCH.end());
    validTags.insert(validTags.end(), ASSET_SYNC_TAGS.begin(), ASSET_SYNC_TAGS.end());
    validTags.insert(validTags.end(), OPTIONAL_TAGS.begin(), OPTIONAL_TAGS.end());
    return CheckAddArgsCommon(env, attrs, validTags);
}

napi_status ParseAttrMap(napi_env env, napi_callback_info info, BaseContext *context)
{
    napi_value argv[MAX_ARGS_NUM] = { 0 };
    IF_ERR_RETURN(ParseJsArgs(env, info, argv, ADD_ARG_COUNT));
    IF_ERR_RETURN(ParseJsMap(env, argv[0], context->attrs));
    IF_ERR_RETURN(CheckAddArgs(env, context->attrs));
    return napi_ok;
}

napi_status ParseAttrMapArray(napi_env env, napi_callback_info info, BaseContext *baseContext)
{
    BatchOperationContext *context = static_cast<BatchOperationContext *>(baseContext);
    napi_value argv[NORMAL_ARGS_NUM] = { 0 };
    IF_ERR_RETURN(ParseJsArgs(env, info, argv, ADD_ARG_COUNT));
    IF_ERR_RETURN(ParseJsMapArray(env, argv[0], context->attrsArray));
    if (context->attrsArray.empty()) {
        LOGE("[FATAL]Batch Add argument empty.");
        return napi_invalid_arg;
    }
    for (const auto &attrs : context->attrsArray) {
        IF_ERR_RETURN(CheckBatchAddArgs(env, attrs));
    }
    return napi_ok;
}

napi_status ParseAttrMapAsUser(napi_env env, napi_callback_info info, BaseContext *context)
{
    napi_value argv[MAX_ARGS_NUM] = { 0 };
    IF_ERR_RETURN(ParseJsArgs(env, info, argv, ADD_ARG_COUNT_AS_USER));
    uint32_t index = 0;
    IF_ERR_RETURN(ParseJsUserId(env, argv[index++], context->attrs));
    IF_ERR_RETURN(ParseJsMap(env, argv[index++], context->attrs));
    IF_ERR_RETURN(CheckAddArgs(env, context->attrs));
    return napi_ok;
}
} // anonymous namespace

napi_value NapiAdd(const napi_env env, napi_callback_info info, bool asUser, bool async)
{
    auto context = std::unique_ptr<BaseContext>(new (std::nothrow)BaseContext());
    NAPI_THROW(env, context == nullptr, SEC_ASSET_OUT_OF_MEMORY, "Unable to allocate memory for Context.");

    context->parse = asUser ? ParseAttrMapAsUser : ParseAttrMap;
    context->execute = [](napi_env env, void *data) {
        if (data == nullptr) {
            LOGE("data is nullptr.");
            return;
        }
        BaseContext *context = static_cast<BaseContext *>(data);
        if (context->attrs.empty()) {
            context->result = AssetAdd(nullptr, context->attrs.size());
            return;
        }
        context->result = AssetAdd(&context->attrs[0], context->attrs.size());
    };

    context->resolve = [](napi_env env, BaseContext *context) -> napi_value {
        return CreateJsUndefined(env);
    };

    if (async) {
        return CreateAsyncWork(env, info, std::move(context), __func__);
    } else {
        return CreateSyncWork(env, info, context.get());
    }
}

napi_value NapiAdd(const napi_env env, napi_callback_info info)
{
    return NapiAdd(env, info, false, true);
}

napi_value NapiAddAsUser(const napi_env env, napi_callback_info info)
{
    return NapiAdd(env, info, true, true);
}

napi_value NapiAddSync(const napi_env env, napi_callback_info info)
{
    return NapiAdd(env, info, false, false);
}

napi_value NapiBatchAdd(const napi_env env, napi_callback_info info)
{
    auto context = std::unique_ptr<BatchOperationContext>(new (std::nothrow)BatchOperationContext());
    NAPI_THROW(env, context == nullptr, SEC_ASSET_OUT_OF_MEMORY, "Unable to allocate memory for Context.");

    context->parse = ParseAttrMapArray;
    context->execute = [](napi_env env, void *data) {
        if (data == nullptr) {
            LOGE("data is nullptr.");
            return;
        }
        BatchOperationContext *context = static_cast<BatchOperationContext *>(data);
        if (context->attrsArray.empty()) {
            context->result = SEC_ASSET_SUCCESS;
            return;
        }

        context->result = AssetBatchAdd(context->attrsArray, context->errInfoArray);
    };

    context->resolve = [](napi_env env, BaseContext *baseContext) -> napi_value {
        BatchOperationContext *context = static_cast<BatchOperationContext *>(baseContext);
        return CreateJsBatchResult(env, context->errInfoArray);
    };

    return CreateAsyncWork(env, info, std::move(context), __func__);
}

} // Asset
} // Security
} // OHOS