ascript 1.3.27__py3-none-win_amd64.whl

ascript/windows/ui/app_store_window.py

@@ -0,0 +1,272 @@
+import json
+import time
+import os
+from .web_window import WebWindow
+from .. import license
+from ..system import KeyValue, R, EnvManager
+
+
+class AppStoreWindow(WebWindow):
+    """应用商店窗口类。
+
+    提供应用搜索、激活码验证、应用下载与启动等功能的 Web 窗口界面。
+    继承自 WebWindow，通过 HTML 页面与 Python 后端进行交互。
+    """
+
+    def __init__(self, default_id: str = None, on_start=None, env_path: str = None):  # 增加默认 ID 参数
+        """初始化应用商店窗口。
+
+        Args:
+            default_id: 默认的应用 ID，用于自动填充搜索框。
+            on_start: 应用启动时的回调函数，接收 (app_info, license_data) 两个参数。
+            env_path: 虚拟环境的根目录路径，传递给 EnvManager。
+        """
+        self.default_id = default_id
+        self.on_start = on_start  # 存储回调
+        self.app_info = None
+        self.license_data = None
+        self.target_path = None
+        self.env_manager = EnvManager(root_path=env_path)
+
+        _root = R.internal("windows", "tools", "web")
+        _filepath = R.internal("windows", "tools", "web", "app_store_window.html")
+
+        super().__init__(_filepath, project_root=_root)
+
+        self.expose(self.get_app_info)
+        self.expose(self.activate_key)
+        self.expose(self.start_app)
+        self.expose(self.get_history)
+        self.expose(self.get_cached_key)
+        # 暴露一个获取初始 ID 的方法
+        self.expose(self.get_init_config)
+        self.expose(self.check_status)
+        self.expose(self.stop_app)
+        self.expose(self.clear_env)
+        self.expose(self.download_app)
+
+    def get_init_config(self):
+        """获取初始配置信息。
+
+        Returns:
+            dict: 包含 default_id 的字典，前端用于初始化界面。
+        """
+        return {"default_id": self.default_id}
+
+    def get_app_info(self, app_id):
+        """根据应用 ID 搜索并获取应用信息。
+
+        Args:
+            app_id: 应用的唯一标识符。
+
+        Returns:
+            dict: 成功时返回 {"code": 1, "data": app_info}；
+                  失败时返回 {"code": 0, "msg": 错误信息}。
+        """
+        app_id = str(app_id).strip()  # 强制转字符串，解决类型报错
+        res = license.achttp.get_apps(app_id)
+        if res and len(res) > 0:
+            self.app_info = res[0]
+            return {"code": 1, "data": self.app_info}
+        return {"code": 0, "msg": f"未能找到 ID 为 '{app_id}' 的应用，请检查输入是否正确。"}
+
+    def get_cached_key(self, app_id):
+        """按应用 ID 获取本地缓存的激活码。
+
+        Args:
+            app_id: 应用的唯一标识符。
+
+        Returns:
+            str: 缓存的激活码字符串，若无缓存则返回空字符串。
+        """
+        return KeyValue.get(f"lic_cache_{str(app_id)}", "")
+
+    def activate_key(self, app_id, key: str):
+        """使用激活码对指定应用进行授权激活。
+
+        Args:
+            app_id: 应用的唯一标识符。
+            key: 用户输入的激活码。
+
+        Returns:
+            dict: 服务端返回的激活结果，成功时 code 为 1 并包含 data 字段。
+        """
+        app_id = str(app_id)
+        res = license.achttp.get_card(app_id, key)
+        if res and res.get('code') == 1:
+            self.license_data = res.get('data')
+            # 区分 ID 存储
+            KeyValue.save(f"lic_cache_{app_id}", key)
+        else:
+            self.license_data = None
+        return res
+
+    def get_history(self):
+        """获取应用运行历史记录列表。
+
+        Returns:
+            list: 运行记录列表，每项包含 id 和 name 字段；异常时返回空列表。
+        """
+        try:
+            history_str = KeyValue.get("app_run_history", "[]")
+            return json.loads(history_str)
+        except:
+            return []
+
+    # 在 AppStoreWindow 类中新增/修改以下方法
+    def download_app(self, app_info):
+        """下载应用文件到本地存储目录。
+
+        根据应用信息中的文件路径下载 whl 等文件，支持断点检测（已存在则跳过），
+        并在下载过程中通过 JS 回调更新进度条。
+
+        Args:
+            app_info: 应用信息字典，需包含 filePath（下载地址）、fileMd5（文件校验值）、id（应用 ID）。
+
+        Returns:
+            dict: 成功时返回 {"code": 1, "msg": ..., "path": 本地路径}；
+                  失败时返回 {"code": 0, "msg": 错误信息}。
+        """
+        url = app_info['filePath']
+        file_md5 = app_info['fileMd5']
+        user_id = str(app_info["id"])
+
+        # 从 URL 中提取原始文件名（例如: myapp-1.0.whl）
+        # 如果 URL 后面带参数，可以用 url.split('?')[0]
+        remote_filename = os.path.basename(url.split('?')[0])
+        # if not remote_filename.endswith('.whl'):
+        #     remote_filename = f"{file_md5}.whl"
+
+        # 1. 获取目标文件夹路径：module/user_id/file_md5/
+        # 这里的 get_module_storage_path 会返回完整的：.../module/1085/md5_string/xxx.whl
+        self.target_path = os.path.join(self.env_manager.get_model_dir(user_id), file_md5, remote_filename)
+
+        # 确保文件夹存在（即 module/user_id/file_md5/ 这一层）
+        storage_dir = os.path.dirname(self.target_path)
+        if not os.path.exists(storage_dir):
+            os.makedirs(storage_dir, exist_ok=True)
+
+        # 2. 如果文件已存在，直接返回
+        if os.path.exists(self.target_path):
+            return {"code": 1, "msg": "已存在", "path": self.target_path}
+
+        # 3. 开始执行下载
+        try:
+            import requests
+            headers = {'User-Agent': 'Mozilla/5.0'}
+            response = requests.get(url, stream=True, timeout=30, headers=headers)
+            response.raise_for_status()
+
+            total_size = int(response.headers.get('content-length', 0))
+            downloaded = 0
+
+            with open(self.target_path, "wb") as f:
+                for chunk in response.iter_content(chunk_size=1024 * 64):
+                    if chunk:
+                        f.write(chunk)
+                        downloaded += len(chunk)
+                        if total_size > 0:
+                            percent = int(downloaded * 100 / total_size)
+                            # 确保 webview_instance 存在
+                            if hasattr(self, 'webview_instance'):
+                                self.webview_instance.evaluate_js(f"updateDownloadProgress({percent})")
+
+            return {"code": 1, "msg": "下载完成", "path": self.target_path}
+        except Exception as e:
+            if os.path.exists(self.target_path):
+                try:
+                    os.remove(self.target_path)
+                except:
+                    pass
+            return {"code": 0, "msg": f"下载失败: {str(e)}"}
+
+    def start_app(self, app_info: dict):
+        """启动应用程序。
+
+        保存运行记录并调用外部回调函数启动应用引擎。
+
+        Args:
+            app_info: 应用信息字典，需包含 id 和 name 字段。
+
+        Returns:
+            dict: 返回 {"code": 1, "msg": "启动成功"}。
+        """
+        # 1. 保存/更新运行记录
+        history = self.get_history()
+        new_item = {"id": str(app_info['id']), "name": app_info['name']}
+
+        # 去重：如果已存在则先删除旧记录
+        history = [i for i in history if str(i.get('id')) != new_item['id']]
+        # 插入到开头
+        history.insert(0, new_item)
+        # 限制数量为 10 条
+        KeyValue.save("app_run_history", json.dumps(history[:10]))
+
+        # 2. 执行回调逻辑
+        if self.on_start:
+            # 将 app_info 和当前激活的 license_data 一并传出
+            # 这样外部调用者就能拿到所有必要信息去启动引擎
+            started = self.on_start(app_info, self.license_data)
+            if started is False:
+                self._return_value = False
+                return {
+                    "code": 0,
+                    "msg": "环境部署或程序启动失败，请查看错误弹窗及部署日志。"
+                }
+
+        # 2. 这里的逻辑可以根据您的业务决定是否关闭窗口
+        self._return_value = True
+        # self.close() # 如果需要启动后关闭商城窗口，请取消注释
+        return {"code": 1, "msg": "启动成功"}
+
+    def check_status(self, app_id):
+        """检查指定应用是否正在运行。
+
+        供前端轮询调用，判断应用进程的运行状态。
+
+        Args:
+            app_id: 应用的唯一标识符。
+
+        Returns:
+            dict: 返回 {"running": bool}，表示应用是否在运行中。
+        """
+        # 调用之前我们完善的 EnvManager.is_running
+        # 假设实例名为 self.env_manager
+        is_run = self.env_manager.is_running(str(app_id))
+        return {"running": is_run}
+
+    def stop_app(self, app_id):
+        """强行停止指定应用的运行。
+
+        Args:
+            app_id: 应用的唯一标识符。
+
+        Returns:
+            dict: 返回 {"code": 1, "msg": "已发送停止指令"}。
+        """
+        self.env_manager.stop_app(str(app_id))
+        return {"code": 1, "msg": "已发送停止指令"}
+
+    def clear_env(self, app_id):
+        """清理指定应用的虚拟环境目录。
+
+        仅在应用未运行时允许清理，运行中会返回错误提示。
+
+        Args:
+            app_id: 应用的唯一标识符。
+
+        Returns:
+            dict: 成功时返回 {"code": 1, "msg": "环境已重置"}；
+                  失败时返回 {"code": 0, "msg": 错误信息}。
+        """
+        # 只有在未运行的情况下才允许清理
+        if self.env_manager.is_running(str(app_id)):
+            return {"code": 0, "msg": "程序运行中，无法清理环境"}
+
+        env_dir = self.env_manager.get_env_dir(str(app_id))
+        import shutil
+        try:
+            shutil.rmtree(env_dir, ignore_errors=True)
+            return {"code": 1, "msg": "环境已重置"}
+        except Exception as e:
+            return {"code": 0, "msg": str(e)}

