-
Notifications
You must be signed in to change notification settings - Fork 28
/
token.go
63 lines (52 loc) · 1.55 KB
/
token.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
package oauth
import (
"errors"
"net/http"
"github.com/authgear/authgear-server/pkg/lib/infra/db/appdb"
"github.com/authgear/authgear-server/pkg/lib/oauth/protocol"
"github.com/authgear/authgear-server/pkg/util/httproute"
"github.com/authgear/authgear-server/pkg/util/httputil"
"github.com/authgear/authgear-server/pkg/util/log"
)
func ConfigureTokenRoute(route httproute.Route) httproute.Route {
return route.
WithMethods("POST", "OPTIONS").
WithPathPattern("/oauth2/token")
}
type ProtocolTokenHandler interface {
Handle(rw http.ResponseWriter, req *http.Request, r protocol.TokenRequest) httputil.Result
}
type TokenHandlerLogger struct{ *log.Logger }
func NewTokenHandlerLogger(lf *log.Factory) TokenHandlerLogger {
return TokenHandlerLogger{lf.New("handler-token")}
}
type TokenHandler struct {
Logger TokenHandlerLogger
Database *appdb.Handle
TokenHandler ProtocolTokenHandler
}
func (h *TokenHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
err := r.ParseForm()
if err != nil {
http.Error(rw, err.Error(), 400)
return
}
req := protocol.TokenRequest{}
for name, values := range r.Form {
req[name] = values[0]
}
var result httputil.Result
err = h.Database.WithTx(func() error {
result = h.TokenHandler.Handle(rw, r, req)
if result.IsInternalError() {
return errAuthzInternalError
}
return nil
})
if err == nil || errors.Is(err, errAuthzInternalError) {
result.WriteResponse(rw, r)
} else {
h.Logger.WithError(err).Error("oauth token handler failed")
http.Error(rw, "Internal Server Error", 500)
}
}