// 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 "base/run_loop.h"
#include "base/test/run_until.h"
#include "chrome/browser/app_mode/test/accelerator_helpers.h"
#include "chrome/browser/ash/app_mode/kiosk_controller.h"
#include "chrome/browser/ash/app_mode/kiosk_system_session.h"
#include "chrome/browser/ash/app_mode/test/kiosk_mixin.h"
#include "chrome/browser/ash/app_mode/test/kiosk_test_utils.h"
#include "chrome/browser/ash/login/app_mode/test/ash_accelerator_helpers.h"
#include "chrome/browser/ui/browser_window/public/global_browser_collection.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/mixin_based_in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/accelerators/accelerator.h"

namespace ash {

using kiosk::test::WaitKioskLaunched;

namespace {

int WindowWidth(content::WebContents* web_contents) {
  return content::EvalJs(web_contents, "window.innerWidth").ExtractInt();
}

void WaitForWindowWidthChanged(content::WebContents* web_contents,
                               int initial_width) {
  ASSERT_TRUE(base::test::RunUntil(
      [&]() { return initial_width != WindowWidth(web_contents); }));
}

int WindowWidthAfterChange(content::WebContents* web_contents,
                           int initial_width) {
  WaitForWindowWidthChanged(web_contents, initial_width);
  return WindowWidth(web_contents);
}

}  // anonymous namespace

// Verifies accelerator behavior in Kiosk sessions in Ash.
class WebKioskAcceleratorTest : public MixinBasedInProcessBrowserTest {
 public:
  WebKioskAcceleratorTest() = default;

  void SetUpOnMainThread() override {
    MixinBasedInProcessBrowserTest::SetUpOnMainThread();
    ui_test_utils::BrowserCreatedObserver browser_created_observer;
    ASSERT_TRUE(WaitKioskLaunched());
    SetBrowser(browser_created_observer.Wait());
  }

  KioskMixin kiosk_{
      &mixin_host_,
      KioskMixin::Config{/*name=*/{},
                         KioskMixin::AutoLaunchAccount{
                             KioskMixin::SimpleWebAppOption().account_id},
                         {KioskMixin::SimpleWebAppOption()}}};
};

IN_PROC_BROWSER_TEST_F(WebKioskAcceleratorTest, AcceleratorsDontCloseSession) {
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 1u);
  ASSERT_FALSE(PressCloseTabAccelerator(browser()));
  ASSERT_FALSE(PressCloseWindowAccelerator(browser()));
  ASSERT_FALSE(ash::PressSignOutAccelerator());
  base::RunLoop loop;
  loop.RunUntilIdle();
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 1u);
  ASSERT_FALSE(
      KioskController::Get().GetKioskSystemSession()->is_shutting_down());
}

IN_PROC_BROWSER_TEST_F(WebKioskAcceleratorTest, ZoomAccelerators) {
  BrowserView* browser_view = BrowserView::GetBrowserViewForBrowser(browser());
  ASSERT_NE(browser_view, nullptr);
  content::WebContents* web_contents = browser_view->GetActiveWebContents();
  ASSERT_NE(web_contents, nullptr);

  // Await page load so accelerators are processed. Prevents a race condition.
  ASSERT_TRUE(WaitForLoadStop(web_contents));

  int initial_width = WindowWidthAfterChange(web_contents, 0);
  ASSERT_GT(initial_width, 0);

  // Zoom in, content becomes larger and window width becomes smaller.
  ASSERT_TRUE(browser_view->AcceleratorPressed(
      ui::Accelerator(ui::VKEY_ADD, ui::EF_CONTROL_DOWN)));
  int zoomed_in_width = WindowWidthAfterChange(web_contents, initial_width);
  ASSERT_LT(zoomed_in_width, initial_width);

  // Restore zoom, window width becomes `initial_width`.
  ASSERT_TRUE(browser_view->AcceleratorPressed(
      ui::Accelerator(ui::VKEY_0, ui::EF_CONTROL_DOWN)));
  int restored_width = WindowWidthAfterChange(web_contents, zoomed_in_width);
  ASSERT_EQ(restored_width, initial_width);

  // Zoom out, content becomes smaller and window width becomes larger.
  ASSERT_TRUE(browser_view->AcceleratorPressed(
      ui::Accelerator(ui::VKEY_SUBTRACT, ui::EF_CONTROL_DOWN)));
  int zoomed_out_width = WindowWidthAfterChange(web_contents, zoomed_in_width);
  ASSERT_GT(zoomed_out_width, initial_width);
}

// Verifies accelerators work in regular non-kiosk sessions in Ash.
class NonKioskAcceleratorTest : public InProcessBrowserTest {};

IN_PROC_BROWSER_TEST_F(NonKioskAcceleratorTest, CloseTabAccelerator) {
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 1u);
  ASSERT_TRUE(PressCloseTabAccelerator(browser()));
  ui_test_utils::BrowserDestroyedObserver settings_browser_closed_observer{
      browser()};
  settings_browser_closed_observer.Wait();
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 0u);
}

IN_PROC_BROWSER_TEST_F(NonKioskAcceleratorTest, CloseWindowAccelerator) {
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 1u);
  ASSERT_TRUE(PressCloseWindowAccelerator(browser()));
  ui_test_utils::BrowserDestroyedObserver settings_browser_closed_observer{
      browser()};
  settings_browser_closed_observer.Wait();
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 0u);
}

IN_PROC_BROWSER_TEST_F(NonKioskAcceleratorTest, SignOutAccelerator) {
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 1u);
  ASSERT_TRUE(ash::PressSignOutAccelerator());
  ui_test_utils::BrowserDestroyedObserver settings_browser_closed_observer{
      browser()};
  settings_browser_closed_observer.Wait();
  ASSERT_EQ(GlobalBrowserCollection::GetInstance()->GetSize(), 0u);
}

}  // namespace ash