ascript/windows/ui/draw_window.py

@@ -0,0 +1,231 @@
+import ctypes
+from ctypes import wintypes
+import threading
+import time
+import struct
+
+# --- 系统环境检查 ---
+IS_64BIT = struct.calcsize("P") == 8
+PTR_TYPE = ctypes.c_int64 if IS_64BIT else ctypes.c_int32
+
+
+class DrawWindow:
+    """屏幕绘制覆盖层窗口。
+
+    使用 Win32 API 创建一个全屏透明置顶窗口，支持在屏幕上绘制矩形、线条和文本。
+    绘制内容不会被截图捕获（使用 WDA_EXCLUDEFROMCAPTURE）。
+    所有方法均为静态方法，全局共享一个绘制窗口实例。
+    """
+
+    _hwnd = None
+    _thread = None
+    _shapes = []  # 存储任务: {'type': 'rect', 'data': (x, y, w, h, color, thickness)}
+    _lock = threading.Lock()
+    _wnd_proc_ptr = None
+
+    # 窗口过程回调定义
+    WNDPROC = ctypes.WINFUNCTYPE(PTR_TYPE, wintypes.HWND, ctypes.c_uint, wintypes.WPARAM, wintypes.LPARAM)
+
+    @staticmethod
+    def _create_window():
+        """创建全屏透明置顶的绘制覆盖层窗口。
+
+        注册自定义窗口类，创建带有 WS_EX_LAYERED、WS_EX_TRANSPARENT、WS_EX_TOPMOST
+        扩展样式的窗口，并进入消息循环。该方法在后台线程中运行。
+        """
+        user32 = ctypes.windll.user32
+        gdi32 = ctypes.windll.gdi32
+        kernel32 = ctypes.windll.kernel32
+
+        # 1. 开启 DPI 感知 (防止 Win10/Win11 缩放导致坐标偏移)
+        try:
+            ctypes.windll.shcore.SetProcessDpiAwareness(1)
+        except Exception:
+            user32.SetProcessDPIAware()
+
+        clsname = f"DrawWindow_{int(time.time())}"
+        hinst = kernel32.GetModuleHandleW(None)
+
+        def wnd_proc(hwnd, msg, wparam, lparam):
+            if msg == 0x000F:  # WM_PAINT
+                ps = ctypes.create_string_buffer(64)
+                hdc = user32.BeginPaint(hwnd, ps)
+
+                with DrawWindow._lock:
+                    for item in DrawWindow._shapes:
+                        t, d = item['type'], item['data']
+
+                        if t in ('rect', 'line'):
+                            color, thick = d[4], d[5]
+                            pen = gdi32.CreatePen(0, thick, color)
+                            old_pen = gdi32.SelectObject(hdc, pen)
+                            if t == 'rect':
+                                brush = gdi32.GetStockObject(5)  # NULL_BRUSH
+                                old_brush = gdi32.SelectObject(hdc, brush)
+                                gdi32.Rectangle(hdc, d[0], d[1], d[0] + d[2], d[1] + d[3])
+                                gdi32.SelectObject(hdc, old_brush)
+                            else:  # line
+                                gdi32.MoveToEx(hdc, d[0], d[1], None)
+                                gdi32.LineTo(hdc, d[2], d[3])
+                            gdi32.SelectObject(hdc, old_pen)
+                            gdi32.DeleteObject(pen)
+
+                        elif t == 'text':
+                            x, y, text, color, size = d
+                            font = gdi32.CreateFontW(size, 0, 0, 0, 400, 0, 0, 0, 1, 0, 0, 0, 0, "Microsoft YaHei")
+                            old_font = gdi32.SelectObject(hdc, font)
+                            gdi32.SetTextColor(hdc, color)
+                            gdi32.SetBkMode(hdc, 1)  # TRANSPARENT
+                            gdi32.TextOutW(hdc, x, y, text, len(text))
+                            gdi32.SelectObject(hdc, old_font)
+                            gdi32.DeleteObject(font)
+
+                user32.EndPaint(hwnd, ps)
+                return 0
+
+            if msg == 0x0002:  # WM_DESTROY
+                user32.PostQuitMessage(0)
+                return 0
+
+            # 安全调用默认窗口过程
+            user32.DefWindowProcW.argtypes = [wintypes.HWND, ctypes.c_uint, wintypes.WPARAM, wintypes.LPARAM]
+            user32.DefWindowProcW.restype = PTR_TYPE
+            return user32.DefWindowProcW(hwnd, msg, wparam, lparam)
+
+        DrawWindow._wnd_proc_ptr = DrawWindow.WNDPROC(wnd_proc)
+
+        from ctypes import Structure, c_uint, c_int
+        class WNDCLASS(Structure):
+            _fields_ = [("style", c_uint), ("lpfnWndProc", DrawWindow.WNDPROC), ("cbClsExtra", c_int),
+                        ("cbWndExtra", c_int), ("hInstance", wintypes.HINSTANCE), ("hIcon", wintypes.HICON),
+                        ("hCursor", wintypes.HANDLE), ("hbrBackground", wintypes.HANDLE),
+                        ("lpszMenuName", wintypes.LPCWSTR), ("lpszClassName", wintypes.LPCWSTR)]
+
+        wc = WNDCLASS()
+        wc.lpfnWndProc = DrawWindow._wnd_proc_ptr
+        wc.hInstance = hinst
+        wc.lpszClassName = clsname
+        wc.hbrBackground = gdi32.GetStockObject(4)  # BLACK_BRUSH
+        wc.hCursor = user32.LoadCursorW(0, 32512)  # IDC_ARROW
+        user32.RegisterClassW(ctypes.byref(wc))
+
+        # 获取当前屏幕物理分辨率
+        sw = user32.GetSystemMetrics(0)
+        sh = user32.GetSystemMetrics(1)
+
+        # 创建全屏透明置顶窗口
+        # WS_EX_LAYERED(0x80000) | WS_EX_TRANSPARENT(0x20) | WS_EX_TOPMOST(0x08)
+        DrawWindow._hwnd = user32.CreateWindowExW(
+            0x00080000 | 0x00000020 | 0x00000008,
+            clsname, "DrawOverlay", 0x80000000,
+            0, 0, sw, sh, 0, 0, hinst, 0
+        )
+
+        # 防截图核心 API
+        user32.SetWindowDisplayAffinity(DrawWindow._hwnd, 0x11)  # WDA_EXCLUDEFROMCAPTURE
+        # 黑色透明
+        user32.SetLayeredWindowAttributes(DrawWindow._hwnd, 0, 255, 0x1)
+        user32.ShowWindow(DrawWindow._hwnd, 5)
+
+        msg = wintypes.MSG()
+        while user32.GetMessageW(ctypes.byref(msg), 0, 0, 0) != 0:
+            user32.TranslateMessage(ctypes.byref(msg))
+            user32.DispatchMessageW(ctypes.byref(msg))
+
+    @staticmethod
+    def _init():
+        """初始化绘制窗口。
+
+        如果窗口线程尚未启动，则创建后台线程并等待窗口句柄就绪。
+        """
+        if DrawWindow._thread is None:
+            DrawWindow._thread = threading.Thread(target=DrawWindow._create_window, daemon=True)
+            DrawWindow._thread.start()
+            while DrawWindow._hwnd is None: time.sleep(0.05)
+
+    @staticmethod
+    def draw_rect(x, y, w, h, color=0x0000FF, thickness=2):
+        """在屏幕上绘制一个空心矩形。
+
+        Args:
+            x: 矩形左上角的 X 坐标（像素）。
+            y: 矩形左上角的 Y 坐标（像素）。
+            w: 矩形的宽度（像素）。
+            h: 矩形的高度（像素）。
+            color: 颜色值，BGR 格式，默认为红色 (0x0000FF)。
+            thickness: 边框线条粗细（像素），默认为 2。
+        """
+        DrawWindow._init()
+        with DrawWindow._lock:
+            DrawWindow._shapes.append({'type': 'rect', 'data': (x, y, w, h, color, thickness)})
+        ctypes.windll.user32.InvalidateRect(DrawWindow._hwnd, None, True)
+
+    @staticmethod
+    def draw_line(x1, y1, x2, y2, color=0x0000FF, thickness=2):
+        """在屏幕上绘制一条直线。
+
+        Args:
+            x1: 起点 X 坐标（像素）。
+            y1: 起点 Y 坐标（像素）。
+            x2: 终点 X 坐标（像素）。
+            y2: 终点 Y 坐标（像素）。
+            color: 颜色值，BGR 格式，默认为红色 (0x0000FF)。
+            thickness: 线条粗细（像素），默认为 2。
+        """
+        DrawWindow._init()
+        with DrawWindow._lock:
+            DrawWindow._shapes.append({'type': 'line', 'data': (x1, y1, x2, y2, color, thickness)})
+        ctypes.windll.user32.InvalidateRect(DrawWindow._hwnd, None, True)
+
+    @staticmethod
+    def draw_text(x, y, text, color=0x0000FF, font_size=20):
+        """在屏幕上绘制文本。
+
+        Args:
+            x: 文本起始位置的 X 坐标（像素）。
+            y: 文本起始位置的 Y 坐标（像素）。
+            text: 要绘制的文本内容。
+            color: 颜色值，BGR 格式，默认为红色 (0x0000FF)。
+            font_size: 字体大小（像素），默认为 20。
+        """
+        DrawWindow._init()
+        with DrawWindow._lock:
+            DrawWindow._shapes.append({'type': 'text', 'data': (x, y, text, color, font_size)})
+        ctypes.windll.user32.InvalidateRect(DrawWindow._hwnd, None, True)
+
+    @staticmethod
+    def clear():
+        """清除屏幕上所有已绘制的图形。"""
+        if DrawWindow._hwnd:
+            with DrawWindow._lock: DrawWindow._shapes.clear()
+            ctypes.windll.user32.InvalidateRect(DrawWindow._hwnd, None, True)
+
+
+# --- 辅助工具：RGB 转 BGR ---
+def RGB(r, g, b):
+    """将 RGB 颜色值转换为 BGR 格式。
+
+    DrawWindow 使用 BGR 颜色格式，此函数将常见的 RGB 值转换为对应的 BGR 值。
+
+    Args:
+        r: 红色通道值 (0-255)。
+        g: 绿色通道值 (0-255)。
+        b: 蓝色通道值 (0-255)。
+
+    Returns:
+        int: BGR 格式的颜色整数值。
+    """
+    return (b << 16) | (g << 8) | r
+
+
+# --- 示例使用 ---
+if __name__ == "__main__":
+    # 画一个三角形（三条线）
+    DrawWindow.draw_line(500, 200, 300, 500, thickness=3)  # 蓝色
+    DrawWindow.draw_line(300, 500, 700, 500, color=0xFF0000, thickness=3)
+    DrawWindow.draw_line(700, 500, 500, 200, color=0xFF0000, thickness=3)
+
+    DrawWindow.draw_text(450, 520, "Triangle Base", color=0xFFFFFF, font_size=25)
+
+    time.sleep(10)
+    DrawWindow.clear()