Mixed Licenses Enrollment

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 <memory>
 #include <string>
 
@@ skipping 38 matching lines @@
 //      state.
 //   4. Download the OAuth2 authorization code for device-level API access.
 //   5. Download the OAuth2 refresh token for device-level API access and store
 //      it.
 //   6. Establish the device lock in installation-time attributes.
 //   7. Store the policy blob and API refresh token.
 class EnrollmentHandlerChromeOS : public CloudPolicyClient::Observer,
                                   public CloudPolicyStore::Observer,
                                   public gaia::GaiaOAuthClient::Delegate {
  public:
-  typedef DeviceCloudPolicyInitializer::EnrollmentCallback
-      EnrollmentCallback;
+  using EnrollmentCallback = DeviceCloudPolicyInitializer::EnrollmentCallback;
+  using AvailableLicensesCallback =
+      DeviceCloudPolicyInitializer::AvailableLicensesCallback;
 
   // |store| and |install_attributes| must remain valid for the life time of the
   // enrollment handler.
   EnrollmentHandlerChromeOS(
       DeviceCloudPolicyStoreChromeOS* store,
       chromeos::InstallAttributes* install_attributes,
       ServerBackedStateKeysBroker* state_keys_broker,
       chromeos::attestation::AttestationFlow* attestation_flow,
       std::unique_ptr<CloudPolicyClient> client,
       scoped_refptr<base::SequencedTaskRunner> background_task_runner,
       chromeos::ActiveDirectoryJoinDelegate* ad_join_delegate,
       const EnrollmentConfig& enrollment_config,
       const std::string& auth_token,
       const std::string& client_id,
       const std::string& requisition,
       const EnrollmentCallback& completion_callback);
   ~EnrollmentHandlerChromeOS() override;
 
+  // Checks license types available for enrollment and reports the result
+  // to |callback|.
+  void CheckAvailableLicenses(
+      const AvailableLicensesCallback& completion_callback);
+
   // Starts the enrollment process and reports the result to
   // |completion_callback_|.
   void StartEnrollment();
 
+  // Starts the enrollment process using user-selected |license_type|
+  // and reports the result to |completion_callback_|.
+  void StartEnrollmentWithLicense(LicenseType license_type);
+
   // Releases the client.
   std::unique_ptr<CloudPolicyClient> ReleaseClient();
 
   // CloudPolicyClient::Observer:
   void OnPolicyFetched(CloudPolicyClient* client) override;
   void OnRegistrationStateChanged(CloudPolicyClient* client) override;
   void OnRobotAuthCodesFetched(CloudPolicyClient* client) override;
   void OnClientError(CloudPolicyClient* client) override;
 
   // CloudPolicyStore::Observer:
@@ skipping 70 matching lines @@
 
   // Calls InstallAttributes::LockDevice() for enterprise enrollment and
   // DeviceSettingsService::SetManagementSettings() for consumer
   // enrollment.
   void StartLockDevice();
 
   // Handle callback from InstallAttributes::LockDevice() and retry on failure.
   void HandleLockDeviceResult(
       chromeos::InstallAttributes::LockResult lock_result);
 
+  // Handles the available licenses request.
+  void HandleAvailableLicensesResult(
+      bool success,
+      const CloudPolicyClient::LicenseMap& license_map);
+
   // Initiates storing DM token. For Active Directory devices only.
   void StartStoreDMToken();
 
   // Called after StartStoreDMtoken() is done.
   void HandleDMTokenStoreResult(bool success);
 
   // Initiates storing of robot auth token.
   void StartStoreRobotAuth();
 
   // Handles completion of the robot token store operation.
@@ skipping 19 matching lines @@
   scoped_refptr<base::SequencedTaskRunner> background_task_runner_;
   chromeos::ActiveDirectoryJoinDelegate* ad_join_delegate_ = nullptr;
   std::unique_ptr<gaia::GaiaOAuthClient> gaia_oauth_client_;
   std::unique_ptr<policy::DMTokenStorage> dm_token_storage_;
 
   EnrollmentConfig enrollment_config_;
   std::string auth_token_;
   std::string client_id_;
   std::string requisition_;
   EnrollmentCallback completion_callback_;
+  AvailableLicensesCallback available_licenses_callback_;
+  enterprise_management::LicenseType::LicenseTypeEnum license_type_ =
+      enterprise_management::LicenseType::UNDEFINED;
 
   // The current state key provided by |state_keys_broker_|.
   std::string current_state_key_;
 
   // The device mode as received in the registration request.
   DeviceMode device_mode_ = DEVICE_MODE_NOT_SET;
 
   // Whether the server signaled to skip robot auth setup.
   bool skip_robot_auth_ = false;
 
@@ skipping 14 matching lines @@
   int lockbox_init_duration_ = 0;
 
   base::WeakPtrFactory<EnrollmentHandlerChromeOS> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(EnrollmentHandlerChromeOS);
 };
 
 }  // namespace policy
 
 #endif  // CHROME_BROWSER_CHROMEOS_POLICY_ENROLLMENT_HANDLER_CHROMEOS_H_