Ties up OOBE controllers and WebUI handlers.

chrome/browser/chromeos/login/base_login_display_host.cc

 // Copyright (c) 2011 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/chromeos/login/base_login_display_host.h"
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/threading/thread_restrictions.h"
@@ skipping 101 matching lines @@
 
   default_host_ = NULL;
 }
 
 // LoginDisplayHost implementation ---------------------------------------------
 
 void BaseLoginDisplayHost::OnSessionStart() {
   MessageLoop::current()->DeleteSoon(FROM_HERE, this);
 }
 
+WizardController* BaseLoginDisplayHost::CreateWizardController() {
+  // TODO(altimofeev): move this code to ViewsLoginDisplayHost when WebUI
+  // implementation will always be used with WebUILoginDisplayHost.
+  oobe_display_.reset(new ViewsOobeDisplay(background_bounds()));
+  WizardController* wizard_controller  =
+      new WizardController(this, oobe_display_.get());
+  oobe_display_->SetScreenObserver(wizard_controller);
+  return wizard_controller;
+}
+
 void BaseLoginDisplayHost::StartWizard(
     const std::string& first_screen_name,
     const GURL& start_url) {
   DVLOG(1) << "Starting wizard, first_screen_name: " << first_screen_name;
   // Create and show the wizard.
   // Note, dtor of the old WizardController should be called before ctor of the
   // new one, because "default_controller()" is updated there. So pure "reset()"
   // is done before new controller creation.
   wizard_controller_.reset();
 
-  ViewsOobeDisplay* oobe_display = new ViewsOobeDisplay(background_bounds_);
-  wizard_controller_.reset(new WizardController(this, oobe_display));
-  oobe_display->SetScreenObserver(wizard_controller_.get());
+  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAllowWebUIOobe)) {
+    wizard_controller_.reset(CreateWizardController());
+  } else {
+    // Force views based implementation.
+    wizard_controller_.reset(BaseLoginDisplayHost::CreateWizardController());
+  }
 
   wizard_controller_->set_start_url(start_url);
   ShowBackground();
   if (!WizardController::IsDeviceRegistered())
     SetOobeProgressBarVisible(true);
   wizard_controller_->Init(first_screen_name);
 }
 
 void BaseLoginDisplayHost::StartSignInScreen() {
+  oobe_display_.reset();
+
   DVLOG(1) << "Starting sign in screen";
   std::vector<chromeos::UserManager::User> users =
       chromeos::UserManager::Get()->GetUsers();
 
   // Fix for users who updated device and thus never passed register screen.
   // If we already have users, we assume that it is not a second part of
   // OOBE. See http://crosbug.com/6289
   if (!WizardController::IsDeviceRegistered() && !users.empty()) {
     VLOG(1) << "Mark device registered because there are remembered users: "
             << users.size();
@@ skipping 153 matching lines @@
     if (!timezone_name.empty()) {
       icu::TimeZone* timezone = icu::TimeZone::createTimeZone(
           icu::UnicodeString::fromUTF8(timezone_name));
       CHECK(timezone) << "Timezone could not be set for " << timezone_name;
       chromeos::SystemAccess::GetInstance()->SetTimezone(*timezone);
     }
   }
 }
 
 }  // namespace browser