[Media Router] Make per-hostname cast mode selections persist

chrome/browser/ui/webui/media_router/media_router_ui.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/media_router/media_router_ui.h"
 
 #include <algorithm>
 #include <string>
 #include <unordered_map>
 #include <utility>
@@ skipping 18 matching lines @@
 #include "chrome/browser/media/router/media_source.h"
 #include "chrome/browser/media/router/media_source_helper.h"
 #include "chrome/browser/media/router/mojo/media_router_mojo_impl.h"
 #include "chrome/browser/media/router/presentation_service_delegate_impl.h"
 #include "chrome/browser/media/router/route_request_result.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/sessions/session_tab_helper.h"
 #include "chrome/browser/ui/webui/media_router/media_router_localized_strings_provider.h"
 #include "chrome/browser/ui/webui/media_router/media_router_resources_provider.h"
 #include "chrome/browser/ui/webui/media_router/media_router_webui_message_handler.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/grit/generated_resources.h"
+#include "components/prefs/pref_service.h"
+#include "components/prefs/scoped_user_pref_update.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui.h"
 #include "content/public/browser/web_ui_data_source.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/common/constants.h"
 #include "net/base/registry_controlled_domains/registry_controlled_domain.h"
 #include "third_party/icu/source/i18n/unicode/coll.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/web_dialogs/web_dialog_delegate.h"
 
@@ skipping 449 matching lines @@
   const std::string source_id = source ? source->id() : "";
 
   // The CreateRoute() part of the function is accomplished in the callback
   // OnSearchSinkResponseReceived().
   router_->SearchSinks(
       sink_id, source_id, search_criteria, domain,
       base::Bind(&MediaRouterUI::OnSearchSinkResponseReceived,
                  weak_factory_.GetWeakPtr(), cast_mode));
 }
 
+MediaCastMode MediaRouterUI::GetCastModeSelectionForCurrentHost() const {
+  const base::DictionaryValue* selections =
+      Profile::FromWebUI(web_ui())->GetPrefs()->GetDictionary(
+          prefs::kMediaRouterCastModeSelections);
+  int cast_mode = MediaCastMode::DEFAULT;
+  selections->GetInteger(GetInitiatorHostname(), &cast_mode);
+  return static_cast<MediaCastMode>(cast_mode);

[imcheng, 2016/11/14 19:32:27]

DCHECK the value from pref that it is a valid cast mode. You can use the
IsValidCastModeNum function.

[takumif, 2016/11/16 01:34:27]

Done.

+}
+
+void MediaRouterUI::RecordCastModeSelection(MediaCastMode cast_mode) {
+  // Desktop mirroring isn't domain-specific, so we don't record the selection.
+  if (cast_mode == MediaCastMode::DESKTOP_MIRROR)
+    return;
+
+  DictionaryPrefUpdate update(Profile::FromWebUI(web_ui())->GetPrefs(),
+                              prefs::kMediaRouterCastModeSelections);
+  update->SetInteger(GetInitiatorHostname(), static_cast<int>(cast_mode));
+}
+
 void MediaRouterUI::OnResultsUpdated(
     const std::vector<MediaSinkWithCastModes>& sinks) {
   sinks_ = sinks;
 
   const icu::Collator* collator_ptr = collator_.get();
   std::sort(sinks_.begin(), sinks_.end(),
             [collator_ptr](const MediaSinkWithCastModes& sink1,
                            const MediaSinkWithCastModes& sink2) {
               return sink1.sink.CompareUsingCollator(sink2.sink, collator_ptr);
             });
@@ skipping 169 matching lines @@
     MediaRouterMetrics::RecordMediaRouterDialogLoaded(
         base::Time::Now() - start_time_);
     start_time_ = base::Time();
   }
 }
 
 void MediaRouterUI::UpdateMaxDialogHeight(int height) {
   handler_->UpdateMaxDialogHeight(height);
 }
 
+std::string MediaRouterUI::GetInitiatorHostname() const {
+  return initiator_ ? initiator_->GetLastCommittedURL().HostNoBrackets() : "";
+}
+
 }  // namespace media_router