Load livereload.js from "/"

Fix #1406
Instead of loading the file from http://localhost:port/, it can be
loaded from /.
This commit is contained in:
Kato Kazuyoshi 2015-10-14 00:41:54 -07:00 committed by spf13
parent 0f438d1852
commit ef4dfcec6c
2 changed files with 22 additions and 7 deletions

View file

@ -2,18 +2,13 @@ package transform
import (
"bytes"
"github.com/spf13/viper"
)
func LiveReloadInject(ct contentTransformer) {
match := []byte("</body>")
port := viper.GetString("port")
replace := []byte(`<script data-no-instant>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)
replace := []byte(`<script data-no-instant>document.write('<script src="/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)

View file

@ -0,0 +1,20 @@
package transform
import (
"bytes"
"github.com/spf13/hugo/helpers"
"testing"
)
func TestLiveReloadInject(t *testing.T) {
out := new(bytes.Buffer)
in := helpers.StringToReader("</body>")
tr := NewChain(LiveReloadInject)
tr.Apply(out, in, []byte("path"))
expected := `<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script></body>`
if string(out.Bytes()) != expected {
t.Errorf("Expected %s got %s", expected, string(out.Bytes()))
}
}