chrome/browser/web_applications/commands/update_file_handler_command.h

// Copyright 2022 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_UPDATE_FILE_HANDLER_COMMAND_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_COMMANDS_UPDATE_FILE_HANDLER_COMMAND_H_

#include <memory>

#include "base/functional/callback.h"
#include "chrome/browser/web_applications/commands/web_app_command.h"
#include "chrome/browser/web_applications/web_app_id.h"
#include "third_party/abseil-cpp/absl/types/optional.h"

namespace web_app {

class AppLock;
class AppLockDescription;
class LockDescription;
enum class Result;

// UpdateFileHandlerCommand updates file handler registration to match with the
// app's setting or user choice.
class UpdateFileHandlerCommand : public WebAppCommandTemplate<AppLock> {
 public:
  // Updates the File Handling API approval state for the given app. If
  // necessary, it also updates the registration with the OS.
  static std::unique_ptr<UpdateFileHandlerCommand> CreateForPersistUserChoice(
      const AppId& app_id,
      bool allowed,
      base::OnceClosure callback);

  // Updates the file handler registration with the OS to match the app's
  // settings. Note that this tries to avoid extra work by no-oping if the
  // current OS state matches what is calculated to be the desired stated. For
  // example, if Chromium has already registered file handlers with the OS, and
  // finds that file handlers *should* be registered with the OS, this function
  // will no-op. This will not account for what the current file handlers
  // actually are. The actual set of file handlers can only change on app
  // update, and that path must go through
  // `OsIntegrationManager::UpdateOsHooks()`, which always clobbers and renews
  // the entire set of OS-registered file handlers (and other OS hooks).
  static std::unique_ptr<UpdateFileHandlerCommand> CreateForUpdate(
      const AppId& app_id,
      base::OnceClosure callback);
  ~UpdateFileHandlerCommand() override;

  void StartWithLock(std::unique_ptr<AppLock> lock) override;

  LockDescription& lock_description() const override;

  base::Value ToDebugValue() const override;

  void OnSyncSourceRemoved() override {}
  void OnShutdown() override;

 private:
  UpdateFileHandlerCommand(const AppId& app_id,
                           absl::optional<bool> user_choice_to_remember,
                           base::OnceClosure callback);
  void OnFileHandlerUpdated(bool file_handling_enabled, Result result);
  void ReportResultAndDestroy(CommandResult result);

  std::unique_ptr<AppLockDescription> lock_description_;
  std::unique_ptr<AppLock> lock_;

  const AppId app_id_;
  absl::optional<bool> user_choice_to_remember_;
  base::OnceClosure callback_;

  base::Value::Dict debug_info_;

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

}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_COMMANDS_UPDATE_FILE_HANDLER_COMMAND_H_