Merge pull request #7 from nregina-hbs/listen-port

Allow selection of listen port

README.md

@@ -37,6 +37,7 @@ Configuration is done via Environment Variables:
 | CODEARTIFACT_DOMAIN  | Yes        | Your CodeArtifact Domain (e.g. sktansandbox) |
 | CODEARTIFACT_TYPE    | No         | Use one of the following: pypi, npm, maven, nuget |
 | CODEARTIFACT_OWNER   | No         | The AWS Account Id of the CodeArtifact Owner (if it's your own account, it can be empty) |
+| LISTEN_PORT          | No         | Port on which the proxy should listen.  Defaults to 8080 |
 
 By default, the proxy will choose to use the Pypi as it's type.
 

src/tools/proxy.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"net/http/httputil"
 	"net/url"
+	"os"
 	"strconv"
 	"strings"
 	"sync"
@@ -125,13 +126,23 @@ func ProxyInit() {
 		panic(err)
 	}
 
+	// Get port from LISTEN_PORT environment variable. If not set, default to 8080.
+	port := getEnv("LISTEN_PORT", "8080")
+
 	proxy := httputil.NewSingleHostReverseProxy(remote)
 
 	proxy.ModifyResponse = ProxyResponseHandler()
 
 	http.HandleFunc("/", ProxyRequestHandler(proxy))
-	err = http.ListenAndServe(":8080", nil)
+	err = http.ListenAndServe(":"+port, nil)
 	if err != nil {
 		panic(err)
 	}
 }
+
+func getEnv(key, fallback string) string {
+	if value, ok := os.LookupEnv(key); ok {
+		return value
+	}
+	return fallback
+}