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

#include "chrome/browser/apps/intent_helper/preferred_apps_test_util.h"

#include "base/check_deref.h"
#include "base/run_loop.h"
#include "base/scoped_observation.h"
#include "chrome/browser/apps/app_service/app_service_proxy.h"
#include "chrome/browser/apps/app_service/app_service_proxy_base.h"
#include "chrome/browser/apps/app_service/app_service_proxy_factory.h"
#include "chrome/browser/profiles/profile.h"

namespace apps_util {

PreferredAppUpdateWaiter::PreferredAppUpdateWaiter(
    apps::PreferredAppsListHandle& handle,
    std::string app_id,
    bool is_preferred_app,
    base::RunLoop::Type run_loop_type)
    : waiting_app_id_(std::move(app_id)),
      waiting_is_preferred_app_(is_preferred_app),
      preferred_apps_(handle),
      run_loop_(run_loop_type) {
  observation_.Observe(&handle);
}

PreferredAppUpdateWaiter::PreferredAppUpdateWaiter(
    apps::AppServiceProxyBase* proxy,
    std::string app_id,
    bool is_preferred_app,
    base::RunLoop::Type run_loop_type)
    : PreferredAppUpdateWaiter(CHECK_DEREF(proxy).PreferredAppsList(),
                               app_id,
                               is_preferred_app,
                               run_loop_type) {}

PreferredAppUpdateWaiter::PreferredAppUpdateWaiter(
    Profile* profile,
    std::string app_id,
    bool is_preferred_app,
    base::RunLoop::Type run_loop_type)
    : PreferredAppUpdateWaiter(
          // The CHECK_DEREF above will catch cases that this returns null.
          apps::AppServiceProxyFactory::GetForProfile(profile),
          app_id,
          is_preferred_app,
          run_loop_type) {}

PreferredAppUpdateWaiter::~PreferredAppUpdateWaiter() = default;

void PreferredAppUpdateWaiter::Wait() {
  if (preferred_apps_->IsPreferredAppForSupportedLinks(waiting_app_id_) !=
      waiting_is_preferred_app_) {
    run_loop_.Run();
  }
}

void PreferredAppUpdateWaiter::OnPreferredAppChanged(const std::string& app_id,
                                                     bool is_preferred_app) {
  if (app_id == waiting_app_id_ &&
      is_preferred_app == waiting_is_preferred_app_) {
    run_loop_.Quit();
  }
}

void PreferredAppUpdateWaiter::OnPreferredAppsListWillBeDestroyed(
    apps::PreferredAppsListHandle* handle) {
  observation_.Reset();
}

PreferredAppsListReadyWaiter::PreferredAppsListReadyWaiter(
    apps::PreferredAppsListHandle& handle)
    : handle_(handle) {
  if (handle_->IsInitialized()) {
    return;
  }
  observation_.Observe(&handle);
}

PreferredAppsListReadyWaiter::~PreferredAppsListReadyWaiter() = default;

void PreferredAppsListReadyWaiter::Wait() {
  if (handle_->IsInitialized()) {
    return;
  }
  run_loop_.Run();
}

void PreferredAppsListReadyWaiter::OnPreferredAppsListInitialized() {
  run_loop_.Quit();
}

void PreferredAppsListReadyWaiter::OnPreferredAppChanged(
    const std::string& app_id,
    bool is_preferred_app) {}

void PreferredAppsListReadyWaiter::OnPreferredAppsListWillBeDestroyed(
    apps::PreferredAppsListHandle* handle) {
  observation_.Reset();
}

void SetSupportedLinksPreferenceAndWait(Profile* profile,
                                        const std::string& app_id) {
  auto* proxy = apps::AppServiceProxyFactory::GetForProfile(profile);
  PreferredAppUpdateWaiter update_waiter(proxy->PreferredAppsList(), app_id);
  proxy->SetSupportedLinksPreference(app_id);
  update_waiter.Wait();
}

void RemoveSupportedLinksPreferenceAndWait(Profile* profile,
                                           const std::string& app_id) {
  auto* proxy = apps::AppServiceProxyFactory::GetForProfile(profile);
  PreferredAppUpdateWaiter update_waiter(proxy->PreferredAppsList(), app_id,
                                         /*is_preferred_app=*/false);
  proxy->RemoveSupportedLinksPreference(app_id);
  update_waiter.Wait();
}

}  // namespace apps_util