templates.go

package timelineTracing

import (
	"bytes"
	"embed"
	"encoding/json"
	"fmt"
	"html/template"
)

//go:embed templateDatadogTraceLink.html
//go:embed template.html
//go:embed templateOpenWithPerfetto.html
var embeddedData embed.FS

type templateName string

const (
	templateDatadogHtml templateName = "templateDatadogTraceLink.html"
	templateHtml        templateName = "template.html"
	templateTraceHtml   templateName = "templateOpenWithPerfetto.html"
)

func loadTemplate(templateName templateName) (*template.Template, error) {
	t, err := template.ParseFS(embeddedData, string(templateName))
	if err != nil {
		return nil, fmt.Errorf("failed to parse the template file %q: %w", templateName, err)
	}
	return t, nil
}

func marshalWithoutEscapingHTML(t interface{}) ([]byte, error) {
	buffer := &bytes.Buffer{}
	encoder := json.NewEncoder(buffer)
	encoder.SetEscapeHTML(false)
	err := encoder.Encode(t)
	return buffer.Bytes(), err
}