Add revocer in LiveReloadInject

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

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://'