// Copyright 2012 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_TABS_TAB_STRIP_MODEL_OBSERVER_H_
#define CHROME_BROWSER_UI_TABS_TAB_STRIP_MODEL_OBSERVER_H_

#include <memory>
#include <optional>
#include <set>
#include <variant>
#include <vector>

#include "base/memory/raw_ptr.h"
#include "chrome/browser/ui/tabs/tab_change_type.h"
#include "components/sessions/core/session_id.h"
#include "components/tab_groups/tab_group_id.h"
#include "components/tab_groups/tab_group_visual_data.h"
#include "components/tabs/public/split_tab_id.h"
#include "components/tabs/public/split_tab_visual_data.h"
#include "components/tabs/public/tab_interface.h"
#include "third_party/perfetto/include/perfetto/tracing/traced_value_forward.h"
#include "ui/base/models/list_selection_model.h"

class TabStripModel;
namespace tabs {
class TabGroupTabCollection;
}  // namespace tabs

namespace content {
class WebContents;
}

////////////////////////////////////////////////////////////////////////////////
//
// TabStripModelChange / TabStripSelectionChange
//
// This observer is not appropriate for most use cases. It's primarily used for
// features that must directly interface with the tab strip, for example: tab
// groups, tab search, etc.
// Most features in Chrome need to hold state on a per-tab basis. In that case,
// add a controller to TabFeatures and use TabInterface to observe for the tab
// events.
//
// The following class and structures are used to inform TabStripModelObservers
// of changes to:
// 1) selection model
// 2) activated tab
// 3) inserted/removed/moved tabs.
// These changes must be bundled together because (1) and (2) consist of indices
// into a list of tabs [determined by (3)]. All three must be kept synchronized.
//
////////////////////////////////////////////////////////////////////////////////
class TabStripModelChange {
 public:
  enum Type { kSelectionOnly, kInserted, kRemoved, kMoved, kReplaced };

  // Used to specify what will happen with the tab after it is removed.
  enum class RemoveReason {
    // Tab will be deleted.
    kDeleted,

    // Tab got detached from a TabStrip and inserted into another TabStrip.
    kInsertedIntoOtherTabStrip
  };

  struct RemovedTab {
    RemovedTab(tabs::TabInterface* tab,
               int index,
               RemoveReason remove_reason,
               tabs::TabInterface::DetachReason tab_detach_reason,
               std::optional<SessionID> session_id);
    virtual ~RemovedTab();
    RemovedTab(RemovedTab&& other);

    void WriteIntoTrace(perfetto::TracedValue context) const;

    raw_ptr<tabs::TabInterface> tab = nullptr;
    raw_ptr<content::WebContents> contents = nullptr;
    int index;
    RemoveReason remove_reason;
    tabs::TabInterface::DetachReason tab_detach_reason;
    std::optional<SessionID> session_id;
  };

  struct ContentsWithIndex {
    raw_ptr<tabs::TabInterface> tab = nullptr;
    raw_ptr<content::WebContents> contents = nullptr;
    int index;

    void WriteIntoTrace(perfetto::TracedValue context) const;
  };

  // WebContents were inserted. This implicitly changes the existing selection
  // model by calling IncrementFrom(index) on each index in |contents[i].index|.
  struct Insert {
    Insert();
    ~Insert();
    Insert(Insert&& other);
    Insert& operator=(Insert&& other);

    // Contains the tabs that were inserted, along with their indexes at the
    // time of insertion. For example, if we inserted elements:
    //
    // Before insertion:
    // A B C D
    // 0 1 2 3
    //
    // After insertion:
    // A X Y B C Z D
    // 0 1 2 3 4 5 6
    //
    // If the tabs were inserted in the order X, Y, Z, `contents` would contain:
    // { X, 1 }, { Y, 2 }, { Z, 5 }
    //
    // But if the contents were inserted in the order Z, Y, X, `contents` would
    // contain:
    // { Z, 3 }, { Y, 1 }, { X, 1 }
    //
    // Therefore all observers which store indices of tabs should update them in
    // the order the tabs appear in `contents`. Observers should not do
    // index-based queries based on their own internally-stored indices until
    // after processing all of `contents`.
    std::vector<ContentsWithIndex> contents;

