generated from datewu/project-abc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
middleware.go
68 lines (64 loc) · 1.47 KB
/
middleware.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
package api
import (
"context"
"net/http"
"github.com/datewu/gtea"
"github.com/datewu/gtea/handler"
"github.com/datewu/set-img/internal/auth"
"github.com/datewu/set-img/internal/auth/github"
)
func (k *k8sHandler) auth(next http.HandlerFunc) http.HandlerFunc {
middle := func(w http.ResponseWriter, r *http.Request) {
if k.app.Env() == gtea.DevEnv {
k.user = "datewu"
next(w, r)
return
}
user := handler.ReadQuery(r, "user", "")
if user == "" {
handler.BadRequestMsg(w, "missing github user query")
return
}
token, err := handler.GetToken(r, "token")
if err != nil {
handler.BadRequestMsg(w, "missing github token query/header/cookie")
return
}
ok, err := auth.Valid(context.Background(), auth.GithubAuth, user, token)
if err != nil {
handler.ServerErr(w, err)
return
}
if !ok {
handler.InvalidAuthenticationToken(w)
return
}
next(w, r)
}
return middle
}
func (m *myHandler) auth(next http.HandlerFunc) http.HandlerFunc {
middle := func(w http.ResponseWriter, r *http.Request) {
if m.app.Env() == gtea.DevEnv {
m.user = "datewu"
next(w, r)
return
}
co, err := r.Cookie(github.CookieName)
if err != nil {
handler.BadRequestMsg(w, "missing github access_token cookie")
return
}
t := co.Value
user, err := github.GetUser(t)
if err != nil {
handler.ClearSimpleCookie(w, github.CookieName)
handler.ServerErr(w, err)
return
}
m.user = user.Login
m.token = t
next(w, r)
}
return middle
}