// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ash/policy/skyvault/policy_utils.h"

#include <optional>

#include "ash/constants/ash_features.h"
#include "ash/constants/ash_pref_names.h"
#include "ash/constants/chrome_pref_names.h"
#include "base/command_line.h"
#include "base/feature_list.h"
#include "base/files/file_path.h"
#include "base/time/time.h"
#include "chrome/browser/ash/policy/skyvault/file_location_utils.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/common/chrome_features.h"
#include "components/prefs/pref_service.h"

namespace policy::local_user_files {

namespace {

FileSaveDestination GetDestinationForPref(
    Profile* profile,
    const std::string& pref_name,
    FileSaveDestination fallback = FileSaveDestination::kDownloads) {
  DCHECK(profile);
  auto* pref = profile->GetPrefs()->FindPreference(pref_name);
  if (!pref || !pref->GetValue() || !pref->IsManaged()) {
    return FileSaveDestination::kNotSpecified;
  }
  const auto value_str = pref->GetValue()->GetString();
  if (!IsValidLocationString(value_str)) {
    return FileSaveDestination::kNotSpecified;
  }
  if (value_str.find(kGoogleDrivePolicyVariableName) != std::string::npos) {
    return FileSaveDestination::kGoogleDrive;
  }
  if (value_str.find(kOneDrivePolicyVariableName) != std::string::npos) {
    return FileSaveDestination::kOneDrive;
  }

  return fallback;
}

}  // namespace

constexpr char kGoogleDrivePolicyVariableName[] = "${google_drive}";
constexpr char kOneDrivePolicyVariableName[] = "${microsoft_onedrive}";

constexpr char kMigrationDestinationGoogleDrive[] = "google_drive";
constexpr char kMigrationDestinationOneDrive[] = "microsoft_onedrive";
constexpr char kMigrationDestinationDelete[] = "delete";

bool LocalUserFilesAllowed(const PrefService& local_state) {
  // If the flag is disabled, ignore the policy value and allow local storage.
  if (!base::FeatureList::IsEnabled(features::kSkyVault)) {
    return true;
  }
  return local_state.GetBoolean(ash::prefs::kLocalUserFilesAllowed);
}

MigrationDestination GetMigrationDestination(const PrefService& local_state) {
  if (!base::FeatureList::IsEnabled(features::kSkyVault) ||
      !base::FeatureList::IsEnabled(ash::features::kSkyVaultV2)) {
    return MigrationDestination::kNotSpecified;
  }

  const std::string destination =
      local_state.GetString(ash::prefs::kLocalUserFilesMigrationDestination);

  if (destination == kMigrationDestinationGoogleDrive) {
    return MigrationDestination::kGoogleDrive;
  }
  if (destination == kMigrationDestinationOneDrive) {
    return MigrationDestination::kOneDrive;
  }
  if (base::FeatureList::IsEnabled(ash::features::kSkyVaultV3) &&
      destination == kMigrationDestinationDelete) {
    return MigrationDestination::kDelete;
  }
  return MigrationDestination::kNotSpecified;
}

bool IsCloudDestination(MigrationDestination destination) {
  return destination == MigrationDestination::kGoogleDrive ||
         destination == MigrationDestination::kOneDrive;
}

FileSaveDestination GetDownloadsDestination(Profile* profile) {
  return GetDestinationForPref(profile,
                               ash::chrome_prefs::kDownloadDefaultDirectory);
}

FileSaveDestination GetScreenCaptureDestination(Profile* profile) {
  return GetDestinationForPref(profile, ash::prefs::kCaptureModePolicySavePath);
}

FileSaveDestination GetCameraDestination(Profile* profile) {
  return GetDestinationForPref(profile, ash::prefs::kCameraSaveLocation,
                               FileSaveDestination::kNotSpecified);
}

bool DownloadToTemp(Profile* profile) {
  return base::FeatureList::IsEnabled(features::kSkyVault) &&
         GetDownloadsDestination(profile) == FileSaveDestination::kOneDrive;
}

base::FilePath GetMyFilesPath(Profile* profile) {
  return profile->GetPath().Append("MyFiles");
}

std::optional<base::Time> GetMigrationStartTime(Profile* profile) {
  if (!base::FeatureList::IsEnabled(ash::features::kSkyVaultV3)) {
    return std::nullopt;
  }
  PrefService* pref_service = profile->GetPrefs();
  base::Time scheduled_start_time =
      pref_service->GetTime(ash::prefs::kSkyVaultMigrationScheduledStartTime);
  if (scheduled_start_time.is_null()) {
    LOG(ERROR) << "Migration/deletion start time cannot be determined.";
    return std::nullopt;
  }
  return scheduled_start_time;
}

}  // namespace policy::local_user_files