[Plugin] astrbot_plugin_inkfusion

[fatsnk]

插件信息

{
  "name": "astrbot_plugin_inkfusion",
  "display_name": "A1111文生图插件",
  "desc": "用于Astrbot的Pollinations、SD（stable diffusion A1111）、llm图片生成插件",
  "author": "F5",
  "repo": "https://github.com/fatsnk/astrbot_plugin_inkfusion",
  "tags": [
    "A1111",
    "stable diffusion",
    "text2img",
    "sd"
  ],
  "social_link": "https://fatsnk.github.io/fatsnk/"
}

插件检查清单

• 我的插件经过完整的测试
• 我的插件不包含恶意代码
• 我已阅读并同意遵守该项目的 行为准则。

[astrpluginreviewer]

🤖 AI代码审核报告 for astrbot_plugin_inkfusion

您好！我已经对你提交的插件代码进行了初步自动化审核，作为初步参考:

main.py

代码质量与编码规范（PEP 8 / 可读性）

• 类体量过大、职责过多：InkfusionPlugin 同时负责配置解析、提示词优化、两套生图后端（Pollinations/SD）、媒体抽取与下载、命令路由与输出格式化。后期很难测试与替换实现，建议拆分为若干服务类（如 PromptOptimizer / PollinationsClient / SDClient / MediaExtractor）。
• 多处超长行：如 logger.info(...)、@register(...) 描述字符串、部分 f-string 输出，明显超过常见行宽约定（79/88），降低可读性与 code review 体验。
• 类型标注不准确/不完整
  • _download_media() 标注返回 str，但实际可能 return None，应为 str | None。
  • _extract_media_urls() 标注 dict，但结构固定为 {"images": list[str], "videos": list[str]}，建议更精确（或用 TypedDict）。
  • self.api_keys: list = ... / self.models: list = ... 过于宽泛；Python 3.10 可直接用 list[str]。
• 配置值缺乏输入校验：width/height/steps/cfg_scale 等直接取配置，未校验范围与类型（例如传入字符串/负数/超大值），容易导致远端报错或资源消耗异常。

功能实现与逻辑正确性

• _extract_full_args() 的“兜底返回 raw”可能导致误解析
  • 当消息不以预期前缀开头时，它会返回整条原消息（含命令本身），调用方又用它覆盖 prompt_text/model_name，会把命令文本当参数。
  • 这在“平台会改写消息前缀/带引用/带机器人@”时尤其容易触发，属于隐性 bug。
• Stable Diffusion base64 解码鲁棒性不足
  • A1111/某些前端或代理可能返回形如 data:image/png;base64,xxxx 的字符串；当前直接 base64.b64decode(images[0]) 可能失败。缺少对前缀的剥离处理。
• 模型切换的“持久化”不明确
  • /ai生图 模型 修改了 self.config["models"]，但没有看到将配置写回 AstrBot 配置系统的动作；重启后可能丢失，造成“用户以为切换成功但重启失效”的体验问题。

安全漏洞与最佳实践

• SSRF / 内网探测风险（高）：_generate_via_llm() 会从 LLM 返回文本中提取任意 http(s) URL 并下载，未做任何限制：
  • 未限制目标域名/IP 段，LLM 可能被提示注入诱导返回 http://127.0.0.1:...、http://169.254.169.254/...（云元数据）或内网地址，导致机器人作为跳板访问内网资源。
  • 建议至少做：解析域名到 IP 并拒绝内网/回环/链路本地地址；可选白名单域名策略。
• 下载文件无大小上限：_download_media() / _generate_image() 直接 await resp.read()，没有基于 Content-Length 或流式读取的上限控制。恶意 URL 可导致内存占用暴涨或磁盘被打满（DoS）。
• 允许自动跳转可能带来降级/绕过：所有 allow_redirects=True，若跳转到不可信域名或从 https 跳到 http，可能引入中间人风险或绕过域名限制（如果你后续加限制但未校验最终 URL，会被绕过）。
• sd_skip_ssl_verify 明显降低安全性：关闭证书校验会让 HTTPS 形同虚设。并且当前 sd_base_url 默认是 http://，此开关容易让用户误以为“更安全/更可用”，实际上是把风险藏起来。
• 日志可能泄露隐私/敏感信息
  • 记录了“完整请求URL”（包含用户 prompt、negative_prompt、尺寸等），在多用户群聊场景可能属于隐私数据。
  • 日志中输出 API Key 前 8 位（api_key[:8]）虽然不完整，但仍可能被用于关联与社工；且在一些 Key 结构固定的系统中可能增加泄露风险。

