Containerize the webserver

1 files changed, 19 insertions, 2 deletions

diff --git a/cmd/main.go b/cmd/main.go
index cd8d4fe..24ba0c9 100644
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -1,14 +1,31 @@
 package main
 
 import (
+	"fmt"
 	"log"
+	"os"
 
 	"git.mithe.dk/web-portfolio/internal"
 )
 
 func main() {
-	server := internal.NewSever()
-	if err := server.Start("localhost:42069"); err != nil {
+	port := os.Getenv("PORT")
+	if port == "" {
+		port = "42069"
+	}
+
+	// In Docker, listen on 0.0.0.0; locally use localhost
+	host := os.Getenv("HOST")
+	if host == "" {
+		host = "0.0.0.0"
+	}
+
+	addr := fmt.Sprintf("%s:%s", host, port)
+
+	server := internal.NewServer()
+	log.Printf("Starting server on %s", addr)
+
+	if err := server.Start(addr); err != nil {
 		log.Fatal(err)
 	}
 }