mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-07 20:30:36 -05:00
23 lines
611 B
Go
23 lines
611 B
Go
package transform
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func LiveReloadInject(ct contentTransformer) {
|
|
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(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)
|
|
}
|