-
Notifications
You must be signed in to change notification settings - Fork 1
/
auth.go
122 lines (111 loc) · 3.38 KB
/
auth.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package main
import (
"fmt"
"math"
"net/http"
"os"
"github.com/gorilla/mux"
"github.com/gorilla/sessions"
"github.com/markbates/goth"
"github.com/markbates/goth/gothic"
"github.com/markbates/goth/providers/twitter"
"github.com/pkg/errors"
)
var sessionName = "ap-user-session"
var store *sessions.FilesystemStore
func initAuth(conf Conf) {
store := sessions.NewFilesystemStore(os.TempDir(), []byte(conf.SessionSecret))
store.MaxLength(math.MaxInt64)
gothic.Store = store
goth.UseProviders(
twitter.New(
conf.Env["TWITTER_KEY"],
conf.Env["TWITTER_SECRET"],
fmt.Sprintf("%s/auth/twitter/callback", conf.URL)),
)
}
func addAuthRoutes(r *mux.Router) *mux.Router {
s := r.PathPrefix("/auth/{provider}").Subrouter()
s.HandleFunc("/callback", authCallbackHandler).Methods("GET")
s.HandleFunc("/logout", logoutHandler).Methods("GET")
s.HandleFunc("", authHandler).Methods("GET")
return r
}
func logoutHandler(w http.ResponseWriter, r *http.Request) {
session, err := gothic.Store.Get(r, sessionName)
if err != nil {
log.Warningln("logoutHandler: Error when session get():", err)
}
session.Values["userid"] = ""
session.Values["twitter"] = ""
session.Values["access_token"] = ""
session.Values["access_token_secret"] = ""
err = session.Save(r, w)
if err != nil {
log.Warningln("logoutHandler: Error on session save: ", err)
}
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
}
func authHandler(w http.ResponseWriter, r *http.Request) {
if gothUser, err := gothic.CompleteUserAuth(w, r); err == nil {
session, err := gothic.Store.Get(r, sessionName)
if err != nil {
log.Warningln("Error when session get():", err)
}
//session.Values["gothUser"] = gothUser
session.Values["userid"] = gothUser.UserID
session.Values["twitter"] = gothUser.NickName
session.Values["access_token"] = gothUser.AccessToken
session.Values["access_token_secret"] = gothUser.AccessTokenSecret
err = session.Save(r, w)
if err != nil {
log.Warningln("Error on session save: ", err)
}
} else {
gothic.BeginAuthHandler(w, r)
}
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
}
func authCallbackHandler(w http.ResponseWriter, r *http.Request) {
gothUser, err := gothic.CompleteUserAuth(w, r)
if err != nil {
log.Warningln("Error completing auth: ", err)
return
}
session, err := gothic.Store.Get(r, sessionName)
if err != nil {
log.Warningln("Error when calling session get():", err)
}
session.Values["userid"] = gothUser.UserID
session.Values["twitter"] = gothUser.NickName
session.Values["access_token"] = gothUser.AccessToken
session.Values["access_token_secret"] = gothUser.AccessTokenSecret
err = session.Save(r, w)
if err != nil {
log.Warningln("Error when saving session: ", err)
}
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
}
func getSessionValues(r *http.Request) map[string]string {
session, err := gothic.Store.Get(r, sessionName)
if err != nil {
log.Error(errors.WithStack(err))
}
vals := make(map[string]string)
vals["userid"] = getString("userid", session.Values)
vals["twitter"] = getString("twitter", session.Values)
vals["access_token"] = getString("access_token", session.Values)
vals["access_token_secret"] = getString("access_token_secret", session.Values)
return vals
}
func getString(key string, m map[interface{}]interface{}) string {
i, ok := m[key]
if !ok {
return ""
}
s, ok := i.(string)
if !ok {
return ""
}
return s
}