Ensure that cookies dialog is populated with new cookies on reload

chrome/browser/ui/webui/options2/cookies_view_handler.cc

 // Copyright (c) 2012 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.
 
 #include "chrome/browser/ui/webui/options2/cookies_view_handler.h"
 
 #include <string>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 102 matching lines @@
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("removeCookie",
       base::Bind(&CookiesViewHandler::Remove,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("loadCookie",
       base::Bind(&CookiesViewHandler::LoadChildren,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("setViewContext",
       base::Bind(&CookiesViewHandler::SetViewContext,
                  base::Unretained(this)));
+  web_ui()->RegisterMessageCallback("reloadCookies",
+      base::Bind(&CookiesViewHandler::ReloadCookies,
+                 base::Unretained(this)));
 }
 
 void CookiesViewHandler::TreeNodesAdded(ui::TreeModel* model,
                                         ui::TreeModelNode* parent,
                                         int start,
                                         int count) {
   // Skip if there is a batch update in progress.
   if (batch_update_)
     return;
 
@@ skipping 155 matching lines @@
   web_ui()->CallJavascriptFunction(
       GetCallback("loadChildren", GetTreeModel()), args);
 }
 
 void CookiesViewHandler::SetViewContext(const base::ListValue* args) {
   bool app_context = false;
   if (args->GetBoolean(0, &app_context))
     app_context_ = app_context;
 }
 
+void CookiesViewHandler::ReloadCookies(const base::ListValue* args) {
+  cookies_tree_model_.reset();
+  app_cookies_tree_model_.reset();
+
+  EnsureCookiesTreeModelCreated();
+}
+
 CookiesTreeModel* CookiesViewHandler::GetTreeModel() {
   CookiesTreeModel* model = app_context_ ?
       app_cookies_tree_model_.get() : cookies_tree_model_.get();
   DCHECK(model);
   return model;
 }
 
 std::string CookiesViewHandler::GetCallback(
      std::string method, CookiesTreeModel* model) {
   std::string callback("CookiesView");
 
   if (model == app_cookies_tree_model_)
     callback.append("App");
   return callback.append(".").append(method);
 }
 
 }  // namespace options2