src/chrome/browser/extensions/api/management/management_api.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_EXTENSIONS_API_MANAGEMENT_MANAGEMENT_API_H_
   6 #define CHROME_BROWSER_EXTENSIONS_API_MANAGEMENT_MANAGEMENT_API_H_
   7
   8 #include "base/compiler_specific.h"
   9 #include "chrome/browser/extensions/chrome_extension_function.h"
  10 #include "chrome/browser/extensions/extension_install_prompt.h"
  11 #include "chrome/browser/extensions/extension_uninstall_dialog.h"
  12 #include "components/keyed_service/core/keyed_service.h"
  13 #include "content/public/browser/notification_observer.h"
  14 #include "content/public/browser/notification_registrar.h"
  15 #include "extensions/browser/browser_context_keyed_api_factory.h"
  16 #include "extensions/browser/event_router.h"
  17
  18 class ExtensionService;
  19 class ExtensionUninstallDialog;
  20
  21 namespace extensions {
  22
  23 class ManagementFunction : public ChromeSyncExtensionFunction {
  24  protected:
  25   virtual ~ManagementFunction() {}
  26
  27   ExtensionService* service();
  28 };
  29
  30 class AsyncManagementFunction : public ChromeAsyncExtensionFunction {
  31  protected:
  32   virtual ~AsyncManagementFunction() {}
  33
  34   ExtensionService* service();
  35 };
  36
  37 class ManagementGetAllFunction : public ManagementFunction {
  38  public:
  39   DECLARE_EXTENSION_FUNCTION("management.getAll", MANAGEMENT_GETALL)
  40
  41  protected:
  42   virtual ~ManagementGetAllFunction() {}
  43
  44   // ExtensionFunction:
  45   virtual bool RunSync() OVERRIDE;
  46 };
  47
  48 class ManagementGetFunction : public ManagementFunction {
  49  public:
  50   DECLARE_EXTENSION_FUNCTION("management.get", MANAGEMENT_GET)
  51
  52  protected:
  53   virtual ~ManagementGetFunction() {}
  54
  55   // ExtensionFunction:
  56   virtual bool RunSync() OVERRIDE;
  57 };
  58
  59 class ManagementGetPermissionWarningsByIdFunction : public ManagementFunction {
  60  public:
  61   DECLARE_EXTENSION_FUNCTION("management.getPermissionWarningsById",
  62                              MANAGEMENT_GETPERMISSIONWARNINGSBYID)
  63
  64  protected:
  65   virtual ~ManagementGetPermissionWarningsByIdFunction() {}
  66
  67   // ExtensionFunction:
  68   virtual bool RunSync() OVERRIDE;
  69 };
  70
  71 class ManagementGetPermissionWarningsByManifestFunction
  72     : public ChromeAsyncExtensionFunction {
  73  public:
  74   DECLARE_EXTENSION_FUNCTION(
  75       "management.getPermissionWarningsByManifest",
  76       MANAGEMENT_GETPERMISSIONWARNINGSBYMANIFEST);
  77
  78   // Called when utility process finishes.
  79   void OnParseSuccess(scoped_ptr<base::DictionaryValue> parsed_manifest);
  80   void OnParseFailure(const std::string& error);
  81
  82  protected:
  83   virtual ~ManagementGetPermissionWarningsByManifestFunction() {}
  84
  85   // ExtensionFunction:
  86   virtual bool RunAsync() OVERRIDE;
  87 };
  88
  89 class ManagementLaunchAppFunction : public ManagementFunction {
  90  public:
  91   DECLARE_EXTENSION_FUNCTION("management.launchApp", MANAGEMENT_LAUNCHAPP)
  92
  93  protected:
  94   virtual ~ManagementLaunchAppFunction() {}
  95
  96   // ExtensionFunction:
  97   virtual bool RunSync() OVERRIDE;
  98 };
  99
 100 class ManagementSetEnabledFunction : public AsyncManagementFunction,
 101                            public ExtensionInstallPrompt::Delegate {
 102  public:
 103   DECLARE_EXTENSION_FUNCTION("management.setEnabled", MANAGEMENT_SETENABLED)
 104
 105   ManagementSetEnabledFunction();
 106
 107  protected:
 108   virtual ~ManagementSetEnabledFunction();
 109
 110   // ExtensionFunction:
 111   virtual bool RunAsync() OVERRIDE;
 112
 113   // ExtensionInstallPrompt::Delegate.
 114   virtual void InstallUIProceed() OVERRIDE;
 115   virtual void InstallUIAbort(bool user_initiated) OVERRIDE;
 116
 117  private:
 118   std::string extension_id_;
 119
 120   // Used for prompting to re-enable items with permissions escalation updates.
 121   scoped_ptr<ExtensionInstallPrompt> install_prompt_;
 122 };
 123
 124 class ManagementUninstallFunctionBase : public AsyncManagementFunction,
 125                           public ExtensionUninstallDialog::Delegate {
 126  public:
 127   ManagementUninstallFunctionBase();
 128
 129   static void SetAutoConfirmForTest(bool should_proceed);
 130
 131   // ExtensionUninstallDialog::Delegate implementation.
 132   virtual void ExtensionUninstallAccepted() OVERRIDE;
 133   virtual void ExtensionUninstallCanceled() OVERRIDE;
 134
 135  protected:
 136   virtual ~ManagementUninstallFunctionBase();
 137
 138   bool Uninstall(const std::string& extension_id, bool show_confirm_dialog);
 139  private:
 140
 141   // If should_uninstall is true, this method does the actual uninstall.
 142   // If |show_uninstall_dialog|, then this function will be called by one of the
 143   // Accepted/Canceled callbacks. Otherwise, it's called directly from RunAsync.
 144   void Finish(bool should_uninstall);
 145
 146   std::string extension_id_;
 147   scoped_ptr<ExtensionUninstallDialog> extension_uninstall_dialog_;
 148 };
 149
 150 class ManagementUninstallFunction : public ManagementUninstallFunctionBase {
 151  public:
 152   DECLARE_EXTENSION_FUNCTION("management.uninstall", MANAGEMENT_UNINSTALL)
 153
 154   ManagementUninstallFunction();
 155
 156  private:
 157   virtual ~ManagementUninstallFunction();
 158
 159   virtual bool RunAsync() OVERRIDE;
 160 };
 161
 162 class ManagementUninstallSelfFunction : public ManagementUninstallFunctionBase {
 163  public:
 164   DECLARE_EXTENSION_FUNCTION("management.uninstallSelf",
 165       MANAGEMENT_UNINSTALLSELF);
 166
 167   ManagementUninstallSelfFunction();
 168
 169  private:
 170   virtual ~ManagementUninstallSelfFunction();
 171
 172   virtual bool RunAsync() OVERRIDE;
 173 };
 174
 175 class ManagementCreateAppShortcutFunction : public AsyncManagementFunction {
 176  public:
 177   DECLARE_EXTENSION_FUNCTION("management.createAppShortcut",
 178       MANAGEMENT_CREATEAPPSHORTCUT);
 179
 180   ManagementCreateAppShortcutFunction();
 181
 182   void OnCloseShortcutPrompt(bool created);
 183
 184   static void SetAutoConfirmForTest(bool should_proceed);
 185
 186  protected:
 187   virtual ~ManagementCreateAppShortcutFunction();
 188
 189   virtual bool RunAsync() OVERRIDE;
 190 };
 191
 192 class ManagementEventRouter : public content::NotificationObserver {
 193  public:
 194   explicit ManagementEventRouter(Profile* profile);
 195   virtual ~ManagementEventRouter();
 196
 197  private:
 198   // content::NotificationObserver implementation.
 199   virtual void Observe(int type,
 200                        const content::NotificationSource& source,
 201                        const content::NotificationDetails& details) OVERRIDE;
 202
 203   content::NotificationRegistrar registrar_;
 204
 205   Profile* profile_;
 206
 207   DISALLOW_COPY_AND_ASSIGN(ManagementEventRouter);
 208 };
 209
 210 class ManagementAPI : public BrowserContextKeyedAPI,
 211                       public extensions::EventRouter::Observer {
 212  public:
 213   explicit ManagementAPI(content::BrowserContext* context);
 214   virtual ~ManagementAPI();
 215
 216   // KeyedService implementation.
 217   virtual void Shutdown() OVERRIDE;
 218
 219   // BrowserContextKeyedAPI implementation.
 220   static BrowserContextKeyedAPIFactory<ManagementAPI>* GetFactoryInstance();
 221
 222   // EventRouter::Observer implementation.
 223   virtual void OnListenerAdded(const extensions::EventListenerInfo& details)
 224       OVERRIDE;
 225
 226  private:
 227   friend class BrowserContextKeyedAPIFactory<ManagementAPI>;
 228
 229   content::BrowserContext* browser_context_;
 230
 231   // BrowserContextKeyedAPI implementation.
 232   static const char* service_name() {
 233     return "ManagementAPI";
 234   }
 235   static const bool kServiceIsNULLWhileTesting = true;
 236
 237   // Created lazily upon OnListenerAdded.
 238   scoped_ptr<ManagementEventRouter> management_event_router_;
 239
 240   DISALLOW_COPY_AND_ASSIGN(ManagementAPI);
 241 };
 242
 243 }  // namespace extensions
 244
 245 #endif  // CHROME_BROWSER_EXTENSIONS_API_MANAGEMENT_MANAGEMENT_API_H_