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/tabs/pinned_tab_service.h" 6 7 #include "chrome/browser/chrome_notification_types.h" 8 #include "chrome/browser/profiles/profile.h" 9 #include "chrome/browser/ui/browser.h" 10 #include "chrome/browser/ui/browser_iterator.h" 11 #include "chrome/browser/ui/tabs/pinned_tab_codec.h" 12 #include "content/public/browser/notification_service.h" 13 14 namespace { 15 16 // Returns true if |browser| is the only normal (tabbed) browser for |browser|'s 17 // profile (across all desktops). 18 bool IsOnlyNormalBrowser(Browser* browser) { 19 for (chrome::BrowserIterator it; !it.done(); it.Next()) { 20 if (*it != browser && it->is_type_tabbed() && 21 it->profile() == browser->profile()) { 22 return false; 23 } 24 } 25 return true; 26 } 27 28 } // namespace 29 30 PinnedTabService::PinnedTabService(Profile* profile) 31 : profile_(profile), 32 save_pinned_tabs_(true), 33 has_normal_browser_(false) { 34 registrar_.Add(this, chrome::NOTIFICATION_BROWSER_OPENED, 35 content::NotificationService::AllBrowserContextsAndSources()); 36 registrar_.Add(this, chrome::NOTIFICATION_BROWSER_CLOSING, 37 content::NotificationService::AllSources()); 38 registrar_.Add(this, chrome::NOTIFICATION_CLOSE_ALL_BROWSERS_REQUEST, 39 content::NotificationService::AllSources()); 40 registrar_.Add(this, chrome::NOTIFICATION_TAB_ADDED, 41 content::NotificationService::AllSources()); 42 } 43 44 void PinnedTabService::Observe(int type, 45 const content::NotificationSource& source, 46 const content::NotificationDetails& details) { 47 // Saving of tabs happens when saving is enabled, and when either the user 48 // exits the application or closes the last browser window. 49 // Saving is disabled when the user exits the application to prevent the 50 // pin state of all the open browsers being overwritten by the state of the 51 // last browser window to close. 52 // Saving is re-enabled when a browser window or tab is opened again. 53 // Note, cancelling a shutdown (via onbeforeunload) will not re-enable pinned 54 // tab saving immediately, to prevent the following situation: 55 // * two windows are open, one with pinned tabs 56 // * user exits 57 // * pinned tabs are saved 58 // * window with pinned tabs is closed 59 // * other window blocks close with onbeforeunload 60 // * user saves work, etc. then closes the window 61 // * pinned tabs are saved, without the window with the pinned tabs, 62 // over-writing the correct state. 63 // Saving is re-enabled if a new tab or window is opened. 64 switch (type) { 65 case chrome::NOTIFICATION_BROWSER_OPENED: { 66 Browser* browser = content::Source<Browser>(source).ptr(); 67 if (!has_normal_browser_ && browser->is_type_tabbed() && 68 browser->profile() == profile_) { 69 has_normal_browser_ = true; 70 } 71 save_pinned_tabs_ = true; 72 break; 73 } 74 75 case chrome::NOTIFICATION_TAB_ADDED: { 76 save_pinned_tabs_ = true; 77 break; 78 } 79 80 case chrome::NOTIFICATION_BROWSER_CLOSING: { 81 Browser* browser = content::Source<Browser>(source).ptr(); 82 if (has_normal_browser_ && save_pinned_tabs_ && 83 browser->profile() == profile_) { 84 if (IsOnlyNormalBrowser(browser)) { 85 has_normal_browser_ = false; 86 PinnedTabCodec::WritePinnedTabs(profile_); 87 } 88 } 89 break; 90 } 91 92 case chrome::NOTIFICATION_CLOSE_ALL_BROWSERS_REQUEST: { 93 if (has_normal_browser_ && save_pinned_tabs_) { 94 PinnedTabCodec::WritePinnedTabs(profile_); 95 save_pinned_tabs_ = false; 96 } 97 break; 98 } 99 100 default: 101 NOTREACHED(); 102 } 103 } 104