Adds geolocaiton support to the location bar content image model and content bubble model.

Most of these edits were lifted out of http://codereview.chromium.org/650180

TODO: add geolocation support to the views in the three UI platforms for the bubble model.
(NOTE this change results in poorly formed bubble contents for the geolocaiton bubble, this will be fixed up in the following CLs)


BUG=11246
TEST=open a site that uses geolocaiton, select allow/deny & click the icon.


Review URL: http://codereview.chromium.org/1344002

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@42665 0039d316-1c4b-4281-b951-d872f2087c98

4 files changed, 158 insertions, 27 deletions

diff --git a/chrome/app/theme/theme_resources.grd b/chrome/app/theme/theme_resources.grd
index a7520a8..da49a1a 100644
--- a/chrome/app/theme/theme_resources.grd
+++ b/chrome/app/theme/theme_resources.grd
@@ -331,6 +331,8 @@
 
       <!-- Geolocation -->
       <include name="IDR_GEOLOCATION_INFOBAR_ICON" file="geolocation_infobar_icon.png" type="BINDATA" />
+      <include name="IDR_GEOLOCATION_ALLOWED_LOCATIONBAR_ICON" file="geolocation_allowed_locationbar_icon.png" type="BINDATA" />
+      <include name="IDR_GEOLOCATION_DENIED_LOCATIONBAR_ICON" file="geolocation_denied_locationbar_icon.png" type="BINDATA" />
 
       <if expr="pp_ifdef('_google_chrome')">
         <include name="IDR_ABOUT_BACKGROUND" file="google_chrome/about_background.png" type="BINDATA" />
diff --git a/chrome/browser/content_setting_bubble_model.cc b/chrome/browser/content_setting_bubble_model.cc
index e361b12..14eda47 100644
--- a/chrome/browser/content_setting_bubble_model.cc
+++ b/chrome/browser/content_setting_bubble_model.cc
@@ -6,6 +6,7 @@
 
 #include "app/l10n_util.h"
 #include "chrome/browser/blocked_popup_container.h"
+#include "chrome/browser/geolocation/geolocation_content_settings_map.h"
 #include "chrome/browser/host_content_settings_map.h"
 #include "chrome/browser/pref_service.h"
 #include "chrome/browser/profile.h"
@@ -33,8 +34,10 @@ class ContentSettingTitleAndLinkModel : public ContentSettingBubbleModel {
       IDS_BLOCKED_JAVASCRIPT_TITLE,
       IDS_BLOCKED_PLUGINS_TITLE,
       IDS_BLOCKED_POPUPS_TITLE,
+      0,  // Geolocation does not have an overall title.
     };
-    set_title(l10n_util::GetStringUTF8(kTitleIDs[content_type()]));
+    if (kTitleIDs[content_type()])
+      set_title(l10n_util::GetStringUTF8(kTitleIDs[content_type()]));
   }
 
   void SetManageLink() {
@@ -44,6 +47,7 @@ class ContentSettingTitleAndLinkModel : public ContentSettingBubbleModel {
       IDS_BLOCKED_JAVASCRIPT_LINK,
       IDS_BLOCKED_PLUGINS_LINK,
       IDS_BLOCKED_POPUPS_LINK,
+      IDS_GEOLOCATION_BUBBLE_MANAGE_LINK,
     };
     set_manage_link(l10n_util::GetStringUTF8(kLinkIDs[content_type()]));
   }