    void WriteIntoTrace(perfetto::TracedValue context) const;
  };

  // WebContents were removed at |indices_before_removal|. This implicitly
  // changes the existing selection model by calling DecrementFrom(index).
  struct Remove {
    Remove();
    ~Remove();
    Remove(Remove&& other);
    Remove& operator=(Remove&& other);

    // Contains the list of tabs removed with their indexes at the time of
    // removal along with flag `remove_reason` that indicates i the tab will be
    // deleted or not after removing. For example, if we removed elements:
    //
    // Before removal:
    // A B C D E F G
    // 0 1 2 3 4 5 6
    //
    // After removal:
    // A D E G
    // 0 1 2 3
    //
    // If the tabs were removed in the order B, C, F, `contents` would contain:
    // { B, 1 }, { C, 1 }, { F, 3 }
    //
    // But if the tabs were removed in the order F, C, B, then `contents` would
    // contain:
    // { F, 5 }, { C, 2 }, { B, 1 }
    //
    // Therefore all observers which store indices of tabs should update them
    // in the order the tabs appear in `contents`. Observers should  not do
    // index-based queries based on their own internally-stored indices until
    // after processing all of `contents`.
    std::vector<RemovedTab> contents;

    void WriteIntoTrace(perfetto::TracedValue context) const;
  };

  // A tab was moved from `from_index` to `to_index`. This implicitly changes
  // the existing selection model by calling Move(from_index, to_index, 1).
  struct Move {
    raw_ptr<tabs::TabInterface> tab = nullptr;
    raw_ptr<content::WebContents> contents = nullptr;
    int from_index;
    int to_index;

    void WriteIntoTrace(perfetto::TracedValue context) const;
  };

  // The tab was replaced at the specified index. This is invoked when
  // prerendering swaps in a prerendered WebContents or when a tab's WebContents
  // is discarded to save memory.
  struct Replace {
    raw_ptr<tabs::TabInterface> tab = nullptr;
    raw_ptr<content::WebContents> old_contents = nullptr;
    raw_ptr<content::WebContents> new_contents = nullptr;
    int index;

    void WriteIntoTrace(perfetto::TracedValue context) const;
  };

  TabStripModelChange();
  explicit TabStripModelChange(Insert delta);
  explicit TabStripModelChange(Remove delta);
  explicit TabStripModelChange(Replace delta);
  explicit TabStripModelChange(Move delta);
  TabStripModelChange(const TabStripModelChange&) = delete;
  TabStripModelChange& operator=(const TabStripModelChange&) = delete;
  ~TabStripModelChange();

  Type type() const { return type_; }
  const Insert* GetInsert() const;
  const Remove* GetRemove() const;
  const Move* GetMove() const;
  const Replace* GetReplace() const;

  void WriteIntoTrace(perfetto::TracedValue context) const;

 private:
  using Delta = std::variant<Insert, Remove, Move, Replace>;

  TabStripModelChange(Type type, Delta delta);

  const Type type_ = kSelectionOnly;

  Delta delta_;
};

// Struct to carry changes on selection/activation.
struct TabStripSelectionChange {
  TabStripSelectionChange();
  TabStripSelectionChange(const TabStripSelectionChange& other);
  ~TabStripSelectionChange();

  TabStripSelectionChange& operator=(const TabStripSelectionChange& other);

  // Fill TabStripSelectionChange with given |contents| and |selection_model|.
  // note that |new_contents| and |new_model| will be filled too so that
  // selection_changed() and active_tab_changed() won't return true.
  TabStripSelectionChange(tabs::TabInterface* tab,
                          const ui::ListSelectionModel& model);

