Move device requisition to EnterpriseEnrollmentHandlerChromeOS.

chrome/browser/chromeos/policy/enrollment_handler_chromeos.h

 // Copyright (c) 2012 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_POLICY_ENROLLMENT_HANDLER_CHROMEOS_H_
 #define CHROME_BROWSER_CHROMEOS_POLICY_ENROLLMENT_HANDLER_CHROMEOS_H_
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/compiler_specific.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "chrome/browser/chromeos/policy/device_cloud_policy_initializer.h"
 #include "chrome/browser/chromeos/policy/device_cloud_policy_validator.h"
 #include "chrome/browser/chromeos/policy/enterprise_install_attributes.h"
 #include "components/policy/core/common/cloud/cloud_policy_client.h"
 #include "components/policy/core/common/cloud/cloud_policy_store.h"
 #include "google_apis/gaia/gaia_oauth_client.h"
 
 namespace base {
 class SequencedTaskRunner;
 }
 
 namespace enterprise_management {
 class PolicyFetchResponse;
 }
 
+class PrefRegistrySimple;
+class PrefService;
+
 namespace policy {
 
 class DeviceCloudPolicyStoreChromeOS;
 class ServerBackedStateKeysBroker;
 
 // Implements the logic that establishes enterprise enrollment for Chromium OS
 // devices. The process is as follows:
 //   1. Given an auth token, register with the policy service.
 //   2. Download the initial policy blob from the service.
 //   3. Verify the policy blob. Everything up to this point doesn't touch device
@@ skipping 11 matching lines @@
       AllowedDeviceModes;
   typedef DeviceCloudPolicyInitializer::EnrollmentCallback
       EnrollmentCallback;
 
   // |store| and |install_attributes| must remain valid for the life time of the
   // enrollment handler. |allowed_device_modes| determines what device modes
   // are acceptable. If the mode specified by the server is not acceptable,
   // enrollment will fail with an EnrollmentStatus indicating
   // STATUS_REGISTRATION_BAD_MODE.
   EnrollmentHandlerChromeOS(
+      PrefService* local_state,
       DeviceCloudPolicyStoreChromeOS* store,
       EnterpriseInstallAttributes* install_attributes,
       ServerBackedStateKeysBroker* state_keys_broker,
       scoped_ptr<CloudPolicyClient> client,
       scoped_refptr<base::SequencedTaskRunner> background_task_runner,
       const std::string& auth_token,
       const std::string& client_id,
       bool is_auto_enrollment,
-      const std::string& requisition,
       const AllowedDeviceModes& allowed_device_modes,
       const EnrollmentCallback& completion_callback);
   virtual ~EnrollmentHandlerChromeOS();
 
+  // Pref registration helper.
+  static void RegisterPrefs(PrefRegistrySimple* registry);
+
   // Starts the enrollment process and reports the result to
   // |completion_callback_|.
   void StartEnrollment();
 
   // Releases the client.
   scoped_ptr<CloudPolicyClient> ReleaseClient();
 
+  // Gets/Sets the device requisition.
+  std::string GetDeviceRequisition() const;
+  void SetDeviceRequisition(const std::string& requisition);
+
+  // Returns true if device requisition is remora/shark.
+  bool IsRemoraRequisition() const;
+  bool IsSharkRequisition() const;
+
   // CloudPolicyClient::Observer:
   virtual void OnPolicyFetched(CloudPolicyClient* client) OVERRIDE;
   virtual void OnRegistrationStateChanged(CloudPolicyClient* client) OVERRIDE;
   virtual void OnRobotAuthCodesFetched(CloudPolicyClient* client) OVERRIDE;
   virtual void OnClientError(CloudPolicyClient* client) OVERRIDE;
 
   // CloudPolicyStore::Observer:
   virtual void OnStoreLoaded(CloudPolicyStore* store) OVERRIDE;
   virtual void OnStoreError(CloudPolicyStore* store) OVERRIDE;
 
@@ skipping 17 matching lines @@
     STEP_POLICY_FETCH,        // Fetching policy.
     STEP_VALIDATION,          // Policy validation.
     STEP_ROBOT_AUTH_FETCH,    // Fetching device API auth code.
     STEP_ROBOT_AUTH_REFRESH,  // Fetching device API refresh token.
     STEP_LOCK_DEVICE,         // Writing installation-time attributes.
     STEP_STORE_ROBOT_AUTH,    // Encrypting & writing robot refresh token.
     STEP_STORE_POLICY,        // Storing policy and API refresh token.
     STEP_FINISHED,            // Enrollment process finished, no further action.
   };
 
+  // Initializes requisition settings at OOBE with values from VPD.
+  void InitializeRequisition();
+
   // Handles the response to a request for server-backed state keys.
   void CheckStateKeys(const std::vector<std::string>& state_keys);
 
   // Starts registration if the store is initialized.
   void AttemptRegistration();
 
   // Handles the policy validation result, proceeding with installation-time
   // attributes locking if successful.
   void PolicyValidated(DeviceCloudPolicyValidator* validator);
 
@@ skipping 14 matching lines @@
 
   // Handles completion of the robot token store operation.
   void HandleRobotAuthTokenStored(bool result);
 
   // Drops any ongoing actions.
   void Stop();
 
   // Reports the result of the enrollment process to the initiator.
   void ReportResult(EnrollmentStatus status);
 
+  PrefService* local_state_;
   DeviceCloudPolicyStoreChromeOS* store_;
   EnterpriseInstallAttributes* install_attributes_;
   ServerBackedStateKeysBroker* state_keys_broker_;
   scoped_ptr<CloudPolicyClient> client_;
   scoped_refptr<base::SequencedTaskRunner> background_task_runner_;
   scoped_ptr<gaia::GaiaOAuthClient> gaia_oauth_client_;
 
   std::string auth_token_;
   std::string client_id_;
   bool is_auto_enrollment_;
@@ skipping 19 matching lines @@
   int lockbox_init_duration_;
 
   base::WeakPtrFactory<EnrollmentHandlerChromeOS> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(EnrollmentHandlerChromeOS);
 };
 
 }  // namespace policy
 
 #endif  // CHROME_BROWSER_CHROMEOS_POLICY_ENROLLMENT_HANDLER_CHROMEOS_H_