chrome/browser/web_applications/commands/generated_icon_fix_command.h

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_WEB_APPLICATIONS_COMMANDS_GENERATED_ICON_FIX_COMMAND_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_COMMANDS_GENERATED_ICON_FIX_COMMAND_H_

#include <memory>

#include "base/functional/callback.h"
#include "base/memory/weak_ptr.h"
#include "base/values.h"
#include "chrome/browser/web_applications/commands/web_app_command.h"
#include "chrome/browser/web_applications/locks/shared_web_contents_with_app_lock.h"
#include "chrome/browser/web_applications/web_app_icon_manager.h"
#include "chrome/browser/web_applications/web_app_id.h"
#include "chrome/browser/web_applications/web_app_install_info.h"

namespace web_app {

class WebAppIconDownloader;

enum class GeneratedIconFixResult {
  kAppUninstalled,
  kShutdown,
  kDownloadFailure,
  kStillGenerated,
  kWriteFailure,
  kSuccess,
};

class GeneratedIconFixCommand
    : public WebAppCommandTemplate<SharedWebContentsWithAppLock> {
 public:
  explicit GeneratedIconFixCommand(
      AppId app_id,
      base::OnceCallback<void(GeneratedIconFixResult)> callback);
  ~GeneratedIconFixCommand() override;

  // WebAppCommandTemplate<SharedWebContentsWithAppLock>:
  void StartWithLock(
      std::unique_ptr<SharedWebContentsWithAppLock> lock) override;
  void OnShutdown() override;
  const LockDescription& lock_description() const override;
  base::Value ToDebugValue() const override;

 private:
  void OnIconsDownloaded(IconsDownloadedResult result,
                         IconsMap icons_map,
                         DownloadedIconsHttpResults icons_http_results);
  void OnIconsWritten(bool success);
  void Stop(GeneratedIconFixResult result, base::Location location);

  AppId app_id_;
  base::OnceCallback<void(GeneratedIconFixResult)> callback_;
  SharedWebContentsWithAppLockDescription lock_description_;
  std::unique_ptr<SharedWebContentsWithAppLock> lock_;

  std::unique_ptr<WebAppIconDownloader> icon_downloader_;
  WebAppInstallInfo install_info_;

  base::Location stop_location_;

  base::WeakPtrFactory<GeneratedIconFixCommand> weak_factory_{this};
};

}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_COMMANDS_GENERATED_ICON_FIX_COMMAND_H_