332 lines
9.3 KiB
C++
332 lines
9.3 KiB
C++
#include "stdafx.h"
|
|
#include "cef_js_bridge.h"
|
|
#include "ipc_string_define.h"
|
|
|
|
namespace nim_comp {
|
|
|
|
CefJSBridge::CefJSBridge()
|
|
{
|
|
|
|
}
|
|
|
|
CefJSBridge::~CefJSBridge()
|
|
{
|
|
|
|
}
|
|
|
|
bool CefJSBridge::CallCppFunction(const CefString& function_name, const CefString& params, CefRefPtr<CefV8Value> callback)
|
|
{
|
|
auto it = render_callback_.find(js_callback_id_);
|
|
if (it == render_callback_.cend())
|
|
{
|
|
CefRefPtr<CefV8Context> context = CefV8Context::GetCurrentContext();
|
|
CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create(kCallCppFunctionMessage);
|
|
|
|
message->GetArgumentList()->SetString(0, function_name);
|
|
message->GetArgumentList()->SetString(1, params);
|
|
message->GetArgumentList()->SetInt(2, js_callback_id_);
|
|
|
|
render_callback_.emplace(js_callback_id_++, std::make_pair(context, callback));
|
|
|
|
// 发送消息到 browser 进程
|
|
CefRefPtr<CefBrowser> browser = context->GetBrowser();
|
|
browser->SendProcessMessage(PID_BROWSER, message);
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void CefJSBridge::RemoveCallbackFuncWithFrame(CefRefPtr<CefFrame> frame)
|
|
{
|
|
if (!render_callback_.empty())
|
|
{
|
|
for (auto it = render_callback_.begin(); it != render_callback_.end();)
|
|
{
|
|
if (it->second.first->IsSame(frame->GetV8Context()))
|
|
{
|
|
it = render_callback_.erase(it);
|
|
}
|
|
else
|
|
{
|
|
++it;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool CefJSBridge::ExecuteJSCallbackFunc(int js_callback_id, bool has_error, const CefString& json_result)
|
|
{
|
|
auto it = render_callback_.find(js_callback_id);
|
|
if (it != render_callback_.cend())
|
|
{
|
|
auto context = it->second.first;
|
|
auto callback = it->second.second;
|
|
|
|
if (context.get() && callback.get())
|
|
{
|
|
context->Enter();
|
|
|
|
CefV8ValueList arguments;
|
|
|
|
// 第一个参数标记函数执行结果是否成功
|
|
arguments.push_back(CefV8Value::CreateBool(has_error));
|
|
|
|
// 第二个参数携带函数执行后返回的数据
|
|
CefV8ValueList json_parse_args;
|
|
json_parse_args.push_back(CefV8Value::CreateString(json_result));
|
|
CefRefPtr<CefV8Value> json_parse = context->GetGlobal()->GetValue("JSON")->GetValue("parse");
|
|
CefRefPtr<CefV8Value> json_object = json_parse->ExecuteFunction(NULL, json_parse_args);
|
|
arguments.push_back(json_object);
|
|
|
|
// 执行 JS 方法
|
|
CefRefPtr<CefV8Value> retval = callback->ExecuteFunction(NULL, arguments);
|
|
if (retval.get())
|
|
{
|
|
if (retval->IsBool())
|
|
{
|
|
retval->GetBoolValue();
|
|
}
|
|
}
|
|
|
|
context->Exit();
|
|
|
|
// 从列表中移除 callback 缓存
|
|
render_callback_.erase(js_callback_id);
|
|
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool CefJSBridge::RegisterJSFunc(const CefString& function_name, CefRefPtr<CefV8Value> function, bool replace/* = false*/)
|
|
{
|
|
CefRefPtr<CefV8Context> context = CefV8Context::GetCurrentContext();
|
|
CefRefPtr<CefFrame> frame = context->GetFrame();
|
|
|
|
if (replace)
|
|
{
|
|
render_registered_function_.emplace(std::make_pair(function_name, frame->GetIdentifier()), function);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
auto it = render_registered_function_.find(std::make_pair(function_name, frame->GetIdentifier()));
|
|
if (it == render_registered_function_.cend())
|
|
{
|
|
render_registered_function_.emplace(std::make_pair(function_name, frame->GetIdentifier()), function);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void CefJSBridge::UnRegisterJSFunc(const CefString& function_name, CefRefPtr<CefFrame> frame)
|
|
{
|
|
render_registered_function_.erase(std::make_pair(function_name, frame->GetIdentifier()));
|
|
}
|
|
|
|
void CefJSBridge::UnRegisterJSFuncWithFrame(CefRefPtr<CefFrame> frame)
|
|
{
|
|
// 由于本类中每一个 render 和 browser 进程都独享一份实例，而不是单例模式
|
|
// 所以这里获取的 browser 都是全局唯一的，可以根据这个 browser 获取所有 frame 和 context
|
|
auto browser = frame->GetBrowser();
|
|
|
|
if (!render_registered_function_.empty())
|
|
{
|
|
for (auto it = render_registered_function_.begin(); it != render_registered_function_.end();)
|
|
{
|
|
auto child_frame = browser->GetFrame(it->first.second);
|
|
if (child_frame.get() && child_frame->GetV8Context()->IsSame(frame->GetV8Context()))
|
|
{
|
|
it = render_registered_function_.erase(it);
|
|
}
|
|
else
|
|
{
|
|
it++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool CefJSBridge::ExecuteJSFunc(const CefString& function_name, const CefString& json_params, CefRefPtr<CefFrame> frame, int cpp_callback_id)
|
|
{
|
|
auto it = render_registered_function_.find(std::make_pair(function_name, frame->GetIdentifier()));
|
|
if (it != render_registered_function_.cend())
|
|
{
|
|
|
|
auto context = frame->GetV8Context();
|
|
auto function = it->second;
|
|
|
|
if (context.get() && function.get())
|
|
{
|
|
context->Enter();
|
|
|
|
CefV8ValueList arguments;
|
|
|
|
// 将 C++ 传递过来的 JSON 转换成 Object
|
|
CefV8ValueList json_parse_args;
|
|
json_parse_args.push_back(CefV8Value::CreateString(json_params));
|
|
CefRefPtr<CefV8Value> json_object = context->GetGlobal()->GetValue("JSON");
|
|
CefRefPtr<CefV8Value> json_parse = json_object->GetValue("parse");
|
|
CefRefPtr<CefV8Value> json_stringify = json_object->GetValue("stringify");
|
|
CefRefPtr<CefV8Value> json_object_args = json_parse->ExecuteFunction(NULL, json_parse_args);
|
|
arguments.push_back(json_object_args);
|
|
|
|
// 执行回调函数
|
|
CefRefPtr<CefV8Value> retval = function->ExecuteFunction(NULL, arguments);
|
|
if (retval.get() && retval->IsObject())
|
|
{
|
|
// 回复调用 JS 后的返回值
|
|
CefV8ValueList json_stringify_args;
|
|
json_stringify_args.push_back(retval);
|
|
CefRefPtr<CefV8Value> json_string = json_stringify->ExecuteFunction(NULL, json_stringify_args);
|
|
CefString str = json_string->GetStringValue();
|
|
|
|
CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create(kExecuteCppCallbackMessage);
|
|
CefRefPtr<CefListValue> args = message->GetArgumentList();
|
|
args->SetString(0, json_string->GetStringValue());
|
|
args->SetInt(1, cpp_callback_id);
|
|
context->GetBrowser()->SendProcessMessage(PID_BROWSER, message);
|
|
}
|
|
|
|
context->Exit();
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool CefJSBridge::CallJSFunction(const CefString& js_function_name, const CefString& params, CefRefPtr<CefFrame> frame, CallJsFunctionCallback callback)
|
|
{
|
|
if (!frame.get())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
auto it = browser_callback_.find(cpp_callback_id_);
|
|
if (it == browser_callback_.cend())
|
|
{
|
|
browser_callback_.emplace(cpp_callback_id_, callback);
|
|
// 发送消息给 render 要求执行一个 js function
|
|
CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create(kCallJsFunctionMessage);
|
|
CefRefPtr<CefListValue> args = message->GetArgumentList();
|
|
args->SetString(0, js_function_name);
|
|
args->SetString(1, params);
|
|
args->SetInt(2, cpp_callback_id_++);
|
|
args->SetInt(3, (int)(frame->GetIdentifier()));
|
|
|
|
frame->GetBrowser()->SendProcessMessage(PID_RENDERER, message);
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool CefJSBridge::ExecuteCppCallbackFunc(int cpp_callback_id, const CefString& json_string)
|
|
{
|
|
auto it = browser_callback_.find(cpp_callback_id);
|
|
if (it != browser_callback_.cend())
|
|
{
|
|
auto callback = it->second;
|
|
if (callback)
|
|
{
|
|
nbase::ThreadManager::PostTask(kThreadUI, [=]() { callback(json_string); });
|
|
}
|
|
|
|
// 执行完成后从缓存中移除
|
|
browser_callback_.erase(cpp_callback_id);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool CefJSBridge::RegisterCppFunc(const CefString& function_name, CppFunction function, CefRefPtr<CefBrowser> browser, bool replace /*= false*/)
|
|
{
|
|
if (replace)
|
|
{
|
|
browser_registered_function_.emplace(std::make_pair(function_name, browser ? browser->GetIdentifier() : -1), function);// = ;
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
auto it = browser_registered_function_.find(std::make_pair(function_name, browser ? browser->GetIdentifier() : -1));
|
|
if (it == browser_registered_function_.cend())
|
|
{
|
|
browser_registered_function_.emplace(std::make_pair(function_name, browser ? browser->GetIdentifier() : -1), function);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void CefJSBridge::UnRegisterCppFunc(const CefString& function_name, CefRefPtr<CefBrowser> browser)
|
|
{
|
|
browser_registered_function_.erase(std::make_pair(function_name, browser ? browser->GetIdentifier() : -1));
|
|
}
|
|
|
|
bool CefJSBridge::ExecuteCppFunc(const CefString& function_name, const CefString& params, int js_callback_id, CefRefPtr<CefBrowser> browser)
|
|
{
|
|
CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create(kExecuteJsCallbackMessage);
|
|
CefRefPtr<CefListValue> args = message->GetArgumentList();
|
|
|
|
auto it = browser_registered_function_.find(std::make_pair(function_name, browser->GetIdentifier()));
|
|
if (it != browser_registered_function_.cend())
|
|
{
|
|
auto function = it->second;
|
|
nbase::ThreadManager::PostTask(kThreadUI, [=]() {
|
|
function(params, [=](bool has_error, const std::string& json_result) {
|
|
// 测试代码，需要封装到管理器中
|
|
args->SetInt(0, js_callback_id);
|
|
args->SetBool(1, has_error);
|
|
args->SetString(2, json_result);
|
|
browser->SendProcessMessage(PID_RENDERER, message);
|
|
});
|
|
});
|
|
return true;
|
|
}
|
|
|
|
it = browser_registered_function_.find(std::make_pair(function_name, -1));
|
|
if (it != browser_registered_function_.cend())
|
|
{
|
|
auto function = it->second;
|
|
nbase::ThreadManager::PostTask(kThreadUI, [=]() {
|
|
function(params, [=](bool has_error, const std::string& json_result) {
|
|
// 测试代码，需要封装到管理器中
|
|
args->SetInt(0, js_callback_id);
|
|
args->SetBool(1, has_error);
|
|
args->SetString(2, json_result);
|
|
browser->SendProcessMessage(PID_RENDERER, message);
|
|
});
|
|
});
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
args->SetInt(0, js_callback_id);
|
|
args->SetBool(1, true);
|
|
args->SetString(2, R"({"message":"Function does not exist."})");
|
|
browser->SendProcessMessage(PID_RENDERER, message);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
}
|