| // 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_manager.h" |
| |
| #include "content/public/browser/browser_context.h" |
| #include "extensions/browser/declarative_user_script_manager_factory.h" |
| #include "extensions/browser/declarative_user_script_set.h" |
| |
| namespace extensions { |
| |
| DeclarativeUserScriptManager::DeclarativeUserScriptManager( |
| content::BrowserContext* browser_context) |
| : browser_context_(browser_context) { |
| extension_registry_observer_.Add(ExtensionRegistry::Get(browser_context)); |
| } |
| |
| DeclarativeUserScriptManager::~DeclarativeUserScriptManager() { |
| } |
| |
| // static |
| DeclarativeUserScriptManager* DeclarativeUserScriptManager::Get( |
| content::BrowserContext* browser_context) { |
| return DeclarativeUserScriptManagerFactory::GetForBrowserContext( |
| browser_context); |
| } |
| |
| DeclarativeUserScriptSet* |
| DeclarativeUserScriptManager::GetDeclarativeUserScriptSetByID( |
| const HostID& host_id) { |
| auto it = declarative_user_script_sets_.find(host_id); |
| |
| if (it != declarative_user_script_sets_.end()) |
| return it->second.get(); |
| |
| return CreateDeclarativeUserScriptSet(host_id); |
| } |
| |
| DeclarativeUserScriptSet* |
| DeclarativeUserScriptManager::CreateDeclarativeUserScriptSet( |
| const HostID& host_id) { |
| // Inserts a new DeclarativeUserScriptManager and returns a ptr to it. |
| return declarative_user_script_sets_ |
| .insert( |
| std::make_pair(host_id, std::make_unique<DeclarativeUserScriptSet>( |
| browser_context_, host_id))) |
| .first->second.get(); |
| } |
| |
| void DeclarativeUserScriptManager::OnExtensionUnloaded( |
| content::BrowserContext* browser_context, |
| const Extension* extension, |
| UnloadedExtensionReason reason) { |
| for (const auto& val : declarative_user_script_sets_) { |
| DeclarativeUserScriptSet* set = val.second.get(); |
| if (set->host_id().id() == extension->id()) |
| set->ClearScripts(); |
| } |
| } |
| |
| } // namespace extensions |