forked from rancher/rancher
-
Notifications
You must be signed in to change notification settings - Fork 1
/
nodeserver.go
62 lines (49 loc) · 1.37 KB
/
nodeserver.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
package rkenodeconfigserver
import (
"net/http"
"encoding/json"
"github.com/rancher/rancher/pkg/tunnelserver"
"github.com/rancher/types/apis/management.cattle.io/v3"
"github.com/sirupsen/logrus"
)
type RKENodeConfigServer struct {
auth *tunnelserver.Authorizer
}
func Handler(auth *tunnelserver.Authorizer) http.Handler {
return &RKENodeConfigServer{
auth: auth,
}
}
func (n *RKENodeConfigServer) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
client, ok, err := n.auth.Authorize(req)
if err != nil {
rw.WriteHeader(http.StatusInternalServerError)
rw.Write([]byte(err.Error()))
return
}
if !ok {
rw.WriteHeader(http.StatusUnauthorized)
return
}
if client.Node == nil {
rw.WriteHeader(http.StatusNotFound)
return
}
if !client.Node.Spec.Imported || client.Cluster.Status.Driver != v3.ClusterDriverLocal {
rw.WriteHeader(http.StatusNotFound)
return
}
if client.Node.Status.NodeConfig == nil {
rw.WriteHeader(http.StatusServiceUnavailable)
}
nodeConfig, err := AgentConfig(req.Context(), *client.Node.Status.NodeConfig, client.Server, client.Token)
if err != nil {
rw.WriteHeader(http.StatusInternalServerError)
rw.Write([]byte(err.Error()))
return
}
rw.Header().Set("Content-Type", "application/json")
if err := json.NewEncoder(rw).Encode(nodeConfig); err != nil {
logrus.Errorf("failed to write nodeConfig to agent: %v", err)
}
}