// 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/ui/views/web_apps/create_shortcut_confirmation_view.h"

#include "base/strings/utf_string_conversions.h"
#include "base/test/bind.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/test_future.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/test/test_browser_dialog.h"
#include "chrome/browser/ui/web_applications/web_app_dialogs.h"
#include "chrome/browser/web_applications/test/web_app_icon_test_utils.h"
#include "chrome/browser/web_applications/web_app_helpers.h"
#include "chrome/browser/web_applications/web_app_icon_generator.h"
#include "chrome/browser/web_applications/web_app_install_info.h"
#include "chrome/common/chrome_features.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "components/webapps/browser/installable/installable_metrics.h"
#include "components/webapps/browser/installable/ml_install_operation_tracker.h"
#include "components/webapps/browser/installable/ml_installability_promoter.h"
#include "content/public/browser/web_contents.h"
#include "content/public/common/content_features.h"
#include "content/public/test/browser_test.h"
#include "third_party/blink/public/common/features.h"

enum CreateShortcutViewParams {
  kTabStripEnabled = 0,
  kTabStripDisabled = 1,
};

std::string ParamsToString(
    const testing::TestParamInfo<CreateShortcutViewParams>& test_params) {
  switch (test_params.param) {
    case kTabStripEnabled:
      return "TabStripEnabled";
    case kTabStripDisabled:
      return "TabStripDisabled";
  }
}

constexpr const char kCreateShortcutIconUrl[] = "https://www.example.com/icon";

class CreateShortcutConfirmationViewBrowserTest
    : public DialogBrowserTest,
      public ::testing::WithParamInterface<CreateShortcutViewParams> {
 public:
  CreateShortcutConfirmationViewBrowserTest() = default;
  CreateShortcutConfirmationViewBrowserTest(
      const CreateShortcutConfirmationViewBrowserTest&) = delete;
  CreateShortcutConfirmationViewBrowserTest& operator=(
      const CreateShortcutConfirmationViewBrowserTest&) = delete;

  // DialogBrowserTest:
  void ShowUi(const std::string& name) override {
    auto app_info = web_app::WebAppInstallInfo::CreateWithStartUrlForTesting(
        GURL("https://example.com"));
    app_info->title = u"Test app";

    const web_app::GeneratedIconsInfo any_icon_info1(
        web_app::IconPurpose::ANY, {web_app::icon_size::k32}, {SK_ColorBLACK});
    const web_app::GeneratedIconsInfo any_icon_info2(
        web_app::IconPurpose::MASKABLE, {web_app::icon_size::k32},
        {SK_ColorBLUE});
    web_app::AddIconsToWebAppInstallInfo(app_info.get(),
                                         GURL(kCreateShortcutIconUrl),
                                         {any_icon_info1, any_icon_info2});

    auto callback = [](bool result,
                       std::unique_ptr<web_app::WebAppInstallInfo>) {};

    content::WebContents* web_contents =
        browser()->GetTabStripModel()->GetActiveWebContents();
    std::unique_ptr<webapps::MlInstallOperationTracker> install_tracker =
        webapps::MLInstallabilityPromoter::FromWebContents(web_contents)
            ->RegisterCurrentInstallForWebContents(
                webapps::WebappInstallSource::MENU_CREATE_SHORTCUT);

    web_app::ShowCreateShortcutDialog(web_contents, std::move(app_info),
                                      std::move(install_tracker),
                                      base::BindLambdaForTesting(callback));
  }

  void SetUp() override {
    base::flat_map<base::test::FeatureRef, bool> features;
    switch (GetParam()) {
      case CreateShortcutViewParams::kTabStripEnabled:
        features.insert({blink::features::kDesktopPWAsTabStrip, true});
        features.insert({features::kDesktopPWAsTabStripSettings, true});
        break;
      case CreateShortcutViewParams::kTabStripDisabled:
        features.insert({blink::features::kDesktopPWAsTabStrip, false});
        features.insert({features::kDesktopPWAsTabStripSettings, false});
        break;
    }

    features.insert({::features::kWebAppInstallDialog, false});
    feature_list.InitWithFeatureStates(features);
    DialogBrowserTest::SetUp();
  }

 private:
  base::test::ScopedFeatureList feature_list;
};

IN_PROC_BROWSER_TEST_P(CreateShortcutConfirmationViewBrowserTest,
                       ShowCreateShortcutDialog) {
  auto app_info = web_app::WebAppInstallInfo::CreateWithStartUrlForTesting(
      GURL("https://example.com"));
  app_info->title = u"Test app";

  base::AutoReset<web_app::InstallDialogTestResponse> auto_accept =
      web_app::SetPwaInstallationAutoRespondForTesting(
          web_app::InstallDialogTestResponse::kAcceptAndLaunch);
  base::AutoReset<web_app::CreateShortcutDialogCheckState> auto_check =
      web_app::SetCreateShortcutDialogCheckStateForTesting(
          web_app::CreateShortcutDialogCheckState::kChecked);
  bool is_accepted = false;
  std::unique_ptr<web_app::WebAppInstallInfo> install_info;
  auto callback = [&is_accepted, &install_info](
                      bool result,
                      std::unique_ptr<web_app::WebAppInstallInfo> info) {
    is_accepted = result;
    install_info = std::move(info);
  };

  content::WebContents* web_contents =
      browser()->GetTabStripModel()->GetActiveWebContents();
  std::unique_ptr<webapps::MlInstallOperationTracker> install_tracker =
      webapps::MLInstallabilityPromoter::FromWebContents(web_contents)
          ->RegisterCurrentInstallForWebContents(
              webapps::WebappInstallSource::MENU_CREATE_SHORTCUT);

  web_app::ShowCreateShortcutDialog(web_contents, std::move(app_info),
                                    std::move(install_tracker),
                                    base::BindLambdaForTesting(callback));
  EXPECT_TRUE(is_accepted);

  EXPECT_EQ(install_info->user_display_mode,
            web_app::mojom::UserDisplayMode::kStandalone);
  EXPECT_TRUE(install_info->is_diy_app);
}

