transform/nav.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

package transform

import (
	htmltran "code.google.com/p/go-html-transform/html/transform"
	"io"
	"fmt"
)

type NavActive struct {
	Section 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
	}

	tr.Apply(htmltran.ModifyAttrib("class", "active"), fmt.Sprintf("li[data-nav=%s]", n.Section))

	return tr.Render(w)
}