|
|
@@ -0,0 +1,227 @@
|
|
|
+"""Affichage et modification des fichiers de mémoire sur Discord : /memoire et /memoire-serveur.
|
|
|
+
|
|
|
+Tout s'affiche dans des messages éphémères : seule la personne qui a lancé la commande les voit."""
|
|
|
+
|
|
|
+from __future__ import annotations
|
|
|
+
|
|
|
+import logging
|
|
|
+from collections.abc import Awaitable, Callable
|
|
|
+from dataclasses import dataclass
|
|
|
+
|
|
|
+import discord
|
|
|
+
|
|
|
+from .members import MemberRegistry, display_name
|
|
|
+from .memory import SERVER_FILE, MemoryStore
|
|
|
+from .text_utils import split_message
|
|
|
+
|
|
|
+logger = logging.getLogger("chatbot.memory")
|
|
|
+
|
|
|
+# Longueur maximale d'une zone de texte dans une fenêtre Discord, et nombre maximal de zones.
|
|
|
+INPUT_MAX = 4000
|
|
|
+MAX_INPUTS = 5
|
|
|
+# Texte maximal d'un embed (4096 pour Discord), et de tous les embeds d'un message (6000).
|
|
|
+EMBED_MAX = 4000
|
|
|
+EMBEDS_TOTAL_MAX = 5500
|
|
|
+# Durée pendant laquelle les boutons répondent (ensuite, il suffit de relancer la commande).
|
|
|
+VIEW_TIMEOUT = 600
|
|
|
+
|
|
|
+Callback = Callable[[discord.Interaction], Awaitable[None]]
|
|
|
+
|
|
|
+
|
|
|
+async def see_user(registry: MemberRegistry, user: discord.User | discord.Member) -> None:
|
|
|
+ """Ajoute un utilisateur Discord au registre, ou met ses noms à jour."""
|
|
|
+ # En message privé, l'auteur n'est pas un membre du serveur : son pseudo n'est pas connu.
|
|
|
+ in_guild = isinstance(user, discord.Member)
|
|
|
+ await registry.see(user.id, user.name, user.global_name, user.nick if in_guild else None, in_guild=in_guild)
|
|
|
+
|
|
|
+
|
|
|
+@dataclass
|
|
|
+class Target:
|
|
|
+ """Fichier affiché : la fiche d'un membre, ou le fichier général (member_id=None)."""
|
|
|
+ file: str
|
|
|
+ member_id: int | None
|
|
|
+ name: str = "" # nom du membre
|
|
|
+ is_self: bool = False # fiche de la personne qui a lancé la commande
|
|
|
+
|
|
|
+
|
|
|
+class MemoryCard:
|
|
|
+ """Ce que voit la personne qui a lancé la commande : le fichier et les boutons pour le modifier."""
|
|
|
+
|
|
|
+ def __init__(self, store: MemoryStore, target: Target, owner_id: int, bot_name: str):
|
|
|
+ self.store = store
|
|
|
+ self.target = target
|
|
|
+ self.owner_id = owner_id
|
|
|
+ self.bot_name = bot_name
|
|
|
+
|
|
|
+ # ---------------------------------------------------------------- état
|
|
|
+
|
|
|
+ @property
|
|
|
+ def memory_enabled(self) -> bool:
|
|
|
+ return self.target.member_id is None or self.store.members.memory_enabled(self.target.member_id)
|
|
|
+
|
|
|
+ def about(self) -> str:
|
|
|
+ """« sur toi », « sur Penta » ou « sur le serveur »."""
|
|
|
+ if self.target.member_id is None:
|
|
|
+ return "sur le serveur"
|
|
|
+ return "sur toi" if self.target.is_self else f"sur {self.target.name}"
|
|
|
+
|
|
|
+ # ----------------------------------------------------------- affichage
|
|
|
+
|
|
|
+ def embeds(self) -> list[discord.Embed]:
|
|
|
+ if self.target.member_id is None:
|
|
|
+ title = "Fichier général du serveur"
|
|
|
+ else:
|
|
|
+ title = "Ta fiche mémoire" if self.target.is_self else f"Fiche mémoire de {self.target.name}"
|
|
|
+ content = self.store.read(self.target.file)
|
|
|
+ if not self.memory_enabled:
|
|
|
+ who = "Tu as" if self.target.is_self else f"{self.target.name} a"
|
|
|
+ text = (f"{who} désactivé la mémoire : {self.bot_name} ne retient rien "
|
|
|
+ f"{self.about()} et ne voit aucune fiche dans ses messages.")
|
|
|
+ return [discord.Embed(title=title, description=text, color=discord.Color.light_grey())]
|
|
|
+ if not content:
|
|
|
+ return [discord.Embed(title=title, description=f"{self.bot_name} n'a encore rien retenu {self.about()}.",
|
|
|
+ color=discord.Color.blurple())]
|
|
|
+ size = len(content)
|
|
|
+ if size > EMBEDS_TOTAL_MAX: # seulement si MEMORY_*_LIMIT dépasse ce que Discord affiche
|
|
|
+ content = content[:EMBEDS_TOTAL_MAX] + "\n[…] (la suite est visible dans « Modifier »)"
|
|
|
+ embeds = [discord.Embed(description=chunk, color=discord.Color.blurple())
|
|
|
+ for chunk in split_message(content, EMBED_MAX)]
|
|
|
+ embeds[0].title = title
|
|
|
+ embeds[-1].set_footer(text=f"{size}/{self.store.limit(self.target.file)} caractères")
|
|
|
+ return embeds
|
|
|
+
|
|
|
+ def view(self) -> discord.ui.View:
|
|
|
+ view = OwnerView(self.owner_id)
|
|
|
+ content = self.store.read(self.target.file)
|
|
|
+ add_button(view, "Modifier", "✏️", discord.ButtonStyle.primary, self.edit, disabled=not self.memory_enabled)
|
|
|
+ add_button(view, "Tout effacer", "🗑️", discord.ButtonStyle.secondary, self.ask_erase,
|
|
|
+ disabled=not content or not self.memory_enabled)
|
|
|
+ # Seul le membre lui-même choisit d'activer ou non la mémoire.
|
|
|
+ if self.target.member_id is not None and self.target.is_self:
|
|
|
+ if self.memory_enabled:
|
|
|
+ add_button(view, "Désactiver la mémoire", "🚫", discord.ButtonStyle.danger, self.ask_disable)
|
|
|
+ else:
|
|
|
+ add_button(view, "Réactiver la mémoire", "✅", discord.ButtonStyle.success, self.enable)
|
|
|
+ return view
|
|
|
+
|
|
|
+ async def send(self, interaction: discord.Interaction) -> None:
|
|
|
+ await interaction.response.send_message(embeds=self.embeds(), view=self.view(), ephemeral=True)
|
|
|
+
|
|
|
+ async def refresh(self, interaction: discord.Interaction, note: str | None = None) -> None:
|
|
|
+ """Réaffiche le fichier à jour dans le même message, avec une courte phrase au-dessus."""
|
|
|
+ await interaction.response.edit_message(content=note, embeds=self.embeds(), view=self.view())
|
|
|
+
|
|
|
+ # -------------------------------------------------------------- actions
|
|
|
+
|
|
|
+ async def edit(self, interaction: discord.Interaction) -> None:
|
|
|
+ await interaction.response.send_modal(EditModal(self))
|
|
|
+
|
|
|
+ async def save(self, interaction: discord.Interaction, text: str) -> None:
|
|
|
+ """Enregistre le texte de la fenêtre « Modifier »."""
|
|
|
+ error = await self.store.overwrite(self.target.file, text, by=str(interaction.user))
|
|
|
+ if error:
|
|
|
+ await interaction.response.send_message(error, ephemeral=True)
|
|
|
+ return
|
|
|
+ await self.refresh(interaction, "C'est enregistré." if text.strip() else "C'est effacé.")
|
|
|
+
|
|
|
+ async def ask_erase(self, interaction: discord.Interaction) -> None:
|
|
|
+ question = f"Effacer tout ce que {self.bot_name} a retenu {self.about()} ? C'est définitif."
|
|
|
+ await confirm(interaction, self, question, self.erase)
|
|
|
+
|
|
|
+ async def erase(self, interaction: discord.Interaction) -> None:
|
|
|
+ error = await self.store.overwrite(self.target.file, "", by=str(interaction.user))
|
|
|
+ await self.refresh(interaction, error or "C'est effacé.")
|
|
|
+
|
|
|
+ async def ask_disable(self, interaction: discord.Interaction) -> None:
|
|
|
+ question = (f"Si tu désactives la mémoire, tout ce que {self.bot_name} a retenu sur toi sera "
|
|
|
+ f"définitivement effacé, et plus rien ne sera retenu sur toi. Tu pourras la réactiver "
|
|
|
+ f"plus tard avec /memoire.")
|
|
|
+ await confirm(interaction, self, question, self.disable)
|
|
|
+
|
|
|
+ async def disable(self, interaction: discord.Interaction) -> None:
|
|
|
+ error = await self.store.overwrite(self.target.file, "", by=str(interaction.user))
|
|
|
+ if error is None:
|
|
|
+ await self.store.members.set_memory(self.target.member_id, False)
|
|
|
+ await self.refresh(interaction, error or "Mémoire désactivée : ta fiche a été effacée.")
|
|
|
+
|
|
|
+ async def enable(self, interaction: discord.Interaction) -> None:
|
|
|
+ await self.store.members.set_memory(self.target.member_id, True)
|
|
|
+ await self.refresh(interaction, "Mémoire réactivée.")
|
|
|
+
|
|
|
+
|
|
|
+class OwnerView(discord.ui.View):
|
|
|
+ """Boutons réservés à la personne qui a lancé la commande."""
|
|
|
+
|
|
|
+ def __init__(self, owner_id: int):
|
|
|
+ super().__init__(timeout=VIEW_TIMEOUT)
|
|
|
+ self.owner_id = owner_id
|
|
|
+
|
|
|
+ async def interaction_check(self, interaction: discord.Interaction) -> bool:
|
|
|
+ return interaction.user.id == self.owner_id
|
|
|
+
|
|
|
+
|
|
|
+def add_button(view: discord.ui.View, label: str, emoji: str, style: discord.ButtonStyle,
|
|
|
+ callback: Callback, disabled: bool = False) -> None:
|
|
|
+ button = discord.ui.Button(label=label, emoji=emoji, style=style, disabled=disabled)
|
|
|
+ button.callback = callback
|
|
|
+ view.add_item(button)
|
|
|
+
|
|
|
+
|
|
|
+async def confirm(interaction: discord.Interaction, card: MemoryCard, question: str, action: Callback) -> None:
|
|
|
+ """Remplace le fichier affiché par une question, avec « Confirmer » et « Annuler »."""
|
|
|
+ view = OwnerView(card.owner_id)
|
|
|
+ add_button(view, "Confirmer", "⚠️", discord.ButtonStyle.danger, action)
|
|
|
+ add_button(view, "Annuler", "↩️", discord.ButtonStyle.secondary, card.refresh)
|
|
|
+ embed = discord.Embed(description=question, color=discord.Color.orange())
|
|
|
+ await interaction.response.edit_message(content=None, embeds=[embed], view=view)
|
|
|
+
|
|
|
+
|
|
|
+class EditModal(discord.ui.Modal):
|
|
|
+ """Fenêtre « Modifier » : le fichier, déjà rempli, dans une ou plusieurs zones de texte
|
|
|
+ (une zone fait 4000 caractères au plus ; le fichier général peut en demander deux)."""
|
|
|
+
|
|
|
+ def __init__(self, card: MemoryCard):
|
|
|
+ title = "Modifier ta fiche" if card.target.is_self else (
|
|
|
+ "Modifier le fichier général" if card.target.member_id is None else f"Fiche de {card.target.name}")
|
|
|
+ super().__init__(title=title[:45], timeout=VIEW_TIMEOUT)
|
|
|
+ self.card = card
|
|
|
+ limit = card.store.limit(card.target.file)
|
|
|
+ count = min(MAX_INPUTS, -(-limit // INPUT_MAX))
|
|
|
+ parts = split_for_inputs(card.store.read(card.target.file), count)
|
|
|
+ self.inputs = []
|
|
|
+ for i in range(count):
|
|
|
+ label = "Contenu" if count == 1 else f"Contenu ({i + 1}/{count})"
|
|
|
+ text_input = discord.ui.TextInput(
|
|
|
+ label=label, style=discord.TextStyle.paragraph, required=False,
|
|
|
+ max_length=min(INPUT_MAX, limit), default=parts[i] or None,
|
|
|
+ placeholder="Une information par ligne. Tout vider efface le fichier." if i == 0 else None,
|
|
|
+ )
|
|
|
+ self.inputs.append(text_input)
|
|
|
+ self.add_item(text_input)
|
|
|
+
|
|
|
+ async def on_submit(self, interaction: discord.Interaction) -> None:
|
|
|
+ text = "\n".join(text_input.value.strip() for text_input in self.inputs if text_input.value.strip())
|
|
|
+ await self.card.save(interaction, text)
|
|
|
+
|
|
|
+ async def on_error(self, interaction: discord.Interaction, error: Exception) -> None:
|
|
|
+ logger.exception("Erreur pendant la modification de %s", self.card.target.file, exc_info=error)
|
|
|
+ if not interaction.response.is_done():
|
|
|
+ await interaction.response.send_message("La modification a échoué, regarde les logs.", ephemeral=True)
|
|
|
+
|
|
|
+
|
|
|
+def split_for_inputs(text: str, count: int) -> list[str]:
|
|
|
+ """Répartit un texte dans count zones de texte, de préférence entre les lignes."""
|
|
|
+ parts = split_message(text, INPUT_MAX)
|
|
|
+ if len(parts) > count: # lignes trop mal réparties : découpage exact
|
|
|
+ parts = [text[i:i + INPUT_MAX] for i in range(0, len(text), INPUT_MAX)]
|
|
|
+ return (parts + [""] * count)[:count]
|
|
|
+
|
|
|
+
|
|
|
+def member_target(store: MemoryStore, member: discord.User | discord.Member, owner_id: int) -> Target:
|
|
|
+ entry = store.members.get(member.id)
|
|
|
+ name = display_name(entry) if entry else member.display_name
|
|
|
+ return Target(store.member_file(member.id), member.id, name, is_self=member.id == owner_id)
|
|
|
+
|
|
|
+
|
|
|
+def server_target() -> Target:
|
|
|
+ return Target(SERVER_FILE, None)
|
Johnounet