Implement CONTENT_SETTING_ASK for database. Also some cleanup of the dialog for local storage.

BUG=34628
TEST=none

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

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

1 files changed, 26 insertions, 1 deletions

diff --git a/chrome/browser/cookie_modal_dialog.cc b/chrome/browser/cookie_modal_dialog.cc
index 496843c..325378d 100644
--- a/chrome/browser/cookie_modal_dialog.cc
+++ b/chrome/browser/cookie_modal_dialog.cc
@@ -8,6 +8,7 @@
 #include "chrome/common/pref_names.h"
 #include "chrome/common/pref_service.h"
 
+// Cookies
 CookiePromptModalDialog::CookiePromptModalDialog(
     TabContents* tab_contents,
     const GURL& origin,
@@ -20,7 +21,7 @@ CookiePromptModalDialog::CookiePromptModalDialog(
       delegate_(delegate) {
 }
 
-
+// LocalStorage
 CookiePromptModalDialog::CookiePromptModalDialog(
     TabContents* tab_contents,
     const GURL& origin,
@@ -35,6 +36,30 @@ CookiePromptModalDialog::CookiePromptModalDialog(
       delegate_(delegate) {
 }
 
+// Database
+CookiePromptModalDialog::CookiePromptModalDialog(
+    TabContents* tab_contents,
+    const GURL& origin,
+    const string16& database_name,
+    CookiePromptModalDialogDelegate* delegate)
+    : AppModalDialog(tab_contents, std::wstring()),
+      dialog_type_(DIALOG_TYPE_DATABASE),
+      origin_(origin),
+      database_name_(database_name),
+      delegate_(delegate) {
+}
+
+void CookiePromptModalDialog::AllowSiteData(bool remember,
+                                            bool session_expire) {
+  delegate_->AllowSiteData(remember, session_expire);
+  delegate_ = NULL;  // It can be deleted at any point now.
+}
+
+void CookiePromptModalDialog::BlockSiteData(bool remember) {
+  delegate_->BlockSiteData(remember);
+  delegate_ = NULL;  // It can be deleted at any point now.
+}
+
 // static
 void CookiePromptModalDialog::RegisterPrefs(PrefService* prefs) {
   prefs->RegisterBooleanPref(prefs::kCookiePromptExpanded, false);