Added OOBE Asset / Naming prompt

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

 // Copyright 2014 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/enterprise_enrollment_helper_impl.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 169 matching lines @@
   CHECK(dcp_initializer);
   dcp_initializer->StartEnrollment(
       policy::MANAGEMENT_MODE_ENTERPRISE_MANAGED,
       connector->device_management_service(),
       nullptr /* owner_settings_service */, enrollment_config_, token,
       device_modes,
       base::Bind(&EnterpriseEnrollmentHelperImpl::OnEnrollmentFinished,
                  weak_ptr_factory_.GetWeakPtr()));
 }
 
+void EnterpriseEnrollmentHelperImpl::GetDeviceAttributeUpdatePermission() {
+  policy::BrowserPolicyConnectorChromeOS* connector =
+      g_browser_process->platform_part()->browser_policy_connector_chromeos();
+  policy::DeviceCloudPolicyManagerChromeOS* policy_manager =
+      connector->GetDeviceCloudPolicyManager();
+  policy::CloudPolicyClient* client = policy_manager->core()->client();
+
+  client->GetDeviceAttributeUpdatePermission(
+      GetOAuthToken(),
+      base::Bind(
+          &EnterpriseEnrollmentHelperImpl::OnDeviceAttributeUpdatePermission,
+          weak_ptr_factory_.GetWeakPtr()));
+}
+
+void EnterpriseEnrollmentHelperImpl::UpdateDeviceAttributes(
+    const std::string& asset_id,
+    const std::string& location) {
+  policy::BrowserPolicyConnectorChromeOS* connector =
+      g_browser_process->platform_part()->browser_policy_connector_chromeos();
+  policy::DeviceCloudPolicyManagerChromeOS* policy_manager =
+      connector->GetDeviceCloudPolicyManager();
+  policy::CloudPolicyClient* client = policy_manager->core()->client();
+
+  client->UpdateDeviceAttributes(
+      GetOAuthToken(), asset_id, location,
+      base::Bind(
+          &EnterpriseEnrollmentHelperImpl::OnDeviceAttributeUploadCompleted,
+          weak_ptr_factory_.GetWeakPtr()));
+}
+
 void EnterpriseEnrollmentHelperImpl::OnTokenFetched(
     size_t fetcher_index,
     const std::string& token,
     const GoogleServiceAuthError& error) {
   CHECK_LT(fetcher_index, oauth_fetchers_.size());
 
   if (error.state() != GoogleServiceAuthError::NONE)
     last_auth_error_ = error;
 
   ++oauth_fetchers_finished_;
@@ skipping 18 matching lines @@
   finished_ = true;
   if (status.status() == policy::EnrollmentStatus::STATUS_SUCCESS) {
     success_ = true;
     DCHECK(!fetch_additional_token_ || !additional_token_.empty());
     status_consumer()->OnDeviceEnrolled(additional_token_);
   } else {
     status_consumer()->OnEnrollmentError(status);
   }
 }
 
+void EnterpriseEnrollmentHelperImpl::OnDeviceAttributeUpdatePermission(
+    bool status) {
+  status_consumer()->OnDeviceAttributeUpdatePermission(status);
+}
+
+void EnterpriseEnrollmentHelperImpl::OnDeviceAttributeUploadCompleted(
+    bool status) {
+  status_consumer()->OnDeviceAttributeUploadCompleted(status);
+}
+
 void EnterpriseEnrollmentHelperImpl::ReportAuthStatus(
     const GoogleServiceAuthError& error) {
   switch (error.state()) {
     case GoogleServiceAuthError::NONE:
     case GoogleServiceAuthError::CAPTCHA_REQUIRED:
     case GoogleServiceAuthError::TWO_FACTOR:
     case GoogleServiceAuthError::HOSTED_NOT_ALLOWED:
     case GoogleServiceAuthError::INVALID_GAIA_CREDENTIALS:
     case GoogleServiceAuthError::REQUEST_CANCELED:
     case GoogleServiceAuthError::UNEXPECTED_SERVICE_RESPONSE:
@@ skipping 133 matching lines @@
       break;
     case policy::EnrollmentStatus::STATUS_ROBOT_REFRESH_STORE_FAILED:
       UMA(policy::kMetricEnrollmentRobotRefreshTokenStoreFailed);
       break;
     case policy::EnrollmentStatus::STATUS_STORE_TOKEN_AND_ID_FAILED:
       // This error should not happen for enterprise enrollment, it only affects
       // consumer enrollment.
       UMA(policy::kMetricEnrollmentStoreTokenAndIdFailed);
       NOTREACHED();
       break;
+    case policy::EnrollmentStatus::STATUS_ATTRIBUTE_UPDATE_FAILED:
+      UMA(policy::kMetricEnrollmentAttributeUpdateFailed);
+      break;
   }
 }
 
 void EnterpriseEnrollmentHelperImpl::UMA(policy::MetricEnrollment sample) {
   EnrollmentUMA(sample, enrollment_config_.mode);
 }
 
 void EnterpriseEnrollmentHelperImpl::OnBrowsingDataRemoverDone() {
   browsing_data_remover_->RemoveObserver(this);
 
   // BrowsingDataRemover deletes itself.
   browsing_data_remover_ = nullptr;
 
   auth_data_cleared_ = true;
 
   std::vector<base::Closure> callbacks_to_run;
   callbacks_to_run.swap(auth_clear_callbacks_);
   for (std::vector<base::Closure>::iterator callback(callbacks_to_run.begin());
        callback != callbacks_to_run.end();
        ++callback) {
     callback->Run();
   }
 }
 
+const std::string& EnterpriseEnrollmentHelperImpl::GetOAuthToken() {
+  DCHECK(oauth_fetchers_.size() > 0);
+  return oauth_fetchers_[0]->oauth2_access_token();
+}
+
 }  // namespace chromeos