  bool active_tab_changed() const {
    // This could be `old_tab != new_tab`, except for tab discarding, where
    // it's the same tab with different contents. Some observers want to
    // treat tab discarding as a selection change, e.g. to update their
    // observations.
    return old_contents != new_contents;
  }

  // TODO(sangwoo.ko) Do we need something to indicate that the change
  // was made implicitly?
  bool selection_changed() const {
    return selected_tabs_were_removed || old_model != new_model;
  }

  raw_ptr<tabs::TabInterface> old_tab = nullptr;
  raw_ptr<tabs::TabInterface> new_tab = nullptr;

  raw_ptr<content::WebContents> old_contents = nullptr;
  raw_ptr<content::WebContents> new_contents = nullptr;

  ui::ListSelectionModel old_model;
  ui::ListSelectionModel new_model;

  bool selected_tabs_were_removed = false;

  int reason = 0;
};

// Struct to carry changes to tab groups. The tab group model is independent of
// the tab strip model, so these changes are not bundled with
// TabStripModelChanges or TabStripSelectionChanges.
struct TabGroupChange {
  // A group is created when the first tab is added to it and closed when the
  // last tab is removed from it. Whenever the set of tabs in the group changes,
  // a kContentsChange event is fired. Whenever the group's visual data changes,
  // such as its title or color, a kVisualsChange event is fired. Whenever the
  // group is moved by interacting with its header, a kMoved event is fired.
  enum Type {
    kCreated,
    kEditorOpened,
    kVisualsChanged,
    kMoved,
    kClosed
  };

  enum class TabGroupCreationReason {
    kNewGroupCreated,
    kInsertedFromAnotherTabstrip
  };

  enum class TabGroupClosureReason { kGroupClosed, kDetachedToAnotherTabstrip };

  // Base class for all changes. Similar to TabStripModelChange::Delta.
  struct Delta {
    virtual ~Delta() = default;
  };

  // The TabGroupVisualData that was changed at the specified group.
  struct VisualsChange : public Delta {
    VisualsChange();
    ~VisualsChange() override;
    raw_ptr<const tab_groups::TabGroupVisualData> old_visuals = nullptr;
    raw_ptr<const tab_groups::TabGroupVisualData> new_visuals = nullptr;
  };

  struct CreateChange : public Delta {
    CreateChange(TabGroupCreationReason reason,
                 tabs::TabGroupTabCollection* detached_group);
    ~CreateChange() override;

    TabGroupCreationReason reason() const { return reason_; }
    std::vector<tabs::TabInterface*> GetDetachedTabs() const;

   private:
    TabGroupCreationReason reason_;
    raw_ptr<tabs::TabGroupTabCollection> detached_group_;
  };

  struct CloseChange : public Delta {
    CloseChange(TabGroupClosureReason reason,
                tabs::TabGroupTabCollection* detached_group);
    ~CloseChange() override;

    TabGroupClosureReason reason() const { return reason_; }
    std::vector<tabs::TabInterface*> GetDetachedTabs() const;

   private:
    TabGroupClosureReason reason_;
    raw_ptr<tabs::TabGroupTabCollection> detached_group_;
  };

  TabGroupChange(TabStripModel* model,
                 tab_groups::TabGroupId group,
                 Type type,
                 std::unique_ptr<Delta> deltap = nullptr);
  TabGroupChange(TabStripModel* model,
                 tab_groups::TabGroupId group,
                 VisualsChange deltap);
  TabGroupChange(TabStripModel* model,
                 tab_groups::TabGroupId group,
                 CreateChange deltap);
  TabGroupChange(TabStripModel* model,
                 tab_groups::TabGroupId group,
                 CloseChange deltap);

  ~TabGroupChange();

  const VisualsChange* GetVisualsChange() const;
  const CreateChange* GetCreateChange() const;
  const CloseChange* GetCloseChange() const;

