configure bitwarden

Author: eumel8

main.go

@@ -2,15 +2,14 @@ package main
 
 import (
 	"bytes"
-	"crypto/tls"
 	"context"
+	"crypto/tls"
 	"encoding/json"
 	"flag"
 	"fmt"
 	"image/color"
 	"math"
 	"net/http"
-	"net/http/httputil"
 	"os"
 	"os/exec"
 	"strconv"
@@ -56,10 +55,10 @@ type basicAuthTransport struct {
 
 // Struct to hold Bitwarden login fields
 type BitwardenItem struct {
-        Login struct {
-                Username string `json:"username"`
-                Password string `json:"password"`
-        } `json:"login"`
+	Login struct {
+		Username string `json:"username"`
+		Password string `json:"password"`
+	} `json:"login"`
 }
 
 func (bat *basicAuthTransport) RoundTrip(req *http.Request) (*http.Response, error) {
@@ -95,23 +94,23 @@ func (bat *basicAuthTransport) transport() http.RoundTripper {
 
 // Get BW_SESSION from env
 func getSessionToken() string {
-        return os.Getenv("BW_SESSION")
+	return os.Getenv("BW_SESSION")
 }
 
 // Run Bitwarden CLI to get the item JSON
 func getBitwardenItemJSON(itemName string) ([]byte, error) {
-        cmd := exec.Command("bw", "get", "item", itemName)
-        cmd.Env = append(os.Environ(), "BW_SESSION="+getSessionToken())
+	cmd := exec.Command("bw", "get", "item", itemName)
+	cmd.Env = append(os.Environ(), "BW_SESSION="+getSessionToken())
 
-        var out bytes.Buffer
-        cmd.Stdout = &out
+	var out bytes.Buffer
+	cmd.Stdout = &out
 
-        err := cmd.Run()
-        if err != nil {
-                return nil, err
-        }
+	err := cmd.Run()
+	if err != nil {
+		return nil, err
+	}
 
-        return out.Bytes(), nil
+	return out.Bytes(), nil
 }
 
 func loadMetricsFromFile(path string) ([]Metric, error) {
@@ -142,28 +141,29 @@ func GetConfig() (*Config, error) {
 	}, nil
 }
 
-func (c *Config) getMetricValue(metric string) (int, error) {
+func (c *Config) getMetricValue(metric string, bitwarden string) (int, error) {
 
 	prometheus := c.PrometheusURL
-        // doing bitwarden stuff here to get prometheus credentials
-        itemName := "Prometheus Agent RemoteWrite"
-        jsonData, err := getBitwardenItemJSON(itemName)
-        if err != nil {
-                fmt.Printf("Failed to get item from Bitwarden: %v\n", err)
-        }
-
-        var item BitwardenItem
-        err = json.Unmarshal(jsonData, &item)
-        if err != nil {
-                fmt.Printf("Failed to parse Bitwarden JSON: %v\n", err)
-        }
-
-        username := item.Login.Username
-        password := item.Login.Password
-
-        if os.Getenv("PROMETHEUS_URL") != "" {
-                prometheus = os.Getenv("PROMETHEUS_URL")
-        }
+	username := os.Getenv("PROM_USER")
+	password := os.Getenv("PROM_PASS")
+
+	if bitwarden != "" {
+		// doing bitwarden stuff here to get prometheus credentials
+		itemName := "Prometheus Agent RemoteWrite"
+		jsonData, err := getBitwardenItemJSON(itemName)
+		if err != nil {
+			fmt.Printf("Failed to get item from Bitwarden: %v\n", err)
+		}
+
+		var item BitwardenItem
+		err = json.Unmarshal(jsonData, &item)
+		if err != nil {
+			fmt.Printf("Failed to parse Bitwarden JSON: %v\n", err)
+		}
+
+		username = item.Login.Username
+		password = item.Login.Password
+	}
 
 	customClient := &http.Client{
 		Transport: &http.Transport{
@@ -226,6 +226,7 @@ func main() {
 
 	verbose := flag.Bool("v", false, "enable verbose logging")
 	debug := flag.Bool("d", false, "enable debugging")
+	bitwarden := flag.Bool("bwd", false, "enable Bitwarden password store")
 	flag.Parse()
 
 	config, err := GetConfig()
@@ -256,7 +257,7 @@ func main() {
 			)
 
 			for _, metric := range config.Metrics {
-				val, err := config.getMetricValue(metric.Name)
+				val, err := config.getMetricValue(metric.Name, *bitwarden)
 
 				var icon *canvas.Text
 				var statusText string