From 7cda72ff5009e35ae3c63fb4d511de8fa88a81bc Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Sun, 4 Dec 2022 22:55:57 +0800
Subject: [PATCH] =?UTF-8?q?sticker=20=E4=BC=98=E5=8C=96=E5=88=9B=E5=BB=BA?=
 =?UTF-8?q?=E7=9A=84=E8=B4=B4=E7=BA=B8=E5=8C=85=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sticker/main.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/sticker/main.py b/sticker/main.py
index 2461530..5bca048 100644
--- a/sticker/main.py
+++ b/sticker/main.py
@@ -67,6 +67,7 @@ class Sticker:
     should_forward: Message
     is_animated: bool
     is_video: bool
+    nums: int
     document: Optional[InputDocument]
     document_path: Optional[str]
     software: str = "PagerMaid-Pyro"
@@ -82,6 +83,7 @@ class Sticker:
         self.should_create = False
         self.is_animated = False
         self.is_video = False
+        self.nums = 1
         self.document = None
         self.document_path = None
 
@@ -103,6 +105,7 @@ class Sticker:
             self.custom_sticker_set = True
 
     async def generate_sticker_set(self, time: int = 1):
+        self.nums = time
         if not self.sticker_set or time > 1:
             me = await bot.get_me()
             if not me.username:
@@ -158,7 +161,11 @@ class Sticker:
 
     async def create_sticker_set(self):
         me = await bot.get_me()
-        title = f"@{me.username} 的私藏" if me.username else self.sticker_set
+        title = f"@{me.username} 的私藏（{self.nums}）" if me.username else self.sticker_set
+        if self.is_video:
+            title += "（Video）"
+        elif self.is_animated:
+            title += "（Animated）"
         try:
             await bot.invoke(
                 CreateStickerSet(
@@ -234,8 +241,12 @@ class Sticker:
                f"使用命令 <code>,{alias_command('s')} cancel</code> 取消自定义保存贴纸包"
 
 
-@listener(command="s",
-          need_admin=True)
+@listener(
+    command="s",
+    parameters="<贴纸包名/cancel>",
+    description="保存贴纸到自己的贴纸包",
+    need_admin=True,
+)
 async def sticker(message: Message):
     one_sticker = Sticker(message, should_forward=message.reply_to_message)
     if not message.reply_to_message: