diff --git a/sidecar/build.gradle.kts b/sidecar/build.gradle.kts index 698eeaa..d39585d 100644 --- a/sidecar/build.gradle.kts +++ b/sidecar/build.gradle.kts @@ -26,6 +26,8 @@ val maestroVersion = "1.40.0" dependencies { implementation("dev.mobile:maestro-client:$maestroVersion") implementation("dev.mobile:maestro-utils:$maestroVersion") + implementation("dev.mobile:maestro-ios:$maestroVersion") + implementation("dev.mobile:maestro-ios-driver:$maestroVersion") implementation("io.grpc:grpc-netty-shaded:$grpcVersion") implementation("io.grpc:grpc-protobuf:$grpcVersion") diff --git a/sidecar/src/main/kotlin/dev/sanderling/sidecar/DriverBackend.kt b/sidecar/src/main/kotlin/dev/sanderling/sidecar/DriverBackend.kt index 62a0a21..3ff34b3 100644 --- a/sidecar/src/main/kotlin/dev/sanderling/sidecar/DriverBackend.kt +++ b/sidecar/src/main/kotlin/dev/sanderling/sidecar/DriverBackend.kt @@ -1,7 +1,7 @@ package dev.sanderling.sidecar interface DriverBackend { - fun launch(bundleId: String, clearState: Boolean) + fun launch(bundleId: String, clearState: Boolean, env: Map = emptyMap()) fun terminate(bundleId: String) fun tap(x: Int, y: Int) fun tapSelector(selector: String) @@ -131,7 +131,7 @@ class StubDriverBackend(private val platform: String) : DriverBackend { @Volatile var lastInputText: String? = null private set - override fun launch(bundleId: String, clearState: Boolean) { + override fun launch(bundleId: String, clearState: Boolean, env: Map) { launchCount++ lastBundleId = bundleId if (clearState) { @@ -386,9 +386,9 @@ class MaestroDriverBackend(private val serial: String?) : DriverBackend { driver.open() } - override fun launch(bundleId: String, clearState: Boolean) { + override fun launch(bundleId: String, clearState: Boolean, env: Map) { if (clearState) driver.clearAppState(bundleId) - driver.launchApp(bundleId, emptyMap(), java.util.UUID.randomUUID()) + driver.launchApp(bundleId, env, java.util.UUID.randomUUID()) } override fun terminate(bundleId: String) = driver.stopApp(bundleId) @@ -486,6 +486,76 @@ private fun pngHeight(bytes: ByteArray): Int { (bytes[22].toInt() and 0xFF shl 8) or (bytes[23].toInt() and 0xFF) } +class IosDriverBackend(private val udid: String) : DriverBackend { + private val driver: maestro.drivers.IOSDriver + + init { + val httpClient = xcuitest.api.OkHttpClientInstance.get() + val metrics = maestro.utils.NoOpMetrics() + val installer = xcuitest.installer.LocalXCTestInstaller( + udid, + "localhost", + false, + 9100, + metrics, + httpClient, + false, + false, + ) + val xcTestDriverClient = xcuitest.XCTestDriverClient(installer, httpClient, false) + val device = ios.xctest.XCTestIOSDevice(udid, xcTestDriverClient) { emptySet() } + driver = maestro.drivers.IOSDriver(device, maestro.utils.NoopInsights, metrics) + driver.open() + } + + override fun launch(bundleId: String, clearState: Boolean, env: Map) { + if (clearState) driver.clearAppState(bundleId) + driver.launchApp(bundleId, env, java.util.UUID.randomUUID()) + } + + override fun terminate(bundleId: String) = driver.stopApp(bundleId) + + override fun tap(x: Int, y: Int) = driver.tap(maestro.Point(x, y)) + + override fun tapSelector(selector: String) { + val root = driver.contentDescriptor(false) + val bounds = findBoundsBySelector(root, selector) ?: return + driver.tap(maestro.Point((bounds[0] + bounds[2]) / 2, (bounds[1] + bounds[3]) / 2)) + } + + override fun inputText(text: String) = driver.inputText(text) + + override fun swipe(fromX: Int, fromY: Int, toX: Int, toY: Int, durationMillis: Long) = + driver.swipe(maestro.Point(fromX, fromY), maestro.Point(toX, toY), maxOf(durationMillis, 250L)) + + override fun pressKey(key: String) { + StubDriverBackend.KEY_MAP[key]?.let { keyCode -> + keyCodeToMaestro(keyCode)?.let { driver.pressKey(it) } + } + } + + override fun screenshot(): Triple { + val buf = okio.Buffer() + driver.takeScreenshot(buf, false) + val bytes = buf.readByteArray() + return Triple(bytes, pngWidth(bytes), pngHeight(bytes)) + } + + override fun hierarchy(): String = + com.fasterxml.jackson.module.kotlin.jacksonObjectMapper() + .writeValueAsString(driver.contentDescriptor(false)) + + override fun recentLogs(sinceUnixMillis: Long, minLevel: String): List = emptyList() + + override fun waitForIdle(durationMillis: Long) { + driver.waitForAppToSettle(null, null, durationMillis.toInt()) + } + + override fun healthy() = runCatching { driver.contentDescriptor(false); true }.getOrElse { false } + + override fun metrics(bundleId: String) = MetricsSample(0.0, 0L, 0L) +} + private fun keyCodeToMaestro(adbKeyCode: String): maestro.KeyCode? { return when (adbKeyCode) { "KEYCODE_BACK" -> maestro.KeyCode.BACK