From: bep <bjorn.erik.pedersen@gmail.com>
Date: Wed, 18 Feb 2015 20:51:32 +0000 (+0100)
Subject: Add revocer in LiveReloadInject
X-Git-Tag: v0.13~8
X-Git-Url: http://git.maquefel.me/?a=commitdiff_plain;h=4d708f096d5df64bf7309cbed67564d98c8e815a;p=brevno-suite%2Fhugo

Add revocer in LiveReloadInject

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

Fixes #911
---

diff --git a/transform/livereloadinject.go b/transform/livereloadinject.go
index 0973841b..4f572b57 100644
--- a/transform/livereloadinject.go
+++ b/transform/livereloadinject.go
@@ -2,11 +2,16 @@ 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://'