From eb64f853bdc1ee99c60e6e983099a8be08be2143 Mon Sep 17 00:00:00 2001
From: Seol <teowhite0423@gmail.com>
Date: Wed, 18 Dec 2024 11:34:17 +0900
Subject: [PATCH] =?UTF-8?q?refactor:=20=EB=93=B1=EB=A1=9D/=ED=83=88?=
 =?UTF-8?q?=ED=87=B4=20=EB=AA=85=EB=A0=B9=EC=96=B4=20=EB=B3=84=EB=8F=84=20?=
 =?UTF-8?q?=ED=8C=8C=EC=9D=BC=EB=A1=9C=20=EB=B6=84=EB=A6=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 functions/account.py | 107 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100644 functions/account.py

diff --git a/functions/account.py b/functions/account.py
new file mode 100644
index 0000000..db18242
--- /dev/null
+++ b/functions/account.py
@@ -0,0 +1,107 @@
+import discord
+from discord.ext import commands
+from utils.commands import slash_command
+from discord.commands import ApplicationContext, Option
+import os
+import pickle
+
+class Account(commands.Cog):
+
+    @slash_command(name="등록", description="유저 정보를 등록합니다.")
+    async def register(self, ctx):
+        if os.path.isfile(f"{ctx.user.id}.pkl") == False:
+            embed = discord.Embed(
+                title="등록", description="봇의 서비스에 등록합니다. 동의하지 않을 경우 서비스 이용에 제한이 있을 수 있습니다."
+            )
+            embed.add_field(name="수집하는 정보", value="유저의 디스코드 id", inline=False)
+
+            class Button(discord.ui.View):
+                @discord.ui.button(label="⭕ 동의", style=discord.ButtonStyle.primary)
+                async def primary(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    UserData = 1
+                    if interaction.user.id == ctx.user.id:
+                        with open(f"{interaction.user.id}.pkl", "wb") as f:
+                            pickle.dump(UserData, f)
+                        new_embed = discord.Embed(
+                            title="등록 완료", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"<@!{interaction.user.id}>님, 등록을 완료하였습니다. 아이디는 {interaction.user.id}입니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+
+                @discord.ui.button(label="❌ 취소", style=discord.ButtonStyle.danger)
+                async def danger(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    if interaction.user.id == ctx.user.id:
+                        new_embed = discord.Embed(
+                            title="등록 취소", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"등록이 취소되었습니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+            await ctx.respond(embed=embed, view=Button())
+
+        else:
+            embed = discord.Embed(title="등록된 유저", description="")
+            embed.add_field(
+                name="", value="이미 등록되었습니다. `/탈퇴` 명령어를 통해 탈퇴할 수 있습니다.", inline=False
+            )
+            await ctx.respond(embed=embed)
+
+
+    @slash_command(name="탈퇴", description="봇 서비스를 탈퇴합니다.")
+    async def delete(self, ctx):
+        if os.path.isfile(f"{ctx.user.id}.pkl"):
+            embed = discord.Embed(
+                title="탈퇴", description="봇의 서비스를 탈퇴합니다. 탈퇴할 경우 모든 데이터가 파기되며, 복구하기 어렵습니다."
+            )
+            embed.add_field(
+                name="파기하는 정보", value="유저의 디스코드 id, 서비스 이용 중 가챠 기록", inline=False
+            )
+
+            class Button(discord.ui.View):
+                @discord.ui.button(label="⭕ 동의", style=discord.ButtonStyle.primary)
+                async def primary(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    if interaction.user.id == ctx.user.id:
+                        os.remove(f"{interaction.user.id}.pkl")
+                        new_embed = discord.Embed(
+                            title="탈퇴 완료", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"<@!{interaction.user.id}>님, 탈퇴를 완료하였습니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+
+                @discord.ui.button(label="❌ 취소", style=discord.ButtonStyle.danger)
+                async def danger(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    if interaction.user.id == ctx.user.id:
+                        new_embed = discord.Embed(
+                            title="탈퇴 취소", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"탈퇴가 취소되었습니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+
+            await ctx.respond(embed=embed, view=Button())
+
+        else:
+            embed = discord.Embed(title="등록하지 않은 유저", description="")
+            embed.add_field(name="", value="`/등록`을 통해 가입해주세요.", inline=False)
+            await ctx.respond(embed=embed)
+
+def setup(bot):
+    bot.add_cog(Account())