wip - more oauth work

server/webserver/discord/oauth.go

@@ -0,0 +1,66 @@
+package discord
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strings"
+
+	"github.com/mgerb/go-discord-bot/server/config"
+)
+
+const discordAPI = "https://discordapp.com/api/v6"
+
+// OauthResp -
+type OauthResp 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"`
+}
+
+// Oauth -
+func Oauth(code string) (OauthResp, error) {
+
+	form := url.Values{}
+
+	form.Set("client_id", config.Config.ClientID)
+	form.Set("client_secret", config.Config.ClientSecret)
+	form.Set("grant_type", "authorization_code")
+	form.Set("code", code)
+	form.Set("redirect_uri", config.Config.RedirectURI)
+
+	req, err := http.NewRequest("POST", discordAPI+"/oauth2/token", strings.NewReader(form.Encode()))
+
+	if err != nil {
+		return OauthResp{}, err
+	}
+
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+
+	resp, err := http.DefaultClient.Do(req)
+
+	if err != nil {
+		return OauthResp{}, err
+	}
+
+	defer resp.Body.Close()
+
+	data, err := ioutil.ReadAll(resp.Body)
+
+	if err != nil {
+		return OauthResp{}, err
+	}
+
+	var oauth OauthResp
+
+	err = json.Unmarshal(data, &oauth)
+
+	if err != nil {
+		return OauthResp{}, err
+	}
+
+	return oauth, nil
+}

server/webserver/discord/user.go

@@ -0,0 +1,60 @@
+package discord
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+
+	log "github.com/sirupsen/logrus"
+)
+
+// User -
+type User struct {
+	Username      string `json:"username"`
+	Verified      bool   `json:"verified"`
+	MFAEnabled    bool   `json:"mfa_enabled"`
+	ID            string `json:"id"`
+	Avatar        string `json:"avatar"`
+	Discriminator string `json:"discriminator"`
+	Email         string `json:"email"`
+}
+
+// GetUserInfo - get user info
+func GetUserInfo(accessToken string) (User, error) {
+	req, err := http.NewRequest("GET", discordAPI+"/users/@me", nil)
+
+	if err != nil {
+		log.Error(err)
+		return User{}, err
+	}
+
+	req.Header.Add("Authorization", "Bearer "+accessToken)
+
+	resp, err := http.DefaultClient.Do(req)
+
+	if err != nil {
+		log.Error(err)
+		return User{}, err
+	}
+
+	defer resp.Body.Close()
+
+	data, err := ioutil.ReadAll(resp.Body)
+
+	if err != nil {
+		log.Error(err)
+		return User{}, err
+	}
+
+	var userInfo User
+
+	err = json.Unmarshal(data, &userInfo)
+
+	if err != nil {
+		log.Error(err)
+		return User{}, err
+	}
+
+	// filter guild based on id
+	return userInfo, nil
+}