@skrillex1224/android-toolkit 0.1.0 → 0.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skrillex1224/android-toolkit",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "type": "module",
   "main": "index.js",
   "exports": {

package/src/frida-client.js

@@ -4,7 +4,7 @@ import {spawn} from 'node:child_process';
 import {Code} from './constants.js';
 import {adbShell} from './device.js';
 import {CrawlerError} from './errors.js';
-import {Logger} from './logger.js';
+import {Logger, sleep} from './logger.js';
 
 /**
  * 启动一个持续运行的 Frida WebView 事件记录器。
@@ -54,6 +54,22 @@ export async function startWebViewEventRecorder(ctx, config = {}, options = {})
  * 收集输出和错误显式化。
  */
 export async function runScript(ctx, source, options = {}) {
+    const attempts = Math.max(1, Number(options.attempts || 3));
+    let lastError = null;
+    for (let attempt = 1; attempt <= attempts; attempt += 1) {
+        try {
+            return await runScriptOnce(ctx, source, options);
+        } catch (error) {
+            lastError = error;
+            if (attempt >= attempts || error?.code !== Code.FridaUnavailable) throw error;
+            Logger.warn(`frida ${options.label || 'script'} 第 ${attempt} 次未返回，准备重试`);
+            await sleep(Number(options.retryDelayMs || 800));
+        }
+    }
+    throw lastError;
+}
+
+async function runScriptOnce(ctx, source, options = {}) {
     assertFridaReady(ctx, ctx.fridaWebViewEventAgentScript, 'frida webview event agent script');
     const marker = options.marker || 'ANDROID_TOOLKIT_SCRIPT_JSON ';
     const scriptPath = writeTempScript(ctx, source, options.label || 'script');
@@ -74,7 +90,7 @@ export async function runScript(ctx, source, options = {}) {
         maxLines: Number(options.maxLines || 1200),
         outputPath: options.outputPath || ''
     });
-    await recorder.waitForExit(Number(options.timeoutMs || 7000));
+    await recorder.waitForExit(Number(options.timeoutMs || 10000));
     const result = await recorder.stop();
     const scriptEvents = result.events.filter((event) => event?.type !== 'parse_error');
     if (scriptEvents.length === 0) {
@@ -187,7 +203,7 @@ Java.perform(function () {
         var view = views[i];
         if (!view.isShown()) continue;
         var value = '';
-        try { value = String(Java.cast(view, TextView).getText()); } catch (_) { continue; }
+        try { value = String(Java.cast(view, TextView).getText().toString()); } catch (_) { continue; }
         if (exact ? value === target : value.indexOf(target) >= 0) {
           var rect = rectOf(view);
           if (rect.right <= rect.left || rect.bottom <= rect.top) continue;
@@ -220,6 +236,205 @@ Java.perform(function () {
     return event.matches || [];
 }
 
+/**
+ * 列出当前 App 内部可见输入框。
+ *
+ * 微信这类 App 往往不给系统 uiautomator 暴露节点，但 App 进程内仍能看到
+ * EditText。业务方可以用 hint/text/bounds 挑选输入框，再交给 Device.click
+ * 或 setFocusedInputText 使用。
+ */
+export async function findVisibleInputs(ctx, options = {}) {
+    const event = await runScript(ctx, `
+Java.perform(function () {
+  function emit(payload) { console.log('ANDROID_TOOLKIT_SCRIPT_JSON ' + JSON.stringify(payload)); }
+  function rectOf(view) {
+    var Rect = Java.use('android.graphics.Rect');
+    var rect = Rect.$new();
+    try { view.getGlobalVisibleRect(rect); } catch (_) { }
+    return { left: rect.left.value, top: rect.top.value, right: rect.right.value, bottom: rect.bottom.value };
+  }
+  function walk(view, out, depth) {
+    if (!view || depth > ${Number(options.maxDepth || 20)} || out.length > ${Number(options.maxNodes || 3000)}) return;
+    out.push(view);
+    try {
+      var ViewGroup = Java.use('android.view.ViewGroup');
+      var group = Java.cast(view, ViewGroup);
+      for (var i = 0; i < group.getChildCount(); i++) walk(group.getChildAt(i), out, depth + 1);
+    } catch (_) { }
+  }
+  function activeActivities() {
+    var ActivityThread = Java.use('android.app.ActivityThread');
+    var Activity = Java.use('android.app.Activity');
+    var ArrayMap = Java.use('android.util.ArrayMap');
+    var thread = ActivityThread.currentActivityThread();
+    var field = thread.getClass().getDeclaredField('mActivities');
+    field.setAccessible(true);
+    var map = Java.cast(field.get(thread), ArrayMap);
+    var out = [];
+    for (var i = 0; i < map.size(); i++) {
+      var record = map.valueAt(i);
+      var recordClass = record.getClass();
+      var activityField = recordClass.getDeclaredField('activity');
+      var pausedField = recordClass.getDeclaredField('paused');
+      activityField.setAccessible(true);
+      pausedField.setAccessible(true);
+      if (!pausedField.getBoolean(record)) out.push(Java.cast(activityField.get(record), Activity));
+    }
+    return out;
+  }
+  Java.scheduleOnMainThread(function () {
+    try {
+      var EditText = Java.use('android.widget.EditText');
+      var inputs = [];
+      var activities = activeActivities();
+      for (var a = 0; a < activities.length; a++) {
+        var views = [];
+        walk(activities[a].getWindow().getDecorView(), views, 0);
+        for (var i = 0; i < views.length; i++) {
+          var view = views[i];
+          if (!view.isShown()) continue;
+          try {
+            var editText = Java.cast(view, EditText);
+            var rect = rectOf(view);
+            if (rect.right <= rect.left || rect.bottom <= rect.top) continue;
+            inputs.push({
+              text: String(editText.getText().toString()),
+              hint: String(editText.getHint() ? editText.getHint().toString() : ''),
+              className: String(view.getClass().getName()),
+              idName: (function () { try { return String(view.getResources().getResourceName(view.getId())); } catch (_) { return ''; } })(),
+              focused: Boolean(view.hasFocus()),
+              bounds: rect
+            });
+          } catch (_) { }
+        }
+      }
+      emit({ ok: true, inputs: inputs });
+    } catch (error) {
+      emit({ ok: false, error: String(error), stack: String(error.stack || '') });
+    }
+  });
+});
+`, {
+        label: options.label || 'find-visible-inputs',
+        packageName: options.packageName || ctx.packageName,
+        timeoutMs: options.timeoutMs || 7000
+    });
+    if (!event.ok) {
+        throw new CrawlerError({
+            message: `automation_failed: findVisibleInputs failed ${event.error || ''}`.trim(),
+            code: Code.AutomationFailed,
+            context: event
+        });
+    }
+    return event.inputs || [];
+}
+
+/**
+ * 给当前获得焦点的 EditText 设置文本，并触发常规输入事件。
+ *
+ * 这比 adb input text 更适合中文，也比直接 setText 更稳：它先 focus 目标
+ * EditText，再用 Editable.replace 更新文本，最后把光标放到末尾。
+ */
+export async function setFocusedInputText(ctx, text, options = {}) {
+    const event = await runScript(ctx, `
+Java.perform(function () {
+  function emit(payload) { console.log('ANDROID_TOOLKIT_SCRIPT_JSON ' + JSON.stringify(payload)); }
+  function rectOf(view) {
+    var Rect = Java.use('android.graphics.Rect');
+    var rect = Rect.$new();
+    try { view.getGlobalVisibleRect(rect); } catch (_) { }
+    return { left: rect.left.value, top: rect.top.value, right: rect.right.value, bottom: rect.bottom.value };
+  }
+  function walk(view, out, depth) {
+    if (!view || depth > ${Number(options.maxDepth || 20)} || out.length > ${Number(options.maxNodes || 3000)}) return;
+    out.push(view);
+    try {
+      var ViewGroup = Java.use('android.view.ViewGroup');
+      var group = Java.cast(view, ViewGroup);
+      for (var i = 0; i < group.getChildCount(); i++) walk(group.getChildAt(i), out, depth + 1);
+    } catch (_) { }
+  }
+  function activeActivities() {
+    var ActivityThread = Java.use('android.app.ActivityThread');
+    var Activity = Java.use('android.app.Activity');
+    var ArrayMap = Java.use('android.util.ArrayMap');
+    var thread = ActivityThread.currentActivityThread();
+    var field = thread.getClass().getDeclaredField('mActivities');
+    field.setAccessible(true);
+    var map = Java.cast(field.get(thread), ArrayMap);
+    var out = [];
+    for (var i = 0; i < map.size(); i++) {
+      var record = map.valueAt(i);
+      var recordClass = record.getClass();
+      var activityField = recordClass.getDeclaredField('activity');
+      var pausedField = recordClass.getDeclaredField('paused');
+      activityField.setAccessible(true);
+      pausedField.setAccessible(true);
+      if (!pausedField.getBoolean(record)) out.push(Java.cast(activityField.get(record), Activity));
+    }
+    return out;
+  }
+  Java.scheduleOnMainThread(function () {
+    try {
+      var EditText = Java.use('android.widget.EditText');
+      var targetText = ${JSON.stringify(String(text || ''))};
+      var activities = activeActivities();
+      var chosen = null;
+      var fallback = null;
+      for (var a = 0; a < activities.length; a++) {
+        var views = [];
+        walk(activities[a].getWindow().getDecorView(), views, 0);
+        for (var i = 0; i < views.length; i++) {
+          var view = views[i];
+          if (!view.isShown()) continue;
+          try {
+            var editText = Java.cast(view, EditText);
+            var rect = rectOf(view);
+            if (rect.right <= rect.left || rect.bottom <= rect.top) continue;
+            var item = { view: editText, bounds: rect, text: String(editText.getText().toString()), hint: String(editText.getHint() ? editText.getHint().toString() : ''), className: String(view.getClass().getName()), focused: Boolean(view.hasFocus()) };
+            if (item.focused) chosen = item;
+            if (!fallback) fallback = item;
+          } catch (_) { }
+        }
+      }
+      chosen = chosen || fallback;
+      if (!chosen) {
+        emit({ ok: false, error: 'visible EditText not found' });
+        return;
+      }
+      var JavaString = Java.use('java.lang.String');
+      var BufferType = Java.use('android.widget.TextView$BufferType');
+      var SpannableStringBuilder = Java.use('android.text.SpannableStringBuilder');
+      chosen.view.requestFocus();
+      chosen.view.setText(SpannableStringBuilder.$new(JavaString.$new(targetText)), BufferType.EDITABLE.value);
+      chosen.view.setSelection(targetText.length);
+      emit({
+        ok: true,
+        text: targetText,
+        hint: String(chosen.view.getHint() ? chosen.view.getHint().toString() : ''),
+        className: chosen.className,
+        bounds: chosen.bounds
+      });
+    } catch (error) {
+      emit({ ok: false, error: String(error), stack: String(error.stack || '') });
+    }
+  });
+});
+`, {
+        label: options.label || 'set-focused-input-text',
+        packageName: options.packageName || ctx.packageName,
+        timeoutMs: options.timeoutMs || 7000
+    });
+    if (!event.ok) {
+        throw new CrawlerError({
+            message: `automation_failed: setFocusedInputText failed ${event.error || ''}`.trim(),
+            code: Code.AutomationFailed,
+            context: event
+        });
+    }
+    return event;
+}
+
 /**
  * 找到目标包名当前可 attach 的 pid。
  *
@@ -425,5 +640,7 @@ export const Frida = {
     runScript,
     startActivityInsideApp,
     findVisibleTextBounds,
+    findVisibleInputs,
+    setFocusedInputText,
     resolvePid
 };