begin move to go-restful
This commit is contained in:
55
cmd/dkl-local-server/ws-host.go
Normal file
55
cmd/dkl-local-server/ws-host.go
Normal file
@ -0,0 +1,55 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"net/http"
|
||||
"path"
|
||||
|
||||
restful "github.com/emicklei/go-restful"
|
||||
)
|
||||
|
||||
type wsHost struct {
|
||||
prefix string
|
||||
getHost func(req *restful.Request) string
|
||||
}
|
||||
|
||||
func (ws *wsHost) register(rws *restful.WebService) {
|
||||
for _, what := range []string{
|
||||
"boot.img",
|
||||
"boot.img.gz",
|
||||
"boot.img.lz4",
|
||||
"boot.iso",
|
||||
"boot.tar",
|
||||
"config",
|
||||
"initrd",
|
||||
"ipxe",
|
||||
"kernel",
|
||||
} {
|
||||
rws.Route(rws.GET(ws.prefix + "/" + what).To(ws.render))
|
||||
}
|
||||
}
|
||||
|
||||
func (ws *wsHost) render(req *restful.Request, resp *restful.Response) {
|
||||
hostname := ws.getHost(req)
|
||||
if hostname == "" {
|
||||
http.NotFound(resp.ResponseWriter, req.Request)
|
||||
return
|
||||
}
|
||||
|
||||
cfg, err := readConfig()
|
||||
if err != nil {
|
||||
writeError(resp.ResponseWriter, err)
|
||||
return
|
||||
}
|
||||
|
||||
host := cfg.Host(hostname)
|
||||
if host == nil {
|
||||
log.Print("no host named ", hostname)
|
||||
http.NotFound(resp.ResponseWriter, req.Request)
|
||||
return
|
||||
}
|
||||
|
||||
what := path.Base(req.Request.URL.Path)
|
||||
|
||||
renderHost(resp.ResponseWriter, req.Request, what, host, cfg)
|
||||
}
|
Reference in New Issue
Block a user