drive: Move IsDriveEnabled() and DisableDrive() to DriveSystemService

chrome/browser/chromeos/gdata/drive_system_service.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/gdata/drive_system_service.h"
 
 #include "base/bind.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/gdata/drive_api_service.h"
 #include "chrome/browser/chromeos/gdata/drive_download_observer.h"
@@ skipping 23 matching lines @@
 using content::BrowserContext;
 using content::BrowserThread;
 
 namespace gdata {
 namespace {
 
 // Used in test to setup system service.
 DriveServiceInterface* g_test_drive_service = NULL;
 const std::string* g_test_cache_root = NULL;
 
+// Map to collect profiles with Drive disabled.
+std::map<Profile*, bool>* g_drive_disabled_map = NULL;
+
+// Disables Drive for the specified profile. Used to disable Drive when
+// needed (ex. initialization of the Drive cache failed).
+// Must be called on UI thread.
+void DisableDrive(Profile* profile) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+
+  // We don't change kDisableGData preference here. If we do, we'll end up
+  // disabling Drive on other devices, as kDisableGData is a syncable
+  // preference. Hence the map is used here.
+  if (!g_drive_disabled_map)
+    g_drive_disabled_map = new std::map<Profile*, bool>;
+
+  g_drive_disabled_map->insert(std::make_pair(profile, true));
+}
+
 }  // namespace
 
 DriveSystemService::DriveSystemService(Profile* profile)
     : profile_(profile),
       cache_(NULL),
       ALLOW_THIS_IN_INITIALIZER_LIST(weak_ptr_factory_(this)) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   base::SequencedWorkerPool* blocking_pool = BrowserThread::GetBlockingPool();
   blocking_task_runner_ = blocking_pool->GetSequencedTaskRunner(
       blocking_pool->GetSequenceToken());
@@ skipping 43 matching lines @@
   stale_cache_files_remover_.reset();
   sync_client_.reset();
   download_observer_.reset();
   file_write_helper_.reset();
   file_system_.reset();
   webapps_registry_.reset();
   uploader_.reset();
   drive_service_.reset();
 }
 
+// static
+bool DriveSystemService::IsDriveEnabled(Profile* profile) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+
+  if (!AuthService::CanAuthenticate(profile))
+    return false;
+
+  // Disable gdata if preference is set.  This can happen with commandline flag
+  // --disable-gdata or enterprise policy, or probably with user settings too
+  // in the future.
+  if (profile->GetPrefs()->GetBoolean(prefs::kDisableGData))
+    return false;
+
+  if (g_drive_disabled_map && g_drive_disabled_map->count(profile) > 0)
+    return false;
+
+  return true;
+}
+
 void DriveSystemService::ClearCacheAndRemountFileSystem(
     const base::Callback<void(bool)>& callback) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   RemoveDriveMountPoint();
   drive_service()->CancelAll();
   cache_->ClearAllOnUIThread(
       base::Bind(&DriveSystemService::AddBackDriveMountPoint,
                  weak_ptr_factory_.GetWeakPtr(),
                  callback));
 }
 
 void DriveSystemService::AddBackDriveMountPoint(
     const base::Callback<void(bool)>& callback,
     DriveFileError error,
     const FilePath& file_path) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   file_system_->Initialize();
   AddDriveMountPoint();
 
   if (!callback.is_null())
     callback.Run(error == DRIVE_FILE_OK);
 }
 
 void DriveSystemService::AddDriveMountPoint() {
-  if (!gdata::util::IsDriveEnabled(profile_))
+  if (!IsDriveEnabled(profile_))
     return;
 
   const FilePath mount_point = gdata::util::GetDriveMountPointPath();
   fileapi::ExternalFileSystemMountPointProvider* provider =
       BrowserContext::GetDefaultStoragePartition(profile_)->
           GetFileSystemContext()->external_provider();
   if (provider && !provider->HasMountPoint(mount_point)) {
     provider->AddRemoteMountPoint(
         mount_point,
         new DriveFileSystemProxy(file_system_.get()));
@@ skipping 12 matching lines @@
           GetFileSystemContext()->external_provider();
   if (provider && provider->HasMountPoint(mount_point))
     provider->RemoveMountPoint(mount_point);
 }
 
 void DriveSystemService::OnCacheInitialized(bool success) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   if (!success) {
     LOG(WARNING) << "Failed to initialize the cache. Disabling Drive";
-    gdata::util::DisableDrive(profile_);
+    DisableDrive(profile_);
     // Change the download directory to the default value if the download
     // destination is set to under Drive mount point.
     //
     // TODO(satorux): This cannot be done in DisableDrive(), as there is a
     // dependency problem. We should move this code to DisableDrive() once
     // the dependency problem is solved. crbug.com/153962
     PrefService* pref_service = profile_->GetPrefs();
     if (gdata::util::IsUnderDriveMountPoint(
             pref_service->GetFilePath(prefs::kDownloadDefaultDirectory))) {
       pref_service->SetFilePath(prefs::kDownloadDefaultDirectory,
@@ skipping 76 matching lines @@
       g_test_cache_root ? FilePath(*g_test_cache_root) :
                           DriveCache::GetCacheRootPath(profile);
   delete g_test_cache_root;
   g_test_cache_root = NULL;
 
   service->Initialize(drive_service, cache_root);
   return service;
 }
 
 }  // namespace gdata