88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
package task
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/localhots/empact/config"
|
|
"github.com/localhots/empact/db"
|
|
)
|
|
|
|
func Authenticate(code string) (token, login string, err error) {
|
|
defer report(time.Now(), "Authenticate")
|
|
if token, err = FetchAccessToken(code); err != nil {
|
|
return
|
|
}
|
|
log.Printf("Got token %q for code %q\n", token, code)
|
|
|
|
var user *db.User
|
|
if user, err = FetchUserInfoWithToken(token); err != nil {
|
|
return
|
|
}
|
|
login = user.Login
|
|
log.Println("Saving user", user)
|
|
db.Queue(func() { user.Save() })
|
|
|
|
tok := &db.Token{
|
|
User: login,
|
|
Token: token,
|
|
}
|
|
log.Println("Saving token", tok)
|
|
db.Queue(func() { tok.Save() })
|
|
|
|
return
|
|
}
|
|
|
|
func FetchAccessToken(code string) (token string, err error) {
|
|
defer report(time.Now(), "FetchAccessToken")
|
|
payload := url.Values{}
|
|
payload.Set("client_id", config.C().ClientID)
|
|
payload.Set("client_secret", config.C().ClientSecret)
|
|
payload.Set("code", code)
|
|
payload.Set("redirect_uri", config.C().RedirectURI)
|
|
|
|
log.Printf("Requesting token for code %q", code)
|
|
buf := bytes.NewBuffer([]byte(payload.Encode()))
|
|
var resp *http.Response
|
|
if resp, err = http.Post(config.C().AccessTokenURL, "application/x-www-form-urlencoded", buf); err != nil {
|
|
return
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
var body []byte
|
|
if body, err = ioutil.ReadAll(resp.Body); err != nil {
|
|
return
|
|
}
|
|
|
|
pairs, _ := url.ParseQuery(string(body))
|
|
if token = pairs.Get("access_token"); token == "" {
|
|
err = fmt.Errorf("Failed to fetch access token usign code %q: %s", code, pairs.Get("error_description"))
|
|
}
|
|
return
|
|
}
|
|
|
|
func FetchUserInfoWithToken(token string) (u *db.User, err error) {
|
|
defer report(time.Now(), "FetchUserInfoWithToken")
|
|
var resp *http.Response
|
|
if resp, err = http.Get("https://api.github.com/user?access_token=" + token); err != nil {
|
|
return
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
var body []byte
|
|
if body, err = ioutil.ReadAll(resp.Body); err != nil {
|
|
return
|
|
}
|
|
|
|
u = &db.User{}
|
|
err = json.Unmarshal(body, &u)
|
|
|
|
return
|
|
}
|