forked from kubevirt/kubevirt
-
Notifications
You must be signed in to change notification settings - Fork 0
/
virt-api.go
94 lines (81 loc) · 3.23 KB
/
virt-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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package main
import (
"flag"
"github.com/emicklei/go-restful"
"github.com/emicklei/go-restful/swagger"
kithttp "github.com/go-kit/kit/transport/http"
"golang.org/x/net/context"
"k8s.io/client-go/pkg/runtime/schema"
"kubevirt.io/kubevirt/pkg/api/v1"
"kubevirt.io/kubevirt/pkg/healthz"
"kubevirt.io/kubevirt/pkg/kubecli"
"kubevirt.io/kubevirt/pkg/logging"
mime "kubevirt.io/kubevirt/pkg/rest"
"kubevirt.io/kubevirt/pkg/rest/endpoints"
"kubevirt.io/kubevirt/pkg/rest/filter"
"kubevirt.io/kubevirt/pkg/virt-api/rest"
"log"
"net/http"
"strconv"
)
func main() {
logging.InitializeLogging("virt-api")
swaggerui := flag.String("swagger-ui", "third_party/swagger-ui", "swagger-ui location")
host := flag.String("listen", "0.0.0.0", "Address and port where to listen on")
port := flag.Int("port", 8183, "Port to listen on")
flag.Parse()
ctx := context.Background()
vmGVR := schema.GroupVersionResource{Group: v1.GroupVersion.Group, Version: v1.GroupVersion.Version, Resource: "vms"}
migrationGVR := schema.GroupVersionResource{Group: v1.GroupVersion.Group, Version: v1.GroupVersion.Version, Resource: "migrations"}
ws, err := rest.GroupVersionProxyBase(ctx, v1.GroupVersion)
if err != nil {
log.Fatal(err)
}
ws, err = rest.GenericResourceProxy(ws, ctx, vmGVR, &v1.VM{}, v1.GroupVersionKind.Kind, &v1.VMList{})
if err != nil {
log.Fatal(err)
}
ws, err = rest.GenericResourceProxy(ws, ctx, migrationGVR, &v1.Migration{}, "Migration", &v1.MigrationList{})
if err != nil {
log.Fatal(err)
}
cli, err := kubecli.GetRESTClient()
if err != nil {
log.Fatal(err)
}
coreCli, err := kubecli.Get()
if err != nil {
log.Fatal(err)
}
// TODO, allow Encoder and Decoders per type and combine the endpoint logic
spice := endpoints.MakeGoRestfulWrapper(endpoints.NewHandlerBuilder().Get().
Endpoint(rest.NewSpiceEndpoint(cli, coreCli, vmGVR)).Encoder(
endpoints.NewMimeTypeAwareEncoder(endpoints.NewEncodeINIResponse(http.StatusOK),
map[string]kithttp.EncodeResponseFunc{
mime.MIME_INI: endpoints.NewEncodeINIResponse(http.StatusOK),
mime.MIME_JSON: endpoints.NewEncodeJsonResponse(http.StatusOK),
mime.MIME_YAML: endpoints.NewEncodeYamlResponse(http.StatusOK),
})).Build(ctx))
ws.Route(ws.GET(rest.ResourcePath(vmGVR)+rest.SubResourcePath("spice")).
To(spice).Produces(mime.MIME_INI, mime.MIME_JSON, mime.MIME_YAML).
Param(rest.NamespaceParam(ws)).Param(rest.NameParam(ws)).
Doc("Returns a remote-viewer configuration file. Run `man 1 remote-viewer` to learn more about the configuration format."))
restful.Add(ws)
ws.Route(ws.GET("/healthz").To(healthz.KubeConnectionHealthzFunc).Consumes(restful.MIME_JSON).Produces(restful.MIME_JSON).Doc("Health endpoint"))
ws, err = rest.ResourceProxyAutodiscovery(ctx, vmGVR)
if err != nil {
log.Fatal(err)
}
restful.Add(ws)
restful.Filter(filter.RequestLoggingFilter())
restful.Filter(restful.OPTIONSFilter())
config := swagger.Config{
WebServices: restful.RegisteredWebServices(), // you control what services are visible
WebServicesUrl: "http://localhost:8183",
ApiPath: "/swaggerapi",
SwaggerPath: "/swagger-ui/",
SwaggerFilePath: *swaggerui,
}
swagger.InstallSwaggerService(config)
log.Fatal(http.ListenAndServe(*host+":"+strconv.Itoa(*port), nil))
}