src/chrome/browser/ui/bookmarks/bookmark_tab_helper.cc

   1 // Copyright (c) 2012 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 #include "chrome/browser/ui/bookmarks/bookmark_tab_helper.h"
   6
   7 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
   8 #include "chrome/browser/defaults.h"
   9 #include "chrome/browser/prefs/pref_service_syncable.h"
  10 #include "chrome/browser/profiles/profile.h"
  11 #include "chrome/browser/search/search.h"
  12 #include "chrome/browser/ui/bookmarks/bookmark_tab_helper_delegate.h"
  13 #include "chrome/browser/ui/bookmarks/bookmark_utils.h"
  14 #include "chrome/browser/ui/sad_tab.h"
  15 #include "chrome/browser/ui/webui/ntp/new_tab_ui.h"
  16 #include "chrome/common/pref_names.h"
  17 #include "components/bookmarks/core/browser/bookmark_model.h"
  18 #include "content/public/browser/navigation_entry.h"
  19 #include "content/public/browser/web_contents.h"
  20
  21 namespace {
  22
  23 bool IsNTPWebUI(content::WebContents* web_contents) {
  24   content::WebUI* web_ui = NULL;
  25   // Use the committed entry so the bookmarks bar disappears at the same time
  26   // the page does.
  27   if (web_contents->GetController().GetLastCommittedEntry())
  28     web_ui = web_contents->GetCommittedWebUI();
  29   else
  30     web_ui = web_contents->GetWebUI();
  31   return web_ui && NewTabUI::FromWebUIController(web_ui->GetController());
  32 }
  33
  34 bool IsInstantNTP(content::WebContents* web_contents) {
  35   // Use the committed entry so the bookmarks bar disappears at the same time
  36   // the page does.
  37   const content::NavigationEntry* entry =
  38       web_contents->GetController().GetLastCommittedEntry();
  39   if (!entry)
  40     entry = web_contents->GetController().GetVisibleEntry();
  41   return chrome::NavEntryIsInstantNTP(web_contents, entry);
  42 }
  43
  44 }  // namespace
  45
  46 DEFINE_WEB_CONTENTS_USER_DATA_KEY(BookmarkTabHelper);
  47
  48 BookmarkTabHelper::~BookmarkTabHelper() {
  49   if (bookmark_model_)
  50     bookmark_model_->RemoveObserver(this);
  51 }
  52
  53 bool BookmarkTabHelper::ShouldShowBookmarkBar() const {
  54   if (web_contents()->ShowingInterstitialPage())
  55     return false;
  56
  57   if (chrome::SadTab::ShouldShow(web_contents()->GetCrashedStatus()))
  58     return false;
  59
  60   if (!browser_defaults::bookmarks_enabled)
  61     return false;
  62
  63   Profile* profile =
  64       Profile::FromBrowserContext(web_contents()->GetBrowserContext());
  65
  66 #if !defined(OS_CHROMEOS)
  67   if (profile->IsGuestSession())
  68     return false;
  69 #endif
  70
  71   PrefService* prefs = profile->GetPrefs();
  72   if (prefs->IsManagedPreference(prefs::kShowBookmarkBar) &&
  73       !prefs->GetBoolean(prefs::kShowBookmarkBar))
  74     return false;
  75
  76   return IsNTPWebUI(web_contents()) || IsInstantNTP(web_contents());
  77 }
  78
  79 BookmarkTabHelper::BookmarkTabHelper(content::WebContents* web_contents)
  80     : content::WebContentsObserver(web_contents),
  81       is_starred_(false),
  82       bookmark_model_(NULL),
  83       delegate_(NULL),
  84       bookmark_drag_(NULL) {
  85   Profile* profile =
  86       Profile::FromBrowserContext(web_contents->GetBrowserContext());
  87   bookmark_model_= BookmarkModelFactory::GetForProfile(profile);
  88   if (bookmark_model_)
  89     bookmark_model_->AddObserver(this);
  90 }
  91
  92 void BookmarkTabHelper::UpdateStarredStateForCurrentURL() {
  93   const bool old_state = is_starred_;
  94   is_starred_ = (bookmark_model_ &&
  95       bookmark_model_->IsBookmarked(chrome::GetURLToBookmark(web_contents())));
  96
  97   if (is_starred_ != old_state && delegate_)
  98     delegate_->URLStarredChanged(web_contents(), is_starred_);
  99 }
 100
 101 void BookmarkTabHelper::BookmarkModelChanged() {
 102 }
 103
 104 void BookmarkTabHelper::BookmarkModelLoaded(BookmarkModel* model,
 105                                             bool ids_reassigned) {
 106   UpdateStarredStateForCurrentURL();
 107 }
 108
 109 void BookmarkTabHelper::BookmarkNodeAdded(BookmarkModel* model,
 110                                           const BookmarkNode* parent,
 111                                           int index) {
 112   UpdateStarredStateForCurrentURL();
 113 }
 114
 115 void BookmarkTabHelper::BookmarkNodeRemoved(
 116     BookmarkModel* model,
 117     const BookmarkNode* parent,
 118     int old_index,
 119     const BookmarkNode* node,
 120     const std::set<GURL>& removed_urls) {
 121   UpdateStarredStateForCurrentURL();
 122 }
 123
 124 void BookmarkTabHelper::BookmarkAllNodesRemoved(
 125     BookmarkModel* model,
 126     const std::set<GURL>& removed_urls) {
 127   UpdateStarredStateForCurrentURL();
 128 }
 129
 130 void BookmarkTabHelper::BookmarkNodeChanged(BookmarkModel* model,
 131                                             const BookmarkNode* node) {
 132   UpdateStarredStateForCurrentURL();
 133 }
 134
 135 void BookmarkTabHelper::DidNavigateMainFrame(
 136     const content::LoadCommittedDetails& /*details*/,
 137     const content::FrameNavigateParams& /*params*/) {
 138   UpdateStarredStateForCurrentURL();
 139 }