From 28fc379842e1f3ff2a5e5006308b65e9a4858d8b Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Sun, 21 Aug 2022 15:19:56 +0800
Subject: [PATCH] =?UTF-8?q?aff=20=E5=BF=AB=E9=80=9F=E5=8F=91=E9=80=81=20af?=
 =?UTF-8?q?f?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 aff/main.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 aff/main.py

diff --git a/aff/main.py b/aff/main.py
new file mode 100644
index 0000000..8474dae
--- /dev/null
+++ b/aff/main.py
@@ -0,0 +1,43 @@
+import contextlib
+
+from pagermaid.enums import Message
+from pagermaid.single_utils import sqlite
+from pagermaid.listener import listener
+
+
+def get_aff() -> str:
+    return sqlite.get("aff.text", "")
+
+
+def set_aff(text: str) -> None:
+    sqlite.update({"aff.text": text})
+
+
+def del_aff() -> None:
+    if get_aff():
+        del sqlite["aff.text"]
+
+
+@listener(command="aff",
+          description="在别人要打算买机场的时候光速发出自己的aff信息(请尽量配合短链接)",
+          parameters="<save|remove> (可选，回复一条消息，用于保存|删除aff信息)")
+async def aff(message: Message):
+    if not message.parameter:
+        msg = get_aff()
+        if not msg:
+            return await message.edit("出错了呜呜呜 ~ Aff消息不存在。\n(你有提前保存好嘛？)")
+        with contextlib.suppress(Exception):
+            await message.edit(msg, disable_web_page_preview=False)
+    elif message.parameter[0] == "save":
+        if not message.reply_to_message:
+            return await message.edit("出错了呜呜呜 ~ 请回复一条消息以保存新的Aff信息。")
+        text = message.reply_to_message.text or message.reply_to_message.caption
+        if not text:
+            return await message.edit("出错了呜呜呜 ~ 请回复一条消息以保存新的Aff信息。")
+        set_aff(text.html)
+        await message.edit("好耶 ！ Aff信息保存成功。")
+    elif message.parameter[0] == "remove":
+        del_aff()
+        await message.edit("好耶 ！ Aff信息删除成功。")
+    else:
+        await message.edit("出错了呜呜呜 ~ 无效的参数。")