  tab_groups::TabGroupId group;
  raw_ptr<TabStripModel> model;
  Type type;

 private:
  std::unique_ptr<Delta> delta;
};

struct SplitTabChange {
  enum class Type { kAdded, kVisualsChanged, kContentsChanged, kRemoved };

  enum class SplitTabAddReason {
    kNewSplitTabAdded,
    kSplitTabUpdated,
    kInsertedFromAnotherTabstrip
  };

  enum class SplitTabRemoveReason {
    kSplitTabRemoved,
    kSplitTabUpdated,
    kDetachedToAnotherTabstrip
  };

  // Base class for all changes. Similar to TabStripModelChange::Delta.
  struct Delta {
    virtual ~Delta() = default;
  };

  struct AddedChange : public Delta {
    AddedChange(const std::vector<std::pair<tabs::TabInterface*, int>>& tabs,
                SplitTabAddReason reason,
                const split_tabs::SplitTabVisualData& visual_data);
    ~AddedChange() override;
    AddedChange(const AddedChange&);

    const std::vector<std::pair<tabs::TabInterface*, int>>& tabs() const {
      return tabs_;
    }
    const split_tabs::SplitTabVisualData& visual_data() const {
      return visual_data_;
    }
    SplitTabAddReason reason() const { return reason_; }

   private:
    std::vector<std::pair<tabs::TabInterface*, int>> tabs_;
    SplitTabAddReason reason_;
    split_tabs::SplitTabVisualData visual_data_;
  };

  struct VisualsChange : public Delta {
    VisualsChange(const split_tabs::SplitTabVisualData& old_visual_data,
                  const split_tabs::SplitTabVisualData& new_visual_data);
    ~VisualsChange() override;

    const split_tabs::SplitTabVisualData& old_visual_data() const {
      return old_visual_data_;
    }
    const split_tabs::SplitTabVisualData& new_visual_data() const {
      return new_visual_data_;
    }

   private:
    split_tabs::SplitTabVisualData old_visual_data_;
    split_tabs::SplitTabVisualData new_visual_data_;
  };

  struct ContentsChange : public Delta {
    ContentsChange(
        const std::vector<std::pair<tabs::TabInterface*, int>>& prev_tabs,
        const std::vector<std::pair<tabs::TabInterface*, int>>& new_tabs);
    ~ContentsChange() override;
    ContentsChange(const ContentsChange&);

    const std::vector<std::pair<tabs::TabInterface*, int>>& prev_tabs() const {
      return prev_tabs_;
    }
    const std::vector<std::pair<tabs::TabInterface*, int>>& new_tabs() const {
      return new_tabs_;
    }

   private:
    std::vector<std::pair<tabs::TabInterface*, int>> prev_tabs_;
    std::vector<std::pair<tabs::TabInterface*, int>> new_tabs_;
  };

  struct RemovedChange : public Delta {
    RemovedChange(const std::vector<std::pair<tabs::TabInterface*, int>>& tabs,
                  SplitTabRemoveReason reason);
    ~RemovedChange() override;
    RemovedChange(const RemovedChange&);

    const std::vector<std::pair<tabs::TabInterface*, int>>& tabs() const {
      return tabs_;
    }
    SplitTabRemoveReason reason() const { return reason_; }

   private:
    std::vector<std::pair<tabs::TabInterface*, int>> tabs_;
    SplitTabRemoveReason reason_;
  };

  SplitTabChange(TabStripModel* model,
                 split_tabs::SplitTabId split_id,
                 Type type,
                 std::unique_ptr<Delta> deltap);
  SplitTabChange(TabStripModel* model,
                 split_tabs::SplitTabId split_id,
                 AddedChange deltap);
  SplitTabChange(TabStripModel* model,
                 split_tabs::SplitTabId split_id,
                 VisualsChange deltap);
  SplitTabChange(TabStripModel* model,
                 split_tabs::SplitTabId split_id,
                 ContentsChange deltap);
  SplitTabChange(TabStripModel* model,
                 split_tabs::SplitTabId split_id,
                 RemovedChange deltap);

