80 lines
2 KiB
C++
80 lines
2 KiB
C++
// Copyright (c) 2013 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "atom/common/api/atom_api_id_weak_map.h"
|
|
|
|
#include "native_mate/constructor.h"
|
|
#include "native_mate/object_template_builder.h"
|
|
|
|
#include "atom/common/node_includes.h"
|
|
|
|
namespace atom {
|
|
|
|
namespace api {
|
|
|
|
IDWeakMap::IDWeakMap() {
|
|
}
|
|
|
|
IDWeakMap::~IDWeakMap() {
|
|
}
|
|
|
|
int32_t IDWeakMap::Add(v8::Isolate* isolate, v8::Local<v8::Object> object) {
|
|
return map_.Add(isolate, object);
|
|
}
|
|
|
|
v8::Local<v8::Value> IDWeakMap::Get(v8::Isolate* isolate, int32_t key) {
|
|
v8::MaybeLocal<v8::Object> result = map_.Get(isolate, key);
|
|
if (result.IsEmpty()) {
|
|
isolate->ThrowException(v8::Exception::Error(
|
|
mate::StringToV8(isolate, "Invalid key")));
|
|
return v8::Undefined(isolate);
|
|
} else {
|
|
return result.ToLocalChecked();
|
|
}
|
|
}
|
|
|
|
bool IDWeakMap::Has(int32_t key) const {
|
|
return map_.Has(key);
|
|
}
|
|
|
|
std::vector<int32_t> IDWeakMap::Keys() const {
|
|
return map_.Keys();
|
|
}
|
|
|
|
void IDWeakMap::Remove(int32_t key) {
|
|
map_.Remove(key);
|
|
}
|
|
|
|
// static
|
|
void IDWeakMap::BuildPrototype(v8::Isolate* isolate,
|
|
v8::Local<v8::ObjectTemplate> prototype) {
|
|
mate::ObjectTemplateBuilder(isolate, prototype)
|
|
.SetMethod("add", &IDWeakMap::Add)
|
|
.SetMethod("get", &IDWeakMap::Get)
|
|
.SetMethod("has", &IDWeakMap::Has)
|
|
.SetMethod("keys", &IDWeakMap::Keys)
|
|
.SetMethod("remove", &IDWeakMap::Remove);
|
|
}
|
|
|
|
} // namespace api
|
|
|
|
} // namespace atom
|
|
|
|
|
|
namespace {
|
|
|
|
void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
|
|
v8::Local<v8::Context> context, void* priv) {
|
|
using atom::api::IDWeakMap;
|
|
v8::Isolate* isolate = context->GetIsolate();
|
|
v8::Local<v8::Function> constructor = mate::CreateConstructor<IDWeakMap>(
|
|
isolate,
|
|
"IDWeakMap",
|
|
base::Bind(&mate::NewOperatorFactory<IDWeakMap>));
|
|
exports->Set(mate::StringToSymbol(isolate, "IDWeakMap"), constructor);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
NODE_MODULE_CONTEXT_AWARE_BUILTIN(atom_common_id_weak_map, Initialize)
|