blob: 39b10d01ce62c5e2c2f3a4b584fc1d6d9bc7ec81 [file] [log] [blame]
// Copyright 2013 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 "chrome/browser/apps/platform_apps/shortcut_manager.h"
#include "base/bind.h"
#include "base/command_line.h"
#include "base/compiler_specific.h"
#include "base/strings/string16.h"
#include "base/strings/utf_string_conversions.h"
#include "base/task/post_task.h"
#include "build/build_config.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_manager.h"
#include "chrome/browser/shell_integration.h"
#include "chrome/browser/web_applications/extensions/web_app_extension_shortcut.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/pref_names.h"
#include "components/pref_registry/pref_registry_syncable.h"
#include "components/prefs/pref_service.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/common/content_switches.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/extension_system.h"
#include "extensions/common/extension_set.h"
#include "extensions/common/one_shot_event.h"
using extensions::Extension;
namespace {
// This version number is stored in local prefs to check whether app shortcuts
// need to be recreated. This might happen when we change various aspects of app
// shortcuts like command-line flags or associated icons, binaries, etc.
#if defined(OS_MACOSX)
const int kCurrentAppShortcutsVersion = 5;
#else
const int kCurrentAppShortcutsVersion = 0;
#endif
// Delay in seconds before running UpdateShortcutsForAllApps.
const int kUpdateShortcutsForAllAppsDelay = 10;
void CreateShortcutsForApp(Profile* profile, const Extension* app) {
web_app::ShortcutLocations creation_locations;
// Creates a shortcut for an app in the Chrome Apps subdir of the
// applications menu, if there is not already one present.
creation_locations.applications_menu_location =
web_app::APP_MENU_LOCATION_SUBDIR_CHROMEAPPS;
web_app::CreateShortcuts(web_app::SHORTCUT_CREATION_AUTOMATED,
creation_locations, profile, app);
}
void SetCurrentAppShortcutsVersion(PrefService* prefs) {
prefs->SetInteger(prefs::kAppShortcutsVersion, kCurrentAppShortcutsVersion);
}
} // namespace
// static
void AppShortcutManager::RegisterProfilePrefs(
user_prefs::PrefRegistrySyncable* registry) {
// Indicates whether app shortcuts have been created.
registry->RegisterIntegerPref(prefs::kAppShortcutsVersion, 0);
}
AppShortcutManager::AppShortcutManager(Profile* profile)
: profile_(profile),
is_profile_attributes_storage_observer_(false),
prefs_(profile->GetPrefs()),
extension_registry_observer_(this),
weak_ptr_factory_(this) {
// Use of g_browser_process requires that we are either on the UI thread, or
// there are no threads initialized (such as in unit tests).
DCHECK(!content::BrowserThread::IsThreadInitialized(
content::BrowserThread::UI) ||
content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
extension_registry_observer_.Add(
extensions::ExtensionRegistry::Get(profile_));
// Wait for extensions to be ready before running
// UpdateShortcutsForAllAppsIfNeeded.
extensions::ExtensionSystem::Get(profile)->ready().Post(
FROM_HERE,
base::Bind(&AppShortcutManager::UpdateShortcutsForAllAppsIfNeeded,
weak_ptr_factory_.GetWeakPtr()));
ProfileManager* profile_manager = g_browser_process->profile_manager();
// profile_manager might be NULL in testing environments.
if (profile_manager) {
profile_manager->GetProfileAttributesStorage().AddObserver(this);
is_profile_attributes_storage_observer_ = true;
}
}
AppShortcutManager::~AppShortcutManager() {
if (g_browser_process && is_profile_attributes_storage_observer_) {
ProfileManager* profile_manager = g_browser_process->profile_manager();
// profile_manager might be NULL in testing environments or during shutdown.
if (profile_manager)
profile_manager->GetProfileAttributesStorage().RemoveObserver(this);
}
}
void AppShortcutManager::OnExtensionWillBeInstalled(
content::BrowserContext* browser_context,
const Extension* extension,
bool is_update,
const std::string& old_name) {
if (!extension->is_app())
return;
// If the app is being updated, update any existing shortcuts but do not
// create new ones. If it is being installed, automatically create a
// shortcut in the applications menu (e.g., Start Menu).
if (is_update) {
web_app::UpdateAllShortcuts(base::UTF8ToUTF16(old_name), profile_,
extension, base::Closure());
} else {
CreateShortcutsForApp(profile_, extension);
}
}
void AppShortcutManager::OnExtensionUninstalled(
content::BrowserContext* browser_context,
const Extension* extension,
extensions::UninstallReason reason) {
web_app::DeleteAllShortcuts(profile_, extension);
}
void AppShortcutManager::OnProfileWillBeRemoved(
const base::FilePath& profile_path) {
if (profile_path != profile_->GetPath())
return;
web_app::internals::GetShortcutIOTaskRunner()->PostTask(
FROM_HERE,
base::BindOnce(&web_app::internals::DeleteAllShortcutsForProfile,
profile_path));
}
void AppShortcutManager::UpdateShortcutsForAllAppsIfNeeded() {
if (base::CommandLine::ForCurrentProcess()->HasSwitch(switches::kTestType))
return;
int last_version = prefs_->GetInteger(prefs::kAppShortcutsVersion);
if (last_version >= kCurrentAppShortcutsVersion)
return;
base::PostDelayedTaskWithTraits(
FROM_HERE, {content::BrowserThread::UI},
base::BindOnce(&web_app::UpdateShortcutsForAllApps, profile_,
base::Bind(&SetCurrentAppShortcutsVersion, prefs_)),
base::TimeDelta::FromSeconds(kUpdateShortcutsForAllAppsDelay));
}