127 lines
4.4 KiB
C++
127 lines
4.4 KiB
C++
|
// Copyright (c) 2011 The Chromium Embedded Framework Authors. All rights
|
||
|
|
// reserved. Use of this source code is governed by a BSD-style license that
|
||
|
|
// can be found in the LICENSE file.
|
||
|
|
|
||
|
|
#include "cefclient/extension_test.h"
|
||
|
|
#include <string>
|
||
|
|
#include "include/cef_browser.h"
|
||
|
|
#include "include/cef_frame.h"
|
||
|
|
#include "include/cef_stream.h"
|
||
|
|
#include "include/cef_v8.h"
|
||
|
|
#include "cefclient/resource_util.h"
|
||
|
|
|
||
|
|
// Implementation of the V8 handler class for the "cef.test" extension.
|
||
|
|
class ClientV8ExtensionHandler : public CefV8Handler {
|
||
|
|
public:
|
||
|
|
ClientV8ExtensionHandler() : test_param_("An initial string value.") {}
|
||
|
|
virtual ~ClientV8ExtensionHandler() {}
|
||
|
|
|
||
|
|
// Execute with the specified argument list and return value. Return true if
|
||
|
|
// the method was handled.
|
||
|
|
virtual bool Execute(const CefString& name,
|
||
|
|
CefRefPtr<CefV8Value> object,
|
||
|
|
const CefV8ValueList& arguments,
|
||
|
|
CefRefPtr<CefV8Value>& retval,
|
||
|
|
CefString& exception) {
|
||
|
|
if (name == "Dummy") {
|
||
|
|
// Used for performance testing.
|
||
|
|
return true;
|
||
|
|
} else if (name == "SetTestParam") {
|
||
|
|
// Handle the SetTestParam native function by saving the string argument
|
||
|
|
// into the local member.
|
||
|
|
if (arguments.size() != 1 || !arguments[0]->IsString())
|
||
|
|
return false;
|
||
|
|
|
||
|
|
test_param_ = arguments[0]->GetStringValue();
|
||
|
|
return true;
|
||
|
|
} else if (name == "GetTestParam") {
|
||
|
|
// Handle the GetTestParam native function by returning the local member
|
||
|
|
// value.
|
||
|
|
retval = CefV8Value::CreateString(test_param_);
|
||
|
|
return true;
|
||
|
|
} else if (name == "GetTestObject") {
|
||
|
|
// Handle the GetTestObject native function by creating and returning a
|
||
|
|
// new V8 object.
|
||
|
|
retval = CefV8Value::CreateObject(NULL);
|
||
|
|
// Add a string parameter to the new V8 object.
|
||
|
|
retval->SetValue("param", CefV8Value::CreateString(
|
||
|
|
"Retrieving a parameter on a native object succeeded."),
|
||
|
|
V8_PROPERTY_ATTRIBUTE_NONE);
|
||
|
|
// Add a function to the new V8 object.
|
||
|
|
retval->SetValue("GetMessage",
|
||
|
|
CefV8Value::CreateFunction("GetMessage", this),
|
||
|
|
V8_PROPERTY_ATTRIBUTE_NONE);
|
||
|
|
return true;
|
||
|
|
} else if (name == "GetMessage") {
|
||
|
|
// Handle the GetMessage object function by returning a string.
|
||
|
|
retval = CefV8Value::CreateString(
|
||
|
|
"Calling a function on a native object succeeded.");
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
CefString test_param_;
|
||
|
|
|
||
|
|
IMPLEMENT_REFCOUNTING(ClientV8ExtensionHandler);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
void InitExtensionTest() {
|
||
|
|
// Register a V8 extension with the below JavaScript code that calls native
|
||
|
|
// methods implemented in ClientV8ExtensionHandler.
|
||
|
|
std::string code = "var cef;"
|
||
|
|
"if (!cef)"
|
||
|
|
" cef = {};"
|
||
|
|
"if (!cef.test)"
|
||
|
|
" cef.test = {};"
|
||
|
|
"(function() {"
|
||
|
|
" cef.test.__defineGetter__('test_param', function() {"
|
||
|
|
" native function GetTestParam();"
|
||
|
|
" return GetTestParam();"
|
||
|
|
" });"
|
||
|
|
" cef.test.__defineSetter__('test_param', function(b) {"
|
||
|
|
" native function SetTestParam();"
|
||
|
|
" if (b) SetTestParam(b);"
|
||
|
|
" });"
|
||
|
|
" cef.test.test_object = function() {"
|
||
|
|
" native function GetTestObject();"
|
||
|
|
" return GetTestObject();"
|
||
|
|
" };"
|
||
|
|
" cef.test.dummy = function() {"
|
||
|
|
" native function Dummy();"
|
||
|
|
" return Dummy();"
|
||
|
|
" };"
|
||
|
|
"})();";
|
||
|
|
CefRegisterExtension("v8/test", code, new ClientV8ExtensionHandler());
|
||
|
|
}
|
||
|
|
|
||
|
|
void RunExtensionTest(CefRefPtr<CefBrowser> browser) {
|
||
|
|
std::string html =
|
||
|
|
"<html><body>ClientV8ExtensionHandler says:<br><pre>"
|
||
|
|
"<script language=\"JavaScript\">"
|
||
|
|
"cef.test.test_param ="
|
||
|
|
" 'Assign and retrieve a value succeeded the first time.';"
|
||
|
|
"document.writeln(cef.test.test_param);"
|
||
|
|
"cef.test.test_param ="
|
||
|
|
" 'Assign and retrieve a value succeeded the second time.';"
|
||
|
|
"document.writeln(cef.test.test_param);"
|
||
|
|
"var obj = cef.test.test_object();"
|
||
|
|
"document.writeln(obj.param);"
|
||
|
|
"document.writeln(obj.GetMessage());"
|
||
|
|
"</script>"
|
||
|
|
"</pre></body></html>";
|
||
|
|
browser->GetMainFrame()->LoadString(html, "about:blank");
|
||
|
|
}
|
||
|
|
|
||
|
|
void RunExtensionPerfTest(CefRefPtr<CefBrowser> browser) {
|
||
|
|
CefRefPtr<CefStreamReader> resourceStream;
|
||
|
|
#if defined(OS_WIN)
|
||
|
|
resourceStream = GetBinaryResourceReader(IDS_EXTENSIONPERF);
|
||
|
|
#elif defined(OS_MACOSX)
|
||
|
|
resourceStream = GetBinaryResourceReader("extensionperf.html");
|
||
|
|
#endif
|
||
|
|
browser->GetMainFrame()->LoadStream(resourceStream, "about:blank");
|
||
|
|
}
|