Home | History | Annotate | Download | only in browser 
      1 // Copyright 2014 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef COMPONENTS_GOOGLE_CORE_BROWSER_GOOGLE_URL_TRACKER_INFOBAR_DELEGATE_H_
      6 #define COMPONENTS_GOOGLE_CORE_BROWSER_GOOGLE_URL_TRACKER_INFOBAR_DELEGATE_H_
      7 
      8 #include "components/infobars/core/confirm_infobar_delegate.h"
      9 #include "url/gurl.h"
     10 
     11 class GoogleURLTracker;
     12 class GoogleURLTrackerNavigationHelper;
     13 
     14 namespace infobars {
     15 class InfoBarManager;
     16 }
     17 
     18 // This infobar is shown by the GoogleURLTracker when the Google base URL has
     19 // changed.
     20 class GoogleURLTrackerInfoBarDelegate : public ConfirmInfoBarDelegate {
     21  public:
     22   // Creates a Google URL tracker infobar and delegate and adds the infobar to
     23   // |infobar_manager|.  Returns the infobar if it was successfully added.
     24   static infobars::InfoBar* Create(
     25       infobars::InfoBarManager* infobar_manager,
     26       GoogleURLTracker* google_url_tracker,
     27       const GURL& search_url);
     28 
     29   // ConfirmInfoBarDelegate:
     30   virtual bool Accept() OVERRIDE;
     31   virtual bool Cancel() OVERRIDE;
     32 
     33   GoogleURLTrackerNavigationHelper* navigation_helper() {
     34     return navigation_helper_weak_ptr_;
     35   }
     36 
     37   void set_navigation_helper(
     38       scoped_ptr<GoogleURLTrackerNavigationHelper> navigation_helper) {
     39     navigation_helper_ = navigation_helper.Pass();
     40     navigation_helper_weak_ptr_ = navigation_helper_.get();
     41   }
     42 
     43   // Other than set_pending_id(), these accessors are only used by test code.
     44   const GURL& search_url() const { return search_url_; }
     45   void set_search_url(const GURL& search_url) { search_url_ = search_url; }
     46   int pending_id() const { return pending_id_; }
     47   void set_pending_id(int pending_id) { pending_id_ = pending_id; }
     48 
     49   // These are virtual so test code can override them in a subclass.
     50   virtual void Update(const GURL& search_url);
     51   virtual void Close(bool redo_search);
     52 
     53  protected:
     54   GoogleURLTrackerInfoBarDelegate(
     55       GoogleURLTracker* google_url_tracker,
     56       const GURL& search_url);
     57   virtual ~GoogleURLTrackerInfoBarDelegate();
     58 
     59  private:
     60   // ConfirmInfoBarDelegate:
     61   virtual base::string16 GetMessageText() const OVERRIDE;
     62   virtual base::string16 GetButtonLabel(InfoBarButton button) const OVERRIDE;
     63   virtual base::string16 GetLinkText() const OVERRIDE;
     64   virtual bool LinkClicked(WindowOpenDisposition disposition) OVERRIDE;
     65   virtual bool ShouldExpireInternal(
     66       const NavigationDetails& details) const OVERRIDE;
     67 
     68   GoogleURLTracker* google_url_tracker_;
     69   scoped_ptr<GoogleURLTrackerNavigationHelper> navigation_helper_;
     70 
     71   // During Close(), this object gives up ownership of |navigation_helper_|,
     72   // which then outlives this object. Sometimes after this point, other classes
     73   // still attempt to call navigation_helper() to access the (still-valid)
     74   // instance. The NavigationHelper instance is stored as a weak pointer in
     75   // addition to a strong pointer to facilitate this case.
     76   GoogleURLTrackerNavigationHelper* navigation_helper_weak_ptr_;
     77   GURL search_url_;
     78   int pending_id_;
     79 
     80   DISALLOW_COPY_AND_ASSIGN(GoogleURLTrackerInfoBarDelegate);
     81 };
     82 
     83 #endif  // COMPONENTS_GOOGLE_CORE_BROWSER_GOOGLE_URL_TRACKER_INFOBAR_DELEGATE_H_
     84