Index: chrome/browser/ui/webui/media_router/media_router_webui_message_handler.cc |
diff --git a/chrome/browser/ui/webui/media_router/media_router_webui_message_handler.cc b/chrome/browser/ui/webui/media_router/media_router_webui_message_handler.cc |
index baeea3dad87c242f764cacd39290ff741d484bb5..8ef375942e4f5ca5ae1a4c0c1f24122d7028db39 100644 |
--- a/chrome/browser/ui/webui/media_router/media_router_webui_message_handler.cc |
+++ b/chrome/browser/ui/webui/media_router/media_router_webui_message_handler.cc |
@@ -29,6 +29,7 @@ const char kActOnIssue[] = "actOnIssue"; |
const char kCloseRoute[] = "closeRoute"; |
const char kCloseDialog[] = "closeDialog"; |
const char kReportSinkCount[] = "reportSinkCount"; |
+const char kOnInitialDataReceived[] = "onInitialDataReceived"; |
// JS function names. |
const char kSetInitialData[] = "media_router.ui.setInitialData"; |
@@ -256,11 +257,16 @@ void MediaRouterWebUIMessageHandler::RegisterMessages() { |
kReportSinkCount, |
base::Bind(&MediaRouterWebUIMessageHandler::OnReportSinkCount, |
base::Unretained(this))); |
+ web_ui()->RegisterMessageCallback( |
+ kOnInitialDataReceived, |
+ base::Bind(&MediaRouterWebUIMessageHandler::OnInitialDataReceived, |
+ base::Unretained(this))); |
} |
void MediaRouterWebUIMessageHandler::OnRequestInitialData( |
const base::ListValue* args) { |
DVLOG(1) << "OnRequestInitialData"; |
+ media_router_ui_->OnUIInitiallyLoaded(); |
base::DictionaryValue initial_data; |
// "No Cast devices found?" Chromecast help center page. |
@@ -392,6 +398,12 @@ void MediaRouterWebUIMessageHandler::OnReportSinkCount( |
UMA_HISTOGRAM_COUNTS_100("MediaRouter.Ui.Device.Count", sink_count); |
} |
+void MediaRouterWebUIMessageHandler::OnInitialDataReceived( |
+ const base::ListValue* args) { |
+ DVLOG(1) << "OnInitialDataReceived"; |
+ media_router_ui_->OnUIInitialDataReceived(); |
+} |
+ |
bool MediaRouterWebUIMessageHandler::ActOnIssueType( |
const IssueAction::Type& action_type, |
const base::DictionaryValue* args) { |