Add revocer in LiveReloadInject
authorbep <bjorn.erik.pedersen@gmail.com>
Wed, 18 Feb 2015 20:51:32 +0000 (21:51 +0100)
committerbep <bjorn.erik.pedersen@gmail.com>
Wed, 18 Feb 2015 20:51:32 +0000 (21:51 +0100)
The panic cannot be reproduced, but add this as protection.

Fixes #911

transform/livereloadinject.go

index 0973841b3b6e42388e0d2aada933f8eb5a30d1a5..4f572b57c734e3f062bf8fccf51c0190079ad56f 100644 (file)
@@ -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://'