mirrors/hugo
1
0
Fork 0
mirror of https://github.com/gohugoio/hugo.git synced 2025-04-28 14:40:43 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

livereload: Inject script without head or body tag

Browse source 
Currently, Hugo does not inject `livereload` script if html does not contain `<head>` or `<body>`. This sometimes happens if you create new sites without `theme` and it is hard to catch the cause soon.

This PR:

* Inject livereload script even if html does not include `<head>`, `<body>`, or `<html>`
    - Modern browsers execute scripts even if they are outside `<html>`
    - Some js frameworks (confirmed with vite) inject HRM script without `<html>` tag
* Append warning script to html if `<head>` or `<body>` is not in html
* Fix bug that livereload cannot be appended to the tags with attrs

Close #10105
This commit is contained in:
satotake 2022-08-07 23:15:28 +09:00 • committed by Bjørn Erik Pedersen
parent 7fb28085ac
commit b017f7cb01
2 changed files with 26 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

22
transform/livereloadinject/livereloadinject.go
View file

@ -24,16 +24,24 @@ import (
"github.com/gohugoio/hugo/transform"
)
const warnMessage = `"head" or "body" tag is required in html to append livereload script. ` +
"As a fallback, Hugo injects it somewhere but it might not work properly."
var warnScript = fmt.Sprintf(`<script data-no-instant defer>console.warn('%s');</script>`, warnMessage)
type tag struct {
markup []byte
appendScript bool
warnRequired bool
}
var tags = []tag{
{markup: []byte("<head>"), appendScript: true},
{markup: []byte("<HEAD>"), appendScript: true},
{markup: []byte("<head"), appendScript: true},
{markup: []byte("<HEAD"), appendScript: true},
{markup: []byte("</body>")},
{markup: []byte("</BODY>")},
{markup: []byte("<html"), appendScript: true, warnRequired: true},
{markup: []byte("<HTML"), appendScript: true, warnRequired: true},
}
// New creates a function that can be used
@ -64,15 +72,19 @@ func New(baseURL url.URL) transform.Transformer {
copy(c, b)
if idx == -1 {
_, err := ft.To().Write(c)
return err
idx = len(b)
match = tag{warnRequired: true}
}
script := []byte(fmt.Sprintf(`<script src="%s" data-no-instant defer></script>`, html.EscapeString(src)))
i := idx
if match.appendScript {
i += len(match.markup)
i += bytes.Index(b[i:], []byte(">")) + 1
}
if match.warnRequired {
script = append(script, []byte(warnScript)...)
}
c = append(c[:i], append(script, c[i:]...)...)