forked from rancher/rancher
-
Notifications
You must be signed in to change notification settings - Fork 1
/
api.go
79 lines (65 loc) · 2.18 KB
/
api.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
package tokens
import (
"context"
"net/http"
normanapi "github.com/rancher/norman/api"
"github.com/rancher/norman/httperror"
"github.com/rancher/norman/types"
managementSchema "github.com/rancher/types/apis/management.cattle.io/v3/schema"
"github.com/rancher/types/client/management/v3"
"github.com/rancher/types/config"
"github.com/sirupsen/logrus"
)
const (
CookieName = "R_SESS"
AuthHeaderName = "Authorization"
AuthValuePrefix = "Bearer"
BasicAuthPrefix = "Basic"
)
var crdVersions = []*types.APIVersion{
&managementSchema.Version,
}
func NewAPIHandler(ctx context.Context, apiContext *config.ScaledContext) (http.Handler, error) {
api := &tokenAPI{
mgr: NewManager(ctx, apiContext),
}
schemas := types.NewSchemas().AddSchemas(managementSchema.TokenSchemas)
schema := schemas.Schema(&managementSchema.Version, client.TokenType)
schema.CollectionActions = map[string]types.Action{
"logout": {},
}
schema.ActionHandler = api.tokenActionHandler
schema.ListHandler = api.tokenListHandler
schema.CreateHandler = api.tokenCreateHandler
schema.DeleteHandler = api.tokenDeleteHandler
server := normanapi.NewAPIServer()
if err := server.AddSchemas(schemas); err != nil {
return nil, err
}
return server, nil
}
type tokenAPI struct {
mgr *Manager
}
func (t *tokenAPI) tokenActionHandler(actionName string, action *types.Action, request *types.APIContext) error {
logrus.Debugf("TokenActionHandler called for action %v", actionName)
if actionName == "logout" {
return t.mgr.logout(actionName, action, request)
}
return httperror.NewAPIError(httperror.ActionNotAvailable, "")
}
func (t *tokenAPI) tokenCreateHandler(request *types.APIContext, _ types.RequestHandler) error {
logrus.Debugf("TokenCreateHandler called")
return t.mgr.deriveToken(request)
}
func (t *tokenAPI) tokenListHandler(request *types.APIContext, _ types.RequestHandler) error {
logrus.Debugf("TokenListHandler called")
if request.ID != "" {
return t.mgr.getTokenFromRequest(request)
}
return t.mgr.listTokens(request)
}
func (t *tokenAPI) tokenDeleteHandler(request *types.APIContext, _ types.RequestHandler) error {
logrus.Debugf("TokenDeleteHandler called")
return t.mgr.removeToken(request)
}