Implement Web Restrictions in WebView.

components/supervised_user_error_page/gin_wrapper.h

+// Copyright 2015 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 COMPONENTS_SUPERVISED_USER_ERROR_PAGE_GIN_WRAPPER_H_
+#define COMPONENTS_SUPERVISED_USER_ERROR_PAGE_GIN_WRAPPER_H_
+
+#include "base/macros.h"
+#include "base/memory/weak_ptr.h"
+#include "components/web_restrictions/interfaces/web_restrictions.mojom.h"
+#include "content/public/renderer/render_frame_observer.h"
+#include "gin/wrappable.h"
+#include "url/gurl.h"
+#include "v8/include/v8.h"
+
+namespace content {
+class RenderFrame;
+}
+
+namespace supervised_user_error_page {
+
+class GinWrapper : public gin::Wrappable<GinWrapper> {
+ public:
+  static gin::WrapperInfo kWrapperInfo;
+
+  static void InstallWhenFrameReady(
+      content::RenderFrame* render_frame,
+      const std::string& url,
+      const web_restrictions::mojom::WebRestrictionsPtr&
+          web_restrictions_service);
+
+ private:
+  class Loader : public content::RenderFrameObserver {
+   public:
+    Loader(content::RenderFrame* render_frame,
+           const std::string& url,
+           const web_restrictions::mojom::WebRestrictionsPtr&
+               web_restrictions_service);
+    ~Loader() override = default;
+
+    void DidClearWindowObject() override;
+    void OnDestruct() override;
+
+   private:
+    void InstallGinWrapper();
+
+    std::string url_;
+    const web_restrictions::mojom::WebRestrictionsPtr&
+        web_restrictions_service_;
+    DISALLOW_COPY_AND_ASSIGN(Loader);
+  };
+
+  GinWrapper(content::RenderFrame* render_frame,
+             const std::string& url,
+             const web_restrictions::mojom::WebRestrictionsPtr&
+                 web_restrictions_service);
+  ~GinWrapper() override;
+
+  // Request permission to allow visiting the currently blocked site.
+  bool RequestPermission(v8::Local<v8::Function> setRequestStatusCallback);
+
+  void OnAccessRequestAdded(bool success);
+
+  // gin::WrappableBase
+  gin::ObjectTemplateBuilder GetObjectTemplateBuilder(
+      v8::Isolate* isolate) override;
+
+  const std::string url_;
+  const web_restrictions::mojom::WebRestrictionsPtr& web_restrictions_service_;
+  v8::Global<v8::Function> setRequestStatusCallback_;
+  base::WeakPtrFactory<GinWrapper> weak_ptr_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(GinWrapper);
+};
+
+}  // namespace supervised_user_error_page.
+
+#endif  // COMPONENTS_SUPERVISED_USER_ERROR_PAGE_GIN_WRAPPER_H_