70 lines
1.2 KiB
Go
70 lines
1.2 KiB
Go
package renderer
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/cortezaproject/corteza-server/system/types"
|
|
)
|
|
|
|
type (
|
|
genericHTML struct {
|
|
def DriverDefinition
|
|
}
|
|
genericHTMLDriver struct{}
|
|
)
|
|
|
|
func newGenericHTML() driverFactory {
|
|
return &genericHTML{
|
|
def: DriverDefinition{
|
|
Name: "genericHTML",
|
|
InputTypes: []types.DocumentType{
|
|
types.DocumentTypePlain,
|
|
types.DocumentTypeHTML,
|
|
},
|
|
OutputTypes: []types.DocumentType{
|
|
types.DocumentTypeHTML,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (d *genericHTML) Define() DriverDefinition {
|
|
return d.def
|
|
}
|
|
|
|
func (d *genericHTML) CanRender(t types.DocumentType) bool {
|
|
for _, i := range d.def.InputTypes {
|
|
if i == t {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (d *genericHTML) CanProduce(t types.DocumentType) bool {
|
|
for _, o := range d.def.OutputTypes {
|
|
if o == t {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (d *genericHTML) Driver() driver {
|
|
return &genericHTMLDriver{}
|
|
}
|
|
|
|
func (d *genericHTMLDriver) Render(ctx context.Context, pl *driverPayload) (io.ReadSeeker, error) {
|
|
t, err := preprocHTMLTemplate(pl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dd := &bytes.Buffer{}
|
|
err = t.Execute(dd, pl.Variables)
|
|
|
|
return bytes.NewReader(dd.Bytes()), err
|
|
}
|