MD Settings: Adding C++ support for editing startup URLs.

chrome/browser/ui/webui/settings/settings_startup_pages_handler.cc

 // Copyright 2015 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/settings/settings_startup_pages_handler.h"
 
 #include <string>
 #include <vector>
 
 #include "chrome/browser/prefs/session_startup_pref.h"
@@ skipping 12 matching lines @@
 StartupPagesHandler::~StartupPagesHandler() {
 }
 
 void StartupPagesHandler::RegisterMessages() {
   if (Profile::FromWebUI(web_ui())->IsOffTheRecord())
     return;
 
   web_ui()->RegisterMessageCallback("addStartupPage",
       base::Bind(&StartupPagesHandler::HandleAddStartupPage,
                  base::Unretained(this)));
+  web_ui()->RegisterMessageCallback("editStartupPage",
+      base::Bind(&StartupPagesHandler::HandleEditStartupPage,
+                 base::Unretained(this)));
   web_ui()->RegisterMessageCallback("onStartupPrefsPageLoad",
       base::Bind(&StartupPagesHandler::HandleOnStartupPrefsPageLoad,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("removeStartupPage",
       base::Bind(&StartupPagesHandler::HandleRemoveStartupPage,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("setStartupPagesToCurrentPages",
       base::Bind(&StartupPagesHandler::HandleSetStartupPagesToCurrentPages,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("validateStartupPage",
@@ skipping 55 matching lines @@
   int row_count = startup_custom_pages_table_model_.RowCount();
   int index;
   if (!args->GetInteger(1, &index) || index > row_count)
     index = row_count;
 
   startup_custom_pages_table_model_.Add(index, url);
   SaveStartupPagesPref();
   ResolveJavascriptCallback(*callback_id, base::FundamentalValue(true));
 }
 
+void StartupPagesHandler::HandleEditStartupPage(const base::ListValue* args) {
+  CHECK_EQ(args->GetSize(), 3U);
+  const base::Value* callback_id;
+  CHECK(args->Get(0, &callback_id));
+  int index;
+  CHECK(args->GetInteger(1, &index));
+
+  if (index < 0 || index > startup_custom_pages_table_model_.RowCount()) {
+    RejectJavascriptCallback(*callback_id, *base::Value::CreateNullValue());
+    NOTREACHED();
+    return;
+  }
+
+  std::string url_string;
+  CHECK(args->GetString(2, &url_string));
+
+  GURL fixed_url;
+  if (settings_utils::FixupAndValidateStartupPage(url_string, &fixed_url)) {
+    std::vector<GURL> urls = startup_custom_pages_table_model_.GetURLs();
+    urls[index] = fixed_url;
+    startup_custom_pages_table_model_.SetURLs(urls);
+    SaveStartupPagesPref();
+    ResolveJavascriptCallback(*callback_id, base::FundamentalValue(true));
+  } else {
+    ResolveJavascriptCallback(*callback_id, base::FundamentalValue(false));
+  }
+}
+
 void StartupPagesHandler::HandleOnStartupPrefsPageLoad(
     const base::ListValue* args) {
   startup_custom_pages_table_model_.SetObserver(this);
 
   PrefService* prefService = Profile::FromWebUI(web_ui())->GetPrefs();
   SessionStartupPref pref = SessionStartupPref::GetStartupPref(
       prefService);
   startup_custom_pages_table_model_.SetURLs(pref.urls);
 
   if (pref.urls.empty())
@@ skipping 61 matching lines @@
 
 void StartupPagesHandler::UpdateStartupPages() {
   const SessionStartupPref startup_pref = SessionStartupPref::GetStartupPref(
       Profile::FromWebUI(web_ui())->GetPrefs());
   startup_custom_pages_table_model_.SetURLs(startup_pref.urls);
   // The change will go to the JS code in the
   // StartupPagesHandler::OnModelChanged() method.
 }
 
 }  // namespace settings