1 files changed, 66 insertions, 0 deletions
diff --git a/chrome/browser/views/sync/sync_setup_wizard.cc b/chrome/browser/views/sync/sync_setup_wizard.cc
index 18ef7b8..2e86bfd 100644
--- a/chrome/browser/views/sync/sync_setup_wizard.cc
+++ b/chrome/browser/views/sync/sync_setup_wizard.cc
@@ -6,12 +6,78 @@
 
 #include "chrome/browser/views/sync/sync_setup_wizard.h"
 
+#include "app/resource_bundle.h"
+#include "base/message_loop.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/common/pref_service.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/dom_ui/chrome_url_data_manager.h"
 #include "chrome/browser/sync/profile_sync_service.h"
 #include "chrome/browser/views/sync/sync_setup_flow.h"
+#include "chrome/common/url_constants.h"
+#include "grit/app_resources.h"
+#include "grit/browser_resources.h"
+
+class SyncResourcesSource : public ChromeURLDataManager::DataSource {
+ public:
+  SyncResourcesSource()
+      : DataSource(chrome::kSyncResourcesPath, MessageLoop::current()) {
+  }
+  virtual ~SyncResourcesSource() { }
+
+  virtual void StartDataRequest(const std::string& path, int request_id);
+
+  virtual std::string GetMimeType(const std::string& path) const {
+    if (path == chrome::kSyncThrobberPath)
+      return "image/png";
+    else
+      return "text/html";
+  }
+ private:
+  DISALLOW_COPY_AND_ASSIGN(SyncResourcesSource);
+};
+
+void SyncResourcesSource::StartDataRequest(const std::string& path_raw,
+                                           int request_id) {
+  scoped_refptr<RefCountedBytes> html_bytes(new RefCountedBytes);
+  if (path_raw == chrome::kSyncThrobberPath) {
+    ResourceBundle::GetSharedInstance().LoadImageResourceBytes(IDR_THROBBER,
+        &html_bytes->data);
+    SendResponse(request_id, html_bytes);
+    return;
+  }
+
+  std::string response;
+  if (path_raw == chrome::kSyncGaiaLoginPath) {
+    static const StringPiece html(ResourceBundle::GetSharedInstance()
+        .GetRawDataResource(IDR_GAIA_LOGIN_HTML));
+    response = html.as_string();
+  } else if (path_raw == chrome::kSyncMergeAndSyncPath) {
+    static const StringPiece html(ResourceBundle::GetSharedInstance()
+        .GetRawDataResource(IDR_MERGE_AND_SYNC_HTML));
+    response = html.as_string();
+  } else if (path_raw == chrome::kSyncSetupFlowPath) {
+    static const StringPiece html(ResourceBundle::GetSharedInstance()
+        .GetRawDataResource(IDR_SYNC_SETUP_FLOW_HTML));
+    response = html.as_string();
+  }
+  // Send the response.
+  html_bytes->data.resize(response.size());
+  std::copy(response.begin(), response.end(), html_bytes->data.begin());
+  SendResponse(request_id, html_bytes);
+}
 
 SyncSetupWizard::SyncSetupWizard(ProfileSyncService* service)
       : service_(service), flow_container_(new SyncSetupFlowContainer()) {
+  // Register data sources for HTML content we require.
+  // g_browser_process and/or io_thread may not exist during testing.
+  if (g_browser_process && g_browser_process->io_thread()) {
+    // Add our network layer data source for 'cloudy' URLs.
+    g_browser_process->io_thread()->message_loop()->PostTask(FROM_HERE,
+        NewRunnableMethod(&chrome_url_data_manager,
+                          &ChromeURLDataManager::AddDataSource,
+                          new SyncResourcesSource()));
+  }
 }
 
 SyncSetupWizard::~SyncSetupWizard() {