From b77d73ddc7b3215cb7cfeda3fbd5b0147c486709 Mon Sep 17 00:00:00 2001
From: tcmofashi <mofashiforzbx@qq.com>
Date: Fri, 7 Mar 2025 01:49:42 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=8E=B0=E5=9C=A8=E5=8F=AF=E4=BB=A5?=
 =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E6=98=AF=E5=90=A6=E8=87=AA=E5=8A=A8=E4=BF=9D?=
 =?UTF-8?q?=E5=AD=98=E8=A1=A8=E6=83=85=E5=8C=85=E4=BA=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config/bot_config_template.toml | 1 +
 src/plugins/chat/config.py      | 2 ++
 src/plugins/chat/utils_image.py | 3 +++
 3 files changed, 6 insertions(+)

diff --git a/config/bot_config_template.toml b/config/bot_config_template.toml
index afc2b5079..4428e1512 100644
--- a/config/bot_config_template.toml
+++ b/config/bot_config_template.toml
@@ -20,6 +20,7 @@ ban_words = [
 [emoji]
 check_interval = 120 # 检查表情包的时间间隔
 register_interval = 10 # 注册表情包的时间间隔
+auto_save = true  # 自动偷表情包
 check_prompt = "不要包含违反公序良俗的内容" # 表情包过滤要求
 
 [cq_code]
diff --git a/src/plugins/chat/config.py b/src/plugins/chat/config.py
index e044edc5e..6fb6045da 100644
--- a/src/plugins/chat/config.py
+++ b/src/plugins/chat/config.py
@@ -30,6 +30,7 @@ class BotConfig:
     forget_memory_interval: int = 300  # 记忆遗忘间隔（秒）
     EMOJI_CHECK_INTERVAL: int = 120  # 表情包检查间隔（分钟）
     EMOJI_REGISTER_INTERVAL: int = 10  # 表情包注册间隔（分钟）
+    EMOJI_SAVE: bool = True  # 偷表情包
     EMOJI_CHECK_PROMPT: str = "不要包含违反公序良俗的内容" # 表情包过滤要求
 
     ban_words = set()
@@ -96,6 +97,7 @@ class BotConfig:
                 config.EMOJI_CHECK_INTERVAL = emoji_config.get("check_interval", config.EMOJI_CHECK_INTERVAL)
                 config.EMOJI_REGISTER_INTERVAL = emoji_config.get("register_interval", config.EMOJI_REGISTER_INTERVAL)
                 config.EMOJI_CHECK_PROMPT = emoji_config.get('check_prompt',config.EMOJI_CHECK_PROMPT)
+                config.EMOJI_SAVE = emoji_config.get('auto_save',config.EMOJI_SAVE)
             
             if "cq_code" in toml_dict:
                 cq_code_config = toml_dict["cq_code"]
diff --git a/src/plugins/chat/utils_image.py b/src/plugins/chat/utils_image.py
index 9a7ef789a..503c2fa85 100644
--- a/src/plugins/chat/utils_image.py
+++ b/src/plugins/chat/utils_image.py
@@ -4,6 +4,7 @@ import hashlib
 import time
 import os
 from ...common.database import Database
+from ..chat.config import global_config
 import zlib  # 用于 CRC32
 import base64
 from nonebot import get_driver
@@ -143,6 +144,8 @@ def storage_emoji(image_data: bytes) -> bytes:
     Returns:
         bytes: 原始图片数据
     """
+    if not global_config.EMOJI_SAVE:
+        return image_data
     try:
         # 使用 CRC32 计算哈希值
         hash_value = format(zlib.crc32(image_data) & 0xFFFFFFFF, 'x')