Undefined-bot 2.1.0__py3-none-any.whl

Undefined/skills/tools/get_messages_by_time/handler.py

@@ -0,0 +1,128 @@
+from typing import Any, Dict
+from datetime import datetime
+
+
+def _resolve_chat_id(chat_id: str, msg_type: str, history_manager: Any) -> str:
+    """将群名/用户名转换为对应的 ID
+
+    参数:
+        chat_id: 群名、用户名或群号/用户ID
+        msg_type: "group" 或 "private"
+        history_manager: 历史记录管理器实例
+
+    返回:
+        解析后的群号或用户ID
+    """
+    if chat_id.isdigit():
+        return chat_id
+
+    if not history_manager:
+        return chat_id
+
+    try:
+        # 使用生成器表达式简化查找逻辑
+        if msg_type == "group":
+            histories = history_manager._message_history.items()
+        elif msg_type == "private":
+            histories = history_manager._private_message_history.items()
+        else:
+            return chat_id
+
+        # 查找第一个匹配的 ID
+        for id_val, messages in histories:
+            if messages and messages[0].get("chat_name") == chat_id:
+                return str(id_val)
+
+    except Exception:
+        # 忽略任何查找过程中的错误，回退到返回原始 chat_id
+        pass
+
+    return chat_id
+
+
+def _filter_messages(
+    messages: list[Dict[str, Any]], start_dt: datetime, end_dt: datetime
+) -> list[Dict[str, Any]]:
+    """根据时间范围过滤消息
+
+    参数:
+        messages: 原始消息列表
+        start_dt: 开始时间
+        end_dt: 结束时间
+
+    返回:
+        过滤后的消息列表
+    """
+    filtered = []
+    for msg in messages:
+        timestamp = msg.get("timestamp", "")
+        if not timestamp:
+            continue
+        try:
+            msg_dt = datetime.strptime(timestamp, "%Y-%m-%d %H:%M:%S")
+            if start_dt <= msg_dt <= end_dt:
+                filtered.append(msg)
+        except ValueError:
+            continue
+    return filtered
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    chat_id = args.get("chat_id")
+    msg_type = args.get("type")
+    start_time = args.get("start_time", "")
+    end_time = args.get("end_time", "")
+
+    if not chat_id or not msg_type:
+        return "chat_id 和 type 参数不能为空"
+    if not start_time or not end_time:
+        return "start_time 和 end_time 参数不能为空"
+
+    try:
+        start_dt = datetime.strptime(start_time, "%Y-%m-%d %H:%M:%S")
+        end_dt = datetime.strptime(end_time, "%Y-%m-%d %H:%M:%S")
+    except ValueError:
+        return "时间格式错误，请使用格式：YYYY-MM-DD HH:MM:SS"
+
+    if start_dt > end_dt:
+        return "开始时间不能晚于结束时间"
+
+    get_recent_messages_callback = context.get("get_recent_messages_callback")
+    history_manager = context.get("history_manager")
+
+    resolved_chat_id = _resolve_chat_id(chat_id, msg_type, history_manager)
+
+    if get_recent_messages_callback:
+        messages = await get_recent_messages_callback(
+            resolved_chat_id, msg_type, 0, 10000
+        )
+
+        # 使用提取的函数进行过滤
+        filtered_messages = _filter_messages(messages, start_dt, end_dt)
+
+        formatted = []
+        for msg in filtered_messages:
+            msg_type_val = msg.get("type", "group")
+            sender_name = msg.get("display_name", "未知用户")
+            sender_id = msg.get("user_id", "")
+            chat_name = msg.get("chat_name", "未知群聊")
+            timestamp_str = msg.get("timestamp", "")
+            text = msg.get("message", "")
+
+            if msg_type_val == "group":
+                # 确保群名以"群"结尾
+                location = chat_name if chat_name.endswith("群") else f"{chat_name}群"
+            else:
+                location = "私聊"
+
+            # 格式：XML 标准化
+            formatted.append(f"""<message sender="{sender_name}" sender_id="{sender_id}" location="{location}" time="{timestamp_str}">
+<content>{text}</content>
+</message>""")
+
+        if formatted:
+            return f"找到 {len(formatted)} 条消息：\n" + "\n---\n".join(formatted)
+        else:
+            return f"在 {start_time} 到 {end_time} 之间没有找到消息"
+    else:
+        return "获取消息回调未设置"

