From ff80ef3f2a1ec03649984fab19a5c946f974031b Mon Sep 17 00:00:00 2001
From: Andre Medeiros <me@andremedeiros.info>
Date: Mon, 5 Jul 2021 20:23:50 -0400
Subject: [PATCH] fix web

---
 cmd/apollo-api/main.go | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/cmd/apollo-api/main.go b/cmd/apollo-api/main.go
index 9f8370f..c4f8b05 100644
--- a/cmd/apollo-api/main.go
+++ b/cmd/apollo-api/main.go
@@ -2,7 +2,6 @@ package main
 
 import (
 	"database/sql"
-	"flag"
 	"fmt"
 	"log"
 	"net/http"
@@ -35,8 +34,6 @@ func main() {
 	}
 
 	var cfg config
-	flag.IntVar(&cfg.port, "port", 4000, "API server port")
-	flag.Parse()
 
 	db, err := sql.Open("postgres", os.Getenv("DATABASE_URL"))
 	if err != nil {
@@ -54,8 +51,13 @@ func main() {
 		rc,
 	}
 
+	port, ok := os.LookupEnv("PORT")
+	if !ok {
+		port = "4000"
+	}
+
 	srv := &http.Server{
-		Addr:    fmt.Sprintf(":%d", cfg.port),
+		Addr:    fmt.Sprintf(":%s", port),
 		Handler: app.routes(),
 	}