arimelody.me/discord/discord.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
}