refactor: Clean up random token implementation

This commit is contained in:
Matthew
2023-11-27 03:03:48 -05:00
parent 523266fc68
commit a4745f05d5

View File

@@ -321,9 +321,16 @@ func parseChatCommand(s *Session, command string) {
} }
case commands["Discord"].Prefix: case commands["Discord"].Prefix:
if commands["Discord"].Enabled { if commands["Discord"].Enabled {
discordToken := make([]byte, 8) randToken := make([]byte, 4)
_, err := rand.Read(discordToken)
s.logger.Info(fmt.Sprint(discordToken)) _, err := rand.Read(randToken)
if err != nil {
sendServerChatMessage(s, fmt.Sprint("An error occurred while processing this command"))
s.logger.Error(fmt.Sprint(err))
return
}
discordToken := fmt.Sprintf("%x-%x", randToken[:2], randToken[2:])
_, err = s.server.db.Exec("UPDATE users u SET discord_token = $1 WHERE u.id=(SELECT c.user_id FROM characters c WHERE c.id=$2)", fmt.Sprint(discordToken), s.charID) _, err = s.server.db.Exec("UPDATE users u SET discord_token = $1 WHERE u.id=(SELECT c.user_id FROM characters c WHERE c.id=$2)", fmt.Sprint(discordToken), s.charID)
if err != nil { if err != nil {
sendServerChatMessage(s, fmt.Sprint("An error occurred while processing this command")) sendServerChatMessage(s, fmt.Sprint("An error occurred while processing this command"))