github/hugo
1
0
Fork 0
mirror of https://github.com/gohugoio/hugo.git synced 2024-11-21 20:46:30 -05:00
Code Issues Projects Releases Packages Wiki Activity

Add revocer in LiveReloadInject

Browse source 
The panic cannot be reproduced, but add this as protection.

Fixes #911
This commit is contained in:
bep 2015-02-18 21:51:32 +01:00
parent d2e022f2a7
commit 4d708f096d
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
transform/livereloadinject.go
View file

@ -2,11 +2,16 @@ package transform
import ( import (
"bytes" "bytes"
jww "github.com/spf13/jwalterweatherman"
"github.com/spf13/viper" "github.com/spf13/viper"
) )
func LiveReloadInject(content []byte) []byte { func LiveReloadInject(content []byte) []byte {
defer func() {
if r := recover(); r != nil {
jww.ERROR.Println("Recovered in LiveReloadInject", r)
}
}()
match := []byte("</body>") match := []byte("</body>")
port := viper.GetString("port") port := viper.GetString("port")
replace := []byte(`<script>document.write('<script src="http://' replace := []byte(`<script>document.write('<script src="http://'