88 lines
2.7 KiB
C
88 lines
2.7 KiB
C
|
// Copyright (c) 2019 GitHub, Inc. All rights reserved.
|
||
|
|
// Use of this source code is governed by the MIT license that can be
|
||
|
|
// found in the LICENSE file.
|
||
|
|
|
||
|
|
#ifndef ATOM_COMMON_NATIVE_MATE_CONVERTERS_ONCE_CALLBACK_H_
|
||
|
|
#define ATOM_COMMON_NATIVE_MATE_CONVERTERS_ONCE_CALLBACK_H_
|
||
|
|
|
||
|
|
#include <utility>
|
||
|
|
|
||
|
|
#include "atom/common/native_mate_converters/callback.h"
|
||
|
|
|
||
|
|
namespace mate {
|
||
|
|
|
||
|
|
namespace internal {
|
||
|
|
|
||
|
|
// Manages the OnceCallback with ref-couting.
|
||
|
|
template <typename Sig>
|
||
|
|
class RefCountedOnceCallback
|
||
|
|
: public base::RefCounted<RefCountedOnceCallback<Sig>> {
|
||
|
|
public:
|
||
|
|
explicit RefCountedOnceCallback(base::OnceCallback<Sig> callback)
|
||
|
|
: callback_(std::move(callback)) {}
|
||
|
|
|
||
|
|
base::OnceCallback<Sig> GetCallback() { return std::move(callback_); }
|
||
|
|
|
||
|
|
private:
|
||
|
|
friend class base::RefCounted<RefCountedOnceCallback<Sig>>;
|
||
|
|
~RefCountedOnceCallback() = default;
|
||
|
|
|
||
|
|
base::OnceCallback<Sig> callback_;
|
||
|
|
};
|
||
|
|
|
||
|
|
// Invokes the OnceCallback.
|
||
|
|
template <typename Sig>
|
||
|
|
struct InvokeOnceCallback {};
|
||
|
|
|
||
|
|
template <typename... ArgTypes>
|
||
|
|
struct InvokeOnceCallback<void(ArgTypes...)> {
|
||
|
|
static void Go(
|
||
|
|
scoped_refptr<RefCountedOnceCallback<void(ArgTypes...)>> holder,
|
||
|
|
ArgTypes... args) {
|
||
|
|
base::OnceCallback<void(ArgTypes...)> callback = holder->GetCallback();
|
||
|
|
DCHECK(!callback.is_null());
|
||
|
|
std::move(callback).Run(std::move(args)...);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename ReturnType, typename... ArgTypes>
|
||
|
|
struct InvokeOnceCallback<ReturnType(ArgTypes...)> {
|
||
|
|
static ReturnType Go(
|
||
|
|
scoped_refptr<RefCountedOnceCallback<ReturnType(ArgTypes...)>> holder,
|
||
|
|
ArgTypes... args) {
|
||
|
|
base::OnceCallback<void(ArgTypes...)> callback = holder->GetCallback();
|
||
|
|
DCHECK(!callback.is_null());
|
||
|
|
return std::move(callback).Run(std::move(args)...);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace internal
|
||
|
|
|
||
|
|
template <typename Sig>
|
||
|
|
struct Converter<base::OnceCallback<Sig>> {
|
||
|
|
static v8::Local<v8::Value> ToV8(v8::Isolate* isolate,
|
||
|
|
base::OnceCallback<Sig> val) {
|
||
|
|
// Reuse the converter of base::RepeatingCallback by storing the callback
|
||
|
|
// with a RefCounted.
|
||
|
|
auto holder = base::MakeRefCounted<internal::RefCountedOnceCallback<Sig>>(
|
||
|
|
std::move(val));
|
||
|
|
return Converter<base::RepeatingCallback<Sig>>::ToV8(
|
||
|
|
isolate,
|
||
|
|
base::BindRepeating(&internal::InvokeOnceCallback<Sig>::Go, holder));
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool FromV8(v8::Isolate* isolate,
|
||
|
|
v8::Local<v8::Value> val,
|
||
|
|
base::OnceCallback<Sig>* out) {
|
||
|
|
if (!val->IsFunction())
|
||
|
|
return false;
|
||
|
|
*out = base::BindOnce(&internal::V8FunctionInvoker<Sig>::Go, isolate,
|
||
|
|
internal::SafeV8Function(isolate, val));
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace mate
|
||
|
|
|
||
|
|
#endif // ATOM_COMMON_NATIVE_MATE_CONVERTERS_ONCE_CALLBACK_H_
|