virid-std 0.1.0__tar.gz

virid_std-0.1.0/.gitignore

@@ -0,0 +1,4 @@
+# --- 依赖与包管理器 ---
+.venv/
+uv.lock
+__pycache__/

virid_std-0.1.0/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: virid-std
+Version: 0.1.0
+Requires-Python: >=3.10
+Requires-Dist: core

virid_std-0.1.0/pyproject.toml

@@ -0,0 +1,18 @@
+[project]
+name = "virid-std"
+version = "0.1.0"
+requires-python = ">=3.10"
+dependencies = [
+    "core",
+]
+
+[tool.uv.sources]
+core = { workspace = true }
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/virid"]
+layout = "src"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"

virid_std-0.1.0/src/virid/std/__init__.py

@@ -0,0 +1,8 @@
+"""
+Copyright (c) 2026-present Ailrid.
+Licensed under the Apache License, Version 2.0.
+Project: Virid
+"""
+
+from .plugin import StdPlugin
+from .utils import execute_group, next_tick, execute_block

virid_std-0.1.0/src/virid/std/plugin.py

@@ -0,0 +1,23 @@
+"""
+Copyright (c) 2026-present Ailrid.
+Licensed under the Apache License, Version 2.0.
+Project: Virid
+"""
+
+from virid.core import ViridApp
+from .utils import activate_utils
+
+
+def activate_plugin(app: ViridApp):
+    activate_utils(app)
+
+
+class Plugin:
+    name = "std"
+
+    # 将返回值改为 None，以契合 Protocol 的声明
+    def install(self, app: ViridApp, options: None) -> None:
+        activate_plugin(app)
+
+
+StdPlugin = Plugin()

virid_std-0.1.0/src/virid/std/utils/__init__.py

@@ -0,0 +1,14 @@
+"""
+Copyright (c) 2026-present Ailrid.
+Licensed under the Apache License, Version 2.0.
+Project: Virid
+"""
+
+from .execute_group import activate_group_messages, execute_group, execute_block
+from .nexttick import activate_next_tick, next_tick
+from virid.core import ViridApp
+
+
+def activate_utils(app: ViridApp):
+    activate_group_messages(app)
+    activate_next_tick(app)

virid_std-0.1.0/src/virid/std/utils/execute_group.py

