Initial Greasemonkey support

chrome/renderer/render_view.cc

 // Copyright (c) 2006-2008 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/render_view.h"
 
 #include <algorithm>
 #include <string>
 #include <vector>
 
@@ skipping 13 matching lines @@
 #include "chrome/common/gfx/color_utils.h"
 #include "chrome/common/jstemplate_builder.h"
 #include "chrome/common/l10n_util.h"
 #include "chrome/common/page_zoom.h"
 #include "chrome/common/resource_bundle.h"
 #include "chrome/common/thumbnail_score.h"
 #include "chrome/common/chrome_plugin_lib.h"
 #include "chrome/renderer/about_handler.h"
 #include "chrome/renderer/chrome_plugin_host.h"
 #include "chrome/renderer/debug_message_handler.h"
+#include "chrome/renderer/greasemonkey_slave.h"
 #include "chrome/renderer/localized_error.h"
 #include "chrome/renderer/renderer_resources.h"
 #include "chrome/renderer/visitedlink_slave.h"
 #include "chrome/renderer/webplugin_delegate_proxy.h"
 #include "chrome/views/message_box_view.h"
 #include "net/base/escape.h"
 #include "net/base/net_errors.h"
 #include "webkit/default_plugin/default_plugin_shared.h"
 #include "webkit/glue/dom_operations.h"
 #include "webkit/glue/dom_serializer.h"
@@ skipping 99 matching lines @@
     opened_by_user_gesture_(true),
     enable_dom_automation_(false),
     enable_dom_ui_bindings_(false),
     target_url_status_(TARGET_NONE),
     printed_document_width_(0),
     first_default_plugin_(NULL),
     navigation_gesture_(NavigationGestureUnknown),
     history_back_list_count_(0),
     history_forward_list_count_(0),
     disable_popup_blocking_(false),
-    has_unload_listener_(false) {
+    has_unload_listener_(false),
+    greasemonkey_enabled_(false) {
   resource_dispatcher_ = new ResourceDispatcher(this);
 #ifdef CHROME_PERSONALIZATION
   personalization_ = Personalization::CreateRendererPersonalization();
 #endif
 }
 
 RenderView::~RenderView() {
   resource_dispatcher_->ClearMessageSender();
   // Clear any back-pointers that might still be held by plugins.
   PluginDelegateList::iterator it = plugin_delegates_.begin();
@@ skipping 90 matching lines @@
   }
 
   host_window_ = parent_hwnd;
   modal_dialog_event_.Set(modal_dialog_event);
 
   CommandLine command_line;
   enable_dom_automation_ =
       command_line.HasSwitch(switches::kDomAutomationController);
   disable_popup_blocking_ =
       command_line.HasSwitch(switches::kDisablePopupBlocking);
+  greasemonkey_enabled_ =
+      command_line.HasSwitch(switches::kEnableGreasemonkey);
 
   debug_message_handler_ = new DebugMessageHandler(this);
   RenderThread::current()->AddFilter(debug_message_handler_);
 }
 
 void RenderView::OnMessageReceived(const IPC::Message& message) {
   // Let the resource dispatcher intercept resource messages first.
   if (resource_dispatcher_->OnMessageReceived(message))
     return;
   IPC_BEGIN_MESSAGE_MAP(RenderView, message)
@@ skipping 1088 matching lines @@
 
 void RenderView::DidFailLoadWithError(WebView* webview,
                                       const WebError& error,
                                       WebFrame* frame) {
 }
 
 void RenderView::DidFinishDocumentLoadForFrame(WebView* webview,
                                                WebFrame* frame) {
   // Check whether we have new encoding name.
   UpdateEncoding(frame, webview->GetMainFrameEncodingName());
+
+  // Inject any Greasemonkey scripts. Do not inject into chrome UI pages, but
+  // do inject into any other document.
+  if (greasemonkey_enabled_) {
+    const GURL &gurl = frame->GetURL();
+    if (gurl.SchemeIs("file") ||
+        gurl.SchemeIs("http") ||
+        gurl.SchemeIs("https")) {
+      RenderThread::current()->greasemonkey_slave()->InjectScripts(frame);
+    }
+  }
 }
 
 void RenderView::DidHandleOnloadEventsForFrame(WebView* webview,
                                                WebFrame* frame) {
 }
 
 void RenderView::DidChangeLocationWithinPageForFrame(WebView* webview,
                                                      WebFrame* frame,
                                                      bool is_new_navigation) {
   DidCommitLoadForFrame(webview, frame, is_new_navigation);
@@ skipping 1206 matching lines @@
 
   if (template_html.empty()) {
     NOTREACHED() << "unable to load template. ID: " << template_resource_id;
     return "";
   }
   // "t" is the id of the templates root node.
   return jstemplate_builder::GetTemplateHtml(
       template_html, &error_strings, "t");
 }