From fcae61039b56bf0e3cdd79642ef14f38f4aca870 Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Fri, 7 Jun 2024 23:33:34 +0800
Subject: [PATCH] feat: donate

---
 modules/donate.py | 55 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 modules/donate.py

diff --git a/modules/donate.py b/modules/donate.py
new file mode 100644
index 0000000..07f2843
--- /dev/null
+++ b/modules/donate.py
@@ -0,0 +1,55 @@
+from pyrogram import filters
+from pyrogram.types import LabeledPrice, PreCheckoutQuery, Message
+
+from defs.glover import admin
+from init import bot
+
+
+@bot.on_message(
+    filters.incoming & filters.command(["donate", f"donate@{bot.me.username}"])
+)
+async def send_donate(_, message: Message):
+    await bot.send_invoice(
+        message.chat.id,
+        title="Donate",
+        description="Support me",
+        currency="XTR",
+        prices=[LabeledPrice(label="Star", amount=10)],
+        payload="stars"
+    )
+
+
+@bot.on_message(
+    filters.successful_payment,
+)
+async def successful_payment(_, message: Message):
+    uuid = message.successful_payment.telegram_payment_charge_id
+    await message.reply_text(
+        f"Thank you for your donation! If you have any questions, please contact @omg_xtao . UUID: `{uuid}`"
+    )
+
+
+@bot.on_message(
+    filters.incoming
+    & filters.private
+    & filters.user(admin)
+    & filters.command(["donate_refund"])
+)
+async def refund_donate(_, message: Message):
+    uid = message.command[1] if len(message.command) > 1 else None
+    uuid = message.command[2] if len(message.command) > 2 else None
+    if not (uuid and uid):
+        return await message.reply_text("Please provide a UID and UUID")
+    try:
+        await bot.refund_star_payment(
+            user_id=int(uid),
+            telegram_payment_charge_id=uuid,
+        )
+        await message.reply_text("Refund successful")
+    except Exception as e:
+        await message.reply_text(f"Refund failed: {e}")
+
+
+@bot.on_pre_checkout_query()
+async def pre_checkout_query_handler(_, query: PreCheckoutQuery):
+    await query.answer(ok=True)