shithub: hugo

ref: 4d708f096d5df64bf7309cbed67564d98c8e815a
dir: /transform/livereloadinject.go/

package transform

import (
	"bytes"
	jww "github.com/spf13/jwalterweatherman"
	"github.com/spf13/viper"
)

func LiveReloadInject(content []byte) []byte {
	defer func() {
		if r := recover(); r != nil {
			jww.ERROR.Println("Recovered in LiveReloadInject", r)
		}
	}()
	match := []byte("</body>")
	port := viper.GetString("port")
	replace := []byte(`<script>document.write('<script src="http://'
        + (location.host || 'localhost').split(':')[0]
		+ ':` + port + `/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
}