Files
js_api_module/textcoder/native_module_util.cpp
T
xliu 72c9b4470d add textdecoder and textencoder
Signed-off-by: xliu <liuxin259@huawei.com>
Change-Id: I8df05485d71536165df3e73d4f0fc12783575dc7
2021-08-21 11:51:07 +08:00

327 lines
12 KiB
C++
Executable File

/*
* 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 <vector>
#include "js_textdecoder.h"
#include "js_textencoder.h"
#include "napi/native_api.h"
#include "napi/native_node_api.h"
#include "utils/log.h"
static const int NUM_OF_DATA = 2;
// Decoder
static void SetVec(const napi_status fatSta, const napi_status bomSta, const bool fat, const bool bom,
std::vector<int> &paraVec)
{
if (paraVec.size() != 2) {
return;
}
if (fatSta == napi_ok) {
if (fat) {
paraVec[0] = 1;
} else {
paraVec[0] = 0;
}
}
if (bomSta == napi_ok) {
if (bom) {
paraVec[1] = 1;
} else {
paraVec[1] = 0;
}
}
}
static napi_value TextdecoderConstructor(napi_env env, napi_callback_info info)
{
size_t tempArgc = 0;
napi_value thisVar = nullptr;
napi_get_cb_info(env, info, &tempArgc, nullptr, &thisVar, nullptr);
size_t argc = 0;
void* data = nullptr;
char* type = nullptr;
size_t typeLen = 0;
std::vector<int> paraVec(2, 0); // 2: Specifies the size of the container to be applied for.
if (tempArgc == 1) {
argc = 1;
napi_value argv = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, &argc, &argv, nullptr, &data));
// first para
NAPI_CALL(env, napi_get_value_string_utf8(env, argv, nullptr, 0, &typeLen));
if (typeLen > 0) {
type = new char[typeLen + 1]();
}
NAPI_CALL(env, napi_get_value_string_utf8(env, argv, type, typeLen + 1, &typeLen));
} else if (tempArgc == NUM_OF_DATA) {
argc = NUM_OF_DATA;
napi_value argv[2] = { 0 };
NAPI_CALL(env, napi_get_cb_info(env, info, &argc, argv, nullptr, &data));
// first para
NAPI_CALL(env, napi_get_value_string_utf8(env, argv[0], nullptr, 0, &typeLen));
if (typeLen > 0) {
type = new char[typeLen + 1]();
}
NAPI_CALL(env, napi_get_value_string_utf8(env, argv[0], type, typeLen + 1, &typeLen));
// second para
napi_value messageKeyFatal = nullptr;
const char* messageKeyStrFatal = "fatal";
napi_value messageKeyIgnorebom = nullptr;
const char* messageKeyStrIgnorebom = "ignoreBOM";
napi_value resultFatal = nullptr;
napi_value resultIgnorebom = nullptr;
bool bResultFat = false;
bool bResultIgnbom = false;
NAPI_CALL(env, napi_create_string_utf8(env, messageKeyStrFatal, strlen(messageKeyStrFatal), &messageKeyFatal));
NAPI_CALL(env, napi_create_string_utf8(env, messageKeyStrIgnorebom, strlen(messageKeyStrIgnorebom),
&messageKeyIgnorebom));
NAPI_CALL(env, napi_get_property(env, argv[1], messageKeyFatal, &resultFatal));
NAPI_CALL(env, napi_get_property(env, argv[1], messageKeyIgnorebom, &resultIgnorebom));
napi_status naFat = napi_get_value_bool(env, resultFatal, &bResultFat);
napi_status naBom = napi_get_value_bool(env, resultIgnorebom, &bResultIgnbom);
SetVec(naFat, naBom, bResultFat, bResultIgnbom, paraVec);
}
std::string enconding = "utf-8";
if (type != nullptr) {
enconding = type;
}
if (type != nullptr) {
delete []type;
type = nullptr;
}
auto objectInfo = new TextDecoder(env, enconding, paraVec);
NAPI_CALL(env, napi_wrap(
env, thisVar, objectInfo,
[](napi_env env, void* data, void* hint) {
auto objectInfo = (TextDecoder*)data;
if (objectInfo != nullptr) {
delete objectInfo;
}
},
nullptr, nullptr));
return thisVar;
}
static napi_value TextdecoderDecode(napi_env env, napi_callback_info info)
{
size_t tempArgc = 2;
napi_value thisVar = nullptr;
napi_get_cb_info(env, info, &tempArgc, nullptr, &thisVar, nullptr);
size_t argc = 0;
void* dataPara = nullptr;
napi_typedarray_type type;
size_t length = 0;
void* data = nullptr;
napi_value arraybuffer = nullptr;
size_t byteOffset = 0;
bool iStream = true;
TextDecoder* textDecoder = nullptr;
NAPI_CALL(env, napi_unwrap(env, thisVar, (void**)&textDecoder));
napi_value valStr = nullptr;
if (tempArgc == 1) {
argc = 1;
napi_value argv = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, &argc, &argv, nullptr, &dataPara));
// first para
NAPI_CALL(env, napi_get_typedarray_info(env, argv, &type, &length, &data, &arraybuffer, &byteOffset));
valStr = textDecoder->Decode(argv, iStream);
} else if (tempArgc == NUM_OF_DATA) {
argc = NUM_OF_DATA;
napi_value argv[2] = { 0 };
NAPI_CALL(env, napi_get_cb_info(env, info, &argc, argv, nullptr, &dataPara));
// first para
NAPI_CALL(env, napi_get_typedarray_info(env, argv[0], &type, &length, &data, &arraybuffer, &byteOffset));
// second para
napi_value messageKeyStream = nullptr;
const char* messageKeyStrStream = "stream";
napi_value resultStream = nullptr;
NAPI_CALL(env, napi_create_string_utf8(env, messageKeyStrStream, strlen(messageKeyStrStream),
&messageKeyStream));
NAPI_CALL(env, napi_get_property(env, argv[1], messageKeyStream, &resultStream));
NAPI_CALL(env, napi_get_value_bool(env, resultStream, &iStream));
valStr = textDecoder->Decode(argv[0], iStream);
}
return valStr;
}
static napi_value TextdecoderGetEncoding(napi_env env, napi_callback_info info)
{
napi_value thisVar = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, nullptr, nullptr, &thisVar, nullptr));
TextDecoder* textDecoder = nullptr;
NAPI_CALL(env, napi_unwrap(env, thisVar, (void**)&textDecoder));
napi_value retVal = textDecoder->GetEncoding();
return retVal;
}
static napi_value TextdecoderGetFatal(napi_env env, napi_callback_info info)
{
napi_value thisVar = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, nullptr, nullptr, &thisVar, nullptr));
TextDecoder* textDecoder = nullptr;
NAPI_CALL(env, napi_unwrap(env, thisVar, (void**)&textDecoder));
napi_value retVal = textDecoder->GetFatal();
return retVal;
}
static napi_value TextdecoderGetIgnoreBOM(napi_env env, napi_callback_info info)
{
napi_value thisVar = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, nullptr, nullptr, &thisVar, nullptr));
TextDecoder* textDecoder = nullptr;
NAPI_CALL(env, napi_unwrap(env, thisVar, (void**)&textDecoder));
napi_value retVal = textDecoder->GetIgnoreBOM();
return retVal;
}
// Encoder
static napi_value TextEncoderConstructor(napi_env env, napi_callback_info info)
{
napi_value thisVar = nullptr;
void* data = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, nullptr, nullptr, &thisVar, &data));
auto object = new TextEncoder(env);
NAPI_CALL(env, napi_wrap(
env, thisVar, object,
[](napi_env env, void* data, void* hint) {
auto object = (TextEncoder*)data;
if (object != nullptr) {
delete object;
}
},
nullptr, nullptr));
return thisVar;
}
static napi_value GetEncoding(napi_env env, napi_callback_info info)
{
napi_value thisVar = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, nullptr, nullptr, &thisVar, nullptr));
TextEncoder* object = nullptr;
NAPI_CALL(env, napi_unwrap(env, thisVar, (void**)&object));
return object->GetEncoding();
}
static napi_value Encode(napi_env env, napi_callback_info info)
{
napi_value thisVar = nullptr;
size_t requireArgc = 1;
size_t argc = 1;
napi_value args = nullptr;
NAPI_CALL(env, napi_get_cb_info(env, info, &argc, &args, &thisVar, nullptr));
NAPI_ASSERT(env, argc >= requireArgc, "Wrong number of arguments");
napi_valuetype valuetype;
NAPI_CALL(env, napi_typeof(env, args, &valuetype));
NAPI_ASSERT(env, valuetype == napi_string, "Wrong argument type. String expected.");
TextEncoder* object = nullptr;
NAPI_CALL(env, napi_unwrap(env, thisVar, (void**)&object));
napi_value result = object->Encode(args);
return result;
}
static napi_value EncodeInto(napi_env env, napi_callback_info info)
{
napi_value thisVar = nullptr;
size_t requireArgc = 2;
size_t argc = 2;
napi_value args[2] = { nullptr };
NAPI_CALL(env, napi_get_cb_info(env, info, &argc, args, &thisVar, nullptr));
NAPI_ASSERT(env, argc >= requireArgc, "Wrong number of arguments");
napi_valuetype valuetype0;
NAPI_CALL(env, napi_typeof(env, args[0], &valuetype0));
napi_typedarray_type valuetype1;
size_t length = 0;
void* data = nullptr;
napi_value arraybuffer = nullptr;
size_t byteOffset = 0;
NAPI_CALL(env, napi_get_typedarray_info(env, args[1], &valuetype1, &length, &data, &arraybuffer, &byteOffset));
NAPI_ASSERT(env, valuetype0 == napi_string, "Wrong argument type. String expected.");
NAPI_ASSERT(env, valuetype1 == napi_uint8_array, "Wrong argument type. napi_uint8_array expected.");
TextEncoder* object = nullptr;
NAPI_CALL(env, napi_unwrap(env, thisVar, (void**)&object));
napi_value result = object->EncodeInto(args[0], args[1]);
return result;
}
static napi_value TextcoderInit(napi_env env, napi_value exports)
{
const char* textEncoderClassName = "TextEncoder";
napi_value textEncoderClass = nullptr;
static napi_property_descriptor textEncoderDesc[] = {
DECLARE_NAPI_GETTER("encoding", GetEncoding),
DECLARE_NAPI_FUNCTION("encode", Encode),
DECLARE_NAPI_FUNCTION("encodeInto", EncodeInto),
};
NAPI_CALL(env, napi_define_class(env, textEncoderClassName, strlen(textEncoderClassName), TextEncoderConstructor,
nullptr, sizeof(textEncoderDesc) / sizeof(textEncoderDesc[0]), textEncoderDesc,
&textEncoderClass));
const char* textDecoderClassName = "TextDecoder";
napi_value textDecoderClass = nullptr;
static napi_property_descriptor textdecoderDesc[] = {
DECLARE_NAPI_FUNCTION("decode", TextdecoderDecode),
DECLARE_NAPI_GETTER("encoding", TextdecoderGetEncoding),
DECLARE_NAPI_GETTER("fatal", TextdecoderGetFatal),
DECLARE_NAPI_GETTER("ignoreBOM", TextdecoderGetIgnoreBOM),
};
NAPI_CALL(env, napi_define_class(env, textDecoderClassName, strlen(textDecoderClassName), TextdecoderConstructor,
nullptr, sizeof(textdecoderDesc) / sizeof(textdecoderDesc[0]), textdecoderDesc,
&textDecoderClass));
static napi_property_descriptor desc[] = {
DECLARE_NAPI_PROPERTY("TextEncoder", textEncoderClass),
DECLARE_NAPI_PROPERTY("TextDecoder", textDecoderClass),
};
NAPI_CALL(env, napi_define_properties(env, exports, sizeof(desc) / sizeof(desc[0]), desc));
return exports;
}
// util modules
static napi_module utilModule = {
.nm_version = 1,
.nm_flags = 0,
.nm_filename = nullptr,
.nm_register_func = TextcoderInit,
.nm_modname = "util",
.nm_priv = ((void*)0),
.reserved = { 0 },
};
// util module register
extern "C" __attribute__((constructor)) void utilRegister()
{
napi_module_register(&utilModule);
}