ui/views/view_tracker.cc

// 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.

#include "ui/views/view_tracker.h"

#include <utility>

#include "base/functional/callback.h"
#include "ui/views/view.h"

namespace views {

ViewTracker::ViewTracker(View* view) {
  SetView(view);
}

ViewTracker::~ViewTracker() = default;

void ViewTracker::SetView(View* view) {
  if (view == view_)
    return;

  observation_.Reset();
  view_ = view;
  if (view_)
    observation_.Observe(view_.get());
}

void ViewTracker::SetOnViewIsDeletingCallback(
    base::OnceClosure on_view_is_deleting_callback) {
  on_view_is_deleting_callback_ = std::move(on_view_is_deleting_callback);
}

void ViewTracker::OnViewIsDeleting(View* observed_view) {
  // View is already in destructor. Set to nullptr first before running any
  // callbacks.
  SetView(nullptr);
  if (on_view_is_deleting_callback_.is_null()) {
    return;
  }
  std::move(on_view_is_deleting_callback_).Run();
  on_view_is_deleting_callback_.Reset();
}

}  // namespace views