package engine
import (
"bytes"
"net/http"
"strings"
"github.com/xyproto/algernon/utils"
)
// InsertAutoRefresh inserts JavaScript code to the page that makes the page
// refresh itself when the source files changes.
// The JavaScript depends on the event server being available.
// If JavaScript can not be inserted, return the original data.
// Assumes that the given htmldata is actually HTML
// (looks for body/head/html tags when inserting a script tag)
func (ac *Config) InsertAutoRefresh(req *http.Request, htmldata []byte) []byte {
fullHost := ac.eventAddr
// If the host+port starts with ":", assume it's only the port number
if strings.HasPrefix(fullHost, ":") {
// Add the hostname in front
if ac.serverHost != "" {
fullHost = ac.serverHost + ac.eventAddr
} else {
fullHost = utils.GetDomain(req) + ac.eventAddr
}
}
// Wait 70% of an event duration before starting to listen for events
multiplier := 0.7
js := `
`
// Reduce the size slightly
js = strings.TrimSpace(strings.ReplaceAll(js, "\n", ""))
// Remove all whitespace that is more than one space
for strings.Contains(js, " ") {
js = strings.ReplaceAll(js, " ", " ")
}
// Place the script at the end of the body, if there is a body
switch {
case bytes.Contains(htmldata, []byte(" ")):
return bytes.Replace(htmldata, []byte(""), []byte(js+""), 1)
case bytes.Contains(htmldata, []byte("