diff options
Diffstat (limited to 'commands/server_errors.go')
| -rw-r--r-- | commands/server_errors.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/commands/server_errors.go b/commands/server_errors.go index 7f467ee1c..0e3bc50de 100644 --- a/commands/server_errors.go +++ b/commands/server_errors.go @@ -16,6 +16,7 @@ package commands import ( "bytes" "io" + "net/url" "github.com/gohugoio/hugo/transform" "github.com/gohugoio/hugo/transform/livereloadinject" @@ -82,9 +83,9 @@ var buildErrorTemplate = `<!doctype html> </html> ` -func injectLiveReloadScript(src io.Reader, port int) string { +func injectLiveReloadScript(src io.Reader, baseURL url.URL) string { var b bytes.Buffer - chain := transform.Chain{livereloadinject.New(port)} + chain := transform.Chain{livereloadinject.New(baseURL)} chain.Apply(&b, src) return b.String() |