异步与网络 I/O 最佳实践（仅指出网络相关阻塞/缺陷）

• 频繁创建 aiohttp.ClientSession()：_generate_image() 每次重试都新建 session；_download_media() 每次下载也新建 session。高并发下会增加连接建立开销、DNS/SSL 开销，且更难做统一连接池/并发限制。建议在插件生命周期内复用 session（或至少在一次操作内复用同一个 session）。
• 缺少并发/速率控制：
  • 对外部服务（Pollinations、SD、本地 WebUI、任意媒体 URL）没有信号量限制；群聊/多用户同时请求时可能把自己和对方服务打挂。
  • 重试策略无 jitter（attempt*2 固定回退），并发场景会出现“同步重试风暴”。
• 响应内容类型校验不完整
  • _generate_image() 仅用 Content-Type 包含 "image" 判断，有些服务会返回 application/octet-stream 携带图片；此时会误判失败并重试，造成不必要流量。
  • _download_media() 同样过度依赖 Content-Type 与扩展名，缺少对实际内容（magic bytes）的最低限度判断（至少对图片/视频可选做简要签名判断或容错策略）。

可维护性与可读性（结构/复用/测试）

• 重复的命令解析与提示文本输出：generate_image_sub、generate_image_shortcut、sd_generate、sd_generate_shortcut 有大量重复（提参、优化、输出“正在生成”），建议抽公共方法减少维护成本。
• 正则抽取 URL 的策略容易误伤/漏报
  • all_urls 的 regex 会把一些带右括号/引号的 URL 截断或包含多余字符（虽然你做了部分排除，但仍常见于 Markdown/HTML 混合文本）。
  • 对“无扩展名但实际是图片”的 URL（常见 CDN）漏报较多，导致“LLM 返回了图但插件说没检测到”。

---

代码评审团

人物一：Linus Torvalds（毒舌、直指痛点）
你这插件像个“上帝类”把所有东西都塞进去：LLM、两套生图后端、下载器、解析器、命令行 UI……短期爽，长期就是维护地狱。把网络客户端拆出来，给每个后端一个清晰接口（generate(prompt)->path），再谈扩展模型/切换提供商，不然以后每加一项功能都在同一个类里继续堆垃圾。

人物二：Grace Hopper（强调工程纪律与安全边界）
从 LLM 输出里抓链接就下载，这是典型“把控制权交给不可信输入”。至少要把 SSRF 这条边界立住：禁止内网地址、限制协议、限制重定向后的最终地址、设置最大下载大小。否则机器人运行在谁的网络里，谁就要承担“被你当扫描器”的风险。

人物三：Geordi La Forge《星际迷航》（偏系统化、面向性能与运行稳定性）
你现在每次请求都开一个新的 ClientSession，重试还会再开，连接池完全没利用起来。并发一上来，连接数、DNS、端口资源都会抖。把 session 做成插件级资源统一管理，再加一个 asyncio.Semaphore 控制外部请求并发，你的稳定性会立刻提升一个量级；再配合带 jitter 的退避，能显著减少服务端与自身被“齐刷刷重试”拖垮。

Disclaimer: 以上评审内容由 AI 自动生成，所涉及人物形象与现实无关，不代表真实人物观点。如果给出的建议无关痛痒请忽略。

---

🔍 审核摘要

统计信息

• 仓库文件总数: 1 个 Python 文件
• 已选择审核文件: 1 / 1
  已发送至 AI 审核的文件清单

main.py

---

此报告由AI自动生成，旨在提供初步反馈和改进建议，不能完全替代人工审核。最终决策以社区维护者的人工审核为准。目前自动审核（仓库地址）处于试验阶段，如遇问题请向维护者反馈。

[fatsnk]

我自用的，懒得优化了。修改自https://github.com/qa296/astrbot_plugin_pollinations_images
pollinations接口没更新，因此做了这个。
astrbot_plugin_pollinations_images插件已经在商店中，仅支持旧的pollinations接口。

[astrbot-plugin-copybara]

插件信息已同步到集合仓库。
PR: AstrBotDevs/AstrBot_Plugins_Collection#232
已自动合并 PR #232

该插件已上架。
This plugin is now published.