// 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 "chrome/browser/accessibility/pdf_ocr_controller.h"

#include <vector>

#include "base/run_loop.h"
#include "base/test/bind.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/with_feature_override.h"
#include "chrome/browser/accessibility/pdf_ocr_controller_factory.h"
#include "chrome/browser/pdf/pdf_extension_test_base.h"
#include "chrome/browser/screen_ai/screen_ai_install_state.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_commands.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/keyed_service/core/keyed_service.h"
#include "components/prefs/pref_service.h"
#include "content/public/browser/browser_accessibility_state.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test.h"
#include "pdf/pdf_features.h"
#include "ui/accessibility/accessibility_features.h"
#include "ui/accessibility/accessibility_switches.h"

#if BUILDFLAG(IS_CHROMEOS)
#include "chrome/browser/ash/accessibility/accessibility_manager.h"
#include "chrome/browser/ash/accessibility/speech_monitor.h"
#else
#include <optional>

#include "chrome/browser/browser_process.h"
#include "content/public/test/scoped_accessibility_mode_override.h"
#endif  // !BUILDFLAG(IS_CHROMEOS)

namespace {

class DownloadObserver : public screen_ai::ScreenAIInstallState::Observer {
 public:
  DownloadObserver() {
    install_state_observer_.Observe(
        screen_ai::ScreenAIInstallState::GetInstance());
  }

  DownloadObserver(const DownloadObserver&) = delete;
  DownloadObserver& operator=(const DownloadObserver&) = delete;

  ~DownloadObserver() override = default;

  // screen_ai::ScreenAIInstallState::Observer:
  void StateChanged(screen_ai::ScreenAIInstallState::State state) override {
    if (state != screen_ai::ScreenAIInstallState::State::kDownloading) {
      return;
    }

    screen_ai::ScreenAIInstallState::GetInstance()->SetState(
        screen_ai::ScreenAIInstallState::State::kDownloadFailed);

    remaining_download_tries_--;
    if (!remaining_download_tries_) {
      run_loop.Quit();
    }
  }

  void WaitForDownloads() {
    if (remaining_download_tries_) {
      run_loop.Run();
    }
  }

  int remaining_download_tries_ = -1;

 private:
  base::RunLoop run_loop;
  base::ScopedObservation<screen_ai::ScreenAIInstallState,
                          screen_ai::ScreenAIInstallState::Observer>
      install_state_observer_{this};
};

}  // namespace

// TODO(crbug.com/360803943): Remove this test when PDF Searchify is launched.
class PdfOcrControllerBrowserTest : public base::test::WithFeatureOverride,
                                    public PDFExtensionTestBase {
 public:
  PdfOcrControllerBrowserTest()
      : base::test::WithFeatureOverride(chrome_pdf::features::kPdfOopif) {}
  ~PdfOcrControllerBrowserTest() override = default;

  PdfOcrControllerBrowserTest(const PdfOcrControllerBrowserTest&) = delete;
  PdfOcrControllerBrowserTest& operator=(const PdfOcrControllerBrowserTest&) =
      delete;

  // PDFExtensionTestBase overrides:
  void TearDownOnMainThread() override {
    PDFExtensionTestBase::TearDownOnMainThread();
    EnableScreenReader(false);
  }

  void EnableScreenReader(bool enabled) {
#if BUILDFLAG(IS_CHROMEOS)
    // Enable Chromevox.
    ash::AccessibilityManager::Get()->EnableSpokenFeedback(enabled);
    if (enabled) {
      // Block until Chromevox is fully loaded.
      speech_monitor_.ExpectSpeechPattern("*");
      speech_monitor_.Call([this]() { DisableEarcons(); });
      speech_monitor_.Replay();
    }
#else
    if (!enabled) {
      scoped_accessibility_override_.reset();
    } else if (!scoped_accessibility_override_) {
      scoped_accessibility_override_.emplace(ui::AXMode::kWebContents |
                                             ui::AXMode::kExtendedProperties |
                                             ui::AXMode::kScreenReader);
    }
#endif  // BUILDFLAG(IS_CHROMEOS)
  }

#if BUILDFLAG(IS_CHROMEOS)
  void EnableSelectToSpeak(bool enabled) {
    ash::AccessibilityManager::Get()->SetSelectToSpeakEnabled(enabled);
  }
#endif  // BUILDFLAG(IS_CHROMEOS)

  bool UseOopif() const override { return GetParam(); }

  std::vector<base::test::FeatureRefAndParams> GetEnabledFeatures()
      const override {
    auto enabled = PDFExtensionTestBase::GetEnabledFeatures();
#if BUILDFLAG(IS_CHROMEOS)
    enabled.push_back({features::kAccessibilityPdfOcrForSelectToSpeak, {}});
#endif  // BUILDFLAG(IS_CHROMEOS)
    return enabled;
  }

 private:
#if BUILDFLAG(IS_CHROMEOS)
  void DisableEarcons() {
    // Playing earcons from within a test is not only annoying if you're
    // running the test locally, but seems to cause crashes
    // (http://crbug.com/396507). Work around this by just telling
    // ChromeVox to not ever play earcons (prerecorded sound effects).
    extensions::browsertest_util::ExecuteScriptInBackgroundPageNoWait(
        browser()->profile(), extension_misc::kChromeVoxExtensionId,
        "ChromeVox.earcons.playEarcon = function() {};");
  }

  ash::test::SpeechMonitor speech_monitor_;
#else
  std::optional<content::ScopedAccessibilityModeOverride>
      scoped_accessibility_override_;
#endif
};

