// 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.

#ifndef CHROME_BROWSER_COMPONENT_UPDATER_TRANSLATE_KIT_COMPONENT_INSTALLER_H_
#define CHROME_BROWSER_COMPONENT_UPDATER_TRANSLATE_KIT_COMPONENT_INSTALLER_H_

#include <string>

#include "base/functional/callback_forward.h"
#include "base/gtest_prod_util.h"
#include "base/memory/raw_ptr.h"
#include "base/values.h"
#include "components/component_updater/component_installer.h"
#include "components/prefs/pref_service.h"
#include "components/update_client/update_client.h"

namespace component_updater {

class ComponentUpdateService;

// The installer policy for the TranslateKit Component.
class TranslateKitComponentInstallerPolicy : public ComponentInstallerPolicy {
 public:
  explicit TranslateKitComponentInstallerPolicy(
      PrefService* pref_service,
      base::RepeatingClosure on_ready_callback);
  ~TranslateKitComponentInstallerPolicy() override;

  // Not Copyable.
  TranslateKitComponentInstallerPolicy(
      const TranslateKitComponentInstallerPolicy&) = delete;
  TranslateKitComponentInstallerPolicy& operator=(
      const TranslateKitComponentInstallerPolicy&) = delete;

  // Requests to update the component.
  static void UpdateComponentOnDemand(ComponentUpdateService* cus);

  static const std::string GetExtensionId();

 private:
  FRIEND_TEST_ALL_PREFIXES(RegisterTranslateKitComponentTest,
                           VerifyInstallationDefaultEmpty);

  // `ComponentInstallerPolicy` overrides:
  bool VerifyInstallation(const base::DictValue& manifest,
                          const base::FilePath& install_dir) const override;
  bool SupportsGroupPolicyEnabledComponentUpdates() const override;
  bool RequiresNetworkEncryption() const override;
  update_client::CrxInstaller::Result OnCustomInstall(
      const base::DictValue& manifest,
      const base::FilePath& install_dir) override;
  void OnCustomUninstall() override;
  void ComponentReady(const base::Version& version,
                      const base::FilePath& install_dir,
                      base::DictValue manifest) override;
  base::FilePath GetRelativeInstallDir() const override;
  void GetHash(std::vector<uint8_t>* hash) const override;
  std::string GetName() const override;
  update_client::InstallerAttributes GetInstallerAttributes() const override;

#if BUILDFLAG(IS_CHROMEOS)
  void OnImageLoaderComponentLoaded(std::optional<base::FilePath> mount_path);
#endif  // BUILDFLAG(IS_CHROMEOS)

  raw_ptr<PrefService> pref_service_;
  base::RepeatingClosure on_ready_callback_;
  base::WeakPtrFactory<TranslateKitComponentInstallerPolicy> weak_factory_{
      this};
};

// Call once during startup to make the component update service aware of
// the TranslateKit component.
void RegisterTranslateKitComponent(ComponentUpdateService* cus,
                                   PrefService* pref_service,
                                   bool force_install,
                                   base::OnceClosure registered_callback,
                                   base::RepeatingClosure on_ready_callback);

}  // namespace component_updater

#endif  // CHROME_BROWSER_COMPONENT_UPDATER_TRANSLATE_KIT_COMPONENT_INSTALLER_H_