From a8afa113cfef80237031916e20df44bc11df2270 Mon Sep 17 00:00:00 2001
From: Kyomotoi <0w0@imki.moe>
Date: Mon, 17 Oct 2022 23:46:00 +0800
Subject: =?UTF-8?q?=E2=99=BB=EF=B8=8F=20=E8=B7=9F=E8=BF=9B=20service=20?=
 =?UTF-8?q?=E7=9A=84=E5=8F=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/rss/__init__.py | 32 +++++++++++++++-----------------
 1 file changed, 15 insertions(+), 17 deletions(-)

(limited to 'ATRI/plugins/rss/__init__.py')

diff --git a/ATRI/plugins/rss/__init__.py b/ATRI/plugins/rss/__init__.py
index a29f385..3f0b906 100644
--- a/ATRI/plugins/rss/__init__.py
+++ b/ATRI/plugins/rss/__init__.py
@@ -1,28 +1,25 @@
 from pathlib import Path
 
 from nonebot.adapters.onebot.v11 import MessageEvent
-from nonebot.permission import SUPERUSER
-from nonebot.adapters.onebot.v11 import GROUP_OWNER, GROUP_ADMIN
 
+from ATRI.permission import ADMIN
 from ATRI.service import Service
+from ATRI.message import MessageBuilder
 
 
 RSS_PLUGIN_DIR = Path(".") / "ATRI" / "plugins" / "rss"
 
 
-class RssHelper(Service):
-    def __init__(self):
-        Service.__init__(
-            self,
-            "rss",
-            "Rss系插件助手",
-            True,
-            permission=SUPERUSER | GROUP_OWNER | GROUP_ADMIN,
-            main_cmd="/rss",
-        )
+rss_helper = (
+    Service("rss")
+    .document("Rss系插件助手")
+    .only_admin(True)
+    .permission(ADMIN)
+    .main_cmd("/rss")
+)
 
 
-rss_menu = RssHelper().on_command("/rss", "Rss帮助菜单")
+rss_menu = rss_helper.on_command("/rss", "Rss帮助菜单")
 
 
 @rss_menu.handle()
@@ -32,8 +29,9 @@ async def _rss_menu(event: MessageEvent):
     if not rss_list:
         rss_list = [str(i).split("\\")[-1] for i in raw_rss_list]
 
-    result = f"""Rss Helper:
-    可用订阅源: {"、".join(map(str, rss_list)).replace("rss_", str())}
-    命令: /rss.(订阅源名称)
-    """.strip()
+    result = (
+        MessageBuilder("Rss Helper:")
+        .text(f"可用订阅源: {', '.join(map(str, rss_list)).replace('rss_', str())}")
+        .text("命令: /rss.(订阅源名称)")
+    )
     await rss_menu.finish(result)
-- 
cgit v1.2.3