  ~SplitTabChange();

  const AddedChange* GetAddedChange() const;
  const VisualsChange* GetVisualsChange() const;
  const ContentsChange* GetContentsChange() const;
  const RemovedChange* GetRemovedChange() const;

  split_tabs::SplitTabId split_id;
  raw_ptr<TabStripModel> model;
  Type type;

 private:
  std::unique_ptr<Delta> delta;
};

////////////////////////////////////////////////////////////////////////////////
//
// TabStripModelObserver
//
//  Objects implement this interface when they wish to be notified of changes
//  to the TabStripModel.
//
//  Two major implementers are the TabStrip, which uses notifications sent
//  via this interface to update the presentation of the strip, and the Browser
//  object, which updates bookkeeping and shows/hides individual WebContentses.
//
//  Register your TabStripModelObserver with the TabStripModel using its
//  Add/RemoveObserver methods.
//
////////////////////////////////////////////////////////////////////////////////
class TabStripModelObserver {
 public:
  enum ChangeReason {
    // Used to indicate that none of the reasons below are responsible for the
    // active tab change.
    CHANGE_REASON_NONE = 0,
    // The active tab changed because the tab's web contents was replaced.
    CHANGE_REASON_REPLACED = 1 << 0,
    // The active tab changed due to a user input event.
    CHANGE_REASON_USER_GESTURE = 1 << 1,
  };

  enum CloseAllStoppedReason {
    // Used to indicate that CloseAllTab event is canceled.
    kCloseAllCanceled = 0,
    // Used to indicate that CloseAllTab event complete successfully.
    kCloseAllCompleted = 1,
  };

  TabStripModelObserver(const TabStripModelObserver&) = delete;
  TabStripModelObserver& operator=(const TabStripModelObserver&) = delete;

  // |change| is a series of changes in tabstrip model. |change| consists
  // of changes with same type and those changes may have caused selection or
  // activation changes. |selection| is determined by comparing the state of
  // TabStripModel before the |change| and after the |change| are applied.
  // When only selection/activation was changed without any change about
  // WebContents, |change| can be empty.
  virtual void OnTabStripModelChanged(TabStripModel* tab_strip_model,
                                      const TabStripModelChange& change,
                                      const TabStripSelectionChange& selection);

  // Notification that a tab will be added to the TabStripModel, which allows
  // an observer to react to an impending change to the TabStripModel. The only
  // use case of this signal that is currently supported is the drag controller
  // cancelling/completing a the drag before a tab is added during header drag.
  virtual void OnTabWillBeAdded();

  // Notification that the tab at |index| will be removed from the
  // TabStripModel, which allows an observer to react to an impending change to
  // the TabStripModel. The only use case of this signal that is currently
  // supported is the drag controller completing a drag before a tab is removed.
  // TODO(crbug.com/40838330): Unify and generalize this and OnTabWillBeAdded,
  // e.g. via OnTabStripModelWillChange().
  virtual void OnTabWillBeRemoved(content::WebContents* contents, int index);

  // |change| is a change in the Tab Group model or metadata. These
  // changes may cause repainting of some Tab Group UI. They are
  // independent of the tabstrip model and do not affect any tab state.
  virtual void OnTabGroupChanged(const TabGroupChange& change);

  // Notfies us when a Tab Group is added to the Tab Group Model.
  virtual void OnTabGroupAdded(const tab_groups::TabGroupId& group_id);

  // Notfies us when a Tab Group will be removed from the Tab Group Model.
  virtual void OnTabGroupWillBeRemoved(const tab_groups::TabGroupId& group_id);

  // Notifies us when there is a change to split tab state in the TabStripModel.
  // The |change| provides details of the change to split tab.
  virtual void OnSplitTabChanged(const SplitTabChange& change);

