From f2860fd684c217f439a96a38c8cbbf3eef48c33e Mon Sep 17 00:00:00 2001
From: Samuel Padgett <spadgett@redhat.com>
Date: Fri, 6 Sep 2019 10:24:46 -0400
Subject: [PATCH] Enable metering proxy for local development

---
 cmd/bridge/main.go | 31 +++++++++++++++++--------------
 1 file changed, 17 insertions(+), 14 deletions(-)

diff --git a/cmd/bridge/main.go b/cmd/bridge/main.go
index 500b22dfc18..f81cc4066d7 100644
--- a/cmd/bridge/main.go
+++ b/cmd/bridge/main.go
@@ -70,6 +70,7 @@ func main() {
 	fK8sModeOffClusterSkipVerifyTLS := fs.Bool("k8s-mode-off-cluster-skip-verify-tls", false, "DEV ONLY. When true, skip verification of certs presented by k8s API server.")
 	fK8sModeOffClusterPrometheus := fs.String("k8s-mode-off-cluster-prometheus", "", "DEV ONLY. URL of the cluster's Prometheus server.")
 	fK8sModeOffClusterAlertmanager := fs.String("k8s-mode-off-cluster-alertmanager", "", "DEV ONLY. URL of the cluster's AlertManager server.")
+	fK8sModeOffClusterMetering := fs.String("k8s-mode-off-cluster-metering", "", "DEV ONLY. URL of the cluster's metering server.")
 
 	fK8sAuth := fs.String("k8s-auth", "service-account", "service-account | bearer-token | oidc | openshift")
 	fK8sAuthBearerToken := fs.String("k8s-auth-bearer-token", "", "Authorization token to send with proxied Kubernetes API requests.")
@@ -146,18 +147,6 @@ func main() {
 		documentationBaseURL = bridge.ValidateFlagIsURL("documentation-base-url", *fDocumentationBaseURL)
 	}
 
-	offClusterPrometheusURL := &url.URL{}
-	if *fK8sModeOffClusterPrometheus != "" && *fK8sMode == "off-cluster" {
-		offClusterPrometheusURL = bridge.ValidateFlagIsURL("k8s-mode-off-cluster-prometheus", *fK8sModeOffClusterPrometheus)
-		offClusterPrometheusURL.Path = "/api"
-	}
-
-	offClusterAlertManagerURL := &url.URL{}
-	if *fK8sModeOffClusterAlertmanager != "" && *fK8sMode == "off-cluster" {
-		offClusterAlertManagerURL = bridge.ValidateFlagIsURL("k8s-mode-off-cluster-alertmanager", *fK8sModeOffClusterAlertmanager)
-		offClusterAlertManagerURL.Path = "/api"
-	}
-
 	branding := *fBranding
 	if branding == "origin" {
 		branding = "okd"
@@ -311,7 +300,9 @@ func main() {
 			Endpoint:        k8sEndpoint,
 		}
 
-		if offClusterPrometheusURL.String() != "" {
+		if *fK8sModeOffClusterPrometheus != "" {
+			offClusterPrometheusURL := bridge.ValidateFlagIsURL("k8s-mode-off-cluster-prometheus", *fK8sModeOffClusterPrometheus)
+			offClusterPrometheusURL.Path = "/api"
 			srv.PrometheusProxyConfig = &proxy.Config{
 				TLSClientConfig: serviceProxyTLSConfig,
 				HeaderBlacklist: []string{"Cookie", "X-CSRFToken"},
@@ -324,7 +315,9 @@ func main() {
 			}
 		}
 
-		if offClusterAlertManagerURL.String() != "" {
+		if *fK8sModeOffClusterAlertmanager != "" {
+			offClusterAlertManagerURL := bridge.ValidateFlagIsURL("k8s-mode-off-cluster-alertmanager", *fK8sModeOffClusterAlertmanager)
+			offClusterAlertManagerURL.Path = "/api"
 			srv.AlertManagerProxyConfig = &proxy.Config{
 				TLSClientConfig: serviceProxyTLSConfig,
 				HeaderBlacklist: []string{"Cookie", "X-CSRFToken"},
@@ -332,6 +325,16 @@ func main() {
 			}
 		}
 
+		if *fK8sModeOffClusterMetering != "" {
+			offClusterMeteringURL := bridge.ValidateFlagIsURL("k8s-mode-off-cluster-metering", *fK8sModeOffClusterMetering)
+			offClusterMeteringURL.Path = "/api"
+			srv.MeteringProxyConfig = &proxy.Config{
+				TLSClientConfig: serviceProxyTLSConfig,
+				HeaderBlacklist: []string{"Cookie", "X-CSRFToken"},
+				Endpoint:        offClusterMeteringURL,
+			}
+		}
+
 	default:
 		bridge.FlagFatalf("k8s-mode", "must be one of: in-cluster, off-cluster")
 	}