Merge pull request #66 from niwaniwa/bugfix/nsfw-button

fix: nsfw button
niri-la · Jul 22, 2023 · 9d1cfec · 9d1cfec
2 parents 72083e5 + 159b55e
commit 9d1cfec
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/component.go b/component.go
@@ -161,7 +161,9 @@ func nsfwComponent() {
 				},
 			})
 			emoji.IsSensitive = true
-			ProcessRequest(emoji, s, i.ChannelID)
+			emoji.RequestState = "Nsfw"
+			emoji.ResponseState = "Nsfw"
+			ProcessNextRequest(emoji, s, i.ChannelID)
 
 		},
 	)
@@ -187,7 +189,6 @@ func nsfwComponent() {
 					"name":  emoji.Name,
 				}).Error(err)
 				return
-
 			}
 
 			if emoji.IsRequested {
@@ -210,7 +211,9 @@ func nsfwComponent() {
 			})
 
 			emoji.IsSensitive = false
-			ProcessRequest(emoji, s, i.ChannelID)
+			emoji.RequestState = "Nsfw"
+			emoji.ResponseState = "Nsfw"
+			ProcessNextRequest(emoji, s, i.ChannelID)
 
 		},
 	)

diff --git a/request-handler.go b/request-handler.go
@@ -472,7 +472,7 @@ func init() {
 	}
 }
 
-func ProcessRequest(emoji *Emoji, s *discordgo.Session, id string) bool {
+func ProcessNextRequest(emoji *Emoji, s *discordgo.Session, id string) bool {
 	requestIndex := reverseWorkflowMap[emoji.RequestState]
 	logger.WithFields(logrus.Fields{
 		"emoji id":       emoji.ID,