goldsmith/plugins/summary/summary.go
2024-02-16 22:35:49 -08:00

91 lines
2.2 KiB
Go

// Package summary generates a summary and title for HTML files using CSS
// selectors. This plugin is useful when combined with other plugins such as
// "collection" to create blog post previews.
package summary
import (
"html/template"
"git.foosoft.net/alex/goldsmith"
"git.foosoft.net/alex/goldsmith/filters/wildcard"
"github.com/PuerkitoBio/goquery"
)
// Summary chainable context.
type Summary struct {
titleKey string
summaryKey string
titlePath string
summaryPath string
}
// New creates a new instance of the Summary plugin.
func New() *Summary {
return &Summary{
titleKey: "Title",
summaryKey: "Summary",
titlePath: "h1",
summaryPath: "p",
}
}
// TitleKey sets the metadata key used to store the file title (default: "Title").
func (self *Summary) TitleKey(key string) *Summary {
self.titleKey = key
return self
}
// SummaryKey sets the metadata key used to store the file summary (default: "Summary").
func (self *Summary) SummaryKey(key string) *Summary {
self.summaryKey = key
return self
}
// TitlePath sets CSS path used to retrieve the file title (default: "h1").
func (self *Summary) TitlePath(path string) *Summary {
self.titlePath = path
return self
}
// SummaryPath sets the CSS path used to retrieve the file summary (default: "p").
func (self *Summary) SummaryPath(path string) *Summary {
self.summaryPath = path
return self
}
func (*Summary) Name() string {
return "summary"
}
func (*Summary) Initialize(context *goldsmith.Context) error {
context.Filter(wildcard.New("**/*.html", "**/*.htm"))
return nil
}
func (self *Summary) Process(context *goldsmith.Context, inputFile *goldsmith.File) error {
doc, err := goquery.NewDocumentFromReader(inputFile)
if err != nil {
return err
}
meta := make(map[string]template.HTML)
if match := doc.Find(self.titlePath); match.Length() > 0 {
if html, err := match.Html(); err == nil {
meta[self.titleKey] = template.HTML(html)
}
}
if match := doc.Find(self.summaryPath); match.Length() > 0 {
if html, err := match.Html(); err == nil {
meta[self.summaryKey] = template.HTML(html)
}
}
for key, value := range meta {
inputFile.SetProp(key, value)
}
context.DispatchFile(inputFile)
return nil
}