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

#include <optional>

#include "base/test/metrics/user_action_tester.h"
#include "base/test/run_until.h"
#include "chrome/app/chrome_command_ids.h"
#include "chrome/browser/compose/chrome_compose_client.h"
#include "chrome/browser/compose/compose_enabling.h"
#include "chrome/browser/compose/compose_session.h"
#include "chrome/browser/media/webrtc/desktop_capture_access_handler.h"
#include "chrome/browser/media/webrtc/media_capture_devices_dispatcher.h"
#include "chrome/browser/renderer_context_menu/render_view_context_menu_test_util.h"
#include "chrome/browser/ui/tabs/tab_enums.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/browser/ui/views/compose/compose_dialog_view.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/browser/ui/webui/feedback/feedback_dialog.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "chrome/test/interaction/interactive_browser_test.h"
#include "components/compose/core/browser/compose_features.h"
#include "components/feature_engagement/public/feature_constants.h"
#include "components/feature_engagement/test/scoped_iph_feature_list.h"
#include "components/optimization_guide/core/optimization_guide_features.h"
#include "components/prefs/pref_service.h"
#include "components/unified_consent/pref_names.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "third_party/blink/public/common/context_menu_data/edit_flags.h"
#include "ui/base/clipboard/clipboard.h"
#include "ui/base/clipboard/scoped_clipboard_writer.h"
#include "ui/base/ime/input_method.h"
#include "ui/base/models/menu_model.h"
#include "ui/gfx/geometry/point_conversions.h"
#include "ui/views/interaction/element_tracker_views.h"
#include "ui/views/interaction/interaction_test_util_views.h"

namespace compose {

class ComposeSessionBrowserTest : public InteractiveBrowserTest {
 public:
  void SetUp() override {
    scoped_compose_enabled_ = ComposeEnabling::ScopedEnableComposeForTesting();
    feature_list()->InitWithExistingFeatures(
        {compose::features::kEnableCompose,
         optimization_guide::features::kOptimizationGuideModelExecution,
         feature_engagement::kIPHComposeMSBBSettingsFeature});
    InteractiveBrowserTest::SetUp();
  }

  void TearDown() override { InteractiveBrowserTest::TearDown(); }

  void SetUpOnMainThread() override {
    InteractiveBrowserTest::SetUpOnMainThread();
    MediaCaptureDevicesDispatcher::GetInstance()
        ->desktop_capture_access_handler_for_test()
        ->SetRequestApprovedForTest(true);
  }

  void TearDownOnMainThread() override {
    MediaCaptureDevicesDispatcher::GetInstance()
        ->desktop_capture_access_handler_for_test()
        ->SetRequestApprovedForTest(false);
    InteractiveBrowserTest::TearDownOnMainThread();
  }

  feature_engagement::test::ScopedIphFeatureList* feature_list() {
    return &feature_list_;
  }

 protected:
  feature_engagement::test::ScopedIphFeatureList feature_list_;
  ComposeEnabling::ScopedOverride scoped_compose_enabled_;
};

IN_PROC_BROWSER_TEST_F(ComposeSessionBrowserTest, LifetimeOfBubbleWrapper) {
  ASSERT_TRUE(embedded_test_server()->Start());
  auto* web_contents = browser()->tab_strip_model()->GetActiveWebContents();
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("/compose/test2.html")));
  ASSERT_NE(nullptr, ChromeComposeClient::FromWebContents(web_contents));

  // get point of element
  gfx::PointF textarea_center =
      content::GetCenterCoordinatesOfElementWithId(web_contents, "elem1");
  autofill::FormFieldData field_data;
  field_data.set_bounds(gfx::RectF((textarea_center), gfx::SizeF(1, 1)));

  auto* client = ChromeComposeClient::FromWebContents(web_contents);
  client->ShowComposeDialog(
      autofill::AutofillComposeDelegate::UiEntryPoint::kAutofillPopup,
      field_data, base::NullCallback());

  // close window right away
  browser()->tab_strip_model()->CloseWebContentsAt(0,
                                                   TabCloseTypes::CLOSE_NONE);
}

