From 7a8b754cada398241fc2d58d63ce58551f6c734c Mon Sep 17 00:00:00 2001
From: spf13 <steve.francia@gmail.com>
Date: Wed, 28 May 2014 19:01:24 -0400
Subject: [PATCH] Fixed #301. LiveReload works on any port now.

---
 commands/server.go            | 2 ++
 transform/livereloadinject.go | 9 +++++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/commands/server.go b/commands/server.go
index 7f0b9585e..89e4c5ba7 100644
--- 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 a84f850ad..0973841b3 100644
--- 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)