goldsmith/plugins/livejs/livejs.go

68 lines
1.6 KiB
Go
Raw Normal View History

2024-02-17 06:35:49 +00:00
// Package livejs injects code to reload the current page when it (or its
// dependencies) are modified. This plugin is helpful for authoring web content
// locally, but should be disabled for site deployment. This can be achieved by
// conditionally including it using the "condition" filter.
package livejs
import (
"bytes"
_ "embed"
"fmt"
"git.foosoft.net/alex/goldsmith"
"git.foosoft.net/alex/goldsmith/filters/wildcard"
"github.com/PuerkitoBio/goquery"
)
//go:embed js/live.js
var livejs string
// LiveJs chainable context.
type LiveJs struct {
html string
}
// New creates a new instance of the LiveJs plugin.
func New() *LiveJs {
return new(LiveJs)
}
func (*LiveJs) Name() string {
return "livejs"
}
func (self *LiveJs) Initialize(context *goldsmith.Context) error {
2024-02-17 06:35:49 +00:00
self.html = fmt.Sprintf("\n<!-- begin livejs code -->\n<script>\n%s\n</script>\n<!-- end livejs code -->\n", livejs)
context.Filter(wildcard.New("**/*.html", "**/*.htm"))
return nil
}
func (self *LiveJs) Process(context *goldsmith.Context, inputFile *goldsmith.File) error {
2024-02-17 06:35:49 +00:00
if outputFile := context.RetrieveCachedFile(inputFile.Path(), inputFile); outputFile != nil {
outputFile.CopyProps(inputFile)
context.DispatchFile(outputFile)
return nil
}
doc, err := goquery.NewDocumentFromReader(inputFile)
if err != nil {
return err
}
doc.Find("body").AppendHtml(self.html)
html, err := doc.Html()
if err != nil {
return err
}
outputFile, err := context.CreateFileFromReader(inputFile.Path(), bytes.NewReader([]byte(html)))
if err != nil {
return err
}
outputFile.CopyProps(inputFile)
context.DispatchAndCacheFile(outputFile)
return nil
}