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.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:
+  struct UserScriptRenderInfo;
+  using UserScriptRenderInfoMap = std::map<int, UserScriptRenderInfo>;
+
+  // UserScriptLoader:
+  void AddScripts(const std::set<extensions::UserScript>& scripts,
+                  int render_process_id,
+                  int render_view_id) override;
+  void LoadScripts(const std::set<HostID>& changed_hosts,
+                   const std::set<int>& added_script_ids) override;
+
+  // Called when every single URL fetch is done. It also traces whether all the
+  // URL fetches are done. If yes, will call OnWebUIURLFetchComplete().
+  void OnSingleWebUIURLFetchComplete(extensions::UserScript::File* script_file,
+                                     bool success,
+                                     const std::string& data);
+
+  // Called when the loads of the user scripts are done.
+  void OnWebUIURLFetchComplete();
+
+  // Caches the render info of script from WebUI when AddScripts is called.
+  // When starting to load the script, we look up this map to retrieve the
+  // render info. It is used for the script from WebUI only, since the fetch
+  // of script content requires the info of associated render.
+  UserScriptRenderInfoMap script_render_info_map_;
+
+  // The number of complete fetchs.
+  int complete_fetchers_;
+  int total_fetchers_;
+
+  ScopedVector<WebUIURLFetcher> fetchers_;
+
+  base::WeakPtrFactory<WebUIUserScriptLoader> weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebUIUserScriptLoader);
+};
+
+#endif  // EXTENSIONS_BROWSER_WEB_UI_USER_SCRIPT_LOADER_H_