local-server/cmd/dkl-local-server/ws-host.go

package main

import (
	"flag"
	"log"
	"net/http"
	"path"

	restful "github.com/emicklei/go-restful"

	"novit.nc/direktil/local-server/pkg/mime"
	"novit.nc/direktil/pkg/localconfig"
)

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
}

func (ws *wsHost) register(rws *restful.WebService, alterRB func(*restful.RouteBuilder)) {
	b := func(what string) *restful.RouteBuilder {
		return rws.GET(ws.prefix + "/" + what).To(ws.render)
	}

	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 " + ws.hostDoc + "'s configuration"),

		b("config.json").
			Doc("Get the " + ws.hostDoc + "'s configuration (as JSON)"),

		// metal/local HDD install
		b("boot.img").
			Produces(mime.DISK).
			Doc("Get the " + ws.hostDoc + "'s boot disk image"),

		b("boot.img.gz").
			Produces(mime.DISK + "+gzip").
			Doc("Get the " + ws.hostDoc + "'s boot disk image (gzip compressed)"),

		b("boot.img.lz4").
			Produces(mime.DISK + "+lz4").
			Doc("Get the " + ws.hostDoc + "'s boot disk image (lz4 compressed)"),

		// metal/local HDD upgrades
		b("boot.tar").
			Produces(mime.TAR).
			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 " + ws.hostDoc + "'s boot CD-ROM image"),

		// netboot support
		b("ipxe").
			Produces(mime.IPXE).
			Doc("Get the " + ws.hostDoc + "'s IPXE code (for netboot)"),

		b("kernel").
			Produces(mime.OCTET).
			Doc("Get the " + ws.hostDoc + "'s kernel (ie: for netboot)"),

		b("initrd").
			Produces(mime.OCTET).
			Doc("Get the " + ws.hostDoc + "'s initial RAM disk (ie: for netboot)"),
	} {
		alterRB(rb)
		rws.Route(rb)
	}
}

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)
		return
	}

	cfg, err := readConfig()
	if err != nil {
		wsError(resp, err)
		return
	}

	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)

	renderHost(resp.ResponseWriter, req.Request, what, host, cfg)
}

func renderHost(w http.ResponseWriter, r *http.Request, what string, host *localconfig.Host, cfg *localconfig.Config) {
	ctx, err := newRenderContext(host, cfg)
	if err != nil {
		log.Printf("host %s: %s: failed to render: %v", what, host.Name, err)
		http.Error(w, "", http.StatusServiceUnavailable)
		return
	}

	switch what {
	case "config":
		err = renderConfig(w, r, ctx, false)

	case "config.json":
		err = renderConfig(w, r, ctx, true)

	case "ipxe":
		err = renderIPXE(w, ctx)

	case "kernel":
		err = renderKernel(w, r, ctx)

	case "initrd":
		err = renderCtx(w, r, ctx, what, buildInitrd)

	case "boot.iso":
		err = renderCtx(w, r, ctx, what, buildBootISO)

	case "boot.tar":
		err = renderCtx(w, r, ctx, what, buildBootTar)

	case "boot.img":
		err = renderCtx(w, r, ctx, what, buildBootImg)

	case "boot.img.gz":
		err = renderCtx(w, r, ctx, what, buildBootImgGZ)

	case "boot.img.lz4":
		err = renderCtx(w, r, ctx, what, buildBootImgLZ4)

	default:
		http.NotFound(w, r)
	}

	if err != nil {
		log.Printf("host %s: %s: failed to render: %v", what, host.Name, err)
		http.Error(w, "", http.StatusServiceUnavailable)
	}
}