cleanup
This commit is contained in:
parent
0ed062a7ff
commit
d79683b130
@ -1,29 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
)
|
||||
|
||||
type notFoundError struct {
|
||||
ref string
|
||||
}
|
||||
|
||||
func (e notFoundError) Error() string {
|
||||
return fmt.Sprintf("not found: %s", e.ref)
|
||||
}
|
||||
|
||||
var _ error = notFoundError{}
|
||||
|
||||
func isNotFound(err error) bool {
|
||||
if err == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
if os.IsNotExist(err) {
|
||||
return true
|
||||
}
|
||||
|
||||
_, ok := err.(notFoundError)
|
||||
return ok
|
||||
}
|
@ -16,6 +16,7 @@ var trustXFF = flag.Bool("trust-xff", true, "Trust the X-Forwarded-For header")
|
||||
|
||||
type wsHost struct {
|
||||
prefix string
|
||||
hostDoc string
|
||||
getHost func(req *restful.Request) string
|
||||
}
|
||||
|
||||
@ -25,53 +26,56 @@ func (ws *wsHost) register(rws *restful.WebService, alterRB func(*restful.RouteB
|
||||
}
|
||||
|
||||
for _, rb := range []*restful.RouteBuilder{
|
||||
rws.GET(ws.prefix).To(ws.get).
|
||||
Doc("Get the " + ws.hostDoc + "'s details"),
|
||||
|
||||
// raw configuration
|
||||
b("config").
|
||||
Produces(mime.YAML).
|
||||
Doc("Get the host's configuration"),
|
||||
Doc("Get the " + ws.hostDoc + "'s configuration"),
|
||||
|
||||
// metal/local HDD install
|
||||
b("boot.img").
|
||||
Produces(mime.DISK).
|
||||
Doc("Get the host's boot disk image"),
|
||||
Doc("Get the " + ws.hostDoc + "'s boot disk image"),
|
||||
|
||||
b("boot.img.gz").
|
||||
Produces(mime.DISK + "+gzip").
|
||||
Doc("Get the host's boot disk image (gzip compressed)"),
|
||||
Doc("Get the " + ws.hostDoc + "'s boot disk image (gzip compressed)"),
|
||||
|
||||
b("boot.img.lz4").
|
||||
Produces(mime.DISK + "+lz4").
|
||||
Doc("Get the host's boot disk image (lz4 compressed)"),
|
||||
Doc("Get the " + ws.hostDoc + "'s boot disk image (lz4 compressed)"),
|
||||
|
||||
// metal/local HDD upgrades
|
||||
b("boot.tar").
|
||||
Produces(mime.TAR).
|
||||
Doc("Get the host's /boot archive (ie: for metal upgrades)"),
|
||||
Doc("Get the " + ws.hostDoc + "'s /boot archive (ie: for metal upgrades)"),
|
||||
|
||||
// read-only ISO support
|
||||
b("boot.iso").
|
||||
Produces(mime.ISO).
|
||||
Doc("Get the host's boot CD-ROM image"),
|
||||
Doc("Get the " + ws.hostDoc + "'s boot CD-ROM image"),
|
||||
|
||||
// netboot support
|
||||
b("ipxe").
|
||||
Produces(mime.IPXE).
|
||||
Doc("Get the host's IPXE code (for netboot)"),
|
||||
Doc("Get the " + ws.hostDoc + "'s IPXE code (for netboot)"),
|
||||
|
||||
b("kernel").
|
||||
Produces(mime.OCTET).
|
||||
Doc("Get the host's kernel (ie: for netboot)"),
|
||||
Doc("Get the " + ws.hostDoc + "'s kernel (ie: for netboot)"),
|
||||
|
||||
b("initrd").
|
||||
Produces(mime.OCTET).
|
||||
Doc("Get the host's initial RAM disk (ie: for netboot)"),
|
||||
Doc("Get the " + ws.hostDoc + "'s initial RAM disk (ie: for netboot)"),
|
||||
} {
|
||||
alterRB(rb)
|
||||
rws.Route(rb)
|
||||
}
|
||||
}
|
||||
|
||||
func (ws *wsHost) render(req *restful.Request, resp *restful.Response) {
|
||||
func (ws *wsHost) host(req *restful.Request, resp *restful.Response) (host *localconfig.Host, cfg *localconfig.Config) {
|
||||
hostname := ws.getHost(req)
|
||||
if hostname == "" {
|
||||
wsNotFound(req, resp)
|
||||
@ -84,12 +88,29 @@ func (ws *wsHost) render(req *restful.Request, resp *restful.Response) {
|
||||
return
|
||||
}
|
||||
|
||||
host := cfg.Host(hostname)
|
||||
host = cfg.Host(hostname)
|
||||
if host == nil {
|
||||
log.Print("no host named ", hostname)
|
||||
wsNotFound(req, resp)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (ws *wsHost) get(req *restful.Request, resp *restful.Response) {
|
||||
host, _ := ws.host(req, resp)
|
||||
if host == nil {
|
||||
return
|
||||
}
|
||||
|
||||
resp.WriteEntity(host)
|
||||
}
|
||||
|
||||
func (ws *wsHost) render(req *restful.Request, resp *restful.Response) {
|
||||
host, cfg := ws.host(req, resp)
|
||||
if host == nil {
|
||||
return
|
||||
}
|
||||
|
||||
what := path.Base(req.Request.URL.Path)
|
||||
|
||||
@ -105,15 +126,9 @@ func renderHost(w http.ResponseWriter, r *http.Request, what string, host *local
|
||||
}
|
||||
|
||||
switch what {
|
||||
case "ipxe":
|
||||
w.Header().Set("Content-Type", "text/x-ipxe")
|
||||
case "config":
|
||||
w.Header().Set("Content-Type", "text/vnd.yaml")
|
||||
default:
|
||||
w.Header().Set("Content-Type", "application/octet-stream")
|
||||
}
|
||||
err = renderConfig(w, r, ctx)
|
||||
|
||||
switch what {
|
||||
case "ipxe":
|
||||
err = renderIPXE(w, ctx)
|
||||
|
||||
@ -138,20 +153,12 @@ func renderHost(w http.ResponseWriter, r *http.Request, what string, host *local
|
||||
case "boot.img.lz4":
|
||||
err = renderCtx(w, r, ctx, what, buildBootImgLZ4)
|
||||
|
||||
case "config":
|
||||
err = renderConfig(w, r, ctx)
|
||||
|
||||
default:
|
||||
http.NotFound(w, r)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
if isNotFound(err) {
|
||||
log.Printf("host %s: %s: %v", what, host.Name, err)
|
||||
http.NotFound(w, r)
|
||||
} else {
|
||||
log.Printf("host %s: %s: failed to render: %v", what, host.Name, err)
|
||||
http.Error(w, "", http.StatusServiceUnavailable)
|
||||
}
|
||||
log.Printf("host %s: %s: failed to render: %v", what, host.Name, err)
|
||||
http.Error(w, "", http.StatusServiceUnavailable)
|
||||
}
|
||||
}
|
||||
|
@ -21,8 +21,10 @@ func buildWS() *restful.WebService {
|
||||
// clusters API
|
||||
ws.Route(ws.GET("/clusters").Filter(adminAuth).To(wsListClusters).
|
||||
Doc("List clusters"))
|
||||
|
||||
ws.Route(ws.GET("/clusters/{cluster-name}").Filter(adminAuth).To(wsCluster).
|
||||
Doc("Get cluster details"))
|
||||
|
||||
ws.Route(ws.GET("/clusters/{cluster-name}/addons").Filter(adminAuth).To(wsClusterAddons).
|
||||
Produces(mime.YAML).
|
||||
Doc("Get cluster addons").
|
||||
@ -34,14 +36,16 @@ func buildWS() *restful.WebService {
|
||||
Doc("List hosts"))
|
||||
|
||||
(&wsHost{
|
||||
prefix: "",
|
||||
prefix: "/me",
|
||||
hostDoc: "detected host",
|
||||
getHost: detectHost,
|
||||
}).register(ws, func(rb *restful.RouteBuilder) {
|
||||
rb.Notes("In this case, the host is detected from the remote IP")
|
||||
})
|
||||
|
||||
(&wsHost{
|
||||
prefix: "/hosts/{host-name}",
|
||||
prefix: "/hosts/{host-name}",
|
||||
hostDoc: "given host",
|
||||
getHost: func(req *restful.Request) string {
|
||||
return req.PathParameter("host-name")
|
||||
},
|
||||
|
@ -5,6 +5,6 @@ const (
|
||||
TAR = "application/tar"
|
||||
DISK = "application/x-diskimage"
|
||||
ISO = "application/x-iso9660-image"
|
||||
IPXE = "application/x-ipxe"
|
||||
IPXE = "text/x-ipxe"
|
||||
OCTET = "application/octet-stream"
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user