Chromium Code Reviews| Index: ios/web/web_state/web_state_observer.cc |
| diff --git a/ios/web/web_state/web_state_observer.cc b/ios/web/web_state/web_state_observer.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..c5c1ab79fa2dc475c7004a9fabead8f937c0139b |
| --- /dev/null |
| +++ b/ios/web/web_state/web_state_observer.cc |
| @@ -0,0 +1,42 @@ |
| +// Copyright 2014 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 "ios/web/public/web_state/web_state_observer.h" |
| + |
| +#include "ios/public/provider/web/web_state.h" |
| +#include "ios/web/public/load_committed_details.h" |
| + |
| +namespace web { |
| + |
| +WebStateObserver::WebStateObserver(ios::WebState* web_state) |
| + : web_state_(nullptr) { |
| + Observe(web_state); |
| +} |
| + |
| +WebStateObserver::WebStateObserver() : web_state_(nullptr) { |
| +} |
| + |
| +WebStateObserver::~WebStateObserver() { |
| + if (web_state_) |
| + web_state_->RemoveObserver(this); |
| +} |
| + |
| +void WebStateObserver::Observe(ios::WebState* web_state) { |
| + if (web_state == web_state_) { |
| + // Early exit to avoid infinite loops if we're in the middle of a callback. |
| + return; |
| + } |
| + if (web_state_) |
| + web_state_->RemoveObserver(this); |
| + web_state_ = web_state; |
| + if (web_state_) |
| + web_state_->AddObserver(this); |
| +} |
| + |
| +void WebStateObserver::ResetWebState() { |
| + web_state_->RemoveObserver(this); |
| + web_state_ = NULL; |
|
stuartmorgan
2014/12/04 23:22:58
Could you change this to nullptr?
|
| +} |
| + |
| +} // namespace web |