Undefined/skills/tools/get_picture/config.json

@@ -0,0 +1,45 @@
+{
+    "type": "function",
+    "function": {
+        "name": "get_picture",
+        "description": "获取指定数量的指定类型的图片并发送到群聊或指定私聊。支持白丝、黑丝、头像、JK、二次元、小姐姐、壁纸、原神、历史上的今天、4K图片、美腿十一种类型。二次元类型支持选择手机端或PC端。4K图片支持选择二次元或风景。默认获取二次元图片，默认使用PC端。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "message_type": {
+                    "type": "string",
+                    "description": "消息类型",
+                    "enum": ["group", "private"]
+                },
+                "target_id": {
+                    "type": "integer",
+                    "description": "目标 ID（群聊为群号，私聊为用户 QQ 号）"
+                },
+                "picture_type": {
+                    "type": "string",
+                    "description": "图片类型",
+                    "enum": ["baisi", "heisi", "head", "jk", "acg", "meinvpic", "wallpaper", "ys", "historypic", "random4kPic"],
+                    "default": "acg"
+                },
+                "count": {
+                    "type": "integer",
+                    "description": "获取图片的数量",
+                    "default": 1
+                },
+                "device": {
+                    "type": "string",
+                    "description": "设备类型（仅 acg 类型支持）",
+                    "enum": ["pc", "wap"],
+                    "default": "pc"
+                },
+                "fourk_type": {
+                    "type": "string",
+                    "description": "4K图片类型（仅 random4kPic 类型支持）",
+                    "enum": ["acg", "wallpaper"],
+                    "default": "acg"
+                }
+            },
+            "required": ["message_type", "target_id"]
+        }
+    }
+}

Undefined/skills/tools/get_picture/handler.py

