src/chrome/browser/ui/tabs/pinned_tab_codec.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/tabs/pinned_tab_codec.h"
   6
   7 #include "base/prefs/pref_service.h"
   8 #include "base/prefs/scoped_user_pref_update.h"
   9 #include "base/values.h"
  10 #include "chrome/browser/extensions/tab_helper.h"
  11 #include "chrome/browser/profiles/profile.h"
  12 #include "chrome/browser/ui/browser.h"
  13 #include "chrome/browser/ui/browser_iterator.h"
  14 #include "chrome/browser/ui/browser_list.h"
  15 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  16 #include "chrome/common/extensions/extension.h"
  17 #include "chrome/common/extensions/manifest_handlers/app_launch_info.h"
  18 #include "chrome/common/pref_names.h"
  19 #include "components/user_prefs/pref_registry_syncable.h"
  20 #include "content/public/browser/navigation_entry.h"
  21 #include "content/public/browser/web_contents.h"
  22
  23 using content::NavigationEntry;
  24
  25 // Key used in dictionaries for the app id.
  26 static const char kAppID[] = "app_id";
  27
  28 // Key used in dictionaries for the url.
  29 static const char kURL[] = "url";
  30
  31 // Returns true if |browser| has any pinned tabs.
  32 static bool HasPinnedTabs(Browser* browser) {
  33   TabStripModel* tab_model = browser->tab_strip_model();
  34   for (int i = 0; i < tab_model->count(); ++i) {
  35     if (tab_model->IsTabPinned(i))
  36       return true;
  37   }
  38   return false;
  39 }
  40
  41 // Adds a DictionaryValue to |values| representing |tab|.
  42 static void EncodeTab(const StartupTab& tab, ListValue* values) {
  43   scoped_ptr<DictionaryValue> value(new DictionaryValue);
  44   value->SetString(kURL, tab.url.spec());
  45   if (tab.is_app)
  46     value->SetString(kAppID, tab.app_id);
  47   values->Append(value.release());
  48 }
  49
  50 // Adds a DictionaryValue to |values| representing the pinned tab at the
  51 // specified index.
  52 static void EncodePinnedTab(TabStripModel* model,
  53                             int index,
  54                             ListValue* values) {
  55   scoped_ptr<DictionaryValue> value(new DictionaryValue());
  56
  57   content::WebContents* web_contents = model->GetWebContentsAt(index);
  58   if (model->IsAppTab(index)) {
  59     const extensions::Extension* extension =
  60         extensions::TabHelper::FromWebContents(web_contents)->extension_app();
  61     DCHECK(extension);
  62     value->SetString(kAppID, extension->id());
  63     // For apps we use the launch url. We do this because the user is
  64     // effectively restarting the app, so returning them to the app's launch
  65     // page seems closest to what they expect.
  66     value->SetString(
  67         kURL, extensions::AppLaunchInfo::GetFullLaunchURL(extension).spec());
  68     values->Append(value.release());
  69   } else {
  70     NavigationEntry* entry = web_contents->GetController().GetActiveEntry();
  71     if (!entry && web_contents->GetController().GetEntryCount())
  72       entry = web_contents->GetController().GetEntryAtIndex(0);
  73     if (entry) {
  74       value->SetString(kURL, entry->GetURL().spec());
  75       values->Append(value.release());
  76     }
  77   }
  78 }
  79
  80 // Invokes EncodePinnedTab for each pinned tab in browser.
  81 static void EncodePinnedTabs(Browser* browser, ListValue* values) {
  82   TabStripModel* tab_model = browser->tab_strip_model();
  83   for (int i = 0; i < tab_model->count() && tab_model->IsTabPinned(i); ++i)
  84     EncodePinnedTab(tab_model, i, values);
  85 }
  86
  87 // Decodes the previously written values in |value| to |tab|, returning true
  88 // on success.
  89 static bool DecodeTab(const DictionaryValue& value, StartupTab* tab) {
  90   tab->is_app = false;
  91
  92   std::string url_string;
  93   if (!value.GetString(kURL, &url_string))
  94     return false;
  95   tab->url = GURL(url_string);
  96
  97   if (value.GetString(kAppID, &(tab->app_id)))
  98     tab->is_app = true;
  99
 100   return true;
 101 }
 102
 103 // static
 104 void PinnedTabCodec::RegisterProfilePrefs(
 105     user_prefs::PrefRegistrySyncable* registry) {
 106   registry->RegisterListPref(prefs::kPinnedTabs,
 107                              user_prefs::PrefRegistrySyncable::UNSYNCABLE_PREF);
 108 }
 109
 110 // static
 111 void PinnedTabCodec::WritePinnedTabs(Profile* profile) {
 112   PrefService* prefs = profile->GetPrefs();
 113   if (!prefs)
 114     return;
 115
 116   ListValue values;
 117   for (chrome::BrowserIterator it; !it.done(); it.Next()) {
 118     Browser* browser = *it;
 119     if (browser->is_type_tabbed() &&
 120         browser->profile() == profile && HasPinnedTabs(browser)) {
 121       EncodePinnedTabs(browser, &values);
 122     }
 123   }
 124   prefs->Set(prefs::kPinnedTabs, values);
 125 }
 126
 127 // static
 128 void PinnedTabCodec::WritePinnedTabs(Profile* profile,
 129                                      const StartupTabs& tabs) {
 130   PrefService* prefs = profile->GetPrefs();
 131   if (!prefs)
 132     return;
 133
 134   ListPrefUpdate update(prefs, prefs::kPinnedTabs);
 135   ListValue* values = update.Get();
 136   values->Clear();
 137   for (StartupTabs::const_iterator i = tabs.begin(); i != tabs.end(); ++i)
 138     EncodeTab(*i, values);
 139 }
 140
 141 // static
 142 StartupTabs PinnedTabCodec::ReadPinnedTabs(Profile* profile) {
 143   PrefService* prefs = profile->GetPrefs();
 144   if (!prefs)
 145     return StartupTabs();
 146   return ReadPinnedTabs(prefs->GetList(prefs::kPinnedTabs));
 147 }
 148
 149 // static
 150 StartupTabs PinnedTabCodec::ReadPinnedTabs(const base::Value* value) {
 151   StartupTabs results;
 152
 153   const base::ListValue* tabs_list = NULL;
 154   if (!value->GetAsList(&tabs_list))
 155     return results;
 156
 157   for (size_t i = 0, max = tabs_list->GetSize(); i < max; ++i) {
 158     const base::DictionaryValue* tab_values = NULL;
 159     if (tabs_list->GetDictionary(i, &tab_values)) {
 160       StartupTab tab;
 161       if (DecodeTab(*tab_values, &tab))
 162         results.push_back(tab);
 163     }
 164   }
 165   return results;
 166 }