mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-29 11:52:10 -05:00
30 lines
782 B
Go
30 lines
782 B
Go
package transform
|
|
|
|
import (
|
|
"bytes"
|
|
jww "github.com/spf13/jwalterweatherman"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func LiveReloadInject(content []byte) (injected []byte) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
jww.ERROR.Println("Recovered in LiveReloadInject", r)
|
|
injected = content
|
|
}
|
|
}()
|
|
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
|
|
}
|