[Plugin] astrbot_plugin_codesnap

[Zeraora-807]

插件信息

{
  "name": "astrbot_plugin_codesnap",
  "display_name": "CodeSnap(?)",
  "desc": "A plugin inspired by CodeSnap",
  "author": "ZeraoraBot",
  "repo": "https://github.com/Zeraora-807/astrbot_plugin_codesnap",
  "tags": [],
  "social_link": ""
}

插件检查清单

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

[astrpluginreviewer]

🤖 AI代码审核报告 for astrbot_plugin_codesnap

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

main.py

代码质量与编码规范

1. CSS 样式截断 (严重): 在 DEFAULT_TEMPLATE 常量中存在大量的 CSS 样式值截断丢失。例如：

   body {
       background: linear-gradient(135deg,  /* <-- 颜色值缺失 */
       font-family: 'Consolas', monospace;
   }
   .code-container {
       background: /* <-- 颜色值缺失 */
       border-radius: 12px;
   }
   .close {
       background: /* <-- 颜色值缺失 */
   }

   这会导致 HTML 页面样式完全失效，生成的代码截图将失去原有的背景和圆点设计，严重影响视觉效果。请补充完整的 CSS 属性值。
2. PEP 8 规范: 类的方法之间、函数定义之间应当保留一定的空行（通常为一个空行）。目前的排版显得过于紧凑，降低了可读性。

功能实现与逻辑正确性

1. HTML 标签重复嵌套: 在 _highlight_code 方法的异常处理回退逻辑中，当遇到错误时返回了 f'<pre>{safe_code}</pre>'。由于在 HTML 模板 DEFAULT_TEMPLATE 中已经存在 <pre>{{ highlighted_code | safe }}</pre> 标签，替换后会变成 <pre><pre>转义后的代码</pre></pre>，导致不规范的嵌套。建议异常时直接返回 safe_code 即可。
2. 命令参数解析稍微冗杂: snap_code 中的指令截取逻辑略显重复。由于 event.message_str 已由 AstrBot 框架处理过，结合 @snap.command("code") 装饰器，实际上可以考虑利用框架自带的参数获取方式来精简逻辑。不过现有的 split 判断逻辑依然可以正常运行。

安全漏洞与最佳实践

1. 潜在的临时文件泄漏: 在 _render_with_playwright 方法中使用了 NamedTemporaryFile(suffix='.png', delete=False)。如果在 await page.screenshot(path=tmp.name, full_page=True) 这一步因为某种原因（例如页面过长导致超时或内存不足）抛出异常并跳出，此时生成的临时文件由于 delete=False 不会被清理。建议将其放入 try...finally 块中捕获异常并在失败时直接手动删除文件，以防长期的临时文件泄漏耗尽磁盘空间。

可维护性与可读性

1. 整体职责划分较为明确，Playwright 的按需懒加载设计 (_get_browser) 和清理机制 (terminate) 都不错。
2. 主题回退机制：在解析主题参数时，若输入的未匹配则退回 default 逻辑写得很清晰，具备良好的容错性。

框架适应性检查

1. 插件注册与主类: 继承自 Star 并使用了正确的 @register，符合要求。
2. 日志记录: 正确使用了 from astrbot.api import logger，符合要求。
3. 数据持久化: 正确使用了 StarTools.get_data_dir("astrbot_plugin_codesnap") 并且通过 Path 对象正确地处理了文件读写，符合要求。
4. 事件监听器签名与导出: 正确引入了 filter 对象，所有指令监听均传入了 event: AstrMessageEvent 参数，符合要求。

---

init.py

审查报告
该文件为空，通常在 Python 项目中仅作为包 (Package) 标识符存在。无需添加任何额外逻辑，符合当前标准和最佳实践。

---

代码评审团

Linus Torvalds：（皱着眉头，指着屏幕上的一堆代码）

“我简直不敢相信我的眼睛。你在 DEFAULT_TEMPLATE 里的 CSS 是怎么回事？background: linear-gradient(135deg, 然后呢？！你的键盘是在敲完逗号之后突然爆炸了吗？这种明显是复制粘贴到一半被截断的错误，连编译器都不会原谅你。你的排版引擎看到这些缺失的颜色值时大概会直接哭出来。赶紧把这些残废的 CSS 补全，别把这种未完成的灾难推向生产环境！”

Steve Jobs：（推了推圆框眼镜，语气平静但挑剔）

“我们在追求极致的美感，而你却在破坏它。看看你的高亮降级逻辑，你竟然把一个 <pre> 标签塞进了另一个 <pre> 标签里！这就像是在精美的包装盒外面套上了一个一模一样的包装盒，毫无意义且笨重。真正的优雅应该是由内而外的一致性。拿掉那层多余的标签，让代码自由呼吸。另外，按需加载 Playwright 是个不错的直觉，但这还不够完美，记住，用户在等待你的每一秒，都是对体验的巨大消耗。”

GLaDOS：（毫无感情的电子音，带着令人发毛的嘲讽）

“测试结果显示：你在这段代码中成功地留下了一个绝妙的磁盘填充漏洞。当 page.screenshot 发生故障时，那些不会被自动删除的 .png 临时文件就会像癌细胞一样在服务器的硬盘里增殖。这难道是某种新型的压力测试吗？如果是的话，我给你打满分。为了庆祝你的这项壮举，我已经启动了实验室的神经毒素释放程序。不用谢，下次记得在 finally 块里加上 os.remove()。”

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

---

🔍 审核摘要

统计信息

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

main.py
__init__.py

---

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

当前仓库：Zeraora-807/astrbot_plugin_codesnap
剩余触发次数：2

[Zeraora-807]

I’ve read the report from AI, now here's my Issue fix report:

1. CSS Style Truncation

Review Comment:
CSS property values are truncated in DEFAULT_TEMPLATE, e.g.,

background: linear-gradient(135deg,

missing color values.

Fix Description:
This was an AI misinterpretation. All CSS property values are complete in the code, for example:

background: linear-gradient(135deg, #4a1d4a 0%, #b45f06 100%);

The AI mistakenly recognized the # in color codes as a comment delimiter. The styles are intact. Tests have been conducted in both local and Docker environments; screenshot rendering is normal with no missing styles.

[Zeraora-807]

2. HTML Tag Redundant Nesting

Review Comment:
_highlight_code returns <pre>{safe_code}</pre> in the exception handling fallback, causing duplicate <pre> tags since the template already includes <pre>.

Fix Description:
Fixed. The exception branch now returns safe_code (plain text) directly without adding <pre> tags. The template handles the wrapping uniformly.

# Before
return f'<pre>{safe_code}</pre>', ... 

# After
return safe_code, ...

[Zeraora-807]

3. Potential Temporary File Leak

Review Comment:
_render_with_playwright uses NamedTemporaryFile(delete=False). If an exception occurs during page.screenshot(), the temporary file is not cleaned up.

Fix Description:
Fixed. A tmp_path variable stores the file path. The file is created in the try block. In the except block, any exception triggers manual cleanup of the temporary file. The finally block closes page and context. This ensures the temporary file is properly handled regardless of success or failure.

# After fix
tmp_path = None
try:
    with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp:
        tmp_path = tmp.name
    await page.screenshot(path=tmp_path, full_page=True)
    return tmp_path
except Exception:
    if tmp_path and Path(tmp_path).exists():
        Path(tmp_path).unlink(missing_ok=True)
    raise
finally:
    await page.close()
    await context.close()

[Zeraora-807]

All issues have been addressed. Please review again.

Functions of the plugin have been checked in Astrbot (Docker)

[astrbot-plugin-copybara]

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

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