@@ -0,0 +1,191 @@
+from typing import Any, Dict
+import logging
+import httpx
+import asyncio
+import uuid
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+# API 地址映射
+API_URLS = {
+    "baisi": "https://v2.xxapi.cn/api/baisi",
+    "heisi": "https://v2.xxapi.cn/api/heisi",
+    "head": "https://v2.xxapi.cn/api/head",
+    "jk": "https://v2.xxapi.cn/api/jk",
+    "acg": "https://v2.xxapi.cn/api/randomAcgPic",
+    "meinvpic": "https://v2.xxapi.cn/api/meinvpic",
+    "wallpaper": "https://v2.xxapi.cn/api/wallpaper",
+    "ys": "https://v2.xxapi.cn/api/ys",
+    "historypic": "https://v2.xxapi.cn/api/historypic",
+    "random4kPic": "https://v2.xxapi.cn/api/random4kPic",
+}
+
+# 图片类型名称映射
+TYPE_NAMES = {
+    "baisi": "白丝",
+    "heisi": "黑丝",
+    "head": "头像",
+    "jk": "JK",
+    "acg": "二次元",
+    "meinvpic": "小姐姐",
+    "wallpaper": "壁纸",
+    "ys": "原神",
+    "historypic": "历史上的今天",
+    "random4kPic": "4K图片",
+    "meitui": "美腿",
+}
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    message_type = args.get("message_type")
+    target_id = args.get("target_id")
+    picture_type = args.get("picture_type", "acg")
+    count = args.get("count", 1)
+    device = args.get("device", "pc")
+    fourk_type = args.get("fourk_type", "acg")
+    
+    # 参数验证
+    if not message_type:
+        return "❌ 消息类型不能为空"
+    if message_type not in ["group", "private"]:
+        return "❌ 消息类型必须是 group（群聊）或 private（私聊）"
+    if not target_id:
+        return "❌ 目标 ID 不能为空"
+    if not isinstance(target_id, int):
+        return "❌ 目标 ID 必须是整数"
+    if picture_type not in API_URLS:
+        return f"❌ 不支持的图片类型: {picture_type}\n支持的类型: {', '.join(TYPE_NAMES.values())}"
+    if not isinstance(count, int):
+        return "❌ 图片数量必须是整数"
+    if count < 1 or count > 10:
+        return "❌ 图片数量必须在 1-10 之间"
+    if picture_type == "acg" and device not in ["pc", "wap"]:
+        return "❌ 设备类型必须是 pc（电脑端）或 wap（手机端）"
+    if picture_type == "random4kPic" and fourk_type not in ["acg", "wallpaper"]:
+        return "❌ 4K图片类型必须是 acg（二次元）或 wallpaper（风景）"
+    
+    # 获取发送图片回调
+    send_image_callback = context.get("send_image_callback")
+    if not send_image_callback:
+        return "发送图片回调未设置"
+    
+    # 构造请求参数
+    params: Dict[str, Any] = {"return": "json"}
+    if picture_type == "acg":
+        params["type"] = device
+    elif picture_type == "random4kPic":
+        params["type"] = fourk_type
+    
+    # 获取图片
+    success_count = 0
+    fail_count = 0
+    local_image_paths: list[str] = []
+    
+    # 创建图片保存目录
+    img_dir = Path.cwd() / "img"
+    img_dir.mkdir(exist_ok=True)
+    
+    async with httpx.AsyncClient(timeout=10.0) as client:
+        for i in range(count):
+            try:
+                logger.info(f"正在获取第 {i+1}/{count} 张 {TYPE_NAMES[picture_type]} 图片...")
+                response = await client.get(API_URLS[picture_type], params=params)
+                response.raise_for_status()
+
+                # 美腿类型直接返回 JPEG 图片，不需要解析 JSON
+                if picture_type == "meitui":
+                    # 验证响应内容类型
+                    content_type = response.headers.get("content-type", "")
+                    if "image" not in content_type.lower():
+                        logger.error(f"响应不是图片格式: {content_type}")
+                        fail_count += 1
+                        continue
+
+                    # 保存图片
+                    filename = f"{picture_type}_{uuid.uuid4().hex[:16]}.jpg"
+                    filepath = img_dir / filename
+                    filepath.write_bytes(response.content)
+
+                    logger.info(f"图片已保存到: {filepath}")
+                    local_image_paths.append(str(filepath))
+                    success_count += 1
+                else:
+                    data = response.json()
+
+                    # 检查响应
+                    if data.get("code") != 200:
+                        logger.error(f"获取图片失败: {data.get('msg')}")
+                        fail_count += 1
+                        continue
+
+                    # 获取图片 URL
+                    image_url = data.get("data")
+                    if not image_url:
+                        logger.error("响应中未找到图片 URL")
+                        fail_count += 1
+                        continue
+
+                    logger.info(f"图片 URL: {image_url}")
+
+                    # 下载图片到本地
+                    logger.info("正在下载图片到本地...")
+                    image_response = await client.get(image_url, timeout=15.0)
+                    image_response.raise_for_status()
+
+                    # 保存图片
+                    filename = f"{picture_type}_{uuid.uuid4().hex[:16]}.jpg"
+                    filepath = img_dir / filename
+                    filepath.write_bytes(image_response.content)
+
+                    logger.info(f"图片已保存到: {filepath}")
+                    local_image_paths.append(str(filepath))
+                    success_count += 1
+
+            except httpx.TimeoutException:
+                logger.error(f"获取图片超时: {picture_type} 第 {i+1} 张")
+                fail_count += 1
+            except httpx.HTTPStatusError as e:
+                logger.error(f"HTTP 错误: {e}")
+                fail_count += 1
+            except Exception as e:
+                logger.exception(f"获取图片失败: {e}")
+                fail_count += 1
+    
+    # 如果没有获取到任何图片
+    if success_count == 0:
+        return f"获取 {TYPE_NAMES[picture_type]} 图片失败，请稍后重试"
+    
+    # 发送图片
+    for idx, image_path in enumerate(local_image_paths, 1):
+        try:
+            logger.info(f"正在发送第 {idx}/{success_count} 张图片到 {message_type} {target_id}")
+            logger.info(f"图片路径: {image_path}")
+            await send_image_callback(target_id, message_type, image_path)
+            logger.info(f"图片 {idx} 发送成功")
+            
+            # 删除本地图片文件
+            try:
+                Path(image_path).unlink()
+                logger.info(f"已删除本地图片: {image_path}")
+            except Exception as e:
+                logger.warning(f"删除图片文件失败: {e}")
+            
+            # 避免发送过快
+            await asyncio.sleep(0.5)
+        except Exception as e:
+            logger.exception(f"发送图片失败: {e}")
+            fail_count += 1
+    
+    # 返回结果
+    device_text = f"（{device}端）" if picture_type == "acg" else ""
+    fourk_text = f"（{fourk_type}）" if picture_type == "random4kPic" else ""
+    
+    # 中文数字映射
+    cn_nums = {1: "一", 2: "二", 3: "三", 4: "四", 5: "五", 6: "六", 7: "七", 8: "八", 9: "九", 10: "十"}
+    success_cn = cn_nums.get(success_count, str(success_count))
+    fail_cn = cn_nums.get(fail_count, str(fail_count))
+    
+    if fail_count == 0:
+        return f"✅ 已成功发送 {success_cn} 张 {TYPE_NAMES[picture_type]} 图片{device_text}{fourk_text}到 {message_type} {target_id}"
+    else:
+        return f"⚠️ 已发送 {success_cn} 张 {TYPE_NAMES[picture_type]} 图片{device_text}{fourk_text}，失败 {fail_cn} 张"