@@ -0,0 +1,136 @@
+"""
+Copyright (c) 2026-present Ailrid.
+Licensed under the Apache License, Version 2.0.
+Project: Virid
+"""
+
+from typing import Any, Callable
+from virid.core import EventMessage, ViridApp, MessageWriter, ExecuteHookContext
+from contextlib import contextmanager
+
+# 每个 key 中缓存的执行组队列及状态映射
+execute_group_map: dict[str, list[dict[str, Any]]] = {}
+message_key_map: dict[int, str] = {}  # 键类型改为 int，用于存放 id(message)
+key_message_map: dict[str, list[EventMessage]] = {}
+call_back_map: dict[str, Callable[[bool], None]] = {}
+
+
+def clear_group(group_id: str) -> None:
+    for message in key_message_map[group_id]:
+        message_key_map.pop(id(message), None)  # 使用 id(message) 释放
+    key_message_map.pop(group_id, None)
+    execute_group_map.pop(group_id, None)
+    call_back_map.pop(group_id, None)
+
+
+def after_execute_hook(
+    message: EventMessage,
+    _hook_context: ExecuteHookContext,
+    success: bool,
+) -> None:
+    key = message_key_map.get(id(message))  # 使用 id(message) 获取
+    if key:
+        execute_group = execute_group_map.get(key)
+        if execute_group is None:
+            return
+
+        if success:
+            # 只有当前任务执行成功，才继续下一个
+            if len(execute_group) > 0:
+                context = execute_group.pop(0)
+                resolve = context["resolve"]
+                resolve()
+            # 全部执行完成
+            if len(execute_group) == 0:
+                callback = call_back_map.get(key)
+                if callback:
+                    callback(True)
+                clear_group(key)
+        else:
+            callback = call_back_map.get(key)
+            if callback:
+                callback(False)
+            clear_group(key)
+            # 如果出错了，直接取消执行队列
+            MessageWriter.error(
+                Exception(
+                    f"[ExecuteGroup] Queue Execution Failed: Due to an error in the System execution triggered by {message.__class__.__name__}, the message group '{key}' has been cancelled"
+                )
+            )
+
+
+def execute_group(
+    messages: list[EventMessage],
+    group_id: str = "default",
+    callback: Callable[[bool], None] | None = None,
+) -> None:
+    # 修复：原代码为 id in execute_group_map，会导致判定错误
+    if group_id in execute_group_map:
+        MessageWriter.error(
+            Exception(
+                f"[ExecuteGroup] Unavailable ID: The id '{group_id}' not yet executed"
+            )
+        )
+        if callback:
+            callback(False)
+        return
+
+    # 注册这一个执行组
+    execute_chain: list[dict[str, Any]] = []
+    key_message_map[group_id] = messages
+
+    for index, message in enumerate(messages):
+        if index == len(messages) - 1:
+            resolve = lambda: None
+        else:
+            # 使用默认参数绑定 next_msg，防止 Python 闭包延迟绑定导致总是拿到最后一条消息
+            next_msg = messages[index + 1]
+            resolve = lambda m=next_msg: MessageWriter.write(m)
+
+        execute_chain.append({"message": message, "resolve": resolve})
+        message_key_map[id(message)] = group_id  # 使用 id(message) 作为键
+
+    if callback:
+        call_back_map[group_id] = callback
+
+    execute_group_map[group_id] = execute_chain
+
+    # 立刻触发第一个
+    MessageWriter.write(messages[0])
+
+
+@contextmanager
+def execute_block(
+    group_id: str = "default", callback: Callable[[bool], None] | None = None
+):
+    #  创建一个临时篮子，用来装 with 期间产生的所有消息实例
+    captured_messages: list[EventMessage] = []
+
+    # 备份原本的全局 MessageWriter.write 方法
+    original_write = MessageWriter.write
+
+    # 定义一个拦截函数
+    def mock_write(message: Any) -> None:
+        if isinstance(message, EventMessage):
+            # 如果是事件消息，拦截下来，存进篮子，先不发送
+            captured_messages.append(message)
+        else:
+            # 如果是其他不归群组管的消息（比如 InfoMessage 等），让它走原渠道正常发
+            original_write(message)
+
+    # 把原本的 write 换成拦截器
+    MessageWriter.write = mock_write # type: ignore
+
+    try:
+        yield  # 此时执行 with 块内部的代码
+    finally:
+        # 离开 with 无论里面是否报错，雷打不动地把原本的 write 恢复回去
+        # 这确保了黑魔法不会污染全局
+        MessageWriter.write = original_write
+
+    if captured_messages:
+        execute_group(captured_messages, group_id=group_id, callback=callback)
+
+
+def activate_group_messages(app: ViridApp) -> None:
+    app.on_after_execute(EventMessage, after_execute_hook)

virid_std-0.1.0/src/virid/std/utils/nexttick.py

@@ -0,0 +1,30 @@
+"""
+Copyright (c) 2026-present Ailrid.
+Licensed under the Apache License, Version 2.0.
+Project: Virid
+"""
+
+from typing import Callable, Any
+from virid.core import ViridApp
+
+# 一个简单的 nextTick 任务队列
+tick_task_queue: list[Callable[[], None]] = []
+
+
+def after_tick_hook(context: Any) -> None:
+    global tick_task_queue
+    if len(tick_task_queue) == 0:
+        return
+
+    # 严格遵循你 TS 原版的双缓冲/清空逻辑
+    for task in tick_task_queue:
+        task()
+    tick_task_queue = []
+
+
+def next_tick(task: Callable[[], None]) -> None:
+    tick_task_queue.append(task)
+
+
+def activate_next_tick(app: ViridApp) -> None:
+    app.on_after_tick(after_tick_hook)