Merge pull request #16 from ascpial/fix-rolereact

[Fix] wrong defer type and wrong edit endpoint in rolereact

1 files changed, 17 insertions, 12 deletions

diff --git a/rolereact/views.go b/rolereact/views.go
index eafa4dc..3ecfe13 100644
--- a/rolereact/views.go
+++ b/rolereact/views.go
@@ -13,14 +13,11 @@ import (
 	"github.com/nyttikord/gokord/interaction"
 )
 
-func MessageModifyData(i *event.InteractionCreate, parameters *EditID) *interaction.ResponseData {
+func MessageModifyComponents(i *event.InteractionCreate, parameters *EditID) []component.Message {
 	message, ok := GetMessageFromEditID(i, parameters.MessageEditID)
 	if !ok {
-		return &interaction.ResponseData{
-			Flags: channel.MessageFlagsIsComponentsV2,
-			Components: []component.Component{
-				&component.TextDisplay{Content: "Cette modification est trop vieille et a été oubliée."},
-			},
+		return []component.Message{
+			&component.TextDisplay{Content: "Cette modification est trop vieille et a été oubliée."},
 		}
 	}
 	var note string
@@ -104,14 +101,22 @@ func MessageModifyData(i *event.InteractionCreate, parameters *EditID) *interact
 				},
 			},
 		}}...)
-	responseData := &interaction.ResponseData{
-		Flags: channel.MessageFlagsIsComponentsV2 | channel.MessageFlagsEphemeral,
-		Components: []component.Component{
-			&component.Container{
-				Components: components,
-			},
+	return []component.Message{
+		&component.Container{
+			Components: components,
 		},
 	}
+}
+
+func MessageModifyData(i *event.InteractionCreate, parameters *EditID) *interaction.ResponseData {
+	components := []component.Component{}
+	for _, component := range MessageModifyComponents(i, parameters) {
+		components = append(components, component)
+	}
+	responseData := &interaction.ResponseData{
+		Flags:      channel.MessageFlagsIsComponentsV2 | channel.MessageFlagsEphemeral,
+		Components: components,
+	}
 	return responseData
 }