133 lines
3.1 KiB
Go
133 lines
3.1 KiB
Go
package root
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/TicketsBot/GoPanel/cache"
|
|
"github.com/TicketsBot/GoPanel/config"
|
|
"github.com/TicketsBot/GoPanel/database/table"
|
|
"github.com/TicketsBot/GoPanel/utils"
|
|
"github.com/TicketsBot/GoPanel/utils/discord"
|
|
"github.com/TicketsBot/GoPanel/utils/discord/endpoints/user"
|
|
"github.com/TicketsBot/GoPanel/utils/discord/objects"
|
|
"github.com/apex/log"
|
|
"github.com/gin-gonic/contrib/sessions"
|
|
"github.com/gin-gonic/gin"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type (
|
|
TokenData struct {
|
|
ClientId string `qs:"client_id"`
|
|
ClientSecret string `qs:"client_secret"`
|
|
GrantType string `qs:"grant_type"`
|
|
Code string `qs:"code"`
|
|
RedirectUri string `qs:"redirect_uri"`
|
|
Scope string `qs:"scope"`
|
|
}
|
|
|
|
TokenResponse struct {
|
|
AccessToken string `json:"access_token"`
|
|
TokenType string `json:"token_type"`
|
|
ExpiresIn int `json:"expires_in"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
Scope string `json:"scope"`
|
|
}
|
|
)
|
|
|
|
func CallbackHandler(ctx *gin.Context) {
|
|
store := sessions.Default(ctx)
|
|
if store == nil {
|
|
return
|
|
}
|
|
defer store.Save()
|
|
|
|
if utils.IsLoggedIn(store) {
|
|
ctx.Redirect(302, config.Conf.Server.BaseUrl)
|
|
return
|
|
}
|
|
|
|
code := ctx.DefaultQuery("code", "")
|
|
if code == "" {
|
|
ctx.String(400, "Discord provided an invalid Oauth2 code")
|
|
return
|
|
}
|
|
|
|
res, err := discord.AccessToken(code)
|
|
if err != nil {
|
|
ctx.String(500, err.Error())
|
|
}
|
|
|
|
store.Set("access_token", res.AccessToken)
|
|
store.Set("refresh_token", res.RefreshToken)
|
|
store.Set("expiry", (time.Now().UnixNano()/int64(time.Second))+int64(res.ExpiresIn))
|
|
|
|
// Get ID + name
|
|
var currentUser objects.User
|
|
err, _ = user.CurrentUser.Request(store, nil, nil, ¤tUser)
|
|
if err != nil {
|
|
ctx.String(500, err.Error())
|
|
return
|
|
}
|
|
|
|
store.Set("csrf", utils.RandStringRunes(32))
|
|
|
|
store.Set("userid", currentUser.Id)
|
|
store.Set("name", currentUser.Username)
|
|
store.Set("avatar", fmt.Sprintf("https://cdn.discordapp.com/avatars/%s/%s.webp", currentUser.Id, currentUser.Avatar))
|
|
if err = store.Save(); err != nil {
|
|
log.Error(err.Error())
|
|
}
|
|
|
|
ctx.Redirect(302, config.Conf.Server.BaseUrl)
|
|
|
|
userId, err := strconv.ParseUint(currentUser.Id, 10, 64); if err != nil { // ???
|
|
return
|
|
}
|
|
|
|
// Cache guilds because Discord takes like 2 whole seconds to return then
|
|
go func() {
|
|
var guilds []objects.Guild
|
|
err, _ = user.CurrentUserGuilds.Request(store, nil, nil, &guilds)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
return
|
|
}
|
|
|
|
for _, guild := range guilds {
|
|
go cache.Client.StoreGuild(guild)
|
|
|
|
// cache roles
|
|
guildId, err := strconv.ParseUint(guild.Id, 10, 64); if err != nil {
|
|
continue
|
|
}
|
|
go cacheRoles(store, guildId, userId)
|
|
}
|
|
|
|
marshalled, err := json.Marshal(guilds)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
return
|
|
}
|
|
|
|
table.UpdateGuilds(currentUser.Id, base64.StdEncoding.EncodeToString(marshalled))
|
|
}()
|
|
}
|
|
|
|
func cacheRoles(store sessions.Session, guildId, userId uint64) {
|
|
roles := utils.GetRolesRest(store, guildId, userId)
|
|
|
|
if roles == nil {
|
|
return
|
|
}
|
|
|
|
// Delete old roles
|
|
table.DeleteRoles(guildId, userId)
|
|
|
|
for _, role := range *roles {
|
|
table.CacheRole(guildId, userId, role)
|
|
}
|
|
}
|