Fixed #301. LiveReload works on any port now.

diff --git a/commands/server.go b/commands/server.go
index 7f0b9585e49d42679cee8835b551e7a1c0d4c2af..89e4c5ba7038003360efe1b215997a73f77822e6 100644 (file)
--- a/commands/server.go
+++ b/commands/server.go
@@ -83,6 +83,8 @@ func server(cmd *cobra.Command, args []string) {
                serverPort = sp.Port
        }
 
+       viper.Set("port", serverPort)
+
        if serverAppend {
                viper.Set("BaseUrl", strings.TrimSuffix(BaseUrl, "/")+":"+strconv.Itoa(serverPort))
        } else {

diff --git a/transform/livereloadinject.go b/transform/livereloadinject.go
index a84f850ad3d73ac0a0d4aefa9a0238b493f2be50..0973841b3b6e42388e0d2aada933f8eb5a30d1a5 100644 (file)
--- a/transform/livereloadinject.go
+++ b/transform/livereloadinject.go
@@ -1,12 +1,17 @@
 package transform
 
-import "bytes"
+import (
+       "bytes"
+
+       "github.com/spf13/viper"
+)
 
 func LiveReloadInject(content []byte) []byte {
        match := []byte("</body>")
+       port := viper.GetString("port")
        replace := []byte(`<script>document.write('<script src="http://'
         + (location.host || 'localhost').split(':')[0]
-        + ':1313/livereload.js?mindelay=10"></'
+               + ':` + port + `/livereload.js?mindelay=10"></'
         + 'script>')</script></body>`)
        newcontent := bytes.Replace(content, match, replace, -1)