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

extensions/browser/web_ui_user_script_loader.h

Index: extensions/browser/web_ui_user_script_loader.h
diff --git a/extensions/browser/web_ui_user_script_loader.h b/extensions/browser/web_ui_user_script_loader.h
new file mode 100644
index 0000000000000000000000000000000000000000..7a13a43fa87fd8f0530d65829650a8559bf478be
--- /dev/null
+++ b/extensions/browser/web_ui_user_script_loader.h
@@ -0,0 +1,69 @@
+// 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"

[Devlin, 2015/04/23 16:34:38]

Forward declaration.

[Xi Han, 2015/04/23 17:50:40]

Done.

+#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(scoped_ptr<extensions::UserScriptList> user_scripts,
+                   const std::set<HostID>& changed_hosts,
+                   const std::set<int>& added_script_ids,
+                   LoadScriptsCallback callback) override;
+
+  // Called when every single URL fetch is done. It also traces whether all the

[Devlin, 2015/04/23 16:34:38]

"when every single URL fetch is done" is ambiguous (it could also mean when all
fetches are done).  Let's say "Called at the end of each fetch, tracking whether
all fetches are done."  And you can omit the part about it calling
OnWebUIURLFetchComplete (that's too much info for the comment).

[Xi Han, 2015/04/23 17:50:40]

Sounds good:)

+  // 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();
+
+  // Creates WebUiURLFetchers for the given |script_files|.
+  void CreateWebUIURLFetchers(extensions::UserScript::FileList* script_files,
+                              int render_process_id,
+                              int render_view_id);
+
+  // 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.
+  size_t complete_fetchers_;
+
+  // Caches |user_scripts_| from UserScriptLoader when loading.
+  scoped_ptr<extensions::UserScriptList> user_scripts_cache_;
+
+  LoadScriptsCallback callback_;

[Devlin, 2015/04/23 16:34:38]

Let's call this scripts_loaded_callback_.

[Xi Han, 2015/04/23 17:50:40]

Done.

+
+  ScopedVector<WebUIURLFetcher> fetchers_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebUIUserScriptLoader);
+};
+
+#endif  // EXTENSIONS_BROWSER_WEB_UI_USER_SCRIPT_LOADER_H_