0fe6767d6b* refactor: convert Menu and globalShortcut to gin * refactor: convert api::Cookies to gin * refactor: convert View and WebContentsView to gin * refactor: convert WebContents related classes to gin * refactor: convert powerMonitor to gin * refactor: prepare for header change * refactor: remove last uses of mate::EventEmitter * refactor: remove mate::EventEmitter * refactor: move trackable_object to gin_helper * fix: custom converter should not use Handle * fix: no more need to check if icon is empty It was a bug that the Handle<NativeImage> can be non-empty when the image file does not exist. The bug was caused by the converter code writing out the image even when the convertion fails. The bug was work-arounded by adding an additional check, but since the original bug had been fixed, the additional check is no longer needed. * fix: should always set frameId even when callback is null * fix: do not mix gin/mate handles for NativeImage
135 lines
3.7 KiB
C++
135 lines
3.7 KiB
C++
// Copyright (c) 2015 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#ifndef SHELL_COMMON_GIN_HELPER_TRACKABLE_OBJECT_H_
|
|
#define SHELL_COMMON_GIN_HELPER_TRACKABLE_OBJECT_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "base/bind.h"
|
|
#include "base/memory/weak_ptr.h"
|
|
#include "shell/common/gin_helper/event_emitter.h"
|
|
#include "shell/common/key_weak_map.h"
|
|
|
|
namespace base {
|
|
class SupportsUserData;
|
|
}
|
|
|
|
namespace gin_helper {
|
|
|
|
// Users should use TrackableObject instead.
|
|
class TrackableObjectBase {
|
|
public:
|
|
TrackableObjectBase();
|
|
|
|
// The ID in weak map.
|
|
int32_t weak_map_id() const { return weak_map_id_; }
|
|
|
|
// Wrap TrackableObject into a class that SupportsUserData.
|
|
void AttachAsUserData(base::SupportsUserData* wrapped);
|
|
|
|
// Get the weak_map_id from SupportsUserData.
|
|
static int32_t GetIDFromWrappedClass(base::SupportsUserData* wrapped);
|
|
|
|
protected:
|
|
virtual ~TrackableObjectBase();
|
|
|
|
// Returns a closure that can destroy the native class.
|
|
base::OnceClosure GetDestroyClosure();
|
|
|
|
int32_t weak_map_id_ = 0;
|
|
|
|
private:
|
|
void Destroy();
|
|
|
|
base::WeakPtrFactory<TrackableObjectBase> weak_factory_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(TrackableObjectBase);
|
|
};
|
|
|
|
// All instances of TrackableObject will be kept in a weak map and can be got
|
|
// from its ID.
|
|
template <typename T>
|
|
class TrackableObject : public TrackableObjectBase, public EventEmitter<T> {
|
|
public:
|
|
// Mark the JS object as destroyed.
|
|
void MarkDestroyed() {
|
|
v8::Local<v8::Object> wrapper = mate::Wrappable<T>::GetWrapper();
|
|
if (!wrapper.IsEmpty()) {
|
|
wrapper->SetAlignedPointerInInternalField(0, nullptr);
|
|
}
|
|
}
|
|
|
|
bool IsDestroyed() {
|
|
v8::Local<v8::Object> wrapper = mate::Wrappable<T>::GetWrapper();
|
|
return wrapper->InternalFieldCount() == 0 ||
|
|
wrapper->GetAlignedPointerFromInternalField(0) == nullptr;
|
|
}
|
|
|
|
// Finds out the TrackableObject from its ID in weak map.
|
|
static T* FromWeakMapID(v8::Isolate* isolate, int32_t id) {
|
|
if (!weak_map_)
|
|
return nullptr;
|
|
|
|
v8::MaybeLocal<v8::Object> object = weak_map_->Get(isolate, id);
|
|
if (object.IsEmpty())
|
|
return nullptr;
|
|
|
|
T* self = nullptr;
|
|
gin::ConvertFromV8(isolate, object.ToLocalChecked(), &self);
|
|
return self;
|
|
}
|
|
|
|
// Finds out the TrackableObject from the class it wraps.
|
|
static T* FromWrappedClass(v8::Isolate* isolate,
|
|
base::SupportsUserData* wrapped) {
|
|
int32_t id = GetIDFromWrappedClass(wrapped);
|
|
if (!id)
|
|
return nullptr;
|
|
return FromWeakMapID(isolate, id);
|
|
}
|
|
|
|
// Returns all objects in this class's weak map.
|
|
static std::vector<v8::Local<v8::Object>> GetAll(v8::Isolate* isolate) {
|
|
if (weak_map_)
|
|
return weak_map_->Values(isolate);
|
|
else
|
|
return std::vector<v8::Local<v8::Object>>();
|
|
}
|
|
|
|
// Removes this instance from the weak map.
|
|
void RemoveFromWeakMap() {
|
|
if (weak_map_ && weak_map_->Has(weak_map_id()))
|
|
weak_map_->Remove(weak_map_id());
|
|
}
|
|
|
|
protected:
|
|
TrackableObject() { weak_map_id_ = ++next_id_; }
|
|
|
|
~TrackableObject() override { RemoveFromWeakMap(); }
|
|
|
|
void InitWith(v8::Isolate* isolate, v8::Local<v8::Object> wrapper) override {
|
|
mate::WrappableBase::InitWith(isolate, wrapper);
|
|
if (!weak_map_) {
|
|
weak_map_ = new electron::KeyWeakMap<int32_t>;
|
|
}
|
|
weak_map_->Set(isolate, weak_map_id_, wrapper);
|
|
}
|
|
|
|
private:
|
|
static int32_t next_id_;
|
|
static electron::KeyWeakMap<int32_t>* weak_map_; // leaked on purpose
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(TrackableObject);
|
|
};
|
|
|
|
template <typename T>
|
|
int32_t TrackableObject<T>::next_id_ = 0;
|
|
|
|
template <typename T>
|
|
electron::KeyWeakMap<int32_t>* TrackableObject<T>::weak_map_ = nullptr;
|
|
|
|
} // namespace gin_helper
|
|
|
|
#endif // SHELL_COMMON_GIN_HELPER_TRACKABLE_OBJECT_H_
|