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

from __future__ import annotations

from typing import ClassVar

from typing_extensions import override

from crossbench.browsers.applescript import AppleScriptBrowser
from crossbench.browsers.attributes import BrowserAttributes
from crossbench.browsers.safari.safari import Safari


class SafariAppleScript(Safari, AppleScriptBrowser):
  APPLE_SCRIPT_ALLOW_JS_MENU: ClassVar[str] = (
      "Develop > Allow JavaScript from Apple Events")
  APPLE_SCRIPT_JS_COMMAND: ClassVar[str] = (
      "tell current tab of front window to do javascript %(js_script)s")
  APPLE_SCRIPT_SET_URL: ClassVar[str] = (
      "set URL of the current tab of front window to %(url)s")

  @override
  def _setup_window(self) -> None:
    self._exec_apple_script(f"""
      tell application "System Events"
          click menu item "New Private Window" of menu "File" of menu bar 1 of process "{self.bundle_name}"
      end tell
      set URL of current tab of front window to ""
    """)
    self.platform.sleep(0.5)
    if self.viewport.is_fullscreen:
      self._exec_apple_script("""
        tell application "System Events"
          keystroke "f" using {command down, control down}
        end tell""")
    elif self.viewport.is_maximized:
      self._exec_apple_script("""
        tell application "System Events"
          keystroke "m"
        end tell""")
    else:
      bounds = (f"{self.viewport.x},{self.viewport.y},"
                f"{self.viewport.width},{self.viewport.height}")
      self._exec_apple_script("set the bounds of the first window to {%s}" %
                              bounds)

  @override
  def quit(self) -> None:
    super().quit()
    # Safari doesn't react to "quit" when using the full app path.
    self.platform.exec_apple_script(f"""
        tell application "{self.bundle_name}"
          quit
        end tell""")

  @classmethod
  @override
  def attributes(cls) -> BrowserAttributes:
    return BrowserAttributes.SAFARI | BrowserAttributes.APPLESCRIPT