// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/strings/utf_string_conversions.h"
#include "base/test/bind.h"
#include "build/build_config.h"
#include "chrome/browser/apps/app_service/app_service_proxy.h"
#include "chrome/browser/apps/app_service/app_service_proxy_factory.h"
#include "chrome/browser/apps/app_service/app_service_test.h"
#include "chrome/browser/apps/app_service/browser_app_launcher.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/web_applications/test/web_app_browsertest_util.h"
#include "chrome/browser/ui/web_applications/web_app_browser_controller.h"
#include "chrome/browser/ui/web_applications/web_app_browsertest_base.h"
#include "chrome/browser/web_applications/mojom/user_display_mode.mojom.h"
#include "chrome/browser/web_applications/test/web_app_install_test_utils.h"
#include "chrome/browser/web_applications/web_app_command_scheduler.h"
#include "chrome/browser/web_applications/web_app_icon_generator.h"
#include "chrome/browser/web_applications/web_app_install_info.h"
#include "chrome/browser/web_applications/web_app_install_utils.h"
#include "chrome/browser/web_applications/web_app_provider.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "components/services/app_service/public/cpp/app_launch_params.h"
#include "components/services/app_service/public/cpp/app_launch_util.h"
#include "components/webapps/browser/install_result_code.h"
#include "components/webapps/browser/installable/installable_metrics.h"
#include "content/public/test/browser_test.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "third_party/skia/include/core/SkColor.h"
#include "ui/base/window_open_disposition.h"
#include "ui/gfx/image/image_skia.h"
#include "url/gurl.h"

namespace web_app {

class WebAppIconManagerBrowserTest : public WebAppBrowserTestBase {
 public:
  WebAppIconManagerBrowserTest() = default;
  WebAppIconManagerBrowserTest(const WebAppIconManagerBrowserTest&) = delete;
  WebAppIconManagerBrowserTest& operator=(const WebAppIconManagerBrowserTest&) =
      delete;

  ~WebAppIconManagerBrowserTest() override = default;

  void SetUpOnMainThread() override {
    Profile* profile = browser()->GetProfile();
    app_service_test_.SetUp(profile);
    web_app::test::WaitUntilReady(WebAppProvider::GetForTest(profile));
    WebAppBrowserTestBase::SetUpOnMainThread();
  }

 protected:
  apps::AppServiceTest& app_service_test() { return app_service_test_; }

 private:
  apps::AppServiceTest app_service_test_;
};

IN_PROC_BROWSER_TEST_F(WebAppIconManagerBrowserTest, SingleIcon) {
  const GURL start_url = embedded_https_test_server().GetURL(
      "/banners/no_manifest_test_page.html");

  webapps::AppId app_id;
  {
    std::unique_ptr<WebAppInstallInfo> install_info =
        WebAppInstallInfo::CreateWithStartUrlForTesting(start_url);
    install_info->scope = start_url.GetWithoutFilename();
    install_info->title = u"App Name";
    install_info->user_display_mode = mojom::UserDisplayMode::kStandalone;

    {
      SkBitmap bitmap;
      bitmap.allocN32Pixels(icon_size::k32, icon_size::k32, true);
      bitmap.eraseColor(SK_ColorBLUE);
      install_info->icon_bitmaps.any[icon_size::k32] = std::move(bitmap);
    }

    base::RunLoop run_loop;

    auto* provider = WebAppProvider::GetForTest(browser()->GetProfile());
    provider->scheduler().InstallFromInfoNoIntegrationForTesting(
        std::move(install_info),
        /*overwrite_existing_manifest_fields=*/false,
        webapps::WebappInstallSource::OMNIBOX_INSTALL_ICON,
        base::BindLambdaForTesting(
            [&app_id, &run_loop](const webapps::AppId& installed_app_id,
                                 webapps::InstallResultCode code) {
              EXPECT_EQ(webapps::InstallResultCode::kSuccessNewInstall, code);
              app_id = installed_app_id;
              run_loop.Quit();
            }));

    run_loop.Run();
  }

  base::RunLoop run_loop;
  WebAppBrowserController::SetIconLoadCallbackForTesting(
      run_loop.QuitClosure());
  BrowserWindowInterface* app_browser = LaunchWebAppBrowser(app_id);
  run_loop.Run();

  gfx::ImageSkia app_icon = web_app::AppBrowserController::From(app_browser)
                                ->GetWindowAppIcon()
                                .Rasterize(nullptr);

#if BUILDFLAG(IS_CHROMEOS)
  gfx::ImageSkia image_skia =
      app_service_test().LoadAppIconBlocking(app_id, kWebAppIconSmall);
  EXPECT_TRUE(app_service_test().AreIconImageEqual(image_skia, app_icon));
#else
  const SkBitmap* bitmap = app_icon.bitmap();
  EXPECT_EQ(SK_ColorBLUE, bitmap->getColor(0, 0));
  EXPECT_EQ(32, bitmap->width());
  EXPECT_EQ(32, bitmap->height());
#endif
}

}  // namespace web_app