59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
|
||
|
|
"github.com/localhots/steward/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
authURL = "https://github.com/login/oauth/authorize"
|
||
|
|
accessTokenURL = "https://github.com/login/oauth/access_token"
|
||
|
|
)
|
||
|
|
|
||
|
|
func authSigninHandler(w http.ResponseWriter, r *http.Request) {
|
||
|
|
params := url.Values{}
|
||
|
|
params.Set("client_id", config.C().ClientID)
|
||
|
|
params.Set("redirect_uri", config.C().RedirectURI)
|
||
|
|
params.Set("scope", "repo")
|
||
|
|
http.Redirect(w, r, authURL+"?"+params.Encode(), 302)
|
||
|
|
}
|
||
|
|
|
||
|
|
func authCallbackHandler(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if r.FormValue("error") != "" {
|
||
|
|
w.Write([]byte(r.FormValue("error_description")))
|
||
|
|
} else {
|
||
|
|
fmt.Println("Got code: ", r.FormValue("code"))
|
||
|
|
token := getAccessToken(r.FormValue("code"))
|
||
|
|
fmt.Println("Got access token: ", token)
|
||
|
|
w.Write([]byte(token))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func getAccessToken(code string) string {
|
||
|
|
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)
|
||
|
|
|
||
|
|
buf := bytes.NewBuffer([]byte(payload.Encode()))
|
||
|
|
resp, err := http.Post(accessTokenURL, "application/x-www-form-urlencoded", buf)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
defer resp.Body.Close()
|
||
|
|
body, err := ioutil.ReadAll(resp.Body)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
pairs, _ := url.ParseQuery(string(body))
|
||
|
|
return pairs.Get("access_token")
|
||
|
|
}
|