Pass in the HostContentSettingsMap to the CookieModalDialog so IsValid can ma...

chrome/browser/cookie_modal_dialog.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_COOKIE_MODAL_DIALOG_H_
 #define CHROME_BROWSER_COOKIE_MODAL_DIALOG_H_
 
 #include <string>
 
+#include "base/ref_counted.h"
 #include "chrome/browser/app_modal_dialog.h"
 #include "chrome/browser/browsing_data_local_storage_helper.h"
 #include "chrome/browser/cookie_prompt_modal_dialog_delegate.h"
 #include "googleurl/src/gurl.h"
 
+class HostContentSettingsMap;
 class PrefService;
 
 // A controller+model class for cookie and local storage warning prompt.
 // |NativeDialog| is a platform specific view.
 class CookiePromptModalDialog : public AppModalDialog {
  public:
   enum DialogType {
     DIALOG_TYPE_COOKIE = 0,
     DIALOG_TYPE_LOCAL_STORAGE,
     DIALOG_TYPE_DATABASE
     // TODO(michaeln): AppCache
   };
 
   // A union of data necessary to determine the type of message box to
   // show.
   CookiePromptModalDialog(TabContents* tab_contents,
+                          HostContentSettingsMap* host_content_settings_map,
                           const GURL& origin,
                           const std::string& cookie_line,
                           CookiePromptModalDialogDelegate* delegate);
   CookiePromptModalDialog(TabContents* tab_contents,
+                          HostContentSettingsMap* host_content_settings_map,
                           const GURL& origin,
                           const string16& key,
                           const string16& value,
                           CookiePromptModalDialogDelegate* delegate);
   CookiePromptModalDialog(TabContents* tab_contents,
+                          HostContentSettingsMap* host_content_settings_map,
                           const GURL& origin,
                           const string16& database_name,
                           CookiePromptModalDialogDelegate* delegate);
-  virtual ~CookiePromptModalDialog() {}
+  virtual ~CookiePromptModalDialog();
 
   static void RegisterPrefs(PrefService* prefs);
 
   // AppModalDialog overrides.
   virtual int GetDialogButtons();
   virtual void AcceptWindow();
   virtual void CancelWindow();
   virtual bool IsValid();
 
   DialogType dialog_type() const { return dialog_type_; }
   const GURL& origin() const { return origin_; }
   const std::string& cookie_line() const { return cookie_line_; }
   const string16& local_storage_key() const { return local_storage_key_; }
   const string16& local_storage_value() const { return local_storage_value_; }
   const string16& database_name() const { return database_name_; }
   TabContents* tab_contents() const { return tab_contents_; }
 
   // Implement CookiePromptModalDialogDelegate.
   void AllowSiteData(bool remember, bool session_expire);
   void BlockSiteData(bool remember);
 
  protected:
   // AppModalDialog overrides.
   virtual NativeDialog CreateNativeDialog();
 #if defined(OS_LINUX)
   virtual void HandleDialogResponse(GtkDialog* dialog, gint response_id);
 #endif
 
  private:
+  // Used to verify our request is still necessary and when the response should
+  // persist.
+  scoped_refptr<HostContentSettingsMap> host_content_settings_map_;
+
   const DialogType dialog_type_;
 
   // The origin connected to this request.
   GURL origin_;
 
   // Cookie to display.
   const std::string cookie_line_;
 
   // LocalStorage key/value.
   const string16 local_storage_key_;
   const string16 local_storage_value_;
 
    // Database name.
   const string16 database_name_;
 
   // Delegate. The caller should provide one in order to receive results
   // from this delegate.  Any time after calling one of these methods, the
   // delegate could be deleted
   CookiePromptModalDialogDelegate* delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(CookiePromptModalDialog);
 };
 
 #endif  // CHROME_BROWSER_COOKIE_MODAL_DIALOG_H_