ascript 1.3.27__py3-none-win_amd64.whl

ascript/windows/ui/web_window.py

@@ -0,0 +1,466 @@
+import threading
+import subprocess
+import tempfile
+import ctypes
+from ctypes import wintypes
+import time
+import webbrowser
+import datetime
+import logging
+import webview
+import os
+from functools import partial
+from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
+from typing import Optional, Callable, Dict
+from urllib.parse import quote
+
+# WebView2 运行时：官方下载页 + 常青版引导安装器（约 2MB，联网拉取完整运行时）
+_WEBVIEW2_DOWNLOAD_URL = "https://developer.microsoft.com/microsoft-edge/webview2/"
+_WEBVIEW2_BOOTSTRAPPER_URL = "https://go.microsoft.com/fwlink/p/?LinkId=2124703"
+
+# 诊断日志：定位握手故障点，写到用户主目录
+_DIAG_LOG = os.path.join(os.path.expanduser("~"), "ascript_diag.log")
+
+
+class _IgnoreMissingWebViewDataFolder(logging.Filter):
+    """Hide pywebview's harmless double-delete warning on Windows shutdown."""
+
+    def filter(self, record):
+        message = record.getMessage()
+        return not (
+            message.startswith("Failed to delete user data folder:")
+            and ("WinError 3" in message or "No such file or directory" in message)
+        )
+
+
+logging.getLogger("pywebview").addFilter(_IgnoreMissingWebViewDataFolder())
+
+
+class _LocalAssetHandler(SimpleHTTPRequestHandler):
+    """Serve one window's local assets without pywebview's global server."""
+
+    def log_message(self, format, *args):
+        pass
+
+    def end_headers(self):
+        self.send_header("Cache-Control", "no-cache, no-store, must-revalidate")
+        self.send_header("Pragma", "no-cache")
+        self.send_header("Expires", "0")
+        super().end_headers()
+
+
+def _diag(msg, reset=False):
+    """把诊断信息写入 ~/ascript_diag.log。reset=True 时清空重写。"""
+    try:
+        mode = "w" if reset else "a"
+        ts = datetime.datetime.now().strftime("%H:%M:%S.%f")[:-3]
+        with open(_DIAG_LOG, mode, encoding="utf-8") as f:
+            f.write(f"[{ts}] {msg}\n")
+    except Exception:
+        pass
+
+
+# Win32 MessageBox 标志位与返回值
+_MB_YESNO = 0x04
+_MB_ICONERROR = 0x10
+_MB_ICONWARNING = 0x30
+_MB_ICONINFORMATION = 0x40
+_IDYES = 6
+
+
+def _msgbox(text: str, title: str = "AScript", style: int = _MB_ICONERROR) -> int:
+    """弹出 Windows 原生消息框。"""
+    return ctypes.windll.user32.MessageBoxW(0, text, title, style)
+
+
+def _install_webview2() -> bool:
+    """下载并静默安装 WebView2 Runtime。成功返回 True。"""
+    try:
+        import requests
+        bootstrapper = os.path.join(tempfile.gettempdir(), "MicrosoftEdgeWebview2Setup.exe")
+        _diag("downloading WebView2 bootstrapper")
+        with requests.get(_WEBVIEW2_BOOTSTRAPPER_URL, stream=True, timeout=120) as resp:
+            resp.raise_for_status()
+            with open(bootstrapper, "wb") as f:
+                for chunk in resp.iter_content(chunk_size=1024 * 64):
+                    f.write(chunk)
+        _diag("running WebView2 bootstrapper /silent /install")
+        result = subprocess.run([bootstrapper, "/silent", "/install"], timeout=600)
+        _diag(f"WebView2 installer returncode={result.returncode}")
+        return result.returncode == 0
+    except Exception as e:
+        _diag(f"WebView2 install failed: {e!r}")
+        return False
+
+
+def _install_webview2_with_ui() -> bool:
+    """带进度窗口地安装 WebView2。MS 静默安装器不汇报百分比，
+    故用滚动式（不确定模式）进度条表示"正在进行"。tkinter 不可用时退化为无窗口安装。"""
+    state = {"ok": False, "done": False}
+
+    def worker():
+        state["ok"] = _install_webview2()
+        state["done"] = True
+
+    t = threading.Thread(target=worker, daemon=True)
+    t.start()
+
+    try:
+        import tkinter as tk
+        from tkinter import ttk
+
+        root = tk.Tk()
+        root.title("AScript - 安装运行环境")
+        root.resizable(False, False)
+        w, h = 400, 140
+        sw, sh = root.winfo_screenwidth(), root.winfo_screenheight()
+        root.geometry(f"{w}x{h}+{(sw - w) // 2}+{(sh - h) // 2}")
+        root.attributes("-topmost", True)
+        root.protocol("WM_DELETE_WINDOW", lambda: None)  # 安装中禁止关闭
+
+        tk.Label(root, text="正在下载并安装 Microsoft Edge WebView2 运行时").pack(pady=(26, 4))
+        tk.Label(root, text="约需 1-2 分钟，请勿关闭此窗口...", fg="#666").pack()
+        bar = ttk.Progressbar(root, mode="indeterminate", length=340)
+        bar.pack(pady=18)
+        bar.start(15)
+
+        def poll():
+            if state["done"]:
+                bar.stop()
+                root.destroy()
+            else:
+                root.after(200, poll)
+
+        root.after(200, poll)
+        root.mainloop()
+    except Exception as e:
+        _diag(f"progress UI unavailable, installing without it: {e!r}")
+        t.join()  # 没有 UI 也要等安装完成
+
+    return state["ok"]
+
+
+# 注入到页面的桥接代码：定义 callPython 并派发 onPythonReady。
+# 故意使用现代语法（async / ...rest）—— 既是功能需要，也是引擎能力探针：
+# 真正的 WebView2/Chromium 能解析，IE/MSHTML 会抛 SyntaxError。
+# IIFE + callPython 存在性判断 → 多次注入也幂等。
+_INJECTION_CODE = """
+(function () {
+    if (!window.callPython) {
+        window.callPython = async function (fnName, ...args) {
+            return await window.pywebview.api.call_internal(fnName, args);
+        };
+        try { window.pywebview.api.call_internal('_diag', ['JS: callPython injected']); } catch (e) {}
+    }
+    if (!window.__ascriptPythonReady) {
+        window.__ascriptPythonReady = true;
+        window.dispatchEvent(new CustomEvent('onPythonReady'));
+    }
+})();
+"""
+
+
+class WebWindow:
+    """基于 pywebview 的 Web 窗口基类。"""
+
+    def __init__(self, html_path: str, project_root: Optional[str] = None, query: Optional[str] = None):
+        self.project_root = os.path.abspath(project_root) if project_root else os.getcwd()
+        if html_path.startswith(("http://", "https://")):
+            self.abs_path = html_path
+        else:
+            self.abs_path = os.path.abspath(html_path)
+        self.query = query.lstrip("?") if query else ""
+        self._exposed_functions: Dict[str, Callable] = {}
+        self.webview_instance = None
+        self._return_value = None
+        self._injected = False
+        self._engine_error_handled = False
+        self._asset_server = None
+        self._exclude_from_capture = False
+        self._capture_exclusion_active = False
+
+    def _native_hwnd(self) -> Optional[int]:
+        """Return the native top-level HWND after pywebview has created it."""
+        if os.name != "nt" or not self.webview_instance:
+            return None
+        try:
+            hwnd = int(self.webview_instance.native.Handle.ToInt64())
+            get_ancestor = ctypes.windll.user32.GetAncestor
+            get_ancestor.argtypes = (wintypes.HWND, wintypes.UINT)
+            get_ancestor.restype = wintypes.HWND
+            root_hwnd = get_ancestor(wintypes.HWND(hwnd), 2)  # GA_ROOT
+            return int(root_hwnd or hwnd)
+        except Exception as error:
+            _diag(f"native window handle unavailable: {error!r}")
+            return None
+
+    def _retry_capture_exclusion(self) -> None:
+        """Retry after WebView2 has finished creating its top-level window."""
+        if not self._exclude_from_capture or self._capture_exclusion_active:
+            return
+
+        def apply():
+            for delay in (0.05, 0.2, 0.5):
+                time.sleep(delay)
+                if self.set_capture_excluded(True):
+                    return
+
+        threading.Thread(target=apply, daemon=True).start()
+
+    def set_capture_excluded(self, excluded: bool = True) -> bool:
+        """Exclude this SDK-owned window from Windows screen capture.
+
+        WDA_EXCLUDEFROMCAPTURE leaves the window visible and interactive for the
+        user, but removes it from supported desktop-capture APIs. Unlike
+        minimizing/hiding, it does not disturb WebView2 focus or composition.
+        """
+        self._exclude_from_capture = bool(excluded)
+        hwnd = self._native_hwnd()
+        if hwnd is None:
+            return False
+        affinity = 0x00000011 if excluded else 0x00000000
+        try:
+            setter = ctypes.windll.user32.SetWindowDisplayAffinity
+            setter.argtypes = (wintypes.HWND, wintypes.DWORD)
+            setter.restype = wintypes.BOOL
+            success = bool(setter(hwnd, affinity))
+            self._capture_exclusion_active = bool(success and excluded)
+            _diag(
+                f"capture exclusion {'enabled' if excluded else 'disabled'}, "
+                f"hwnd={hwnd}, success={success}"
+            )
+            return success
+        except (AttributeError, OSError) as error:
+            self._capture_exclusion_active = False
+            _diag(f"capture exclusion failed, hwnd={hwnd}: {error!r}")
+            return False
+
+    def _window_url(self) -> str:
+        """Return a URL for the configured page, starting a private local server if needed."""
+        if self.abs_path.startswith(("http://", "https://")):
+            url = self.abs_path
+        else:
+            if not os.path.isfile(self.abs_path):
+                raise FileNotFoundError(f"WebWindow 页面不存在: {self.abs_path}")
+
+            project_root = os.path.abspath(self.project_root)
+            try:
+                inside_project = os.path.normcase(os.path.commonpath([project_root, self.abs_path])) == os.path.normcase(project_root)
+            except ValueError:
+                inside_project = False
+            server_root = project_root if inside_project else os.path.dirname(self.abs_path)
+            relative_path = os.path.relpath(self.abs_path, server_root).replace(os.sep, "/")
+
+            handler = partial(_LocalAssetHandler, directory=server_root)
+            self._asset_server = ThreadingHTTPServer(("127.0.0.1", 0), handler)
+            self._asset_server.daemon_threads = True
+            threading.Thread(target=self._asset_server.serve_forever, daemon=True).start()
+            host, port = self._asset_server.server_address
+            url = f"http://{host}:{port}/{quote(relative_path, safe='/')}"
+            _diag(f"local asset server started, root={server_root}, url={url}")
+
+        if self.query:
+            separator = "&" if "?" in url else "?"
+            url += separator + self.query
+        return url
+
+    def _stop_asset_server(self):
+        server = self._asset_server
+        self._asset_server = None
+        if server is None:
+            return
+        try:
+            server.shutdown()
+            server.server_close()
+            _diag("local asset server stopped")
+        except Exception as error:
+            _diag(f"local asset server stop failed: {error!r}")
+
+    def expose(self, func: Callable):
+        """将 Python 函数暴露给前端 JavaScript 调用。"""
+        self._exposed_functions[func.__name__] = func
+        return func
+
+    def _inject(self, source: str):
+        """注入 callPython 桥接并派发 onPythonReady。可被多个时机触发，幂等。"""
+        if self._engine_error_handled:
+            return
+        _diag(f"_inject START, source={source}")
+        try:
+            self.webview_instance.evaluate_js(_INJECTION_CODE)
+            self._injected = True
+            _diag(f"_inject DONE, source={source}")
+        except Exception as e:
+            _diag(f"_inject ERROR, source={source}: {e!r}")
+            # SyntaxError 是确凿信号：现代引擎绝不会对合法的现代 JS 报 SyntaxError，
+            # 只有 IE/MSHTML 才会 → 据此判定引擎不可用。不查注册表，不会误判。
+            # 其它异常（多为时机问题）只记录，留给另一个注入时机重试。
+            if "SyntaxError" in str(e):
+                self._handle_engine_error()
+
+    def _handle_engine_error(self):
+        """检测到当前是 IE/MSHTML 引擎（WebView2 不可用）。
+        现代 JS 在此引擎下无法运行，关闭卡死的窗口并引导用户安装 WebView2。"""
+        if self._engine_error_handled:
+            return
+        self._engine_error_handled = True
+        _diag("ENGINE ERROR: modern JS rejected -> WebView2 not in use (MSHTML fallback)")
+        # 先隐藏卡死的窗口，但不要 destroy —— destroy 会让 webview.start 返回、
+        # 主线程退出，把后面的安装进度窗口一起带走。等安装流程走完再关。
+        try:
+            self.webview_instance.hide()
+        except Exception:
+            pass
+        choice = _msgbox(
+            "AScript 无法启动：界面引擎回退到了过时的 IE 内核，\n"
+            "通常是缺少 Microsoft Edge WebView2 运行时。\n\n"
+            "是否现在自动下载并安装？（约 1-2 分钟，需联网）",
+            "缺少运行环境", _MB_YESNO | _MB_ICONWARNING)
+        if choice == _IDYES:
+            if _install_webview2_with_ui():
+                _msgbox("WebView2 安装完成，请重新启动 AScript。", "AScript", _MB_ICONINFORMATION)
+            else:
+                _msgbox("自动安装失败，即将打开下载页面，请手动安装后重新启动程序。")
+                webbrowser.open(_WEBVIEW2_DOWNLOAD_URL)
+        else:
+            _msgbox("已取消。请手动安装 WebView2 运行时后重新启动程序：\n" + _WEBVIEW2_DOWNLOAD_URL)
+        # 全部处理完，关闭窗口让 webview.start 返回、进程干净退出
+        try:
+            self.webview_instance.destroy()
+        except Exception:
+            pass
+
+    def _bring_to_front(self, keep_topmost: bool = False):
+        """Briefly raise the native window so launches from a browser stay visible."""
+        if os.name != "nt" or not self.webview_instance:
+            return
+        hwnd = self._native_hwnd()
+        if hwnd is None:
+            return
+
+        user32 = ctypes.windll.user32
+        flags = 0x0001 | 0x0002 | 0x0040  # SWP_NOSIZE | SWP_NOMOVE | SWP_SHOWWINDOW
+        try:
+            user32.ShowWindow(hwnd, 9)  # SW_RESTORE
+            user32.SetWindowPos(hwnd, -1, 0, 0, 0, 0, flags)  # HWND_TOPMOST
+            user32.BringWindowToTop(hwnd)
+            user32.SetForegroundWindow(hwnd)
+            _diag(f"bring-to-front requested, hwnd={hwnd}")
+        except Exception as e:
+            _diag(f"bring-to-front failed: {e!r}")
+            return
+
+        if keep_topmost:
+            _diag(f"permanent topmost preserved, hwnd={hwnd}")
+            return
+
+        def release_topmost():
+            time.sleep(0.8)
+            try:
+                user32.SetWindowPos(hwnd, -2, 0, 0, 0, 0, flags)  # HWND_NOTOPMOST
+                _diag(f"temporary topmost released, hwnd={hwnd}")
+            except Exception as e:
+                _diag(f"release temporary topmost failed: {e!r}")
+
+        threading.Thread(target=release_topmost, daemon=True).start()
+
+    def show(
+        self,
+        title: str = "AScript Window",
+        debug=False,
+        block: bool = True,
+        bring_to_front: bool = True,
+        exclude_from_capture: bool = False,
+        **kwargs,
+    ):
+        """创建并显示 Web 窗口。"""
+        _diag(f"show() called, html={self.abs_path}, block={block}", reset=True)
+        self._exclude_from_capture = bool(exclude_from_capture)
+
+        class ApiBridge:
+            """JavaScript 与 Python 之间的 API 桥接。"""
+
+            def __init__(self, funcs):
+                self._funcs = funcs
+
+            def call_internal(self, func_name, args):
+                _diag(f"call_internal ENTER: {func_name} args={args!r}")
+                f = self._funcs.get(func_name)
+                if not f:
+                    _diag(f"call_internal: '{func_name}' NOT FOUND")
+                    return None
+                try:
+                    result = f(*args)
+                    _diag(f"call_internal EXIT: {func_name} -> {repr(result)[:200]}")
+                    return result
+                except Exception as e:
+                    _diag(f"call_internal ERROR: {func_name}: {e!r}")
+                    print(f"Error calling {func_name}: {e}")
+                    return None
+
+        # 诊断用：让 JS 也能往同一份日志里写
+        self._exposed_functions.setdefault("_diag", lambda m: _diag(str(m)))
+
+        window_url = self._window_url()
+        self.webview_instance = webview.create_window(
+            title=title,
+            url=window_url,
+            js_api=ApiBridge(self._exposed_functions),
+            **kwargs
+        )
+        _diag("window created")
+
+        def on_loaded(*args):
+            _diag(f"events.loaded fired, args={args!r}")
+            if self._exclude_from_capture and not self._capture_exclusion_active:
+                self.set_capture_excluded(True)
+                self._retry_capture_exclusion()
+            self._inject("events.loaded")
+
+        def on_shown(*args):
+            _diag(f"events.shown fired, args={args!r}")
+            if self._exclude_from_capture:
+                self.set_capture_excluded(True)
+                self._retry_capture_exclusion()
+            if bring_to_front:
+                self._bring_to_front(keep_topmost=bool(kwargs.get("on_top")))
+
+        def on_start():
+            # webview.start 的启动钩子，作为 events.loaded 的兜底注入时机
+            _diag("webview.start hook fired")
+            self._inject("start-hook")
+
+        self.webview_instance.events.loaded += on_loaded
+        self.webview_instance.events.shown += on_shown
+        _diag("events.loaded subscribed")
+
+        # 强制 edgechromium：装了 WebView2 时确保用它。
+        # 但 pywebview 在 WebView2 缺失时会"静默回退"到 MSHTML 且不抛异常，
+        # 所以真正的判定靠 _inject 里对真实引擎的运行时探测，而非这里。
+        if block:
+            try:
+                _diag("calling webview.start (block) gui=edgechromium")
+                webview.start(on_start, debug=debug, gui='edgechromium')
+                _diag("webview.start returned")
+            except Exception as e:
+                _diag(f"webview.start raised: {e!r}")
+            finally:
+                self._stop_asset_server()
+            return self._return_value
+        else:
+            def _run():
+                try:
+                    _diag("calling webview.start (thread) gui=edgechromium")
+                    webview.start(on_start, debug=debug, gui='edgechromium')
+                except Exception as e:
+                    _diag(f"webview.start raised (thread): {e!r}")
+                finally:
+                    self._stop_asset_server()
+
+            t = threading.Thread(target=_run, daemon=True)
+            t.start()
+
+    def close(self):
+        """关闭并销毁 Web 窗口。"""
+        if self.webview_instance:
+            self.webview_instance.destroy()

ascript/windows/utils/node_utils.py

@@ -0,0 +1,20 @@
+"""窗口节点工具模块。
+
+提供与 Windows 窗口枚举和节点操作相关的实用函数。
+"""
+
+import win32gui
+
+
+def get_all_window(model: int):
+    """获取所有窗口句柄。
+
+    根据指定的模式枚举系统中所有顶层窗口。
+
+    Args:
+        model: 枚举模式，用于控制窗口枚举的方式。
+
+    Returns:
+        None: 当前实现未返回值（EnumWindows 缺少回调函数）。
+    """
+    win32gui.EnumWindows()