mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-21 20:46:30 -05:00
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:
parent
0f438d1852
commit
ef4dfcec6c
2 changed files with 22 additions and 7 deletions
|
@ -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)
|
||||
|
|
20
transform/livereloadinject_test.go
Normal file
20
transform/livereloadinject_test.go
Normal 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()))
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue