mirror of
https://github.com/gohugoio/hugo.git
synced 2024-07-07 17:46:05 +00:00
34 lines
569 B
Go
34 lines
569 B
Go
package transform
|
|
|
|
import (
|
|
htmltran "code.google.com/p/go-html-transform/html/transform"
|
|
"io"
|
|
"fmt"
|
|
)
|
|
|
|
type NavActive struct {
|
|
Section string
|
|
AttrName string
|
|
}
|
|
|
|
func (n *NavActive) Apply(r io.Reader, w io.Writer) (err error) {
|
|
var tr *htmltran.Transformer
|
|
|
|
if n.Section == "" {
|
|
_, err = io.Copy(w, r)
|
|
return
|
|
}
|
|
|
|
if tr, err = htmltran.NewFromReader(r); err != nil {
|
|
return
|
|
}
|
|
|
|
if n.AttrName == "" {
|
|
n.AttrName = "hugo-nav"
|
|
}
|
|
|
|
tr.Apply(htmltran.ModifyAttrib("class", "active"), fmt.Sprintf("li[%s=%s]", n.AttrName, n.Section))
|
|
|
|
return tr.Render(w)
|
|
}
|