more

endpoints/tokens.go

@@ -0,0 +1,68 @@
+package endpoints
+
+import (
+	"fmt"
+
+	"git.zomo.dev/zomo/discord-retokenizer/storage"
+	"github.com/gin-gonic/gin"
+)
+
+func getTokens(c *gin.Context) {
+	tokens := storage.GetTokens()
+	c.JSON(200, tokens)
+}
+
+func getToken(c *gin.Context) {
+	id := c.Param("id")
+	token := storage.GetToken(id)
+	c.JSON(200, token)
+}
+
+type TokenBody struct {
+	BotID string `json:"bot_id" binding:"required"`
+}
+
+func addToken(c *gin.Context) {
+	var tokenBody TokenBody
+	if err := c.BindJSON(&tokenBody); err != nil {
+		fmt.Println(err)
+		return
+	}
+	token := storage.GenerateToken(tokenBody.BotID)
+	if token == "" {
+		c.JSON(400, gin.H{
+			"message": "bot id not found",
+		})
+		return
+	}
+	c.JSON(200, gin.H{
+		"token": token,
+	})
+}
+
+func deleteToken(c *gin.Context) {
+	id := c.Param("id")
+	storage.DeleteToken(id)
+	c.JSON(200, gin.H{
+		"message": "success",
+	})
+}
+
+func updateToken(c *gin.Context) {
+	id := c.Param("id")
+	var tokenBody TokenBody
+	if err := c.BindJSON(&tokenBody); err != nil {
+		fmt.Println(err)
+		return
+	}
+	didSet := storage.UpdateToken(id, tokenBody.BotID)
+	if !didSet {
+		c.JSON(400, gin.H{
+			"message": "token or bot id not found",
+		})
+		return
+	}
+	c.JSON(200, gin.H{
+		"message": "success",
+	})
+}