// 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/web_applications/test/web_app_install_test_utils.h"

#include "base/command_line.h"
#include "base/functional/bind.h"
#include "base/functional/concurrent_callbacks.h"
#include "base/run_loop.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/bind.h"
#include "base/test/test_future.h"
#include "build/build_config.h"
#include "chrome/browser/apps/app_service/app_registry_cache_waiter.h"
#include "chrome/browser/apps/app_service/app_service_proxy_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/web_applications/jobs/manifest_to_web_app_install_info_job.h"
#include "chrome/browser/web_applications/mojom/user_display_mode.mojom.h"
#include "chrome/browser/web_applications/os_integration/os_integration_manager.h"
#include "chrome/browser/web_applications/os_integration/os_integration_test_override.h"
#include "chrome/browser/web_applications/proto/web_app_install_state.pb.h"
#include "chrome/browser/web_applications/test/fake_web_app_provider.h"
#include "chrome/browser/web_applications/test/web_app_icon_test_utils.h"
#include "chrome/browser/web_applications/web_app_command_scheduler.h"
#include "chrome/browser/web_applications/web_app_constants.h"
#include "chrome/browser/web_applications/web_app_helpers.h"
#include "chrome/browser/web_applications/web_app_install_info.h"
#include "chrome/browser/web_applications/web_app_install_params.h"
#include "chrome/browser/web_applications/web_app_install_utils.h"
#include "chrome/browser/web_applications/web_app_management_type.h"
#include "chrome/browser/web_applications/web_app_provider.h"
#include "chrome/browser/web_applications/web_app_registrar.h"
#include "chrome/common/buildflags.h"
#include "chrome/common/chrome_switches.h"
#include "components/services/app_service/public/cpp/app_types.h"
#include "components/services/app_service/public/cpp/types_util.h"
#include "components/webapps/browser/install_result_code.h"
#include "components/webapps/browser/installable/installable_metrics.h"
#include "components/webapps/browser/uninstall_result_code.h"
#include "components/webapps/common/web_app_id.h"
#include "content/public/browser/web_contents.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/mojom/manifest/display_mode.mojom-shared.h"
#include "url/gurl.h"

