Implement <webview>.addContentScript/removeContentScript API [2]

extensions/browser/web_ui_user_script_loader.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 EXTENSIONS_BROWSER_WEB_UI_USER_SCRIPT_LOADER_H_
+#define EXTENSIONS_BROWSER_WEB_UI_USER_SCRIPT_LOADER_H_
+
+#include "base/memory/scoped_vector.h"
+#include "extensions/browser/guest_view/web_view/web_ui/web_ui_url_fetcher_group.h"
+#include "extensions/browser/user_script_loader.h"
+
+namespace content {
+class BrowserContext;
+}
+
+// UserScriptLoader for WebUI.
+class WebUIUserScriptLoader : public extensions::UserScriptLoader {
+ public:
+  WebUIUserScriptLoader(content::BrowserContext* browser_context,
+                        const HostID& host_id);
+  ~WebUIUserScriptLoader() override;
+
+ private:
+  // UserScriptLoader:
+  void LoadScripts(const std::set<HostID>& changed_hosts,
+                   const std::set<int>& added_script_ids) override;
+
+  // Called when the loads of the user scripts are done.
+  void OnWebUIURLFetchComplete(
+      scoped_ptr<extensions::UserScriptList> user_scripts);
+
+  std::set<WebUIURLFetcherGroup*> fetcher_groups_;
+  base::WeakPtrFactory<WebUIUserScriptLoader> weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebUIUserScriptLoader);
+};
+
+#endif  // EXTENSIONS_BROWSER_WEB_UI_USER_SCRIPT_LOADER_H_