automation.cs支持不同条件查找windows，前端依然限制只通过句柄查找

plugin/lib/csharp/automation.cs

@@ -10,6 +10,7 @@ using System.Linq;
 using System.Threading;
 using System.Windows.Forms;
 using System.Drawing;
+using System.Diagnostics;
 
 public class AutomationManager
 {
@@ -72,6 +73,17 @@ public class AutomationManager
   [DllImport("user32.dll")]
   static extern int SetWindowLong(IntPtr hwnd, int index, int newStyle);
 
+  [DllImport("user32.dll")]
+  static extern bool IsWindow(IntPtr hWnd);
+
+  [DllImport("user32.dll")]
+  private static extern bool EnumWindows(EnumWindowsProc lpEnumFunc, IntPtr lParam);
+
+  [DllImport("user32.dll")]
+  private static extern bool IsWindowVisible(IntPtr hWnd);
+
+  private delegate bool EnumWindowsProc(IntPtr hWnd, IntPtr lParam);
+
   private const int GWL_EXSTYLE = -20;
   private const int WS_EX_TRANSPARENT = 0x20;
   private const int WS_EX_LAYERED = 0x80000;
@@ -342,26 +354,23 @@ public class AutomationManager
   {
     // 获取起始窗口
     AutomationElement root;
-    string windowArg = GetArgumentValue(args, "-window");
-    if (!string.IsNullOrEmpty(windowArg))
-    {
-      // 通过窗口句柄查找
-      int handle = int.Parse(windowArg);
     try
     {
-        root = AutomationElement.FromHandle(new IntPtr(handle));
+      List<IntPtr> targetWindows = GetTargetWindows(args);
+      if (targetWindows.Count == 0)
+      {
+        throw new Exception("未找到目标窗口");
+      }
+
+      // 总是使用第一个窗口
+      IntPtr handle = targetWindows[0];
+
+      root = AutomationElement.FromHandle(handle);
     }
     catch
     {
       throw new Exception("无法获取指定的窗口");
     }
-    }
-    else
-    {
-      // 使用当前活动窗口
-      IntPtr activeHandle = GetForegroundWindow();
-      root = AutomationElement.FromHandle(activeHandle);
-    }
 
     if (root == null)
     {
@@ -456,6 +465,89 @@ public class AutomationManager
     throw new Exception("必须指定元素的识别方式: -xpath, -id, -name 或 -condition");
   }
 
+  private static List<IntPtr> GetTargetWindows(string[] args)
+  {
+    List<IntPtr> targetWindows = new List<IntPtr>();
+    string method = GetArgumentValue(args, "-method") ?? "handle";
+    string value = GetArgumentValue(args, "-window") ?? "";
+
+    switch (method.ToLower())
+    {
+      case "handle":
+        IntPtr handle = new IntPtr(long.Parse(value));
+        if (!IsWindow(handle))
+        {
+          throw new Exception("指定的句柄不是一个有效的窗口句柄");
+        }
+        targetWindows.Add(handle);
+        break;
+
+      case "active":
+        targetWindows.Add(GetForegroundWindow());
+        break;
+
+      case "process":
+        var processes = Process.GetProcessesByName(value);
+        foreach (var process in processes)
+        {
+          if (process.MainWindowHandle != IntPtr.Zero)
+          {
+            targetWindows.Add(process.MainWindowHandle);
+          }
+        }
+        break;
+
+      case "class":
+        EnumWindows((hwnd, param) =>
+        {
+          if (!IsWindowVisible(hwnd))
+          {
+            return true;
+          }
+
+          StringBuilder className = new StringBuilder(256);
+          GetClassName(hwnd, className, className.Capacity);
+          string windowClassName = className.ToString();
+
+          if (windowClassName.IndexOf(value, StringComparison.OrdinalIgnoreCase) >= 0)
+          {
+            targetWindows.Add(hwnd);
+          }
+          return true;
+        }, IntPtr.Zero);
+        break;
+
+      case "title":
+      default:
+        if (!string.IsNullOrEmpty(value))
+        {
+          EnumWindows((hwnd, param) =>
+          {
+            StringBuilder title = new StringBuilder(256);
+            GetWindowText(hwnd, title, title.Capacity);
+            string windowTitle = title.ToString();
+
+            if (!string.IsNullOrEmpty(windowTitle) &&
+                          windowTitle.IndexOf(value, StringComparison.OrdinalIgnoreCase) >= 0)
+            {
+              targetWindows.Add(hwnd);
+            }
+            return true;
+          }, IntPtr.Zero);
+        }
+        break;
+    }
+
+    if (targetWindows.Count == 0)
+    {
+      throw new Exception("未找到匹配的窗口");
+    }
+
+    return targetWindows;
+  }
+
+
+
   private static void ShowHelp()
   {
     Console.WriteLine(@"UI自动化工具 v1.0

plugin/lib/quickcomposer/windows/automation.js

@@ -20,7 +20,9 @@ async function runAutomation(
   const args = [];
 
   if (window) {
-    args.push("-window", window);
+    args.push("-method", "handle", "-window", window);
+  } else {
+    args.push("-method", "active");
   }
 
   args.push("-type", type);

src/js/composer/commands/windowsCommands.js

@@ -157,7 +157,7 @@ const windowHandleConfig = [
 ];
 
 const searchElementConfig = [
-  windowHandleConfig,
+  ...windowHandleConfig,
   {
     label: "元素查找方式",
     component: "q-select",