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
19ContentSettingsUsagesState::ContentSettingsUsagesState(Profile* profile,
20                                                       ContentSettingsType type)
21    : profile_(profile),
22      type_(type) {
23}
24
25ContentSettingsUsagesState::~ContentSettingsUsagesState() {
26}
27
28void ContentSettingsUsagesState::OnPermissionSet(
29    const GURL& requesting_origin, bool allowed) {
30  state_map_[requesting_origin] =
31      allowed ? CONTENT_SETTING_ALLOW : CONTENT_SETTING_BLOCK;
32}
33
34void 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
52void ContentSettingsUsagesState::ClearStateMap() {
53  state_map_.clear();
54}
55
56void 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
102std::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}
109