summaryrefslogtreecommitdiffstats
path: root/transform/livereloadinject.go
blob: 2fb130c17cae0f9407a4cb5327a0613d930fb7a2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
package transform

import (
	"bytes"
)

func LiveReloadInject(ct contentTransformer) {
	match := []byte("</body>")
	replace := []byte(`<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script></body>`)

	newcontent := bytes.Replace(ct.Content(), match, replace, -1)
	if len(newcontent) == len(ct.Content()) {
		match := []byte("</BODY>")
		newcontent = bytes.Replace(ct.Content(), match, replace, -1)
	}

	ct.Write(newcontent)
}