From 4338aebac26032ce709b5515f7c0cbf027d20870 Mon Sep 17 00:00:00 2001
From: franchioping <franchioping@gmail.com>
Date: Wed, 24 Jul 2024 14:16:51 +0100
Subject: [PATCH] commit

---
 embed_util.py | 49 ++++++++++++++++++++++++++++++++++++++++---------
 main.py       |  7 +++++--
 2 files changed, 45 insertions(+), 11 deletions(-)

diff --git a/embed_util.py b/embed_util.py
index 620a37e..f862bb4 100644
--- a/embed_util.py
+++ b/embed_util.py
@@ -5,6 +5,9 @@ import os.path
 import requests
 from util import parallel_downloads
 
+prev_label = "Prev"
+next_label = "Next"
+
 
 class ListManga(discord.ui.View):
     def __init__(self, manga_list: list[Manga]):
@@ -13,30 +16,58 @@ class ListManga(discord.ui.View):
         self.index = 0
         self.manga_list = manga_list
         self.manga_files = gen_manga_files(self.manga_list)
+        self.comp = self.children
+        self.prev: discord.Button = None
+        self.next: discord.Button = None
+        for comp in self.comp:
+            if comp.type.name == "button":
+                button: discord.Button = comp
+                if button.label == prev_label:
+                    self.prev = button
+                if button.label == next_label:
+                    self.next = button
+        self.msg: discord.Message = None
+        print(self.manga_list)
 
-    async def force_reload(self, msg: discord.Message):
-        await msg.edit(
+    def set_msg(self, msg):
+        self.msg = msg
+
+    async def update_buttons(self):
+        self.prev.disabled = self.index != 0
+        self.next.disabled = self.index != len(self.manga_list) - 1
+
+        print(f"index: {self.index}, prev: {self.prev.disabled}, next {self.next.disabled}")
+
+
+    async def force_reload(self):
+        await self.update_buttons()
+
+        self.msg = await self.msg.edit(
             embed=manga_embed(self.manga_list[self.index]),
-            attachments=[self.manga_files[self.index]]
+            attachments=[self.manga_files[self.index]],
+            view=self
         )
 
     async def print_manga(self, interaction: discord.Interaction):
+
         await interaction.response.defer()
-        await interaction.message.edit(
-            embed=manga_embed(self.manga_list[self.index]),
-            attachments=[self.manga_files[self.index]]
-        )
 
+        await self.update_buttons()
 
-    @discord.ui.button(label='Prev', style=discord.ButtonStyle.grey)
+        await self.force_reload()
+
+    @discord.ui.button(label=prev_label, style=discord.ButtonStyle.grey)
     async def previous(self, interaction: discord.Interaction, button: discord.ui.Button):
+        self.prev = button
         if self.index > 0:
             self.index -= 1
+            print("cant go anymore back")
 
         await self.print_manga(interaction)
 
-    @discord.ui.button(label='Next', style=discord.ButtonStyle.grey)
+    @discord.ui.button(label=next_label, style=discord.ButtonStyle.grey)
     async def next(self, interaction: discord.Interaction, button: discord.ui.Button):
+        self.next = button
         if self.index < len(self.manga_list) - 1:
             self.index += 1
 
diff --git a/main.py b/main.py
index 30cf5d5..5dd54a4 100644
--- a/main.py
+++ b/main.py
@@ -31,7 +31,6 @@ async def first_command(interaction: discord.Interaction):
 
 
 async def render_manga_list_in_dm(interaction: discord.Interaction, manga_list: list[manga_api.Manga]):
-    await interaction.followup.send("Search done, Check your DM's")
     chanel = await interaction.user.create_dm()
 
     if len(manga_list) == 0:
@@ -40,7 +39,11 @@ async def render_manga_list_in_dm(interaction: discord.Interaction, manga_list:
 
     view = embed_util.ListManga(manga_list)
     msg = await chanel.send(view=view, embed=embed_util.manga_embed(manga_list[0]))
-    await view.force_reload(msg)
+    view.set_msg(msg)
+    await view.force_reload()
+
+    await interaction.followup.send("Query done, Check your DM's")
+
     await view.wait()
 
     print("Done.. Checking Returns")