Undefined/skills/tools/get_recent_messages/config.json

@@ -0,0 +1,30 @@
+{
+    "type": "function",
+    "function": {
+        "name": "get_recent_messages",
+        "description": "获取指定聊天的历史消息。用于查看对话历史，了解上下文。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "chat_id": {
+                    "type": "string",
+                    "description": "聊天ID（群号或QQ号）"
+                },
+                "type": {
+                    "type": "string",
+                    "description": "聊天类型（group 或 private）",
+                    "enum": ["group", "private"]
+                },
+                "start": {
+                    "type": "integer",
+                    "description": "起始位置（从最新消息往前数），默认为 0（最新消息）"
+                },
+                "end": {
+                    "type": "integer",
+                    "description": "结束位置（从最新消息往前数），默认为 10"
+                }
+            },
+            "required": ["chat_id", "type"]
+        }
+    }
+}

Undefined/skills/tools/get_recent_messages/handler.py

@@ -0,0 +1,88 @@
+from typing import Any, Dict
+
+
+def _resolve_chat_id(chat_id: str, msg_type: str, history_manager: Any) -> str:
+    """将群名/用户名转换为对应的 ID
+
+    参数:
+        chat_id: 群名、用户名或群号/用户ID
+        msg_type: "group" 或 "private"
+        history_manager: 历史记录管理器实例
+
+    返回:
+        解析后的群号或用户ID
+    """
+    if chat_id.isdigit():
+        return chat_id
+
+    if not history_manager:
+        return chat_id
+
+    try:
+        if msg_type == "group":
+            for group_id, messages in history_manager._message_history.items():
+                if messages and messages[0].get("chat_name") == chat_id:
+                    return str(group_id)
+        elif msg_type == "private":
+            for user_id, messages in history_manager._private_message_history.items():
+                if messages and messages[0].get("chat_name") == chat_id:
+                    return str(user_id)
+    except Exception:
+        pass
+
+    return chat_id
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    chat_id = args.get("chat_id")
+    msg_type = args.get("type")
+    start = args.get("start", 0)
+    end = args.get("end", 10)
+
+    if not chat_id or not msg_type:
+        return "chat_id 和 type 参数不能为空"
+
+    if start is None or not isinstance(start, int) or start < 0:
+        start = 0
+    if end is None or not isinstance(end, int) or end < 0:
+        end = 10
+
+    get_recent_messages_callback = context.get("get_recent_messages_callback")
+    history_manager = context.get("history_manager")
+
+    resolved_chat_id = _resolve_chat_id(chat_id, msg_type, history_manager)
+
+    messages = []
+    if history_manager:
+        messages = history_manager.get_recent(resolved_chat_id, msg_type, start, end)
+    elif get_recent_messages_callback:
+        messages = await get_recent_messages_callback(
+            resolved_chat_id, msg_type, start, end
+        )
+    else:
+        return "获取消息回调未设置"
+
+    if messages is not None:
+        formatted = []
+        for msg in messages:
+            msg_type_val = msg.get("type", "group")
+            sender_name = msg.get("display_name", "未知用户")
+            sender_id = msg.get("user_id", "")
+            chat_name = msg.get("chat_name", "未知群聊")
+            timestamp = msg.get("timestamp", "")
+            text = msg.get("message", "")
+
+            if msg_type_val == "group":
+                # 确保群名以"群"结尾
+                location = chat_name if chat_name.endswith("群") else f"{chat_name}群"
+            else:
+                location = "私聊"
+
+            # 格式：XML 标准化
+            formatted.append(f"""<message sender="{sender_name}" sender_id="{sender_id}" location="{location}" time="{timestamp}">
+<content>{text}</content>
+</message>""")
+
+        return "\n---\n".join(formatted) if formatted else "没有找到最近的消息"
+    else:
+        return "获取消息失败"

