Index: chrome/renderer/automation/automation_renderer_helper.cc |
diff --git a/chrome/renderer/automation/automation_renderer_helper.cc b/chrome/renderer/automation/automation_renderer_helper.cc |
deleted file mode 100644 |
index e856f63d35d678768e8f7e6c92a4524eb46f4e2e..0000000000000000000000000000000000000000 |
--- a/chrome/renderer/automation/automation_renderer_helper.cc |
+++ /dev/null |
@@ -1,288 +0,0 @@ |
-// Copyright (c) 2012 The Chromium 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 "chrome/renderer/automation/automation_renderer_helper.h" |
- |
-#include <algorithm> |
- |
-#include "base/basictypes.h" |
-#include "base/json/json_writer.h" |
-#include "base/stringprintf.h" |
-#include "base/strings/string_split.h" |
-#include "base/utf_string_conversions.h" |
-#include "base/values.h" |
-#include "chrome/common/automation_events.h" |
-#include "chrome/common/automation_messages.h" |
-#include "content/public/renderer/render_view.h" |
-#include "content/public/renderer/v8_value_converter.h" |
-#include "ipc/ipc_channel.h" |
-#include "ipc/ipc_message.h" |
-#include "skia/ext/platform_canvas.h" |
-#include "third_party/WebKit/Source/Platform/chromium/public/WebSize.h" |
-#include "third_party/WebKit/Source/Platform/chromium/public/WebURL.h" |
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h" |
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebInputEvent.h" |
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebScriptSource.h" |
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h" |
-#include "ui/gfx/codec/png_codec.h" |
-#include "ui/gfx/point.h" |
-#include "ui/gfx/rect.h" |
-#include "v8/include/v8.h" |
-#include "webkit/glue/webkit_glue.h" |
- |
-#if !defined(NO_TCMALLOC) && (defined(OS_LINUX) || defined(OS_CHROMEOS)) |
-#include "third_party/tcmalloc/chromium/src/gperftools/heap-profiler.h" |
-#endif // !defined(NO_TCMALLOC) && (defined(OS_LINUX) || defined(OS_CHROMEOS)) |
- |
-using WebKit::WebFrame; |
-using WebKit::WebSize; |
-using WebKit::WebURL; |
-using WebKit::WebView; |
- |
-AutomationRendererHelper::AutomationRendererHelper( |
- content::RenderView* render_view) |
- : content::RenderViewObserver(render_view) { |
-} |
- |
-AutomationRendererHelper::~AutomationRendererHelper() { } |
- |
-bool AutomationRendererHelper::SnapshotEntirePage( |
- WebView* view, |
- std::vector<unsigned char>* png_data, |
- std::string* error_msg) { |
- WebFrame* frame = view->mainFrame(); |
- WebSize old_size = view->size(); |
- WebSize new_size = frame->contentsSize(); |
- WebSize old_scroll = frame->scrollOffset(); |
- bool fixed_layout_enabled = view->isFixedLayoutModeEnabled(); |
- WebSize fixed_size = view->fixedLayoutSize(); |
- |
- frame->setCanHaveScrollbars(false); |
- view->setFixedLayoutSize(old_size); |
- view->enableFixedLayoutMode(true); |
- view->resize(new_size); |
- view->layout(); |
- frame->setScrollOffset(frame->minimumScrollOffset()); |
- |
- skia::RefPtr<SkCanvas> canvas = skia::AdoptRef( |
- skia::CreatePlatformCanvas(new_size.width, new_size.height, true)); |
- |
- view->paint(webkit_glue::ToWebCanvas(canvas.get()), |
- gfx::Rect(0, 0, new_size.width, new_size.height)); |
- |
- frame->setCanHaveScrollbars(true); |
- view->setFixedLayoutSize(fixed_size); |
- view->enableFixedLayoutMode(fixed_layout_enabled); |
- view->resize(old_size); |
- view->layout(); |
- frame->setScrollOffset(old_scroll); |
- |
- const SkBitmap& bmp = skia::GetTopDevice(*canvas)->accessBitmap(false); |
- SkAutoLockPixels lock_pixels(bmp); |
- // EncodeBGRA uses FORMAT_SkBitmap, which doesn't work on windows for some |
- // cases dealing with transparency. See crbug.com/96317. Use FORMAT_BGRA. |
- bool encode_success = gfx::PNGCodec::Encode( |
- reinterpret_cast<unsigned char*>(bmp.getPixels()), |
- gfx::PNGCodec::FORMAT_BGRA, |
- gfx::Size(bmp.width(), bmp.height()), |
- bmp.rowBytes(), |
- true, // discard_transparency |
- std::vector<gfx::PNGCodec::Comment>(), |
- png_data); |
- if (!encode_success) |
- *error_msg = "failed to encode image as png"; |
- return encode_success; |
-} |
- |
-void AutomationRendererHelper::OnSnapshotEntirePage() { |
- std::vector<unsigned char> png_data; |
- std::string error_msg; |
- bool success = false; |
- if (render_view()->GetWebView()) { |
- success = SnapshotEntirePage( |
- render_view()->GetWebView(), &png_data, &error_msg); |
- } else { |
- error_msg = "cannot snapshot page because webview is null"; |
- } |
- |
- // Check that the image is not too large, allowing a 1kb buffer for other |
- // message data. |
- if (success && png_data.size() > IPC::Channel::kMaximumMessageSize - 1024) { |
- png_data.clear(); |
- success = false; |
- error_msg = "image is too large to be transferred over ipc"; |
- } |
- Send(new AutomationMsg_SnapshotEntirePageACK( |
- routing_id(), success, png_data, error_msg)); |
-} |
- |
-namespace { |
- |
-scoped_ptr<base::Value> EvaluateScriptInFrame(WebFrame* web_frame, |
- const std::string& script) { |
- v8::HandleScope handle_scope; |
- v8::Local<v8::Value> result = v8::Local<v8::Value>::New( |
- web_frame->executeScriptAndReturnValue( |
- WebKit::WebScriptSource(UTF8ToUTF16(script)))); |
- if (!result.IsEmpty()) { |
- v8::Local<v8::Context> context = web_frame->mainWorldScriptContext(); |
- v8::Context::Scope context_scope(context); |
- scoped_ptr<content::V8ValueConverter> converter( |
- content::V8ValueConverter::create()); |
- return scoped_ptr<base::Value>(converter->FromV8Value(result, context)); |
- } else { |
- return scoped_ptr<base::Value>(base::Value::CreateNullValue()); |
- } |
-} |
- |
-WebFrame* FrameFromXPath(WebView* view, const string16& frame_xpath) { |
- WebFrame* frame = view->mainFrame(); |
- if (frame_xpath.empty()) |
- return frame; |
- |
- std::vector<string16> xpaths; |
- base::SplitString(frame_xpath, '\n', &xpaths); |
- for (std::vector<string16>::const_iterator i = xpaths.begin(); |
- frame && i != xpaths.end(); ++i) { |
- frame = frame->findChildByExpression(*i); |
- } |
- return frame; |
-} |
- |
-bool EvaluateScriptChainHelper( |
- WebView* web_view, |
- const std::string& script, |
- const std::string& frame_xpath, |
- scoped_ptr<base::DictionaryValue>* result, |
- std::string* error_msg) { |
- WebFrame* web_frame = FrameFromXPath(web_view, UTF8ToUTF16(frame_xpath)); |
- if (!web_frame) { |
- *error_msg = "Failed to locate frame by xpath: " + frame_xpath; |
- return false; |
- } |
- scoped_ptr<base::Value> script_value = |
- EvaluateScriptInFrame(web_frame, script); |
- base::DictionaryValue* dict; |
- if (!script_value.get() || !script_value->GetAsDictionary(&dict)) { |
- *error_msg = "Script did not return an object"; |
- return false; |
- } |
- base::Value* error_value; |
- if (dict->Get("error", &error_value)) { |
- base::JSONWriter::Write(error_value, error_msg); |
- return false; |
- } |
- result->reset(static_cast<base::DictionaryValue*>(script_value.release())); |
- return true; |
-} |
- |
-} // namespace |
- |
-bool AutomationRendererHelper::EvaluateScriptChain( |
- const std::vector<ScriptEvaluationRequest>& script_chain, |
- scoped_ptr<base::DictionaryValue>* result, |
- std::string* error_msg) { |
- CHECK(!script_chain.empty()); |
- WebView* web_view = render_view()->GetWebView(); |
- scoped_ptr<base::DictionaryValue> temp_result; |
- for (size_t i = 0; i < script_chain.size(); ++i) { |
- std::string args_utf8 = "null"; |
- if (temp_result.get()) |
- base::JSONWriter::Write(temp_result.get(), &args_utf8); |
- std::string wrapper_script = base::StringPrintf( |
- "(function(){return %s\n}).apply(null, [%s])", |
- script_chain[i].script.c_str(), args_utf8.c_str()); |
- if (!EvaluateScriptChainHelper(web_view, wrapper_script, |
- script_chain[i].frame_xpath, |
- &temp_result, error_msg)) { |
- return false; |
- } |
- } |
- std::string result_str; |
- base::JSONWriter::Write(temp_result.get(), &result_str); |
- result->reset(temp_result.release()); |
- return true; |
-} |
- |
-bool AutomationRendererHelper::ProcessMouseEvent( |
- const AutomationMouseEvent& event, |
- std::string* error_msg) { |
- WebView* web_view = render_view()->GetWebView(); |
- if (!web_view) { |
- *error_msg = "Failed to process mouse event because webview does not exist"; |
- return false; |
- } |
- WebKit::WebMouseEvent mouse_event = event.mouse_event; |
- if (!event.location_script_chain.empty()) { |
- scoped_ptr<base::DictionaryValue> result; |
- if (!EvaluateScriptChain(event.location_script_chain, &result, error_msg)) |
- return false; |
- int x, y; |
- if (!result->GetInteger("x", &x) || |
- !result->GetInteger("y", &y)) { |
- *error_msg = "Script did not return an (x,y) location"; |
- return false; |
- } |
- mouse_event.x = x; |
- mouse_event.y = y; |
- } |
- Send(new AutomationMsg_WillProcessMouseEventAt( |
- routing_id(), |
- gfx::Point(mouse_event.x, mouse_event.y))); |
- web_view->handleInputEvent(mouse_event); |
- return true; |
-} |
- |
-#if !defined(NO_TCMALLOC) && (defined(OS_LINUX) || defined(OS_CHROMEOS)) |
-void AutomationRendererHelper::OnHeapProfilerDump(const std::string& reason) { |
- if (!::IsHeapProfilerRunning()) { |
- return; |
- } |
- ::HeapProfilerDump(reason.c_str()); |
-} |
-#endif // !defined(NO_TCMALLOC) && (defined(OS_LINUX) || defined(OS_CHROMEOS)) |
- |
-bool AutomationRendererHelper::OnMessageReceived(const IPC::Message& message) { |
- bool handled = true; |
- bool deserialize_success = true; |
- IPC_BEGIN_MESSAGE_MAP_EX(AutomationRendererHelper, message, |
- deserialize_success) |
- IPC_MESSAGE_HANDLER(AutomationMsg_SnapshotEntirePage, OnSnapshotEntirePage) |
-#if !defined(NO_TCMALLOC) && (defined(OS_LINUX) || defined(OS_CHROMEOS)) |
- IPC_MESSAGE_HANDLER(AutomationMsg_HeapProfilerDump, OnHeapProfilerDump) |
-#endif // !defined(NO_TCMALLOC) && (defined(OS_LINUX) || defined(OS_CHROMEOS)) |
- IPC_MESSAGE_HANDLER(AutomationMsg_ProcessMouseEvent, OnProcessMouseEvent) |
- IPC_MESSAGE_UNHANDLED(handled = false) |
- IPC_END_MESSAGE_MAP_EX() |
- if (!deserialize_success) { |
- LOG(ERROR) << "Failed to deserialize an IPC message"; |
- } |
- return handled; |
-} |
- |
-void AutomationRendererHelper::WillPerformClientRedirect( |
- WebFrame* frame, const WebURL& from, const WebURL& to, double interval, |
- double fire_time) { |
- Send(new AutomationMsg_WillPerformClientRedirect( |
- routing_id(), frame->identifier(), interval)); |
-} |
- |
-void AutomationRendererHelper::DidCancelClientRedirect(WebFrame* frame) { |
- Send(new AutomationMsg_DidCompleteOrCancelClientRedirect( |
- routing_id(), frame->identifier())); |
-} |
- |
-void AutomationRendererHelper::DidCompleteClientRedirect( |
- WebFrame* frame, const WebURL& from) { |
- Send(new AutomationMsg_DidCompleteOrCancelClientRedirect( |
- routing_id(), frame->identifier())); |
-} |
- |
-void AutomationRendererHelper::OnProcessMouseEvent( |
- const AutomationMouseEvent& event) { |
- std::string error_msg; |
- bool success = ProcessMouseEvent(event, &error_msg); |
- Send(new AutomationMsg_ProcessMouseEventACK( |
- routing_id(), success, error_msg)); |
-} |