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
14namespace {
15
16// Returns true if |browser| is the only normal (tabbed) browser for |browser|'s
17// profile (across all desktops).
18bool 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
30PinnedTabService::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
44void 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