diff options
Diffstat (limited to 'transform/livereloadinject.go')
| -rw-r--r-- | transform/livereloadinject.go | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/transform/livereloadinject.go b/transform/livereloadinject.go index 789ff9d22..9eab45f78 100644 --- a/transform/livereloadinject.go +++ b/transform/livereloadinject.go @@ -15,15 +15,20 @@ package transform import ( "bytes" + "fmt" ) func LiveReloadInject(ct contentTransformer) { - match := []byte("</body>") - replace := []byte(`<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script></body>`) + endBodyTag := "</body>" + match := []byte(endBodyTag) + replaceTemplate := `<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script>%s` + replace := []byte(fmt.Sprintf(replaceTemplate, endBodyTag)) newcontent := bytes.Replace(ct.Content(), match, replace, -1) if len(newcontent) == len(ct.Content()) { - match := []byte("</BODY>") + endBodyTag = "</BODY>" + replace := []byte(fmt.Sprintf(replaceTemplate, endBodyTag)) + match := []byte(endBodyTag) newcontent = bytes.Replace(ct.Content(), match, replace, -1) } |