| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/infobars/infobar_responder.h" | 5 #include "chrome/browser/infobars/infobar_responder.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 #include "chrome/browser/infobars/infobar_service.h" | 9 #include "chrome/browser/infobars/infobar_service.h" |
| 10 #include "components/infobars/core/confirm_infobar_delegate.h" | 10 #include "components/infobars/core/confirm_infobar_delegate.h" |
| 11 #include "components/infobars/core/infobar.h" | 11 #include "components/infobars/core/infobar.h" |
| 12 | 12 |
| 13 InfoBarResponder::InfoBarResponder(InfoBarService* infobar_service, | 13 InfoBarResponder::InfoBarResponder(InfoBarService* infobar_service, |
| 14 AutoResponseType response) | 14 bool should_accept) |
| 15 : infobar_service_(infobar_service), response_(response) { | 15 : infobar_service_(infobar_service), should_accept_(should_accept) { |
| 16 infobar_service_->AddObserver(this); | 16 infobar_service_->AddObserver(this); |
| 17 } | 17 } |
| 18 | 18 |
| 19 InfoBarResponder::~InfoBarResponder() { | 19 InfoBarResponder::~InfoBarResponder() { |
| 20 // This is safe even if we were already removed as an observer in | 20 // This is safe even if we were already removed as an observer in |
| 21 // OnInfoBarAdded(). | 21 // OnInfoBarAdded(). |
| 22 infobar_service_->RemoveObserver(this); | 22 infobar_service_->RemoveObserver(this); |
| 23 } | 23 } |
| 24 | 24 |
| 25 void InfoBarResponder::OnInfoBarAdded(infobars::InfoBar* infobar) { | 25 void InfoBarResponder::OnInfoBarAdded(infobars::InfoBar* infobar) { |
| 26 infobar_service_->RemoveObserver(this); | 26 infobar_service_->RemoveObserver(this); |
| 27 ConfirmInfoBarDelegate* delegate = | 27 ConfirmInfoBarDelegate* delegate = |
| 28 infobar->delegate()->AsConfirmInfoBarDelegate(); | 28 infobar->delegate()->AsConfirmInfoBarDelegate(); |
| 29 DCHECK(delegate); | 29 DCHECK(delegate); |
| 30 | 30 |
| 31 base::MessageLoop::current()->PostTask( | 31 base::MessageLoop::current()->PostTask( |
| 32 FROM_HERE, | 32 FROM_HERE, |
| 33 base::Bind(&InfoBarResponder::Respond, base::Unretained(this), delegate)); | 33 base::Bind(&InfoBarResponder::Respond, base::Unretained(this), delegate)); |
| 34 } | 34 } |
| 35 | 35 |
| 36 void InfoBarResponder::OnInfoBarReplaced(infobars::InfoBar* old_infobar, | 36 void InfoBarResponder::Respond(ConfirmInfoBarDelegate* delegate) { |
| 37 infobars::InfoBar* new_infobar) { | 37 if (should_accept_) |
| 38 OnInfoBarAdded(new_infobar); | 38 delegate->Accept(); |
| 39 else |
| 40 delegate->Cancel(); |
| 39 } | 41 } |
| 40 | |
| 41 void InfoBarResponder::Respond(ConfirmInfoBarDelegate* delegate) { | |
| 42 switch (response_) { | |
| 43 case ACCEPT: | |
| 44 delegate->Accept(); | |
| 45 break; | |
| 46 case DENY: | |
| 47 delegate->Cancel(); | |
| 48 break; | |
| 49 case DISMISS: | |
| 50 delegate->InfoBarDismissed(); | |
| 51 break; | |
| 52 } | |
| 53 } | |
| OLD | NEW |