WebExtensionDemos.hs

module BiDi.Demos.WebExtensionDemos where

import BiDi.Actions (BiDiActions (..))
import BiDi.DemoUtils ( demo, BiDiDemo, runDemo )
import Control.Exception (SomeException, catch)
import IOUtils (DemoActions (..))
import TestData (demoExtensionAsBase64, demoExtensionDirPath, demoExtensionZipPath)
import WebDriverPreCore.BiDi.Protocol
    ( WebExtensionID(..),
      WebExtensionInstall(..),
      WebExtensionResult(..),
      WebExtensionUninstall(..) )
import Utils (txt)
import Prelude hiding (log, putStrLn)

-- this is just to silence "defined but not used" warnings for runDemo
_stopDemoUnusedWarning :: BiDiDemo -> IO ()
_stopDemoUnusedWarning = runDemo

-- >>> runDemo webExtensionInstallPathDemo
-- *** Exception: BiDIError (ProtocolException {error = UnknownError, description = "An unknown error occurred in the remote end while processing the command", message = "Method not available.", stacktrace = Just "Error\n    at new UnknownErrorException (<anonymous>:65:5630)\n    at CommandProcessor.processCommand (<anonymous>:485:13324)", errorData = Nothing, response = Object (fromList [("error",String "unknown error"),("id",Number 1.0),("message",String "Method not available."),("stacktrace",String "Error\n    at new UnknownErrorException (<anonymous>:65:5630)\n    at CommandProcessor.processCommand (<anonymous>:485:13324)"),("type",String "error")])})
webExtensionInstallPathDemo :: BiDiDemo
webExtensionInstallPathDemo =
  demo "WebExtension - Install from Path" action
  where
    action :: DemoActions -> BiDiActions -> IO ()
    action MkDemoActions {..} MkBiDiActions {..} = do
      logTxt "Test 1: Install extension from filesystem path"
      exPath <- demoExtensionDirPath
      result <- webExtensionInstall $ ExtensionPath exPath
      logShow "Extension installed from path" result
      pause

      logTxt "Test 2: Uninstall the extension"
      uninstallResult <- webExtensionUninstall $ MkWebExtensionUninstall result.extension
      logShow "Extension uninstalled" uninstallResult
      pause

      logTxt "Test 3: Attempt to uninstall non-existent extension"
      webExtensionUninstall (MkWebExtensionUninstall $ MkWebExtensionID "non-existent-extension-id")
        `catch` \(e :: SomeException) -> do
          logShow "Expected error for non-existent extension: " e
      pause

-- >>> runDemo webExtensionInstallArchiveDemo
webExtensionInstallArchiveDemo :: BiDiDemo
webExtensionInstallArchiveDemo =
  demo "WebExtension - Install from Archive" action
  where
    action :: DemoActions -> BiDiActions -> IO ()
    action MkDemoActions {..} MkBiDiActions {..} = do
      logTxt "Test 1: Install extension from zip archive"
      zipPath <- demoExtensionZipPath
      result1 <- webExtensionInstall $ ExtensionArchivePath zipPath
      logShow "Extension installed from archive" result1
      pause

-- >>> runDemo webExtensionInstallBase64Demo
webExtensionInstallBase64Demo :: BiDiDemo
webExtensionInstallBase64Demo =
  demo "WebExtension - Install from Base64" action
  where
    action :: DemoActions -> BiDiActions -> IO ()
    action MkDemoActions {..} MkBiDiActions {..} = do
      logTxt "Test 1: Install extension from base64 encoded data"

      base64Data <- demoExtensionAsBase64
      result1 <- webExtensionInstall $ ExtensionBase64Encoded base64Data
      logShow "Extension installed from base64" result1
      pause

-- >>> runDemo webExtensionValidationDemo
webExtensionValidationDemo :: BiDiDemo
webExtensionValidationDemo =
  demo "WebExtension - Extension Validation" action
  where
    action :: DemoActions -> BiDiActions -> IO ()
    action MkDemoActions {..} MkBiDiActions {..} = do
      logTxt "Test 1: Invalid extension path"
      let invalidExtension = ExtensionPath "/non/existent/path"
      invalidResult <-
        webExtensionInstall invalidExtension `catch` \(e :: SomeException) -> do
          logTxt $ "Installation failed as expected: " <> txt e
          pure $ WebExtensionInstallResult {extension = MkWebExtensionID "failed-extension"}
      logShow "Invalid extension handling" invalidResult
      pause

      logTxt "Test 2: Malformed base64 data"
      let malformedBase64 = ExtensionBase64Encoded "invalid-base64-data!!!"
      malformedResult <-
        webExtensionInstall malformedBase64 `catch` \(e :: SomeException) -> do
          logTxt $ "Base64 error (expected): " <> txt e
          pure $ WebExtensionInstallResult {extension = MkWebExtensionID "malformed-extension"}
      logShow "Malformed base64 handling" malformedResult
      pause