@@ -80,6 +84,7 @@ class ContentSettingSingleRadioGroup : public ContentSettingTitleAndLinkModel {
       IDS_BLOCKED_JAVASCRIPT_UNBLOCK,
       IDS_BLOCKED_PLUGINS_UNBLOCK,
       IDS_BLOCKED_POPUPS_UNBLOCK,
+      0,  // We don't manage geolocation here.
     };
     std::string radio_allow_label;
     radio_allow_label = l10n_util::GetStringFUTF8(
@@ -91,6 +96,7 @@ class ContentSettingSingleRadioGroup : public ContentSettingTitleAndLinkModel {
       IDS_BLOCKED_JAVASCRIPT_NO_ACTION,
       IDS_BLOCKED_PLUGINS_NO_ACTION,
       IDS_BLOCKED_POPUPS_NO_ACTION,
+      0,  // We don't manage geolocation here.
     };
     std::string radio_block_label;
     radio_block_label = l10n_util::GetStringFUTF8(
@@ -149,6 +155,63 @@ class ContentSettingPopupBubbleModel : public ContentSettingSingleRadioGroup {
   }
 };
 
+class ContentSettingDomainListBubbleModel
+    : public ContentSettingTitleAndLinkModel {
+ public:
+  ContentSettingDomainListBubbleModel(TabContents* tab_contents,
+                                      Profile* profile,
+                                      ContentSettingsType content_type)
+      : ContentSettingTitleAndLinkModel(tab_contents, profile, content_type) {
+    DCHECK_EQ(CONTENT_SETTINGS_TYPE_GEOLOCATION, content_type) <<
+        "SetDomains currently only supports geolocation content type";
+    SetDomains();
+    SetClearLink();
+  }
+
+ private:
+  void MaybeAddDomainList(DomainList* domain_list, int title_id) {
+    if (!domain_list->hosts.empty()) {
+      domain_list->title = l10n_util::GetStringUTF8(title_id);
+      add_domain_list(*domain_list);
+    }
+  }
+  void SetDomains() {
+    const TabContents::GeolocationContentSettings& settings =
+        tab_contents()->geolocation_content_settings();
+
+    // Divide the tab's current geolocation users into sets according to their
+    // permission state.
+    DomainList domain_lists[CONTENT_SETTING_NUM_SETTINGS];
+    for (TabContents::GeolocationContentSettings::const_iterator it =
+        settings.begin(); it != settings.end(); ++it) {
+      domain_lists[it->second].hosts.push_back(it->first);
+    }
+    MaybeAddDomainList(&domain_lists[CONTENT_SETTING_ALLOW],
+                       IDS_GEOLOCATION_BUBBLE_SECTION_ALLOWED);
+    MaybeAddDomainList(&domain_lists[CONTENT_SETTING_BLOCK],
+                       IDS_GEOLOCATION_BUBBLE_SECTION_DENIED);
+  }
+  void SetClearLink() {
+    set_clear_link(l10n_util::GetStringUTF8(IDS_GEOLOCATION_BUBBLE_CLEAR_LINK));
+  }
+  virtual void OnClearLinkClicked() {
+    if (!tab_contents())
+      return;
+    // Reset this embedder's entry to default for each of the requesting
+    // origins currently on the page.
+    const GURL& embedder_url = tab_contents()->GetURL();
+    const TabContents::GeolocationContentSettings& settings =
+        tab_contents()->geolocation_content_settings();
+    GeolocationContentSettingsMap* settings_map =
+        profile()->GetGeolocationContentSettingsMap();
+    for (TabContents::GeolocationContentSettings::const_iterator it =
+        settings.begin(); it != settings.end(); ++it) {
+      settings_map->SetContentSetting(it->first, embedder_url,
+                                      CONTENT_SETTING_DEFAULT);
+    }
+  }
+};
+
 // static
 ContentSettingBubbleModel*
     ContentSettingBubbleModel::CreateContentSettingBubbleModel(
@@ -163,6 +226,10 @@ ContentSettingBubbleModel*
     return new ContentSettingPopupBubbleModel(tab_contents, profile,
                                               content_type);
   }
+  if (content_type == CONTENT_SETTINGS_TYPE_GEOLOCATION) {
+    return new ContentSettingDomainListBubbleModel(tab_contents, profile,
+                                                   content_type);
+  }
   return new ContentSettingSingleRadioGroup(tab_contents, profile,
                                             content_type);
 }
diff --git a/chrome/browser/content_setting_bubble_model.h b/chrome/browser/content_setting_bubble_model.h
index dcbafb7..b318859 100644
--- a/chrome/browser/content_setting_bubble_model.h
+++ b/chrome/browser/content_setting_bubble_model.h
@@ -13,6 +13,7 @@
 #include "chrome/common/notification_registrar.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 
+class GURL;
 class Profile;
 class SkBitmap;
 class TabContents;
@@ -46,11 +47,18 @@ class ContentSettingBubbleModel : public NotificationObserver {
   };
   typedef std::vector<RadioGroup> RadioGroups;
 
+  struct DomainList {
+    std::string title;
+    std::vector<GURL> hosts;
+  };
+
   struct BubbleContent {
     std::string title;
     PopupItems popup_items;
     RadioGroups radio_groups;
+    std::vector<DomainList> domain_lists;
     std::string manage_link;
+    std::string clear_link;
   };
 
   const BubbleContent& bubble_content() const { return bubble_content_; }
