matchbox/api/ipxe.go

package api

import (
	"bytes"
	"fmt"
	"net/http"
	"text/template"
)

const ipxeBootstrap = `#!ipxe
chain ipxe?uuid=${uuid}
`

var ipxeTemplate = template.Must(template.New("ipxe boot").Parse(`#!ipxe
kernel {{.Kernel}}{{range $key, $value := .Cmdline}} {{if $value}}{{$key}}={{$value}}{{else}}{{$key}}{{end}}{{end}}
initrd {{ range $element := .Initrd }}{{$element}}{{end}}
boot
`))

// ipxeInspect returns a handler that responds with an iPXE script to gather
// client machine data and chain load the real boot script.
func ipxeInspect() http.Handler {
	fn := func(w http.ResponseWriter, req *http.Request) {
		log.Info("iPXE boot script request")
		fmt.Fprintf(w, ipxeBootstrap)
	}
	return http.HandlerFunc(fn)
}

// ipxeBoot returns a handler which renders an iPXE boot config script based
// on the machine attribtue query parameters.
func ipxeHandler(store Store) http.Handler {
	fn := func(w http.ResponseWriter, req *http.Request) {
		attrs := attrsFromRequest(req)
		log.Infof("iPXE boot config request for %+v", attrs)

		config, err := store.BootConfig(attrs)
		if err != nil {
			http.NotFound(w, req)
			return
		}

		var buf bytes.Buffer
		err = ipxeTemplate.Execute(&buf, config)
		if err != nil {
			log.Errorf("iPXE template render error: %s", err)
			http.NotFound(w, req)
			return
		}
		if _, err := buf.WriteTo(w); err != nil {
			log.Infof("error writing to response, %s", err)
		}
	}
	return http.HandlerFunc(fn)
}