From 5f93e13469d36d25d0f07b549d2c1ef3c6cd57cb Mon Sep 17 00:00:00 2001
From: Conatum <cona@thewisewolf.dev>
Date: Fri, 29 Sep 2023 15:21:01 +0300
Subject: [PATCH] fix(rmenus): Implement editmenu style pathway.

---
 src/modules/rolemenus/cog.py | 41 +++++++++++++++++++++++++++++++++++-
 1 file changed, 40 insertions(+), 1 deletion(-)

diff --git a/src/modules/rolemenus/cog.py b/src/modules/rolemenus/cog.py
index 0fdc944d..588a45a2 100644
--- a/src/modules/rolemenus/cog.py
+++ b/src/modules/rolemenus/cog.py
@@ -971,7 +971,41 @@ class RoleMenuCog(LionCog):
                 )
                 # TODO: Generate the custom message from the template if it doesn't exist
 
-        # TODO: Pathway for setting menu style
+        if menu_style is not None:
+            if not managed and not reposting:
+                raise UserInputError(
+                    t(_p(
+                        'cmd:rolemenu_edit|parse:style|error:not_managed',
+                        "Cannot change the style of a role menu attached to a message I did not send."
+                    ))
+                )
+            if menu_style is MenuType.REACTION:
+                # Check menu is suitable for moving to reactions
+                roles = target.roles
+                if len(roles) > 20:
+                    raise UserInputError(
+                        t(_p(
+                            'cmd:rolemenu_edit|parse:style|error:too_many_reactions',
+                            "Too many roles! Reaction role menus can have at most `20` roles."
+                        ))
+                    )
+                emojis = [mrole.config.emoji.value for mrole in roles]
+                emojis = [emoji for emoji in emojis if emoji]
+                uniq = set(emojis)
+                if len(uniq) != len(roles):
+                    raise UserInputError(
+                        t(_p(
+                            "cmd:rolemenu_edit|parse:style|error:incomplete_emojis",
+                            "Cannot switch to the reaction role style! Every role needs a distinct emoji first."
+                        ))
+                    )
+            update_args[self.data.RoleMenu.menutype.name] = menu_style
+            ack_lines.append(
+                t(_p(
+                    'cmd:rolemenu_edit|parse:style|success',
+                    "Updated role menu style."
+                ))
+            )
 
         if rawmessage is not None:
             msg_config = target.config.rawmessage
@@ -1019,6 +1053,11 @@ class RoleMenuCog(LionCog):
                 )).format(channel=channel.mention, exception=e.text))
         else:
             await target.update_message()
+            if menu_style is not None:
+                try:
+                    await target.update_reactons()
+                except SafeCancellation as e:
+                    error_lines.append(e.msg)
 
         # Ack the updates
         if ack_lines or error_lines: