chrome/browser/content_settings/content_settings_usages_state.cc

   1 // Copyright 2013 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/content_settings/content_settings_usages_state.h"
   6
   7 #include <string>
   8
   9 #include "base/prefs/pref_service.h"
  10 #include "base/strings/string_piece.h"
  11 #include "base/strings/utf_string_conversions.h"
  12 #include "chrome/browser/content_settings/host_content_settings_map.h"
  13 #include "chrome/browser/profiles/profile.h"
  14 #include "chrome/common/pref_names.h"
  15 #include "content/public/browser/navigation_details.h"
  16 #include "content/public/browser/navigation_entry.h"
  17 #include "net/base/net_util.h"
  18
  19 ContentSettingsUsagesState::ContentSettingsUsagesState(Profile* profile,
  20                                                        ContentSettingsType type)
  21     : profile_(profile),
  22       type_(type) {
  23 }
  24
  25 ContentSettingsUsagesState::~ContentSettingsUsagesState() {
  26 }
  27
  28 void ContentSettingsUsagesState::OnPermissionSet(
  29     const GURL& requesting_origin, bool allowed) {
  30   state_map_[requesting_origin] =
  31       allowed ? CONTENT_SETTING_ALLOW : CONTENT_SETTING_BLOCK;
  32 }
  33
  34 void ContentSettingsUsagesState::DidNavigate(
  35     const content::LoadCommittedDetails& details) {
  36   if (details.entry)
  37     embedder_url_ = details.entry->GetURL();
  38   if (state_map_.empty())
  39     return;
  40   if (!details.entry ||
  41       details.previous_url.GetOrigin() != details.entry->GetURL().GetOrigin()) {
  42     state_map_.clear();
  43     return;
  44   }
  45   // We're in the same origin, check if there's any icon to be displayed.
  46   unsigned int tab_state_flags = 0;
  47   GetDetailedInfo(NULL, &tab_state_flags);
  48   if (!(tab_state_flags & TABSTATE_HAS_ANY_ICON))
  49     state_map_.clear();
  50 }
  51
  52 void ContentSettingsUsagesState::ClearStateMap() {
  53   state_map_.clear();
  54 }
  55
  56 void ContentSettingsUsagesState::GetDetailedInfo(
  57     FormattedHostsPerState* formatted_hosts_per_state,
  58     unsigned int* tab_state_flags) const {
  59   DCHECK(tab_state_flags);
  60   DCHECK(embedder_url_.is_valid());
  61   ContentSetting default_setting =
  62       profile_->GetHostContentSettingsMap()->GetDefaultContentSetting(
  63           type_, NULL);
  64   std::set<std::string> formatted_hosts;
  65   std::set<std::string> repeated_formatted_hosts;
  66
  67   // Build a set of repeated formatted hosts
  68   for (StateMap::const_iterator i(state_map_.begin());
  69        i != state_map_.end(); ++i) {
  70     std::string formatted_host = GURLToFormattedHost(i->first);
  71     if (!formatted_hosts.insert(formatted_host).second) {
  72       repeated_formatted_hosts.insert(formatted_host);
  73     }
  74   }
  75
  76   for (StateMap::const_iterator i(state_map_.begin());
  77        i != state_map_.end(); ++i) {
  78     if (i->second == CONTENT_SETTING_ALLOW)
  79       *tab_state_flags |= TABSTATE_HAS_ANY_ALLOWED;
  80     if (formatted_hosts_per_state) {
  81       std::string formatted_host = GURLToFormattedHost(i->first);
  82       std::string final_formatted_host =
  83           repeated_formatted_hosts.find(formatted_host) ==
  84           repeated_formatted_hosts.end() ?
  85           formatted_host :
  86           i->first.spec();
  87       (*formatted_hosts_per_state)[i->second].insert(final_formatted_host);
  88     }
  89
  90     const ContentSetting saved_setting =
  91         profile_->GetHostContentSettingsMap()->GetContentSetting(
  92             i->first, embedder_url_, type_, std::string());
  93     if (saved_setting != default_setting)
  94       *tab_state_flags |= TABSTATE_HAS_EXCEPTION;
  95     if (saved_setting != i->second)
  96       *tab_state_flags |= TABSTATE_HAS_CHANGED;
  97     if (saved_setting != CONTENT_SETTING_ASK)
  98       *tab_state_flags |= TABSTATE_HAS_ANY_ICON;
  99   }
 100 }
 101
 102 std::string ContentSettingsUsagesState::GURLToFormattedHost(
 103     const GURL& url) const {
 104   base::string16 display_host;
 105   net::AppendFormattedHost(url,
 106       profile_->GetPrefs()->GetString(prefs::kAcceptLanguages), &display_host);
 107   return base::UTF16ToUTF8(display_host);
 108 }