127 lines
3.6 KiB
Go
127 lines
3.6 KiB
Go
package discord
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
const API_ENDPOINT = "https://discord.com/api/v10"
|
|
|
|
var CLIENT_ID = func() string {
|
|
envvar := os.Getenv("DISCORD_CLIENT_ID")
|
|
if envvar == "" {
|
|
fmt.Printf("DISCORD_CLIENT_ID was not provided. Admin login will be unavailable.\n")
|
|
}
|
|
return envvar
|
|
}()
|
|
var CLIENT_SECRET = func() string {
|
|
envvar := os.Getenv("DISCORD_CLIENT_SECRET")
|
|
if envvar == "" {
|
|
fmt.Printf("DISCORD_CLIENT_SECRET was not provided. Admin login will be unavailable.\n")
|
|
}
|
|
return envvar
|
|
}()
|
|
var REDIRECT_URI = func() string {
|
|
envvar := os.Getenv("DISCORD_REDIRECT_URI")
|
|
if envvar == "" {
|
|
fmt.Printf("DISCORD_REDIRECT_URI was not provided. Admin login will be unavailable.\n")
|
|
}
|
|
return envvar
|
|
}()
|
|
var OAUTH_CALLBACK_URI = func() string {
|
|
envvar := os.Getenv("OAUTH_CALLBACK_URI")
|
|
if envvar == "" {
|
|
fmt.Printf("OAUTH_CALLBACK_URI was not provided. Admin login will be unavailable.\n")
|
|
}
|
|
return envvar
|
|
}()
|
|
|
|
type (
|
|
AccessTokenResponse struct {
|
|
TokenType string `json:"token_type"`
|
|
AccessToken string `json:"access_token"`
|
|
ExpiresIn int `json:"expires_in"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
Scope string `json:"scope"`
|
|
}
|
|
|
|
AuthInfoResponse struct {
|
|
Application struct {
|
|
Id string
|
|
Name string
|
|
Icon string
|
|
Description string
|
|
Hook bool
|
|
BotPublic bool
|
|
botRequireCodeGrant bool
|
|
VerifyKey bool
|
|
}
|
|
Scopes []string
|
|
Expires string
|
|
User DiscordUser
|
|
}
|
|
|
|
DiscordUser struct {
|
|
Id string
|
|
Username string
|
|
Avatar string
|
|
Discriminator string
|
|
GlobalName string
|
|
PublicFlags int
|
|
}
|
|
)
|
|
|
|
func GetOAuthTokenFromCode(code string) (string, error) {
|
|
// let's get an oauth token!
|
|
req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("%s/oauth2/token", API_ENDPOINT),
|
|
strings.NewReader(url.Values{
|
|
"client_id": {CLIENT_ID},
|
|
"client_secret": {CLIENT_SECRET},
|
|
"grant_type": {"authorization_code"},
|
|
"code": {code},
|
|
"redirect_uri": {OAUTH_CALLBACK_URI},
|
|
}.Encode()))
|
|
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
|
|
|
|
res, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return "", errors.New(fmt.Sprintf("Failed while contacting discord API: %s", err))
|
|
}
|
|
|
|
oauth := AccessTokenResponse{}
|
|
|
|
err = json.NewDecoder(res.Body).Decode(&oauth)
|
|
if err != nil {
|
|
return "", errors.New(fmt.Sprintf("Failed to parse OAuth response data from discord: %s\n", err))
|
|
}
|
|
res.Body.Close()
|
|
|
|
return oauth.AccessToken, nil
|
|
}
|
|
|
|
func GetDiscordUserFromAuth(token string) (DiscordUser, error) {
|
|
// let's get authorisation information!
|
|
req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%s/oauth2/@me", API_ENDPOINT), nil)
|
|
req.Header.Add("Authorization", "Bearer " + token)
|
|
|
|
res, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return DiscordUser{}, errors.New(fmt.Sprintf("Failed to retrieve discord auth information: %s\n", err))
|
|
}
|
|
|
|
auth_info := AuthInfoResponse{}
|
|
|
|
err = json.NewDecoder(res.Body).Decode(&auth_info)
|
|
if err != nil {
|
|
return DiscordUser{}, errors.New(fmt.Sprintf("Failed to parse auth information from discord: %s\n", err))
|
|
}
|
|
defer res.Body.Close()
|
|
|
|
return auth_info.User, nil
|
|
}
|