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

#include "chrome/browser/component_updater/first_party_sets_component_installer.h"

#include "base/feature_list.h"
#include "base/logging.h"
#include "base/task/thread_pool.h"
#include "build/branding_buildflags.h"
#include "components/component_updater/installer_policies/first_party_sets_component_installer_policy.h"
#include "content/public/browser/first_party_sets_handler.h"
#include "content/public/common/content_features.h"
#include "net/base/features.h"

namespace {

base::TaskPriority GetTaskPriority() {
#if BUILDFLAG(CHROME_FOR_TESTING)
  // In Chrome for Testing, component updates are either completely disabled or
  // delay browser startup until all required components are up to date. Since
  // BEST_EFFORT tasks are not run until after the browser starts, the
  // 'Related Website Sets' component is not updated along with other required
  // components unless its priority is elevated.
  return base::TaskPriority::USER_BLOCKING;
#else
  return base::TaskPriority::BEST_EFFORT;
#endif  // BUILDFLAG(CHROME_FOR_TESTING)
}

}  // namespace

namespace component_updater {

void RegisterFirstPartySetsComponent(ComponentUpdateService* cus) {
  if (!content::FirstPartySetsHandler::GetInstance()->IsEnabled()) {
    return;
  }

  VLOG(1) << "Registering Related Website Sets component.";

  auto policy = std::make_unique<FirstPartySetsComponentInstallerPolicy>(
      /*on_sets_ready=*/base::BindOnce([](base::Version version,
                                          base::File sets_file) {
        VLOG(1) << "Received Related Website Sets";
        content::FirstPartySetsHandler::GetInstance()->SetPublicFirstPartySets(
            version, std::move(sets_file));
      }),
      GetTaskPriority());

  FirstPartySetsComponentInstallerPolicy* raw_policy = policy.get();
  // Dereferencing `raw_policy` this way is safe because the closure is invoked
  // by the ComponentInstaller instance, which owns `policy` (so they have the
  // same lifetime). Therefore if/when the closure is invoked, `policy` is still
  // alive.
  base::MakeRefCounted<ComponentInstaller>(
      std::move(policy), /*action_handler=*/nullptr, GetTaskPriority())
      ->Register(cus, base::BindOnce(
                          [](FirstPartySetsComponentInstallerPolicy* policy) {
                            policy->OnRegistrationComplete();
                          },
                          raw_policy));
}

}  // namespace component_updater