// TODO(crbug.com/507917198): Re-enable this test
#if BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_MAC)
#define MAYBE_ContextMenuPasteEnabled DISABLED_ContextMenuPasteEnabled
#else
#define MAYBE_ContextMenuPasteEnabled ContextMenuPasteEnabled
#endif  // BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_MAC)
IN_PROC_BROWSER_TEST_F(ComposeSessionBrowserTest,
                       MAYBE_ContextMenuPasteEnabled) {
  content::BrowserTestClipboardScope test_clipboard_scope;
  ASSERT_TRUE(embedded_test_server()->Start());
  auto* web_contents = browser()->tab_strip_model()->GetActiveWebContents();
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("/compose/test2.html")));
  ASSERT_NE(nullptr, ChromeComposeClient::FromWebContents(web_contents));

  test_clipboard_scope.SetText("Test clipboard text");

  ASSERT_TRUE(base::test::RunUntil([&]() {
    std::string clipboard_text;
    test_clipboard_scope.GetText(&clipboard_text);
    return clipboard_text == "Test clipboard text";
  }));

  // get point of element
  gfx::PointF textarea_center =
      content::GetCenterCoordinatesOfElementWithId(web_contents, "elem1");
  autofill::FormFieldData field_data;
  field_data.set_bounds(gfx::RectF((textarea_center), gfx::SizeF(1, 1)));

  auto* client = ChromeComposeClient::FromWebContents(web_contents);
  client->ShowComposeDialog(
      autofill::AutofillComposeDelegate::UiEntryPoint::kAutofillPopup,
      field_data, base::NullCallback());

  ASSERT_TRUE(base::test::RunUntil([&]() {
    return views::ElementTrackerViews::GetInstance()->GetFirstMatchingView(
               ComposeDialogView::kComposeDialogId,
               BrowserView::GetBrowserViewForBrowser(browser())
                   ->GetElementContext()) != nullptr;
  }));

  ComposeDialogView* dialog_view = static_cast<ComposeDialogView*>(
      views::ElementTrackerViews::GetInstance()->GetFirstMatchingView(
          ComposeDialogView::kComposeDialogId,
          BrowserView::GetBrowserViewForBrowser(browser())
              ->GetElementContext()));
  ASSERT_TRUE(dialog_view);

  content::ContextMenuParams params;
  params.is_editable = true;
  params.edit_flags = blink::ContextMenuDataEditFlags::kCanPaste;
  // Simulate the context menu request on the dialog's WebContents
  content::RenderFrameHost* render_frame_host =
      dialog_view->bubble_wrapper()->web_contents()->GetPrimaryMainFrame();
  dialog_view->HandleContextMenu(*render_frame_host, params);

  // Wait for the async clipboard read to complete
  ASSERT_TRUE(base::test::RunUntil([&]() {
    return dialog_view->GetContextMenuModelForTesting() != nullptr;
  }));

  const ui::MenuModel* menu_model =
      dialog_view->GetContextMenuModelForTesting();
  ASSERT_TRUE(menu_model);

  bool paste_found = false;
  for (size_t i = 0; i < menu_model->GetItemCount(); ++i) {
    if (menu_model->GetCommandIdAt(i) == IDC_CONTENT_CONTEXT_PASTE ||
        menu_model->GetCommandIdAt(i) ==
            IDC_CONTENT_CONTEXT_PASTE_AND_MATCH_STYLE) {
      paste_found = true;
    }
  }
  EXPECT_TRUE(paste_found);

  // Dismiss the menu to ensure LIFO destruction order on Wayland.
  ui::KeyEvent escape_key(ui::EventType::kKeyPressed, ui::VKEY_ESCAPE,
                          ui::EF_NONE);
  std::ignore =
      dialog_view->GetWidget()->GetInputMethod()->DispatchKeyEvent(&escape_key);
}

IN_PROC_BROWSER_TEST_F(ComposeSessionBrowserTest, OpenFeedbackPage) {
  // Feedback page can only be opened from a dialog state where MSSB is enabled.
  // TODO(b/316601302): Without directly setting the MSBB pref value this test
  // is flaky on Linux MSan builders. This requires further investigation, but
  // the MSBB dialog state is not on the feedback page testing path so the
  // current state still satisfies the test requirement.
  PrefService* prefs = browser()->GetProfile()->GetPrefs();
  prefs->SetBoolean(
      unified_consent::prefs::kUrlKeyedAnonymizedDataCollectionEnabled, true);

  ASSERT_TRUE(embedded_test_server()->Start());
  auto* web_contents = browser()->tab_strip_model()->GetActiveWebContents();
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("/compose/test2.html")));
  ASSERT_NE(nullptr, ChromeComposeClient::FromWebContents(web_contents));

  // get point of element
  gfx::PointF textarea_center =
      content::GetCenterCoordinatesOfElementWithId(web_contents, "elem1");
  autofill::FormFieldData field_data;
  field_data.set_bounds(gfx::RectF((textarea_center), gfx::SizeF(1, 1)));

  auto* client = ChromeComposeClient::FromWebContents(web_contents);
  client->ShowComposeDialog(
      autofill::AutofillComposeDelegate::UiEntryPoint::kAutofillPopup,
      field_data, base::NullCallback());

  client->OpenFeedbackPageForTest("test_id");

  RunTestSequence(
      InAnyContext(WaitForShow(FeedbackDialog::kFeedbackDialogForTesting)));
}

