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

extensions/browser/extension_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_EXTENSION_USER_SCRIPT_LOADER_H_
 #define EXTENSIONS_BROWSER_EXTENSION_USER_SCRIPT_LOADER_H_
 
 #include "base/memory/shared_memory.h"
 #include "extensions/browser/extension_registry_observer.h"
 #include "extensions/browser/user_script_loader.h"
 #include "extensions/common/extension.h"
 
 namespace content {
 class BrowserContext;
 }
 
 namespace extensions {
 
 class ExtensionRegistry;
 
 // UserScriptLoader for extensions.
 class ExtensionUserScriptLoader : public UserScriptLoader,
                                   public ExtensionRegistryObserver {
  public:
+  using PathAndDefaultLocale = std::pair<base::FilePath, std::string>;

[Devlin, 2015/04/24 17:50:59]

Why do these need to be public?  And can't SubstitutionMap be in the .cc file in
the anonymous namespace?

[Xi Han, 2015/04/24 18:28:54]

There are several functions in the anonymous namespace of .cc file use
HostsInfo, so we have to declare HostsInfo and PathAndDefaultLocale as public.
Yes, we should definitely move SubstitutionMap to the .cc file.

+  using HostsInfo = std::map<HostID, PathAndDefaultLocale>;
+
+  using SubstitutionMap = std::map<std::string, std::string>;
+
   // The listen_for_extension_system_loaded is only set true when initilizing
   // the Extension System, e.g, when constructs SharedUserScriptMaster in
   // ExtensionSystemImpl.
   ExtensionUserScriptLoader(content::BrowserContext* browser_context,
                             const HostID& host_id,
                             bool listen_for_extension_system_loaded);
   ~ExtensionUserScriptLoader() override;
 
+  // A wrapper around the method to load user scripts, which is normally run on
+  // the file thread. Exposed only for tests.
+  void LoadScriptsForTest(UserScriptList* user_scripts);
+
  private:
   // UserScriptLoader:
-  void UpdateHostsInfo(const std::set<HostID>& changed_hosts) override;
-  LoadUserScriptsContentFunction GetLoadUserScriptsFunction() override;
+  void LoadScripts(scoped_ptr<UserScriptList> user_scripts,
+                   const std::set<HostID>& changed_hosts,
+                   const std::set<int>& added_script_ids,
+                   LoadScriptsCallback callback) override;
+
+  // Updates |hosts_info_| to contain info for each element of
+  //  |changed_hosts_|.
+  void UpdateHostsInfo(const std::set<HostID>& changed_hosts);
 
   // ExtensionRegistryObserver:
   void OnExtensionUnloaded(content::BrowserContext* browser_context,
                            const Extension* extension,
                            UnloadedExtensionInfo::Reason reason) override;
 
   // Initiates script load when we have been waiting for the extension system
   // to be ready.
   void OnExtensionSystemReady();
 
+  // Maps host info needed for localization to a host ID.
+  HostsInfo hosts_info_;
+
+  // Manages content verification of the loaded user scripts.
+  scoped_refptr<ContentVerifier> content_verifier_;
+
   ScopedObserver<ExtensionRegistry, ExtensionRegistryObserver>
       extension_registry_observer_;
 
   base::WeakPtrFactory<ExtensionUserScriptLoader> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ExtensionUserScriptLoader);
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_EXTENSION_USER_SCRIPT_LOADER_H_