// 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_STARTUP_HELPER_H_
#define CHROME_BROWSER_EXTENSIONS_STARTUP_HELPER_H_

#include "base/memory/ref_counted.h"
#include "chrome/browser/extensions/pack_extension_job.h"

class Profile;

namespace base {
class CommandLine;
}

namespace extensions {

// Initialization helpers for various Extension startup actions.
class StartupHelper : public PackExtensionJob::Client {
 public:
  StartupHelper();
  virtual ~StartupHelper();

  virtual void OnPackSuccess(
      const base::FilePath& crx_path,
      const base::FilePath& output_private_key_path) OVERRIDE;
  virtual void OnPackFailure(const std::string& error_message,
                             ExtensionCreator::ErrorType type) OVERRIDE;

  // Handle --pack-extension flag from the |cmd_line| by packing the specified
  // extension. Returns false if the pack job failed.
  bool PackExtension(const base::CommandLine& cmd_line);

  // Validates a crx at the path given by the --validate-extension flag - can
  // it be installed? Returns true if the crx is valid, or false otherwise.
  // If the return value is false, a description of the problem may be written
  // into |error|.
  bool ValidateCrx(const base::CommandLine& cmd_line, std::string* error);

  // Handle --install-from-webstore flag from |cmd_line| by downloading
  // metadata from the webstore for the given id, prompting the user to
  // confirm, and then downloading the crx and installing it.
  bool InstallFromWebstore(const base::CommandLine& cmd_line, Profile* profile);

  // Handle --limited-install-from-webstore flag from |cmd_line| by downloading
  // metadata from the webstore for the given id, prompting the user to
  // confirm, and then downloading the crx and installing it.
  // This whole process is only kicked off by this function and completed
  // asynchronously unlike InstallFromWebstore which finishes everything before
  // returning.
  void LimitedInstallFromWebstore(const base::CommandLine& cmd_line,
                                  Profile* profile,
                                  base::Callback<void()> done_callback);

  // Maps the command line argument to the extension id. Returns an empty string
  // in the case when there is no mapping.
  std::string WebStoreIdFromLimitedInstallCmdLine(
      const base::CommandLine& cmd_line);
 private:
  scoped_refptr<PackExtensionJob> pack_job_;
  bool pack_job_succeeded_;

  DISALLOW_COPY_AND_ASSIGN(StartupHelper);
};

}  // namespace extensions

#endif  // CHROME_BROWSER_EXTENSIONS_STARTUP_HELPER_H_