IN_PROC_BROWSER_TEST_P(PdfOcrControllerBrowserTest, TestGetAllPdfWebContents) {
  // Load a HTML webpage.
  constexpr char kTestHtml[] =
      "<html><head><title>TEST</title></head><body></body></html>";
  ASSERT_TRUE(ui_test_utils::NavigateToURL(
      browser(), GURL(std::string("data:text/html,") + kTestHtml)));

  std::vector<content::WebContents*> pdf_web_contents_vector =
      screen_ai::PdfOcrController::GetAllPdfWebContentsForTesting(
          browser()->profile());
  ASSERT_EQ(0u, pdf_web_contents_vector.size());

  // Load test PDF.
  ASSERT_TRUE(LoadPdf(embedded_test_server()->GetURL("/pdf/test.pdf")));

  pdf_web_contents_vector =
      screen_ai::PdfOcrController::GetAllPdfWebContentsForTesting(
          browser()->profile());
  ASSERT_EQ(1u, pdf_web_contents_vector.size());
}

// Enabling screen reader should affect the accessibility mode of a new
// WebContents of PDF Viewer Mimehandler.
IN_PROC_BROWSER_TEST_P(PdfOcrControllerBrowserTest,
                       OpenPDFAfterTurningOnScreenReader) {
  // Forced accessibility contradicts with turning off the screen reader.
  // When PDF Searchify is enabled, PDF OCR controller will not trigger OCR in
  // renderer.
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kForceRendererAccessibility) ||
      base::FeatureList::IsEnabled(chrome_pdf::features::kPdfSearchify)) {
    GTEST_SKIP();
  }

  ui::AXMode ax_mode =
      content::BrowserAccessibilityState::GetInstance()->GetAccessibilityMode();
  EXPECT_FALSE(ax_mode.has_mode(ui::AXMode::kPDFOcr));

  EnableScreenReader(true);
  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->set_ocr_ready_for_testing();
  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->Activate();

  // Load test PDF.
  ASSERT_TRUE(LoadPdf(embedded_test_server()->GetURL("/pdf/test.pdf")));

  content::WebContents* pdf_contents = GetActiveWebContents();
  ax_mode = pdf_contents->GetAccessibilityMode();
  EXPECT_TRUE(ax_mode.has_mode(ui::AXMode::kPDFOcr));
}

