zot/pkg/debug/gqlplayground/gqlplayground.go

//go:build debug
// +build debug

package debug

import (
	"embed"
	"html/template"
	"net/http"

	"github.com/gorilla/mux"
	"zotregistry.io/zot/pkg/api/config"
	"zotregistry.io/zot/pkg/api/constants"
	debugCst "zotregistry.io/zot/pkg/debug/constants"
	"zotregistry.io/zot/pkg/log"
	"zotregistry.io/zot/pkg/storage"
)

//go:embed index.html.tmpl
var playgroundHTML embed.FS

// SetupGQLPlaygroundRoutes ...
func SetupGQLPlaygroundRoutes(conf *config.Config, router *mux.Router,
	storeController storage.StoreController, l log.Logger,
) {
	log := log.Logger{Logger: l.With().Caller().Timestamp().Logger()}
	log.Info().Msg("setting up graphql playground route")

	templ, err := template.ParseFS(playgroundHTML, "index.html.tmpl")
	if err != nil {
		log.Fatal().Err(err)
	}

	//nolint:lll
	router.PathPrefix(constants.RoutePrefix + debugCst.GQLPlaygroundEndpoint).HandlerFunc(func(writer http.ResponseWriter, req *http.Request) {
		writer.Header().Add("Content-Type", "text/html")

		proto := ""

		if req.TLS == nil {
			proto += "http://"
		} else {
			proto += "https://"
		}

		target := proto + req.Host + constants.ExtSearchPrefix

		// respond with the output of template execution
		_ = templ.Execute(writer, struct {
			Target string
		}{Target: target})
	})
}