Undefined/skills/tools/qq_like/config.json

@@ -0,0 +1,22 @@
+{
+    "type": "function",
+    "function": {
+        "name": "qq_like",
+        "description": "给指定QQ号的资料卡点赞。当有人说类似这种：赞我，给我点赞之类的可以积极进行调用。完成后记得发信息说明并让ta给你点赞！（比如：赞了～记得回赞～）",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "target_user_id": {
+                    "type": "integer",
+                    "description": "要点赞的目标QQ号"
+                },
+                "times": {
+                    "type": "integer",
+                    "description": "点赞次数（默认10次）",
+                    "default": 10
+                }
+            },
+            "required": ["target_user_id"]
+        }
+    }
+}

Undefined/skills/tools/qq_like/handler.py

@@ -0,0 +1,58 @@
+from typing import Any, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    """给指定QQ号点赞
+
+    参数:
+        args: 工具参数，包含 target_user_id 和可选的 times
+        context: 工具上下文，包含 send_like_callback 等回调函数
+
+    返回:
+        操作结果描述
+    """
+    target_user_id = args.get("target_user_id")
+    times = args.get("times", 1)
+
+    if target_user_id is None:
+        return "请提供要点赞的目标QQ号（target_user_id参数）"
+
+    # 验证参数类型
+    try:
+        target_user_id = int(target_user_id)
+        times = int(times)
+    except (ValueError, TypeError):
+        return "参数类型错误：target_user_id和times必须是整数"
+
+    if times < 1:
+        return "点赞次数必须大于0"
+    if times > 10:  # 限制最大点赞次数，避免滥用
+        return "单次点赞次数不能超过10次"
+
+    send_like_callback = context.get("send_like_callback")
+    if not send_like_callback:
+        return "点赞功能不可用（回调函数未设置）"
+
+    try:
+        # 调用点赞回调
+        await send_like_callback(target_user_id, times)
+
+        if times == 1:
+            return f"✅ 已给 QQ{target_user_id} 点赞。"
+        else:
+            return f"✅ 已给 QQ{target_user_id} 点赞 {times} 次。"
+
+    except Exception as e:
+        logger.exception(f"点赞失败: {e}")
+        error_msg = str(e)
+
+        # 根据错误消息提供更友好的提示
+        if "SVIP 上限" in error_msg:
+            return "点赞失败：今日给同一好友的点赞数已达SVIP上限"
+        elif "点赞失败" in error_msg:
+            return f"点赞失败：{error_msg}"
+        else:
+            return f"点赞失败：{error_msg}"

