now supports multiple IDs and restructures the json send

vaporvee · Aug 18, 2023 · aa486d3 · aa486d3
1 parent 68cb2dd
commit aa486d3
Showing 1 changed file with 117 additions and 31 deletions.
diff --git a/extended-user-info-api/main.go b/extended-user-info-api/main.go
@@ -2,50 +2,136 @@ package main
 
 import (
 	"encoding/json"
-	"fmt"
 	"io"
 	"net/http"
 	"os"
+	"strconv"
 	"strings"
+	"time"
 
 	"github.com/gin-gonic/gin"
 	"github.com/joho/godotenv"
 )
 
+type User struct {
+	GlobalName          string `json:"global_name"`
+	PublicFlags         int    `json:"public_flags"`
+	Flags               int    `json:"flags"`
+	AccentColor         string `json:"accent_color"`
+	AvatarDecorationURL string `json:"avatar_decoration"`
+	BannerURL           string `json:"banner_url"`
+	BannerColor         string `json:"banner_color"`
+}
+
 func main() {
 	r := gin.Default()
-	r.GET("/*userID", func(c *gin.Context) {
-		if c.Param("userID") == "/" {
-			c.Writer.WriteHeader(http.StatusOK)
-			c.Writer.Write([]byte("<html><body>Please provide a Discord user id after the slash. <a href=\"https://github.com/vaporvee/discord-sdk-godot/\">GitHub project</a></body></html>"))
+	r.GET("/:userIDs", func(c *gin.Context) {
+		userIDs := strings.Split(c.Param("userIDs"), ",")
+
+		if len(userIDs) == 0 {
+			c.JSON(http.StatusBadRequest, gin.H{"error": "No user IDs provided"})
 			return
-		} else {
-			godotenv.Load()
-			bot_token := os.Getenv("BOT_TOKEN")
-			var params string
-			params = strings.TrimLeft(c.Param("userID"), "/")
-			client := &http.Client{}
-			req, err := http.NewRequest("GET", "https://discord.com/api/v10/users/"+params, nil)
-			if err != nil {
-				fmt.Println("Error creating HTTP request:", err)
-				return
-			}
-			req.Header.Add("Authorization", "Bot "+bot_token)
-			req.Header.Add("Content-Type", "application/json")
-			res, err := client.Do(req)
-			if err != nil {
-				fmt.Println("Error sending HTTP request:", err)
-				return
-			}
-			body, err := io.ReadAll(res.Body)
-			if err != nil {
-				fmt.Println("Error reading HTTP response body:", err)
-				return
+		}
+
+		godotenv.Load()
+		botToken := os.Getenv("BOT_TOKEN")
+		client := &http.Client{}
+
+		userResponses := make(map[string]User)
+
+		for _, userID := range userIDs {
+			for {
+				req, err := http.NewRequest("GET", "https://discord.com/api/v10/users/"+userID, nil)
+				if err != nil {
+					c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+					return
+				}
+				req.Header.Add("Authorization", "Bot "+botToken)
+				req.Header.Add("Content-Type", "application/json")
+
+				res, err := client.Do(req)
+				if err != nil {
+					c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+					return
+				}
+				defer res.Body.Close()
+
+				if res.StatusCode == http.StatusTooManyRequests {
+					retryAfter := parseRetryAfterHeader(res.Header)
+					if retryAfter > 0 {
+						time.Sleep(retryAfter)
+						continue // Retry the request after waiting
+					}
+				}
+
+				body, err := io.ReadAll(res.Body)
+				if err != nil {
+					c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+					return
+				}
+
+				var jsonMap map[string]interface{}
+				if err := json.Unmarshal(body, &jsonMap); err != nil {
+					c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+					return
+				}
+
+				user := User{
+					GlobalName:  getStringFromMap(jsonMap, "global_name"),
+					PublicFlags: getIntFromMap(jsonMap, "public_flags"),
+					Flags:       getIntFromMap(jsonMap, "flags"),
+					AccentColor: getStringFromMap(jsonMap, "accent_color"),
+					BannerColor: getStringFromMap(jsonMap, "banner_color"),
+				}
+
+				if avatarDecoration, ok := jsonMap["avatar_decoration"].(string); ok {
+					user.AvatarDecorationURL = "https://cdn.discordapp.com/avatar-decorations/" + userID + "/" + avatarDecoration + ".png"
+				}
+
+				if banner, ok := jsonMap["banner"].(string); ok {
+					user.BannerURL = "https://cdn.discordapp.com/banners/" + userID + "/" + banner + ".png"
+				}
+
+				userResponses[userID] = user
+				break // Exit the loop and process the next user ID
 			}
-			var jsonMap map[string]interface{}
-			json.Unmarshal([]byte(string(body)), &jsonMap)
-			c.JSON(http.StatusOK, jsonMap)
 		}
+
+		c.JSON(http.StatusOK, userResponses)
 	})
-	r.Run()
+
+	r.GET("/", func(c *gin.Context) {
+		c.Writer.WriteHeader(http.StatusOK)
+		c.Writer.Write([]byte("<html><body>Please provide a comma-separated list of Discord user IDs or a single user ID in the URL. Example: /688778053408784415,369216554950328320 <br> <a href=\"https://github.com/vaporvee/discord-sdk-godot/\">GitHub project</a></body></html>"))
+	})
+
+	r.Run(":8080")
+}
+
+func parseRetryAfterHeader(headers http.Header) time.Duration {
+	retryAfterStr := headers.Get("Retry-After")
+	if retryAfterStr == "" {
+		return 0
+	}
+
+	retryAfter, err := strconv.Atoi(retryAfterStr)
+	if err != nil {
+		return 0
+	}
+
+	return time.Duration(retryAfter) * time.Millisecond
+}
+
+func getStringFromMap(m map[string]interface{}, key string) string {
+	if val, ok := m[key].(string); ok {
+		return val
+	}
+	return ""
+}
+
+func getIntFromMap(m map[string]interface{}, key string) int {
+	if val, ok := m[key].(float64); ok {
+		return int(val)
+	}
+	return 0
 }