refactor!: Properly support contexts

The new AWS SDK uses contexts for all API functions and for loading the
SDK configuration. When chamber was migrated to the SDK, all of them
were set to `context.TODO()`. This commit removes those temporary values
and threads contexts properly throughout the codebase, starting from the
chamber commands.

BREAKING CHANGE: This commit changes the signature of many exported
functions to add a context argument.

Author: bhavanki

cmd/delete.go

@@ -51,7 +51,7 @@ func delete(cmd *cobra.Command, args []string) error {
 				Set("backend", backend),
 		})
 	}
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
@@ -60,5 +60,5 @@ func delete(cmd *cobra.Command, args []string) error {
 		Key:     key,
 	}
 
-	return secretStore.Delete(secretId)
+	return secretStore.Delete(cmd.Context(), secretId)
 }

cmd/env.go

@@ -64,12 +64,12 @@ func exportEnv(cmd *cobra.Command, args []string) ([]string, error) {
 		return nil, fmt.Errorf("Failed to validate service: %w", err)
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return nil, fmt.Errorf("Failed to get secret store: %w", err)
 	}
 
-	rawSecrets, err := secretStore.ListRaw(service)
+	rawSecrets, err := secretStore.ListRaw(cmd.Context(), service)
 	if err != nil {
 		return nil, fmt.Errorf("Failed to list store contents: %w", err)
 	}

cmd/exec.go

@@ -92,7 +92,7 @@ func execRun(cmd *cobra.Command, args []string) error {
 		}
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
@@ -108,7 +108,7 @@ func execRun(cmd *cobra.Command, args []string) error {
 		}
 		var err error
 		env = environ.Environ(os.Environ())
-		err = env.LoadStrict(secretStore, strictValue, pristine, services...)
+		err = env.LoadStrict(cmd.Context(), secretStore, strictValue, pristine, services...)
 		if err != nil {
 			return err
 		}
@@ -120,7 +120,7 @@ func execRun(cmd *cobra.Command, args []string) error {
 			collisions := make([]string, 0)
 			var err error
 			// TODO: these interfaces should look the same as Strict*, so move pristine in there
-			err = env.Load(secretStore, service, &collisions)
+			err = env.Load(cmd.Context(), secretStore, service, &collisions)
 			if err != nil {
 				return fmt.Errorf("Failed to list store contents: %w", err)
 			}

cmd/export.go

@@ -53,7 +53,7 @@ func runExport(cmd *cobra.Command, args []string) error {
 		})
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return err
 	}
@@ -64,7 +64,7 @@ func runExport(cmd *cobra.Command, args []string) error {
 			return fmt.Errorf("Failed to validate service %s: %w", service, err)
 		}
 
-		rawSecrets, err := secretStore.ListRaw(service)
+		rawSecrets, err := secretStore.ListRaw(cmd.Context(), service)
 		if err != nil {
 			return fmt.Errorf("Failed to list store contents for service %s: %w", service, err)
 		}

cmd/find.go

@@ -39,18 +39,18 @@ func find(cmd *cobra.Command, args []string) error {
 		includeSecrets = true
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
-	services, err := secretStore.ListServices(blankService, includeSecrets)
+	services, err := secretStore.ListServices(cmd.Context(), blankService, includeSecrets)
 	if err != nil {
 		return fmt.Errorf("Failed to list store contents: %w", err)
 	}
 
 	if byValue {
 		for _, service := range services {
-			allSecrets, err := secretStore.List(service, true)
+			allSecrets, err := secretStore.List(cmd.Context(), service, true)
 			if err == nil {
 				matches = append(matches, findValueMatch(allSecrets, findSecret)...)
 			}

cmd/history.go

@@ -47,7 +47,7 @@ func history(cmd *cobra.Command, args []string) error {
 		})
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
@@ -56,7 +56,7 @@ func history(cmd *cobra.Command, args []string) error {
 		Key:     key,
 	}
 
-	events, err := secretStore.History(secretId)
+	events, err := secretStore.History(cmd.Context(), secretId)
 	if err != nil {
 		return fmt.Errorf("Failed to get history: %w", err)
 	}

cmd/import.go

@@ -65,7 +65,7 @@ func importRun(cmd *cobra.Command, args []string) error {
 		})
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
@@ -78,7 +78,7 @@ func importRun(cmd *cobra.Command, args []string) error {
 			Service: service,
 			Key:     key,
 		}
-		if err := secretStore.Write(secretId, value); err != nil {
+		if err := secretStore.Write(cmd.Context(), secretId, value); err != nil {
 			return fmt.Errorf("Failed to write secret: %w", err)
 		}
 	}

cmd/list-services.go

@@ -34,11 +34,11 @@ func listServices(cmd *cobra.Command, args []string) error {
 		service = utils.NormalizeService(args[0])
 
 	}
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
-	secrets, err := secretStore.ListServices(service, includeSecretName)
+	secrets, err := secretStore.ListServices(cmd.Context(), service, includeSecretName)
 	if err != nil {
 		return fmt.Errorf("Failed to list store contents: %w", err)
 	}

cmd/list.go

@@ -54,11 +54,11 @@ func list(cmd *cobra.Command, args []string) error {
 		})
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
-	secrets, err := secretStore.List(service, withValues)
+	secrets, err := secretStore.List(cmd.Context(), service, withValues)
 	if err != nil {
 		return fmt.Errorf("Failed to list store contents: %w", err)
 	}

cmd/read.go

@@ -54,7 +54,7 @@ func read(cmd *cobra.Command, args []string) error {
 		})
 	}
 
-	secretStore, err := getSecretStore()
+	secretStore, err := getSecretStore(cmd.Context())
 	if err != nil {
 		return fmt.Errorf("Failed to get secret store: %w", err)
 	}
@@ -64,7 +64,7 @@ func read(cmd *cobra.Command, args []string) error {
 		Key:     key,
 	}
 
-	secret, err := secretStore.Read(secretId, version)
+	secret, err := secretStore.Read(cmd.Context(), secretId, version)
 	if err != nil {
 		return fmt.Errorf("Failed to read: %w", err)
 	}