Add option to use Metabase API key (#8)

Author: saligrama

README.md

@@ -13,13 +13,15 @@ docker build -t formal-metabase-sync-worker .
 Then run the docker image:
 
 ```bash
-docker run -e METABASE_HOSTNAME=""  -e METABASE_USERNAME="" -e METABASE_PASSWORD="" -e METABASE_VERSION="" -e FORMAL_API_KEY="" -e FORMAL_APP_ID="" -e VERIFY_TLS="" formal-metabase-sync-worker 
+docker run -e METABASE_HOSTNAME="" -e METABASE_USE_API_KEY="" -e METABASE_API_KEY=""  -e METABASE_USERNAME="" -e METABASE_PASSWORD="" -e METABASE_VERSION="" -e FORMAL_API_KEY="" -e FORMAL_APP_ID="" -e VERIFY_TLS="" formal-metabase-sync-worker 
 ```
 
 ## Environment Variables
+- ```METABASE_USE_API_KEY``` (optional): Whether or not to use the Metabase API key instead of username/password auth. Set to `true` or `false`. Default value is `false`.
+- ```METABASE_API_KEY``` (optional): The API key for the metabase instance; required if not providing username/password auth.
 - ```METABASE_HOSTNAME```: The hostname of the metabase instance 
-- ```METABASE_USERNAME```: The username of the metabase instance
-- ```METABASE_PASSWORD```: The password of the metabase instance
+- ```METABASE_USERNAME```: (optional) The username of the metabase instance; required if not providing API key.
+- ```METABASE_PASSWORD```: (optional) The password of the metabase instance; required if not providing API key.
 - ```METABASE_VERSION```: The version of the metabase instance (e.g.: 0.35.4)
 - ```FORMAL_API_KEY```: The API key of the formal instance
 - ```FORMAL_APP_ID```: The app id of the Formal Metabase integration

main.go

@@ -9,7 +9,14 @@ import (
 )
 
 func main() {
+	metabaseUseApiKey, err := strconv.ParseBool(os.Getenv("METABASE_USE_API_KEY"))
+	if err != nil {
+		metabaseUseApiKey = false
+	}
+
 	metabaseIntegration := MetabaseIntegration{
+		UseAPIKey:        metabaseUseApiKey,
+		MetabaseAPIKey:   os.Getenv("METABASE_API_KEY"),
 		MetabaseHostname: os.Getenv("METABASE_HOSTNAME"),
 		MetabaseUsername: os.Getenv("METABASE_USERNAME"),
 		MetabasePwd:      os.Getenv("METABASE_PASSWORD"),

metabase_api.go

@@ -30,7 +30,7 @@ type MetabaseUser struct {
 
 const METABASE_THRESHOLD_VERSION = "0.40.0"
 
-func GetMetabaseRoles(hostname, metabaseVersion, sessionKey string, verifyTLS bool) (map[string]MetabaseUser, error) {
+func GetMetabaseRoles(hostname string, metabaseVersion string, apiKey string, sessionKey string, useAPIKey bool, verifyTLS bool) (map[string]MetabaseUser, error) {
 	baseUrl := "https://" + hostname + "/api/user"
 
 	roles := map[string]MetabaseUser{}
@@ -44,7 +44,11 @@ func GetMetabaseRoles(hostname, metabaseVersion, sessionKey string, verifyTLS bo
 				return nil, err
 			}
 			req.Header.Set("Content-Type", "application/json")
-			req.Header.Set("X-Metabase-Session", sessionKey)
+			if useAPIKey {
+				req.Header.Set("X-API-Key", apiKey)
+			} else {
+				req.Header.Set("X-Metabase-Session", sessionKey)
+			}
 
 			// Send Request
 			client := &http.Client{
@@ -91,7 +95,11 @@ func GetMetabaseRoles(hostname, metabaseVersion, sessionKey string, verifyTLS bo
 			return nil, err
 		}
 		req.Header.Set("Content-Type", "application/json")
-		req.Header.Set("X-Metabase-Session", sessionKey)
+		if useAPIKey {
+			req.Header.Set("X-API-Key", apiKey)
+		} else {
+			req.Header.Set("X-Metabase-Session", sessionKey)
+		}
 
 		// Send Request
 		client := &http.Client{

workflow.go

@@ -7,6 +7,8 @@ import (
 )
 
 type MetabaseIntegration struct {
+	UseAPIKey        bool
+	MetabaseAPIKey   string
 	MetabaseHostname string
 	MetabaseUsername string
 	MetabasePwd      string
@@ -16,12 +18,25 @@ type MetabaseIntegration struct {
 func MetabaseWorkflow(metabaseIntegration MetabaseIntegration, apiKey, integrationID string, verifyTLS bool) error {
 	client := New(apiKey)
 
-	sessionKey, err := RefreshMetabaseSessionKey(metabaseIntegration, verifyTLS)
-	if err != nil {
-		return err
+	sessionKey := ""
+
+	if !metabaseIntegration.UseAPIKey {
+		key, err := RefreshMetabaseSessionKey(metabaseIntegration, verifyTLS)
+		if err != nil {
+			return err
+		}
+
+		sessionKey = key
 	}
 
-	metabaseRoles, err := GetMetabaseRoles(metabaseIntegration.MetabaseHostname, metabaseIntegration.Version, sessionKey, verifyTLS)
+	metabaseRoles, err := GetMetabaseRoles(
+		metabaseIntegration.MetabaseHostname,
+		metabaseIntegration.Version,
+		metabaseIntegration.MetabaseAPIKey,
+		sessionKey,
+		metabaseIntegration.UseAPIKey,
+		verifyTLS,
+	)
 	if err != nil {
 		return err
 	}