package route

import (
	"net/http"

	"github.com/pkg/errors"
	"gitlab.com/wpetit/goweb/middleware/container"
	"gitlab.com/wpetit/goweb/service/template"
	"gitlab.com/wpetit/goweb/template/html"
)

func extendTemplateData(w http.ResponseWriter, r *http.Request, data template.Data) template.Data {
	ctn := container.Must(r.Context())
	data, err := template.Extend(data,
		html.WithFlashes(w, r, ctn),
		template.WithBuildInfo(w, r, ctn),
	)

	if err != nil {
		panic(errors.Wrap(err, "could not extend template data"))
	}

	return data
}