Added OOBE Asset / Naming prompt

chrome/browser/chromeos/login/enrollment/enrollment_screen.cc

 // 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.
 
 #include "chrome/browser/chromeos/login/enrollment/enrollment_screen.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/callback.h"
 #include "base/logging.h"
@@ skipping 209 matching lines @@
 
 void EnrollmentScreen::OnOtherError(
     EnterpriseEnrollmentHelper::OtherError error) {
   OnAnyEnrollmentError();
   actor_->ShowOtherError(error);
 }
 
 void EnrollmentScreen::OnDeviceEnrolled(const std::string& additional_token) {
   if (!additional_token.empty())
     SendEnrollmentAuthToken(additional_token);
-  StartupUtils::MarkDeviceRegistered(
-      base::Bind(&EnrollmentScreen::ShowEnrollmentStatusOnSuccess,
-                 weak_ptr_factory_.GetWeakPtr()));
+
+  enrollment_helper_->GetDeviceAttributeUpdatePermission();
+}
+
+void EnrollmentScreen::OnDeviceAttributeProvided(const std::string& asset_id,
+                                                 const std::string& location) {
+  enrollment_helper_->UpdateDeviceAttributes(asset_id, location);
+}
+
+void EnrollmentScreen::OnDeviceAttributeUpdatePermission(bool granted) {
+  // If user is permitted to update device attributes
+  // Show attribute prompt screen
+  if (granted) {
+    StartupUtils::MarkDeviceRegistered(
+        base::Bind(&EnrollmentScreen::ShowAttributePromptScreen,
+                   weak_ptr_factory_.GetWeakPtr()));
+  } else {
+    StartupUtils::MarkDeviceRegistered(
+        base::Bind(&EnrollmentScreen::ShowEnrollmentStatusOnSuccess,
+                   weak_ptr_factory_.GetWeakPtr()));
+  }
+
   if (remora_controller_) {
     remora_controller_->OnEnrollmentStatusChanged(
         HostPairingController::ENROLLMENT_STATUS_SUCCESS);
   }
 }
 
+void EnrollmentScreen::OnDeviceAttributeUploadCompleted(bool success) {
+  if (success) {
+    actor_->ShowEnrollmentStatus(policy::EnrollmentStatus::ForStatus(
+        policy::EnrollmentStatus::STATUS_SUCCESS));
+  } else {
+    actor_->ShowEnrollmentStatus(policy::EnrollmentStatus::ForStatus(
+        policy::EnrollmentStatus::STATUS_ATTRIBUTE_UPDATE_FAILED));
+  }
+}
+
+void EnrollmentScreen::ShowAttributePromptScreen() {
+  policy::BrowserPolicyConnectorChromeOS* connector =
+      g_browser_process->platform_part()->browser_policy_connector_chromeos();
+  policy::DeviceCloudPolicyManagerChromeOS* policy_manager =
+      connector->GetDeviceCloudPolicyManager();
+
+  policy::CloudPolicyStore* store = policy_manager->core()->store();
+
+  const enterprise_management::PolicyData* policy = store->policy();
+
+  std::string asset_id = policy ? policy->annotated_asset_id() : std::string();
+  std::string location = policy ? policy->annotated_location() : std::string();
+  actor_->ShowAttributePromptScreen(asset_id, location);
+}
+
 void EnrollmentScreen::SendEnrollmentAuthToken(const std::string& token) {
   // TODO(achuith, zork): Extract and send domain.
   DCHECK(shark_controller_);
   shark_controller_->OnAuthenticationDone("", token);
 }
 
 void EnrollmentScreen::ShowEnrollmentStatusOnSuccess() {
   StartupUtils::MarkOobeCompleted();
   if (elapsed_timer_)
     UMA_ENROLLMENT_TIME(kMetricEnrollmentTimeSuccess, elapsed_timer_);
@@ skipping 14 matching lines @@
   enrollment_failed_once_ = true;
   if (elapsed_timer_)
     UMA_ENROLLMENT_TIME(kMetricEnrollmentTimeFailure, elapsed_timer_);
   if (remora_controller_) {
     remora_controller_->OnEnrollmentStatusChanged(
         HostPairingController::ENROLLMENT_STATUS_FAILURE);
   }
 }
 
 }  // namespace chromeos