feat: Add mobile endpoint to retrieve UIAutomator page source dump

espresso-server/app/src/androidTest/java/io/appium/espressoserver/lib/handlers/UiautomatorPageSource.kt

@@ -0,0 +1,44 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.appium.espressoserver.lib.handlers
+
+import io.appium.espressoserver.lib.handlers.exceptions.AppiumException
+import io.appium.espressoserver.lib.helpers.InteractionHelper
+import io.appium.espressoserver.lib.model.AppiumParams
+import java.io.File
+
+class UiautomatorPageSource : RequestHandler<AppiumParams, String> {
+
+    @Throws(AppiumException::class)
+    override fun handleInternal(params: AppiumParams): String {
+        var dumpXml = ""
+        var dumpView: File? = null
+
+        try {
+            val uiDevice = InteractionHelper.getUiDevice()
+            dumpView = File.createTempFile("window_dump", "uix")
+            uiDevice.dumpWindowHierarchy(dumpView)
+            dumpXml = dumpView.inputStream().readBytes().toString(Charsets.UTF_8)
+            dumpView.delete()
+        } catch (e: Exception) {
+            throw AppiumException("Could not get page source with UiAutomator", e)
+        } finally {
+            dumpView?.delete()
+        }
+        return dumpXml
+    }
+}

espresso-server/app/src/androidTest/java/io/appium/espressoserver/lib/http/Router.kt

@@ -130,6 +130,7 @@ internal class Router {
         routeMap.addRoute(RouteDefinition(Method.POST, "/session/:sessionId/appium/execute_mobile/backdoor", MobileBackdoor(), MobileBackdoorParams::class.java))
         routeMap.addRoute(RouteDefinition(Method.POST, "/session/:sessionId/appium/execute_mobile/:elementId/flash", MobileViewFlash(), ViewFlashParams::class.java))
         routeMap.addRoute(RouteDefinition(Method.POST, "/session/:sessionId/appium/execute_mobile/uiautomator", Uiautomator(), UiautomatorParams::class.java))
+        routeMap.addRoute(RouteDefinition(Method.GET, "/session/:sessionId/appium/execute_mobile/uiautomator_page_source", UiautomatorPageSource(), AppiumParams::class.java))
         routeMap.addRoute(RouteDefinition(Method.POST, "/session/:sessionId/appium/execute_mobile/:elementId/click_action", MobileClickAction(), MobileClickActionParams::class.java))
         routeMap.addRoute(RouteDefinition(Method.POST, "/session/:sessionId/appium/execute_mobile/web_atoms", WebAtoms(), WebAtomsParams::class.java))
         routeMap.addRoute(RouteDefinition(Method.POST, "/session/:sessionId/appium/execute_mobile/:elementId/dismiss_autofill", PerformAutofillDismissal(), AppiumParams::class.java))

lib/commands/execute.js

@@ -43,6 +43,8 @@ extensions.executeMobile = async function executeMobile (mobileCommand, opts = {
 
     uiautomator: 'mobileUiautomator',
 
+    uiautomatorPageSource: 'mobileUiAutomatorPageSource',
+
     clickAction: 'mobileClickAction',
 
     webAtoms: 'mobileWebAtoms',

lib/commands/general.js

@@ -244,6 +244,14 @@ commands.mobileUiautomator = async function mobileUiautomator (opts = {}) {
   return await this.espresso.jwproxy.command(`/appium/execute_mobile/uiautomator`, 'POST', {strategy, locator, index, action});
 };
 
+/**
+ *  Execute UiAutomator2 command to return the UI dump when AUT is in background.
+ *
+ */
+commands.mobileUiAutomatorPageSource = async function mobileUiAutomatorPageSource () {
+  return await this.espresso.jwproxy.command(`/appium/execute_mobile/uiautomator_page_source`, 'GET');
+};
+
 /**
  *  Flash the element with given id.
  *  durationMillis and repeatCount are optional