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

#ifndef UI_VIEWS_VIEW_TRACKER_H_
#define UI_VIEWS_VIEW_TRACKER_H_

#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/scoped_observation.h"
#include "ui/views/view_observer.h"
#include "ui/views/views_export.h"

namespace views {

class View;

// ViewTracker tracks a single View. When the View is deleted it's removed.
class VIEWS_EXPORT ViewTracker : public ViewObserver {
 public:
  explicit ViewTracker(View* view = nullptr);
  ViewTracker(const ViewTracker&) = delete;
  ViewTracker& operator=(const ViewTracker&) = delete;
  ~ViewTracker() override;

  void SetView(View* view);
  View* view() { return view_; }
  const View* view() const { return view_; }
  // If `track_entire_view_hierarchy_` is true this will be run before
  // destruction of the view hierarchy. If `track_entire_view_hierarchy_` is
  // false the callback will be run right before the view itself will be deleted
  // (child views will already be destroyed at that point).
  void SetIsDeletingCallback(base::OnceClosure is_deleting_callback);
  // If set to true, the view tracker will remove the view it is tracking before
  // destruction of the view hierarchy, including child views, starts.
  void SetTrackEntireViewHierarchy(bool track_entire_view_hierarchy) {
    track_entire_view_hierarchy_ = track_entire_view_hierarchy;
  }

  operator bool() const { return !!view_; }

  // ViewObserver:
  void OnViewIsDeleting(View* observed_view) override;
  void OnViewHierarchyWillBeDeleted(View* observed_view) override;

 private:
  void HandleViewDestruction();

  raw_ptr<View> view_ = nullptr;

  base::OnceClosure is_deleting_callback_;

  bool track_entire_view_hierarchy_ = false;

  base::ScopedObservation<View, ViewObserver> observation_{this};
};

}  // namespace views

#endif  // UI_VIEWS_VIEW_TRACKER_H_