Store creating url in origin nodes and use it for content settings.

chrome/browser/cookies_tree_model.h

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_COOKIES_TREE_MODEL_H_
 #define CHROME_BROWSER_COOKIES_TREE_MODEL_H_
 
 #include <string>
 #include <vector>
 
@@ skipping 94 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(CookieTreeNode);
 };
 
 // CookieTreeRootNode ---------------------------------------------------------
 // The node at the root of the CookieTree that gets inserted into the view.
 class CookieTreeRootNode : public CookieTreeNode {
  public:
   explicit CookieTreeRootNode(CookiesTreeModel* model) : model_(model) {}
   virtual ~CookieTreeRootNode() {}
 
-  CookieTreeOriginNode* GetOrCreateOriginNode(const std::wstring& origin);
+  CookieTreeOriginNode* GetOrCreateOriginNode(const GURL& url);
 
   // CookieTreeNode methods:
   virtual CookiesTreeModel* GetModel() const { return model_; }
   virtual DetailedInfo GetDetailedInfo() const {
     return DetailedInfo(std::wstring(), DetailedInfo::TYPE_ROOT, NULL, NULL,
                         NULL, NULL);
   }
  private:
 
   CookiesTreeModel* model_;
 
   DISALLOW_COPY_AND_ASSIGN(CookieTreeRootNode);
 };
 
 // CookieTreeOriginNode -------------------------------------------------------
 class CookieTreeOriginNode : public CookieTreeNode {
  public:
-  explicit CookieTreeOriginNode(const std::wstring& origin)
-      : CookieTreeNode(origin),
-        cookies_child_(NULL),
-        databases_child_(NULL),
-        local_storages_child_(NULL),
-        appcaches_child_(NULL) {}
+  // Returns the origin node's title to use for a given URL.
+  static std::wstring TitleForUrl(const GURL& url);
+
+  explicit CookieTreeOriginNode(const GURL& url);
   virtual ~CookieTreeOriginNode() {}
 
   // CookieTreeNode methods:
   virtual DetailedInfo GetDetailedInfo() const {
     return DetailedInfo(GetTitle(), DetailedInfo::TYPE_ORIGIN, NULL, NULL,
                         NULL, NULL);
   }
 
   // CookieTreeOriginNode methods:
   CookieTreeCookiesNode* GetOrCreateCookiesNode();
   CookieTreeDatabasesNode* GetOrCreateDatabasesNode();
   CookieTreeLocalStoragesNode* GetOrCreateLocalStoragesNode();
   CookieTreeAppCachesNode* GetOrCreateAppCachesNode();
+
+  // Creates an content exception for this origin of type
+  // CONTENT_SETTINGS_TYPE_COOKIES.
   void CreateContentException(HostContentSettingsMap* content_settings,
-                              ContentSetting setting);
+                              ContentSetting setting) const;
+
+  // True if a content exception can be created for this origin.
+  bool CanCreateContentException() const;
 
  private:
   // Pointers to the cookies, databases, local storage and appcache nodes.
   // When we build up the tree we need to quickly get a reference to the COOKIES
   // node to add children. Checking each child and interrogating them to see if
   // they are a COOKIES, APPCACHES, DATABASES etc node seems less preferable
   // than storing an extra pointer per origin.
   CookieTreeCookiesNode* cookies_child_;
   CookieTreeDatabasesNode* databases_child_;
   CookieTreeLocalStoragesNode* local_storages_child_;
   CookieTreeAppCachesNode* appcaches_child_;
 
+  // The URL for which this node was initially created.
+  GURL url_;
+
   DISALLOW_COPY_AND_ASSIGN(CookieTreeOriginNode);
 };
 
 // CookieTreeCookieNode ------------------------------------------------------
 class CookieTreeCookieNode : public CookieTreeNode {
  public:
   friend class CookieTreeCookiesNode;
 
   // Does not take ownership of cookie, and cookie should remain valid at least
   // as long as the CookieTreeCookieNode is valid.
@@ skipping 263 matching lines @@
 
   friend class CookieTreeAppCacheNode;
   friend class CookieTreeCookieNode;
   friend class CookieTreeDatabaseNode;
   friend class CookieTreeLocalStorageNode;
 
   DISALLOW_COPY_AND_ASSIGN(CookiesTreeModel);
 };
 
 #endif  // CHROME_BROWSER_COOKIES_TREE_MODEL_H_