  // The specified WebContents at |index| changed in some way. |contents|
  // may be an entirely different object and the old value is no longer
  // available by the time this message is delivered.
  //
  // See tab_change_type.h for a description of |change_type|.
  virtual void TabChangedAt(content::WebContents* contents,
                            int index,
                            TabChangeType change_type);

  // Invoked when the pinned state of a tab changes.
  virtual void TabPinnedStateChanged(TabStripModel* tab_strip_model,
                                     content::WebContents* contents,
                                     int index);

  // Invoked when the blocked state of a tab changes.
  // NOTE: This is invoked when a tab becomes blocked/unblocked by a tab modal
  // window.
  virtual void TabBlockedStateChanged(content::WebContents* contents,
                                      int index);

  // Called when the tab at `index` is added to the group with id `new_group` or
  // removed from a group with id `old_group`.
  virtual void TabGroupedStateChanged(
      TabStripModel* tab_strip_model,
      std::optional<tab_groups::TabGroupId> old_group,
      std::optional<tab_groups::TabGroupId> new_group,
      tabs::TabInterface* tab,
      int index);

  // The TabStripModel now no longer has any tabs. The implementer may
  // use this as a trigger to try and close the window containing the
  // TabStripModel, for example...
  virtual void TabStripEmpty();

  // Called when a tab is attempted to be closed but the closure is not
  // permitted by the `TabStripModel::IsTabClosable` oracle.
  virtual void TabCloseCancelled(const content::WebContents* contents);

  // Sent any time an attempt is made to close all the tabs. This is not
  // necessarily the result of CloseAllTabs(). For example, if the user closes
  // the last tab WillCloseAllTabs() is sent. If the close does not succeed
  // during the current event (say unload handlers block it) then
  // CloseAllTabsStopped() is sent with reason 'CANCELED'. On the other hand if
  // the close does finish then CloseAllTabsStopped() is sent with reason
  // 'COMPLETED'. Also note that if the last tab is detached
  // (DetachAndDeleteWebContentsAt()) then
  // this is not sent.
  virtual void WillCloseAllTabs(TabStripModel* tab_strip_model);
  virtual void CloseAllTabsStopped(TabStripModel* tab_strip_model,
                                   CloseAllStoppedReason reason);

  // The specified tab at |index| requires the display of a UI indication to the
  // user that it needs their attention. The UI indication is set iff
  // |attention| is true.
  virtual void SetTabNeedsAttentionAt(int index, bool attention);

  // Called when an observed TabStripModel is beginning destruction.
  virtual void OnTabStripModelDestroyed(TabStripModel* tab_strip_model);

  static void StopObservingAll(TabStripModelObserver* observer);
  static bool IsObservingAny(TabStripModelObserver* observer);
  static int CountObservedModels(TabStripModelObserver* observer);

  // A passkey for TabStripModel to access some methods on this class - see
  // </docs/patterns/passkey.md>.
  class ModelPasskey {
   private:
    friend class TabStripModel;
    ModelPasskey() = default;
    ~ModelPasskey() = default;
  };

  // These methods are used by TabStripModel to notify this class of lifecycle
  // events on the TabStripModelObserver or the TabStripModel itself. The first
  // two are used to allow TabStripModelObserver to track which models it is
  // observing. The third is used to allow TabStripModelObserver to clean up
  // when an observed TabStripModel is destroyed, and to send the
  // OnTabStripModelDestroyed notification above.
  void StartedObserving(ModelPasskey, TabStripModel* model);
  void StoppedObserving(ModelPasskey, TabStripModel* model);
  void ModelDestroyed(ModelPasskey, TabStripModel* model);

 protected:
  TabStripModelObserver();
  virtual ~TabStripModelObserver();

 private:
  std::set<raw_ptr<TabStripModel, SetExperimental>> observed_models_;
};

#endif  // CHROME_BROWSER_UI_TABS_TAB_STRIP_MODEL_OBSERVER_H_