plugins: (role-menu) Fix issues with creating multiple menus in one g…

…uild
5HT2 · Jun 9, 2022 · 9098433 · 9098433
1 parent fc249ec
commit 9098433
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/plugins/role-menu/role-menu.go b/plugins/role-menu/role-menu.go
@@ -19,7 +19,7 @@ import (
 var p *plugins.Plugin
 
 type config struct {
-	Menus map[string]map[string]Menu `json:"menus,omitempty"` // [guild id][message id]Menu
+	Menus map[string]map[string]Menu `json:"menus"` // [guild id][message id]Menu
 }
 
 type Menu struct {
@@ -115,12 +115,23 @@ func RoleMenuCommand(c bot.Command) error {
 			//
 			// Save final menu in config
 
-			finalMenu := Menu{Channel: int64(c.E.ChannelID), Roles: roles}
-			messageMenu := make(map[string]Menu)
-			messageMenu[msg.ID.String()] = finalMenu
-
 			menus := make(map[string]map[string]Menu, 0)
-			menus[c.E.GuildID.String()] = messageMenu
+			if p.Config != nil {
+				if menu, ok := p.Config.(config).Menus[c.E.GuildID.String()]; ok {
+					menus[c.E.GuildID.String()] = menu
+				}
+			}
+
+			createdMenu := Menu{Channel: int64(c.E.ChannelID), Roles: roles}
+
+			if _, ok := menus[c.E.GuildID.String()]; ok {
+				menus[c.E.GuildID.String()][msg.ID.String()] = createdMenu
+			} else {
+				messageMenu := make(map[string]Menu)
+				messageMenu[msg.ID.String()] = createdMenu
+				menus[c.E.GuildID.String()] = messageMenu
+			}
+
 			p.Config = config{Menus: menus}
 
 			// Add reactions to menu