Enable port configuration with environment variable.

hscells · Oct 12, 2018 · 5deae71 · 5deae71
1 parent 7a97468
commit 5deae71
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/cmd/bigbro/main.go b/cmd/bigbro/main.go
@@ -7,6 +7,7 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/hscells/bigbro"
 	"log"
+	"os"
 )
 
 type server struct {
@@ -78,6 +79,11 @@ func main() {
 		c.Next()
 	})
 
+	port := os.Getenv("BIGBRO_PORT")
+	if len(port) == 0 {
+		port = "1984"
+	}
+
 	g.GET("/event", s.handleEvent)
 	if goterm.Width() > 91 {
 		fmt.Print(`
@@ -90,16 +96,16 @@ func main() {
                                                                     ...is always watching
 
  Harry Scells 2018
- version 09.Aug.2018
+ version 13.Oct.2018
 
 `)
 	} else {
 		fmt.Print(`Big Brother
 ...is always watching
 
 Harry Scells 2018
-version 18.Sep.2018
+version 13.Oct.2018
 `)
 	}
-	g.Run("0.0.0.0:1984")
+	g.Run(fmt.Sprintf("0.0.0.0:%s", port))
 }