Prevent duplicate concurrent installs of the same extension

chrome/browser/extensions/webstore_standalone_installer.h

 // 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_WEBSTORE_STANDALONE_INSTALLER_H_
 #define CHROME_BROWSER_EXTENSIONS_WEBSTORE_STANDALONE_INSTALLER_H_
 
 #include <string>
 
 #include "base/callback.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "chrome/browser/extensions/extension_install_prompt.h"
+#include "chrome/browser/extensions/install_tracker.h"
 #include "chrome/browser/extensions/webstore_data_fetcher_delegate.h"
 #include "chrome/browser/extensions/webstore_install_helper.h"
 #include "chrome/browser/extensions/webstore_install_result.h"
 #include "chrome/browser/extensions/webstore_installer.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 
 class GURL;
 
 namespace base {
@@ skipping 33 matching lines @@
                               Profile* profile,
                               const Callback& callback);
   void BeginInstall();
 
  protected:
   virtual ~WebstoreStandaloneInstaller();
 
   // Called when the install should be aborted. The callback is cleared.
   void AbortInstall();
 
+  // Checks InstallTracker and returns true if the same extension is not
+  // currently being installed. Registers this install with the InstallTracker.
+  bool EnsureUniqueInstall(webstore_install::Result* reason,
+                           std::string* error);
+
   // Called when the install is complete.
   virtual void CompleteInstall(webstore_install::Result result,
                                const std::string& error);
 
   // Called when the installer should proceed to prompt the user.
   void ProceedWithInstallPrompt();
 
   // Lazily creates a dummy extension for display from the parsed manifest. This
   // is safe to call from OnManifestParsed() onwards. The manifest may be
   // invalid, thus the caller must check that the return value is not NULL.
   scoped_refptr<const Extension> GetLocalizedExtensionForDisplay();
 
   // Template Method's hooks to be implemented by subclasses.
 
+  // Called when this install is about to be registered with the InstallTracker.
+  // Allows subclasses to set properties of the install data.
+  virtual void InitInstallData(
+      InstallTracker::InstallProgressData* install_data) const;
+
   // Called at certain check points of the workflow to decide whether it makes
   // sense to proceed with installation. A requestor can be a website that
   // initiated an inline installation, or a command line option.
   virtual bool CheckRequestorAlive() const = 0;
 
   // Requestor's URL, if any. Should be an empty GURL if URL is meaningless
   // (e.g. for a command line option).
   virtual const GURL& GetRequestorURL() const = 0;
 
   // Should a new tab be opened after installation to show the newly installed
@@ skipping 105 matching lines @@
 
   // WebstoreInstaller::Delegate interface implementation.
   virtual void OnExtensionInstallSuccess(const std::string& id) OVERRIDE;
   virtual void OnExtensionInstallFailure(
       const std::string& id,
       const std::string& error,
       WebstoreInstaller::FailureReason reason) OVERRIDE;
 
   void ShowInstallUI();
   void OnWebStoreDataFetcherDone();
+  void DeregisterActiveInstall();
 
   // Input configuration.
   std::string id_;
   Callback callback_;
   Profile* profile_;
   WebstoreInstaller::InstallSource install_source_;
 
   // Installation dialog and its underlying prompt.
   scoped_ptr<ExtensionInstallPrompt> install_ui_;
   scoped_refptr<ExtensionInstallPrompt::Prompt> install_prompt_;
 
   // For fetching webstore JSON data.
   scoped_ptr<WebstoreDataFetcher> webstore_data_fetcher_;
 
   // Extracted from the webstore JSON data response.
   std::string localized_name_;
   std::string localized_description_;
   bool show_user_count_;
   std::string localized_user_count_;
   double average_rating_;
   int rating_count_;
   scoped_ptr<base::DictionaryValue> webstore_data_;
   scoped_ptr<base::DictionaryValue> manifest_;
   SkBitmap icon_;
 
+  // True if this install was registered with InstallTracker.
+  bool install_registered_;
+
   // Created by ShowInstallUI() when a prompt is shown (if
   // the implementor returns a non-NULL in CreateInstallPrompt()).
   scoped_refptr<Extension> localized_extension_for_display_;
 
   DISALLOW_IMPLICIT_CONSTRUCTORS(WebstoreStandaloneInstaller);
 };
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_WEBSTORE_STANDALONE_INSTALLER_H_