namespace web_app {
namespace test {
namespace {

WebAppInstallDialogCallback CreateAcceptDialogCallback() {
  return base::BindOnce(
      [](base::WeakPtr<WebAppScreenshotFetcher> screenshot_fetcher,
         content::WebContents* initiator_web_contents,
         std::unique_ptr<WebAppInstallInfo> web_app_info,
         WebAppInstallationAcceptanceCallback acceptance_callback) {
        web_app_info->user_display_mode = mojom::UserDisplayMode::kStandalone;
        std::move(acceptance_callback)
            .Run(/*accept=*/true, std::move(web_app_info),
                 base::BindOnce([](bool success,
                                   base::OnceClosure reparent_or_launch_app) {
                   if (success && reparent_or_launch_app) {
                     std::move(reparent_or_launch_app).Run();
                   }
                 }));
      });
}

bool IsValidInstallSourceForWebContentsInstall(
    webapps::WebappInstallSource install_source) {
  switch (install_source) {
    case webapps::WebappInstallSource::OMNIBOX_INSTALL_ICON:
      return true;
    case webapps::WebappInstallSource::MENU_BROWSER_TAB:
      return true;
    case webapps::WebappInstallSource::AUTOMATIC_PROMPT_BROWSER_TAB:
      return true;
    case webapps::WebappInstallSource::DEVTOOLS:
      return true;
    default:
      return false;
  }
}

web_app::FallbackBehavior GetFallbackBehaviorFromInstallSource(
    webapps::WebappInstallSource install_source) {
  switch (install_source) {
    case webapps::WebappInstallSource::OMNIBOX_INSTALL_ICON:
      return web_app::FallbackBehavior::kCraftedManifestOnly;
    case webapps::WebappInstallSource::MENU_BROWSER_TAB:
      return web_app::FallbackBehavior::kUseFallbackInfoWhenNotInstallable;
    case webapps::WebappInstallSource::AUTOMATIC_PROMPT_BROWSER_TAB:
      return web_app::FallbackBehavior::kCraftedManifestOnly;
    case webapps::WebappInstallSource::DEVTOOLS:
      return web_app::FallbackBehavior::kUseFallbackInfoWhenNotInstallable;
    default:
      return web_app::FallbackBehavior::kCraftedManifestOnly;
  }
}

}  // namespace

void WaitUntilReady(WebAppProvider* provider) {
  if (provider->on_registry_ready().is_signaled())
    return;

  base::RunLoop run_loop(base::RunLoop::Type::kNestableTasksAllowed);
  provider->on_registry_ready().Post(FROM_HERE, run_loop.QuitClosure());
  run_loop.Run();
}

void WaitUntilWebAppProviderAndSubsystemsReady(WebAppProvider* provider) {
  WaitUntilReady(provider);

  if (provider->on_external_managers_synchronized().is_signaled()) {
    return;
  }

  base::RunLoop run_loop(base::RunLoop::Type::kNestableTasksAllowed);
  provider->on_external_managers_synchronized().Post(FROM_HERE,
                                                     run_loop.QuitClosure());
  run_loop.Run();
}

void AwaitStartWebAppProviderAndSubsystems(Profile* profile) {
  base::CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kDisableDefaultApps);
  FakeWebAppProvider* provider = FakeWebAppProvider::Get(profile);
  DCHECK(provider);
  provider->StartWithSubsystems();
  WaitUntilWebAppProviderAndSubsystemsReady(provider);
}

webapps::AppId InstallDummyWebApp(
    Profile* profile,
    const std::string& app_name,
    const GURL& start_url,
    const webapps::WebappInstallSource install_source) {
  auto web_app_info =
      WebAppInstallInfo::CreateWithStartUrlForTesting(start_url);
  web_app_info->scope = start_url.GetWithoutFilename();
  web_app_info->title = base::UTF8ToUTF16(app_name);
  web_app_info->description = base::UTF8ToUTF16(app_name);
  web_app_info->user_display_mode = mojom::UserDisplayMode::kStandalone;
  web_app_info->install_url = start_url;

  return InstallWebApp(profile, std::move(web_app_info),
                       /*overwrite_existing_manifest_fields=*/true,
                       install_source);
}

webapps::AppId InstallWebApp(Profile* profile,
                             std::unique_ptr<WebAppInstallInfo> web_app_info,
                             bool overwrite_existing_manifest_fields,
                             webapps::WebappInstallSource install_source) {
  // The sync system requires that sync entity name is never empty.
  if (web_app_info->title.empty())
    web_app_info->title = u"WebAppInstallInfo App Name";

  // Ensure web apps can never be installed with an empty scope.
  if (web_app_info->scope.is_empty()) {
    web_app_info->scope = web_app_info->start_url().GetWithoutFilename();
  }

  base::test::TestFuture<const webapps::AppId&, webapps::InstallResultCode>
      future;
  auto* provider = WebAppProvider::GetForTest(profile);
  DCHECK(provider);
  WaitUntilReady(provider);

  // In unit tests, we do not have Browser or WebContents instances. Hence we
  // use `InstallFromInfoCommand` instead of `FetchManifestAndInstallCommand` or
  // `WebAppInstallCommand` to install the web app.

  // If OS integration is being handed by the test (as in, it is suppressing it
  // in the OsIntegrationManager or it is using the OsIntegrationTestOverride),
  // then we can safely install with os integration. Otherwise, keep it off
  // until we can solve this better later. Note: On ChromeOS, we never need to
  // skip OS integration, so always keep it enabled.
  // TODO(https://crbug.com/328524602): Have all installs do OS integration, and
  // ensure that all browsertest / unit tests classes have the execution of the
  // integration suppressed or overridden.
  WebAppInstallParams params;
#if !BUILDFLAG(IS_CHROMEOS)
  auto does_test_handle_os_integration = []() {
    return OsIntegrationTestOverride::Get() ||
           OsIntegrationManager::AreOsHooksSuppressedForTesting();
  };

  if (!does_test_handle_os_integration()) {
    params.install_state =
        proto::InstallState::INSTALLED_WITHOUT_OS_INTEGRATION;
    params.add_to_applications_menu = false;
    params.add_to_desktop = false;
    params.add_to_quick_launch_bar = false;
  }
#endif
  provider->scheduler().InstallFromInfoWithParams(
      std::move(web_app_info), overwrite_existing_manifest_fields,
      install_source, future.GetCallback(), params);

  EXPECT_EQ(webapps::InstallResultCode::kSuccessNewInstall,
            future.Get<webapps::InstallResultCode>());
  // Allow updates to be published to App Service listeners.
  base::RunLoop().RunUntilIdle();

  webapps::AppId app_id = future.Get<webapps::AppId>();
  if (apps::AppServiceProxyFactory::IsAppServiceAvailableForProfile(profile)) {
    apps::AppReadinessWaiter(profile, app_id,
                             base::BindRepeating(apps_util::IsInstalled))
        .Await();
  }
  return app_id;
}

webapps::AppId InstallWebAppWithoutOsIntegration(
    Profile* profile,
    std::unique_ptr<WebAppInstallInfo> web_app_info,
    bool overwrite_existing_manifest_fields,
    webapps::WebappInstallSource install_source) {
  // Use InstallShortcut for Create Shortcut install source.
  CHECK_NE(install_source, webapps::WebappInstallSource::MENU_CREATE_SHORTCUT);

  // The sync system requires that sync entity name is never empty.
  if (web_app_info->title.empty()) {
    web_app_info->title = u"WebAppInstallInfo App Name";
  }

  webapps::AppId app_id;
  base::test::TestFuture<const webapps::AppId&, webapps::InstallResultCode>
      future;
  auto* provider = WebAppProvider::GetForTest(profile);
  DCHECK(provider);
  WaitUntilReady(provider);
  provider->scheduler().InstallFromInfoNoIntegrationForTesting(
      std::move(web_app_info), overwrite_existing_manifest_fields,
      install_source, future.GetCallback());

  EXPECT_EQ(webapps::InstallResultCode::kSuccessNewInstall,
            future.Get<webapps::InstallResultCode>());
  // Allow updates to be published to App Service listeners.
  base::RunLoop().RunUntilIdle();

  return future.Get<webapps::AppId>();
}

void UninstallWebApp(Profile* profile,
                     const webapps::AppId& app_id,
                     webapps::WebappUninstallSource uninstall_source) {
  WebAppProvider* const provider = WebAppProvider::GetForTest(profile);
  base::test::TestFuture<webapps::UninstallResultCode> future;

  DCHECK(provider->registrar_unsafe().CanUserUninstallWebApp(app_id));
  provider->scheduler().RemoveUserUninstallableManagements(
      app_id, uninstall_source, future.GetCallback());

  EXPECT_TRUE(UninstallSucceeded(future.Get()));
  // Allow updates to be published to App Service listeners.
  base::RunLoop().RunUntilIdle();
}

bool UninstallAllWebApps(Profile* profile) {
  bool success = true;
  auto* provider = WebAppProvider::GetForTest(profile);
  if (!provider)
    return false;
  // Wait for any existing commands to complete before reading the registrar.
  provider->command_manager().AwaitAllCommandsCompleteForTesting();
  std::vector<webapps::AppId> app_ids =
      provider->registrar_unsafe().GetAppIds();

  base::ConcurrentCallbacks<webapps::UninstallResultCode> uninstall_callbacks;
  for (auto& app_id : app_ids) {
    const WebApp* app = provider->registrar_unsafe().GetAppById(app_id);
    WebAppManagementTypes sources = app->GetSources();

    // Non-user installs first, as they block user uninstalls.
    for (WebAppManagement::Type app_source : sources) {
      if (kUserUninstallableSources.Has(app_source)) {
        continue;
      }
      provider->scheduler().RemoveInstallManagementMaybeUninstall(
          app_id, app_source, webapps::WebappUninstallSource::kTestCleanup,
          uninstall_callbacks.CreateCallback());
    }
    // Then schedule the user uninstalls if applicable.
    for (WebAppManagement::Type user_uninstallable_source :
         kUserUninstallableSources) {
      if (!sources.Has(user_uninstallable_source)) {
        continue;
      }
      provider->scheduler().RemoveInstallManagementMaybeUninstall(
          app_id, user_uninstallable_source,
          webapps::WebappUninstallSource::kTestCleanup,
          uninstall_callbacks.CreateCallback());
    }
  }
  base::test::TestFuture<std::vector<webapps::UninstallResultCode>>
      uninstalls_future;
  std::move(uninstall_callbacks).Done(uninstalls_future.GetCallback());

  if (!uninstalls_future.Wait()) {
    LOG(ERROR) << "Uninstall timeout";
    return false;
  }

  for (webapps::UninstallResultCode result : uninstalls_future.Get()) {
    if (result == webapps::UninstallResultCode::kError) {
      LOG(ERROR) << "Error uninstalling";
      success = false;
    }
  }
  return success;
}

webapps::AppId InstallForWebContents(
    Profile* profile,
    content::WebContents* web_contents,
    webapps::WebappInstallSource install_source) {
  auto* provider = WebAppProvider::GetForTest(profile);
  base::test::TestFuture<const webapps::AppId&, webapps::InstallResultCode>
      install_future;

  CHECK(IsValidInstallSourceForWebContentsInstall(install_source))
      << "Incorrect WebappInstallSource used to trigger an user triggered "
         "install from tests";

  web_app::FallbackBehavior fallback_behavior =
      GetFallbackBehaviorFromInstallSource(install_source);

  provider->scheduler().FetchManifestAndInstall(
      install_source, web_contents->GetWeakPtr(), CreateAcceptDialogCallback(),
      install_future.GetCallback(), fallback_behavior);
  EXPECT_TRUE(install_future.Wait());
  provider->command_manager().AwaitAllCommandsCompleteForTesting();
  return install_future.Get<webapps::AppId>();
}

std::unique_ptr<WebAppInstallInfo> GetInstallInfoForCurrentManifest(
    base::WeakPtr<content::WebContents> web_contents,
    const blink::mojom::Manifest& manifest,
    WebAppInstallInfoConstructOptions construct_options) {
  Profile* profile =
      Profile::FromBrowserContext(web_contents->GetBrowserContext());
  auto* provider = WebAppProvider::GetForTest(profile);
  base::DictValue debug_data;
  base::test::TestFuture<std::unique_ptr<WebAppInstallInfo>> test_future;
  std::unique_ptr<WebAppDataRetriever> retriever =
      provider->web_contents_manager().CreateDataRetriever();
  auto job = ManifestToWebAppInstallInfoJob::CreateAndStart(
      manifest, *retriever.get(), /*background_installation=*/false,
      webapps::WebappInstallSource::OMNIBOX_INSTALL_ICON, web_contents,
      [](IconUrlSizeSet&) {}, debug_data, test_future.GetCallback(),
      construct_options);
  EXPECT_TRUE(test_future.Wait(base::RunLoop::Type::kNestableTasksAllowed));
  return test_future.Take();
}

}  // namespace test
}  // namespace web_app