From 52fd7bb7cb728b6bf068c75b6ab5d7a595c30413 Mon Sep 17 00:00:00 2001
From: Farhan Khan <farhan@farhan.codes>
Date: Sun, 6 Dec 2020 00:01:53 -0500
Subject: [PATCH] adding oauth file, need to get better at this

---
 poll/oauth.go | 138 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 138 insertions(+)
 create mode 100644 poll/oauth.go

diff --git a/poll/oauth.go b/poll/oauth.go
new file mode 100644
index 0000000..c3f3553
--- /dev/null
+++ b/poll/oauth.go
@@ -0,0 +1,138 @@
+package main
+
+import (
+	"net/http"
+	"encoding/json"
+	"bytes"
+	"log"
+	"io/ioutil"
+)
+
+type OAuth struct {
+	Access_token	string	`"json:access_token"`
+	Created_at	int	`"json:created_at"`
+	Expires_in	int64	`"json:Expires_in"`
+	Refresh_token	string	`"json:refresh_token"`
+}
+
+type authError struct {
+	msg	string
+}
+
+func (e *authError) Error() string {
+	return e.msg
+}
+
+func register_client(endpoint string, http_client *http.Client) (string, string, error) {
+	requestBodymap, _ := json.Marshal(map[string]string{
+		"client_name":   "Tusky", // Hard-coded in for now...
+		"scopes":        "read write follow push",
+		"redirect_uris": "urn:ietf:wg:oauth:2.0:oob",
+	})
+	requestBodybytes := bytes.NewBuffer(requestBodymap)
+
+	api_base_apps := "https://" + endpoint + "/api/v1/apps"
+
+	resp, err := http_client.Post(api_base_apps, "application/json", requestBodybytes)
+	if err != nil {
+		log.Fatal("Unable to connect to " + api_base_apps + " ", err)
+	}
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		log.Print("Unable to read HTTP response: ", err)
+		return "", "", err
+	}
+	defer resp.Body.Close()
+
+	bodymap := make(map[string]string)
+	err = json.Unmarshal(body, &bodymap)
+	if err != nil {
+		log.Print("Unable to Unmarshal response: ", err)
+		return "", "", err
+	}
+
+	return bodymap["client_id"], bodymap["client_secret"], nil
+}
+
+func oauth_login(endpoint string, username string, password string, client_id string, client_secret string) (OAuth, error) {
+	authMap, err := json.Marshal(map[string]string{
+		"username": username,
+		"password": password,
+		"redirect_uri": "urn:ietf:wg:oauth:2.0:oob",
+		"grant_type": "password",
+		"client_name": "Tusky",
+		"scope": "read write follow push",
+		"client_id": client_id,
+		"client_secret": client_secret,
+	})
+
+	if err != nil {
+		log.Print("Unable to create Authentication map")
+		return OAuth{}, err
+	}
+
+	authMapbytes := bytes.NewBuffer(authMap)
+
+	resp, err := http.Post("https://" + endpoint + "/oauth/token", "application/json", authMapbytes)
+	if err != nil {
+		log.Print("Cannot connect to " + endpoint + ": ", err)
+		return OAuth{}, err
+	}
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		log.Print("Unable to read response data: ", err)
+		return OAuth{}, err
+	}
+
+	if resp.StatusCode == 400 {
+		log.Print("Unable to authenticate")
+		return OAuth{}, &authError{"Authentication error"}
+	}
+//	log.Fatal(resp.StatusCode)
+
+	oauthData := OAuth{}
+	err = json.Unmarshal(body, &oauthData)
+	if err != nil {
+		log.Print("Unable to Unmarshal json data: ", err)
+		return OAuth{}, err
+	}
+
+	return oauthData, nil
+
+}
+
+func oauth_refresh(endpoint string, client_id string, client_secret string, refresh_token string) (OAuth, error) {
+	authMap, err := json.Marshal(map[string]string{
+		"redirect_uri": "urn:ietf:wg:oauth:2.0:oob",
+		"grant_type": "refresh_token",
+		"scope": "read write follow push",
+		"refresh_token": refresh_token,
+		"client_id": client_id,
+		"client_secret": client_secret,
+	})
+
+	authMapbytes := bytes.NewBuffer(authMap)
+
+	resp, err := http.Post("https://" + endpoint + "/oauth/token", "application/json", authMapbytes)
+	if err != nil {
+		log.Print("Cannot connect to " + endpoint + ": ", err)
+		return OAuth{}, err
+	}
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		log.Print("Unable to read response data: ", err)
+		return OAuth{}, err
+	}
+
+	oauthData := OAuth{}
+	err = json.Unmarshal(body, &oauthData)
+	if err != nil {
+		log.Print("Unable to Unmarshal json data: ", err)
+		return oauthData, err
+	}
+
+	return oauthData, nil
+}