@@ -63,6 +71,7 @@ class ContentSettingBubbleModel : public NotificationObserver {
   virtual void OnRadioClicked(int radio_group, int radio_index) {}
   virtual void OnPopupClicked(int index) {}
   virtual void OnManageLinkClicked() {}
+  virtual void OnClearLinkClicked() {}
 
  protected:
   ContentSettingBubbleModel(TabContents* tab_contents, Profile* profile,
@@ -78,9 +87,15 @@ class ContentSettingBubbleModel : public NotificationObserver {
   void add_radio_group(const RadioGroup& radio_group) {
     bubble_content_.radio_groups.push_back(radio_group);
   }
+  void add_domain_list(const DomainList& domain_list) {
+    bubble_content_.domain_lists.push_back(domain_list);
+  }
   void set_manage_link(const std::string& link) {
     bubble_content_.manage_link = link;
   }
+  void set_clear_link(const std::string& link) {
+    bubble_content_.clear_link = link;
+  }
 
  private:
   TabContents* tab_contents_;
diff --git a/chrome/browser/content_setting_image_model.cc b/chrome/browser/content_setting_image_model.cc
index 96430f7..77b2f1a 100644
--- a/chrome/browser/content_setting_image_model.cc
+++ b/chrome/browser/content_setting_image_model.cc
@@ -12,40 +12,21 @@
 class ContentSettingBlockedImageModel : public ContentSettingImageModel {
  public:
   explicit ContentSettingBlockedImageModel(
-      ContentSettingsType content_settings_type)
-      : ContentSettingImageModel(content_settings_type) {
-  }
+      ContentSettingsType content_settings_type);
 
-  virtual void UpdateFromTabContents(const TabContents* tab_contents) {
-    if (tab_contents &&
-        tab_contents->IsContentBlocked(get_content_settings_type())) {
-      set_icon(kBlockedIconIDs[get_content_settings_type()]);
-      set_tooltip(
-          l10n_util::GetStringUTF8(kTooltipIDs[get_content_settings_type()]));
-      set_visible(true);
-    } else {
-      set_visible(false);
-    }
-  }
+  virtual void UpdateFromTabContents(const TabContents* tab_contents);
 
  private:
   static const int kBlockedIconIDs[];
   static const int kTooltipIDs[];
-
 };
 
-ContentSettingImageModel::ContentSettingImageModel(
-    ContentSettingsType content_settings_type)
-    : content_settings_type_(content_settings_type),
-      is_visible_(false),
-      icon_(0) {
-}
+class ContentSettingGeolocationImageModel : public ContentSettingImageModel {
+ public:
+  ContentSettingGeolocationImageModel();
 
-// static
-ContentSettingImageModel* ContentSettingImageModel::
-    CreateContentSettingImageModel(ContentSettingsType content_settings_type) {
-  return new ContentSettingBlockedImageModel(content_settings_type);
-}
+  virtual void UpdateFromTabContents(const TabContents* tab_contents);
+};
 
 const int ContentSettingBlockedImageModel::kBlockedIconIDs[] = {
     IDR_BLOCKED_COOKIES,
@@ -62,3 +43,69 @@ const int ContentSettingBlockedImageModel::kTooltipIDs[] = {
   IDS_BLOCKED_PLUGINS_TITLE,
   IDS_BLOCKED_POPUPS_TOOLTIP,
 };
+
+
+ContentSettingBlockedImageModel::ContentSettingBlockedImageModel(
+    ContentSettingsType content_settings_type)
+    : ContentSettingImageModel(content_settings_type) {
+}
+
+void ContentSettingBlockedImageModel::UpdateFromTabContents(
+    const TabContents* tab_contents) {
+  if (!tab_contents ||
+      !tab_contents->IsContentBlocked(get_content_settings_type())) {
+    set_visible(false);
+    return;
+  }
+  set_icon(kBlockedIconIDs[get_content_settings_type()]);
+  set_tooltip(
+      l10n_util::GetStringUTF8(kTooltipIDs[get_content_settings_type()]));
+  set_visible(true);
+}
+
+ContentSettingGeolocationImageModel::ContentSettingGeolocationImageModel()
+    : ContentSettingImageModel(CONTENT_SETTINGS_TYPE_GEOLOCATION) {
+}
+
+void ContentSettingGeolocationImageModel::UpdateFromTabContents(
+    const TabContents* tab_contents) {
+  if (!tab_contents) {
+    set_visible(false);
+    return;
+  }
+  const TabContents::GeolocationContentSettings& settings =
+      tab_contents->geolocation_content_settings();
+  if (settings.empty()) {
+    set_visible(false);
+    return;
+  }
+  set_visible(true);
+  // If any embedded site has access the allowed icon takes priority over the
+  // blocked icon.
+  for (TabContents::GeolocationContentSettings::const_iterator it =
+       settings.begin(); it != settings.end(); ++it ) {
+    if (it->second == CONTENT_SETTING_ALLOW) {
+      set_icon(IDR_GEOLOCATION_ALLOWED_LOCATIONBAR_ICON);
+      set_tooltip(l10n_util::GetStringUTF8(IDS_GEOLOCATION_ALLOWED_TOOLTIP));
+      return;
+    }
+  }
+  set_icon(IDR_GEOLOCATION_DENIED_LOCATIONBAR_ICON);
+  set_tooltip(l10n_util::GetStringUTF8(IDS_GEOLOCATION_BLOCKED_TOOLTIP));
+}
+
+ContentSettingImageModel::ContentSettingImageModel(
+    ContentSettingsType content_settings_type)
+    : content_settings_type_(content_settings_type),
+      is_visible_(false),
+      icon_(0) {
+}
+
+// static
+ContentSettingImageModel*
+    ContentSettingImageModel::CreateContentSettingImageModel(
+    ContentSettingsType content_settings_type) {
+  if (content_settings_type == CONTENT_SETTINGS_TYPE_GEOLOCATION)
+    return new ContentSettingGeolocationImageModel();
+  return new ContentSettingBlockedImageModel(content_settings_type);
+}