Re-landing mistakenly reverted CL http://codereview.chromium.org/2101021....

chrome/browser/chromeos/login/wizard_controller.h

 // Copyright (c) 2010 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.
 
 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_WIZARD_CONTROLLER_H_
 #define CHROME_BROWSER_CHROMEOS_LOGIN_WIZARD_CONTROLLER_H_
 
 #include <string>
 
 #include "base/scoped_ptr.h"
 #include "chrome/browser/chromeos/login/screen_observer.h"
 #include "chrome/browser/chromeos/login/view_screen.h"
 #include "chrome/browser/chromeos/login/wizard_screen.h"
 #include "testing/gtest/include/gtest/gtest_prod.h"
 
 class WizardContentsView;
 class WizardScreen;
 
 namespace chromeos {
 class AccountScreen;
 class BackgroundView;
 class LoginScreen;
 class NetworkScreen;
 class UpdateScreen;
+class StartupCustomizationDocument;
 }
 
 namespace gfx {
 class Rect;
 }
 
 namespace views {
 class Views;
 class Widget;
 }
@@ skipping 45 matching lines @@
   void ShowAccountScreen();
   void ShowUpdateScreen();
 
   // Returns a pointer to the current screen or NULL if there's no such
   // screen.
   WizardScreen* current_screen() const { return current_screen_; }
 
   // Overrides observer for testing.
   void set_observer(ScreenObserver* observer) { observer_ = observer; }
 
+  // Sets partner startup customization. WizardController takes ownership
+  // of the document object.
+  void SetCustomization(
+      const chromeos::StartupCustomizationDocument* customization);
+
   static const char kNetworkScreenName[];
   static const char kLoginScreenName[];
   static const char kAccountScreenName[];
   static const char kUpdateScreenName[];
   static const char kOutOfBoxScreenName[];
   static const char kTestNoScreenName[];
 
  private:
   // Exit handlers:
   void OnLoginSignInSelected();
@@ skipping 49 matching lines @@
 
   // True if full OOBE flow should be shown.
   bool is_out_of_box_;
 
   // NULL by default - controller itself is observer. Mock could be assigned.
   ScreenObserver* observer_;
 
   // Default WizardController.
   static WizardController* default_controller_;
 
+  // Partner startup customizations.
+  scoped_ptr<const chromeos::StartupCustomizationDocument> customization_;
+
   FRIEND_TEST(WizardControllerFlowTest, ControlFlowErrorNetwork);
   FRIEND_TEST(WizardControllerFlowTest, ControlFlowErrorUpdate);
   FRIEND_TEST(WizardControllerFlowTest, ControlFlowLanguageOnLogin);
   FRIEND_TEST(WizardControllerFlowTest, ControlFlowLanguageOnNetwork);
   FRIEND_TEST(WizardControllerFlowTest, ControlFlowMain);
   FRIEND_TEST(WizardControllerTest, SwitchLanguage);
   friend class WizardControllerFlowTest;
   DISALLOW_COPY_AND_ASSIGN(WizardController);
 };
 
 #endif  // CHROME_BROWSER_CHROMEOS_LOGIN_WIZARD_CONTROLLER_H_