// Copyright 2024 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/ash/mahi/web_contents/mahi_web_contents_manager_impl.h"

#include <memory>
#include <optional>
#include <string>
#include <utility>

#include "ash/system/mahi/test/mock_mahi_manager.h"
#include "base/callback_list.h"
#include "base/run_loop.h"
#include "base/strings/strcat.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/bind.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/scoped_feature_list.h"
#include "base/unguessable_token.h"
#include "build/build_config.h"
#include "chrome/browser/ash/mahi/web_contents/test_support/fake_mahi_web_contents_manager.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "chromeos/components/mahi/public/cpp/mahi_browser_util.h"
#include "chromeos/components/mahi/public/cpp/mahi_manager.h"
#include "chromeos/components/mahi/public/cpp/mahi_types.h"
#include "chromeos/components/mahi/public/cpp/mahi_util.h"
#include "chromeos/components/mahi/public/cpp/mahi_web_contents_manager.h"
#include "chromeos/constants/chromeos_features.h"
#include "components/content_settings/core/common/features.h"
#include "content/public/browser/navigation_entry.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test.h"
#include "net/dns/mock_host_resolver.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/page_transition_types.h"
#include "ui/gfx/image/image_skia.h"
#include "url/gurl.h"

namespace mahi {

namespace {

using chromeos::mahi::ButtonType;
using chromeos::mahi::kMahiContentExtractionTriggeringLatency;
using chromeos::mahi::kMahiContextMenuActivated;
using chromeos::mahi::kMahiContextMenuActivatedFailed;

// Fake context menu click action.
constexpr int64_t kDisplayID = 1;
constexpr ButtonType kButtonType = ButtonType::kQA;
constexpr char16_t kQuestion[] = u"dump question";

// Fake web content.
constexpr char kUrl[] = "data:text/html,<p>kittens!</p>";

constexpr char kPDFFilename[] = "paragraphs-and-heading-untagged.pdf";

}  // namespace

class MahiWebContentsManagerBrowserTest : public InProcessBrowserTest {
 public:
  MahiWebContentsManagerBrowserTest() {
    scoped_feature_list_.InitWithFeatures(
        {chromeos::features::kFeatureManagementMahi}, {});
  }
  ~MahiWebContentsManagerBrowserTest() override = default;

  // InProcessBrowserTest:
  void SetUpCommandLine(base::CommandLine* command_line) override {
    embedded_test_server()->ServeFilesFromSourceDirectory("content/test/data");
    ASSERT_TRUE(embedded_test_server()->InitializeAndListen());
  }

  void SetUpOnMainThread() override {
    host_resolver()->AddRule("*", "127.0.0.1");
    embedded_test_server()->StartAcceptingConnections();

    InProcessBrowserTest::SetUpOnMainThread();

    fake_mahi_web_contents_manager_ =
        std::make_unique<FakeMahiWebContentsManager>();
    scoped_mahi_web_contents_manager_ =
        std::make_unique<chromeos::ScopedMahiWebContentsManagerOverride>(
            fake_mahi_web_contents_manager_.get());
  }

  // InProcessBrowserTest:
  void TearDownOnMainThread() override {
    scoped_mahi_web_contents_manager_.reset();
    fake_mahi_web_contents_manager_.reset();
    InProcessBrowserTest::TearDownOnMainThread();
  }

  // Simulates opening a new tab with url.
  void CreateWebContent() {
    // Simulates chrome open.
    ASSERT_TRUE(
        ui_test_utils::NavigateToURL(browser(), GURL("chrome://newtab/")));
    // Then navigates to the target page.
    EXPECT_TRUE(AddTabAtIndex(0, GURL(kUrl), ui::PAGE_TRANSITION_TYPED));
  }

