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

#ifndef CHROME_BROWSER_PROFILES_PROFILE_KEY_H_
#define CHROME_BROWSER_PROFILES_PROFILE_KEY_H_

#include "base/files/file_path.h"
#include "base/memory/raw_ptr.h"
#include "build/build_config.h"
#include "components/keyed_service/core/simple_factory_key.h"
#include "components/leveldb_proto/public/proto_database_provider.h"

class PrefService;

#if BUILDFLAG(IS_ANDROID)
class ProfileKeyAndroid;
#endif  // BUILDFLAG(IS_ANDROID)

// An embryonic Profile with only fields accessible in reduced mode.
// Used as a SimpleFactoryKey.
class ProfileKey : public SimpleFactoryKey {
 public:
  ProfileKey(const base::FilePath& path,
             ProfileKey* original_key = nullptr);

  ProfileKey(const ProfileKey&) = delete;
  ProfileKey& operator=(const ProfileKey&) = delete;

  ~ProfileKey() override;

  // Profile-specific APIs needed in reduced mode:
  ProfileKey* GetOriginalKey();
  PrefService* GetPrefs();
  void SetPrefs(PrefService* prefs);

  // Gets a pointer to a ProtoDatabaseProvider, this instance is owned by
  // StartupData in Android's reduced mode, and by StoragePartition in all other
  // cases. Virtual for testing.
  virtual leveldb_proto::ProtoDatabaseProvider* GetProtoDatabaseProvider();
  void SetProtoDatabaseProvider(
      leveldb_proto::ProtoDatabaseProvider* db_provider);

  static ProfileKey* FromSimpleFactoryKey(SimpleFactoryKey* key);

#if BUILDFLAG(IS_ANDROID)
  ProfileKeyAndroid* GetProfileKeyAndroid();
#endif  // BUILDFLAG(IS_ANDROID)

 private:
  raw_ptr<PrefService> prefs_ = nullptr;
  raw_ptr<leveldb_proto::ProtoDatabaseProvider, AcrossTasksDanglingUntriaged>
      db_provider_ = nullptr;

  // Points to the original (non off-the-record) ProfileKey.
  raw_ptr<ProfileKey> original_key_ = nullptr;

#if BUILDFLAG(IS_ANDROID)
  std::unique_ptr<ProfileKeyAndroid> profile_key_android_;
#endif  // BUILDFLAG(IS_ANDROID)
};

#endif  // CHROME_BROWSER_PROFILES_PROFILE_KEY_H_