✨ 添加获取会话状态内容的依赖注入 (#23)

kirami/depends.py

@@ -54,6 +54,8 @@
     extract_plain_text,
 )
 
+T = TypeVar("T")
+
 
 class DependsInner(BaseDependsInner):
     def __call__(
@@ -292,13 +294,20 @@ async def client_session() -> AsyncGenerator[AsyncClient, None]:
 """网络连接会话对象"""
 
 
+def useStateData(key: str | None = None) -> Any:
+    """提取会话状态内容"""
+    return ArgInner(key, type="state")  # type: ignore
+
+
+StateData = Annotated[T, useStateData()]
+"""会话状态内容"""
+
+
 def useArgot(key: str | None = None) -> Any:
     """提取暗语的内容"""
     return ArgInner(key, type="argot")  # type: ignore
 
 
-T = TypeVar("T")
-
 Argot = Annotated[T, useArgot()]
 """暗语内容"""
 

kirami/patch.py

@@ -202,8 +202,10 @@ def get_plaintext(self: Event) -> str:
 
 
 async def _solve(self: ArgParam, matcher: Matcher, **_kwargs: Any) -> Any:
-    """支持 Argot"""
+    """支持 State、Argot 内容提取"""
     key: str = self.extra["key"]
+    if self.extra["type"] == "state":
+        return matcher.state.get(key)
     if self.extra["type"] == "argot":
         return matcher.get_argot(key)
     message = matcher.get_arg(key)