  void CreateWebContentWithPDF() {
    GURL pdf_url = embedded_test_server()->GetURL(
        base::StrCat({"/pdf/accessibility/", kPDFFilename}));

    // Simulates chrome open.
    ASSERT_TRUE(
        ui_test_utils::NavigateToURL(browser(), GURL("chrome://newtab/")));
    EXPECT_TRUE(AddTabAtIndex(0, pdf_url, ui::PAGE_TRANSITION_TYPED));
  }

  void ExpectOnContextMenuClicked(ButtonType button_type) {
    base::RunLoop run_loop;
    EXPECT_CALL(mock_mahi_manager_, OnContextMenuClicked)
        .WillOnce([&run_loop](chromeos::MahiContextMenuRequest request) {
          run_loop.Quit();
        });
    fake_mahi_web_contents_manager_->OnContextMenuClicked(
        kDisplayID, button_type,
        /*question=*/kQuestion, /*mahi_menu_bounds=*/gfx::Rect());
    run_loop.Run();
  }

  base::test::ScopedFeatureList scoped_feature_list_;
  testing::StrictMock<ash::MockMahiManager> mock_mahi_manager_;
  chromeos::ScopedMahiManagerSetter scoped_manager_setter_{&mock_mahi_manager_};

  std::unique_ptr<FakeMahiWebContentsManager> fake_mahi_web_contents_manager_;
  std::unique_ptr<chromeos::ScopedMahiWebContentsManagerOverride>
      scoped_mahi_web_contents_manager_;
};

IN_PROC_BROWSER_TEST_F(MahiWebContentsManagerBrowserTest,
                       OnContextMenuClicked) {
  base::RunLoop run_loop;
  // Expects that `MahiManager` should receive the context menu click action.
  EXPECT_CALL(mock_mahi_manager_, OnContextMenuClicked)
      .WillOnce([&run_loop](chromeos::MahiContextMenuRequest request) {
        EXPECT_EQ(kDisplayID, request.display_id);
        EXPECT_EQ(MatchButtonTypeToActionType(kButtonType),
                  request.action_type);
        EXPECT_EQ(kQuestion, request.question);
        run_loop.Quit();
      });

  fake_mahi_web_contents_manager_->OnContextMenuClicked(
      kDisplayID, kButtonType, kQuestion, /*mahi_menu_bounds=*/gfx::Rect());
  run_loop.Run();

  EXPECT_EQ(GURL(),
            fake_mahi_web_contents_manager_->focused_web_content_state().url);
  EXPECT_EQ(u"",
            fake_mahi_web_contents_manager_->focused_web_content_state().title);
}

IN_PROC_BROWSER_TEST_F(MahiWebContentsManagerBrowserTest,
                       PDFContentIsDetectedCorrectly) {
  base::HistogramTester histogram;

  base::RunLoop run_loop;
  // Expects that `MahiManager` should receive the focused page change.
  EXPECT_CALL(mock_mahi_manager_, SetCurrentFocusedPageInfo)
      // When browser opens with `chrome://newtab`, we should be notified to
      // clear the previous focus info.
      .WillOnce([](chromeos::MahiPageInfo page_info) {
        EXPECT_EQ(GURL(), page_info.url);
        EXPECT_FALSE(page_info.is_distillable.has_value());
      })
      // When a PDF is opened, the `MahiManager` should be notified without the
      // distillability check.
      .WillOnce([&run_loop, &histogram](chromeos::MahiPageInfo page_info) {
        EXPECT_TRUE(page_info.is_distillable.has_value());
        EXPECT_EQ(page_info.url.ExtractFileName(), kPDFFilename);
        run_loop.Quit();
        // Since there is no distillability check for PDFs, triggering metric is
        // not logged.
        histogram.ExpectTotalCount(kMahiContentExtractionTriggeringLatency, 0);
      });

  CreateWebContentWithPDF();
  run_loop.Run();
  EXPECT_TRUE(fake_mahi_web_contents_manager_->is_pdf_focused_web_contents());
}

IN_PROC_BROWSER_TEST_F(MahiWebContentsManagerBrowserTest,
                       OpenNewPageToChangePageFocus) {
  base::HistogramTester histogram;

  // Initially, the focused state's favicon is empty.
  EXPECT_TRUE(fake_mahi_web_contents_manager_->focused_web_content_state()
                  .favicon.isNull());

  base::RunLoop run_loop;

  EXPECT_CALL(mock_mahi_manager_,
              SetCurrentFocusedPageInfo(testing::Field(
                  &chromeos::MahiPageInfo::url, testing::Not(GURL(kUrl)))))
      // When browser opens with `chrome://newtab` or `about:blank`, we should
      // be notified to clear the previous focus info.
      .Times(testing::AtLeast(1))
      .WillRepeatedly([](chromeos::MahiPageInfo page_info) {
        EXPECT_FALSE(page_info.is_distillable.has_value());
      });

  // Expects that `MahiManager` should receive the focused page change.
  EXPECT_CALL(mock_mahi_manager_,
              SetCurrentFocusedPageInfo(
                  testing::Field(&chromeos::MahiPageInfo::url, GURL(kUrl))))
      // When a new page gets focus, the `MahiManager` should be notified
      // without the distillability check.
      .WillOnce([&histogram](chromeos::MahiPageInfo page_info) {
        EXPECT_EQ(GURL(kUrl), page_info.url);
        EXPECT_FALSE(page_info.is_distillable.has_value());
        // Before distillability check finishes, triggering metric is not
        // logged.
        histogram.ExpectTotalCount(kMahiContentExtractionTriggeringLatency, 0);
      })
      // When the focused page finishes loading, the `MahiManager` should be
      // notified with the distillability check.
      .WillOnce([&run_loop, &histogram](chromeos::MahiPageInfo page_info) {
        EXPECT_EQ(GURL(kUrl), page_info.url);
        EXPECT_TRUE(page_info.is_distillable.has_value());
        EXPECT_FALSE(page_info.is_distillable.value());
        // The favicon is not empty.
        EXPECT_FALSE(page_info.favicon_image.isNull());

        run_loop.Quit();

        // When distillability check finishes, triggering metric is logged.
        histogram.ExpectTotalCount(kMahiContentExtractionTriggeringLatency, 1);
      });

  CreateWebContent();
  run_loop.Run();
}

IN_PROC_BROWSER_TEST_F(MahiWebContentsManagerBrowserTest, GetPageContents) {
  // Initially, the focused state and the requested state should be different.
  base::UnguessableToken focused_page_id =
      fake_mahi_web_contents_manager_->focused_web_content_state().page_id;

  // First create a web page so there is a place to extract the contents from.
  base::RunLoop run_loop;
  EXPECT_CALL(mock_mahi_manager_,
              SetCurrentFocusedPageInfo(testing::Field(
                  &chromeos::MahiPageInfo::url, testing::Not(GURL(kUrl)))))
      // When browser opens with `chrome://newtab` and/or `about:blank`, we
      // should be notified to clear the previous focus info.
      .Times(testing::AtLeast(1))
      .WillRepeatedly([](chromeos::MahiPageInfo page_info) {
        EXPECT_FALSE(page_info.is_distillable.has_value());
      });

  EXPECT_CALL(mock_mahi_manager_,
              SetCurrentFocusedPageInfo(
                  testing::Field(&chromeos::MahiPageInfo::url, (GURL(kUrl)))))
      .WillOnce([](chromeos::MahiPageInfo page_info) {})
      .WillOnce([&run_loop, &focused_page_id,
                 this](chromeos::MahiPageInfo page_info) {
        EXPECT_TRUE(page_info.is_distillable.has_value());
        EXPECT_FALSE(page_info.is_distillable.value());

        // Gets the page id of the newly opened page.
        focused_page_id = page_info.page_id;
        // When distillability check is returned, simulates the content request
        // from the mahi manager.
        fake_mahi_web_contents_manager_->RequestContent(
            focused_page_id,
            base::BindLambdaForTesting(
                [&](std::optional<chromeos::MahiPageContent> page_content) {
                  run_loop.Quit();
                }));
      });
  CreateWebContent();
  run_loop.Run();

  EXPECT_EQ(
      focused_page_id,
      fake_mahi_web_contents_manager_->focused_web_content_state().page_id);
  EXPECT_EQ(GURL(kUrl),
            fake_mahi_web_contents_manager_->focused_web_content_state().url);
  EXPECT_EQ(u"data:text/html,<p>kittens!</p>",
            fake_mahi_web_contents_manager_->focused_web_content_state().title);
}

IN_PROC_BROWSER_TEST_F(MahiWebContentsManagerBrowserTest, GetPDFContents) {
  // Initially, the focused state and the requested state should be different.
  base::UnguessableToken focused_page_id =
      fake_mahi_web_contents_manager_->focused_web_content_state().page_id;

  // First create a web page so there is a place to extract the contents from.
  base::RunLoop run_loop;
  EXPECT_CALL(mock_mahi_manager_, SetCurrentFocusedPageInfo)
      .WillOnce([](chromeos::MahiPageInfo page_info) {})
      .WillOnce([&run_loop, &focused_page_id,
                 this](chromeos::MahiPageInfo page_info) {
        EXPECT_TRUE(page_info.is_distillable.has_value());
        EXPECT_TRUE(page_info.is_distillable.value());
        EXPECT_EQ(page_info.url.ExtractFileName(), kPDFFilename);

        focused_page_id = page_info.page_id;
        // Simulate a request to extract content from client.
        fake_mahi_web_contents_manager_->RequestContent(
            focused_page_id,
            base::BindLambdaForTesting(
                [&](std::optional<chromeos::MahiPageContent> page_content) {
                  run_loop.Quit();
                }));
      });
  CreateWebContentWithPDF();

  EXPECT_EQ(
      focused_page_id,
      fake_mahi_web_contents_manager_->focused_web_content_state().page_id);
  EXPECT_EQ(fake_mahi_web_contents_manager_->focused_web_content_state()
                .url.ExtractFileName(),
            kPDFFilename);
}

IN_PROC_BROWSER_TEST_F(MahiWebContentsManagerBrowserTest, ContextMenuMetrics) {
  base::HistogramTester histogram;

  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kSettings,
                              0);
  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kOutline,
                              0);
  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kSummary,
                              0);
  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kQA, 0);
  histogram.ExpectBucketCount(kMahiContextMenuActivatedFailed,
                              ButtonType::kSettings, 0);
  histogram.ExpectBucketCount(kMahiContextMenuActivatedFailed,
                              ButtonType::kOutline, 0);
  histogram.ExpectBucketCount(kMahiContextMenuActivatedFailed,
                              ButtonType::kSummary, 0);

  // QA section.
  // With a successful click.
  ExpectOnContextMenuClicked(ButtonType::kQA);
  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kQA, 1);
  histogram.ExpectTotalCount(kMahiContextMenuActivated, 1);
  testing::Mock::VerifyAndClearExpectations(this);

  // Outline.
  // With a successful click.
  ExpectOnContextMenuClicked(ButtonType::kOutline);
  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kOutline,
                              1);
  histogram.ExpectTotalCount(kMahiContextMenuActivated, 2);
  testing::Mock::VerifyAndClearExpectations(this);

  // Summary button.
  // With a successful click.
  ExpectOnContextMenuClicked(ButtonType::kSummary);
  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kSummary,
                              1);
  histogram.ExpectTotalCount(kMahiContextMenuActivated, 3);
  testing::Mock::VerifyAndClearExpectations(this);

  // Settings button.
  // With a successful click.
  ExpectOnContextMenuClicked(ButtonType::kSettings);
  histogram.ExpectBucketCount(kMahiContextMenuActivated, ButtonType::kSettings,
                              1);
  histogram.ExpectTotalCount(kMahiContextMenuActivated, 4);
  testing::Mock::VerifyAndClearExpectations(this);
}

}  // namespace mahi