Chromium Code Reviews| Index: chrome/browser/extensions/bundle_installer.h |
| diff --git a/chrome/browser/extensions/bundle_installer.h b/chrome/browser/extensions/bundle_installer.h |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..b36c075b589ae3470d5d07070e2f1c175d5c19d0 |
| --- /dev/null |
| +++ b/chrome/browser/extensions/bundle_installer.h |
| @@ -0,0 +1,186 @@ |
| +// Copyright (c) 2012 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 CHROME_BROWSER_EXTENSIONS_BUNDLE_INSTALLER_H_ |
| +#define CHROME_BROWSER_EXTENSIONS_BUNDLE_INSTALLER_H_ |
| +#pragma once |
| + |
| +#include <string> |
| +#include <vector> |
| + |
| +#include "base/memory/linked_ptr.h" |
| +#include "chrome/browser/extensions/extension_install_ui.h" |
| +#include "chrome/browser/extensions/webstore_installer.h" |
| +#include "chrome/browser/extensions/webstore_install_helper.h" |
| +#include "chrome/browser/ui/browser_list.h" |
| +#include "chrome/common/extensions/extension.h" |
| + |
| +namespace base { |
| +class DictionaryValue; |
| +} // namespace base |
| + |
| +namespace content { |
| +class NavigationController; |
| +} // namespace content |
| + |
| +class Browser; |
| +class Profile; |
| + |
| +namespace extensions { |
| + |
| +// Manages the installation life cycle for extension bundles. |
| +// |
| +// We install bundles in two steps: |
| +// 1) PromptForApproval: parse manifests and prompt the user |
| +// 2) CompleteInstall: install the CRXs and show confirmation bubble |
| +// |
| +class BundleInstaller : public WebstoreInstallHelper::Delegate, |
| + public ExtensionInstallUI::Delegate, |
| + public WebstoreInstaller::Delegate, |
| + public BrowserList::Observer, |
| + public base::RefCountedThreadSafe<BundleInstaller> { |
| + public: |
| + // Auto approve or cancel the permission prompt. |
| + static void SetAutoApproveForTesting(bool approve); |
| + |
| + class Delegate { |
| + public: |
| + virtual void OnBundleInstallApproved() {} |
| + virtual void OnBundleInstallCanceled(bool user_initiated) {} |
| + virtual void OnBundleInstallCompleted() {} |
| + }; |
| + |
| + // Represents an individual member of the bundle. |
| + struct Item { |
| + // Items are in the PENDING state until they've been installed, or the |
| + // install has failed or been canceled. |
| + enum State { |
| + STATE_PENDING, |
| + STATE_INSTALLED, |
| + STATE_FAILED |
| + }; |
| + |
| + Item(); |
| + |
| + // Creates a dummy extension and sets the manifest's name to the item's |
| + // localized name. |
| + scoped_refptr<Extension> CreateDummyExtension( |
| + base::DictionaryValue* manifest); |
| + |
| + std::string id; |
| + std::string manifest; |
| + std::string localized_name; |
| + State state; |
| + }; |
| + |
| + typedef std::vector<Item> ItemList; |
| + |
| + BundleInstaller(Profile* profile, const ItemList& items); |
| + virtual ~BundleInstaller(); |
| + |
| + // Returns true if the user has approved the bundle's permissions. |
| + bool approved() const { return approved_; } |
| + |
| + // Gets the items in the given state. |
| + ItemList GetItemsByState(Item::State state) const; |
|
Yoyo Zhou
2012/02/22 00:20:26
GetItemsWithState, perhaps?
jstritar
2012/02/22 15:45:48
Done.
|
| + |
| + // Parses the extension manifests and then prompts the user to approve their |
| + // permissions. One of OnBundleInstallApproved or OnBundleInstallCanceled |
| + // will be called when complete if |delegate| is not NULL. |
| + // Note: the |delegate| must stay alive until receiving the callback. |
| + void PromptForApproval(Delegate* delegate); |
| + |
| + // If the bundle has been approved, this downloads and installs the member |
| + // extensions. OnBundleInstallComplete will be called when the process is |
| + // complete and |delegate| is not NULL. The download process uses the |
| + // specified |controller|. When complete, we show a confirmation bubble in |
| + // the specified |browser|. |
| + // Note: the |delegate| must stay alive until receiving the callback. |
| + void CompleteInstall(content::NavigationController* controller, |
| + Browser* browser, |
| + Delegate* delegate); |
| + |
| + private: |
| + friend class base::RefCountedThreadSafe<BundleInstaller>; |
| + |
| + typedef std::map<std::string, Item> ItemMap; |
| + typedef std::map<std::string, linked_ptr<base::DictionaryValue> > ManifestMap; |
| + |
| + // Displays the install bubble for |bundle| on |browser|. |
| + // Note: this is a platform specific implementation. |
| + static void ShowInstalledBubble(const BundleInstaller* bundle, |
| + Browser* browser); |
| + |
| + // Parses the manifests using WebstoreInstallHelper. |
| + void ParseManifests(); |
| + |
| + // Notifies the delegate that the installation has been approved. |
| + void ReportApproved(); |
| + |
| + // Notifies the delegate that the installation was canceled. |
| + void ReportCanceled(bool user_initiated); |
| + |
| + // Notifies the delegate that the installation is complete. |
| + void ReportComplete(); |
| + |
| + // Prompts the user to install the bundle once we have dummy extensions for |
| + // all the pending items. |
| + void ShowPromptIfDoneParsing(); |
| + |
| + // Prompts the user to install the bundle. |
| + void ShowPrompt(); |
| + |
| + // Displays the installed bubble once all items have installed or failed. |
| + void ShowInstalledBubbleIfDone(); |
| + |
| + // WebstoreInstallHelper::Delegate implementation: |
| + virtual void OnWebstoreParseSuccess( |
| + const std::string& id, |
| + const SkBitmap& icon, |
| + base::DictionaryValue* parsed_manifest) OVERRIDE; |
| + virtual void OnWebstoreParseFailure( |
| + const std::string& id, |
| + InstallHelperResultCode result_code, |
| + const std::string& error_message) OVERRIDE; |
| + |
| + // ExtensionInstallUI::Delegate implementation: |
| + virtual void InstallUIProceed() OVERRIDE; |
| + virtual void InstallUIAbort(bool user_initiated) OVERRIDE; |
| + |
| + // WebstoreInstaller::Delegate implementation: |
| + virtual void OnExtensionInstallSuccess(const std::string& id) OVERRIDE; |
| + virtual void OnExtensionInstallFailure(const std::string& id, |
| + const std::string& error) OVERRIDE; |
| + |
| + // BrowserList::observer implementation: |
| + virtual void OnBrowserAdded(const Browser* browser) OVERRIDE; |
| + virtual void OnBrowserRemoved(const Browser* browser) OVERRIDE; |
| + virtual void OnBrowserSetLastActive(const Browser* browser) OVERRIDE; |
| + |
| + // Holds the Extensions used to generate the permission warnings. |
| + ExtensionList dummy_extensions_; |
| + |
| + // Holds the parsed manifests, indexed by the extension ids. |
| + ManifestMap parsed_manifests_; |
| + |
| + // True if the user has approved the bundle. |
| + bool approved_; |
| + |
| + // Holds the bundle's Items, indexed by their ids. |
| + ItemMap items_; |
| + |
| + // The browser to show the confirmation bubble for. |
| + Browser* browser_; |
| + |
| + // The profile that the bundle should be installed in. |
| + Profile* profile_; |
| + |
| + Delegate* delegate_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(BundleInstaller); |
| +}; |
| + |
| +} // namespace extensions |
| + |
| +#endif // CHROME_BROWSER_EXTENSIONS_BUNDLE_INSTALLER_H_ |