// Copyright 2022 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_UPDATER_POLICY_POLICY_MANAGER_H_
#define CHROME_UPDATER_POLICY_POLICY_MANAGER_H_

#include <optional>
#include <string>
#include <vector>

#include "base/memory/scoped_refptr.h"
#include "base/values.h"
#include "chrome/updater/policy/manager.h"

namespace updater {

// A policy manager that holds all policies in-memory. Main purposes for this
// class:
//   1) Provides a way for policy override, esp. for testing.
//   2) Cache policies for those providers when loading policies is expensive.
//
class PolicyManager : public PolicyManagerInterface {
 public:
  explicit PolicyManager(base::DictValue policies);
  PolicyManager(const PolicyManager&) = delete;
  PolicyManager& operator=(const PolicyManager&) = delete;

  std::optional<int> GetIntegerPolicy(const std::string& key) const;
  std::optional<std::string> GetStringPolicy(const std::string& key) const;

  // Overrides for PolicyManagerInterface.
  std::string source() const override;

  bool HasActiveDevicePolicies() const override;

  std::optional<bool> CloudPolicyOverridesPlatformPolicy() const override;
  std::optional<base::TimeDelta> GetLastCheckPeriod() const override;
  std::optional<UpdatesSuppressedTimes> GetUpdatesSuppressedTimes()
      const override;
  std::optional<std::string> GetDownloadPreference() const override;
  std::optional<int> GetPackageCacheSizeLimitMBytes() const override;
  std::optional<int> GetPackageCacheExpirationTimeDays() const override;
  std::optional<int> GetEffectivePolicyForAppInstalls(
      const std::string& app_id) const override;
  std::optional<int> GetEffectivePolicyForAppUpdates(
      const std::string& app_id) const override;
  std::optional<std::string> GetTargetVersionPrefix(
      const std::string& app_id) const override;
  std::optional<bool> IsRollbackToTargetVersionAllowed(
      const std::string& app_id) const override;
  std::optional<int> GetMajorVersionRolloutPolicy(
      const std::string& app_id) const override;
  std::optional<int> GetMinorVersionRolloutPolicy(
      const std::string& app_id) const override;
  std::optional<std::string> GetProxyMode() const override;
  std::optional<std::string> GetProxyPacUrl() const override;
  std::optional<std::string> GetProxyServer() const override;
  std::optional<std::string> GetTargetChannel(
      const std::string& app_id) const override;
  std::optional<std::vector<std::string>> GetForceInstallApps() const override;
  std::optional<std::vector<std::string>> GetAppsWithPolicy() const override;

 protected:
  ~PolicyManager() override;

 private:
  const base::DictValue policies_;
  std::vector<std::string> force_install_apps_;
};

// A factory method to create a dict policy manager.
scoped_refptr<PolicyManagerInterface> CreateDictPolicyManager(
    base::DictValue policies);

}  // namespace updater

#endif  // CHROME_UPDATER_POLICY_POLICY_MANAGER_H_