chromium/chrome/browser/component_updater/smart_dim_component_installer.h

// Copyright 2020 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_SMART_DIM_COMPONENT_INSTALLER_H_
#define CHROME_BROWSER_COMPONENT_UPDATER_SMART_DIM_COMPONENT_INSTALLER_H_

#include <stdint.h>

#include <memory>
#include <string>
#include <utility>
#include <vector>

#include "base/functional/callback.h"
#include "base/values.h"
#include "components/component_updater/component_installer.h"

namespace base {
class FilePath;
}  // namespace base

namespace component_updater {

class ComponentUpdateService;

class SmartDimComponentInstallerPolicy : public ComponentInstallerPolicy {
 public:
  explicit SmartDimComponentInstallerPolicy(std::string expected_version);

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

  ~SmartDimComponentInstallerPolicy() override;

  static const std::string GetExtensionId();

 private:
  // ComponentInstallerPolicy overrides:
  bool SupportsGroupPolicyEnabledComponentUpdates() const override;
  bool RequiresNetworkEncryption() const override;
  update_client::CrxInstaller::Result OnCustomInstall(
      const base::Value::Dict& manifest,
      const base::FilePath& install_dir) override;
  void OnCustomUninstall() override;
  bool VerifyInstallation(const base::Value::Dict& manifest,
                          const base::FilePath& install_dir) const override;
  void ComponentReady(const base::Version& version,
                      const base::FilePath& install_dir,
                      base::Value::Dict 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;

  // This installer requests exact expected_version_ from the server.
  // Only expected_version_ can pass VerifyInstallation and be fed to
  // SmartDimMlAgent.
  std::string expected_version_;
};

// Call once during startup to make the component update service aware of
// the smart dim component. The optional `callback` is invoked when registration
// is complete.
void RegisterSmartDimComponent(ComponentUpdateService* cus,
                               base::OnceClosure callback = {});

}  // namespace component_updater

#endif  // CHROME_BROWSER_COMPONENT_UPDATER_SMART_DIM_COMPONENT_INSTALLER_H_