template.go

package golang

import (
	"bytes"
	_ "embed"
	"text/template"
)

// MainFuncRxTmpl the rxstream serverless of the main function template
//
//go:embed templates/main_rx.tmpl
var MainFuncRxTmpl []byte

// MainFuncTmpl the raw bytes serverless of the main function template
//
//go:embed templates/main.tmpl
var MainFuncTmpl []byte

//go:embed templates/init.tmpl
var InitTmpl []byte

//go:embed templates/init_test.tmpl
var InitTestTmpl []byte

//go:embed templates/init_rx.tmpl
var InitRxTmpl []byte

//go:embed templates/wasi_main.tmpl
var WasiMainFuncTmpl []byte

// Context defines context for the template
type Context struct {
	// Name of the servcie
	Name string
	// ZipperAddr is the address of the zipper server
	ZipperAddr string
	// Client credential
	Credential string
	// WithInitFunc determines whether to work with init function
	WithInitFunc bool
	// WithWantedTarget determines whether to work with SetWantedTarget
	WithWantedTarget bool
	// WithDescription determines whether to work with description
	WithDescription bool
	// WithInputSchema determines whether to work with input schema
	WithInputSchema bool
}

// RenderTmpl renders the template with the given context
func RenderTmpl(tpl string, ctx *Context) ([]byte, error) {
	t := template.Must(template.New("tpl").Parse(tpl))
	buf := bytes.NewBuffer([]byte{})
	err := t.Execute(buf, ctx)
	if err != nil {
		return nil, err
	}

	return buf.Bytes(), nil
}