// 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 <memory>
#include <utility>

#include "base/files/file_util.h"
#include "base/task/thread_pool.h"
#include "base/test/bind.h"
#include "base/test/metrics/histogram_tester.h"
#include "chrome/browser/apps/app_service/app_registry_cache_waiter.h"
#include "chrome/browser/ui/web_applications/test/web_app_browsertest_util.h"
#include "chrome/browser/ui/web_applications/web_app_browsertest_base.h"
#include "chrome/browser/ui/web_applications/web_app_metrics.h"
#include "chrome/browser/web_applications/commands/web_app_icon_diagnostic_command.h"
#include "chrome/browser/web_applications/os_integration/os_integration_manager.h"
#include "chrome/browser/web_applications/test/web_app_install_test_utils.h"
#include "chrome/browser/web_applications/web_app_provider.h"
#include "chrome/browser/web_applications/web_app_registry_update.h"
#include "chrome/browser/web_applications/web_app_sync_bridge.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "components/webapps/common/web_app_id.h"
#include "content/public/test/browser_test.h"

namespace web_app {

class WebAppIconHealthChecksBrowserTest : public WebAppBrowserTestBase {
 public:
  WebAppIconHealthChecksBrowserTest() {
    WebAppMetrics::DisableAutomaticIconHealthChecksForTesting();
  }

  ~WebAppIconHealthChecksBrowserTest() override = default;

  void SetUpOnMainThread() override {
    WebAppBrowserTestBase::SetUpOnMainThread();
    ASSERT_TRUE(embedded_test_server()->Start());
  }

  Profile* profile() { return browser()->GetProfile(); }

  ScopedRegistryUpdate CreateUpdateScope() {
    return WebAppProvider::GetForTest(profile())
        ->sync_bridge_unsafe()
        .BeginUpdate();
  }

  void RunIconChecksWithMetricExpectations(
      WebAppIconDiagnosticResult expected_result) {
    base::HistogramTester histogram_tester;

    base::RunLoop run_loop;
    WebAppMetrics::Get(profile())->icon_health_checks_for_testing().Start(
        run_loop.QuitClosure());
    run_loop.Run();

    auto check_histogram = [&](const char* histogram, bool subresult) {
      histogram_tester.ExpectUniqueSample(histogram, subresult, 1);
    };
    check_histogram("WebApp.Icon.AppsWithEmptyDownloadedIconSizes",
                    expected_result.has_empty_downloaded_icon_sizes);
    check_histogram("WebApp.Icon.AppsWithGeneratedIconFlag",
                    expected_result.has_generated_icon_flag);
    check_histogram("WebApp.Icon.AppsWithGeneratedIconBitmap",
                    expected_result.has_generated_icon_bitmap);
    check_histogram("WebApp.Icon.AppsWithGeneratedIconFlagFalseNegative",
                    expected_result.has_generated_icon_flag_false_negative);
    check_histogram("WebApp.Icon.AppsWithEmptyIconBitmap",
                    expected_result.has_empty_icon_bitmap);
    check_histogram("WebApp.Icon.AppsWithEmptyIconFile",
                    expected_result.has_empty_icon_file);
    check_histogram("WebApp.Icon.AppsWithMissingIconFile",
                    expected_result.has_missing_icon_file);
  }

