Implement setSupervisorAuthHeader for setting bearer token

Author: sairon

http.go

@@ -112,7 +112,7 @@ func httpSupervisorProxy(w http.ResponseWriter, r *http.Request) {
 	proxy := httputil.NewSingleHostReverseProxy(u)
 
 	// Add authorization header
-	r.Header.Add("Authorization", "Bearer "+os.Getenv("SUPERVISOR_TOKEN"))
+	setSupervisorAuthHeader(r)
 
 	switch cleanPath {
 	case "/logs":

main.go

@@ -24,6 +24,15 @@ func getSupervisorHost() string {
 	return supervisorHost
 }
 
+func setSupervisorAuthHeader(r *http.Request) {
+	token := os.Getenv("SUPERVISOR_TOKEN")
+	if token != "" {
+		r.Header.Add("Authorization", "Bearer "+token)
+	} else {
+		log.Println("No SUPERVISOR_TOKEN set, request will be unauthenticated")
+	}
+}
+
 func main() {
 	development = (os.Getenv("DEVELOPMENT") == "True")
 

mdns.go

@@ -7,7 +7,6 @@ import (
 	"log"
 	"net"
 	"net/http"
-	"os"
 	"time"
 
 	"github.com/grandcat/zeroconf"
@@ -91,7 +90,7 @@ func getOutboundIP() (net.IP, error) {
 		return nil, fmt.Errorf("can't create request to Supervisor: %s", err)
 	}
 
-	req.Header.Add("Authorization", "Bearer "+os.Getenv("SUPERVISOR_TOKEN"))
+	setSupervisorAuthHeader(req)
 
 	response, err := client.Do(req)
 	if err == nil {