90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
package endpoints
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.zomo.dev/zomo/discord-retokenizer/storage"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
|
|
type LoginPasswordBody struct {
|
|
Username string `json:"username" binding:"required"`
|
|
Password string `json:"password" binding:"required"`
|
|
}
|
|
|
|
type LoginTokenBody struct {
|
|
Token string `json:"token" binding:"required"`
|
|
}
|
|
|
|
func loginPassword(c *gin.Context) {
|
|
var loginBody LoginPasswordBody
|
|
if err := c.BindJSON(&loginBody); err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
userAgentString := c.GetHeader("User-Agent")
|
|
if userAgentString == "" {
|
|
return
|
|
}
|
|
ua := storage.ParseUA(userAgentString)
|
|
|
|
loggedIn, token := storage.CheckLoginPassword(loginBody.Username, loginBody.Password, c.ClientIP(), ua)
|
|
|
|
if loggedIn {
|
|
c.JSON(200, gin.H{
|
|
"token": token,
|
|
})
|
|
} else {
|
|
c.JSON(401, gin.H{
|
|
"error": "invalid username or password",
|
|
})
|
|
}
|
|
}
|
|
|
|
|
|
func loginToken(c *gin.Context) {
|
|
var loginBody LoginTokenBody
|
|
if err := c.BindJSON(&loginBody); err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
userAgentString := c.GetHeader("User-Agent")
|
|
if userAgentString == "" {
|
|
return
|
|
}
|
|
ua := storage.ParseUA(userAgentString)
|
|
|
|
loggedIn := storage.CheckLoginToken(loginBody.Token, c.ClientIP(), ua)
|
|
|
|
if loggedIn {
|
|
c.JSON(200, gin.H{
|
|
"token": loginBody.Token,
|
|
})
|
|
} else {
|
|
c.JSON(401, gin.H{
|
|
"error": "invalid username or password",
|
|
})
|
|
}
|
|
}
|
|
|
|
func updateLogin(c *gin.Context) {
|
|
var updateLogin LoginPasswordBody
|
|
if err := c.BindJSON(&updateLogin); err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
storage.UpdateUsername(updateLogin.Username)
|
|
storage.UpdatePassword(updateLogin.Password)
|
|
}
|
|
|
|
func getLoginTokens(c *gin.Context) {
|
|
tokens := storage.GetLoginTokensSimple()
|
|
c.JSON(200, tokens)
|
|
}
|
|
|
|
func clearLoginTokens(c *gin.Context) {
|
|
storage.ClearLoginTokens()
|
|
} |