// 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_OS_INTEGRATION_SHORTCUT_MENU_HANDLING_SUB_MANAGER_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_OS_INTEGRATION_SHORTCUT_MENU_HANDLING_SUB_MANAGER_H_

#include <optional>
#include <vector>

#include "base/files/file_path.h"
#include "base/functional/callback_forward.h"
#include "chrome/browser/web_applications/os_integration/os_integration_sub_manager.h"
#include "chrome/browser/web_applications/proto/web_app_os_integration_state.pb.h"
#include "chrome/browser/web_applications/web_app_icon_manager.h"
#include "chrome/browser/web_applications/web_app_install_info.h"
#include "components/webapps/common/web_app_id.h"

namespace web_app {

class WebAppProvider;

// Used to track when information, like shortcut menu icons, app title and app
// launch url in shortcut menu were last updated at and update them once they
// are changed.
class ShortcutMenuHandlingSubManager : public OsIntegrationSubManager {
 public:
  ShortcutMenuHandlingSubManager(const base::FilePath& profile_path,
                                 WebAppProvider& provider);
  ~ShortcutMenuHandlingSubManager() override;

  void Configure(const webapps::AppId& app_id,
                 proto::os_state::WebAppOsIntegration& desired_state,
                 base::OnceClosure configure_done) override;
  void Execute(const webapps::AppId& app_id,
               const std::optional<SynchronizeOsOptions>& synchronize_options,
               const proto::os_state::WebAppOsIntegration& desired_state,
               const proto::os_state::WebAppOsIntegration& current_state,
               base::OnceClosure execute_complete) override;
  void ForceUnregister(const webapps::AppId& app_id,
                       base::OnceClosure callback) override;

 private:
  void StoreShortcutMenuData(
      const webapps::AppId& app_id,
      std::vector<WebAppShortcutsMenuItemInfo> shortcut_menu_item_info,
      proto::os_state::ShortcutMenus* shortcut_menus,
      WebAppIconManager::ShortcutIconDataVector data);
  void StartShortcutsMenuUnregistration(
      const webapps::AppId& app_id,
      const proto::os_state::WebAppOsIntegration& current_state,
      base::OnceClosure registration_callback);
  void ReadIconDataForShortcutsMenu(
      const webapps::AppId& app_id,
      const proto::os_state::WebAppOsIntegration& desired_state,
      base::OnceClosure execute_complete);
  void OnIconDataLoadedRegisterShortcutsMenu(
      const webapps::AppId& app_id,
      const proto::os_state::WebAppOsIntegration& desired_state,
      base::OnceClosure execute_complete,
      ShortcutsMenuIconBitmaps shortcut_menu_icon_bitmaps);

  const base::FilePath profile_path_;
  const raw_ref<WebAppProvider> provider_;

  base::WeakPtrFactory<ShortcutMenuHandlingSubManager> weak_ptr_factory_{this};
};
}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_OS_INTEGRATION_SHORTCUT_MENU_HANDLING_SUB_MANAGER_H_