// 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.

#ifndef CHROME_BROWSER_UI_ASH_TEST_UTIL_H_
#define CHROME_BROWSER_UI_ASH_TEST_UTIL_H_

#include "ash/wm/splitview/split_view_types.h"
#include "base/test/test_future.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/mixin_based_in_process_browser_test.h"
#include "chromeos/ui/base/window_state_type.h"
#include "ui/aura/window.h"
#include "ui/aura/window_observer.h"

class BrowserFrameViewChromeOS;
class BrowserView;
class BrowserWindowInterface;

namespace views {
class Widget;
}

// This class was meant primarily for writing UI-centric ChromeOS browser tests
// that work for both Ash and Lacros. With Lacros gone, it has no purpose
// anymore and could be removed (but it's still used by tests).
class ChromeOSBrowserUITest : public MixinBasedInProcessBrowserTest {
 public:
  ChromeOSBrowserUITest() = default;
  ChromeOSBrowserUITest(const ChromeOSBrowserUITest&) = delete;
  ChromeOSBrowserUITest& operator=(const ChromeOSBrowserUITest&) = delete;
  ~ChromeOSBrowserUITest() override = default;

  // MixinBasedInProcessBrowserTest:
  void SetUpDefaultCommandLine(base::CommandLine* command_line) override;
  void TearDownOnMainThread() override;

  // Checks whether the ChromeOS UI is currently in tablet mode (as opposed to
  // clamshell mode).
  static bool InTabletMode();

  // Makes the ChromeOS UI enter/exit tablet mode.
  void EnterTabletMode();
  void ExitTabletMode();

  // Makes the ChromeOS UI enter/exit overview mode.
  static void EnterOverviewMode();
  static void ExitOverviewMode();

  // Snaps the given window to the given part of the screen
  static void SnapWindow(aura::Window* window, ash::SnapPosition position);

  // Pins the given window. This is also known as locked fullscreen mode.
  static void PinWindow(aura::Window* window, bool trusted);

  // Checks whether the ChromeOS UI currently shows the shelf (on the primary
  // display).
  static bool IsShelfVisible();

  // Deactivates the given widget.
  static void DeactivateWidget(views::Widget* widget);

  // Enters/exits immersive fullscreen mode for the given browser.
  static void EnterImmersiveFullscreenMode(BrowserWindowInterface* browser);
  static void ExitImmersiveFullscreenMode(BrowserWindowInterface* browser);

  // Enters/exits fullscreen mode in th tab associated with the given contents.
  static void EnterTabFullscreenMode(BrowserWindowInterface* browser,
                                     content::WebContents* web_contents);
  static void ExitTabFullscreenMode(BrowserWindowInterface* browser,
                                    content::WebContents* web_contents);

  // Returns the non-client frame view for `browser_view`.
  static BrowserFrameViewChromeOS* GetFrameViewChromeOS(
      BrowserView* browser_view);

 private:
  void SetTabletMode(bool enabled);
  static void SetOverviewMode(bool enabled);
};

#endif  // CHROME_BROWSER_UI_ASH_TEST_UTIL_H_