chrome/browser/infobars/infobar_observer.cc

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

#include "chrome/browser/infobars/infobar_observer.h"

InfoBarObserver::InfoBarObserver(infobars::InfoBarManager* manager, Type type)
    : type_(type), infobar_observer_(this) {
  // There may be no |manager| if the browser window is currently closing.
  if (manager)
    infobar_observer_.Add(manager);
}

InfoBarObserver::~InfoBarObserver() {}

void InfoBarObserver::Wait() {
  // When there is no manager being observed, there is nothing to wait on, so
  // return immediately.
  if (infobar_observer_.IsObservingSources())
    run_loop_.Run();
}

void InfoBarObserver::OnInfoBarAdded(infobars::InfoBar* infobar) {
  OnNotified(Type::kInfoBarAdded);
}
void InfoBarObserver::OnInfoBarRemoved(infobars::InfoBar* infobar,
                                       bool animate) {
  OnNotified(Type::kInfoBarRemoved);
}
void InfoBarObserver::OnInfoBarReplaced(infobars::InfoBar* old_infobar,
                                        infobars::InfoBar* new_infobar) {
  OnNotified(Type::kInfoBarReplaced);
}

void InfoBarObserver::OnManagerShuttingDown(infobars::InfoBarManager* manager) {
  if (run_loop_.running())
    run_loop_.Quit();
  infobar_observer_.Remove(manager);
}

void InfoBarObserver::OnNotified(Type type) {
  if (type == type_)
    run_loop_.Quit();
}