  webapps::AppId InstallWebAppAndAwaitAppService(const char* path) {
    webapps::AppId app_id =
        InstallWebAppFromPage(browser(), embedded_test_server()->GetURL(path));
    apps::AppReadinessWaiter(profile(), app_id, apps::Readiness::kReady)
        .Await();
    return app_id;
  }
};

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, HealthyIcons) {
  InstallWebAppAndAwaitAppService("/web_apps/basic.html");
  RunIconChecksWithMetricExpectations({});
}

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, EmptyAppName) {
  webapps::AppId app_id =
      InstallWebAppAndAwaitAppService("/web_apps/basic.html");

  // Delete the app name (some users may have corrupt web app databases with
  // missing app names).
  {
    WebAppSyncBridge& sync_bridge =
        WebAppProvider::GetForTest(profile())->sync_bridge_unsafe();
    sync_bridge.set_disable_checks_for_testing(true);
    ScopedRegistryUpdate update = sync_bridge.BeginUpdate();
    WebApp* web_app = update->UpdateApp(app_id);
    web_app->SetName("");
  }

  // Check that we don't crash on empty app names.
  RunIconChecksWithMetricExpectations({});
}

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest,
                       MissingDownloadedIconSizes) {
  webapps::AppId app_id =
      InstallWebAppAndAwaitAppService("/web_apps/basic.html");
  CreateUpdateScope()->UpdateApp(app_id)->SetDownloadedIconSizes(
      IconPurpose::ANY, {});
  RunIconChecksWithMetricExpectations(
      {.has_empty_downloaded_icon_sizes = true, .has_empty_icon_bitmap = true});
}

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, GeneratedIcon) {
  InstallWebAppAndAwaitAppService("/web_apps/get_manifest.html?no_icons.json");
  RunIconChecksWithMetricExpectations(
      {.has_generated_icon_flag = true, .has_generated_icon_bitmap = true});
}

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest,
                       GeneratedIconFlagFalseNegative) {
  webapps::AppId app_id = InstallWebAppAndAwaitAppService(
      "/web_apps/get_manifest.html?no_icons.json");
  // In https://crbug.com/40835055 manifest update erroneously set
  // is_generated_icon to false.
  CreateUpdateScope()->UpdateApp(app_id)->SetIsGeneratedIcon(false);
  RunIconChecksWithMetricExpectations(
      {.has_generated_icon_flag_false_negative = true,
       .has_generated_icon_bitmap = true});
}

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest,
                       PRE_DeletedIconFiles) {
  webapps::AppId app_id =
      InstallWebAppAndAwaitAppService("/web_apps/basic.html");
  RunIconChecksWithMetricExpectations({});

  // Delete the icons.
  base::RunLoop run_loop;
  WebAppProvider::GetForTest(profile())->icon_manager().DeleteData(
      app_id,
      base::BindLambdaForTesting([&](bool success) { run_loop.Quit(); }));
  run_loop.Run();

  // Restart to reload the app.
}
IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, DeletedIconFiles) {
  RunIconChecksWithMetricExpectations(
      {.has_empty_icon_bitmap = true, .has_missing_icon_file = true});
}

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, PRE_EmptyIconFile) {
  webapps::AppId app_id =
      InstallWebAppAndAwaitAppService("/web_apps/basic.html");
  RunIconChecksWithMetricExpectations({});

  // Empty the contents of one of the icon files.
  base::FilePath icon_path =
      WebAppProvider::GetForTest(profile())
          ->icon_manager()
          .GetIconFilePathForTesting(app_id, IconPurpose::ANY, 32);
  ASSERT_FALSE(icon_path.empty());
  base::RunLoop run_loop;
  base::ThreadPool::CreateSequencedTaskRunner(
      {base::MayBlock(), base::TaskPriority::USER_VISIBLE,
       base::TaskShutdownBehavior::BLOCK_SHUTDOWN})
      ->PostTaskAndReply(FROM_HERE, base::BindLambdaForTesting([icon_path]() {
                           base::WriteFile(icon_path, "");
                         }),
                         run_loop.QuitClosure());
  run_loop.Run();

  // Restart to reload the app.
}
IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, EmptyIconFile) {
  RunIconChecksWithMetricExpectations(
      {.has_empty_icon_bitmap = true, .has_empty_icon_file = true});
}

IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, PRE_CorruptIconFile) {
  webapps::AppId app_id =
      InstallWebAppAndAwaitAppService("/web_apps/basic.html");
  RunIconChecksWithMetricExpectations({});

  // Corrupt the contents of one of the icon files.
  base::FilePath icon_path =
      WebAppProvider::GetForTest(profile())
          ->icon_manager()
          .GetIconFilePathForTesting(app_id, IconPurpose::ANY, 32);
  ASSERT_FALSE(icon_path.empty());
  base::RunLoop run_loop;
  base::ThreadPool::CreateSequencedTaskRunner(
      {base::MayBlock(), base::TaskPriority::USER_VISIBLE,
       base::TaskShutdownBehavior::BLOCK_SHUTDOWN})
      ->PostTaskAndReply(
          FROM_HERE, base::BindLambdaForTesting([icon_path]() {
            base::WriteFile(icon_path, "This is invalid data for a PNG file.");
          }),
          run_loop.QuitClosure());
  run_loop.Run();

  // Restart to reload the app.
}
IN_PROC_BROWSER_TEST_F(WebAppIconHealthChecksBrowserTest, CorruptIconFile) {
  RunIconChecksWithMetricExpectations({.has_empty_icon_bitmap = true});
}

}  // namespace web_app