From 3af8414ae37f93751b2090a98801f75b153d95e0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Wojciech=20Tyczy=C5=84ski?= <tykus160@gmail.com>
Date: Thu, 10 Mar 2022 23:25:31 +0100
Subject: [PATCH] feat: Add Safari Technology Preview, iOS and iPadOS with
 simulators (#33)

Works OK with Technology Preview and simulators. Wasn't able to check on actual devices.
---
 index.js | 36 +++++++++++++++++++++++++++++++++---
 1 file changed, 33 insertions(+), 3 deletions(-)

diff --git a/index.js b/index.js
index 071babc..abcabf7 100644
--- a/index.js
+++ b/index.js
@@ -220,6 +220,18 @@ function generateSubclass(
   return subclass;
 }
 
+function generateSafariDriver(name, device = 'mac', simulator = false) {
+  const config = device == 'mac' ? {} : {
+    platformName: 'iOS',
+    'safari:deviceType': device,
+    'safari:useSimulator': simulator,
+  };
+  return generateSubclass('Safari', name,
+      '/usr/bin/safaridriver',
+      (port) => ['--port=' + port],
+      config);
+}
+
 const LocalWebDriverChrome = generateSubclass(
     'Chrome', 'Chrome',
     'chromedriver',
@@ -276,9 +288,19 @@ const LocalWebDriverFirefoxHeadless = generateSubclass(
       },
     });
 
-const LocalWebDriverSafari = generateSubclass(
-    'Safari', 'Safari',
-    '/usr/bin/safaridriver',
+const LocalWebDriverSafari = generateSafariDriver('Safari');
+
+const LocalWebDriverSafariIOS = generateSafariDriver('SafariIOS', 'iPhone');
+const LocalWebDriverSafariIOSSim =
+    generateSafariDriver('SafariIOSSim', 'iPhone', true);
+const LocalWebDriverSafariIPadOS =
+    generateSafariDriver('SafariIPadOS', 'iPad');
+const LocalWebDriverSafariIPadOSSim =
+    generateSafariDriver('SafariIPadOSSim', 'iPad', true);
+
+const LocalWebDriverSafariTP = generateSubclass(
+    'Safari Technology Preview', 'Safari Technology Preview',
+    '/Applications/Safari Technology Preview.app/Contents/MacOS/safaridriver',
     (port) => ['-p', port]);
 
 module.exports = {
@@ -293,4 +315,12 @@ module.exports = {
 // Safari is only supported on Mac.
 if (os.platform() == 'darwin') {
   module.exports['launcher:Safari'] = ['type', LocalWebDriverSafari];
+  module.exports['launcher:SafariIOS'] = ['type', LocalWebDriverSafariIOS];
+  module.exports['launcher:SafariIOSSim'] =
+      ['type', LocalWebDriverSafariIOSSim];
+  module.exports['launcher:SafariIPadOS'] =
+      ['type', LocalWebDriverSafariIPadOS];
+  module.exports['launcher:SafariIPadOSSim'] =
+      ['type', LocalWebDriverSafariIPadOSSim];
+  module.exports['launcher:SafariTP'] = ['type', LocalWebDriverSafariTP];
 }