text: // Copyright 2014 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.
  
  #include "extensions/browser/declarative_user_script_set.h"
  
  #include "content/public/browser/browser_context.h"
  #include "extensions/browser/extension_user_script_loader.h"
  #include "extensions/browser/user_script_loader.h"
  #include "extensions/browser/web_ui_user_script_loader.h"
  #include "extensions/common/user_script.h"
  
  namespace extensions {
  
  DeclarativeUserScriptSet::DeclarativeUserScriptSet(
      content::BrowserContext* browser_context,
      const HostID& host_id)
      : host_id_(host_id) {
    switch (host_id_.type()) {
      case HostID::EXTENSIONS:
        loader_.reset(new ExtensionUserScriptLoader(
            browser_context, host_id,
            false /* listen_for_extension_system_loaded */));
        break;
      case HostID::WEBUI:
        loader_.reset(new WebUIUserScriptLoader(browser_context, host_id));
        break;
    }
  }
  
  DeclarativeUserScriptSet::~DeclarativeUserScriptSet() {}
  
  void DeclarativeUserScriptSet::AddScript(std::unique_ptr<UserScript> script) {
    std::unique_ptr<UserScriptList> scripts(new UserScriptList());
    scripts->push_back(std::move(script));
    loader_->AddScripts(std::move(scripts));
  }
  
  void DeclarativeUserScriptSet::AddScripts(
      std::unique_ptr<UserScriptList> scripts,
      int render_process_id,
      int render_view_id) {
    loader_->AddScripts(std::move(scripts), render_process_id, render_view_id);
  }
  
  void DeclarativeUserScriptSet::RemoveScript(const UserScriptIDPair& script) {
    std::set<UserScriptIDPair> scripts;
    scripts.insert(script);
    RemoveScripts(scripts);
  }
  
  void DeclarativeUserScriptSet::RemoveScripts(
      const std::set<UserScriptIDPair>& scripts) {
    loader_->RemoveScripts(scripts);
  }
  
  void DeclarativeUserScriptSet::ClearScripts() {
    loader_->ClearScripts();
  }
  
  }  // namespace extensions
  