// Enabling screen reader should affect the accessibility mode of an exiting
// WebContents of PDF Viewer Mimehandler.
IN_PROC_BROWSER_TEST_P(PdfOcrControllerBrowserTest,
                       OpenPDFBeforeTurningOnScreenReader) {
  // Forced accessibility contradicts with turning off the screen reader.
  // When PDF Searchify is enabled, PDF OCR controller will not trigger OCR in
  // renderer.
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kForceRendererAccessibility) ||
      base::FeatureList::IsEnabled(chrome_pdf::features::kPdfSearchify)) {
    GTEST_SKIP();
  }

  ui::AXMode ax_mode =
      content::BrowserAccessibilityState::GetInstance()->GetAccessibilityMode();
  EXPECT_FALSE(ax_mode.has_mode(ui::AXMode::kPDFOcr));

  // Load test PDF.
  ASSERT_TRUE(LoadPdf((embedded_test_server()->GetURL("/pdf/test.pdf"))));
  content::WebContents* pdf_contents = GetActiveWebContents();
  ax_mode = pdf_contents->GetAccessibilityMode();
  EXPECT_FALSE(ax_mode.has_mode(ui::AXMode::kPDFOcr));

  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->set_ocr_ready_for_testing();
  EnableScreenReader(true);

  pdf_contents = GetActiveWebContents();
  ax_mode = pdf_contents->GetAccessibilityMode();
  EXPECT_TRUE(ax_mode.has_mode(ui::AXMode::kPDFOcr));
}

IN_PROC_BROWSER_TEST_P(PdfOcrControllerBrowserTest, WithoutScreenReader) {
  // Forced accessibility contradicts with turning off the screen reader.
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kForceRendererAccessibility)) {
    GTEST_SKIP();
  }

  EnableScreenReader(false);

  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->set_ocr_ready_for_testing();
  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->Activate();

  ASSERT_TRUE(LoadPdf(embedded_test_server()->GetURL("/pdf/test.pdf")));
  content::WebContents* pdf_contents = GetActiveWebContents();
  ui::AXMode ax_mode = pdf_contents->GetAccessibilityMode();
  EXPECT_FALSE(ax_mode.has_mode(ui::AXMode::kPDFOcr));
}

// Retry download if it fails.
IN_PROC_BROWSER_TEST_P(PdfOcrControllerBrowserTest, DownloadRetry) {
  // Forced accessibility affects counting.
  // When PDF Searchify is enabled, PDF OCR controller will not trigger ScreenAI
  // service initialization.
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kForceRendererAccessibility) ||
      base::FeatureList::IsEnabled(chrome_pdf::features::kPdfSearchify)) {
    GTEST_SKIP();
  }

  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->set_initialization_retry_wait_for_testing(base::Milliseconds(1));

  DownloadObserver observer;
  observer.remaining_download_tries_ = 3;

  EnableScreenReader(true);

  observer.WaitForDownloads();
}

#if BUILDFLAG(IS_CHROMEOS)
IN_PROC_BROWSER_TEST_P(PdfOcrControllerBrowserTest, WithoutSelectToSpeak) {
  EnableSelectToSpeak(false);

  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->set_ocr_ready_for_testing();
  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->Activate();

  ASSERT_TRUE(LoadPdf(embedded_test_server()->GetURL("/pdf/test.pdf")));
  content::WebContents* pdf_contents = GetActiveWebContents();
  ui::AXMode ax_mode = pdf_contents->GetAccessibilityMode();
  EXPECT_FALSE(ax_mode.has_mode(ui::AXMode::kPDFOcr));
}

IN_PROC_BROWSER_TEST_P(PdfOcrControllerBrowserTest, WithSelectToSpeak) {
  // When PDF Searchify is enabled, PDF OCR controller will not trigger OCR in
  // renderer.
  if (base::FeatureList::IsEnabled(chrome_pdf::features::kPdfSearchify)) {
    GTEST_SKIP();
  }

  EnableSelectToSpeak(true);

  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->set_ocr_ready_for_testing();
  screen_ai::PdfOcrControllerFactory::GetForProfile(browser()->profile())
      ->Activate();

  ASSERT_TRUE(LoadPdf(embedded_test_server()->GetURL("/pdf/test.pdf")));
  content::WebContents* pdf_contents = GetActiveWebContents();
  ui::AXMode ax_mode = pdf_contents->GetAccessibilityMode();
  EXPECT_TRUE(ax_mode.has_mode(ui::AXMode::kPDFOcr));
}
#endif  // BUILDFLAG(IS_CHROMEOS)

// TODO(crbug.com/40268279): Stop testing both modes after OOPIF PDF viewer
// launches.
INSTANTIATE_FEATURE_OVERRIDE_TEST_SUITE(PdfOcrControllerBrowserTest);