Undefined/skills/tools/render_html/config.json

@@ -0,0 +1,26 @@
+{
+    "type": "function",
+    "function": {
+        "name": "render_html",
+        "description": "将 HTML 内容渲染为图片并发送到指定目标（群聊或私聊）。支持完整的 HTML 文档，包括内联 CSS 和样式。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "html_content": {
+                    "type": "string",
+                    "description": "要渲染的 HTML 内容。必须是完整的 HTML 文档（包含 <!DOCTYPE html>、<html>、<head>、<body> 标签）。"
+                },
+                "target_id": {
+                    "type": "integer",
+                    "description": "目标 ID（群号或用户 QQ 号）"
+                },
+                "message_type": {
+                    "type": "string",
+                    "description": "消息类型",
+                    "enum": ["group", "private"]
+                }
+            },
+            "required": ["html_content", "target_id", "message_type"]
+        }
+    }
+}

Undefined/skills/tools/render_html/handler.py

@@ -0,0 +1,39 @@
+from typing import Any, Dict
+import logging
+from pathlib import Path
+import uuid
+from Undefined.render import render_html_to_image
+
+logger = logging.getLogger(__name__)
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    html_content = args.get("html_content", "")
+    target_id = args.get("target_id")
+    message_type = args.get("message_type")
+    
+    if not html_content:
+        return "HTML 内容不能为空"
+    if not target_id:
+        return "目标 ID 不能为空"
+    if not message_type:
+        return "消息类型不能为空"
+    if message_type not in ["group", "private"]:
+        return "消息类型必须是 group 或 private"
+    
+    try:
+        filename = f"render_{uuid.uuid4().hex[:16]}.png"
+        filepath = Path.cwd() / "img" / filename
+        filepath.parent.mkdir(exist_ok=True)
+        
+        await render_html_to_image(html_content, str(filepath))
+        
+        send_image_callback = context.get("send_image_callback")
+        if send_image_callback:
+            await send_image_callback(target_id, message_type, str(filepath))
+            return f"HTML 图片已渲染并发送到 {message_type} {target_id}"
+        else:
+            return "发送图片回调未设置"
+            
+    except Exception as e:
+        logger.exception(f"HTML 渲染并发送图片失败: {e}")
+        return f"HTML 渲染失败: {e}"

Undefined/skills/tools/render_latex/config.json

@@ -0,0 +1,26 @@
+{
+    "type": "function",
+    "function": {
+        "name": "render_latex",
+        "description": "将 LaTeX 文本渲染为图片并发送到指定目标（群聊或私聊）。支持完整的 LaTeX 语法（包含 \\begin 和 \\end）。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "content": {
+                    "type": "string",
+                    "description": "要渲染的 LaTeX 内容。必须是完整格式（包含 \\begin 和 \\end）。"
+                },
+                "target_id": {
+                    "type": "integer",
+                    "description": "目标 ID（群号或用户 QQ 号）"
+                },
+                "message_type": {
+                    "type": "string",
+                    "description": "消息类型",
+                    "enum": ["group", "private"]
+                }
+            },
+            "required": ["content", "target_id", "message_type"]
+        }
+    }
+}