IN_PROC_BROWSER_TEST_F(ComposeSessionBrowserTest,
                       TestDialogClosedAfterPageScrolled) {
  ASSERT_TRUE(embedded_test_server()->Start());
  auto* web_contents = browser()->tab_strip_model()->GetActiveWebContents();
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("/compose/test2.html")));
  ASSERT_NE(nullptr, ChromeComposeClient::FromWebContents(web_contents));
  auto* client = ChromeComposeClient::FromWebContents(web_contents);

  // get point of element
  gfx::PointF textarea_center =
      content::GetCenterCoordinatesOfElementWithId(web_contents, "elem1");
  autofill::FormFieldData field_data;
  field_data.set_bounds(gfx::RectF((textarea_center), gfx::SizeF(1, 1)));

  client->ShowComposeDialog(
      autofill::AutofillComposeDelegate::UiEntryPoint::kAutofillPopup,
      field_data, base::NullCallback());

  EXPECT_TRUE(client->IsDialogShowing());

  // Scroll on page
  blink::WebGestureEvent event;
  event.SetType(blink::WebInputEvent::Type::kGestureScrollBegin);
  client->DidGetUserInteraction(event);

  EXPECT_FALSE(client->IsDialogShowing());
}

IN_PROC_BROWSER_TEST_F(ComposeSessionBrowserTest, SettingsLaunchedTest) {
  base::UserActionTester user_action_tester;
  ASSERT_TRUE(embedded_test_server()->Start());
  auto* web_contents = browser()->tab_strip_model()->GetActiveWebContents();
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("/compose/test2.html")));
  ASSERT_NE(nullptr, ChromeComposeClient::FromWebContents(web_contents));

  // get point of element
  gfx::PointF textarea_center =
      content::GetCenterCoordinatesOfElementWithId(web_contents, "elem1");
  autofill::FormFieldData field_data;
  field_data.set_bounds(gfx::RectF((textarea_center), gfx::SizeF(1, 1)));

  auto* client = ChromeComposeClient::FromWebContents(web_contents);
  client->ShowComposeDialog(
      autofill::AutofillComposeDelegate::UiEntryPoint::kAutofillPopup,
      field_data, base::NullCallback());

  client->OpenComposeSettings();

  EXPECT_EQ(1, user_action_tester.GetActionCount(
                   "Compose.SessionPaused.MSBBSettingsShown"));

  int tab_index = browser()->tab_strip_model()->count() - 1;

  content::WebContentsDestroyedWatcher destroyed_watcher(
      browser()->tab_strip_model()->GetWebContentsAt(tab_index));
  browser()->tab_strip_model()->CloseWebContentsAt(
      tab_index, TabCloseTypes::CLOSE_CREATE_HISTORICAL_TAB);
  destroyed_watcher.Wait();

  EXPECT_EQ(embedded_test_server()->GetURL("/compose/test2.html"),
            web_contents->GetVisibleURL());
  EXPECT_TRUE(client->IsDialogShowing());
}

IN_PROC_BROWSER_TEST_F(ComposeSessionBrowserTest,
                       PractiveNudgeSettingsLaunchedTest) {
  base::HistogramTester histogram_tester;
  ASSERT_TRUE(embedded_test_server()->Start());
  auto* web_contents = browser()->tab_strip_model()->GetActiveWebContents();
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("/compose/test2.html")));

  auto* client = ChromeComposeClient::FromWebContents(web_contents);
  ASSERT_NE(nullptr, client);

  autofill::FormFieldData field_data;

  client->OpenProactiveNudgeSettings();

  ASSERT_EQ(browser()->tab_strip_model()->count(), 2);

  histogram_tester.ExpectUniqueSample(
      compose::kComposeProactiveNudgeCtr,
      compose::ComposeNudgeCtrEvent::kOpenSettings, 1);
}

IN_PROC_BROWSER_TEST_F(ComposeSessionBrowserTest,
                       SelectionNudgeSettingsLaunchedTest) {
  base::HistogramTester histogram_tester;
  ASSERT_TRUE(embedded_test_server()->Start());
  auto* web_contents = browser()->tab_strip_model()->GetActiveWebContents();
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), embedded_test_server()->GetURL("/compose/test2.html")));

  auto* client = ChromeComposeClient::FromWebContents(web_contents);
  ASSERT_NE(nullptr, client);

  autofill::FormFieldData field_data;
  autofill::FormData form_data;

  // Set the most recent nudge to the selection nudge.
  client->ShowProactiveNudge(form_data.global_id(), field_data.global_id(),
                             compose::ComposeEntryPoint::kSelectionNudge);
  client->OpenProactiveNudgeSettings();

  ASSERT_EQ(browser()->tab_strip_model()->count(), 2);

  histogram_tester.ExpectUniqueSample(
      compose::kComposeSelectionNudgeCtr,
      compose::ComposeNudgeCtrEvent::kOpenSettings, 1);
}

}  // namespace compose