hugo/transform/livereloadinject.go

20 lines
527 B
Go
Raw Normal View History

package transform
import "bytes"
func LiveReloadInject(content []byte) []byte {
match := []byte("</body>")
replace := []byte(`<script>document.write('<script src="http://'
+ (location.host || 'localhost').split(':')[0]
+ ':1313/livereload.js?mindelay=10"></'
+ 'script>')</script></body>`)
newcontent := bytes.Replace(content, match, replace, -1)
if len(newcontent) == len(content) {
match := []byte("</BODY>")
newcontent = bytes.Replace(content, match, replace, -1)
}
return newcontent
}