chrome/browser/ui/toasts/api/toast_specification.h

// Copyright 2024 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_UI_TOASTS_API_TOAST_SPECIFICATION_H_
#define CHROME_BROWSER_UI_TOASTS_API_TOAST_SPECIFICATION_H_

#include <memory>
#include <optional>

#include "base/functional/callback.h"
#include "base/functional/callback_forward.h"
#include "base/memory/raw_ref.h"
#include "base/types/pass_key.h"
#include "ui/base/accelerators/accelerator.h"
#include "ui/gfx/vector_icon_types.h"

// ToastSpecification details the supported actions and elements it contains.
class ToastSpecification {
 public:
  class Builder final {
   public:
    Builder(const gfx::VectorIcon& icon, int body_string_id);
    explicit Builder(const gfx::VectorIcon& icon);
    Builder(const Builder& other) = delete;
    Builder& operator=(const Builder&) = delete;

    ~Builder();

    // Adds a "X" close button to the toast. However, if the toast have a menu,
    // the close button will be a rounded dismiss button instead.
    Builder& AddCloseButton();

    // Adds a rounded action button toast with the `action_button_string_id`
    // for the label. All toasts with an action button must also have a close
    // button.
    Builder& AddActionButton(int action_button_string_id,
                             base::RepeatingClosure closure);

    // Adds a three dot menu to the toast. Toasts with an action button are not
    // allowed to have menu because they must have an "X" close button instead.
    // If the specification includes a menu, the `ToastParams` that are used
    // to show the Toast must have a non-null `menu_model` member.
    Builder& AddMenu();

    // Toasts by default are scoped to the active tab when they are triggered.
    // Globally scoped toasts will not immediately dismiss when the user
    // switches tabs or navigates to another site and will rely on timing out to
    // dismiss.
    Builder& AddGlobalScoped();

    // Adds the accelerator that is used to trigger an action when this toast is
    // displayed. The accelerator is handled by the view.
    Builder& AddAccelerator(ui::Accelerator accelerator,
                            base::RepeatingClosure callback);

    std::unique_ptr<ToastSpecification> Build();

   private:
    void ValidateSpecification();

    std::unique_ptr<ToastSpecification> toast_specification_;
  };

  ToastSpecification(base::PassKey<ToastSpecification::Builder>,
                     const gfx::VectorIcon& icon,
                     int string_id);
  ToastSpecification(base::PassKey<ToastSpecification::Builder>,
                     const gfx::VectorIcon& icon);
  ~ToastSpecification();

  int body_string_id() const { return body_string_id_; }
  const gfx::VectorIcon& icon() const { return *icon_; }
  bool has_close_button() const { return has_close_button_; }
  std::optional<int> action_button_string_id() const {
    return action_button_string_id_;
  }
  base::RepeatingClosure action_button_callback() const {
    return action_button_closure_;
  }
  const ui::Accelerator& accelerator() const { return accelerator_; }
  base::RepeatingClosure accelerator_callback() const {
    return accelerator_callback_;
  }

  bool has_menu() const { return has_menu_; }
  bool is_global_scope() const { return is_global_scope_; }
  bool has_accelerator() const { return !accelerator_.IsEmpty(); }

  void AddCloseButton();
  void AddActionButton(int string_id, base::RepeatingClosure closure);
  void AddMenu();
  void AddGlobalScope();
  void AddAccelerator(ui::Accelerator accelerator,
                      base::RepeatingClosure callback);

 private:
  const base::raw_ref<const gfx::VectorIcon> icon_;
  int body_string_id_ = 0;
  bool has_close_button_ = false;
  bool has_menu_ = false;
  std::optional<int> action_button_string_id_;
  base::RepeatingClosure action_button_closure_;
  bool is_global_scope_ = false;
  ui::Accelerator accelerator_;
  base::RepeatingClosure accelerator_callback_;
};

#endif  // CHROME_BROWSER_UI_TOASTS_API_TOAST_SPECIFICATION_H_