1 files changed, 87 insertions, 0 deletions
diff --git a/chromium/content/renderer/web_ui_mojo.h b/chromium/content/renderer/web_ui_mojo.h
new file mode 100644
index 00000000000..2a645429a2f
--- /dev/null
+++ b/chromium/content/renderer/web_ui_mojo.h
@@ -0,0 +1,87 @@
+// Copyright 2014 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.
+
+#ifndef CONTENT_RENDERER_WEB_UI_MOJO_H_
+#define CONTENT_RENDERER_WEB_UI_MOJO_H_
+
+#include <string>
+
+#include "content/public/renderer/render_frame_observer.h"
+#include "content/public/renderer/render_view_observer.h"
+#include "content/public/renderer/render_view_observer_tracker.h"
+#include "mojo/public/cpp/system/core.h"
+
+namespace gin {
+class PerContextData;
+}
+
+namespace content {
+
+class WebUIMojoContextState;
+
+// WebUIMojo is responsible for enabling the renderer side of mojo bindings.
+// It creates (and destroys) a WebUIMojoContextState at the appropriate times
+// and handles the necessary browser messages. WebUIMojo destroys itself when
+// the RendererView it is created with is destroyed.
+class WebUIMojo
+    : public RenderViewObserver,
+      public RenderViewObserverTracker<WebUIMojo> {
+ public:
+  explicit WebUIMojo(RenderView* render_view);
+
+  // Sets the handle to the current WebUI.
+  void SetBrowserHandle(mojo::ScopedMessagePipeHandle handle);
+
+ private:
+  class MainFrameObserver : public RenderFrameObserver {
+   public:
+    explicit MainFrameObserver(WebUIMojo* web_ui_mojo);
+    virtual ~MainFrameObserver();
+
+    // RenderFrameObserver overrides:
+    virtual void WillReleaseScriptContext(v8::Handle<v8::Context> context,
+                                          int world_id) OVERRIDE;
+    virtual void DidFinishDocumentLoad() OVERRIDE;
+
+   private:
+    WebUIMojo* web_ui_mojo_;
+
+    DISALLOW_COPY_AND_ASSIGN(MainFrameObserver);
+  };
+
+  virtual ~WebUIMojo();
+
+  void CreateContextState();
+  void DestroyContextState(v8::Handle<v8::Context> context);
+
+  // Invoked when the frame finishes loading. Invokes SetHandleOnContextState()
+  // if necessary.
+  void OnDidFinishDocumentLoad();
+
+  // Invokes SetHandle() on the WebUIMojoContextState (if there is one).
+  void SetHandleOnContextState(mojo::ScopedMessagePipeHandle handle);
+
+  WebUIMojoContextState* GetContextState();
+
+  // RenderViewObserver overrides:
+  virtual void DidClearWindowObject(blink::WebLocalFrame* frame) OVERRIDE;
+
+  MainFrameObserver main_frame_observer_;
+
+  // Set to true in DidFinishDocumentLoad(). 'main' is only executed once this
+  // happens.
+  bool did_finish_document_load_;
+
+  // If SetBrowserHandle() is invoked before the document finishes loading the
+  // MessagePipeHandle is stored here. When the document finishes loading
+  // SetHandleOnContextState() is invoked to send the handle to the
+  // WebUIMojoContextState and ultimately the page.
+  mojo::ScopedMessagePipeHandle pending_handle_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebUIMojo);
+};
+
+}  // namespace content
+
+#endif  // CONTENT_RENDERER_WEB_UI_MOJO_H_