IN_PROC_BROWSER_TEST_P(CreateShortcutConfirmationViewBrowserTest,
                       VerifyCreateShortcutDialogContents) {
  auto app_info = web_app::WebAppInstallInfo::CreateWithStartUrlForTesting(
      GURL("https://example.com"));
  app_info->title = u"Test app";

  base::test::TestFuture<bool, std::unique_ptr<web_app::WebAppInstallInfo>>
      install_result;

  content::WebContents* web_contents =
      browser()->GetTabStripModel()->GetActiveWebContents();
  std::unique_ptr<webapps::MlInstallOperationTracker> install_tracker =
      webapps::MLInstallabilityPromoter::FromWebContents(web_contents)
          ->RegisterCurrentInstallForWebContents(
              webapps::WebappInstallSource::MENU_CREATE_SHORTCUT);

  web_app::ShowCreateShortcutDialog(web_contents, std::move(app_info),
                                    std::move(install_tracker),
                                    install_result.GetCallback());

  CreateShortcutConfirmationView* dialog =
      CreateShortcutConfirmationView::GetDialogForTesting();

  ASSERT_TRUE(dialog);
  EXPECT_TRUE(dialog->GetVisible());

  dialog->Accept();

  EXPECT_TRUE(install_result.Get<bool /*is_accepted*/>());

  EXPECT_EQ(install_result.Get<std::unique_ptr<web_app::WebAppInstallInfo>>()
                ->user_display_mode,
            web_app::mojom::UserDisplayMode::kBrowser);
  EXPECT_TRUE(install_result.Get<std::unique_ptr<web_app::WebAppInstallInfo>>()
                  ->is_diy_app);
}

IN_PROC_BROWSER_TEST_P(CreateShortcutConfirmationViewBrowserTest,
                       DialogCancelNotDiy) {
  auto app_info = web_app::WebAppInstallInfo::CreateWithStartUrlForTesting(
      GURL("https://example.com"));
  app_info->title = u"Test app";

  base::test::TestFuture<bool, std::unique_ptr<web_app::WebAppInstallInfo>>
      install_result;

  content::WebContents* web_contents =
      browser()->GetTabStripModel()->GetActiveWebContents();
  std::unique_ptr<webapps::MlInstallOperationTracker> install_tracker =
      webapps::MLInstallabilityPromoter::FromWebContents(web_contents)
          ->RegisterCurrentInstallForWebContents(
              webapps::WebappInstallSource::MENU_CREATE_SHORTCUT);

  web_app::ShowCreateShortcutDialog(web_contents, std::move(app_info),
                                    std::move(install_tracker),
                                    install_result.GetCallback());

  CreateShortcutConfirmationView* dialog =
      CreateShortcutConfirmationView::GetDialogForTesting();

  ASSERT_TRUE(dialog);
  EXPECT_TRUE(dialog->GetVisible());

  dialog->Cancel();

  EXPECT_FALSE(install_result.Get<bool /*is_accepted*/>());
  EXPECT_FALSE(
      install_result.Get<std::unique_ptr<web_app::WebAppInstallInfo>>()
          ->is_diy_app);
}

IN_PROC_BROWSER_TEST_P(CreateShortcutConfirmationViewBrowserTest,
                       InvokeUi_default) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(CreateShortcutConfirmationViewBrowserTest,
                       NormalizeTitles) {
  base::AutoReset<web_app::InstallDialogTestResponse> auto_accept =
      web_app::SetPwaInstallationAutoRespondForTesting(
          web_app::InstallDialogTestResponse::kAcceptAndLaunch);
  base::AutoReset<web_app::CreateShortcutDialogCheckState> auto_check =
      web_app::SetCreateShortcutDialogCheckStateForTesting(
          web_app::CreateShortcutDialogCheckState::kChecked);

  struct TestCases {
    std::u16string input;
    std::u16string expected_result;
  } test_cases[] = {
      {u"App Title", u"App Title"},
      {u"http://example.com", u"example.com"},
      {u"https://example.com", u"example.com"},
  };

  for (const TestCases& test_case : test_cases) {
    auto app_info = web_app::WebAppInstallInfo::CreateWithStartUrlForTesting(
        GURL("https://example.com"));
    app_info->title = test_case.input;

    bool is_accepted = false;
    std::u16string title;
    auto callback = [&is_accepted, &title](
                        bool result,
                        std::unique_ptr<web_app::WebAppInstallInfo> info) {
      is_accepted = result;
      title = info->title.value();
    };

    content::WebContents* web_contents =
        browser()->GetTabStripModel()->GetActiveWebContents();
    std::unique_ptr<webapps::MlInstallOperationTracker> install_tracker =
        webapps::MLInstallabilityPromoter::FromWebContents(web_contents)
            ->RegisterCurrentInstallForWebContents(
                webapps::WebappInstallSource::MENU_CREATE_SHORTCUT);

    web_app::ShowCreateShortcutDialog(web_contents, std::move(app_info),
                                      std::move(install_tracker),
                                      base::BindLambdaForTesting(callback));
    EXPECT_TRUE(is_accepted) << test_case.input;
    EXPECT_EQ(test_case.expected_result, title) << test_case.input;
  }
}

INSTANTIATE_TEST_SUITE_P(
    All,
    CreateShortcutConfirmationViewBrowserTest,
    ::testing::Values(CreateShortcutViewParams::kTabStripDisabled,
                      CreateShortcutViewParams::kTabStripEnabled),
    ParamsToString);